From e7c81ef34e9db426d6ba72e2feae4180233631e5 Mon Sep 17 00:00:00 2001 From: Adam Ierymenko Date: Fri, 3 Oct 2014 22:03:19 -0700 Subject: Turns out that needed to be a list after all. Also clean up Multicaster::gather(). --- node/Multicaster.cpp | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) (limited to 'node/Multicaster.cpp') diff --git a/node/Multicaster.cpp b/node/Multicaster.cpp index b3c0abd8..6f3c4395 100644 --- a/node/Multicaster.cpp +++ b/node/Multicaster.cpp @@ -66,12 +66,13 @@ unsigned int Multicaster::gather(const RuntimeEnvironment *RR,const Address &que if (limit > gs->second.members.size()) limit = (unsigned int)gs->second.members.size(); - if (limit > 0xffff) // sanity check -- this won't fit in a packet anyway - limit = 0xffff; + if (limit > ((ZT_PROTO_MAX_PACKET_LENGTH / 5) + 1)) + limit = (ZT_PROTO_MAX_PACKET_LENGTH / 5) + 1; - appendTo.append((uint32_t)gs->second.members.size()); + unsigned int totalAt = appendTo.size(); + appendTo.addSize(4); // sizeof(uint32_t) unsigned int nAt = appendTo.size(); - appendTo.append((uint16_t)0); // set to n later + appendTo.addSize(2); // sizeof(uint16_t) while ((n < limit)&&((appendTo.size() + ZT_ADDRESS_LENGTH) <= ZT_PROTO_MAX_PACKET_LENGTH)) { // Pick a member at random -- if we've already picked it, @@ -103,6 +104,7 @@ restart_member_scan: } } + appendTo.setAt(totalAt,(uint32_t)(gs->second.members.size() - skipped)); appendTo.setAt(nAt,(uint16_t)(n - skipped)); return n; -- cgit v1.2.3