LinuxNetLink.cpp 27 KB

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