LinuxNetLink.cpp 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080
  1. /*
  2. * ZeroTier One - Network Virtualization Everywhere
  3. * Copyright (C) 2011-2018 ZeroTier, Inc. https://www.zerotier.com/
  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. * You can be released from the requirements of the license by purchasing
  21. * a commercial license. Buying such a license is mandatory as soon as you
  22. * develop commercial closed-source software that incorporates or links
  23. * directly against ZeroTier software without disclosing the source code
  24. * of your own application.
  25. */
  26. #include "LinuxNetLink.hpp"
  27. #include <unistd.h>
  28. #include <linux/if_tun.h>
  29. namespace ZeroTier {
  30. struct nl_route_req {
  31. struct nlmsghdr nl;
  32. struct rtmsg rt;
  33. char buf[8192];
  34. };
  35. struct nl_if_req {
  36. struct nlmsghdr nl;
  37. struct ifinfomsg ifa;
  38. char buf[8192];
  39. };
  40. struct nl_adr_req {
  41. struct nlmsghdr nl;
  42. struct ifaddrmsg ifa;
  43. char buf[8192];
  44. };
  45. LinuxNetLink::LinuxNetLink()
  46. : _t()
  47. , _running(false)
  48. , _routes_ipv4()
  49. , _rv4_m()
  50. , _routes_ipv6()
  51. , _rv6_m()
  52. , _seq(0)
  53. , _interfaces()
  54. , _if_m()
  55. , _fd(socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE))
  56. , _la({0})
  57. {
  58. // set socket timeout to 1 sec so we're not permablocking recv() calls
  59. _setSocketTimeout(_fd, 1);
  60. _la.nl_family = AF_NETLINK;
  61. _la.nl_pid = getpid()+1;
  62. _la.nl_groups = RTMGRP_LINK|RTMGRP_IPV4_IFADDR|RTMGRP_IPV6_IFADDR|RTMGRP_IPV4_ROUTE|RTMGRP_IPV6_ROUTE|RTMGRP_NOTIFY;
  63. if (bind(_fd, (struct sockaddr*)&_la, sizeof(_la))) {
  64. fprintf(stderr, "Error connecting to RTNETLINK: %s\n", strerror(errno));
  65. ::exit(1);
  66. }
  67. fprintf(stderr, "Requesting IPV4 Routes\n");
  68. _requestIPv4Routes();
  69. fprintf(stderr, "Requesting IPV6 Routes\n");
  70. _requestIPv6Routes();
  71. fprintf(stderr, "Requesting Interface List\n");
  72. _requestInterfaceList();
  73. _running = true;
  74. _t = Thread::start(this);
  75. }
  76. LinuxNetLink::~LinuxNetLink()
  77. {
  78. _running = false;
  79. Thread::join(_t);
  80. ::close(_fd);
  81. }
  82. void LinuxNetLink::_setSocketTimeout(int fd, int seconds)
  83. {
  84. struct timeval tv;
  85. tv.tv_sec = seconds;
  86. tv.tv_usec = 0;
  87. if(setsockopt(fd, SOL_SOCKET, SO_RCVTIMEO, (const char*)&tv, sizeof(tv)) != 0) {
  88. #ifdef ZT_TRACE
  89. fprintf(stderr, "setsockopt failed: %s\n", strerror(errno));
  90. #endif
  91. }
  92. }
  93. int LinuxNetLink::_doRecv(int fd)
  94. {
  95. char buf[8192];
  96. char *p = NULL;
  97. struct nlmsghdr *nlp;
  98. int nll = 0;
  99. int rtn = 0;
  100. p = buf;
  101. while(true) {
  102. rtn = recv(fd, p, sizeof(buf) - nll, 0);
  103. if (rtn > 0) {
  104. nlp = (struct nlmsghdr *)p;
  105. if(nlp->nlmsg_type == NLMSG_ERROR && (nlp->nlmsg_flags & NLM_F_ACK) != NLM_F_ACK) {
  106. struct nlmsgerr *err = (struct nlmsgerr*)NLMSG_DATA(nlp);
  107. if (err->error != 0) {
  108. #ifdef ZT_TRACE
  109. fprintf(stderr, "rtnetlink error: %s\n", strerror(-(err->error)));
  110. #endif
  111. }
  112. p = buf;
  113. nll = 0;
  114. break;
  115. }
  116. if (nlp->nlmsg_type == NLMSG_NOOP) {
  117. break;
  118. }
  119. if( (nlp->nlmsg_flags & NLM_F_MULTI) == NLM_F_MULTI || (nlp->nlmsg_type == NLMSG_DONE))
  120. {
  121. if (nlp->nlmsg_type == NLMSG_DONE) {
  122. _processMessage(nlp, nll);
  123. p = buf;
  124. nll = 0;
  125. break;
  126. }
  127. p += rtn;
  128. nll += rtn;
  129. }
  130. if (nlp->nlmsg_type == NLMSG_OVERRUN) {
  131. #ifdef ZT_TRACE
  132. fprintf(stderr, "NLMSG_OVERRUN: Data lost\n");
  133. #endif
  134. p = buf;
  135. nll = 0;
  136. break;
  137. }
  138. nll += rtn;
  139. _processMessage(nlp, nll);
  140. p = buf;
  141. nll = 0;
  142. break;
  143. } else {
  144. break;
  145. }
  146. }
  147. return rtn;
  148. }
  149. void LinuxNetLink::threadMain() throw()
  150. {
  151. int rtn = 0;
  152. while(_running) {
  153. rtn = _doRecv(_fd);
  154. if (rtn <= 0) {
  155. Thread::sleep(100);
  156. continue;
  157. }
  158. }
  159. }
  160. void LinuxNetLink::_processMessage(struct nlmsghdr *nlp, int nll)
  161. {
  162. for(; NLMSG_OK(nlp, nll); nlp=NLMSG_NEXT(nlp, nll))
  163. {
  164. switch(nlp->nlmsg_type)
  165. {
  166. case RTM_NEWLINK:
  167. _linkAdded(nlp);
  168. break;
  169. case RTM_DELLINK:
  170. _linkDeleted(nlp);
  171. break;
  172. case RTM_NEWADDR:
  173. _ipAddressAdded(nlp);
  174. break;
  175. case RTM_DELADDR:
  176. _ipAddressDeleted(nlp);
  177. break;
  178. case RTM_NEWROUTE:
  179. _routeAdded(nlp);
  180. break;
  181. case RTM_DELROUTE:
  182. _routeDeleted(nlp);
  183. break;
  184. default:
  185. break;
  186. }
  187. }
  188. }
  189. void LinuxNetLink::_ipAddressAdded(struct nlmsghdr *nlp)
  190. {
  191. struct ifaddrmsg *ifap = (struct ifaddrmsg *)NLMSG_DATA(nlp);
  192. struct rtattr *rtap = (struct rtattr *)IFA_RTA(ifap);
  193. int ifal = IFA_PAYLOAD(nlp);
  194. char addr[40] = {0};
  195. char local[40] = {0};
  196. char label[40] = {0};
  197. char bcast[40] = {0};
  198. for(;RTA_OK(rtap, ifal); rtap=RTA_NEXT(rtap,ifal))
  199. {
  200. switch(rtap->rta_type) {
  201. case IFA_ADDRESS:
  202. inet_ntop(ifap->ifa_family, RTA_DATA(rtap), addr, 40);
  203. break;
  204. case IFA_LOCAL:
  205. inet_ntop(ifap->ifa_family, RTA_DATA(rtap), local, 40);
  206. break;
  207. case IFA_LABEL:
  208. memcpy(label, RTA_DATA(rtap), 40);
  209. break;
  210. case IFA_BROADCAST:
  211. inet_ntop(ifap->ifa_family, RTA_DATA(rtap), bcast, 40);
  212. break;
  213. }
  214. }
  215. #ifdef ZT_TRACE
  216. fprintf(stderr,"Added IP Address %s local: %s label: %s broadcast: %s\n", addr, local, label, bcast);
  217. #endif
  218. }
  219. void LinuxNetLink::_ipAddressDeleted(struct nlmsghdr *nlp)
  220. {
  221. struct ifaddrmsg *ifap = (struct ifaddrmsg *)NLMSG_DATA(nlp);
  222. struct rtattr *rtap = (struct rtattr *)IFA_RTA(ifap);
  223. int ifal = IFA_PAYLOAD(nlp);
  224. char addr[40] = {0};
  225. char local[40] = {0};
  226. char label[40] = {0};
  227. char bcast[40] = {0};
  228. for(;RTA_OK(rtap, ifal); rtap=RTA_NEXT(rtap,ifal))
  229. {
  230. switch(rtap->rta_type) {
  231. case IFA_ADDRESS:
  232. inet_ntop(ifap->ifa_family, RTA_DATA(rtap), addr, 40);
  233. break;
  234. case IFA_LOCAL:
  235. inet_ntop(ifap->ifa_family, RTA_DATA(rtap), local, 40);
  236. break;
  237. case IFA_LABEL:
  238. memcpy(label, RTA_DATA(rtap), 40);
  239. break;
  240. case IFA_BROADCAST:
  241. inet_ntop(ifap->ifa_family, RTA_DATA(rtap), bcast, 40);
  242. break;
  243. }
  244. }
  245. #ifdef ZT_TRACE
  246. fprintf(stderr, "Removed IP Address %s local: %s label: %s broadcast: %s\n", addr, local, label, bcast);
  247. #endif
  248. }
  249. void LinuxNetLink::_routeAdded(struct nlmsghdr *nlp)
  250. {
  251. char dsts[40] = {0};
  252. char gws[40] = {0};
  253. char srcs[40] = {0};
  254. char ifs[16] = {0};
  255. char ms[24] = {0};
  256. struct rtmsg *rtp = (struct rtmsg *) NLMSG_DATA(nlp);
  257. struct rtattr *rtap = (struct rtattr *)RTM_RTA(rtp);
  258. int rtl = RTM_PAYLOAD(nlp);
  259. for(;RTA_OK(rtap, rtl); rtap=RTA_NEXT(rtap, rtl))
  260. {
  261. switch(rtap->rta_type)
  262. {
  263. case RTA_DST:
  264. inet_ntop(rtp->rtm_family, RTA_DATA(rtap), dsts, rtp->rtm_family == AF_INET ? 24 : 40);
  265. break;
  266. case RTA_SRC:
  267. inet_ntop(rtp->rtm_family, RTA_DATA(rtap), srcs, rtp->rtm_family == AF_INET ? 24: 40);
  268. break;
  269. case RTA_GATEWAY:
  270. inet_ntop(rtp->rtm_family, RTA_DATA(rtap), gws, rtp->rtm_family == AF_INET ? 24 : 40);
  271. break;
  272. case RTA_OIF:
  273. sprintf(ifs, "%d", *((int*)RTA_DATA(rtap)));
  274. break;
  275. }
  276. }
  277. sprintf(ms, "%d", rtp->rtm_dst_len);
  278. #ifdef ZT_TRACE
  279. fprintf(stderr, "Route Added: dst %s/%s gw %s src %s if %s\n", dsts, ms, gws, srcs, ifs);
  280. #endif
  281. }
  282. void LinuxNetLink::_routeDeleted(struct nlmsghdr *nlp)
  283. {
  284. char dsts[40] = {0};
  285. char gws[40] = {0};
  286. char srcs[40] = {0};
  287. char ifs[16] = {0};
  288. char ms[24] = {0};
  289. struct rtmsg *rtp = (struct rtmsg *) NLMSG_DATA(nlp);
  290. struct rtattr *rtap = (struct rtattr *)RTM_RTA(rtp);
  291. int rtl = RTM_PAYLOAD(nlp);
  292. for(;RTA_OK(rtap, rtl); rtap=RTA_NEXT(rtap, rtl))
  293. {
  294. switch(rtap->rta_type)
  295. {
  296. case RTA_DST:
  297. inet_ntop(rtp->rtm_family, RTA_DATA(rtap), dsts, rtp->rtm_family == AF_INET ? 24 : 40);
  298. break;
  299. case RTA_SRC:
  300. inet_ntop(rtp->rtm_family, RTA_DATA(rtap), srcs, rtp->rtm_family == AF_INET ? 24 : 40);
  301. break;
  302. case RTA_GATEWAY:
  303. inet_ntop(rtp->rtm_family, RTA_DATA(rtap), gws, rtp->rtm_family == AF_INET ? 24 : 40);
  304. break;
  305. case RTA_OIF:
  306. sprintf(ifs, "%d", *((int*)RTA_DATA(rtap)));
  307. break;
  308. }
  309. }
  310. sprintf(ms, "%d", rtp->rtm_dst_len);
  311. #ifdef ZT_TRACE
  312. fprintf(stderr, "Route Deleted: dst %s/%s gw %s src %s if %s\n", dsts, ms, gws, srcs, ifs);
  313. #endif
  314. }
  315. void LinuxNetLink::_linkAdded(struct nlmsghdr *nlp)
  316. {
  317. char mac[18] = {0};
  318. char mac_bin[6] = {0};
  319. unsigned int mtu = 0;
  320. char ifname[IFNAMSIZ] = {0};
  321. struct ifinfomsg *ifip = (struct ifinfomsg *)NLMSG_DATA(nlp);
  322. struct rtattr *rtap = (struct rtattr *)IFLA_RTA(ifip);
  323. int ifil = RTM_PAYLOAD(nlp);
  324. const char *ptr;
  325. unsigned char *ptr2;
  326. for(;RTA_OK(rtap, ifil);rtap=RTA_NEXT(rtap, ifil))
  327. {
  328. switch(rtap->rta_type) {
  329. case IFLA_ADDRESS:
  330. ptr2 = (unsigned char*)RTA_DATA(rtap);
  331. snprintf(mac, 20, "%02x:%02x:%02x:%02x:%02x:%02x",
  332. ptr2[0], ptr2[1], ptr2[2], ptr2[3], ptr2[4], ptr2[5]);
  333. memcpy(mac_bin, ptr, 6);
  334. break;
  335. case IFLA_IFNAME:
  336. ptr = (const char*)RTA_DATA(rtap);
  337. memcpy(ifname, ptr, strlen(ptr));
  338. break;
  339. case IFLA_MTU:
  340. memcpy(&mtu, RTA_DATA(rtap), sizeof(unsigned int));
  341. break;
  342. }
  343. }
  344. {
  345. Mutex::Lock l(_if_m);
  346. struct iface_entry &entry = _interfaces[ifip->ifi_index];
  347. entry.index = ifip->ifi_index;
  348. memcpy(entry.ifacename, ifname, sizeof(ifname));
  349. memcpy(entry.mac, mac, sizeof(mac));
  350. memcpy(entry.mac_bin, mac_bin, 6);
  351. entry.mtu = mtu;
  352. }
  353. #ifdef ZT_TRACE
  354. fprintf(stderr, "Link Added: %s mac: %s, mtu: %d\n", ifname, mac, mtu);
  355. #endif
  356. }
  357. void LinuxNetLink::_linkDeleted(struct nlmsghdr *nlp)
  358. {
  359. char mac[18] = {0};
  360. unsigned int mtu = 0;
  361. char ifname[40] = {0};
  362. struct ifinfomsg *ifip = (struct ifinfomsg *)NLMSG_DATA(nlp);
  363. struct rtattr *rtap = (struct rtattr *)IFLA_RTA(ifip);
  364. int ifil = RTM_PAYLOAD(nlp);
  365. const char *ptr;
  366. unsigned char *ptr2;
  367. for(;RTA_OK(rtap, ifil);rtap=RTA_NEXT(rtap, ifil))
  368. {
  369. switch(rtap->rta_type) {
  370. case IFLA_ADDRESS:
  371. ptr2 = (unsigned char*)RTA_DATA(rtap);
  372. snprintf(mac, 20, "%02x:%02x:%02x:%02x:%02x:%02x",
  373. ptr2[0], ptr2[1], ptr2[2], ptr2[3], ptr2[4], ptr2[5]);
  374. break;
  375. case IFLA_IFNAME:
  376. ptr = (const char*)RTA_DATA(rtap);
  377. memcpy(ifname, ptr, strlen(ptr));
  378. break;
  379. case IFLA_MTU:
  380. memcpy(&mtu, RTA_DATA(rtap), sizeof(unsigned int));
  381. break;
  382. }
  383. }
  384. #ifdef ZT_TRACE
  385. fprintf(stderr, "Link Deleted: %s mac: %s, mtu: %d\n", ifname, mac, mtu);
  386. #endif
  387. {
  388. Mutex::Lock l(_if_m);
  389. if(_interfaces.contains(ifip->ifi_index)) {
  390. _interfaces.erase(ifip->ifi_index);
  391. }
  392. }
  393. }
  394. void LinuxNetLink::_requestIPv4Routes()
  395. {
  396. int fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
  397. if (fd == -1) {
  398. fprintf(stderr, "Error opening RTNETLINK socket: %s\n", strerror(errno));
  399. return;
  400. }
  401. _setSocketTimeout(fd);
  402. struct sockaddr_nl la;
  403. la.nl_family = AF_NETLINK;
  404. la.nl_pid = getpid();
  405. la.nl_groups = RTMGRP_IPV4_ROUTE;
  406. if(bind(fd, (struct sockaddr*)&la, sizeof(la))) {
  407. fprintf(stderr, "Error binding RTNETLINK: %s\n", strerror(errno));
  408. return;
  409. }
  410. struct nl_route_req req;
  411. bzero(&req, sizeof(req));
  412. req.nl.nlmsg_len = NLMSG_LENGTH(sizeof(struct rtmsg));
  413. req.nl.nlmsg_flags = NLM_F_REQUEST | NLM_F_DUMP;
  414. req.nl.nlmsg_type = RTM_GETROUTE;
  415. req.nl.nlmsg_pid = 0;
  416. req.nl.nlmsg_seq = ++_seq;
  417. req.rt.rtm_family = AF_INET;
  418. req.rt.rtm_table = RT_TABLE_MAIN;
  419. struct sockaddr_nl pa;
  420. bzero(&pa, sizeof(pa));
  421. pa.nl_family = AF_NETLINK;
  422. struct msghdr msg;
  423. bzero(&msg, sizeof(msg));
  424. msg.msg_name = (void*)&pa;
  425. msg.msg_namelen = sizeof(pa);
  426. struct iovec iov;
  427. bzero(&iov, sizeof(iov));
  428. iov.iov_base = (void*)&req.nl;
  429. iov.iov_len = req.nl.nlmsg_len;
  430. msg.msg_iov = &iov;
  431. msg.msg_iovlen = 1;
  432. sendmsg(fd, &msg, 0);
  433. _doRecv(fd);
  434. close(fd);
  435. }
  436. void LinuxNetLink::_requestIPv6Routes()
  437. {
  438. int fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
  439. if (fd == -1) {
  440. fprintf(stderr, "Error opening RTNETLINK socket: %s\n", strerror(errno));
  441. return;
  442. }
  443. _setSocketTimeout(fd);
  444. struct sockaddr_nl la;
  445. la.nl_family = AF_NETLINK;
  446. la.nl_pid = getpid();
  447. la.nl_groups = RTMGRP_IPV6_ROUTE;
  448. if(bind(fd, (struct sockaddr*)&la, sizeof(struct sockaddr_nl))) {
  449. fprintf(stderr, "Error binding RTNETLINK: %s\n", strerror(errno));
  450. return;
  451. }
  452. struct nl_route_req req;
  453. bzero(&req, sizeof(req));
  454. req.nl.nlmsg_len = NLMSG_LENGTH(sizeof(struct rtmsg));
  455. req.nl.nlmsg_flags = NLM_F_REQUEST | NLM_F_DUMP;
  456. req.nl.nlmsg_type = RTM_GETROUTE;
  457. req.nl.nlmsg_pid = 0;
  458. req.nl.nlmsg_seq = ++_seq;
  459. req.rt.rtm_family = AF_INET6;
  460. req.rt.rtm_table = RT_TABLE_MAIN;
  461. struct sockaddr_nl pa;
  462. bzero(&pa, sizeof(pa));
  463. pa.nl_family = AF_NETLINK;
  464. struct msghdr msg;
  465. bzero(&msg, sizeof(msg));
  466. msg.msg_name = (void*)&pa;
  467. msg.msg_namelen = sizeof(pa);
  468. struct iovec iov;
  469. bzero(&iov, sizeof(iov));
  470. iov.iov_base = (void*)&req.nl;
  471. iov.iov_len = req.nl.nlmsg_len;
  472. msg.msg_iov = &iov;
  473. msg.msg_iovlen = 1;
  474. sendmsg(fd, &msg, 0);
  475. _doRecv(fd);
  476. close(fd);
  477. }
  478. void LinuxNetLink::_requestInterfaceList()
  479. {
  480. int fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
  481. if (fd == -1) {
  482. fprintf(stderr, "Error opening RTNETLINK socket: %s\n", strerror(errno));
  483. return;
  484. }
  485. _setSocketTimeout(fd);
  486. struct sockaddr_nl la;
  487. la.nl_family = AF_NETLINK;
  488. la.nl_pid = getpid();
  489. la.nl_groups = RTMGRP_LINK;
  490. if(bind(fd, (struct sockaddr*)&la, sizeof(struct sockaddr_nl))) {
  491. fprintf(stderr, "Error binding RTNETLINK: %s\n", strerror(errno));
  492. return;
  493. }
  494. struct nl_if_req req;
  495. bzero(&req, sizeof(req));
  496. req.nl.nlmsg_len = NLMSG_LENGTH(sizeof(struct ifinfomsg));
  497. req.nl.nlmsg_flags = NLM_F_REQUEST | NLM_F_DUMP;
  498. req.nl.nlmsg_type = RTM_GETLINK;
  499. req.nl.nlmsg_pid = 0;
  500. req.nl.nlmsg_seq = ++_seq;
  501. req.ifa.ifi_family = AF_UNSPEC;
  502. struct sockaddr_nl pa;
  503. bzero(&pa, sizeof(pa));
  504. pa.nl_family = AF_NETLINK;
  505. struct msghdr msg;
  506. bzero(&msg, sizeof(msg));
  507. msg.msg_name = (void*)&pa;
  508. msg.msg_namelen = sizeof(pa);
  509. struct iovec iov;
  510. bzero(&iov, sizeof(iov));
  511. iov.iov_base = (void*)&req.nl;
  512. iov.iov_len = req.nl.nlmsg_len;
  513. msg.msg_iov = &iov;
  514. msg.msg_iovlen = 1;
  515. sendmsg(fd, &msg, 0);
  516. _doRecv(fd);
  517. close(fd);
  518. }
  519. void LinuxNetLink::addRoute(const InetAddress &target, const InetAddress &via, const InetAddress &src, const char *ifaceName)
  520. {
  521. int fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
  522. if (fd == -1) {
  523. fprintf(stderr, "Error opening RTNETLINK socket: %s\n", strerror(errno));
  524. return;
  525. }
  526. _setSocketTimeout(fd);
  527. struct sockaddr_nl la;
  528. bzero(&la, sizeof(la));
  529. la.nl_family = AF_NETLINK;
  530. la.nl_pid = getpid();
  531. if(bind(fd, (struct sockaddr*)&la, sizeof(struct sockaddr_nl))) {
  532. fprintf(stderr, "Error binding RTNETLINK: %s\n", strerror(errno));
  533. return;
  534. }
  535. #ifdef ZT_TRACE
  536. char tmp[64];
  537. char tmp2[64];
  538. char tmp3[64];
  539. fprintf(stderr, "Adding Route. target: %s via: %s src: %s iface: %s\n", target.toString(tmp), via.toString(tmp2), src.toString(tmp3), ifaceName);
  540. #endif
  541. if(!target) {
  542. #ifdef ZT_TRACE
  543. fprintf(stderr, "Uhhhh adding an empty route?!?!?");
  544. #endif
  545. return;
  546. }
  547. int rtl = sizeof(struct rtmsg);
  548. struct nl_route_req req;
  549. bzero(&req, sizeof(req));
  550. struct rtattr *rtap = (struct rtattr *)req.buf;
  551. rtap->rta_type = RTA_DST;
  552. if (target.isV4()) {
  553. rtap->rta_len = RTA_LENGTH(sizeof(struct in_addr));
  554. memcpy(RTA_DATA(rtap), &((struct sockaddr_in*)&target)->sin_addr, sizeof(struct in_addr));
  555. } else {
  556. rtap->rta_len = RTA_LENGTH(sizeof(struct in6_addr));
  557. memcpy(RTA_DATA(rtap), &((struct sockaddr_in6*)&target)->sin6_addr, sizeof(struct in6_addr));
  558. }
  559. rtl += rtap->rta_len;
  560. if(via) {
  561. rtap = (struct rtattr *)(((char*)rtap)+rtap->rta_len);
  562. rtap->rta_type = RTA_GATEWAY;
  563. if(via.isV4()) {
  564. rtap->rta_len = RTA_LENGTH(sizeof(struct in_addr));
  565. memcpy(RTA_DATA(rtap), &((struct sockaddr_in*)&via)->sin_addr, sizeof(struct in_addr));
  566. } else {
  567. rtap->rta_len = RTA_LENGTH(sizeof(struct in6_addr));
  568. memcpy(RTA_DATA(rtap), &((struct sockaddr_in6*)&via)->sin6_addr, sizeof(struct in6_addr));
  569. }
  570. rtl += rtap->rta_len;
  571. } else if (src) {
  572. rtap = (struct rtattr *)(((char*)rtap)+rtap->rta_len);
  573. rtap->rta_type = RTA_SRC;
  574. if(src.isV4()) {
  575. rtap->rta_len = RTA_LENGTH(sizeof(struct in_addr));
  576. memcpy(RTA_DATA(rtap), &((struct sockaddr_in*)&src)->sin_addr, sizeof(struct in_addr));
  577. } else {
  578. rtap->rta_len = RTA_LENGTH(sizeof(struct in6_addr));
  579. memcpy(RTA_DATA(rtap), &((struct sockaddr_in6*)&src)->sin6_addr, sizeof(struct in6_addr));
  580. }
  581. req.rt.rtm_src_len = src.netmaskBits();
  582. }
  583. if (ifaceName != NULL) {
  584. int interface_index = _indexForInterface(ifaceName);
  585. if (interface_index != -1) {
  586. rtap = (struct rtattr *) (((char*)rtap) + rtap->rta_len);
  587. rtap->rta_type = RTA_OIF;
  588. rtap->rta_len = RTA_LENGTH(sizeof(int));
  589. memcpy(RTA_DATA(rtap), &interface_index, sizeof(int));
  590. rtl += rtap->rta_len;
  591. }
  592. }
  593. req.nl.nlmsg_len = NLMSG_LENGTH(rtl);
  594. req.nl.nlmsg_flags = NLM_F_REQUEST | NLM_F_EXCL | NLM_F_CREATE | NLM_F_ACK;
  595. req.nl.nlmsg_type = RTM_NEWROUTE;
  596. req.nl.nlmsg_pid = 0;
  597. req.nl.nlmsg_seq = ++_seq;
  598. req.rt.rtm_family = target.ss_family;
  599. req.rt.rtm_table = RT_TABLE_MAIN;
  600. req.rt.rtm_protocol = RTPROT_STATIC;
  601. req.rt.rtm_scope = RT_SCOPE_UNIVERSE;
  602. req.rt.rtm_type = RTN_UNICAST;
  603. req.rt.rtm_dst_len = target.netmaskBits();
  604. req.rt.rtm_flags = 0;
  605. struct sockaddr_nl pa;
  606. bzero(&pa, sizeof(pa));
  607. pa.nl_family = AF_NETLINK;
  608. struct msghdr msg;
  609. bzero(&msg, sizeof(msg));
  610. msg.msg_name = (void*)&pa;
  611. msg.msg_namelen = sizeof(pa);
  612. struct iovec iov;
  613. bzero(&iov, sizeof(iov));
  614. iov.iov_base = (void*)&req.nl;
  615. iov.iov_len = req.nl.nlmsg_len;
  616. msg.msg_iov = &iov;
  617. msg.msg_iovlen = 1;
  618. sendmsg(fd, &msg, 0);
  619. _doRecv(fd);
  620. close(fd);
  621. }
  622. void LinuxNetLink::delRoute(const InetAddress &target, const InetAddress &via, const InetAddress &src, const char *ifaceName)
  623. {
  624. int fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
  625. if (fd == -1) {
  626. fprintf(stderr, "Error opening RTNETLINK socket: %s\n", strerror(errno));
  627. return;
  628. }
  629. _setSocketTimeout(fd);
  630. struct sockaddr_nl la;
  631. la.nl_family = AF_NETLINK;
  632. la.nl_pid = getpid();
  633. if(bind(fd, (struct sockaddr*)&la, sizeof(struct sockaddr_nl))) {
  634. fprintf(stderr, "Error binding RTNETLINK: %s\n", strerror(errno));
  635. return;
  636. }
  637. #ifdef ZT_TRACE
  638. char tmp[64];
  639. char tmp2[64];
  640. char tmp3[64];
  641. fprintf(stderr, "Removing Route. target: %s via: %s src: %s iface: %s\n", target.toString(tmp), via.toString(tmp2), src.toString(tmp3), ifaceName);
  642. #endif
  643. if(!target) {
  644. #ifdef ZT_TRACE
  645. fprintf(stderr, "Uhhhh deleting an empty route?!?!?");
  646. #endif
  647. return;
  648. }
  649. int rtl = sizeof(struct rtmsg);
  650. struct nl_route_req req;
  651. bzero(&req, sizeof(req));
  652. struct rtattr *rtap = (struct rtattr *)req.buf;
  653. rtap->rta_type = RTA_DST;
  654. if (target.isV4()) {
  655. rtap->rta_len = RTA_LENGTH(sizeof(struct in_addr));
  656. memcpy(RTA_DATA(rtap), &((struct sockaddr_in*)&target)->sin_addr, sizeof(struct in_addr));
  657. } else {
  658. rtap->rta_len = RTA_LENGTH(sizeof(struct in6_addr));
  659. memcpy(RTA_DATA(rtap), &((struct sockaddr_in6*)&target)->sin6_addr, sizeof(struct in6_addr));
  660. }
  661. rtl += rtap->rta_len;
  662. if(via) {
  663. rtap = (struct rtattr *)(((char*)rtap)+rtap->rta_len);
  664. rtap->rta_type = RTA_GATEWAY;
  665. if(via.isV4()) {
  666. rtap->rta_len = RTA_LENGTH(sizeof(struct in_addr));
  667. memcpy(RTA_DATA(rtap), &((struct sockaddr_in*)&via)->sin_addr, sizeof(struct in_addr));
  668. } else {
  669. rtap->rta_len = RTA_LENGTH(sizeof(struct in6_addr));
  670. memcpy(RTA_DATA(rtap), &((struct sockaddr_in6*)&via)->sin6_addr, sizeof(struct in6_addr));
  671. }
  672. rtl += rtap->rta_len;
  673. } else if (src) {
  674. rtap = (struct rtattr *)(((char*)rtap)+rtap->rta_len);
  675. rtap->rta_type = RTA_SRC;
  676. if(src.isV4()) {
  677. rtap->rta_len = RTA_LENGTH(sizeof(struct in_addr));
  678. memcpy(RTA_DATA(rtap), &((struct sockaddr_in*)&src)->sin_addr, sizeof(struct in_addr));
  679. } else {
  680. rtap->rta_len = RTA_LENGTH(sizeof(struct in6_addr));
  681. memcpy(RTA_DATA(rtap), &((struct sockaddr_in6*)&src)->sin6_addr, sizeof(struct in6_addr));
  682. }
  683. req.rt.rtm_src_len = src.netmaskBits();
  684. }
  685. if (ifaceName != NULL) {
  686. int interface_index = _indexForInterface(ifaceName);
  687. if (interface_index != -1) {
  688. rtap = (struct rtattr *) (((char*)rtap) + rtap->rta_len);
  689. rtap->rta_type = RTA_OIF;
  690. rtap->rta_len = RTA_LENGTH(sizeof(int));
  691. memcpy(RTA_DATA(rtap), &interface_index, sizeof(int));
  692. rtl += rtap->rta_len;
  693. }
  694. }
  695. req.nl.nlmsg_len = NLMSG_LENGTH(rtl);
  696. req.nl.nlmsg_flags = NLM_F_REQUEST;
  697. req.nl.nlmsg_type = RTM_DELROUTE;
  698. req.nl.nlmsg_pid = 0;
  699. req.nl.nlmsg_seq = ++_seq;
  700. req.rt.rtm_family = target.ss_family;
  701. req.rt.rtm_table = RT_TABLE_MAIN;
  702. req.rt.rtm_protocol = RTPROT_STATIC;
  703. req.rt.rtm_scope = RT_SCOPE_UNIVERSE;
  704. req.rt.rtm_type = RTN_UNICAST;
  705. req.rt.rtm_dst_len = target.netmaskBits();
  706. req.rt.rtm_flags = 0;
  707. struct sockaddr_nl pa;
  708. bzero(&pa, sizeof(pa));
  709. pa.nl_family = AF_NETLINK;
  710. struct msghdr msg;
  711. bzero(&msg, sizeof(msg));
  712. msg.msg_name = (void*)&pa;
  713. msg.msg_namelen = sizeof(pa);
  714. struct iovec iov;
  715. bzero(&iov, sizeof(iov));
  716. iov.iov_base = (void*)&req.nl;
  717. iov.iov_len = req.nl.nlmsg_len;
  718. msg.msg_iov = &iov;
  719. msg.msg_iovlen = 1;
  720. sendmsg(fd, &msg, 0);
  721. _doRecv(fd);
  722. close(fd);
  723. }
  724. void LinuxNetLink::addAddress(const InetAddress &addr, const char *iface)
  725. {
  726. int fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
  727. if (fd == -1) {
  728. fprintf(stderr, "Error opening RTNETLINK socket: %s\n", strerror(errno));
  729. return;
  730. }
  731. _setSocketTimeout(fd);
  732. struct sockaddr_nl la;
  733. la.nl_family = AF_NETLINK;
  734. la.nl_pid = getpid();
  735. if (addr.isV4()) {
  736. la.nl_groups = RTMGRP_IPV4_IFADDR;
  737. } else {
  738. la.nl_groups = RTMGRP_IPV6_IFADDR;
  739. }
  740. if(bind(fd, (struct sockaddr*)&la, sizeof(struct sockaddr_nl))) {
  741. fprintf(stderr, "Error binding RTNETLINK: %s\n", strerror(errno));
  742. return;
  743. }
  744. #ifdef ZT_TRACE
  745. char tmp[128];
  746. fprintf(stderr, "Adding IP address %s to interface %s", addr.toString(tmp), iface);
  747. #endif
  748. int interface_index = _indexForInterface(iface);
  749. if (interface_index == -1) {
  750. fprintf(stderr, "Unable to find index for interface %s\n", iface);
  751. return;
  752. }
  753. int rtl = sizeof(struct ifaddrmsg);
  754. struct nl_adr_req req;
  755. bzero(&req, sizeof(struct nl_adr_req));
  756. struct rtattr *rtap = (struct rtattr *)req.buf;;
  757. if(addr.isV4()) {
  758. struct sockaddr_in *addr_v4 = (struct sockaddr_in*)&addr;
  759. rtap->rta_type = IFA_ADDRESS;
  760. rtap->rta_len = RTA_LENGTH(sizeof(struct in_addr));
  761. memcpy(RTA_DATA(rtap), &addr_v4->sin_addr, sizeof(struct in_addr));
  762. rtl += rtap->rta_len;
  763. rtap = (struct rtattr*)(((char*)rtap) + rtap->rta_len);
  764. rtap->rta_type = IFA_LOCAL;
  765. rtap->rta_len = RTA_LENGTH(sizeof(struct in_addr));
  766. memcpy(RTA_DATA(rtap), &addr_v4->sin_addr, sizeof(struct in_addr));
  767. rtl += rtap->rta_len;
  768. InetAddress broadcast = addr.broadcast();
  769. if(broadcast) {
  770. rtap = (struct rtattr*)(((char*)rtap)+rtap->rta_len);
  771. struct sockaddr_in *bcast = (struct sockaddr_in*)&broadcast;
  772. rtap->rta_type = IFA_BROADCAST;
  773. rtap->rta_len = RTA_LENGTH(sizeof(struct in_addr));
  774. memcpy(RTA_DATA(rtap), &bcast->sin_addr, sizeof(struct in_addr));
  775. rtl += rtap->rta_len;
  776. }
  777. } else { //V6
  778. rtap->rta_type = IFA_ADDRESS;
  779. struct sockaddr_in6 *addr_v6 = (struct sockaddr_in6*)&addr;
  780. rtap->rta_len = RTA_LENGTH(sizeof(struct in6_addr));
  781. memcpy(RTA_DATA(rtap), &addr_v6->sin6_addr, sizeof(struct in6_addr));
  782. rtl += rtap->rta_len;
  783. }
  784. if (iface) {
  785. rtap = (struct rtattr*)(((char*)rtap)+rtap->rta_len);
  786. rtap->rta_type = IFA_LABEL;
  787. rtap->rta_len = RTA_LENGTH(strlen(iface));
  788. memcpy(RTA_DATA(rtap), iface, strlen(iface));
  789. rtl += rtap->rta_len;
  790. }
  791. req.nl.nlmsg_len = NLMSG_LENGTH(rtl);
  792. req.nl.nlmsg_flags = NLM_F_REQUEST | NLM_F_CREATE | NLM_F_EXCL;
  793. req.nl.nlmsg_type = RTM_NEWADDR;
  794. req.nl.nlmsg_pid = 0;
  795. req.nl.nlmsg_seq = ++_seq;
  796. req.ifa.ifa_family = addr.ss_family;
  797. req.ifa.ifa_prefixlen = addr.port();
  798. req.ifa.ifa_flags = IFA_F_PERMANENT;
  799. req.ifa.ifa_scope = 0;
  800. req.ifa.ifa_index = interface_index;
  801. struct sockaddr_nl pa;
  802. bzero(&pa, sizeof(sockaddr_nl));
  803. pa.nl_family = AF_NETLINK;
  804. struct msghdr msg;
  805. bzero(&msg, sizeof(msg));
  806. msg.msg_name = (void*)&pa;
  807. msg.msg_namelen = sizeof(pa);
  808. struct iovec iov;
  809. iov.iov_base = (void*)&req.nl;
  810. iov.iov_len = req.nl.nlmsg_len;
  811. msg.msg_iov = &iov;
  812. msg.msg_iovlen = 1;
  813. sendmsg(fd, &msg, 0);
  814. _doRecv(fd);
  815. close(fd);
  816. }
  817. void LinuxNetLink::removeAddress(const InetAddress &addr, const char *iface)
  818. {
  819. int fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
  820. if (fd == -1) {
  821. fprintf(stderr, "Error opening RTNETLINK socket: %s\n", strerror(errno));
  822. return;
  823. }
  824. _setSocketTimeout(fd);
  825. struct sockaddr_nl la;
  826. la.nl_family = AF_NETLINK;
  827. la.nl_pid = getpid();
  828. if (addr.isV4()) {
  829. la.nl_groups = RTMGRP_IPV4_IFADDR;
  830. } else {
  831. la.nl_groups = RTMGRP_IPV6_IFADDR;
  832. }
  833. if(bind(fd, (struct sockaddr*)&la, sizeof(struct sockaddr_nl))) {
  834. fprintf(stderr, "Error binding RTNETLINK: %s\n", strerror(errno));
  835. return;
  836. }
  837. #ifdef ZT_TRACE
  838. char tmp[128];
  839. fprintf(stderr, "Removing IP address %s from interface %s", addr.toString(tmp), iface);
  840. #endif
  841. int interface_index = _indexForInterface(iface);
  842. if (interface_index == -1) {
  843. fprintf(stderr, "Unable to find index for interface %s\n", iface);
  844. return;
  845. }
  846. int rtl = sizeof(struct ifaddrmsg);
  847. struct nl_adr_req req;
  848. bzero(&req, sizeof(struct nl_adr_req));
  849. struct rtattr *rtap = (struct rtattr *)req.buf;
  850. if(addr.isV4()) {
  851. struct sockaddr_in *addr_v4 = (struct sockaddr_in*)&addr;
  852. rtap->rta_type = IFA_ADDRESS;
  853. rtap->rta_len = RTA_LENGTH(sizeof(struct in_addr));
  854. memcpy(RTA_DATA(rtap), &addr_v4->sin_addr, sizeof(struct in_addr));
  855. rtl += rtap->rta_len;
  856. rtap = (struct rtattr*)(((char*)rtap) + rtap->rta_len);
  857. rtap->rta_type = IFA_LOCAL;
  858. rtap->rta_len = RTA_LENGTH(sizeof(struct in_addr));
  859. memcpy(RTA_DATA(rtap), &addr_v4->sin_addr, sizeof(struct in_addr));
  860. rtl += rtap->rta_len;
  861. InetAddress broadcast = addr.broadcast();
  862. if(broadcast) {
  863. rtap = (struct rtattr*)(((char*)rtap)+rtap->rta_len);
  864. struct sockaddr_in *bcast = (struct sockaddr_in*)&broadcast;
  865. rtap->rta_type = IFA_BROADCAST;
  866. rtap->rta_len = RTA_LENGTH(sizeof(struct in_addr));
  867. memcpy(RTA_DATA(rtap), &bcast->sin_addr, sizeof(struct in_addr));
  868. rtl += rtap->rta_len;
  869. }
  870. } else { //V6
  871. rtap->rta_type = IFA_ADDRESS;
  872. struct sockaddr_in6 *addr_v6 = (struct sockaddr_in6*)&addr;
  873. rtap->rta_len = RTA_LENGTH(sizeof(struct in6_addr));
  874. memcpy(RTA_DATA(rtap), &addr_v6->sin6_addr, sizeof(struct in6_addr));
  875. rtl += rtap->rta_len;
  876. }
  877. if (iface) {
  878. rtap = (struct rtattr*)(((char*)rtap)+rtap->rta_len);
  879. rtap->rta_type = IFA_LABEL;
  880. rtap->rta_len = RTA_LENGTH(strlen(iface));
  881. memcpy(RTA_DATA(rtap), iface, strlen(iface));
  882. rtl += rtap->rta_len;
  883. }
  884. req.nl.nlmsg_len = NLMSG_LENGTH(rtl);
  885. req.nl.nlmsg_flags = NLM_F_REQUEST;
  886. req.nl.nlmsg_type = RTM_DELADDR;
  887. req.nl.nlmsg_pid = 0;
  888. req.nl.nlmsg_seq = ++_seq;
  889. req.ifa.ifa_family = addr.ss_family;
  890. req.ifa.ifa_prefixlen = addr.port();
  891. req.ifa.ifa_flags = IFA_F_PERMANENT;
  892. req.ifa.ifa_scope = 0;
  893. req.ifa.ifa_index = interface_index;
  894. struct sockaddr_nl pa;
  895. bzero(&pa, sizeof(sockaddr_nl));
  896. pa.nl_family = AF_NETLINK;
  897. struct msghdr msg;
  898. bzero(&msg, sizeof(msg));
  899. msg.msg_name = (void*)&pa;
  900. msg.msg_namelen = sizeof(pa);
  901. struct iovec iov;
  902. iov.iov_base = (void*)&req.nl;
  903. iov.iov_len = req.nl.nlmsg_len;
  904. msg.msg_iov = &iov;
  905. msg.msg_iovlen = 1;
  906. sendmsg(fd, &msg, 0);
  907. _doRecv(fd);
  908. close(fd);
  909. }
  910. RouteList LinuxNetLink::getIPV4Routes() const
  911. {
  912. return _routes_ipv4;
  913. }
  914. RouteList LinuxNetLink::getIPV6Routes() const
  915. {
  916. return _routes_ipv6;
  917. }
  918. int LinuxNetLink::_indexForInterface(const char *iface)
  919. {
  920. Mutex::Lock l(_if_m);
  921. int interface_index = -1;
  922. Hashtable<int, iface_entry>::Iterator iter(_interfaces);
  923. int *k = NULL;
  924. iface_entry *v = NULL;
  925. while(iter.next(k,v)) {
  926. if(strcmp(iface, v->ifacename) == 0) {
  927. interface_index = v->index;
  928. break;
  929. }
  930. }
  931. return interface_index;
  932. }
  933. } // namespace ZeroTier