Network.cpp 51 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435
  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: 2023-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 <stdio.h>
  14. #include <string.h>
  15. #include <stdlib.h>
  16. #include <math.h>
  17. #include "../include/ZeroTierDebug.h"
  18. #include "Constants.hpp"
  19. #include "Network.hpp"
  20. #include "RuntimeEnvironment.hpp"
  21. #include "MAC.hpp"
  22. #include "Address.hpp"
  23. #include "InetAddress.hpp"
  24. #include "Switch.hpp"
  25. #include "Buffer.hpp"
  26. #include "Packet.hpp"
  27. #include "NetworkController.hpp"
  28. #include "Node.hpp"
  29. #include "Peer.hpp"
  30. #include "Trace.hpp"
  31. #include "ScopedPtr.hpp"
  32. #include <set>
  33. namespace ZeroTier {
  34. namespace {
  35. // Returns true if packet appears valid; pos and proto will be set
  36. static bool _ipv6GetPayload(const uint8_t *frameData,unsigned int frameLen,unsigned int &pos,unsigned int &proto)
  37. {
  38. if (frameLen < 40)
  39. return false;
  40. pos = 40;
  41. proto = frameData[6];
  42. while (pos <= frameLen) {
  43. switch(proto) {
  44. case 0: // hop-by-hop options
  45. case 43: // routing
  46. case 60: // destination options
  47. case 135: // mobility options
  48. if ((pos + 8) > frameLen)
  49. return false; // invalid!
  50. proto = frameData[pos];
  51. pos += ((unsigned int)frameData[pos + 1] * 8) + 8;
  52. break;
  53. //case 44: // fragment -- we currently can't parse these and they are deprecated in IPv6 anyway
  54. //case 50:
  55. //case 51: // IPSec ESP and AH -- we have to stop here since this is encrypted stuff
  56. default:
  57. return true;
  58. }
  59. }
  60. return false; // overflow == invalid
  61. }
  62. enum _doZtFilterResult
  63. {
  64. DOZTFILTER_NO_MATCH,
  65. DOZTFILTER_DROP,
  66. DOZTFILTER_REDIRECT,
  67. DOZTFILTER_ACCEPT,
  68. DOZTFILTER_SUPER_ACCEPT
  69. };
  70. static _doZtFilterResult _doZtFilter(
  71. const RuntimeEnvironment *RR,
  72. Trace::RuleResultLog &rrl,
  73. const NetworkConfig &nconf,
  74. const Membership *membership, // can be NULL
  75. const bool inbound,
  76. const Address &ztSource,
  77. Address &ztDest, // MUTABLE -- is changed on REDIRECT actions
  78. const MAC &macSource,
  79. const MAC &macDest,
  80. const uint8_t *const frameData,
  81. const unsigned int frameLen,
  82. const unsigned int etherType,
  83. const unsigned int vlanId,
  84. const ZT_VirtualNetworkRule *rules, // cannot be NULL
  85. const unsigned int ruleCount,
  86. Address &cc, // MUTABLE -- set to TEE destination if TEE action is taken or left alone otherwise
  87. unsigned int &ccLength, // MUTABLE -- set to length of packet payload to TEE
  88. bool &ccWatch, // MUTABLE -- set to true for WATCH target as opposed to normal TEE
  89. uint8_t &qosBucket) // MUTABLE -- set to the value of the argument provided to PRIORITY
  90. {
  91. // Set to true if we are a TEE/REDIRECT/WATCH target
  92. bool superAccept = false;
  93. // The default match state for each set of entries starts as 'true' since an
  94. // ACTION with no MATCH entries preceding it is always taken.
  95. uint8_t thisSetMatches = 1;
  96. rrl.clear();
  97. for(unsigned int rn=0;rn<ruleCount;++rn) {
  98. const ZT_VirtualNetworkRuleType rt = (ZT_VirtualNetworkRuleType)(rules[rn].t & 0x3f);
  99. // First check if this is an ACTION
  100. if ((unsigned int)rt <= (unsigned int)ZT_NETWORK_RULE_ACTION__MAX_ID) {
  101. if (thisSetMatches) {
  102. switch(rt) {
  103. case ZT_NETWORK_RULE_ACTION_PRIORITY:
  104. qosBucket = (rules[rn].v.qosBucket >= 0 || rules[rn].v.qosBucket <= 8) ? rules[rn].v.qosBucket : 4; // 4 = default bucket (no priority)
  105. return DOZTFILTER_ACCEPT;
  106. case ZT_NETWORK_RULE_ACTION_DROP:
  107. return DOZTFILTER_DROP;
  108. case ZT_NETWORK_RULE_ACTION_ACCEPT:
  109. return (superAccept ? DOZTFILTER_SUPER_ACCEPT : DOZTFILTER_ACCEPT); // match, accept packet
  110. // These are initially handled together since preliminary logic is common
  111. case ZT_NETWORK_RULE_ACTION_TEE:
  112. case ZT_NETWORK_RULE_ACTION_WATCH:
  113. case ZT_NETWORK_RULE_ACTION_REDIRECT: {
  114. const Address fwdAddr(rules[rn].v.fwd.address);
  115. if (fwdAddr == ztSource) {
  116. // Skip as no-op since source is target
  117. } else if (fwdAddr == RR->identity.address()) {
  118. if (inbound) {
  119. return DOZTFILTER_SUPER_ACCEPT;
  120. } else {
  121. }
  122. } else if (fwdAddr == ztDest) {
  123. } else {
  124. if (rt == ZT_NETWORK_RULE_ACTION_REDIRECT) {
  125. ztDest = fwdAddr;
  126. return DOZTFILTER_REDIRECT;
  127. } else {
  128. cc = fwdAddr;
  129. ccLength = (rules[rn].v.fwd.length != 0) ? ((frameLen < (unsigned int)rules[rn].v.fwd.length) ? frameLen : (unsigned int)rules[rn].v.fwd.length) : frameLen;
  130. ccWatch = (rt == ZT_NETWORK_RULE_ACTION_WATCH);
  131. }
  132. }
  133. } continue;
  134. case ZT_NETWORK_RULE_ACTION_BREAK:
  135. return DOZTFILTER_NO_MATCH;
  136. // Unrecognized ACTIONs are ignored as no-ops
  137. default:
  138. continue;
  139. }
  140. } else {
  141. // If this is an incoming packet and we are a TEE or REDIRECT target, we should
  142. // super-accept if we accept at all. This will cause us to accept redirected or
  143. // tee'd packets in spite of MAC and ZT addressing checks.
  144. if (inbound) {
  145. switch(rt) {
  146. case ZT_NETWORK_RULE_ACTION_TEE:
  147. case ZT_NETWORK_RULE_ACTION_WATCH:
  148. case ZT_NETWORK_RULE_ACTION_REDIRECT:
  149. if (RR->identity.address() == rules[rn].v.fwd.address)
  150. superAccept = true;
  151. break;
  152. default:
  153. break;
  154. }
  155. }
  156. thisSetMatches = 1; // reset to default true for next batch of entries
  157. continue;
  158. }
  159. }
  160. // Circuit breaker: no need to evaluate an AND if the set's match state
  161. // is currently false since anything AND false is false.
  162. if ((!thisSetMatches)&&(!(rules[rn].t & 0x40))) {
  163. rrl.logSkipped(rn,thisSetMatches);
  164. continue;
  165. }
  166. // If this was not an ACTION evaluate next MATCH and update thisSetMatches with (AND [result])
  167. uint8_t thisRuleMatches = 0;
  168. uint64_t ownershipVerificationMask = 1; // this magic value means it hasn't been computed yet -- this is done lazily the first time it's needed
  169. switch(rt) {
  170. case ZT_NETWORK_RULE_MATCH_SOURCE_ZEROTIER_ADDRESS:
  171. thisRuleMatches = (uint8_t)(rules[rn].v.zt == ztSource.toInt());
  172. break;
  173. case ZT_NETWORK_RULE_MATCH_DEST_ZEROTIER_ADDRESS:
  174. thisRuleMatches = (uint8_t)(rules[rn].v.zt == ztDest.toInt());
  175. break;
  176. case ZT_NETWORK_RULE_MATCH_VLAN_ID:
  177. thisRuleMatches = (uint8_t)(rules[rn].v.vlanId == (uint16_t)vlanId);
  178. break;
  179. case ZT_NETWORK_RULE_MATCH_VLAN_PCP:
  180. // NOT SUPPORTED YET
  181. thisRuleMatches = (uint8_t)(rules[rn].v.vlanPcp == 0);
  182. break;
  183. case ZT_NETWORK_RULE_MATCH_VLAN_DEI:
  184. // NOT SUPPORTED YET
  185. thisRuleMatches = (uint8_t)(rules[rn].v.vlanDei == 0);
  186. break;
  187. case ZT_NETWORK_RULE_MATCH_MAC_SOURCE:
  188. thisRuleMatches = (uint8_t)(MAC(rules[rn].v.mac,6) == macSource);
  189. break;
  190. case ZT_NETWORK_RULE_MATCH_MAC_DEST:
  191. thisRuleMatches = (uint8_t)(MAC(rules[rn].v.mac,6) == macDest);
  192. break;
  193. case ZT_NETWORK_RULE_MATCH_IPV4_SOURCE:
  194. if ((etherType == ZT_ETHERTYPE_IPV4)&&(frameLen >= 20)) {
  195. thisRuleMatches = (uint8_t)(InetAddress((const void *)&(rules[rn].v.ipv4.ip),4,rules[rn].v.ipv4.mask).containsAddress(InetAddress((const void *)(frameData + 12),4,0)));
  196. } else {
  197. thisRuleMatches = 0;
  198. }
  199. break;
  200. case ZT_NETWORK_RULE_MATCH_IPV4_DEST:
  201. if ((etherType == ZT_ETHERTYPE_IPV4)&&(frameLen >= 20)) {
  202. thisRuleMatches = (uint8_t)(InetAddress((const void *)&(rules[rn].v.ipv4.ip),4,rules[rn].v.ipv4.mask).containsAddress(InetAddress((const void *)(frameData + 16),4,0)));
  203. } else {
  204. thisRuleMatches = 0;
  205. }
  206. break;
  207. case ZT_NETWORK_RULE_MATCH_IPV6_SOURCE:
  208. if ((etherType == ZT_ETHERTYPE_IPV6)&&(frameLen >= 40)) {
  209. thisRuleMatches = (uint8_t)(InetAddress((const void *)rules[rn].v.ipv6.ip,16,rules[rn].v.ipv6.mask).containsAddress(InetAddress((const void *)(frameData + 8),16,0)));
  210. } else {
  211. thisRuleMatches = 0;
  212. }
  213. break;
  214. case ZT_NETWORK_RULE_MATCH_IPV6_DEST:
  215. if ((etherType == ZT_ETHERTYPE_IPV6)&&(frameLen >= 40)) {
  216. thisRuleMatches = (uint8_t)(InetAddress((const void *)rules[rn].v.ipv6.ip,16,rules[rn].v.ipv6.mask).containsAddress(InetAddress((const void *)(frameData + 24),16,0)));
  217. } else {
  218. thisRuleMatches = 0;
  219. }
  220. break;
  221. case ZT_NETWORK_RULE_MATCH_IP_TOS:
  222. if ((etherType == ZT_ETHERTYPE_IPV4)&&(frameLen >= 20)) {
  223. const uint8_t tosMasked = frameData[1] & rules[rn].v.ipTos.mask;
  224. thisRuleMatches = (uint8_t)((tosMasked >= rules[rn].v.ipTos.value[0])&&(tosMasked <= rules[rn].v.ipTos.value[1]));
  225. } else if ((etherType == ZT_ETHERTYPE_IPV6)&&(frameLen >= 40)) {
  226. const uint8_t tosMasked = (((frameData[0] << 4) & 0xf0) | ((frameData[1] >> 4) & 0x0f)) & rules[rn].v.ipTos.mask;
  227. thisRuleMatches = (uint8_t)((tosMasked >= rules[rn].v.ipTos.value[0])&&(tosMasked <= rules[rn].v.ipTos.value[1]));
  228. } else {
  229. thisRuleMatches = 0;
  230. }
  231. break;
  232. case ZT_NETWORK_RULE_MATCH_IP_PROTOCOL:
  233. if ((etherType == ZT_ETHERTYPE_IPV4)&&(frameLen >= 20)) {
  234. thisRuleMatches = (uint8_t)(rules[rn].v.ipProtocol == frameData[9]);
  235. } else if (etherType == ZT_ETHERTYPE_IPV6) {
  236. unsigned int pos = 0,proto = 0;
  237. if (_ipv6GetPayload(frameData,frameLen,pos,proto)) {
  238. thisRuleMatches = (uint8_t)(rules[rn].v.ipProtocol == (uint8_t)proto);
  239. } else {
  240. thisRuleMatches = 0;
  241. }
  242. } else {
  243. thisRuleMatches = 0;
  244. }
  245. break;
  246. case ZT_NETWORK_RULE_MATCH_ETHERTYPE:
  247. thisRuleMatches = (uint8_t)(rules[rn].v.etherType == (uint16_t)etherType);
  248. break;
  249. case ZT_NETWORK_RULE_MATCH_ICMP:
  250. if ((etherType == ZT_ETHERTYPE_IPV4)&&(frameLen >= 20)) {
  251. if (frameData[9] == 0x01) { // IP protocol == ICMP
  252. const unsigned int ihl = (frameData[0] & 0xf) * 4;
  253. if (frameLen >= (ihl + 2)) {
  254. if (rules[rn].v.icmp.type == frameData[ihl]) {
  255. if ((rules[rn].v.icmp.flags & 0x01) != 0) {
  256. thisRuleMatches = (uint8_t)(frameData[ihl+1] == rules[rn].v.icmp.code);
  257. } else {
  258. thisRuleMatches = 1;
  259. }
  260. } else {
  261. thisRuleMatches = 0;
  262. }
  263. } else {
  264. thisRuleMatches = 0;
  265. }
  266. } else {
  267. thisRuleMatches = 0;
  268. }
  269. } else if (etherType == ZT_ETHERTYPE_IPV6) {
  270. unsigned int pos = 0,proto = 0;
  271. if (_ipv6GetPayload(frameData,frameLen,pos,proto)) {
  272. if ((proto == 0x3a)&&(frameLen >= (pos+2))) {
  273. if (rules[rn].v.icmp.type == frameData[pos]) {
  274. if ((rules[rn].v.icmp.flags & 0x01) != 0) {
  275. thisRuleMatches = (uint8_t)(frameData[pos+1] == rules[rn].v.icmp.code);
  276. } else {
  277. thisRuleMatches = 1;
  278. }
  279. } else {
  280. thisRuleMatches = 0;
  281. }
  282. } else {
  283. thisRuleMatches = 0;
  284. }
  285. } else {
  286. thisRuleMatches = 0;
  287. }
  288. } else {
  289. thisRuleMatches = 0;
  290. }
  291. break;
  292. case ZT_NETWORK_RULE_MATCH_IP_SOURCE_PORT_RANGE:
  293. case ZT_NETWORK_RULE_MATCH_IP_DEST_PORT_RANGE:
  294. if ((etherType == ZT_ETHERTYPE_IPV4)&&(frameLen >= 20)) {
  295. const unsigned int headerLen = 4 * (frameData[0] & 0xf);
  296. int p = -1;
  297. switch(frameData[9]) { // IP protocol number
  298. // All these start with 16-bit source and destination port in that order
  299. case 0x06: // TCP
  300. case 0x11: // UDP
  301. case 0x84: // SCTP
  302. case 0x88: // UDPLite
  303. if (frameLen > (headerLen + 4)) {
  304. unsigned int pos = headerLen + ((rt == ZT_NETWORK_RULE_MATCH_IP_DEST_PORT_RANGE) ? 2 : 0);
  305. p = (int)frameData[pos++] << 8;
  306. p |= (int)frameData[pos];
  307. }
  308. break;
  309. }
  310. thisRuleMatches = (p >= 0) ? (uint8_t)((p >= (int)rules[rn].v.port[0])&&(p <= (int)rules[rn].v.port[1])) : (uint8_t)0;
  311. } else if (etherType == ZT_ETHERTYPE_IPV6) {
  312. unsigned int pos = 0,proto = 0;
  313. if (_ipv6GetPayload(frameData,frameLen,pos,proto)) {
  314. int p = -1;
  315. switch(proto) { // IP protocol number
  316. // All these start with 16-bit source and destination port in that order
  317. case 0x06: // TCP
  318. case 0x11: // UDP
  319. case 0x84: // SCTP
  320. case 0x88: // UDPLite
  321. if (frameLen > (pos + 4)) {
  322. if (rt == ZT_NETWORK_RULE_MATCH_IP_DEST_PORT_RANGE) pos += 2;
  323. p = (int)frameData[pos++] << 8;
  324. p |= (int)frameData[pos];
  325. }
  326. break;
  327. }
  328. thisRuleMatches = (p > 0) ? (uint8_t)((p >= (int)rules[rn].v.port[0])&&(p <= (int)rules[rn].v.port[1])) : (uint8_t)0;
  329. } else {
  330. thisRuleMatches = 0;
  331. }
  332. } else {
  333. thisRuleMatches = 0;
  334. }
  335. break;
  336. case ZT_NETWORK_RULE_MATCH_CHARACTERISTICS: {
  337. uint64_t cf = (inbound) ? ZT_RULE_PACKET_CHARACTERISTICS_INBOUND : 0ULL;
  338. if (macDest.isMulticast()) cf |= ZT_RULE_PACKET_CHARACTERISTICS_MULTICAST;
  339. if (macDest.isBroadcast()) cf |= ZT_RULE_PACKET_CHARACTERISTICS_BROADCAST;
  340. if (ownershipVerificationMask == 1) {
  341. ownershipVerificationMask = 0;
  342. InetAddress src;
  343. if ((etherType == ZT_ETHERTYPE_IPV4)&&(frameLen >= 20)) {
  344. src.set((const void *)(frameData + 12),4,0);
  345. } else if ((etherType == ZT_ETHERTYPE_IPV6)&&(frameLen >= 40)) {
  346. // IPv6 NDP requires special handling, since the src and dest IPs in the packet are empty or link-local.
  347. if ( (frameLen >= (40 + 8 + 16)) && (frameData[6] == 0x3a) && ((frameData[40] == 0x87)||(frameData[40] == 0x88)) ) {
  348. if (frameData[40] == 0x87) {
  349. // Neighbor solicitations contain no reliable source address, so we implement a small
  350. // hack by considering them authenticated. Otherwise you would pretty much have to do
  351. // this manually in the rule set for IPv6 to work at all.
  352. ownershipVerificationMask |= ZT_RULE_PACKET_CHARACTERISTICS_SENDER_IP_AUTHENTICATED;
  353. } else {
  354. // Neighbor advertisements on the other hand can absolutely be authenticated.
  355. src.set((const void *)(frameData + 40 + 8),16,0);
  356. }
  357. } else {
  358. // Other IPv6 packets can be handled normally
  359. src.set((const void *)(frameData + 8),16,0);
  360. }
  361. } else if ((etherType == ZT_ETHERTYPE_ARP)&&(frameLen >= 28)) {
  362. src.set((const void *)(frameData + 14),4,0);
  363. }
  364. if (inbound) {
  365. if (membership) {
  366. if ((src)&&(membership->peerOwnsAddress<InetAddress>(nconf,src)))
  367. ownershipVerificationMask |= ZT_RULE_PACKET_CHARACTERISTICS_SENDER_IP_AUTHENTICATED;
  368. if (membership->peerOwnsAddress<MAC>(nconf,macSource))
  369. ownershipVerificationMask |= ZT_RULE_PACKET_CHARACTERISTICS_SENDER_MAC_AUTHENTICATED;
  370. }
  371. } else {
  372. for(unsigned int i=0;i<nconf.certificateOfOwnershipCount;++i) {
  373. if ((src)&&(nconf.certificatesOfOwnership[i].owns(src)))
  374. ownershipVerificationMask |= ZT_RULE_PACKET_CHARACTERISTICS_SENDER_IP_AUTHENTICATED;
  375. if (nconf.certificatesOfOwnership[i].owns(macSource))
  376. ownershipVerificationMask |= ZT_RULE_PACKET_CHARACTERISTICS_SENDER_MAC_AUTHENTICATED;
  377. }
  378. }
  379. }
  380. cf |= ownershipVerificationMask;
  381. if ((etherType == ZT_ETHERTYPE_IPV4)&&(frameLen >= 20)&&(frameData[9] == 0x06)) {
  382. const unsigned int headerLen = 4 * (frameData[0] & 0xf);
  383. cf |= (uint64_t)frameData[headerLen + 13];
  384. cf |= (((uint64_t)(frameData[headerLen + 12] & 0x0f)) << 8);
  385. } else if (etherType == ZT_ETHERTYPE_IPV6) {
  386. unsigned int pos = 0,proto = 0;
  387. if (_ipv6GetPayload(frameData,frameLen,pos,proto)) {
  388. if ((proto == 0x06)&&(frameLen > (pos + 14))) {
  389. cf |= (uint64_t)frameData[pos + 13];
  390. cf |= (((uint64_t)(frameData[pos + 12] & 0x0f)) << 8);
  391. }
  392. }
  393. }
  394. thisRuleMatches = (uint8_t)((cf & rules[rn].v.characteristics) != 0);
  395. } break;
  396. case ZT_NETWORK_RULE_MATCH_FRAME_SIZE_RANGE:
  397. thisRuleMatches = (uint8_t)((frameLen >= (unsigned int)rules[rn].v.frameSize[0])&&(frameLen <= (unsigned int)rules[rn].v.frameSize[1]));
  398. break;
  399. case ZT_NETWORK_RULE_MATCH_RANDOM:
  400. thisRuleMatches = (uint8_t)((uint32_t)(Utils::random() & 0xffffffffULL) <= rules[rn].v.randomProbability);
  401. break;
  402. case ZT_NETWORK_RULE_MATCH_TAGS_DIFFERENCE:
  403. case ZT_NETWORK_RULE_MATCH_TAGS_BITWISE_AND:
  404. case ZT_NETWORK_RULE_MATCH_TAGS_BITWISE_OR:
  405. case ZT_NETWORK_RULE_MATCH_TAGS_BITWISE_XOR:
  406. case ZT_NETWORK_RULE_MATCH_TAGS_EQUAL: {
  407. const Tag *const localTag = std::lower_bound(&(nconf.tags[0]),&(nconf.tags[nconf.tagCount]),rules[rn].v.tag.id,Tag::IdComparePredicate());
  408. if ((localTag != &(nconf.tags[nconf.tagCount]))&&(localTag->id() == rules[rn].v.tag.id)) {
  409. const Tag *const remoteTag = ((membership) ? membership->getTag(nconf,rules[rn].v.tag.id) : (const Tag *)0);
  410. if (remoteTag) {
  411. const uint32_t ltv = localTag->value();
  412. const uint32_t rtv = remoteTag->value();
  413. if (rt == ZT_NETWORK_RULE_MATCH_TAGS_DIFFERENCE) {
  414. const uint32_t diff = (ltv > rtv) ? (ltv - rtv) : (rtv - ltv);
  415. thisRuleMatches = (uint8_t)(diff <= rules[rn].v.tag.value);
  416. } else if (rt == ZT_NETWORK_RULE_MATCH_TAGS_BITWISE_AND) {
  417. thisRuleMatches = (uint8_t)((ltv & rtv) == rules[rn].v.tag.value);
  418. } else if (rt == ZT_NETWORK_RULE_MATCH_TAGS_BITWISE_OR) {
  419. thisRuleMatches = (uint8_t)((ltv | rtv) == rules[rn].v.tag.value);
  420. } else if (rt == ZT_NETWORK_RULE_MATCH_TAGS_BITWISE_XOR) {
  421. thisRuleMatches = (uint8_t)((ltv ^ rtv) == rules[rn].v.tag.value);
  422. } else if (rt == ZT_NETWORK_RULE_MATCH_TAGS_EQUAL) {
  423. thisRuleMatches = (uint8_t)((ltv == rules[rn].v.tag.value)&&(rtv == rules[rn].v.tag.value));
  424. } else { // sanity check, can't really happen
  425. thisRuleMatches = 0;
  426. }
  427. } else {
  428. if ((inbound)&&(!superAccept)) {
  429. thisRuleMatches = 0;
  430. } else {
  431. // Outbound side is not strict since if we have to match both tags and
  432. // we are sending a first packet to a recipient, we probably do not know
  433. // about their tags yet. They will filter on inbound and we will filter
  434. // once we get their tag. If we are a tee/redirect target we are also
  435. // not strict since we likely do not have these tags.
  436. thisRuleMatches = 1;
  437. }
  438. }
  439. } else {
  440. thisRuleMatches = 0;
  441. }
  442. } break;
  443. case ZT_NETWORK_RULE_MATCH_TAG_SENDER:
  444. case ZT_NETWORK_RULE_MATCH_TAG_RECEIVER: {
  445. if (superAccept) {
  446. thisRuleMatches = 1;
  447. } else if ( ((rt == ZT_NETWORK_RULE_MATCH_TAG_SENDER)&&(inbound)) || ((rt == ZT_NETWORK_RULE_MATCH_TAG_RECEIVER)&&(!inbound)) ) {
  448. const Tag *const remoteTag = ((membership) ? membership->getTag(nconf,rules[rn].v.tag.id) : (const Tag *)0);
  449. if (remoteTag) {
  450. thisRuleMatches = (uint8_t)(remoteTag->value() == rules[rn].v.tag.value);
  451. } else {
  452. if (rt == ZT_NETWORK_RULE_MATCH_TAG_RECEIVER) {
  453. // If we are checking the receiver and this is an outbound packet, we
  454. // can't be strict since we may not yet know the receiver's tag.
  455. thisRuleMatches = 1;
  456. } else {
  457. thisRuleMatches = 0;
  458. }
  459. }
  460. } else { // sender and outbound or receiver and inbound
  461. const Tag *const localTag = std::lower_bound(&(nconf.tags[0]),&(nconf.tags[nconf.tagCount]),rules[rn].v.tag.id,Tag::IdComparePredicate());
  462. if ((localTag != &(nconf.tags[nconf.tagCount]))&&(localTag->id() == rules[rn].v.tag.id)) {
  463. thisRuleMatches = (uint8_t)(localTag->value() == rules[rn].v.tag.value);
  464. } else {
  465. thisRuleMatches = 0;
  466. }
  467. }
  468. } break;
  469. case ZT_NETWORK_RULE_MATCH_INTEGER_RANGE: {
  470. uint64_t integer = 0;
  471. const unsigned int bits = (rules[rn].v.intRange.format & 63) + 1;
  472. const unsigned int bytes = ((bits + 8 - 1) / 8); // integer ceiling of division by 8
  473. if ((rules[rn].v.intRange.format & 0x80) == 0) {
  474. // Big-endian
  475. unsigned int idx = rules[rn].v.intRange.idx + (8 - bytes);
  476. const unsigned int eof = idx + bytes;
  477. if (eof <= frameLen) {
  478. while (idx < eof) {
  479. integer <<= 8;
  480. integer |= frameData[idx++];
  481. }
  482. }
  483. integer &= 0xffffffffffffffffULL >> (64 - bits);
  484. } else {
  485. // Little-endian
  486. unsigned int idx = rules[rn].v.intRange.idx;
  487. const unsigned int eof = idx + bytes;
  488. if (eof <= frameLen) {
  489. while (idx < eof) {
  490. integer >>= 8;
  491. integer |= ((uint64_t)frameData[idx++]) << 56;
  492. }
  493. }
  494. integer >>= (64 - bits);
  495. }
  496. thisRuleMatches = (uint8_t)((integer >= rules[rn].v.intRange.start)&&(integer <= (rules[rn].v.intRange.start + (uint64_t)rules[rn].v.intRange.end)));
  497. } break;
  498. // The result of an unsupported MATCH is configurable at the network
  499. // level via a flag.
  500. default:
  501. thisRuleMatches = (uint8_t)((nconf.flags & ZT_NETWORKCONFIG_FLAG_RULES_RESULT_OF_UNSUPPORTED_MATCH) != 0);
  502. break;
  503. }
  504. rrl.log(rn,thisRuleMatches,thisSetMatches);
  505. if ((rules[rn].t & 0x40))
  506. thisSetMatches |= (thisRuleMatches ^ ((rules[rn].t >> 7) & 1));
  507. else thisSetMatches &= (thisRuleMatches ^ ((rules[rn].t >> 7) & 1));
  508. }
  509. return DOZTFILTER_NO_MATCH;
  510. }
  511. } // anonymous namespace
  512. const ZeroTier::MulticastGroup Network::BROADCAST(ZeroTier::MAC(0xffffffffffffULL),0);
  513. Network::Network(const RuntimeEnvironment *renv,void *tPtr,uint64_t nwid,void *uptr,const NetworkConfig *nconf) :
  514. RR(renv),
  515. _uPtr(uptr),
  516. _id(nwid),
  517. _mac(renv->identity.address(),nwid),
  518. _portInitialized(false),
  519. _lastConfigUpdate(0),
  520. _destroyed(false),
  521. _netconfFailure(NETCONF_FAILURE_NONE),
  522. _portError(0)
  523. {
  524. for(int i=0;i<ZT_NETWORK_MAX_INCOMING_UPDATES;++i)
  525. _incomingConfigChunks[i].ts = 0;
  526. if (nconf) {
  527. this->setConfiguration(tPtr,*nconf,false);
  528. _lastConfigUpdate = 0; // still want to re-request since it's likely outdated
  529. } else {
  530. uint64_t tmp[2];
  531. tmp[0] = nwid; tmp[1] = 0;
  532. bool got = false;
  533. ScopedPtr< Dictionary<ZT_NETWORKCONFIG_DICT_CAPACITY> > dict(new Dictionary<ZT_NETWORKCONFIG_DICT_CAPACITY>());
  534. try {
  535. int n = RR->node->stateObjectGet(tPtr,ZT_STATE_OBJECT_NETWORK_CONFIG,tmp,dict->unsafeData(),ZT_NETWORKCONFIG_DICT_CAPACITY - 1);
  536. if (n > 1) {
  537. try {
  538. ScopedPtr<NetworkConfig> nconf(new NetworkConfig());
  539. if (nconf->fromDictionary(*dict)) {
  540. this->setConfiguration(tPtr,*nconf,false);
  541. _lastConfigUpdate = 0; // still want to re-request an update since it's likely outdated
  542. got = true;
  543. }
  544. } catch ( ... ) {}
  545. }
  546. } catch ( ... ) {}
  547. if (!got)
  548. RR->node->stateObjectPut(tPtr,ZT_STATE_OBJECT_NETWORK_CONFIG,tmp,"\n",1);
  549. }
  550. if (!_portInitialized) {
  551. ZT_VirtualNetworkConfig ctmp;
  552. _externalConfig(&ctmp);
  553. _portError = RR->node->configureVirtualNetworkPort(tPtr,_id,&_uPtr,ZT_VIRTUAL_NETWORK_CONFIG_OPERATION_UP,&ctmp);
  554. _portInitialized = true;
  555. }
  556. }
  557. Network::~Network()
  558. {
  559. _memberships_l.lock();
  560. _config_l.lock();
  561. _config_l.unlock();
  562. _memberships_l.unlock();
  563. ZT_VirtualNetworkConfig ctmp;
  564. _externalConfig(&ctmp);
  565. if (_destroyed) {
  566. // This is done in Node::leave() so we can pass tPtr properly
  567. //RR->node->configureVirtualNetworkPort((void *)0,_id,&_uPtr,ZT_VIRTUAL_NETWORK_CONFIG_OPERATION_DESTROY,&ctmp);
  568. } else {
  569. RR->node->configureVirtualNetworkPort((void *)0,_id,&_uPtr,ZT_VIRTUAL_NETWORK_CONFIG_OPERATION_DOWN,&ctmp);
  570. }
  571. }
  572. bool Network::filterOutgoingPacket(
  573. void *tPtr,
  574. const bool noTee,
  575. const Address &ztSource,
  576. const Address &ztDest,
  577. const MAC &macSource,
  578. const MAC &macDest,
  579. const uint8_t *frameData,
  580. const unsigned int frameLen,
  581. const unsigned int etherType,
  582. const unsigned int vlanId,
  583. uint8_t &qosBucket)
  584. {
  585. Address ztFinalDest(ztDest);
  586. int localCapabilityIndex = -1;
  587. int accept = 0;
  588. Trace::RuleResultLog rrl,crrl;
  589. Address cc;
  590. unsigned int ccLength = 0;
  591. bool ccWatch = false;
  592. Mutex::Lock l1(_memberships_l);
  593. Mutex::Lock l2(_config_l);
  594. Membership *const membership = (ztDest) ? _memberships.get(ztDest) : (Membership *)0;
  595. switch(_doZtFilter(RR,rrl,_config,membership,false,ztSource,ztFinalDest,macSource,macDest,frameData,frameLen,etherType,vlanId,_config.rules,_config.ruleCount,cc,ccLength,ccWatch,qosBucket)) {
  596. case DOZTFILTER_NO_MATCH: {
  597. for(unsigned int c=0;c<_config.capabilityCount;++c) {
  598. ztFinalDest = ztDest; // sanity check, shouldn't be possible if there was no match
  599. Address cc2;
  600. unsigned int ccLength2 = 0;
  601. bool ccWatch2 = false;
  602. switch (_doZtFilter(RR,crrl,_config,membership,false,ztSource,ztFinalDest,macSource,macDest,frameData,frameLen,etherType,vlanId,_config.capabilities[c].rules(),_config.capabilities[c].ruleCount(),cc2,ccLength2,ccWatch2,qosBucket)) {
  603. case DOZTFILTER_NO_MATCH:
  604. case DOZTFILTER_DROP: // explicit DROP in a capability just terminates its evaluation and is an anti-pattern
  605. break;
  606. case DOZTFILTER_REDIRECT: // interpreted as ACCEPT but ztFinalDest will have been changed in _doZtFilter()
  607. case DOZTFILTER_ACCEPT:
  608. case DOZTFILTER_SUPER_ACCEPT: // no difference in behavior on outbound side in capabilities
  609. localCapabilityIndex = (int)c;
  610. accept = 1;
  611. if ((!noTee)&&(cc2)) {
  612. Packet outp(cc2,RR->identity.address(),Packet::VERB_EXT_FRAME);
  613. outp.append(_id);
  614. outp.append((uint8_t)(ccWatch2 ? 0x16 : 0x02));
  615. macDest.appendTo(outp);
  616. macSource.appendTo(outp);
  617. outp.append((uint16_t)etherType);
  618. outp.append(frameData,ccLength2);
  619. outp.compress();
  620. RR->sw->send(tPtr,outp,true);
  621. }
  622. break;
  623. }
  624. if (accept)
  625. break;
  626. }
  627. } break;
  628. case DOZTFILTER_DROP:
  629. if (_config.remoteTraceTarget)
  630. RR->t->networkFilter(tPtr,*this,rrl,(Trace::RuleResultLog *)0,(Capability *)0,ztSource,ztDest,macSource,macDest,frameData,frameLen,etherType,vlanId,noTee,false,0);
  631. return false;
  632. case DOZTFILTER_REDIRECT: // interpreted as ACCEPT but ztFinalDest will have been changed in _doZtFilter()
  633. case DOZTFILTER_ACCEPT:
  634. accept = 1;
  635. break;
  636. case DOZTFILTER_SUPER_ACCEPT:
  637. accept = 2;
  638. break;
  639. }
  640. if (accept) {
  641. if (membership)
  642. membership->logSentBytes(frameLen);
  643. if ((!noTee)&&(cc)) {
  644. Packet outp(cc,RR->identity.address(),Packet::VERB_EXT_FRAME);
  645. outp.append(_id);
  646. outp.append((uint8_t)(ccWatch ? 0x16 : 0x02));
  647. macDest.appendTo(outp);
  648. macSource.appendTo(outp);
  649. outp.append((uint16_t)etherType);
  650. outp.append(frameData,ccLength);
  651. outp.compress();
  652. RR->sw->send(tPtr,outp,true);
  653. }
  654. if ((ztDest != ztFinalDest)&&(ztFinalDest)) {
  655. Packet outp(ztFinalDest,RR->identity.address(),Packet::VERB_EXT_FRAME);
  656. outp.append(_id);
  657. outp.append((uint8_t)0x04);
  658. macDest.appendTo(outp);
  659. macSource.appendTo(outp);
  660. outp.append((uint16_t)etherType);
  661. outp.append(frameData,frameLen);
  662. outp.compress();
  663. RR->sw->send(tPtr,outp,true);
  664. if (_config.remoteTraceTarget)
  665. RR->t->networkFilter(tPtr,*this,rrl,(localCapabilityIndex >= 0) ? &crrl : (Trace::RuleResultLog *)0,(localCapabilityIndex >= 0) ? &(_config.capabilities[localCapabilityIndex]) : (Capability *)0,ztSource,ztDest,macSource,macDest,frameData,frameLen,etherType,vlanId,noTee,false,0);
  666. return false; // DROP locally, since we redirected
  667. } else {
  668. if (_config.remoteTraceTarget)
  669. RR->t->networkFilter(tPtr,*this,rrl,(localCapabilityIndex >= 0) ? &crrl : (Trace::RuleResultLog *)0,(localCapabilityIndex >= 0) ? &(_config.capabilities[localCapabilityIndex]) : (Capability *)0,ztSource,ztDest,macSource,macDest,frameData,frameLen,etherType,vlanId,noTee,false,1);
  670. return true;
  671. }
  672. } else {
  673. if (_config.remoteTraceTarget)
  674. RR->t->networkFilter(tPtr,*this,rrl,(localCapabilityIndex >= 0) ? &crrl : (Trace::RuleResultLog *)0,(localCapabilityIndex >= 0) ? &(_config.capabilities[localCapabilityIndex]) : (Capability *)0,ztSource,ztDest,macSource,macDest,frameData,frameLen,etherType,vlanId,noTee,false,0);
  675. return false;
  676. }
  677. }
  678. int Network::filterIncomingPacket(
  679. void *tPtr,
  680. const SharedPtr<Peer> &sourcePeer,
  681. const Address &ztDest,
  682. const MAC &macSource,
  683. const MAC &macDest,
  684. const uint8_t *frameData,
  685. const unsigned int frameLen,
  686. const unsigned int etherType,
  687. const unsigned int vlanId)
  688. {
  689. Address ztFinalDest(ztDest);
  690. Trace::RuleResultLog rrl,crrl;
  691. int accept = 0;
  692. Address cc;
  693. unsigned int ccLength = 0;
  694. bool ccWatch = false;
  695. const Capability *c = (Capability *)0;
  696. uint8_t qosBucket = 255; // For incoming packets this is a dummy value
  697. Mutex::Lock l1(_memberships_l);
  698. Mutex::Lock l2(_config_l);
  699. Membership &membership = _memberships[sourcePeer->address()];
  700. switch (_doZtFilter(RR,rrl,_config,&membership,true,sourcePeer->address(),ztFinalDest,macSource,macDest,frameData,frameLen,etherType,vlanId,_config.rules,_config.ruleCount,cc,ccLength,ccWatch,qosBucket)) {
  701. case DOZTFILTER_NO_MATCH: {
  702. Membership::CapabilityIterator mci(membership,_config);
  703. while ((c = mci.next())) {
  704. ztFinalDest = ztDest; // sanity check, should be unmodified if there was no match
  705. Address cc2;
  706. unsigned int ccLength2 = 0;
  707. bool ccWatch2 = false;
  708. switch(_doZtFilter(RR,crrl,_config,&membership,true,sourcePeer->address(),ztFinalDest,macSource,macDest,frameData,frameLen,etherType,vlanId,c->rules(),c->ruleCount(),cc2,ccLength2,ccWatch2,qosBucket)) {
  709. case DOZTFILTER_NO_MATCH:
  710. case DOZTFILTER_DROP: // explicit DROP in a capability just terminates its evaluation and is an anti-pattern
  711. break;
  712. case DOZTFILTER_REDIRECT: // interpreted as ACCEPT but ztDest will have been changed in _doZtFilter()
  713. case DOZTFILTER_ACCEPT:
  714. accept = 1; // ACCEPT
  715. break;
  716. case DOZTFILTER_SUPER_ACCEPT:
  717. accept = 2; // super-ACCEPT
  718. break;
  719. }
  720. if (accept) {
  721. if (cc2) {
  722. Packet outp(cc2,RR->identity.address(),Packet::VERB_EXT_FRAME);
  723. outp.append(_id);
  724. outp.append((uint8_t)(ccWatch2 ? 0x1c : 0x08));
  725. macDest.appendTo(outp);
  726. macSource.appendTo(outp);
  727. outp.append((uint16_t)etherType);
  728. outp.append(frameData,ccLength2);
  729. outp.compress();
  730. RR->sw->send(tPtr,outp,true);
  731. }
  732. break;
  733. }
  734. }
  735. } break;
  736. case DOZTFILTER_DROP:
  737. if (_config.remoteTraceTarget)
  738. RR->t->networkFilter(tPtr,*this,rrl,(Trace::RuleResultLog *)0,(Capability *)0,sourcePeer->address(),ztDest,macSource,macDest,frameData,frameLen,etherType,vlanId,false,true,0);
  739. return 0; // DROP
  740. case DOZTFILTER_REDIRECT: // interpreted as ACCEPT but ztFinalDest will have been changed in _doZtFilter()
  741. case DOZTFILTER_ACCEPT:
  742. accept = 1; // ACCEPT
  743. break;
  744. case DOZTFILTER_SUPER_ACCEPT:
  745. accept = 2; // super-ACCEPT
  746. break;
  747. }
  748. if (accept) {
  749. membership.logReceivedBytes(frameLen);
  750. if (cc) {
  751. Packet outp(cc,RR->identity.address(),Packet::VERB_EXT_FRAME);
  752. outp.append(_id);
  753. outp.append((uint8_t)(ccWatch ? 0x1c : 0x08));
  754. macDest.appendTo(outp);
  755. macSource.appendTo(outp);
  756. outp.append((uint16_t)etherType);
  757. outp.append(frameData,ccLength);
  758. outp.compress();
  759. RR->sw->send(tPtr,outp,true);
  760. }
  761. if ((ztDest != ztFinalDest)&&(ztFinalDest)) {
  762. Packet outp(ztFinalDest,RR->identity.address(),Packet::VERB_EXT_FRAME);
  763. outp.append(_id);
  764. outp.append((uint8_t)0x0a);
  765. macDest.appendTo(outp);
  766. macSource.appendTo(outp);
  767. outp.append((uint16_t)etherType);
  768. outp.append(frameData,frameLen);
  769. outp.compress();
  770. RR->sw->send(tPtr,outp,true);
  771. if (_config.remoteTraceTarget)
  772. RR->t->networkFilter(tPtr,*this,rrl,(c) ? &crrl : (Trace::RuleResultLog *)0,c,sourcePeer->address(),ztDest,macSource,macDest,frameData,frameLen,etherType,vlanId,false,true,0);
  773. return 0; // DROP locally, since we redirected
  774. }
  775. }
  776. if (_config.remoteTraceTarget)
  777. RR->t->networkFilter(tPtr,*this,rrl,(c) ? &crrl : (Trace::RuleResultLog *)0,c,sourcePeer->address(),ztDest,macSource,macDest,frameData,frameLen,etherType,vlanId,false,true,accept);
  778. return accept;
  779. }
  780. uint64_t Network::handleConfigChunk(void *tPtr,const uint64_t packetId,const Address &source,const Buffer<ZT_PROTO_MAX_PACKET_LENGTH> &chunk,unsigned int ptr)
  781. {
  782. if (_destroyed)
  783. return 0;
  784. const unsigned int start = ptr;
  785. ptr += 8; // skip network ID, which is already obviously known
  786. const unsigned int chunkLen = chunk.at<uint16_t>(ptr); ptr += 2;
  787. const void *chunkData = chunk.field(ptr,chunkLen); ptr += chunkLen;
  788. uint64_t configUpdateId;
  789. {
  790. Mutex::Lock l1(_config_l);
  791. _IncomingConfigChunk *c = (_IncomingConfigChunk *)0;
  792. uint64_t chunkId = 0;
  793. unsigned long totalLength,chunkIndex;
  794. if (ptr < chunk.size()) {
  795. const bool fastPropagate = ((chunk[ptr++] & 0x01) != 0);
  796. configUpdateId = chunk.at<uint64_t>(ptr); ptr += 8;
  797. totalLength = chunk.at<uint32_t>(ptr); ptr += 4;
  798. chunkIndex = chunk.at<uint32_t>(ptr); ptr += 4;
  799. if (((chunkIndex + chunkLen) > totalLength)||(totalLength >= ZT_NETWORKCONFIG_DICT_CAPACITY)) // >= since we need room for a null at the end
  800. return 0;
  801. if ((chunk[ptr] != 1)||(chunk.at<uint16_t>(ptr + 1) != ZT_C25519_SIGNATURE_LEN))
  802. return 0;
  803. const uint8_t *sig = reinterpret_cast<const uint8_t *>(chunk.field(ptr + 3,ZT_C25519_SIGNATURE_LEN));
  804. // We can use the signature, which is unique per chunk, to get a per-chunk ID for local deduplication use
  805. for(unsigned int i=0;i<16;++i)
  806. reinterpret_cast<uint8_t *>(&chunkId)[i & 7] ^= sig[i];
  807. // Find existing or new slot for this update and check if this is a duplicate chunk
  808. for(int i=0;i<ZT_NETWORK_MAX_INCOMING_UPDATES;++i) {
  809. if (_incomingConfigChunks[i].updateId == configUpdateId) {
  810. c = &(_incomingConfigChunks[i]);
  811. for(unsigned long j=0;j<c->haveChunks;++j) {
  812. if (c->haveChunkIds[j] == chunkId)
  813. return 0;
  814. }
  815. break;
  816. } else if ((!c)||(_incomingConfigChunks[i].ts < c->ts)) {
  817. c = &(_incomingConfigChunks[i]);
  818. }
  819. }
  820. // If it's not a duplicate, check chunk signature
  821. const Identity controllerId(RR->topology->getIdentity(tPtr,controller()));
  822. if (!controllerId) // we should always have the controller identity by now, otherwise how would we have queried it the first time?
  823. return 0;
  824. if (!controllerId.verify(chunk.field(start,ptr - start),ptr - start,sig,ZT_C25519_SIGNATURE_LEN))
  825. return 0;
  826. // New properly verified chunks can be flooded "virally" through the network
  827. if (fastPropagate) {
  828. Mutex::Lock l2(_memberships_l);
  829. Address *a = (Address *)0;
  830. Membership *m = (Membership *)0;
  831. Hashtable<Address,Membership>::Iterator i(_memberships);
  832. while (i.next(a,m)) {
  833. if ((*a != source)&&(*a != controller())) {
  834. Packet outp(*a,RR->identity.address(),Packet::VERB_NETWORK_CONFIG);
  835. outp.append(reinterpret_cast<const uint8_t *>(chunk.data()) + start,chunk.size() - start);
  836. RR->sw->send(tPtr,outp,true);
  837. }
  838. }
  839. }
  840. } else if ((source == controller())||(!source)) { // since old chunks aren't signed, only accept from controller itself (or via cluster backplane)
  841. // Legacy support for OK(NETWORK_CONFIG_REQUEST) from older controllers
  842. chunkId = packetId;
  843. configUpdateId = chunkId;
  844. totalLength = chunkLen;
  845. chunkIndex = 0;
  846. if (totalLength >= ZT_NETWORKCONFIG_DICT_CAPACITY)
  847. return 0;
  848. for(int i=0;i<ZT_NETWORK_MAX_INCOMING_UPDATES;++i) {
  849. if ((!c)||(_incomingConfigChunks[i].ts < c->ts))
  850. c = &(_incomingConfigChunks[i]);
  851. }
  852. } else {
  853. // Single-chunk unsigned legacy configs are only allowed from the controller itself
  854. return 0;
  855. }
  856. ++c->ts; // newer is higher, that's all we need
  857. if (c->updateId != configUpdateId) {
  858. c->updateId = configUpdateId;
  859. c->haveChunks = 0;
  860. c->haveBytes = 0;
  861. }
  862. if (c->haveChunks >= ZT_NETWORK_MAX_UPDATE_CHUNKS)
  863. return false;
  864. c->haveChunkIds[c->haveChunks++] = chunkId;
  865. memcpy(c->data.unsafeData() + chunkIndex,chunkData,chunkLen);
  866. c->haveBytes += chunkLen;
  867. if (c->haveBytes == totalLength) {
  868. c->data.unsafeData()[c->haveBytes] = (char)0; // ensure null terminated
  869. ScopedPtr<NetworkConfig> nc(new NetworkConfig());
  870. try {
  871. if (nc->fromDictionary(c->data)) {
  872. this->setConfiguration(tPtr,*nc,true);
  873. return configUpdateId;
  874. }
  875. } catch ( ... ) {}
  876. }
  877. }
  878. return 0;
  879. }
  880. int Network::setConfiguration(void *tPtr,const NetworkConfig &nconf,bool saveToDisk)
  881. {
  882. if (_destroyed)
  883. return 0;
  884. // _lock is NOT locked when this is called
  885. try {
  886. if ((nconf.issuedTo != RR->identity.address())||(nconf.networkId != _id))
  887. return 0; // invalid config that is not for us or not for this network
  888. if (_config == nconf)
  889. return 1; // OK config, but duplicate of what we already have
  890. ZT_VirtualNetworkConfig ctmp;
  891. bool oldPortInitialized;
  892. { // do things that require lock here, but unlock before calling callbacks
  893. Mutex::Lock l1(_config_l);
  894. _config = nconf;
  895. _lastConfigUpdate = RR->node->now();
  896. _netconfFailure = NETCONF_FAILURE_NONE;
  897. oldPortInitialized = _portInitialized;
  898. _portInitialized = true;
  899. _externalConfig(&ctmp);
  900. }
  901. _portError = RR->node->configureVirtualNetworkPort(tPtr,_id,&_uPtr,(oldPortInitialized) ? ZT_VIRTUAL_NETWORK_CONFIG_OPERATION_CONFIG_UPDATE : ZT_VIRTUAL_NETWORK_CONFIG_OPERATION_UP,&ctmp);
  902. if (saveToDisk) {
  903. try {
  904. ScopedPtr< Dictionary<ZT_NETWORKCONFIG_DICT_CAPACITY> > d(new Dictionary<ZT_NETWORKCONFIG_DICT_CAPACITY>());
  905. if (nconf.toDictionary(*d,false)) {
  906. uint64_t tmp[2];
  907. tmp[0] = _id; tmp[1] = 0;
  908. RR->node->stateObjectPut(tPtr,ZT_STATE_OBJECT_NETWORK_CONFIG,tmp,d->data(),d->sizeBytes());
  909. }
  910. } catch ( ... ) {}
  911. }
  912. return 2; // OK and configuration has changed
  913. } catch ( ... ) {} // ignore invalid configs
  914. return 0;
  915. }
  916. bool Network::gate(void *tPtr,const SharedPtr<Peer> &peer)
  917. {
  918. const int64_t now = RR->node->now();
  919. Mutex::Lock l(_memberships_l);
  920. try {
  921. if (_config) {
  922. Membership *m = _memberships.get(peer->address());
  923. if ( (_config.isPublic()) || ((m)&&(m->isAllowedOnNetwork(_config))) ) {
  924. if (!m)
  925. m = &(_memberships[peer->address()]);
  926. if (m->multicastLikeGate(now)) {
  927. Mutex::Lock l2(_myMulticastGroups_l);
  928. _announceMulticastGroupsTo(tPtr,peer->address(),_allMulticastGroups());
  929. }
  930. return true;
  931. }
  932. }
  933. } catch ( ... ) {}
  934. return false;
  935. }
  936. void Network::doPeriodicTasks(void *tPtr,const int64_t now)
  937. {
  938. if (_destroyed)
  939. return;
  940. if ((now - _lastConfigUpdate) >= ZT_NETWORK_AUTOCONF_DELAY)
  941. _requestConfiguration(tPtr);
  942. {
  943. Mutex::Lock l1(_memberships_l);
  944. {
  945. Address *a = (Address *)0;
  946. Membership *m = (Membership *)0;
  947. Hashtable<Address,Membership>::Iterator i(_memberships);
  948. while (i.next(a,m))
  949. m->clean(now,_config);
  950. }
  951. {
  952. Mutex::Lock l2(_myMulticastGroups_l);
  953. Hashtable< MulticastGroup,uint64_t >::Iterator i(_multicastGroupsBehindMe);
  954. MulticastGroup *mg = (MulticastGroup *)0;
  955. uint64_t *ts = (uint64_t *)0;
  956. while (i.next(mg,ts)) {
  957. if ((now - *ts) > (ZT_MULTICAST_LIKE_EXPIRE * 2))
  958. _multicastGroupsBehindMe.erase(*mg);
  959. }
  960. _announceMulticastGroups(tPtr,false);
  961. }
  962. }
  963. }
  964. void Network::learnBridgeRoute(const MAC &mac,const Address &addr)
  965. {
  966. Mutex::Lock _l(_remoteBridgeRoutes_l);
  967. _remoteBridgeRoutes[mac] = addr;
  968. // Anti-DOS circuit breaker to prevent nodes from spamming us with absurd numbers of bridge routes
  969. while (_remoteBridgeRoutes.size() > ZT_MAX_BRIDGE_ROUTES) {
  970. Hashtable< Address,unsigned long > counts;
  971. Address maxAddr;
  972. unsigned long maxCount = 0;
  973. MAC *m = (MAC *)0;
  974. Address *a = (Address *)0;
  975. // Find the address responsible for the most entries
  976. {
  977. Hashtable<MAC,Address>::Iterator i(_remoteBridgeRoutes);
  978. while (i.next(m,a)) {
  979. const unsigned long c = ++counts[*a];
  980. if (c > maxCount) {
  981. maxCount = c;
  982. maxAddr = *a;
  983. }
  984. }
  985. }
  986. // Kill this address from our table, since it's most likely spamming us
  987. {
  988. Hashtable<MAC,Address>::Iterator i(_remoteBridgeRoutes);
  989. while (i.next(m,a)) {
  990. if (*a == maxAddr)
  991. _remoteBridgeRoutes.erase(*m);
  992. }
  993. }
  994. }
  995. }
  996. Membership::AddCredentialResult Network::addCredential(void *tPtr,const Address &sentFrom,const Revocation &rev)
  997. {
  998. if (rev.networkId() != _id)
  999. return Membership::ADD_REJECTED;
  1000. Mutex::Lock l1(_memberships_l);
  1001. Membership &m = _memberships[rev.target()];
  1002. const Membership::AddCredentialResult result = m.addCredential(RR,tPtr,_config,rev);
  1003. if ((result == Membership::ADD_ACCEPTED_NEW)&&(rev.fastPropagate())) {
  1004. Address *a = (Address *)0;
  1005. Membership *m = (Membership *)0;
  1006. Hashtable<Address,Membership>::Iterator i(_memberships);
  1007. while (i.next(a,m)) {
  1008. if ((*a != sentFrom)&&(*a != rev.signer())) {
  1009. Packet outp(*a,RR->identity.address(),Packet::VERB_NETWORK_CREDENTIALS);
  1010. outp.append((uint8_t)0x00); // no COM
  1011. outp.append((uint16_t)0); // no capabilities
  1012. outp.append((uint16_t)0); // no tags
  1013. outp.append((uint16_t)1); // one revocation!
  1014. rev.serialize(outp);
  1015. outp.append((uint16_t)0); // no certificates of ownership
  1016. RR->sw->send(tPtr,outp,true);
  1017. }
  1018. }
  1019. }
  1020. return result;
  1021. }
  1022. void Network::_requestConfiguration(void *tPtr)
  1023. {
  1024. if (_destroyed)
  1025. return;
  1026. if ((_id >> 56) == 0xff) {
  1027. if ((_id & 0xffffff) == 0) {
  1028. const uint16_t startPortRange = (uint16_t)((_id >> 40) & 0xffff);
  1029. const uint16_t endPortRange = (uint16_t)((_id >> 24) & 0xffff);
  1030. if (endPortRange >= startPortRange) {
  1031. ScopedPtr<NetworkConfig> nconf(new NetworkConfig());
  1032. nconf->networkId = _id;
  1033. nconf->timestamp = RR->node->now();
  1034. nconf->credentialTimeMaxDelta = ZT_NETWORKCONFIG_DEFAULT_CREDENTIAL_TIME_MAX_MAX_DELTA;
  1035. nconf->revision = 1;
  1036. nconf->issuedTo = RR->identity.address();
  1037. nconf->flags = ZT_NETWORKCONFIG_FLAG_ENABLE_IPV6_NDP_EMULATION;
  1038. nconf->mtu = ZT_DEFAULT_MTU;
  1039. nconf->multicastLimit = 0;
  1040. nconf->staticIpCount = 1;
  1041. nconf->ruleCount = 14;
  1042. nconf->staticIps[0] = InetAddress::makeIpv66plane(_id,RR->identity.address().toInt());
  1043. // Drop everything but IPv6
  1044. nconf->rules[0].t = (uint8_t)ZT_NETWORK_RULE_MATCH_ETHERTYPE | 0x80; // NOT
  1045. nconf->rules[0].v.etherType = 0x86dd; // IPv6
  1046. nconf->rules[1].t = (uint8_t)ZT_NETWORK_RULE_ACTION_DROP;
  1047. // Allow ICMPv6
  1048. nconf->rules[2].t = (uint8_t)ZT_NETWORK_RULE_MATCH_IP_PROTOCOL;
  1049. nconf->rules[2].v.ipProtocol = 0x3a; // ICMPv6
  1050. nconf->rules[3].t = (uint8_t)ZT_NETWORK_RULE_ACTION_ACCEPT;
  1051. // Allow destination ports within range
  1052. nconf->rules[4].t = (uint8_t)ZT_NETWORK_RULE_MATCH_IP_PROTOCOL;
  1053. nconf->rules[4].v.ipProtocol = 0x11; // UDP
  1054. nconf->rules[5].t = (uint8_t)ZT_NETWORK_RULE_MATCH_IP_PROTOCOL | 0x40; // OR
  1055. nconf->rules[5].v.ipProtocol = 0x06; // TCP
  1056. nconf->rules[6].t = (uint8_t)ZT_NETWORK_RULE_MATCH_IP_DEST_PORT_RANGE;
  1057. nconf->rules[6].v.port[0] = startPortRange;
  1058. nconf->rules[6].v.port[1] = endPortRange;
  1059. nconf->rules[7].t = (uint8_t)ZT_NETWORK_RULE_ACTION_ACCEPT;
  1060. // Allow non-SYN TCP packets to permit non-connection-initiating traffic
  1061. nconf->rules[8].t = (uint8_t)ZT_NETWORK_RULE_MATCH_CHARACTERISTICS | 0x80; // NOT
  1062. nconf->rules[8].v.characteristics = ZT_RULE_PACKET_CHARACTERISTICS_TCP_SYN;
  1063. nconf->rules[9].t = (uint8_t)ZT_NETWORK_RULE_ACTION_ACCEPT;
  1064. // Also allow SYN+ACK which are replies to SYN
  1065. nconf->rules[10].t = (uint8_t)ZT_NETWORK_RULE_MATCH_CHARACTERISTICS;
  1066. nconf->rules[10].v.characteristics = ZT_RULE_PACKET_CHARACTERISTICS_TCP_SYN;
  1067. nconf->rules[11].t = (uint8_t)ZT_NETWORK_RULE_MATCH_CHARACTERISTICS;
  1068. nconf->rules[11].v.characteristics = ZT_RULE_PACKET_CHARACTERISTICS_TCP_ACK;
  1069. nconf->rules[12].t = (uint8_t)ZT_NETWORK_RULE_ACTION_ACCEPT;
  1070. nconf->rules[13].t = (uint8_t)ZT_NETWORK_RULE_ACTION_DROP;
  1071. nconf->type = ZT_NETWORK_TYPE_PUBLIC;
  1072. nconf->name[0] = 'a';
  1073. nconf->name[1] = 'd';
  1074. nconf->name[2] = 'h';
  1075. nconf->name[3] = 'o';
  1076. nconf->name[4] = 'c';
  1077. nconf->name[5] = '-';
  1078. Utils::hex((uint16_t)startPortRange,nconf->name + 6);
  1079. nconf->name[10] = '-';
  1080. Utils::hex((uint16_t)endPortRange,nconf->name + 11);
  1081. nconf->name[15] = (char)0;
  1082. this->setConfiguration(tPtr,*nconf,false);
  1083. } else {
  1084. this->setNotFound();
  1085. }
  1086. } else if ((_id & 0xff) == 0x01) {
  1087. // ffAAaaaaaaaaaa01 -- where AA is the IPv4 /8 to use and aaaaaaaaaa is the anchor node for multicast gather and replication
  1088. const uint64_t myAddress = RR->identity.address().toInt();
  1089. const uint64_t networkHub = (_id >> 8) & 0xffffffffffULL;
  1090. uint8_t ipv4[4];
  1091. ipv4[0] = (uint8_t)((_id >> 48) & 0xff);
  1092. ipv4[1] = (uint8_t)((myAddress >> 16) & 0xff);
  1093. ipv4[2] = (uint8_t)((myAddress >> 8) & 0xff);
  1094. ipv4[3] = (uint8_t)(myAddress & 0xff);
  1095. char v4ascii[24];
  1096. Utils::decimal(ipv4[0],v4ascii);
  1097. ScopedPtr<NetworkConfig> nconf(new NetworkConfig());
  1098. nconf->networkId = _id;
  1099. nconf->timestamp = RR->node->now();
  1100. nconf->credentialTimeMaxDelta = ZT_NETWORKCONFIG_DEFAULT_CREDENTIAL_TIME_MAX_MAX_DELTA;
  1101. nconf->revision = 1;
  1102. nconf->issuedTo = RR->identity.address();
  1103. nconf->flags = ZT_NETWORKCONFIG_FLAG_ENABLE_IPV6_NDP_EMULATION;
  1104. nconf->mtu = ZT_DEFAULT_MTU;
  1105. nconf->multicastLimit = 1024;
  1106. nconf->specialistCount = (networkHub == 0) ? 0 : 1;
  1107. nconf->staticIpCount = 2;
  1108. nconf->ruleCount = 1;
  1109. if (networkHub != 0)
  1110. nconf->specialists[0] = networkHub;
  1111. nconf->staticIps[0] = InetAddress::makeIpv66plane(_id,myAddress);
  1112. nconf->staticIps[1].set(ipv4,4,8);
  1113. nconf->rules[0].t = (uint8_t)ZT_NETWORK_RULE_ACTION_ACCEPT;
  1114. nconf->type = ZT_NETWORK_TYPE_PUBLIC;
  1115. nconf->name[0] = 'a';
  1116. nconf->name[1] = 'd';
  1117. nconf->name[2] = 'h';
  1118. nconf->name[3] = 'o';
  1119. nconf->name[4] = 'c';
  1120. nconf->name[5] = '-';
  1121. unsigned long nn = 6;
  1122. while ((nconf->name[nn] = v4ascii[nn - 6])) ++nn;
  1123. nconf->name[nn++] = '.';
  1124. nconf->name[nn++] = '0';
  1125. nconf->name[nn++] = '.';
  1126. nconf->name[nn++] = '0';
  1127. nconf->name[nn++] = '.';
  1128. nconf->name[nn++] = '0';
  1129. nconf->name[nn++] = (char)0;
  1130. this->setConfiguration(tPtr,*nconf,false);
  1131. }
  1132. return;
  1133. }
  1134. const Address ctrl(controller());
  1135. ScopedPtr< Dictionary<ZT_NETWORKCONFIG_METADATA_DICT_CAPACITY> > rmd(new Dictionary<ZT_NETWORKCONFIG_METADATA_DICT_CAPACITY>());
  1136. rmd->add(ZT_NETWORKCONFIG_REQUEST_METADATA_KEY_NODE_VENDOR,(uint64_t)ZT_VENDOR_ZEROTIER);
  1137. rmd->add(ZT_NETWORKCONFIG_REQUEST_METADATA_KEY_PROTOCOL_VERSION,(uint64_t)ZT_PROTO_VERSION);
  1138. rmd->add(ZT_NETWORKCONFIG_REQUEST_METADATA_KEY_NODE_MAJOR_VERSION,(uint64_t)ZEROTIER_ONE_VERSION_MAJOR);
  1139. rmd->add(ZT_NETWORKCONFIG_REQUEST_METADATA_KEY_NODE_MINOR_VERSION,(uint64_t)ZEROTIER_ONE_VERSION_MINOR);
  1140. rmd->add(ZT_NETWORKCONFIG_REQUEST_METADATA_KEY_NODE_REVISION,(uint64_t)ZEROTIER_ONE_VERSION_REVISION);
  1141. rmd->add(ZT_NETWORKCONFIG_REQUEST_METADATA_KEY_MAX_NETWORK_RULES,(uint64_t)ZT_MAX_NETWORK_RULES);
  1142. rmd->add(ZT_NETWORKCONFIG_REQUEST_METADATA_KEY_MAX_NETWORK_CAPABILITIES,(uint64_t)ZT_MAX_NETWORK_CAPABILITIES);
  1143. rmd->add(ZT_NETWORKCONFIG_REQUEST_METADATA_KEY_MAX_CAPABILITY_RULES,(uint64_t)ZT_MAX_CAPABILITY_RULES);
  1144. rmd->add(ZT_NETWORKCONFIG_REQUEST_METADATA_KEY_MAX_NETWORK_TAGS,(uint64_t)ZT_MAX_NETWORK_TAGS);
  1145. rmd->add(ZT_NETWORKCONFIG_REQUEST_METADATA_KEY_FLAGS,(uint64_t)0);
  1146. rmd->add(ZT_NETWORKCONFIG_REQUEST_METADATA_KEY_RULES_ENGINE_REV,(uint64_t)ZT_RULES_ENGINE_REVISION);
  1147. RR->t->networkConfigRequestSent(tPtr,*this,ctrl);
  1148. if (ctrl == RR->identity.address()) {
  1149. if (RR->localNetworkController) {
  1150. RR->localNetworkController->request(_id,InetAddress(),0xffffffffffffffffULL,RR->identity,*rmd);
  1151. } else {
  1152. this->setNotFound();
  1153. }
  1154. return;
  1155. }
  1156. Packet outp(ctrl,RR->identity.address(),Packet::VERB_NETWORK_CONFIG_REQUEST);
  1157. outp.append((uint64_t)_id);
  1158. const unsigned int rmdSize = rmd->sizeBytes();
  1159. outp.append((uint16_t)rmdSize);
  1160. outp.append((const void *)rmd->data(),rmdSize);
  1161. if (_config) {
  1162. outp.append((uint64_t)_config.revision);
  1163. outp.append((uint64_t)_config.timestamp);
  1164. } else {
  1165. outp.append((unsigned char)0,16);
  1166. }
  1167. outp.compress();
  1168. RR->node->expectReplyTo(outp.packetId());
  1169. RR->sw->send(tPtr,outp,true);
  1170. }
  1171. ZT_VirtualNetworkStatus Network::_status() const
  1172. {
  1173. if (_portError)
  1174. return ZT_NETWORK_STATUS_PORT_ERROR;
  1175. switch(_netconfFailure) {
  1176. case NETCONF_FAILURE_ACCESS_DENIED:
  1177. return ZT_NETWORK_STATUS_ACCESS_DENIED;
  1178. case NETCONF_FAILURE_NOT_FOUND:
  1179. return ZT_NETWORK_STATUS_NOT_FOUND;
  1180. case NETCONF_FAILURE_NONE:
  1181. return ((_config) ? ZT_NETWORK_STATUS_OK : ZT_NETWORK_STATUS_REQUESTING_CONFIGURATION);
  1182. default:
  1183. return ZT_NETWORK_STATUS_PORT_ERROR;
  1184. }
  1185. }
  1186. void Network::_externalConfig(ZT_VirtualNetworkConfig *ec) const
  1187. {
  1188. // assumes _config_l is locked
  1189. ec->nwid = _id;
  1190. ec->mac = _mac.toInt();
  1191. if (_config)
  1192. Utils::scopy(ec->name,sizeof(ec->name),_config.name);
  1193. else ec->name[0] = (char)0;
  1194. ec->status = _status();
  1195. ec->type = (_config) ? (_config.isPrivate() ? ZT_NETWORK_TYPE_PRIVATE : ZT_NETWORK_TYPE_PUBLIC) : ZT_NETWORK_TYPE_PRIVATE;
  1196. ec->mtu = (_config) ? _config.mtu : ZT_DEFAULT_MTU;
  1197. ec->dhcp = 0;
  1198. std::vector<Address> ab;
  1199. for(unsigned int i=0;i<_config.specialistCount;++i) {
  1200. if ((_config.specialists[i] & ZT_NETWORKCONFIG_SPECIALIST_TYPE_ACTIVE_BRIDGE) != 0)
  1201. ab.push_back(Address(_config.specialists[i]));
  1202. }
  1203. ec->bridge = (std::find(ab.begin(),ab.end(),RR->identity.address()) != ab.end()) ? 1 : 0;
  1204. ec->broadcastEnabled = (_config) ? (_config.enableBroadcast() ? 1 : 0) : 0;
  1205. ec->portError = _portError;
  1206. ec->netconfRevision = (_config) ? (unsigned long)_config.revision : 0;
  1207. ec->assignedAddressCount = 0;
  1208. for(unsigned int i=0;i<ZT_MAX_ZT_ASSIGNED_ADDRESSES;++i) {
  1209. if (i < _config.staticIpCount) {
  1210. memcpy(&(ec->assignedAddresses[i]),&(_config.staticIps[i]),sizeof(struct sockaddr_storage));
  1211. ++ec->assignedAddressCount;
  1212. } else {
  1213. memset(&(ec->assignedAddresses[i]),0,sizeof(struct sockaddr_storage));
  1214. }
  1215. }
  1216. ec->routeCount = 0;
  1217. for(unsigned int i=0;i<ZT_MAX_NETWORK_ROUTES;++i) {
  1218. if (i < _config.routeCount) {
  1219. memcpy(&(ec->routes[i]),&(_config.routes[i]),sizeof(ZT_VirtualNetworkRoute));
  1220. ++ec->routeCount;
  1221. } else {
  1222. memset(&(ec->routes[i]),0,sizeof(ZT_VirtualNetworkRoute));
  1223. }
  1224. }
  1225. ec->multicastSubscriptionCount = (unsigned int)_myMulticastGroups.size();
  1226. for(unsigned long i=0;i<(unsigned long)_myMulticastGroups.size();++i) {
  1227. ec->multicastSubscriptions[i].mac = _myMulticastGroups[i].mac().toInt();
  1228. ec->multicastSubscriptions[i].adi = _myMulticastGroups[i].adi();
  1229. }
  1230. }
  1231. void Network::_announceMulticastGroups(void *tPtr,bool force)
  1232. {
  1233. // Assumes _myMulticastGroups_l and _memberships_l are locked
  1234. const int64_t now = RR->node->now();
  1235. const std::vector<MulticastGroup> groups(_allMulticastGroups());
  1236. _announceMulticastGroupsTo(tPtr,controller(),groups);
  1237. {
  1238. Address *a = (Address *)0;
  1239. Membership *m = (Membership *)0;
  1240. Hashtable<Address,Membership>::Iterator i(_memberships);
  1241. while (i.next(a,m)) {
  1242. bool announce = m->multicastLikeGate(now); // force this to be called even if 'force' is true since it updates last push time
  1243. if ((!announce)&&(force))
  1244. announce = true;
  1245. if ((announce)&&(m->isAllowedOnNetwork(_config)))
  1246. _announceMulticastGroupsTo(tPtr,*a,groups);
  1247. }
  1248. }
  1249. }
  1250. void Network::_announceMulticastGroupsTo(void *tPtr,const Address &peer,const std::vector<MulticastGroup> &allMulticastGroups)
  1251. {
  1252. // Assumes _myMulticastGroups_l and _memberships_l are locked
  1253. ScopedPtr<Packet> outp(new Packet(peer,RR->identity.address(),Packet::VERB_MULTICAST_LIKE));
  1254. for(std::vector<MulticastGroup>::const_iterator mg(allMulticastGroups.begin());mg!=allMulticastGroups.end();++mg) {
  1255. if ((outp->size() + 24) >= ZT_PROTO_MAX_PACKET_LENGTH) {
  1256. outp->compress();
  1257. RR->sw->send(tPtr,*outp,true);
  1258. outp->reset(peer,RR->identity.address(),Packet::VERB_MULTICAST_LIKE);
  1259. }
  1260. // network ID, MAC, ADI
  1261. outp->append((uint64_t)_id);
  1262. mg->mac().appendTo(*outp);
  1263. outp->append((uint32_t)mg->adi());
  1264. }
  1265. if (outp->size() > ZT_PROTO_MIN_PACKET_LENGTH) {
  1266. outp->compress();
  1267. RR->sw->send(tPtr,*outp,true);
  1268. }
  1269. }
  1270. std::vector<MulticastGroup> Network::_allMulticastGroups() const
  1271. {
  1272. // Assumes _myMulticastGroups_l is locked
  1273. std::vector<MulticastGroup> mgs;
  1274. mgs.reserve(_myMulticastGroups.size() + _multicastGroupsBehindMe.size() + 1);
  1275. mgs.insert(mgs.end(),_myMulticastGroups.begin(),_myMulticastGroups.end());
  1276. _multicastGroupsBehindMe.appendKeys(mgs);
  1277. if ((_config)&&(_config.enableBroadcast()))
  1278. mgs.push_back(Network::BROADCAST);
  1279. std::sort(mgs.begin(),mgs.end());
  1280. mgs.erase(std::unique(mgs.begin(),mgs.end()),mgs.end());
  1281. return mgs;
  1282. }
  1283. } // namespace ZeroTier