Browse Source

Fixes to PUSH_DIRECT_PATHS.

Adam Ierymenko 10 years ago
parent
commit
4bf3bcbd55
2 changed files with 9 additions and 5 deletions
  1. 8 4
      node/IncomingPacket.cpp
  2. 1 1
      node/Peer.cpp

+ 8 - 4
node/IncomingPacket.cpp

@@ -906,7 +906,7 @@ bool IncomingPacket::_doPUSH_DIRECT_PATHS(const RuntimeEnvironment *RR,const Sha
 		unsigned int count = at<uint16_t>(ZT_PACKET_IDX_PAYLOAD);
 		unsigned int count = at<uint16_t>(ZT_PACKET_IDX_PAYLOAD);
 		unsigned int ptr = ZT_PACKET_IDX_PAYLOAD + 2;
 		unsigned int ptr = ZT_PACKET_IDX_PAYLOAD + 2;
 
 
-		while (count) { // if ptr overflows Buffer will throw
+		while (count--) { // if ptr overflows Buffer will throw
 			// TODO: properly handle blacklisting, support other features... see Packet.hpp.
 			// TODO: properly handle blacklisting, support other features... see Packet.hpp.
 
 
 			unsigned int flags = (*this)[ptr++];
 			unsigned int flags = (*this)[ptr++];
@@ -914,18 +914,22 @@ bool IncomingPacket::_doPUSH_DIRECT_PATHS(const RuntimeEnvironment *RR,const Sha
 			unsigned int extLen = at<uint16_t>(ptr); ptr += 2;
 			unsigned int extLen = at<uint16_t>(ptr); ptr += 2;
 			ptr += extLen; // unused right now
 			ptr += extLen; // unused right now
 			unsigned int addrType = (*this)[ptr++];
 			unsigned int addrType = (*this)[ptr++];
-
 			unsigned int addrLen = (*this)[ptr++];
 			unsigned int addrLen = (*this)[ptr++];
+
 			switch(addrType) {
 			switch(addrType) {
 				case 4: {
 				case 4: {
 					InetAddress a(field(ptr,4),4,at<uint16_t>(ptr + 4));
 					InetAddress a(field(ptr,4),4,at<uint16_t>(ptr + 4));
-					if ( ((flags & (0x01 | 0x02)) == 0) && (Path::isAddressValidForPath(a)) )
+					if ( ((flags & (0x01 | 0x02)) == 0) && (Path::isAddressValidForPath(a)) ) {
+						TRACE("attempting to contact %s at pushed direct path %s",peer->address().toString().c_str(),a.toString().c_str());
 						peer->attemptToContactAt(RR,a,RR->node->now());
 						peer->attemptToContactAt(RR,a,RR->node->now());
+					}
 				}	break;
 				}	break;
 				case 6: {
 				case 6: {
 					InetAddress a(field(ptr,16),16,at<uint16_t>(ptr + 16));
 					InetAddress a(field(ptr,16),16,at<uint16_t>(ptr + 16));
-					if ( ((flags & (0x01 | 0x02)) == 0) && (Path::isAddressValidForPath(a)) )
+					if ( ((flags & (0x01 | 0x02)) == 0) && (Path::isAddressValidForPath(a)) ) {
+						TRACE("attempting to contact %s at pushed direct path %s",peer->address().toString().c_str(),a.toString().c_str());
 						peer->attemptToContactAt(RR,a,RR->node->now());
 						peer->attemptToContactAt(RR,a,RR->node->now());
+					}
 				}	break;
 				}	break;
 			}
 			}
 			ptr += addrLen;
 			ptr += addrLen;

+ 1 - 1
node/Peer.cpp

@@ -255,7 +255,7 @@ void Peer::pushDirectPaths(const RuntimeEnvironment *RR,RemotePath *path,uint64_
 
 
 				outp.append(flags);
 				outp.append(flags);
 				outp.append((uint8_t)((p->metric() >= 0) ? ((p->metric() <= 255) ? p->metric() : 255) : 0));
 				outp.append((uint8_t)((p->metric() >= 0) ? ((p->metric() <= 255) ? p->metric() : 255) : 0));
-				outp.append((uint16_t)0);
+				outp.append((uint16_t)0); // no extensions
 				outp.append(addressType);
 				outp.append(addressType);
 				outp.append((uint8_t)((addressType == 4) ? 6 : 18));
 				outp.append((uint8_t)((addressType == 4) ? 6 : 18));
 				outp.append(p->address().rawIpData(),((addressType == 4) ? 4 : 16));
 				outp.append(p->address().rawIpData(),((addressType == 4) ? 4 : 16));