EthernetTap.cpp 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904
  1. /*
  2. * ZeroTier One - Global Peer to Peer Ethernet
  3. * Copyright (C) 2012-2013 ZeroTier Networks LLC
  4. *
  5. * This program is free software: you can redistribute it and/or modify
  6. * it under the terms of the GNU General Public License as published by
  7. * the Free Software Foundation, either version 3 of the License, or
  8. * (at your option) any later version.
  9. *
  10. * This program is distributed in the hope that it will be useful,
  11. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13. * GNU General Public License for more details.
  14. *
  15. * You should have received a copy of the GNU General Public License
  16. * along with this program. If not, see <http://www.gnu.org/licenses/>.
  17. *
  18. * --
  19. *
  20. * ZeroTier may be used and distributed under the terms of the GPLv3, which
  21. * are available at: http://www.gnu.org/licenses/gpl-3.0.html
  22. *
  23. * If you would like to embed ZeroTier into a commercial application or
  24. * redistribute it in a modified binary form, please contact ZeroTier Networks
  25. * LLC. Start here: http://www.zerotier.com/
  26. */
  27. #include <string>
  28. #include <map>
  29. #include <set>
  30. #include <algorithm>
  31. #include "Constants.hpp"
  32. #include "EthernetTap.hpp"
  33. #include "Logger.hpp"
  34. #include "RuntimeEnvironment.hpp"
  35. #include "Utils.hpp"
  36. #include "Mutex.hpp"
  37. // ff:ff:ff:ff:ff:ff with no ADI
  38. static const ZeroTier::MulticastGroup _blindWildcardMulticastGroup(ZeroTier::MAC(0xff),0);
  39. //
  40. // TAP implementation for *nix OSes, with some specialization for different flavors
  41. //
  42. #ifdef __UNIX_LIKE__ /////////////////////////////////////////////////////////
  43. #include <stdint.h>
  44. #include <stdio.h>
  45. #include <stdlib.h>
  46. #include <string.h>
  47. #include <unistd.h>
  48. #include <signal.h>
  49. #include <fcntl.h>
  50. #include <errno.h>
  51. #include <sys/types.h>
  52. #include <sys/stat.h>
  53. #include <sys/ioctl.h>
  54. #include <sys/wait.h>
  55. #include <sys/select.h>
  56. #include <netinet/in.h>
  57. #include <net/if_arp.h>
  58. #include <arpa/inet.h>
  59. // Command identifiers used with command finder static (on various *nixes)
  60. #define ZT_UNIX_IP_COMMAND 1
  61. #define ZT_UNIX_IFCONFIG_COMMAND 2
  62. #define ZT_MAC_KEXTLOAD_COMMAND 3
  63. #define ZT_MAC_IPCONFIG_COMMAND 4
  64. // Finds external commands on startup
  65. class _CommandFinder
  66. {
  67. public:
  68. _CommandFinder()
  69. {
  70. _findCmd(ZT_UNIX_IFCONFIG_COMMAND,"ifconfig");
  71. #ifdef __LINUX__
  72. _findCmd(ZT_UNIX_IP_COMMAND,"ip");
  73. #endif
  74. #ifdef __APPLE__
  75. _findCmd(ZT_MAC_KEXTLOAD_COMMAND,"kextload");
  76. _findCmd(ZT_MAC_IPCONFIG_COMMAND,"ipconfig");
  77. #endif
  78. }
  79. // returns NULL if command was not found
  80. inline const char *operator[](int id) const
  81. throw()
  82. {
  83. std::map<int,std::string>::const_iterator c(_paths.find(id));
  84. if (c == _paths.end())
  85. return (const char *)0;
  86. return c->second.c_str();
  87. }
  88. private:
  89. inline void _findCmd(int id,const char *name)
  90. {
  91. char tmp[4096];
  92. sprintf(tmp,"/sbin/%s",name);
  93. if (ZeroTier::Utils::fileExists(tmp)) {
  94. _paths[id] = tmp;
  95. return;
  96. }
  97. sprintf(tmp,"/usr/sbin/%s",name);
  98. if (ZeroTier::Utils::fileExists(tmp)) {
  99. _paths[id] = tmp;
  100. return;
  101. }
  102. sprintf(tmp,"/bin/%s",name);
  103. if (ZeroTier::Utils::fileExists(tmp)) {
  104. _paths[id] = tmp;
  105. return;
  106. }
  107. sprintf(tmp,"/usr/bin/%s",name);
  108. if (ZeroTier::Utils::fileExists(tmp)) {
  109. _paths[id] = tmp;
  110. return;
  111. }
  112. }
  113. std::map<int,std::string> _paths;
  114. };
  115. static const _CommandFinder UNIX_COMMANDS;
  116. #ifdef __LINUX__
  117. #include <linux/if.h>
  118. #include <linux/if_tun.h>
  119. #include <linux/if_addr.h>
  120. #include <linux/if_ether.h>
  121. #endif // __LINUX__
  122. #ifdef __APPLE__
  123. #include <sys/uio.h>
  124. #include <sys/param.h>
  125. #include <sys/sysctl.h>
  126. #include <net/route.h>
  127. #include <net/if_dl.h>
  128. #include <ifaddrs.h>
  129. #endif // __APPLE__
  130. namespace ZeroTier {
  131. // Only permit one tap to be opened concurrently across the entire process
  132. static Mutex __tapCreateLock;
  133. #ifdef __LINUX__
  134. EthernetTap::EthernetTap(
  135. const RuntimeEnvironment *renv,
  136. const char *tag,
  137. const MAC &mac,
  138. unsigned int mtu,
  139. void (*handler)(void *,const MAC &,const MAC &,unsigned int,const Buffer<4096> &),
  140. void *arg)
  141. throw(std::runtime_error) :
  142. _mac(mac),
  143. _mtu(mtu),
  144. _r(renv),
  145. _handler(handler),
  146. _arg(arg),
  147. _fd(0)
  148. {
  149. char procpath[128];
  150. Mutex::Lock _l(__tapCreateLock); // create only one tap at a time, globally
  151. if (mtu > 4096)
  152. throw std::runtime_error("max tap MTU is 4096");
  153. _fd = ::open("/dev/net/tun",O_RDWR);
  154. if (_fd <= 0)
  155. throw std::runtime_error(std::string("could not open TUN/TAP device: ") + strerror(errno));
  156. struct ifreq ifr;
  157. memset(&ifr,0,sizeof(ifr));
  158. { // pick an unused device name
  159. int devno = 0;
  160. struct stat sbuf;
  161. do {
  162. sprintf(ifr.ifr_name,"zt%d",devno++);
  163. sprintf(procpath,"/proc/sys/net/ipv4/conf/%s",ifr.ifr_name);
  164. } while (stat(procpath,&sbuf) == 0);
  165. }
  166. ifr.ifr_flags = IFF_TAP | IFF_NO_PI;
  167. if (ioctl(_fd,TUNSETIFF,(void *)&ifr) < 0) {
  168. ::close(_fd);
  169. throw std::runtime_error("unable to configure TUN/TAP device for TAP operation");
  170. }
  171. strcpy(_dev,ifr.ifr_name);
  172. ioctl(_fd,TUNSETPERSIST,0); // valgrind may generate a false alarm here
  173. // Open an arbitrary socket to talk to netlink
  174. int sock = socket(AF_INET,SOCK_DGRAM,0);
  175. if (sock <= 0) {
  176. ::close(_fd);
  177. throw std::runtime_error("unable to open netlink socket");
  178. }
  179. // Set MAC address
  180. ifr.ifr_ifru.ifru_hwaddr.sa_family = ARPHRD_ETHER;
  181. memcpy(ifr.ifr_ifru.ifru_hwaddr.sa_data,mac.data,6);
  182. if (ioctl(sock,SIOCSIFHWADDR,(void *)&ifr) < 0) {
  183. ::close(_fd);
  184. ::close(sock);
  185. throw std::runtime_error("unable to configure TAP hardware (MAC) address");
  186. return;
  187. }
  188. // Set MTU
  189. ifr.ifr_ifru.ifru_mtu = (int)mtu;
  190. if (ioctl(sock,SIOCSIFMTU,(void *)&ifr) < 0) {
  191. ::close(_fd);
  192. ::close(sock);
  193. throw std::runtime_error("unable to configure TAP MTU");
  194. }
  195. if (fcntl(_fd,F_SETFL,fcntl(_fd,F_GETFL) & ~O_NONBLOCK) == -1) {
  196. ::close(_fd);
  197. throw std::runtime_error("unable to set flags on file descriptor for TAP device");
  198. }
  199. /* Bring interface up */
  200. if (ioctl(sock,SIOCGIFFLAGS,(void *)&ifr) < 0) {
  201. ::close(_fd);
  202. ::close(sock);
  203. throw std::runtime_error("unable to get TAP interface flags");
  204. }
  205. ifr.ifr_flags |= IFF_UP;
  206. if (ioctl(sock,SIOCSIFFLAGS,(void *)&ifr) < 0) {
  207. ::close(_fd);
  208. ::close(sock);
  209. throw std::runtime_error("unable to set TAP interface flags");
  210. }
  211. ::close(sock);
  212. ::pipe(_shutdownSignalPipe);
  213. TRACE("tap %s created",_dev);
  214. _thread = Thread::start(this);
  215. }
  216. #endif // __LINUX__
  217. #ifdef __APPLE__
  218. EthernetTap::EthernetTap(
  219. const RuntimeEnvironment *renv,
  220. const char *tag,
  221. const MAC &mac,
  222. unsigned int mtu,
  223. void (*handler)(void *,const MAC &,const MAC &,unsigned int,const Buffer<4096> &),
  224. void *arg)
  225. throw(std::runtime_error) :
  226. _mac(mac),
  227. _mtu(mtu),
  228. _r(renv),
  229. _handler(handler),
  230. _arg(arg),
  231. _fd(0)
  232. {
  233. char devpath[64],ethaddr[64],mtustr[16];
  234. struct stat tmp;
  235. Mutex::Lock _l(__tapCreateLock); // create only one tap at a time, globally
  236. if (mtu > 4096)
  237. throw std::runtime_error("max tap MTU is 4096");
  238. // Check for existence of ZT tap devices, try to load module if not there
  239. const char *kextload = UNIX_COMMANDS[ZT_MAC_KEXTLOAD_COMMAND];
  240. if ((stat("/dev/zt0",&tmp))&&(kextload)) {
  241. long kextpid;
  242. char tmp[4096];
  243. strcpy(tmp,_r->homePath.c_str());
  244. if ((kextpid = (long)vfork()) == 0) {
  245. chdir(tmp);
  246. execl(kextload,kextload,"-q","-repository",tmp,"tap.kext",(const char *)0);
  247. _exit(-1);
  248. } else {
  249. int exitcode = -1;
  250. waitpid(kextpid,&exitcode,0);
  251. usleep(500);
  252. }
  253. }
  254. if (stat("/dev/zt0",&tmp))
  255. throw std::runtime_error("/dev/zt# tap devices do not exist and unable to load kernel extension");
  256. // Open the first available device (ones in use will fail with resource busy)
  257. for(int i=0;i<256;++i) {
  258. sprintf(devpath,"/dev/zt%d",i);
  259. if (stat(devpath,&tmp))
  260. throw std::runtime_error("no more TAP devices available");
  261. _fd = ::open(devpath,O_RDWR);
  262. if (_fd > 0) {
  263. sprintf(_dev,"zt%d",i);
  264. break;
  265. }
  266. }
  267. if (_fd <= 0)
  268. throw std::runtime_error("unable to open TAP device or no more devices available");
  269. if (fcntl(_fd,F_SETFL,fcntl(_fd,F_GETFL) & ~O_NONBLOCK) == -1) {
  270. ::close(_fd);
  271. throw std::runtime_error("unable to set flags on file descriptor for TAP device");
  272. }
  273. const char *ifconfig = UNIX_COMMANDS[ZT_UNIX_IFCONFIG_COMMAND];
  274. if (!ifconfig) {
  275. ::close(_fd);
  276. throw std::runtime_error("unable to find 'ifconfig' command on system");
  277. }
  278. // Configure MAC address and MTU, bring interface up
  279. sprintf(ethaddr,"%.2x:%.2x:%.2x:%.2x:%.2x:%.2x",(int)mac[0],(int)mac[1],(int)mac[2],(int)mac[3],(int)mac[4],(int)mac[5]);
  280. sprintf(mtustr,"%u",mtu);
  281. long cpid;
  282. if ((cpid = (long)vfork()) == 0) {
  283. execl(ifconfig,ifconfig,_dev,"lladdr",ethaddr,"mtu",mtustr,"up",(const char *)0);
  284. _exit(-1);
  285. } else {
  286. int exitcode = -1;
  287. waitpid(cpid,&exitcode,0);
  288. if (exitcode) {
  289. ::close(_fd);
  290. throw std::runtime_error("ifconfig failure setting link-layer address and activating tap interface");
  291. }
  292. }
  293. whack(); // turns on IPv6 on OSX
  294. ::pipe(_shutdownSignalPipe);
  295. _thread = Thread::start(this);
  296. }
  297. #endif // __APPLE__
  298. EthernetTap::~EthernetTap()
  299. {
  300. ::write(_shutdownSignalPipe[1],"\0",1); // causes thread to exit
  301. Thread::join(_thread);
  302. ::close(_fd);
  303. }
  304. #ifdef __APPLE__
  305. void EthernetTap::whack()
  306. {
  307. const char *ipconfig = UNIX_COMMANDS[ZT_MAC_IPCONFIG_COMMAND];
  308. if (ipconfig) {
  309. long cpid = (long)vfork();
  310. if (cpid == 0) {
  311. execl(ipconfig,ipconfig,"set",_dev,"AUTOMATIC-V6",(const char *)0);
  312. _exit(-1);
  313. } else {
  314. int exitcode = -1;
  315. waitpid(cpid,&exitcode,0);
  316. }
  317. }
  318. }
  319. #else
  320. void EthernetTap::whack() {}
  321. #endif // __APPLE__ / !__APPLE__
  322. #ifdef __LINUX__
  323. static bool ___removeIp(const char *_dev,const InetAddress &ip)
  324. {
  325. const char *ipcmd = UNIX_COMMANDS[ZT_UNIX_IP_COMMAND];
  326. if (!ipcmd)
  327. return false;
  328. long cpid = (long)vfork();
  329. if (cpid == 0) {
  330. execl(ipcmd,ipcmd,"addr","del",ip.toString().c_str(),"dev",_dev,(const char *)0);
  331. _exit(-1);
  332. } else {
  333. int exitcode = -1;
  334. waitpid(cpid,&exitcode,0);
  335. return (exitcode == 0);
  336. }
  337. }
  338. bool EthernetTap::addIP(const InetAddress &ip)
  339. {
  340. const char *ipcmd = UNIX_COMMANDS[ZT_UNIX_IP_COMMAND];
  341. if (!ipcmd) {
  342. LOG("ERROR: could not configure IP address for %s: unable to find 'ip' command on system (checked /sbin, /bin, /usr/sbin, /usr/bin)",_dev);
  343. return false;
  344. }
  345. Mutex::Lock _l(_ips_m);
  346. if (!ip)
  347. return false;
  348. if (_ips.count(ip) > 0)
  349. return true;
  350. // Remove and reconfigure if address is the same but netmask is different
  351. for(std::set<InetAddress>::iterator i(_ips.begin());i!=_ips.end();++i) {
  352. if (i->ipsEqual(ip)) {
  353. if (___removeIp(_dev,*i)) {
  354. _ips.erase(i);
  355. break;
  356. } else {
  357. LOG("WARNING: failed to remove old IP/netmask %s to replace with %s",i->toString().c_str(),ip.toString().c_str());
  358. }
  359. }
  360. }
  361. long cpid;
  362. if ((cpid = (long)vfork()) == 0) {
  363. execl(ipcmd,ipcmd,"addr","add",ip.toString().c_str(),"dev",_dev,(const char *)0);
  364. _exit(-1);
  365. } else {
  366. int exitcode = -1;
  367. waitpid(cpid,&exitcode,0);
  368. if (exitcode == 0) {
  369. _ips.insert(ip);
  370. return true;
  371. } else return false;
  372. }
  373. return false;
  374. }
  375. #endif // __LINUX__
  376. #ifdef __APPLE__
  377. static bool ___removeIp(const char *_dev,const InetAddress &ip)
  378. {
  379. const char *ifconfig = UNIX_COMMANDS[ZT_UNIX_IFCONFIG_COMMAND];
  380. if (!ifconfig)
  381. return false;
  382. long cpid;
  383. if ((cpid = (long)vfork()) == 0) {
  384. execl(ifconfig,ifconfig,_dev,"inet",ip.toIpString().c_str(),"-alias",(const char *)0);
  385. _exit(-1);
  386. } else {
  387. int exitcode = -1;
  388. waitpid(cpid,&exitcode,0);
  389. return (exitcode == 0);
  390. }
  391. return false; // never reached, make compiler shut up about return value
  392. }
  393. bool EthernetTap::addIP(const InetAddress &ip)
  394. {
  395. const char *ifconfig = UNIX_COMMANDS[ZT_UNIX_IFCONFIG_COMMAND];
  396. if (!ifconfig) {
  397. LOG("ERROR: could not configure IP address for %s: unable to find 'ifconfig' command on system (checked /sbin, /bin, /usr/sbin, /usr/bin)",_dev);
  398. return false;
  399. }
  400. Mutex::Lock _l(_ips_m);
  401. if (!ip)
  402. return false;
  403. if (_ips.count(ip) > 0)
  404. return true; // IP/netmask already assigned
  405. // Remove and reconfigure if address is the same but netmask is different
  406. for(std::set<InetAddress>::iterator i(_ips.begin());i!=_ips.end();++i) {
  407. if ((i->ipsEqual(ip))&&(i->netmaskBits() != ip.netmaskBits())) {
  408. if (___removeIp(_dev,*i)) {
  409. _ips.erase(i);
  410. break;
  411. } else {
  412. LOG("WARNING: failed to remove old IP/netmask %s to replace with %s",i->toString().c_str(),ip.toString().c_str());
  413. }
  414. }
  415. }
  416. long cpid;
  417. if ((cpid = (long)vfork()) == 0) {
  418. execl(ifconfig,ifconfig,_dev,ip.isV4() ? "inet" : "inet6",ip.toString().c_str(),"alias",(const char *)0);
  419. _exit(-1);
  420. } else {
  421. int exitcode = -1;
  422. waitpid(cpid,&exitcode,0);
  423. if (exitcode == 0) {
  424. _ips.insert(ip);
  425. return true;
  426. }
  427. }
  428. return false;
  429. }
  430. #endif // __APPLE__
  431. bool EthernetTap::removeIP(const InetAddress &ip)
  432. {
  433. Mutex::Lock _l(_ips_m);
  434. if (_ips.count(ip) > 0) {
  435. if (___removeIp(_dev,ip)) {
  436. _ips.erase(ip);
  437. return true;
  438. }
  439. }
  440. return false;
  441. }
  442. void EthernetTap::put(const MAC &from,const MAC &to,unsigned int etherType,const void *data,unsigned int len)
  443. {
  444. char putBuf[4096 + 14];
  445. if ((_fd > 0)&&(len <= _mtu)) {
  446. for(int i=0;i<6;++i)
  447. putBuf[i] = to.data[i];
  448. for(int i=0;i<6;++i)
  449. putBuf[i+6] = from.data[i];
  450. *((uint16_t *)(putBuf + 12)) = htons((uint16_t)etherType);
  451. memcpy(putBuf + 14,data,len);
  452. len += 14;
  453. int n = ::write(_fd,putBuf,len);
  454. if (n <= 0) {
  455. LOG("error writing packet to Ethernet tap device: %s",strerror(errno));
  456. } else if (n != (int)len) {
  457. // Saw this gremlin once, so log it if we see it again... OSX tap
  458. // or something seems to have goofy issues with certain MTUs.
  459. LOG("ERROR: write underrun: %s tap write() wrote %d of %u bytes of frame",_dev,n,len);
  460. }
  461. }
  462. }
  463. std::string EthernetTap::deviceName() const
  464. {
  465. return std::string(_dev);
  466. }
  467. #ifdef __LINUX__
  468. bool EthernetTap::updateMulticastGroups(std::set<MulticastGroup> &groups)
  469. {
  470. char *ptr,*ptr2;
  471. unsigned char mac[6];
  472. std::set<MulticastGroup> newGroups;
  473. int fd = ::open("/proc/net/dev_mcast",O_RDONLY);
  474. if (fd > 0) {
  475. char buf[131072];
  476. int n = (int)::read(fd,buf,sizeof(buf));
  477. if ((n > 0)&&(n < (int)sizeof(buf))) {
  478. buf[n] = (char)0;
  479. for(char *l=strtok_r(buf,"\r\n",&ptr);(l);l=strtok_r((char *)0,"\r\n",&ptr)) {
  480. int fno = 0;
  481. char *devname = (char *)0;
  482. char *mcastmac = (char *)0;
  483. for(char *f=strtok_r(l," \t",&ptr2);(f);f=strtok_r((char *)0," \t",&ptr2)) {
  484. if (fno == 1)
  485. devname = f;
  486. else if (fno == 4)
  487. mcastmac = f;
  488. ++fno;
  489. }
  490. if ((devname)&&(!strcmp(devname,_dev))&&(mcastmac)&&(Utils::unhex(mcastmac,mac,6) == 6))
  491. newGroups.insert(MulticastGroup(MAC(mac),0));
  492. }
  493. }
  494. ::close(fd);
  495. }
  496. {
  497. Mutex::Lock _l(_ips_m);
  498. for(std::set<InetAddress>::const_iterator i(_ips.begin());i!=_ips.end();++i)
  499. newGroups.insert(MulticastGroup::deriveMulticastGroupForAddressResolution(*i));
  500. }
  501. bool changed = false;
  502. newGroups.insert(_blindWildcardMulticastGroup); // always join this
  503. for(std::set<MulticastGroup>::iterator mg(newGroups.begin());mg!=newGroups.end();++mg) {
  504. if (!groups.count(*mg)) {
  505. groups.insert(*mg);
  506. changed = true;
  507. }
  508. }
  509. for(std::set<MulticastGroup>::iterator mg(groups.begin());mg!=groups.end();) {
  510. if (!newGroups.count(*mg)) {
  511. groups.erase(mg++);
  512. changed = true;
  513. } else ++mg;
  514. }
  515. return changed;
  516. }
  517. #endif // __LINUX__
  518. #ifdef __APPLE__
  519. bool EthernetTap::updateMulticastGroups(std::set<MulticastGroup> &groups)
  520. {
  521. std::set<MulticastGroup> newGroups;
  522. struct ifmaddrs *ifmap = (struct ifmaddrs *)0;
  523. if (!getifmaddrs(&ifmap)) {
  524. struct ifmaddrs *p = ifmap;
  525. while (p) {
  526. if (p->ifma_addr->sa_family == AF_LINK) {
  527. struct sockaddr_dl *in = (struct sockaddr_dl *)p->ifma_name;
  528. struct sockaddr_dl *la = (struct sockaddr_dl *)p->ifma_addr;
  529. if ((la->sdl_alen == 6)&&(in->sdl_nlen <= sizeof(_dev))&&(!memcmp(_dev,in->sdl_data,in->sdl_nlen)))
  530. newGroups.insert(MulticastGroup(MAC(la->sdl_data + la->sdl_nlen),0));
  531. }
  532. p = p->ifma_next;
  533. }
  534. freeifmaddrs(ifmap);
  535. }
  536. {
  537. Mutex::Lock _l(_ips_m);
  538. for(std::set<InetAddress>::const_iterator i(_ips.begin());i!=_ips.end();++i)
  539. newGroups.insert(MulticastGroup::deriveMulticastGroupForAddressResolution(*i));
  540. }
  541. bool changed = false;
  542. newGroups.insert(_blindWildcardMulticastGroup); // always join this
  543. for(std::set<MulticastGroup>::iterator mg(newGroups.begin());mg!=newGroups.end();++mg) {
  544. if (!groups.count(*mg)) {
  545. groups.insert(*mg);
  546. changed = true;
  547. }
  548. }
  549. for(std::set<MulticastGroup>::iterator mg(groups.begin());mg!=groups.end();) {
  550. if (!newGroups.count(*mg)) {
  551. groups.erase(mg++);
  552. changed = true;
  553. } else ++mg;
  554. }
  555. return changed;
  556. }
  557. #endif // __APPLE__
  558. void EthernetTap::threadMain()
  559. throw()
  560. {
  561. fd_set readfds,nullfds;
  562. MAC to,from;
  563. char getBuf[4096 + 14];
  564. Buffer<4096> data;
  565. // Wait for a moment after startup -- wait for Network to finish
  566. // constructing itself.
  567. Thread::sleep(500);
  568. FD_ZERO(&readfds);
  569. FD_ZERO(&nullfds);
  570. int nfds = (int)std::max(_shutdownSignalPipe[0],_fd) + 1;
  571. for(;;) {
  572. FD_SET(_shutdownSignalPipe[0],&readfds);
  573. FD_SET(_fd,&readfds);
  574. select(nfds,&readfds,&nullfds,&nullfds,(struct timeval *)0);
  575. if (FD_ISSET(_shutdownSignalPipe[0],&readfds)) // writes to shutdown pipe terminate thread
  576. break;
  577. if (FD_ISSET(_fd,&readfds)) {
  578. int n = (int)::read(_fd,getBuf,_mtu + 14);
  579. if (n > 14) {
  580. for(int i=0;i<6;++i)
  581. to.data[i] = (unsigned char)getBuf[i];
  582. for(int i=0;i<6;++i)
  583. from.data[i] = (unsigned char)getBuf[i + 6];
  584. data.copyFrom(getBuf + 14,(unsigned int)n - 14);
  585. _handler(_arg,from,to,ntohs(((const uint16_t *)getBuf)[6]),data);
  586. } else if (n < 0) {
  587. if ((errno != EINTR)&&(errno != ETIMEDOUT)) {
  588. TRACE("unexpected error reading from tap: %s",strerror(errno));
  589. break;
  590. }
  591. }
  592. }
  593. }
  594. }
  595. } // namespace ZeroTier
  596. #endif // __UNIX_LIKE__ //////////////////////////////////////////////////////
  597. //////////////////////////////////////////////////////////////////////////////
  598. #ifdef __WINDOWS__ ///////////////////////////////////////////////////////////
  599. #include <stdio.h>
  600. #include <stdlib.h>
  601. #include <stdint.h>
  602. #include <string.h>
  603. #include <WinSock2.h>
  604. #include <Windows.h>
  605. #include <iphlpapi.h>
  606. #include <ws2ipdef.h>
  607. #include <WS2tcpip.h>
  608. #include <tchar.h>
  609. #include <winreg.h>
  610. namespace ZeroTier {
  611. static Mutex _systemTapInitLock;
  612. EthernetTap::EthernetTap(
  613. const RuntimeEnvironment *renv,
  614. const char *tag,
  615. const MAC &mac,
  616. unsigned int mtu,
  617. void (*handler)(void *,const MAC &,const MAC &,unsigned int,const Buffer<4096> &),
  618. void *arg)
  619. throw(std::runtime_error) :
  620. _mac(mac),
  621. _mtu(mtu),
  622. _r(renv),
  623. _handler(handler),
  624. _arg(arg)
  625. {
  626. char subkeyName[4096];
  627. char subkeyClass[4096];
  628. char data[4096];
  629. Mutex::Lock _l(_systemTapInitLock); // only init one tap at a time, process-wide
  630. HKEY nwAdapters;
  631. if (RegOpenKeyExA(HKEY_LOCAL_MACHINE,"SYSTEM\\CurrentControlSet\\Control\\Class\\{4D36E972-E325-11CE-BFC1-08002BE10318}",0,KEY_READ|KEY_WRITE,&nwAdapters) != ERROR_SUCCESS)
  632. throw std::runtime_error("unable to open registry key for network adapter enumeration");
  633. std::set<std::string> existingDeviceInstances;
  634. std::string mySubkeyName;
  635. // Enumerate all Microsoft Loopback Adapter instances and look for one
  636. // that matches our tag.
  637. for(DWORD subkeyIndex=0;subkeyIndex!=-1;) {
  638. DWORD type;
  639. DWORD dataLen;
  640. DWORD subkeyNameLen = sizeof(subkeyName);
  641. DWORD subkeyClassLen = sizeof(subkeyClass);
  642. FILETIME lastWriteTime;
  643. switch (RegEnumKeyExA(nwAdapters,subkeyIndex++,subkeyName,&subkeyNameLen,(DWORD *)0,subkeyClass,&subkeyClassLen,&lastWriteTime)) {
  644. case ERROR_NO_MORE_ITEMS: subkeyIndex = -1; break;
  645. case ERROR_SUCCESS:
  646. type = 0;
  647. dataLen = sizeof(data);
  648. if (RegGetValueA(nwAdapters,subkeyName,"ComponentId",RRF_RT_ANY,&type,(PVOID)data,&dataLen) == ERROR_SUCCESS) {
  649. data[dataLen] = '\0';
  650. if (!strcmpi(data,"*msloop")) {
  651. std::string instanceId;
  652. type = 0;
  653. dataLen = sizeof(data);
  654. if (RegGetValueA(nwAdapters,subkeyName,"NetCfgInstanceId",RRF_RT_ANY,&type,(PVOID)data,&dataLen) == ERROR_SUCCESS) {
  655. instanceId.assign(data,dataLen);
  656. existingDeviceInstances.insert(instanceId);
  657. }
  658. if ((_myDeviceInstanceId.length() == 0)&&(instanceId.length() != 0)) {
  659. type = 0;
  660. dataLen = sizeof(data);
  661. if (RegGetValueA(nwAdapters,subkeyName,"_ZeroTierTapIdentifier",RRF_RT_ANY,&type,(PVOID)data,&dataLen) == ERROR_SUCCESS) {
  662. data[dataLen] = '\0';
  663. if (!strcmp(data,tag)) {
  664. type = 0;
  665. dataLen = sizeof(data);
  666. if (RegGetValueA(nwAdapters,subkeyName,"NetCfgInstanceId",RRF_RT_ANY,&type,(PVOID)data,&dataLen) == ERROR_SUCCESS) {
  667. _myDeviceInstanceId = instanceId;
  668. mySubkeyName = subkeyName;
  669. subkeyIndex = -1; // break outer loop
  670. }
  671. }
  672. }
  673. }
  674. }
  675. }
  676. break;
  677. }
  678. }
  679. // If there is no device, try to create one
  680. if (_myDeviceInstanceId.length() == 0) {
  681. // Execute devcon to install an instance of the Microsoft Loopback Adapter
  682. #ifdef _WIN64
  683. const char *devcon = "\\devcon64.exe";
  684. #else
  685. BOOL f64 = FALSE;
  686. const char *devcon = ((IsWow64Process(GetCurrentProcess(),&f64) == TRUE) ? "\\devcon64.exe" : "\\devcon32.exe");
  687. #endif
  688. char windir[4096];
  689. windir[0] = '\0';
  690. GetWindowsDirectoryA(windir,sizeof(windir));
  691. STARTUPINFOA startupInfo;
  692. startupInfo.cb = sizeof(startupInfo);
  693. PROCESS_INFORMATION processInfo;
  694. memset(&startupInfo,0,sizeof(STARTUPINFOA));
  695. memset(&processInfo,0,sizeof(PROCESS_INFORMATION));
  696. if (!CreateProcessA(NULL,(LPSTR)(std::string("\"") + _r->homePath + devcon + "\" install " + windir + "\\inf\\netloop.inf *msloop").c_str(),NULL,NULL,FALSE,0,NULL,NULL,&startupInfo,&processInfo)) {
  697. RegCloseKey(nwAdapters);
  698. throw std::runtime_error(std::string("unable to find or execute devcon at ")+devcon);
  699. }
  700. WaitForSingleObject(processInfo.hProcess,INFINITE);
  701. CloseHandle(processInfo.hProcess);
  702. CloseHandle(processInfo.hThread);
  703. // Scan for that new instance by looking for adapters of type
  704. // *msloop that we did not already see on the first scan.
  705. for(DWORD subkeyIndex=0;subkeyIndex!=-1;) {
  706. DWORD type;
  707. DWORD dataLen;
  708. DWORD subkeyNameLen = sizeof(subkeyName);
  709. DWORD subkeyClassLen = sizeof(subkeyClass);
  710. FILETIME lastWriteTime;
  711. switch (RegEnumKeyExA(nwAdapters,subkeyIndex++,subkeyName,&subkeyNameLen,(DWORD *)0,subkeyClass,&subkeyClassLen,&lastWriteTime)) {
  712. case ERROR_NO_MORE_ITEMS: subkeyIndex = -1; break;
  713. case ERROR_SUCCESS:
  714. type = 0;
  715. dataLen = sizeof(data);
  716. if (RegGetValueA(nwAdapters,subkeyName,"ComponentId",RRF_RT_ANY,&type,(PVOID)data,&dataLen) == ERROR_SUCCESS) {
  717. data[dataLen] = '\0';
  718. if (!strcmpi(data,"*msloop")) {
  719. type = 0;
  720. dataLen = sizeof(data);
  721. if (RegGetValueA(nwAdapters,subkeyName,"NetCfgInstanceId",RRF_RT_ANY,&type,(PVOID)data,&dataLen) == ERROR_SUCCESS) {
  722. if (existingDeviceInstances.count(std::string(data,dataLen)) == 0) {
  723. RegSetKeyValueA(nwAdapters,subkeyName,"_ZeroTierTapIdentifier",REG_SZ,tag,strlen(tag)+1);
  724. _myDeviceInstanceId.assign(data,dataLen);
  725. mySubkeyName = subkeyName;
  726. subkeyIndex = -1; // break outer loop
  727. }
  728. }
  729. }
  730. }
  731. break;
  732. }
  733. }
  734. }
  735. if (_myDeviceInstanceId.length() > 0) {
  736. char tmps[4096];
  737. sprintf_s(tmps,"%.2X-%.2X-%.2X-%.2X-%.2X-%.2X",(unsigned int)mac.data[0],(unsigned int)mac.data[1],(unsigned int)mac.data[2],(unsigned int)mac.data[3],(unsigned int)mac.data[4],(unsigned int)mac.data[5]);
  738. RegSetKeyValueA(nwAdapters,mySubkeyName.c_str(),"NetworkAddress",REG_SZ,tmps,strlen(tmps)+1);
  739. DWORD tmp = mtu;
  740. RegSetKeyValueA(nwAdapters,mySubkeyName.c_str(),"MTU",REG_DWORD,(LPCVOID)&tmp,sizeof(tmp));
  741. tmp = 0;
  742. RegSetKeyValueA(nwAdapters,mySubkeyName.c_str(),"EnableDHCP",REG_DWORD,(LPCVOID)&tmp,sizeof(tmp));
  743. RegSetKeyValueA(nwAdapters,mySubkeyName.c_str(),"DriverDesc",REG_SZ,"ZeroTier One Virtual LAN",25);
  744. }
  745. RegCloseKey(nwAdapters);
  746. if (_myDeviceInstanceId.length() == 0)
  747. throw std::runtime_error("unable to create new loopback adapter for tap");
  748. //Thread::start(this);
  749. }
  750. EthernetTap::~EthernetTap()
  751. {
  752. }
  753. void EthernetTap::whack()
  754. {
  755. }
  756. bool EthernetTap::addIP(const InetAddress &ip)
  757. {
  758. return false;
  759. }
  760. bool EthernetTap::removeIP(const InetAddress &ip)
  761. {
  762. return false;
  763. }
  764. void EthernetTap::put(const MAC &from,const MAC &to,unsigned int etherType,const void *data,unsigned int len)
  765. {
  766. if (len > (ZT_IF_MTU))
  767. return;
  768. {
  769. Mutex::Lock _l(_injectPending_m);
  770. _injectPending.push( std::pair<Array<char,ZT_IF_MTU + 32>,unsigned int>(Array<char,ZT_IF_MTU + 32>(),len + 14) );
  771. char *d = _injectPending.back().first.data;
  772. memcpy(d,to.data,6);
  773. memcpy(d + 6,from.data,6);
  774. *((uint16_t *)(d + 12)) = Utils::hton(etherType);
  775. memcpy(d + 14,data,len);
  776. }
  777. //ReleaseSemaphore(_pcapIoThread.updateSem,1,NULL);
  778. }
  779. std::string EthernetTap::deviceName() const
  780. {
  781. return _myDeviceInstanceId;
  782. }
  783. bool EthernetTap::updateMulticastGroups(std::set<MulticastGroup> &groups)
  784. {
  785. // TODO
  786. return false;
  787. }
  788. void EthernetTap::threadMain()
  789. throw()
  790. {
  791. }
  792. } // namespace ZeroTier
  793. #endif // __WINDOWS__ ////////////////////////////////////////////////////////