ZeroTierOne/osdep/ManagedRoute.cpp

/*
 * ZeroTier One - Network Virtualization Everywhere
 * Copyright (C) 2011-2016  ZeroTier, Inc.  https://www.zerotier.com/
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "../node/Constants.hpp"

#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#ifdef __WINDOWS__
#include <WinSock2.h>
#include <Windows.h>
#include <netioapi.h>
#include <IPHlpApi.h>
#endif

#ifdef __UNIX_LIKE__
#include <unistd.h>
#include <sys/param.h>
#include <sys/sysctl.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <net/route.h>
#include <net/if.h>
#include <net/if_dl.h>
#include <ifaddrs.h>
#endif

#include <vector>
#include <algorithm>
#include <utility>

#include "ManagedRoute.hpp"

#define ZT_BSD_ROUTE_CMD "/sbin/route"
#define ZT_LINUX_IP_COMMAND "/sbin/ip"

namespace ZeroTier {

namespace {

// Fork a target into two more specific targets e.g. 0.0.0.0/0 -> 0.0.0.0/1, 128.0.0.0/1
// If the target is already maximally-specific, 'right' will be unchanged and 'left' will be 't'
static void _forkTarget(const InetAddress &t,InetAddress &left,InetAddress &right)
{
	const unsigned int bits = t.netmaskBits() + 1;
	left = t;
	if ((t.ss_family == AF_INET)&&(bits <= 32)) {
		left.setPort(bits);
		right = t;
		reinterpret_cast<struct sockaddr_in *>(&right)->sin_addr.s_addr ^= Utils::hton((uint32_t)(1 << (32 - bits)));
		right.setPort(bits);
	} else if ((t.ss_family == AF_INET6)&&(bits <= 128)) {
		left.setPort(bits);
		right = t;
		uint8_t *b = reinterpret_cast<uint8_t *>(reinterpret_cast<struct sockaddr_in6 *>(&right)->sin6_addr.s6_addr);
		b[bits / 8] ^= 1 << (8 - (bits % 8));
		right.setPort(bits);
	}
}

#ifdef __BSD__ // ------------------------------------------------------------
#define ZT_ROUTING_SUPPORT_FOUND 1

struct _RTE
{
	InetAddress target;
	InetAddress via;
	char device[128];
	int metric;
	bool ifscope;
};

static std::vector<_RTE> _getRTEs(const InetAddress &target,bool contains)
{
	std::vector<_RTE> rtes;
	int mib[6];
	size_t needed;

	mib[0] = CTL_NET;
	mib[1] = PF_ROUTE;
	mib[2] = 0;
	mib[3] = 0;
	mib[4] = NET_RT_DUMP;
	mib[5] = 0;
	if (!sysctl(mib,6,NULL,&needed,NULL,0)) {
		if (needed <= 0)
			return rtes;

		char *buf = (char *)::malloc(needed);
		if (buf) {
			if (!sysctl(mib,6,buf,&needed,NULL,0)) {
		    struct rt_msghdr *rtm;
				for(char *next=buf,*end=buf+needed;next<end;) {
					rtm = (struct rt_msghdr *)next;
					char *saptr = (char *)(rtm + 1);
					char *saend = next + rtm->rtm_msglen;

					InetAddress sa_t,sa_v;
					int deviceIndex = -9999;

					if (((rtm->rtm_flags & RTF_LLINFO) == 0)&&((rtm->rtm_flags & RTF_HOST) == 0)&&((rtm->rtm_flags & RTF_UP) != 0)&&((rtm->rtm_flags & RTF_MULTICAST) == 0)) {
						int which = 0;
						while (saptr < saend) {
							struct sockaddr *sa = (struct sockaddr *)saptr;
							unsigned int salen = sa->sa_len;
							if (!salen)
								break;

							// Skip missing fields in rtm_addrs bit field
							while ((rtm->rtm_addrs & 1) == 0) {
								rtm->rtm_addrs >>= 1;
								++which;
								if (which > 6)
									break;
							}
							if (which > 6)
								break;

							rtm->rtm_addrs >>= 1;
							switch(which++) {
								case 0:
									//printf("RTA_DST\n");
									if (sa->sa_family == AF_INET6) {
										struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)sa;
										if ((sin6->sin6_addr.s6_addr[0] == 0xfe)&&((sin6->sin6_addr.s6_addr[1] & 0xc0) == 0x80)) {
											// BSD uses this fucking strange in-band signaling method to encode device scope IDs for IPv6 addresses... probably a holdover from very early versions of the spec.
											unsigned int interfaceIndex = ((((unsigned int)sin6->sin6_addr.s6_addr[2]) << 8) & 0xff) | (((unsigned int)sin6->sin6_addr.s6_addr[3]) & 0xff);
											sin6->sin6_addr.s6_addr[2] = 0;
											sin6->sin6_addr.s6_addr[3] = 0;
											if (!sin6->sin6_scope_id)
												sin6->sin6_scope_id = interfaceIndex;
										}
									}
									sa_t = *sa;
									break;
								case 1:
									//printf("RTA_GATEWAY\n");
									switch(sa->sa_family) {
										case AF_LINK:
											deviceIndex = (int)((const struct sockaddr_dl *)sa)->sdl_index;
											break;
										case AF_INET:
										case AF_INET6:
											sa_v = *sa;
											break;
									}
									break;
								case 2: {
									//printf("RTA_NETMASK\n");
									if (sa_t.ss_family == AF_INET6) {
										salen = sizeof(struct sockaddr_in6);
										unsigned int bits = 0;
										for(int i=0;i<16;++i) {
											unsigned char c = (unsigned char)((const struct sockaddr_in6 *)sa)->sin6_addr.s6_addr[i];
											if (c == 0xff)
												bits += 8;
											else break;
										}
										sa_t.setPort(bits);
									} else if (sa_t.ss_family == AF_INET) {
										salen = sizeof(struct sockaddr_in);
										sa_t.setPort((unsigned int)Utils::countBits((uint32_t)((const struct sockaddr_in *)sa)->sin_addr.s_addr));
									}
								}	break;
								/*
								case 3:
									//printf("RTA_GENMASK\n");
									break;
								case 4:
									//printf("RTA_IFP\n");
									break;
								case 5:
									//printf("RTA_IFA\n");
									break;
								case 6:
									//printf("RTA_AUTHOR\n");
									break;
								*/
							}

							saptr += salen;
						}

						if (((contains)&&(sa_t.containsAddress(target)))||(sa_t == target)) {
							rtes.push_back(_RTE());
							rtes.back().target = sa_t;
							rtes.back().via = sa_v;
							if (deviceIndex >= 0) {
								if_indextoname(deviceIndex,rtes.back().device);
							} else {
								rtes.back().device[0] = (char)0;
							}
							rtes.back().metric = ((int)rtm->rtm_rmx.rmx_hopcount < 0) ? 0 : (int)rtm->rtm_rmx.rmx_hopcount;
						}
					}

					next = saend;
				}
			}

			::free(buf);
		}
	}

	return rtes;
}

static void _routeCmd(const char *op,const InetAddress &target,const InetAddress &via,const char *ifscope,const char *localInterface)
{
	long p = (long)fork();
	if (p > 0) {
		int exitcode = -1;
		::waitpid(p,&exitcode,0);
	} else if (p == 0) {
		//::close(STDOUT_FILENO);
		//::close(STDERR_FILENO);
		if (via) {
			if ((ifscope)&&(ifscope[0])) {
				::execl(ZT_BSD_ROUTE_CMD,ZT_BSD_ROUTE_CMD,op,"-ifscope",ifscope,((target.ss_family == AF_INET6) ? "-inet6" : "-inet"),target.toString().c_str(),via.toIpString().c_str(),(const char *)0);
			} else {
				::execl(ZT_BSD_ROUTE_CMD,ZT_BSD_ROUTE_CMD,op,((target.ss_family == AF_INET6) ? "-inet6" : "-inet"),target.toString().c_str(),via.toIpString().c_str(),(const char *)0);
			}
		} else if ((localInterface)&&(localInterface[0])) {
			if ((ifscope)&&(ifscope[0])) {
				::execl(ZT_BSD_ROUTE_CMD,ZT_BSD_ROUTE_CMD,op,"-ifscope",ifscope,((target.ss_family == AF_INET6) ? "-inet6" : "-inet"),target.toString().c_str(),"-interface",localInterface,(const char *)0);
			} else {
				::execl(ZT_BSD_ROUTE_CMD,ZT_BSD_ROUTE_CMD,op,((target.ss_family == AF_INET6) ? "-inet6" : "-inet"),target.toString().c_str(),"-interface",localInterface,(const char *)0);
			}
		}
		::_exit(-1);
	}
}

#endif // __BSD__ ------------------------------------------------------------

#ifdef __LINUX__ // ----------------------------------------------------------
#define ZT_ROUTING_SUPPORT_FOUND 1

#endif // __LINUX__ ----------------------------------------------------------

#ifdef __WINDOWS__ // --------------------------------------------------------
#define ZT_ROUTING_SUPPORT_FOUND 1

#endif // __WINDOWS__ --------------------------------------------------------

} // anonymous namespace

bool ManagedRoute::sync()
{
	if (this->target.isDefaultRoute()) {
		/* In ZeroTier we use a forked-route trick to override the default
		 * with a more specific one while leaving the original system route
		 * intact. We also create a shadow more specific route to the
		 * original gateway that is device-bound so that ZeroTier's device
		 * bound ports go via the physical Internet link. This has to be
		 * done *slightly* differently on different platforms. */

		InetAddress leftt,rightt;
		_forkTarget(this->target,leftt,rightt);

#ifdef __BSD__ // ------------------------------------------------------------

		InetAddress systemVia;
		char systemDevice[128];
		int systemMetric = 9999999;
		systemDevice[0] = (char)0;

		std::vector<_RTE> rtes(_getRTEs(this->target,false));
		for(std::vector<_RTE>::iterator r(rtes.begin());r!=rtes.end();++r) {
			if (r->via) {
				if ((!systemVia)||(r->metric < systemMetric)) {
					systemVia = r->via;
					Utils::scopy(systemDevice,sizeof(systemDevice),r->device);
					systemMetric = r->metric;
				}
			}
		}

		if (!systemDevice[0]) {
			rtes = _getRTEs(systemVia,true);
			for(std::vector<_RTE>::iterator r(rtes.begin());r!=rtes.end();++r) {
				if (r->device[0])
					Utils::scopy(systemDevice,sizeof(systemDevice),r->device);
			}
		}

		if ((!systemVia)||(!systemDevice[0]))
			return false;

		_routeCmd("add",leftt,systemVia,systemDevice,(const char *)0);
		_routeCmd("change",leftt,systemVia,systemDevice,(const char *)0);
		_routeCmd("add",rightt,systemVia,systemDevice,(const char *)0);
		_routeCmd("change",rightt,systemVia,systemDevice,(const char *)0);

		if (this->via) {
			_routeCmd("add",leftt,this->via,(const char *)0,(const char *)0);
			_routeCmd("change",leftt,this->via,(const char *)0,(const char *)0);
			_routeCmd("add",rightt,this->via,(const char *)0,(const char *)0);
			_routeCmd("change",rightt,this->via,(const char *)0,(const char *)0);
		} else if ((this->device)&&(this->device[0])) {
			_routeCmd("add",leftt,this->via,(const char *)0,this->device);
			_routeCmd("change",leftt,this->via,(const char *)0,this->device);
			_routeCmd("add",rightt,this->via,(const char *)0,this->device);
			_routeCmd("change",rightt,this->via,(const char *)0,this->device);
		}

#endif // __BSD__ ------------------------------------------------------------

#ifdef __LINUX__ // ----------------------------------------------------------

#endif // __LINUX__ ----------------------------------------------------------

#ifdef __WINDOWS__ // --------------------------------------------------------

#endif // __WINDOWS__ --------------------------------------------------------

	} else {

		// TODO

#ifdef __BSD__ // ------------------------------------------------------------

#endif // __BSD__ ------------------------------------------------------------

#ifdef __LINUX__ // ----------------------------------------------------------

#endif // __LINUX__ ----------------------------------------------------------

#ifdef __WINDOWS__ // --------------------------------------------------------

#endif // __WINDOWS__ --------------------------------------------------------

	}

	return true;
}

void ManagedRoute::remove()
{
	if (!this->applied)
		return;

	if (this->target.isDefaultRoute()) {
		/* In ZeroTier we use a forked-route trick to override the default
		 * with a more specific one while leaving the original system route
		 * intact. We also create a shadow more specific route to the
		 * original gateway that is device-bound so that ZeroTier's device
		 * bound ports go via the physical Internet link. This has to be
		 * done *slightly* differently on different platforms. */

		InetAddress leftt,rightt;
		_forkTarget(this->target,leftt,rightt);

#ifdef __BSD__ // ------------------------------------------------------------

		InetAddress systemVia;
		char systemDevice[128];
		int systemMetric = 9999999;
		systemDevice[0] = (char)0;

		std::vector<_RTE> rtes(_getRTEs(this->target,false));
		for(std::vector<_RTE>::iterator r(rtes.begin());r!=rtes.end();++r) {
			if (r->via) {
				if ((!systemVia)||(r->metric < systemMetric)) {
					systemVia = r->via;
					Utils::scopy(systemDevice,sizeof(systemDevice),r->device);
					systemMetric = r->metric;
				}
			}
		}

		if (!systemDevice[0]) {
			rtes = _getRTEs(systemVia,true);
			for(std::vector<_RTE>::iterator r(rtes.begin());r!=rtes.end();++r) {
				if (r->device[0])
					Utils::scopy(systemDevice,sizeof(systemDevice),r->device);
			}
		}

		if ((!systemVia)||(!systemDevice[0]))
			return false;

		_routeCmd("delete",leftt,systemVia,systemDevice,(const char *)0);
		_routeCmd("delete",rightt,systemVia,systemDevice,(const char *)0);

		if (this->via) {
			_routeCmd("delete",leftt,this->via,(const char *)0,(const char *)0);
			_routeCmd("delete",rightt,this->via,(const char *)0,(const char *)0);
		} else if ((this->device)&&(this->device[0])) {
			_routeCmd("delete",leftt,this->via,(const char *)0,this->device);
			_routeCmd("delete",rightt,this->via,(const char *)0,this->device);
		}

#endif // __BSD__ ------------------------------------------------------------

#ifdef __LINUX__ // ----------------------------------------------------------

#endif // __LINUX__ ----------------------------------------------------------

#ifdef __WINDOWS__ // --------------------------------------------------------

#endif // __WINDOWS__ --------------------------------------------------------

	} else {

		// TODO

#ifdef __BSD__ // ------------------------------------------------------------

#endif // __BSD__ ------------------------------------------------------------

#ifdef __LINUX__ // ----------------------------------------------------------

#endif // __LINUX__ ----------------------------------------------------------

#ifdef __WINDOWS__ // --------------------------------------------------------

#endif // __WINDOWS__ --------------------------------------------------------

	}
}

} // namespace ZeroTier

#ifndef ZT_ROUTING_SUPPORT_FOUND
#error ManagedRoute.cpp has no support for managing routes on this platform! You'll need to check and see if one of the existing ones will work and make sure proper defines are set, or write one. Please do a Github pull request if you do this for a new OS!
#endif

/*
int main(int argc,char **argv)
{
	ZeroTier::ManagedRoute t;
	t.set(ZeroTier::InetAddress("0.0.0.0/0"),ZeroTier::InetAddress("10.6.6.112"),"zt2");
	sleep(10000);
}
*/