LinuxNetLink.cpp 27 KB

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