PacketDecoder.cpp 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559
  1. /*
  2. * ZeroTier One - Global Peer to Peer Ethernet
  3. * Copyright (C) 2012-2013 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 "RuntimeEnvironment.hpp"
  28. #include "Topology.hpp"
  29. #include "PacketDecoder.hpp"
  30. #include "Switch.hpp"
  31. #include "Peer.hpp"
  32. #include "NodeConfig.hpp"
  33. #include "Filter.hpp"
  34. namespace ZeroTier {
  35. bool PacketDecoder::tryDecode(const RuntimeEnvironment *_r)
  36. throw(std::out_of_range,std::runtime_error)
  37. {
  38. if ((!encrypted())&&(verb() == Packet::VERB_HELLO)) {
  39. // Unencrypted HELLOs are handled here since they are used to
  40. // populate our identity cache in the first place. Thus we might get
  41. // a HELLO for someone for whom we don't have a Peer record.
  42. TRACE("HELLO from %s(%s)",source().toString().c_str(),_remoteAddress.toString().c_str());
  43. return _doHELLO(_r);
  44. }
  45. SharedPtr<Peer> peer = _r->topology->getPeer(source());
  46. if (peer) {
  47. if (_step == DECODE_STEP_WAITING_FOR_ORIGINAL_SUBMITTER_LOOKUP) {
  48. // This means we've already decoded, decrypted, decompressed, and
  49. // validated, and we're processing a MULTICAST_FRAME. We're waiting
  50. // for a lookup on the frame's original submitter. So try again and
  51. // see if we have it.
  52. return _doMULTICAST_FRAME(_r,peer);
  53. }
  54. if (!hmacVerify(peer->macKey())) {
  55. TRACE("dropped packet from %s(%s), HMAC authentication failed (size: %u)",source().toString().c_str(),_remoteAddress.toString().c_str(),size());
  56. return true;
  57. }
  58. if (encrypted()) {
  59. decrypt(peer->cryptKey());
  60. } else {
  61. // Unencrypted is tolerated in case we want to run this on
  62. // devices where squeezing out cycles matters. HMAC is
  63. // what's really important. But log it in debug to catch any
  64. // packets being mistakenly sent in the clear.
  65. TRACE("ODD: %s from %s(%s) wasn't encrypted",Packet::verbString(verb()),source().toString().c_str(),_remoteAddress.toString().c_str());
  66. }
  67. if (!uncompress()) {
  68. TRACE("dropped packet from %s(%s), compressed data invalid",source().toString().c_str(),_remoteAddress.toString().c_str());
  69. return true;
  70. }
  71. Packet::Verb v = verb();
  72. // Validated packets that have passed HMAC can result in us learning a new
  73. // path to this peer.
  74. peer->onReceive(_r,_localPort,_remoteAddress,hops(),v,Utils::now());
  75. switch(v) {
  76. case Packet::VERB_NOP:
  77. TRACE("NOP from %s(%s)",source().toString().c_str(),_remoteAddress.toString().c_str());
  78. return true;
  79. case Packet::VERB_HELLO:
  80. return _doHELLO(_r);
  81. case Packet::VERB_ERROR:
  82. return _doERROR(_r,peer);
  83. case Packet::VERB_OK:
  84. return _doOK(_r,peer);
  85. case Packet::VERB_WHOIS:
  86. return _doWHOIS(_r,peer);
  87. case Packet::VERB_RENDEZVOUS:
  88. return _doRENDEZVOUS(_r,peer);
  89. case Packet::VERB_FRAME:
  90. return _doFRAME(_r,peer);
  91. case Packet::VERB_MULTICAST_LIKE:
  92. return _doMULTICAST_LIKE(_r,peer);
  93. case Packet::VERB_MULTICAST_FRAME:
  94. return _doMULTICAST_FRAME(_r,peer);
  95. case Packet::VERB_NETWORK_MEMBERSHIP_CERTIFICATE:
  96. return _doNETWORK_MEMBERSHIP_CERTIFICATE(_r,peer);
  97. case Packet::VERB_NETWORK_CONFIG_REQUEST:
  98. return _doNETWORK_CONFIG_REQUEST(_r,peer);
  99. case Packet::VERB_NETWORK_CONFIG_REFRESH:
  100. return _doNETWORK_CONFIG_REFRESH(_r,peer);
  101. default:
  102. // This might be something from a new or old version of the protocol.
  103. // Technically it passed HMAC so the packet is still valid, but we
  104. // ignore it.
  105. TRACE("ignored unrecognized verb %.2x from %s(%s)",(unsigned int)v,source().toString().c_str(),_remoteAddress.toString().c_str());
  106. return true;
  107. }
  108. } else {
  109. _step = DECODE_STEP_WAITING_FOR_SENDER_LOOKUP;
  110. _r->sw->requestWhois(source());
  111. return false;
  112. }
  113. }
  114. void PacketDecoder::_CBaddPeerFromHello(void *arg,const SharedPtr<Peer> &p,Topology::PeerVerifyResult result)
  115. {
  116. _CBaddPeerFromHello_Data *req = (_CBaddPeerFromHello_Data *)arg;
  117. const RuntimeEnvironment *_r = req->renv;
  118. switch(result) {
  119. case Topology::PEER_VERIFY_ACCEPTED_NEW:
  120. case Topology::PEER_VERIFY_ACCEPTED_ALREADY_HAVE:
  121. case Topology::PEER_VERIFY_ACCEPTED_DISPLACED_INVALID_ADDRESS: {
  122. _r->sw->doAnythingWaitingForPeer(p);
  123. Packet outp(req->source,_r->identity.address(),Packet::VERB_OK);
  124. outp.append((unsigned char)Packet::VERB_HELLO);
  125. outp.append(req->helloPacketId);
  126. outp.append(req->helloTimestamp);
  127. outp.encrypt(p->cryptKey());
  128. outp.hmacSet(p->macKey());
  129. _r->demarc->send(req->localPort,req->remoteAddress,outp.data(),outp.size(),-1);
  130. } break;
  131. case Topology::PEER_VERIFY_REJECTED_INVALID_IDENTITY: {
  132. Packet outp(req->source,_r->identity.address(),Packet::VERB_ERROR);
  133. outp.append((unsigned char)Packet::VERB_HELLO);
  134. outp.append(req->helloPacketId);
  135. outp.append((unsigned char)Packet::ERROR_IDENTITY_INVALID);
  136. outp.encrypt(p->cryptKey());
  137. outp.hmacSet(p->macKey());
  138. _r->demarc->send(req->localPort,req->remoteAddress,outp.data(),outp.size(),-1);
  139. } break;
  140. case Topology::PEER_VERIFY_REJECTED_DUPLICATE:
  141. case Topology::PEER_VERIFY_REJECTED_DUPLICATE_TRIAGED: {
  142. Packet outp(req->source,_r->identity.address(),Packet::VERB_ERROR);
  143. outp.append((unsigned char)Packet::VERB_HELLO);
  144. outp.append(req->helloPacketId);
  145. outp.append((unsigned char)Packet::ERROR_IDENTITY_COLLISION);
  146. outp.encrypt(p->cryptKey());
  147. outp.hmacSet(p->macKey());
  148. _r->demarc->send(req->localPort,req->remoteAddress,outp.data(),outp.size(),-1);
  149. } break;
  150. }
  151. delete req;
  152. }
  153. void PacketDecoder::_CBaddPeerFromWhois(void *arg,const SharedPtr<Peer> &p,Topology::PeerVerifyResult result)
  154. {
  155. switch(result) {
  156. case Topology::PEER_VERIFY_ACCEPTED_NEW:
  157. case Topology::PEER_VERIFY_ACCEPTED_ALREADY_HAVE:
  158. case Topology::PEER_VERIFY_ACCEPTED_DISPLACED_INVALID_ADDRESS:
  159. ((const RuntimeEnvironment *)arg)->sw->doAnythingWaitingForPeer(p);
  160. break;
  161. default:
  162. break;
  163. }
  164. }
  165. bool PacketDecoder::_doERROR(const RuntimeEnvironment *_r,const SharedPtr<Peer> &peer)
  166. {
  167. try {
  168. #ifdef ZT_TRACE
  169. Packet::Verb inReVerb = (Packet::Verb)(*this)[ZT_PROTO_VERB_ERROR_IDX_IN_RE_VERB];
  170. Packet::ErrorCode errorCode = (Packet::ErrorCode)(*this)[ZT_PROTO_VERB_ERROR_IDX_ERROR_CODE];
  171. TRACE("ERROR %s from %s(%s) in-re %s",Packet::errorString(errorCode),source().toString().c_str(),_remoteAddress.toString().c_str(),Packet::verbString(inReVerb));
  172. #endif
  173. // TODO (sorta):
  174. // The fact is that the protocol works fine without error handling.
  175. // The only error that really needs to be handled here is duplicate
  176. // identity collision, which if it comes from a supernode should cause
  177. // us to restart and regenerate a new identity.
  178. } catch (std::exception &ex) {
  179. TRACE("dropped ERROR from %s(%s): unexpected exception: %s",source().toString().c_str(),_remoteAddress.toString().c_str(),ex.what());
  180. } catch ( ... ) {
  181. TRACE("dropped ERROR from %s(%s): unexpected exception: (unknown)",source().toString().c_str(),_remoteAddress.toString().c_str());
  182. }
  183. return true;
  184. }
  185. bool PacketDecoder::_doHELLO(const RuntimeEnvironment *_r)
  186. {
  187. try {
  188. //unsigned int protoVersion = (*this)[ZT_PROTO_VERB_HELLO_IDX_PROTOCOL_VERSION];
  189. unsigned int vMajor = (*this)[ZT_PROTO_VERB_HELLO_IDX_MAJOR_VERSION];
  190. unsigned int vMinor = (*this)[ZT_PROTO_VERB_HELLO_IDX_MINOR_VERSION];
  191. unsigned int vRevision = at<uint16_t>(ZT_PROTO_VERB_HELLO_IDX_REVISION);
  192. uint64_t timestamp = at<uint64_t>(ZT_PROTO_VERB_HELLO_IDX_TIMESTAMP);
  193. Identity id(*this,ZT_PROTO_VERB_HELLO_IDX_IDENTITY);
  194. SharedPtr<Peer> candidate(new Peer(_r->identity,id));
  195. candidate->setPathAddress(_remoteAddress,false);
  196. // Initial sniff test
  197. if (id.address().isReserved()) {
  198. TRACE("rejected HELLO from %s(%s): identity has reserved address",source().toString().c_str(),_remoteAddress.toString().c_str());
  199. Packet outp(source(),_r->identity.address(),Packet::VERB_ERROR);
  200. outp.append((unsigned char)Packet::VERB_HELLO);
  201. outp.append(packetId());
  202. outp.append((unsigned char)Packet::ERROR_IDENTITY_INVALID);
  203. outp.encrypt(candidate->cryptKey());
  204. outp.hmacSet(candidate->macKey());
  205. _r->demarc->send(_localPort,_remoteAddress,outp.data(),outp.size(),-1);
  206. return true;
  207. }
  208. if (id.address() != source()) {
  209. TRACE("rejected HELLO from %s(%s): identity is not for sender of packet (HELLO is a self-announcement)",source().toString().c_str(),_remoteAddress.toString().c_str());
  210. Packet outp(source(),_r->identity.address(),Packet::VERB_ERROR);
  211. outp.append((unsigned char)Packet::VERB_HELLO);
  212. outp.append(packetId());
  213. outp.append((unsigned char)Packet::ERROR_INVALID_REQUEST);
  214. outp.encrypt(candidate->cryptKey());
  215. outp.hmacSet(candidate->macKey());
  216. _r->demarc->send(_localPort,_remoteAddress,outp.data(),outp.size(),-1);
  217. return true;
  218. }
  219. // Is this a HELLO for a peer we already know? If so just update its
  220. // packet receive stats and send an OK.
  221. SharedPtr<Peer> existingPeer(_r->topology->getPeer(id.address()));
  222. if ((existingPeer)&&(existingPeer->identity() == id)) {
  223. existingPeer->onReceive(_r,_localPort,_remoteAddress,hops(),Packet::VERB_HELLO,Utils::now());
  224. existingPeer->setRemoteVersion(vMajor,vMinor,vRevision);
  225. Packet outp(source(),_r->identity.address(),Packet::VERB_OK);
  226. outp.append((unsigned char)Packet::VERB_HELLO);
  227. outp.append(packetId());
  228. outp.append(timestamp);
  229. outp.encrypt(existingPeer->cryptKey());
  230. outp.hmacSet(existingPeer->macKey());
  231. _r->demarc->send(_localPort,_remoteAddress,outp.data(),outp.size(),-1);
  232. return true;
  233. }
  234. // Otherwise we call addPeer() and set up a callback to handle the verdict
  235. _CBaddPeerFromHello_Data *arg = new _CBaddPeerFromHello_Data;
  236. arg->renv = _r;
  237. arg->source = source();
  238. arg->remoteAddress = _remoteAddress;
  239. arg->localPort = _localPort;
  240. arg->vMajor = vMajor;
  241. arg->vMinor = vMinor;
  242. arg->vRevision = vRevision;
  243. arg->helloPacketId = packetId();
  244. arg->helloTimestamp = timestamp;
  245. _r->topology->addPeer(candidate,&PacketDecoder::_CBaddPeerFromHello,arg);
  246. } catch (std::exception &ex) {
  247. TRACE("dropped HELLO from %s(%s): %s",source().toString().c_str(),_remoteAddress.toString().c_str(),ex.what());
  248. } catch ( ... ) {
  249. TRACE("dropped HELLO from %s(%s): unexpected exception",source().toString().c_str(),_remoteAddress.toString().c_str());
  250. }
  251. return true;
  252. }
  253. bool PacketDecoder::_doOK(const RuntimeEnvironment *_r,const SharedPtr<Peer> &peer)
  254. {
  255. try {
  256. Packet::Verb inReVerb = (Packet::Verb)(*this)[ZT_PROTO_VERB_OK_IDX_IN_RE_VERB];
  257. switch(inReVerb) {
  258. case Packet::VERB_HELLO: {
  259. // OK from HELLO permits computation of latency.
  260. unsigned int latency = std::min((unsigned int)(Utils::now() - at<uint64_t>(ZT_PROTO_VERB_HELLO__OK__IDX_TIMESTAMP)),(unsigned int)0xffff);
  261. TRACE("%s(%s): OK(HELLO), latency: %u",source().toString().c_str(),_remoteAddress.toString().c_str(),latency);
  262. peer->setLatency(_remoteAddress,latency);
  263. } break;
  264. case Packet::VERB_WHOIS:
  265. // Right now we only query supernodes for WHOIS and only accept
  266. // OK back from them. If we query other nodes, we'll have to
  267. // do something to prevent WHOIS cache poisoning such as
  268. // using the packet ID field in the OK packet to match with the
  269. // original query. Technically we should be doing this anyway.
  270. TRACE("%s(%s): OK(%s)",source().toString().c_str(),_remoteAddress.toString().c_str(),Packet::verbString(inReVerb));
  271. if (_r->topology->isSupernode(source()))
  272. _r->topology->addPeer(SharedPtr<Peer>(new Peer(_r->identity,Identity(*this,ZT_PROTO_VERB_WHOIS__OK__IDX_IDENTITY))),&PacketDecoder::_CBaddPeerFromWhois,const_cast<void *>((const void *)_r));
  273. break;
  274. default:
  275. //TRACE("%s(%s): OK(%s)",source().toString().c_str(),_remoteAddress.toString().c_str(),Packet::verbString(inReVerb));
  276. break;
  277. }
  278. } catch (std::exception &ex) {
  279. TRACE("dropped OK from %s(%s): unexpected exception: %s",source().toString().c_str(),_remoteAddress.toString().c_str(),ex.what());
  280. } catch ( ... ) {
  281. TRACE("dropped OK from %s(%s): unexpected exception: (unknown)",source().toString().c_str(),_remoteAddress.toString().c_str());
  282. }
  283. return true;
  284. }
  285. bool PacketDecoder::_doWHOIS(const RuntimeEnvironment *_r,const SharedPtr<Peer> &peer)
  286. {
  287. if (payloadLength() == ZT_ADDRESS_LENGTH) {
  288. SharedPtr<Peer> p(_r->topology->getPeer(Address(payload(),ZT_ADDRESS_LENGTH)));
  289. if (p) {
  290. Packet outp(source(),_r->identity.address(),Packet::VERB_OK);
  291. outp.append((unsigned char)Packet::VERB_WHOIS);
  292. outp.append(packetId());
  293. p->identity().serialize(outp,false);
  294. outp.encrypt(peer->cryptKey());
  295. outp.hmacSet(peer->macKey());
  296. _r->demarc->send(_localPort,_remoteAddress,outp.data(),outp.size(),-1);
  297. TRACE("sent WHOIS response to %s for %s",source().toString().c_str(),Address(payload(),ZT_ADDRESS_LENGTH).toString().c_str());
  298. } else {
  299. Packet outp(source(),_r->identity.address(),Packet::VERB_ERROR);
  300. outp.append((unsigned char)Packet::VERB_WHOIS);
  301. outp.append(packetId());
  302. outp.append((unsigned char)Packet::ERROR_NOT_FOUND);
  303. outp.append(payload(),ZT_ADDRESS_LENGTH);
  304. outp.encrypt(peer->cryptKey());
  305. outp.hmacSet(peer->macKey());
  306. _r->demarc->send(_localPort,_remoteAddress,outp.data(),outp.size(),-1);
  307. TRACE("sent WHOIS ERROR to %s for %s (not found)",source().toString().c_str(),Address(payload(),ZT_ADDRESS_LENGTH).toString().c_str());
  308. }
  309. } else {
  310. TRACE("dropped WHOIS from %s(%s): missing or invalid address",source().toString().c_str(),_remoteAddress.toString().c_str());
  311. }
  312. return true;
  313. }
  314. bool PacketDecoder::_doRENDEZVOUS(const RuntimeEnvironment *_r,const SharedPtr<Peer> &peer)
  315. {
  316. try {
  317. Address with(field(ZT_PROTO_VERB_RENDEZVOUS_IDX_ZTADDRESS,ZT_ADDRESS_LENGTH),ZT_ADDRESS_LENGTH);
  318. SharedPtr<Peer> withPeer(_r->topology->getPeer(with));
  319. if (withPeer) {
  320. unsigned int port = at<uint16_t>(ZT_PROTO_VERB_RENDEZVOUS_IDX_PORT);
  321. unsigned int addrlen = (*this)[ZT_PROTO_VERB_RENDEZVOUS_IDX_ADDRLEN];
  322. if ((port > 0)&&((addrlen == 4)||(addrlen == 16))) {
  323. InetAddress atAddr(field(ZT_PROTO_VERB_RENDEZVOUS_IDX_ADDRESS,addrlen),addrlen,port);
  324. TRACE("RENDEZVOUS from %s says %s might be at %s, starting NAT-t",source().toString().c_str(),with.toString().c_str(),atAddr.toString().c_str());
  325. _r->sw->contact(withPeer,atAddr);
  326. } else {
  327. TRACE("dropped corrupt RENDEZVOUS from %s(%s) (bad address or port)",source().toString().c_str(),_remoteAddress.toString().c_str());
  328. }
  329. } else {
  330. TRACE("ignored RENDEZVOUS from %s(%s) to meet unknown peer %s",source().toString().c_str(),_remoteAddress.toString().c_str(),with.toString().c_str());
  331. }
  332. } catch (std::exception &ex) {
  333. TRACE("dropped RENDEZVOUS from %s(%s): %s",source().toString().c_str(),_remoteAddress.toString().c_str(),ex.what());
  334. } catch ( ... ) {
  335. TRACE("dropped RENDEZVOUS from %s(%s): unexpected exception",source().toString().c_str(),_remoteAddress.toString().c_str());
  336. }
  337. return true;
  338. }
  339. bool PacketDecoder::_doFRAME(const RuntimeEnvironment *_r,const SharedPtr<Peer> &peer)
  340. {
  341. try {
  342. SharedPtr<Network> network(_r->nc->network(at<uint64_t>(ZT_PROTO_VERB_FRAME_IDX_NETWORK_ID)));
  343. if (network) {
  344. if (network->isAllowed(source())) {
  345. unsigned int etherType = at<uint16_t>(ZT_PROTO_VERB_FRAME_IDX_ETHERTYPE);
  346. if ((etherType != ZT_ETHERTYPE_ARP)&&(etherType != ZT_ETHERTYPE_IPV4)&&(etherType != ZT_ETHERTYPE_IPV6)) {
  347. TRACE("dropped FRAME from %s: unsupported ethertype",source().toString().c_str());
  348. } else if (size() > ZT_PROTO_VERB_FRAME_IDX_PAYLOAD) {
  349. network->tap().put(source().toMAC(),network->tap().mac(),etherType,data() + ZT_PROTO_VERB_FRAME_IDX_PAYLOAD,size() - ZT_PROTO_VERB_FRAME_IDX_PAYLOAD);
  350. }
  351. } else {
  352. TRACE("dropped FRAME from %s(%s): not a member of closed network %llu",source().toString().c_str(),_remoteAddress.toString().c_str(),network->id());
  353. }
  354. } else {
  355. TRACE("dropped FRAME from %s(%s): network %llu unknown",source().toString().c_str(),_remoteAddress.toString().c_str(),at<uint64_t>(ZT_PROTO_VERB_FRAME_IDX_NETWORK_ID));
  356. }
  357. } catch (std::exception &ex) {
  358. TRACE("dropped FRAME from %s(%s): unexpected exception: %s",source().toString().c_str(),_remoteAddress.toString().c_str(),ex.what());
  359. } catch ( ... ) {
  360. TRACE("dropped FRAME from %s(%s): unexpected exception: (unknown)",source().toString().c_str(),_remoteAddress.toString().c_str());
  361. }
  362. return true;
  363. }
  364. bool PacketDecoder::_doMULTICAST_LIKE(const RuntimeEnvironment *_r,const SharedPtr<Peer> &peer)
  365. {
  366. try {
  367. unsigned int ptr = ZT_PACKET_IDX_PAYLOAD;
  368. unsigned int numAccepted = 0;
  369. uint64_t now = Utils::now();
  370. // Iterate through 18-byte network,MAC,ADI tuples:
  371. while ((ptr + 18) <= size()) {
  372. uint64_t nwid = at<uint64_t>(ptr); ptr += 8;
  373. SharedPtr<Network> network(_r->nc->network(nwid));
  374. if (network) {
  375. if (network->isAllowed(source())) {
  376. MAC mac(field(ptr,6)); ptr += 6;
  377. uint32_t adi = at<uint32_t>(ptr); ptr += 4;
  378. //TRACE("peer %s likes multicast group %s:%.8lx on network %llu",source().toString().c_str(),mac.toString().c_str(),(unsigned long)adi,nwid);
  379. _r->multicaster->likesMulticastGroup(nwid,MulticastGroup(mac,adi),source(),now);
  380. ++numAccepted;
  381. } else {
  382. TRACE("ignored MULTICAST_LIKE from %s(%s): not a member of closed network %llu",source().toString().c_str(),_remoteAddress.toString().c_str(),nwid);
  383. }
  384. } else {
  385. TRACE("ignored MULTICAST_LIKE from %s(%s): network %llu unknown or we are not a member",source().toString().c_str(),_remoteAddress.toString().c_str(),nwid);
  386. }
  387. }
  388. Packet outp(source(),_r->identity.address(),Packet::VERB_OK);
  389. outp.append((unsigned char)Packet::VERB_MULTICAST_LIKE);
  390. outp.append(packetId());
  391. outp.append((uint16_t)numAccepted);
  392. outp.encrypt(peer->cryptKey());
  393. outp.hmacSet(peer->macKey());
  394. _r->demarc->send(_localPort,_remoteAddress,outp.data(),outp.size(),-1);
  395. } catch (std::exception &ex) {
  396. TRACE("dropped MULTICAST_LIKE from %s(%s): unexpected exception: %s",source().toString().c_str(),_remoteAddress.toString().c_str(),ex.what());
  397. } catch ( ... ) {
  398. TRACE("dropped MULTICAST_LIKE from %s(%s): unexpected exception: (unknown)",source().toString().c_str(),_remoteAddress.toString().c_str());
  399. }
  400. return true;
  401. }
  402. bool PacketDecoder::_doMULTICAST_FRAME(const RuntimeEnvironment *_r,const SharedPtr<Peer> &peer)
  403. {
  404. try {
  405. SharedPtr<Network> network(_r->nc->network(at<uint64_t>(ZT_PROTO_VERB_MULTICAST_FRAME_IDX_NETWORK_ID)));
  406. if (network) {
  407. if (network->isAllowed(source())) {
  408. if (size() > ZT_PROTO_VERB_MULTICAST_FRAME_IDX_PAYLOAD) {
  409. Address originalSubmitterAddress(field(ZT_PROTO_VERB_MULTICAST_FRAME_IDX_SUBMITTER_ADDRESS,ZT_ADDRESS_LENGTH),ZT_ADDRESS_LENGTH);
  410. MAC fromMac(field(ZT_PROTO_VERB_MULTICAST_FRAME_IDX_SOURCE_MAC,6));
  411. MulticastGroup mg(MAC(field(ZT_PROTO_VERB_MULTICAST_FRAME_IDX_DESTINATION_MAC,6)),at<uint32_t>(ZT_PROTO_VERB_MULTICAST_FRAME_IDX_ADI));
  412. unsigned int hops = (*this)[ZT_PROTO_VERB_MULTICAST_FRAME_IDX_HOP_COUNT];
  413. unsigned int etherType = at<uint16_t>(ZT_PROTO_VERB_MULTICAST_FRAME_IDX_ETHERTYPE);
  414. unsigned int datalen = at<uint16_t>(ZT_PROTO_VERB_MULTICAST_FRAME_IDX_PAYLOAD_LENGTH);
  415. unsigned int signaturelen = at<uint16_t>(ZT_PROTO_VERB_MULTICAST_FRAME_IDX_SIGNATURE_LENGTH);
  416. unsigned char *dataAndSignature = field(ZT_PROTO_VERB_MULTICAST_FRAME_IDX_PAYLOAD,datalen + signaturelen);
  417. uint64_t mccrc = Multicaster::computeMulticastDedupCrc(network->id(),fromMac,mg,etherType,dataAndSignature,datalen);
  418. uint64_t now = Utils::now();
  419. bool isDuplicate = _r->multicaster->checkDuplicate(mccrc,now);
  420. if (originalSubmitterAddress == _r->identity.address()) {
  421. // Technically should not happen, since the original submitter is
  422. // excluded from consideration as a propagation recipient.
  423. TRACE("dropped boomerang MULTICAST_FRAME received from %s(%s)",source().toString().c_str(),_remoteAddress.toString().c_str());
  424. } else if ((!isDuplicate)||(_r->topology->amSupernode())) {
  425. //
  426. // If I am a supernode, I will repeatedly propagate duplicates. That's
  427. // because supernodes are used to bridge sparse multicast groups. Non-
  428. // supernodes will ignore duplicates completely.
  429. //
  430. // TODO: supernodes should keep a local bloom filter too and OR it with
  431. // the bloom from the packet in order to pick different recipients each
  432. // time a multicast returns to them for repropagation.
  433. //
  434. SharedPtr<Peer> originalSubmitter(_r->topology->getPeer(originalSubmitterAddress));
  435. if (!originalSubmitter) {
  436. TRACE("requesting WHOIS on original multicast frame submitter %s",originalSubmitterAddress.toString().c_str());
  437. _r->sw->requestWhois(originalSubmitterAddress);
  438. _step = DECODE_STEP_WAITING_FOR_ORIGINAL_SUBMITTER_LOOKUP;
  439. return false; // try again if/when we get OK(WHOIS)
  440. } else if (Multicaster::verifyMulticastPacket(originalSubmitter->identity(),network->id(),fromMac,mg,etherType,dataAndSignature,datalen,dataAndSignature + datalen,signaturelen)) {
  441. _r->multicaster->addToDedupHistory(mccrc,now);
  442. // Even if we are a supernode, we still don't repeatedly inject
  443. // duplicates into our own tap.
  444. if (!isDuplicate)
  445. network->tap().put(fromMac,mg.mac(),etherType,dataAndSignature,datalen);
  446. if (++hops < ZT_MULTICAST_PROPAGATION_DEPTH) {
  447. Address upstream(source()); // save this since we mangle it
  448. Multicaster::MulticastBloomFilter bloom(field(ZT_PROTO_VERB_MULTICAST_FRAME_IDX_BLOOM_FILTER,ZT_PROTO_VERB_MULTICAST_FRAME_BLOOM_FILTER_SIZE_BYTES));
  449. SharedPtr<Peer> propPeers[ZT_MULTICAST_PROPAGATION_BREADTH];
  450. unsigned int np = _r->multicaster->pickNextPropagationPeers(
  451. *(_r->prng),
  452. *(_r->topology),
  453. network->id(),
  454. mg,
  455. originalSubmitterAddress,
  456. upstream,
  457. bloom,
  458. ZT_MULTICAST_PROPAGATION_BREADTH,
  459. propPeers,
  460. now);
  461. // In a bit of a hack, we re-use this packet to repeat it
  462. // to our multicast propagation recipients. Afterwords we
  463. // return true just to be sure this is the end of this
  464. // packet's life cycle, since it is now mangled.
  465. setSource(_r->identity.address());
  466. (*this)[ZT_PROTO_VERB_MULTICAST_FRAME_IDX_HOP_COUNT] = hops;
  467. memcpy(field(ZT_PROTO_VERB_MULTICAST_FRAME_IDX_BLOOM_FILTER,ZT_PROTO_VERB_MULTICAST_FRAME_BLOOM_FILTER_SIZE_BYTES),bloom.data(),ZT_PROTO_VERB_MULTICAST_FRAME_BLOOM_FILTER_SIZE_BYTES);
  468. compress();
  469. for(unsigned int i=0;i<np;++i) {
  470. //TRACE("propagating multicast from original node %s: %s -> %s",originalSubmitterAddress.toString().c_str(),upstream.toString().c_str(),propPeers[i]->address().toString().c_str());
  471. // Re-use this packet to re-send multicast frame to everyone
  472. // downstream from us.
  473. newInitializationVector();
  474. setDestination(propPeers[i]->address());
  475. _r->sw->send(*this,true);
  476. }
  477. return true;
  478. } else {
  479. //TRACE("terminating MULTICAST_FRAME propagation from %s(%s): max depth reached",source().toString().c_str(),_remoteAddress.toString().c_str());
  480. }
  481. } else {
  482. LOG("rejected MULTICAST_FRAME from %s(%s) due to failed signature check (claims original sender %s)",source().toString().c_str(),_remoteAddress.toString().c_str(),originalSubmitterAddress.toString().c_str());
  483. }
  484. } else {
  485. TRACE("dropped redundant MULTICAST_FRAME from %s(%s)",source().toString().c_str(),_remoteAddress.toString().c_str());
  486. }
  487. } else {
  488. TRACE("dropped MULTICAST_FRAME from %s(%s): invalid short packet",source().toString().c_str(),_remoteAddress.toString().c_str());
  489. }
  490. } else {
  491. TRACE("dropped MULTICAST_FRAME from %s(%s): not a member of closed network %llu",source().toString().c_str(),_remoteAddress.toString().c_str(),network->id());
  492. }
  493. } else {
  494. TRACE("dropped MULTICAST_FRAME from %s(%s): network %llu unknown or we are not a member",source().toString().c_str(),_remoteAddress.toString().c_str(),at<uint64_t>(ZT_PROTO_VERB_MULTICAST_FRAME_IDX_NETWORK_ID));
  495. }
  496. } catch (std::exception &ex) {
  497. TRACE("dropped MULTICAST_FRAME from %s(%s): unexpected exception: %s",source().toString().c_str(),_remoteAddress.toString().c_str(),ex.what());
  498. } catch ( ... ) {
  499. TRACE("dropped MULTICAST_FRAME from %s(%s): unexpected exception: (unknown)",source().toString().c_str(),_remoteAddress.toString().c_str());
  500. }
  501. return true;
  502. }
  503. bool PacketDecoder::_doNETWORK_MEMBERSHIP_CERTIFICATE(const RuntimeEnvironment *_r,const SharedPtr<Peer> &peer)
  504. {
  505. }
  506. bool PacketDecoder::_doNETWORK_CONFIG_REQUEST(const RuntimeEnvironment *_r,const SharedPtr<Peer> &peer)
  507. {
  508. }
  509. bool PacketDecoder::_doNETWORK_CONFIG_REFRESH(const RuntimeEnvironment *_r,const SharedPtr<Peer> &peer)
  510. {
  511. }
  512. } // namespace ZeroTier