Switch.cpp 35 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929
  1. /*
  2. * ZeroTier One - Global Peer to Peer Ethernet
  3. * Copyright (C) 2011-2014 ZeroTier Networks LLC
  4. *
  5. * This program is free software: you can redistribute it and/or modify
  6. * it under the terms of the GNU General Public License as published by
  7. * the Free Software Foundation, either version 3 of the License, or
  8. * (at your option) any later version.
  9. *
  10. * This program is distributed in the hope that it will be useful,
  11. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13. * GNU General Public License for more details.
  14. *
  15. * You should have received a copy of the GNU General Public License
  16. * along with this program. If not, see <http://www.gnu.org/licenses/>.
  17. *
  18. * --
  19. *
  20. * ZeroTier may be used and distributed under the terms of the GPLv3, which
  21. * are available at: http://www.gnu.org/licenses/gpl-3.0.html
  22. *
  23. * If you would like to embed ZeroTier into a commercial application or
  24. * redistribute it in a modified binary form, please contact ZeroTier Networks
  25. * LLC. Start here: http://www.zerotier.com/
  26. */
  27. #include <stdio.h>
  28. #include <stdlib.h>
  29. #include <algorithm>
  30. #include <utility>
  31. #include <stdexcept>
  32. #include "Constants.hpp"
  33. #ifdef __WINDOWS__
  34. #include <WinSock2.h>
  35. #include <Windows.h>
  36. #endif
  37. #include "Switch.hpp"
  38. #include "Node.hpp"
  39. #include "EthernetTap.hpp"
  40. #include "InetAddress.hpp"
  41. #include "Topology.hpp"
  42. #include "RuntimeEnvironment.hpp"
  43. #include "Peer.hpp"
  44. #include "NodeConfig.hpp"
  45. #include "CMWC4096.hpp"
  46. #include "AntiRecursion.hpp"
  47. #include "../version.h"
  48. namespace ZeroTier {
  49. Switch::Switch(const RuntimeEnvironment *renv) :
  50. _r(renv),
  51. _lastBeacon(0),
  52. _multicastIdCounter((unsigned int)renv->prng->next32()) // start a random spot to minimize possible collisions on startup
  53. {
  54. }
  55. Switch::~Switch()
  56. {
  57. }
  58. void Switch::onRemotePacket(const SharedPtr<Socket> &fromSock,const InetAddress &fromAddr,Buffer<ZT_SOCKET_MAX_MESSAGE_LEN> &data)
  59. {
  60. try {
  61. if (data.size() == ZT_PROTO_BEACON_LENGTH) {
  62. _handleBeacon(fromSock,fromAddr,data);
  63. } else if (data.size() > ZT_PROTO_MIN_FRAGMENT_LENGTH) {
  64. if (data[ZT_PACKET_FRAGMENT_IDX_FRAGMENT_INDICATOR] == ZT_PACKET_FRAGMENT_INDICATOR)
  65. _handleRemotePacketFragment(fromSock,fromAddr,data);
  66. else if (data.size() >= ZT_PROTO_MIN_PACKET_LENGTH)
  67. _handleRemotePacketHead(fromSock,fromAddr,data);
  68. }
  69. } catch (std::exception &ex) {
  70. TRACE("dropped packet from %s: unexpected exception: %s",fromAddr.toString().c_str(),ex.what());
  71. } catch ( ... ) {
  72. TRACE("dropped packet from %s: unexpected exception: (unknown)",fromAddr.toString().c_str());
  73. }
  74. }
  75. void Switch::onLocalEthernet(const SharedPtr<Network> &network,const MAC &from,const MAC &to,unsigned int etherType,const Buffer<4096> &data)
  76. {
  77. SharedPtr<NetworkConfig> nconf(network->config2());
  78. if (!nconf)
  79. return;
  80. // Sanity check -- bridge loop? OS problem?
  81. if (to == network->mac())
  82. return;
  83. /* Check anti-recursion module to ensure that this is not ZeroTier talking over its own links.
  84. * Note: even when we introduce a more purposeful binding of the main UDP port, this can
  85. * still happen because Windows likes to send broadcasts over interfaces that have little
  86. * to do with their intended target audience. :P */
  87. if (!_r->antiRec->checkEthernetFrame(data.data(),data.size())) {
  88. TRACE("%s: rejected recursively addressed ZeroTier packet by tail match (type %s, length: %u)",network->tapDeviceName().c_str(),etherTypeName(etherType),data.size());
  89. return;
  90. }
  91. // Check to make sure this protocol is allowed on this network
  92. if (!nconf->permitsEtherType(etherType)) {
  93. TRACE("%s: ignored tap: %s -> %s: ethertype %s not allowed on network %.16llx",network->tapDeviceName().c_str(),from.toString().c_str(),to.toString().c_str(),etherTypeName(etherType),(unsigned long long)network->id());
  94. return;
  95. }
  96. // Check if this packet is from someone other than the tap -- i.e. bridged in
  97. bool fromBridged = false;
  98. if (from != network->mac()) {
  99. if (!network->permitsBridging(_r->identity.address())) {
  100. LOG("%s: %s -> %s %s not forwarded, bridging disabled on %.16llx or this peer not a bridge",network->tapDeviceName().c_str(),from.toString().c_str(),to.toString().c_str(),etherTypeName(etherType),network->id());
  101. return;
  102. }
  103. fromBridged = true;
  104. }
  105. if (to.isMulticast()) {
  106. // Destination is a multicast address (including broadcast)
  107. uint64_t now = Utils::now();
  108. MulticastGroup mg(to,0);
  109. if (to.isBroadcast()) {
  110. if ((etherType == ZT_ETHERTYPE_ARP)&&(data.size() >= 28)&&(data[2] == 0x08)&&(data[3] == 0x00)&&(data[4] == 6)&&(data[5] == 4)&&(data[7] == 0x01)) {
  111. // Cram IPv4 IP into ADI field to make IPv4 ARP broadcast channel specific and scalable
  112. // Also: enableBroadcast() does not apply to ARP since it's required for IPv4
  113. mg = MulticastGroup::deriveMulticastGroupForAddressResolution(InetAddress(data.field(24,4),4,0));
  114. } else if (!nconf->enableBroadcast()) {
  115. // Don't transmit broadcasts if this network doesn't want them
  116. TRACE("%s: dropped broadcast since ff:ff:ff:ff:ff:ff is not enabled on network %.16llx",network->tapDeviceName().c_str(),network->id());
  117. return;
  118. }
  119. }
  120. /* Learn multicast groups for bridged-in hosts.
  121. * Note that some OSes, most notably Linux, do this for you by learning
  122. * multicast addresses on bridge interfaces and subscribing each slave.
  123. * But in that case this does no harm, as the sets are just merged. */
  124. if (fromBridged)
  125. network->learnBridgedMulticastGroup(mg,now);
  126. // Check multicast/broadcast bandwidth quotas and reject if quota exceeded
  127. if (!network->updateAndCheckMulticastBalance(_r->identity.address(),mg,data.size())) {
  128. TRACE("%s: didn't multicast %d bytes, quota exceeded for multicast group %s",network->tapDeviceName().c_str(),(int)data.size(),mg.toString().c_str());
  129. return;
  130. }
  131. TRACE("%s: MULTICAST %s -> %s %s %d",network->tapDeviceName().c_str(),from.toString().c_str(),mg.toString().c_str(),etherTypeName(etherType),(int)data.size());
  132. /* old P5 multicast algorithm
  133. const unsigned int mcid = ++_multicastIdCounter & 0xffffff;
  134. const uint16_t bloomNonce = (uint16_t)(_r->prng->next32() & 0xffff); // doesn't need to be cryptographically strong
  135. unsigned char bloom[ZT_PROTO_VERB_MULTICAST_FRAME_LEN_PROPAGATION_BLOOM];
  136. unsigned char fifo[ZT_PROTO_VERB_MULTICAST_FRAME_LEN_PROPAGATION_FIFO + ZT_ADDRESS_LENGTH]; // extra ZT_ADDRESS_LENGTH is for first hop, not put in packet but serves as destination for packet
  137. unsigned char *const fifoEnd = fifo + sizeof(fifo);
  138. const unsigned int signedPartLen = (ZT_PROTO_VERB_MULTICAST_FRAME_IDX_FRAME - ZT_PROTO_VERB_MULTICAST_FRAME_IDX__START_OF_SIGNED_PORTION) + data.size();
  139. const SharedPtr<Peer> supernode(_r->topology->getBestSupernode());
  140. // For each bit prefix send a packet to a list of destinations within it
  141. for(unsigned int prefix=0,np=((unsigned int)2 << (nconf->multicastPrefixBits() - 1));prefix<np;++prefix) {
  142. memset(bloom,0,sizeof(bloom));
  143. unsigned char *fifoPtr = fifo;
  144. // Add all active bridges and then next hops we know about to propagation queue
  145. Multicaster::AddToPropagationQueue appender(
  146. &fifoPtr,
  147. fifoEnd,
  148. bloom,
  149. bloomNonce,
  150. _r->identity.address(),
  151. nconf->multicastPrefixBits(),
  152. prefix,
  153. _r->topology,
  154. now);
  155. for(std::set<Address>::const_iterator ab(nconf->activeBridges().begin());ab!=nconf->activeBridges().end();++ab) {
  156. if (!appender(*ab))
  157. break;
  158. }
  159. _r->mc->getNextHops(network->id(),mg,appender);
  160. // Pad remainder of FIFO with zeroes
  161. while (fifoPtr != fifoEnd)
  162. *(fifoPtr++) = (unsigned char)0;
  163. // First element in FIFO is first hop, rest of FIFO is sent in packet *to* first hop
  164. Address firstHop(fifo,ZT_ADDRESS_LENGTH);
  165. if (!firstHop) {
  166. if (supernode)
  167. firstHop = supernode->address();
  168. else continue; // nowhere to go
  169. }
  170. Packet outp(firstHop,_r->identity.address(),Packet::VERB_MULTICAST_FRAME);
  171. outp.append((uint16_t)0);
  172. outp.append(fifo + ZT_ADDRESS_LENGTH,ZT_PROTO_VERB_MULTICAST_FRAME_LEN_PROPAGATION_FIFO); // remainder of fifo is loaded into packet
  173. outp.append(bloom,ZT_PROTO_VERB_MULTICAST_FRAME_LEN_PROPAGATION_BLOOM);
  174. outp.append((nconf->com()) ? (unsigned char)ZT_PROTO_VERB_MULTICAST_FRAME_FLAGS_HAS_MEMBERSHIP_CERTIFICATE : (unsigned char)0);
  175. outp.append(network->id());
  176. outp.append(bloomNonce);
  177. outp.append((unsigned char)nconf->multicastPrefixBits());
  178. outp.append((unsigned char)prefix);
  179. _r->identity.address().appendTo(outp); // lower 40 bits of MCID are my address
  180. outp.append((unsigned char)((mcid >> 16) & 0xff));
  181. outp.append((unsigned char)((mcid >> 8) & 0xff));
  182. outp.append((unsigned char)(mcid & 0xff)); // upper 24 bits of MCID are from our counter
  183. from.appendTo(outp);
  184. mg.mac().appendTo(outp);
  185. outp.append(mg.adi());
  186. outp.append((uint16_t)etherType);
  187. outp.append((uint16_t)data.size());
  188. outp.append(data);
  189. C25519::Signature sig(_r->identity.sign(outp.field(ZT_PROTO_VERB_MULTICAST_FRAME_IDX__START_OF_SIGNED_PORTION,signedPartLen),signedPartLen));
  190. outp.append((uint16_t)sig.size());
  191. outp.append(sig.data,(unsigned int)sig.size());
  192. // FIXME: now we send the netconf cert with every single multicast,
  193. // which pretty much ensures everyone has it ahead of time but adds
  194. // some redundant payload. Maybe think abouut this in the future.
  195. if (nconf->com())
  196. nconf->com().serialize(outp);
  197. outp.compress();
  198. send(outp,true);
  199. }
  200. */
  201. return;
  202. }
  203. if (to[0] == MAC::firstOctetForNetwork(network->id())) {
  204. // Destination is another ZeroTier node
  205. Address toZT(to.toAddress(network->id()));
  206. if (network->isAllowed(toZT)) {
  207. network->pushMembershipCertificate(toZT,false,Utils::now());
  208. if (fromBridged) {
  209. // Must use EXT_FRAME if source is not myself
  210. Packet outp(toZT,_r->identity.address(),Packet::VERB_EXT_FRAME);
  211. outp.append(network->id());
  212. outp.append((unsigned char)0);
  213. to.appendTo(outp);
  214. from.appendTo(outp);
  215. outp.append((uint16_t)etherType);
  216. outp.append(data);
  217. outp.compress();
  218. send(outp,true);
  219. } else {
  220. // VERB_FRAME is really just lighter weight EXT_FRAME, can use for direct-to-direct (before bridging this was the only unicast method)
  221. Packet outp(toZT,_r->identity.address(),Packet::VERB_FRAME);
  222. outp.append(network->id());
  223. outp.append((uint16_t)etherType);
  224. outp.append(data);
  225. outp.compress();
  226. send(outp,true);
  227. }
  228. } else {
  229. TRACE("%s: UNICAST: %s -> %s %s dropped, destination not a member of closed network %.16llx",network->tapDeviceName().c_str(),from.toString().c_str(),to.toString().c_str(),etherTypeName(etherType),network->id());
  230. }
  231. return;
  232. }
  233. {
  234. // Destination is behind another bridge
  235. Address bridges[ZT_MAX_BRIDGE_SPAM];
  236. unsigned int numBridges = 0;
  237. bridges[0] = network->findBridgeTo(to);
  238. if ((bridges[0])&&(bridges[0] != _r->identity.address())&&(network->isAllowed(bridges[0]))&&(network->permitsBridging(bridges[0]))) {
  239. // We have a known bridge route for this MAC.
  240. ++numBridges;
  241. } else if (!nconf->activeBridges().empty()) {
  242. /* If there is no known route, spam to up to ZT_MAX_BRIDGE_SPAM active
  243. * bridges. This is similar to what many switches do -- if they do not
  244. * know which port corresponds to a MAC, they send it to all ports. If
  245. * there aren't any active bridges, numBridges will stay 0 and packet
  246. * is dropped. */
  247. std::set<Address>::const_iterator ab(nconf->activeBridges().begin());
  248. if (nconf->activeBridges().size() <= ZT_MAX_BRIDGE_SPAM) {
  249. // If there are <= ZT_MAX_BRIDGE_SPAM active bridges, spam them all
  250. while (ab != nconf->activeBridges().end()) {
  251. if (network->isAllowed(*ab)) // config sanity check
  252. bridges[numBridges++] = *ab;
  253. ++ab;
  254. }
  255. } else {
  256. // Otherwise pick a random set of them
  257. while (numBridges < ZT_MAX_BRIDGE_SPAM) {
  258. if (ab == nconf->activeBridges().end())
  259. ab = nconf->activeBridges().begin();
  260. if (((unsigned long)_r->prng->next32() % (unsigned long)nconf->activeBridges().size()) == 0) {
  261. if (network->isAllowed(*ab)) // config sanity check
  262. bridges[numBridges++] = *ab;
  263. ++ab;
  264. } else ++ab;
  265. }
  266. }
  267. }
  268. for(unsigned int b=0;b<numBridges;++b) {
  269. Packet outp(bridges[b],_r->identity.address(),Packet::VERB_EXT_FRAME);
  270. outp.append(network->id());
  271. outp.append((unsigned char)0);
  272. to.appendTo(outp);
  273. from.appendTo(outp);
  274. outp.append((uint16_t)etherType);
  275. outp.append(data);
  276. outp.compress();
  277. send(outp,true);
  278. }
  279. }
  280. }
  281. void Switch::send(const Packet &packet,bool encrypt)
  282. {
  283. if (packet.destination() == _r->identity.address()) {
  284. TRACE("BUG: caught attempt to send() to self, ignored");
  285. return;
  286. }
  287. if (!_trySend(packet,encrypt)) {
  288. Mutex::Lock _l(_txQueue_m);
  289. _txQueue.insert(std::pair< Address,TXQueueEntry >(packet.destination(),TXQueueEntry(Utils::now(),packet,encrypt)));
  290. }
  291. }
  292. void Switch::sendHELLO(const Address &dest)
  293. {
  294. Packet outp(dest,_r->identity.address(),Packet::VERB_HELLO);
  295. outp.append((unsigned char)ZT_PROTO_VERSION);
  296. outp.append((unsigned char)ZEROTIER_ONE_VERSION_MAJOR);
  297. outp.append((unsigned char)ZEROTIER_ONE_VERSION_MINOR);
  298. outp.append((uint16_t)ZEROTIER_ONE_VERSION_REVISION);
  299. outp.append(Utils::now());
  300. _r->identity.serialize(outp,false);
  301. send(outp,false);
  302. }
  303. bool Switch::sendHELLO(const SharedPtr<Peer> &dest,const Path &path)
  304. {
  305. uint64_t now = Utils::now();
  306. Packet outp(dest->address(),_r->identity.address(),Packet::VERB_HELLO);
  307. outp.append((unsigned char)ZT_PROTO_VERSION);
  308. outp.append((unsigned char)ZEROTIER_ONE_VERSION_MAJOR);
  309. outp.append((unsigned char)ZEROTIER_ONE_VERSION_MINOR);
  310. outp.append((uint16_t)ZEROTIER_ONE_VERSION_REVISION);
  311. outp.append(now);
  312. _r->identity.serialize(outp,false);
  313. outp.armor(dest->key(),false);
  314. _r->antiRec->logOutgoingZT(outp.data(),outp.size());
  315. return _r->sm->send(path.address(),path.tcp(),path.type() == Path::PATH_TYPE_TCP_OUT,outp.data(),outp.size());
  316. }
  317. bool Switch::sendHELLO(const SharedPtr<Peer> &dest,const InetAddress &destUdp)
  318. {
  319. uint64_t now = Utils::now();
  320. Packet outp(dest->address(),_r->identity.address(),Packet::VERB_HELLO);
  321. outp.append((unsigned char)ZT_PROTO_VERSION);
  322. outp.append((unsigned char)ZEROTIER_ONE_VERSION_MAJOR);
  323. outp.append((unsigned char)ZEROTIER_ONE_VERSION_MINOR);
  324. outp.append((uint16_t)ZEROTIER_ONE_VERSION_REVISION);
  325. outp.append(now);
  326. _r->identity.serialize(outp,false);
  327. outp.armor(dest->key(),false);
  328. _r->antiRec->logOutgoingZT(outp.data(),outp.size());
  329. return _r->sm->send(destUdp,false,false,outp.data(),outp.size());
  330. }
  331. bool Switch::unite(const Address &p1,const Address &p2,bool force)
  332. {
  333. if ((p1 == _r->identity.address())||(p2 == _r->identity.address()))
  334. return false;
  335. SharedPtr<Peer> p1p = _r->topology->getPeer(p1);
  336. if (!p1p)
  337. return false;
  338. SharedPtr<Peer> p2p = _r->topology->getPeer(p2);
  339. if (!p2p)
  340. return false;
  341. uint64_t now = Utils::now();
  342. std::pair<InetAddress,InetAddress> cg(Peer::findCommonGround(*p1p,*p2p,now));
  343. if (!(cg.first))
  344. return false;
  345. // Addresses are sorted in key for last unite attempt map for order
  346. // invariant lookup: (p1,p2) == (p2,p1)
  347. Array<Address,2> uniteKey;
  348. if (p1 >= p2) {
  349. uniteKey[0] = p2;
  350. uniteKey[1] = p1;
  351. } else {
  352. uniteKey[0] = p1;
  353. uniteKey[1] = p2;
  354. }
  355. {
  356. Mutex::Lock _l(_lastUniteAttempt_m);
  357. std::map< Array< Address,2 >,uint64_t >::const_iterator e(_lastUniteAttempt.find(uniteKey));
  358. if ((!force)&&(e != _lastUniteAttempt.end())&&((now - e->second) < ZT_MIN_UNITE_INTERVAL))
  359. return false;
  360. else _lastUniteAttempt[uniteKey] = now;
  361. }
  362. TRACE("unite: %s(%s) <> %s(%s)",p1.toString().c_str(),cg.second.toString().c_str(),p2.toString().c_str(),cg.first.toString().c_str());
  363. /* Tell P1 where to find P2 and vice versa, sending the packets to P1 and
  364. * P2 in randomized order in terms of which gets sent first. This is done
  365. * since in a few cases NAT-t can be sensitive to slight timing differences
  366. * in terms of when the two peers initiate. Normally this is accounted for
  367. * by the nearly-simultaneous RENDEZVOUS kickoff from the supernode, but
  368. * given that supernodes are hosted on cloud providers this can in some
  369. * cases have a few ms of latency between packet departures. By randomizing
  370. * the order we make each attempted NAT-t favor one or the other going
  371. * first, meaning if it doesn't succeed the first time it might the second
  372. * and so forth. */
  373. unsigned int alt = _r->prng->next32() & 1;
  374. unsigned int completed = alt + 2;
  375. while (alt != completed) {
  376. if ((alt & 1) == 0) {
  377. // Tell p1 where to find p2.
  378. Packet outp(p1,_r->identity.address(),Packet::VERB_RENDEZVOUS);
  379. outp.append((unsigned char)0);
  380. p2.appendTo(outp);
  381. outp.append((uint16_t)cg.first.port());
  382. if (cg.first.isV6()) {
  383. outp.append((unsigned char)16);
  384. outp.append(cg.first.rawIpData(),16);
  385. } else {
  386. outp.append((unsigned char)4);
  387. outp.append(cg.first.rawIpData(),4);
  388. }
  389. outp.armor(p1p->key(),true);
  390. p1p->send(_r,outp.data(),outp.size(),now);
  391. } else {
  392. // Tell p2 where to find p1.
  393. Packet outp(p2,_r->identity.address(),Packet::VERB_RENDEZVOUS);
  394. outp.append((unsigned char)0);
  395. p1.appendTo(outp);
  396. outp.append((uint16_t)cg.second.port());
  397. if (cg.second.isV6()) {
  398. outp.append((unsigned char)16);
  399. outp.append(cg.second.rawIpData(),16);
  400. } else {
  401. outp.append((unsigned char)4);
  402. outp.append(cg.second.rawIpData(),4);
  403. }
  404. outp.armor(p2p->key(),true);
  405. p2p->send(_r,outp.data(),outp.size(),now);
  406. }
  407. ++alt; // counts up and also flips LSB
  408. }
  409. return true;
  410. }
  411. void Switch::contact(const SharedPtr<Peer> &peer,const InetAddress &atAddr)
  412. {
  413. // Send simple packet directly to indicated address -- works for most NATs
  414. sendHELLO(peer,atAddr);
  415. TRACE("sending NAT-t HELLO to %s(%s)",peer->address().toString().c_str(),atAddr.toString().c_str());
  416. // If we have not punched through after this timeout, open refreshing can of whupass
  417. {
  418. Mutex::Lock _l(_contactQueue_m);
  419. _contactQueue.push_back(ContactQueueEntry(peer,Utils::now() + ZT_NAT_T_TACTICAL_ESCALATION_DELAY,atAddr));
  420. }
  421. // Kick main loop out of wait so that it can pick up this
  422. // change to our scheduled timer tasks.
  423. _r->sm->whack();
  424. }
  425. unsigned long Switch::doTimerTasks()
  426. {
  427. unsigned long nextDelay = ~((unsigned long)0); // big number, caller will cap return value
  428. uint64_t now = Utils::now();
  429. {
  430. Mutex::Lock _l(_contactQueue_m);
  431. for(std::list<ContactQueueEntry>::iterator qi(_contactQueue.begin());qi!=_contactQueue.end();) {
  432. if (now >= qi->fireAtTime) {
  433. if (!qi->peer->hasActiveDirectPath(now)) {
  434. TRACE("deploying aggressive NAT-t against %s(%s)",qi->peer->address().toString().c_str(),qi->inaddr.toString().c_str());
  435. /* Shotgun approach -- literally -- against symmetric NATs. Most of these
  436. * either increment or decrement ports so this gets a good number. Also try
  437. * the original port one more time for good measure, since sometimes it
  438. * fails first time around. */
  439. int p = (int)qi->inaddr.port() - 2;
  440. for(int k=0;k<5;++k) {
  441. if ((p > 0)&&(p <= 0xffff)) {
  442. qi->inaddr.setPort((unsigned int)p);
  443. sendHELLO(qi->peer,qi->inaddr);
  444. }
  445. ++p;
  446. }
  447. }
  448. _contactQueue.erase(qi++);
  449. } else {
  450. nextDelay = std::min(nextDelay,(unsigned long)(qi->fireAtTime - now));
  451. ++qi;
  452. }
  453. }
  454. }
  455. {
  456. Mutex::Lock _l(_outstandingWhoisRequests_m);
  457. for(std::map< Address,WhoisRequest >::iterator i(_outstandingWhoisRequests.begin());i!=_outstandingWhoisRequests.end();) {
  458. unsigned long since = (unsigned long)(now - i->second.lastSent);
  459. if (since >= ZT_WHOIS_RETRY_DELAY) {
  460. if (i->second.retries >= ZT_MAX_WHOIS_RETRIES) {
  461. TRACE("WHOIS %s timed out",i->first.toString().c_str());
  462. _outstandingWhoisRequests.erase(i++);
  463. continue;
  464. } else {
  465. i->second.lastSent = now;
  466. i->second.peersConsulted[i->second.retries] = _sendWhoisRequest(i->first,i->second.peersConsulted,i->second.retries);
  467. ++i->second.retries;
  468. TRACE("WHOIS %s (retry %u)",i->first.toString().c_str(),i->second.retries);
  469. nextDelay = std::min(nextDelay,(unsigned long)ZT_WHOIS_RETRY_DELAY);
  470. }
  471. } else nextDelay = std::min(nextDelay,ZT_WHOIS_RETRY_DELAY - since);
  472. ++i;
  473. }
  474. }
  475. {
  476. Mutex::Lock _l(_txQueue_m);
  477. for(std::multimap< Address,TXQueueEntry >::iterator i(_txQueue.begin());i!=_txQueue.end();) {
  478. if (_trySend(i->second.packet,i->second.encrypt))
  479. _txQueue.erase(i++);
  480. else if ((now - i->second.creationTime) > ZT_TRANSMIT_QUEUE_TIMEOUT) {
  481. TRACE("TX %s -> %s timed out",i->second.packet.source().toString().c_str(),i->second.packet.destination().toString().c_str());
  482. _txQueue.erase(i++);
  483. } else ++i;
  484. }
  485. }
  486. {
  487. Mutex::Lock _l(_rxQueue_m);
  488. for(std::list< SharedPtr<IncomingPacket> >::iterator i(_rxQueue.begin());i!=_rxQueue.end();) {
  489. if ((now - (*i)->receiveTime()) > ZT_RECEIVE_QUEUE_TIMEOUT) {
  490. TRACE("RX %s -> %s timed out",(*i)->source().toString().c_str(),(*i)->destination().toString().c_str());
  491. _rxQueue.erase(i++);
  492. } else ++i;
  493. }
  494. }
  495. {
  496. Mutex::Lock _l(_defragQueue_m);
  497. for(std::map< uint64_t,DefragQueueEntry >::iterator i(_defragQueue.begin());i!=_defragQueue.end();) {
  498. if ((now - i->second.creationTime) > ZT_FRAGMENTED_PACKET_RECEIVE_TIMEOUT) {
  499. TRACE("incomplete fragmented packet %.16llx timed out, fragments discarded",i->first);
  500. _defragQueue.erase(i++);
  501. } else ++i;
  502. }
  503. }
  504. return std::max(nextDelay,(unsigned long)10); // minimum delay
  505. }
  506. void Switch::announceMulticastGroups(const std::map< SharedPtr<Network>,std::set<MulticastGroup> > &allMemberships)
  507. {
  508. std::vector< SharedPtr<Peer> > directPeers;
  509. _r->topology->eachPeer(Topology::CollectPeersWithActiveDirectPath(directPeers,Utils::now()));
  510. #ifdef ZT_TRACE
  511. unsigned int totalMulticastGroups = 0;
  512. for(std::map< SharedPtr<Network>,std::set<MulticastGroup> >::const_iterator i(allMemberships.begin());i!=allMemberships.end();++i)
  513. totalMulticastGroups += (unsigned int)i->second.size();
  514. TRACE("announcing %u multicast groups for %u networks to %u peers",totalMulticastGroups,(unsigned int)allMemberships.size(),(unsigned int)directPeers.size());
  515. #endif
  516. uint64_t now = Utils::now();
  517. for(std::vector< SharedPtr<Peer> >::iterator p(directPeers.begin());p!=directPeers.end();++p) {
  518. Packet outp((*p)->address(),_r->identity.address(),Packet::VERB_MULTICAST_LIKE);
  519. for(std::map< SharedPtr<Network>,std::set<MulticastGroup> >::const_iterator nwmgs(allMemberships.begin());nwmgs!=allMemberships.end();++nwmgs) {
  520. nwmgs->first->pushMembershipCertificate((*p)->address(),false,now);
  521. if ((_r->topology->isSupernode((*p)->address()))||(nwmgs->first->isAllowed((*p)->address()))) {
  522. for(std::set<MulticastGroup>::iterator mg(nwmgs->second.begin());mg!=nwmgs->second.end();++mg) {
  523. if ((outp.size() + 18) > ZT_UDP_DEFAULT_PAYLOAD_MTU) {
  524. send(outp,true);
  525. outp.reset((*p)->address(),_r->identity.address(),Packet::VERB_MULTICAST_LIKE);
  526. }
  527. // network ID, MAC, ADI
  528. outp.append((uint64_t)nwmgs->first->id());
  529. mg->mac().appendTo(outp);
  530. outp.append((uint32_t)mg->adi());
  531. }
  532. }
  533. }
  534. if (outp.size() > ZT_PROTO_MIN_PACKET_LENGTH)
  535. send(outp,true);
  536. }
  537. }
  538. void Switch::announceMulticastGroups(const SharedPtr<Peer> &peer)
  539. {
  540. Packet outp(peer->address(),_r->identity.address(),Packet::VERB_MULTICAST_LIKE);
  541. std::vector< SharedPtr<Network> > networks(_r->nc->networks());
  542. uint64_t now = Utils::now();
  543. for(std::vector< SharedPtr<Network> >::iterator n(networks.begin());n!=networks.end();++n) {
  544. if (((*n)->isAllowed(peer->address()))||(_r->topology->isSupernode(peer->address()))) {
  545. (*n)->pushMembershipCertificate(peer->address(),false,now);
  546. std::set<MulticastGroup> mgs((*n)->multicastGroups());
  547. for(std::set<MulticastGroup>::iterator mg(mgs.begin());mg!=mgs.end();++mg) {
  548. if ((outp.size() + 18) > ZT_UDP_DEFAULT_PAYLOAD_MTU) {
  549. send(outp,true);
  550. outp.reset(peer->address(),_r->identity.address(),Packet::VERB_MULTICAST_LIKE);
  551. }
  552. // network ID, MAC, ADI
  553. outp.append((uint64_t)(*n)->id());
  554. mg->mac().appendTo(outp);
  555. outp.append((uint32_t)mg->adi());
  556. }
  557. }
  558. }
  559. if (outp.size() > ZT_PROTO_MIN_PACKET_LENGTH)
  560. send(outp,true);
  561. }
  562. void Switch::requestWhois(const Address &addr)
  563. {
  564. //TRACE("requesting WHOIS for %s",addr.toString().c_str());
  565. bool inserted = false;
  566. {
  567. Mutex::Lock _l(_outstandingWhoisRequests_m);
  568. std::pair< std::map< Address,WhoisRequest >::iterator,bool > entry(_outstandingWhoisRequests.insert(std::pair<Address,WhoisRequest>(addr,WhoisRequest())));
  569. if ((inserted = entry.second))
  570. entry.first->second.lastSent = Utils::now();
  571. entry.first->second.retries = 0; // reset retry count if entry already existed
  572. }
  573. if (inserted)
  574. _sendWhoisRequest(addr,(const Address *)0,0);
  575. }
  576. void Switch::cancelWhoisRequest(const Address &addr)
  577. {
  578. Mutex::Lock _l(_outstandingWhoisRequests_m);
  579. _outstandingWhoisRequests.erase(addr);
  580. }
  581. void Switch::doAnythingWaitingForPeer(const SharedPtr<Peer> &peer)
  582. {
  583. { // cancel pending WHOIS since we now know this peer
  584. Mutex::Lock _l(_outstandingWhoisRequests_m);
  585. _outstandingWhoisRequests.erase(peer->address());
  586. }
  587. { // finish processing any packets waiting on peer's public key / identity
  588. Mutex::Lock _l(_rxQueue_m);
  589. for(std::list< SharedPtr<IncomingPacket> >::iterator rxi(_rxQueue.begin());rxi!=_rxQueue.end();) {
  590. if ((*rxi)->tryDecode(_r))
  591. _rxQueue.erase(rxi++);
  592. else ++rxi;
  593. }
  594. }
  595. { // finish sending any packets waiting on peer's public key / identity
  596. Mutex::Lock _l(_txQueue_m);
  597. std::pair< std::multimap< Address,TXQueueEntry >::iterator,std::multimap< Address,TXQueueEntry >::iterator > waitingTxQueueItems(_txQueue.equal_range(peer->address()));
  598. for(std::multimap< Address,TXQueueEntry >::iterator txi(waitingTxQueueItems.first);txi!=waitingTxQueueItems.second;) {
  599. if (_trySend(txi->second.packet,txi->second.encrypt))
  600. _txQueue.erase(txi++);
  601. else ++txi;
  602. }
  603. }
  604. }
  605. const char *Switch::etherTypeName(const unsigned int etherType)
  606. throw()
  607. {
  608. switch(etherType) {
  609. case ZT_ETHERTYPE_IPV4: return "IPV4";
  610. case ZT_ETHERTYPE_ARP: return "ARP";
  611. case ZT_ETHERTYPE_RARP: return "RARP";
  612. case ZT_ETHERTYPE_ATALK: return "ATALK";
  613. case ZT_ETHERTYPE_AARP: return "AARP";
  614. case ZT_ETHERTYPE_IPX_A: return "IPX_A";
  615. case ZT_ETHERTYPE_IPX_B: return "IPX_B";
  616. case ZT_ETHERTYPE_IPV6: return "IPV6";
  617. }
  618. return "UNKNOWN";
  619. }
  620. void Switch::_handleRemotePacketFragment(const SharedPtr<Socket> &fromSock,const InetAddress &fromAddr,const Buffer<4096> &data)
  621. {
  622. Packet::Fragment fragment(data);
  623. Address destination(fragment.destination());
  624. if (destination != _r->identity.address()) {
  625. // Fragment is not for us, so try to relay it
  626. if (fragment.hops() < ZT_RELAY_MAX_HOPS) {
  627. fragment.incrementHops();
  628. // Note: we don't bother initiating NAT-t for fragments, since heads will set that off.
  629. // It wouldn't hurt anything, just redundant and unnecessary.
  630. SharedPtr<Peer> relayTo = _r->topology->getPeer(destination);
  631. if ((!relayTo)||(relayTo->send(_r,fragment.data(),fragment.size(),Utils::now()) == Path::PATH_TYPE_NULL)) {
  632. // Don't know peer or no direct path -- so relay via supernode
  633. relayTo = _r->topology->getBestSupernode();
  634. if (relayTo)
  635. relayTo->send(_r,fragment.data(),fragment.size(),Utils::now());
  636. }
  637. } else {
  638. TRACE("dropped relay [fragment](%s) -> %s, max hops exceeded",fromAddr.toString().c_str(),destination.toString().c_str());
  639. }
  640. } else {
  641. // Fragment looks like ours
  642. uint64_t pid = fragment.packetId();
  643. unsigned int fno = fragment.fragmentNumber();
  644. unsigned int tf = fragment.totalFragments();
  645. if ((tf <= ZT_MAX_PACKET_FRAGMENTS)&&(fno < ZT_MAX_PACKET_FRAGMENTS)&&(fno > 0)&&(tf > 1)) {
  646. // Fragment appears basically sane. Its fragment number must be
  647. // 1 or more, since a Packet with fragmented bit set is fragment 0.
  648. // Total fragments must be more than 1, otherwise why are we
  649. // seeing a Packet::Fragment?
  650. Mutex::Lock _l(_defragQueue_m);
  651. std::map< uint64_t,DefragQueueEntry >::iterator dqe(_defragQueue.find(pid));
  652. if (dqe == _defragQueue.end()) {
  653. // We received a Packet::Fragment without its head, so queue it and wait
  654. DefragQueueEntry &dq = _defragQueue[pid];
  655. dq.creationTime = Utils::now();
  656. dq.frags[fno - 1] = fragment;
  657. dq.totalFragments = tf; // total fragment count is known
  658. dq.haveFragments = 1 << fno; // we have only this fragment
  659. //TRACE("fragment (%u/%u) of %.16llx from %s",fno + 1,tf,pid,fromAddr.toString().c_str());
  660. } else if (!(dqe->second.haveFragments & (1 << fno))) {
  661. // We have other fragments and maybe the head, so add this one and check
  662. dqe->second.frags[fno - 1] = fragment;
  663. dqe->second.totalFragments = tf;
  664. //TRACE("fragment (%u/%u) of %.16llx from %s",fno + 1,tf,pid,fromAddr.toString().c_str());
  665. if (Utils::countBits(dqe->second.haveFragments |= (1 << fno)) == tf) {
  666. // We have all fragments -- assemble and process full Packet
  667. //TRACE("packet %.16llx is complete, assembling and processing...",pid);
  668. SharedPtr<IncomingPacket> packet(dqe->second.frag0);
  669. for(unsigned int f=1;f<tf;++f)
  670. packet->append(dqe->second.frags[f - 1].payload(),dqe->second.frags[f - 1].payloadLength());
  671. _defragQueue.erase(dqe);
  672. if (!packet->tryDecode(_r)) {
  673. Mutex::Lock _l(_rxQueue_m);
  674. _rxQueue.push_back(packet);
  675. }
  676. }
  677. } // else this is a duplicate fragment, ignore
  678. }
  679. }
  680. }
  681. void Switch::_handleRemotePacketHead(const SharedPtr<Socket> &fromSock,const InetAddress &fromAddr,const Buffer<4096> &data)
  682. {
  683. SharedPtr<IncomingPacket> packet(new IncomingPacket(data,fromSock,fromAddr));
  684. Address source(packet->source());
  685. Address destination(packet->destination());
  686. //TRACE("<< %.16llx %s -> %s (size: %u)",(unsigned long long)packet->packetId(),source.toString().c_str(),destination.toString().c_str(),packet->size());
  687. if (destination != _r->identity.address()) {
  688. // Packet is not for us, so try to relay it
  689. if (packet->hops() < ZT_RELAY_MAX_HOPS) {
  690. packet->incrementHops();
  691. SharedPtr<Peer> relayTo = _r->topology->getPeer(destination);
  692. Path::Type relayedVia;
  693. if ((relayTo)&&((relayedVia = relayTo->send(_r,packet->data(),packet->size(),Utils::now())) != Path::PATH_TYPE_NULL)) {
  694. /* If both paths are UDP, attempt to invoke UDP NAT-t between peers
  695. * by sending VERB_RENDEZVOUS. Do not do this for TCP due to GitHub
  696. * issue #63. */
  697. if ((fromSock->udp())&&(relayedVia == Path::PATH_TYPE_UDP))
  698. unite(source,destination,false);
  699. } else {
  700. // Don't know peer or no direct path -- so relay via supernode
  701. relayTo = _r->topology->getBestSupernode(&source,1,true);
  702. if (relayTo)
  703. relayTo->send(_r,packet->data(),packet->size(),Utils::now());
  704. }
  705. } else {
  706. TRACE("dropped relay %s(%s) -> %s, max hops exceeded",packet->source().toString().c_str(),fromAddr.toString().c_str(),destination.toString().c_str());
  707. }
  708. } else if (packet->fragmented()) {
  709. // Packet is the head of a fragmented packet series
  710. uint64_t pid = packet->packetId();
  711. Mutex::Lock _l(_defragQueue_m);
  712. std::map< uint64_t,DefragQueueEntry >::iterator dqe(_defragQueue.find(pid));
  713. if (dqe == _defragQueue.end()) {
  714. // If we have no other fragments yet, create an entry and save the head
  715. DefragQueueEntry &dq = _defragQueue[pid];
  716. dq.creationTime = Utils::now();
  717. dq.frag0 = packet;
  718. dq.totalFragments = 0; // 0 == unknown, waiting for Packet::Fragment
  719. dq.haveFragments = 1; // head is first bit (left to right)
  720. //TRACE("fragment (0/?) of %.16llx from %s",pid,fromAddr.toString().c_str());
  721. } else if (!(dqe->second.haveFragments & 1)) {
  722. // If we have other fragments but no head, see if we are complete with the head
  723. if ((dqe->second.totalFragments)&&(Utils::countBits(dqe->second.haveFragments |= 1) == dqe->second.totalFragments)) {
  724. // We have all fragments -- assemble and process full Packet
  725. //TRACE("packet %.16llx is complete, assembling and processing...",pid);
  726. // packet already contains head, so append fragments
  727. for(unsigned int f=1;f<dqe->second.totalFragments;++f)
  728. packet->append(dqe->second.frags[f - 1].payload(),dqe->second.frags[f - 1].payloadLength());
  729. _defragQueue.erase(dqe);
  730. if (!packet->tryDecode(_r)) {
  731. Mutex::Lock _l(_rxQueue_m);
  732. _rxQueue.push_back(packet);
  733. }
  734. } else {
  735. // Still waiting on more fragments, so queue the head
  736. dqe->second.frag0 = packet;
  737. }
  738. } // else this is a duplicate head, ignore
  739. } else {
  740. // Packet is unfragmented, so just process it
  741. if (!packet->tryDecode(_r)) {
  742. Mutex::Lock _l(_rxQueue_m);
  743. _rxQueue.push_back(packet);
  744. }
  745. }
  746. }
  747. void Switch::_handleBeacon(const SharedPtr<Socket> &fromSock,const InetAddress &fromAddr,const Buffer<4096> &data)
  748. {
  749. Address beaconAddr(data.field(ZT_PROTO_BEACON_IDX_ADDRESS,ZT_ADDRESS_LENGTH),ZT_ADDRESS_LENGTH);
  750. if (beaconAddr == _r->identity.address())
  751. return;
  752. SharedPtr<Peer> peer(_r->topology->getPeer(beaconAddr));
  753. if (peer) {
  754. uint64_t now = Utils::now();
  755. if (peer->haveUdpPath(fromAddr)) {
  756. if ((now - peer->lastDirectReceive()) >= ZT_PEER_DIRECT_PING_DELAY)
  757. peer->sendPing(_r,now);
  758. } else {
  759. if ((now - _lastBeacon) < ZT_MIN_BEACON_RESPONSE_INTERVAL)
  760. return;
  761. _lastBeacon = now;
  762. sendHELLO(peer,fromAddr);
  763. }
  764. }
  765. }
  766. Address Switch::_sendWhoisRequest(const Address &addr,const Address *peersAlreadyConsulted,unsigned int numPeersAlreadyConsulted)
  767. {
  768. SharedPtr<Peer> supernode(_r->topology->getBestSupernode(peersAlreadyConsulted,numPeersAlreadyConsulted,false));
  769. if (supernode) {
  770. Packet outp(supernode->address(),_r->identity.address(),Packet::VERB_WHOIS);
  771. addr.appendTo(outp);
  772. outp.armor(supernode->key(),true);
  773. uint64_t now = Utils::now();
  774. if (supernode->send(_r,outp.data(),outp.size(),now) != Path::PATH_TYPE_NULL)
  775. return supernode->address();
  776. }
  777. return Address();
  778. }
  779. bool Switch::_trySend(const Packet &packet,bool encrypt)
  780. {
  781. SharedPtr<Peer> peer(_r->topology->getPeer(packet.destination()));
  782. if (peer) {
  783. uint64_t now = Utils::now();
  784. SharedPtr<Peer> via;
  785. if (peer->hasActiveDirectPath(now)) {
  786. via = peer;
  787. } else {
  788. via = _r->topology->getBestSupernode();
  789. if (!via)
  790. return false;
  791. }
  792. Packet tmp(packet);
  793. unsigned int chunkSize = std::min(tmp.size(),(unsigned int)ZT_UDP_DEFAULT_PAYLOAD_MTU);
  794. tmp.setFragmented(chunkSize < tmp.size());
  795. tmp.armor(peer->key(),encrypt);
  796. if (via->send(_r,tmp.data(),chunkSize,now) != Path::PATH_TYPE_NULL) {
  797. if (chunkSize < tmp.size()) {
  798. // Too big for one bite, fragment the rest
  799. unsigned int fragStart = chunkSize;
  800. unsigned int remaining = tmp.size() - chunkSize;
  801. unsigned int fragsRemaining = (remaining / (ZT_UDP_DEFAULT_PAYLOAD_MTU - ZT_PROTO_MIN_FRAGMENT_LENGTH));
  802. if ((fragsRemaining * (ZT_UDP_DEFAULT_PAYLOAD_MTU - ZT_PROTO_MIN_FRAGMENT_LENGTH)) < remaining)
  803. ++fragsRemaining;
  804. unsigned int totalFragments = fragsRemaining + 1;
  805. for(unsigned int f=0;f<fragsRemaining;++f) {
  806. chunkSize = std::min(remaining,(unsigned int)(ZT_UDP_DEFAULT_PAYLOAD_MTU - ZT_PROTO_MIN_FRAGMENT_LENGTH));
  807. Packet::Fragment frag(tmp,fragStart,chunkSize,f + 1,totalFragments);
  808. via->send(_r,frag.data(),frag.size(),now);
  809. fragStart += chunkSize;
  810. remaining -= chunkSize;
  811. }
  812. }
  813. return true;
  814. }
  815. } else requestWhois(packet.destination());
  816. return false;
  817. }
  818. } // namespace ZeroTier