don't send address for initial PeerTest

This commit is contained in:
orignal 2014-09-22 16:27:56 -04:00
parent 11402708da
commit 1d9fdca9ce

24
SSU.cpp
View File

@ -755,7 +755,7 @@ namespace ssu
buf += 4; // nonce buf += 4; // nonce
uint8_t size = *buf; uint8_t size = *buf;
buf++; // size buf++; // size
uint8_t * address = (size == 4) ? buf : nullptr; uint32_t address = (size == 4) ? *(uint32_t *)buf : 0; // use it as is
buf += size; // address buf += size; // address
uint16_t port = *(uint16_t *)buf; // use it as is uint16_t port = *(uint16_t *)buf; // use it as is
buf += 2; // port buf += 2; // port
@ -777,7 +777,7 @@ namespace ssu
else if (port) else if (port)
{ {
LogPrint ("SSU peer test from Charlie. We are Bob"); LogPrint ("SSU peer test from Charlie. We are Bob");
boost::asio::ip::udp::endpoint ep (boost::asio::ip::address_v4 (be32toh (*(uint32_t *)address)), be16toh (port)); // Alice's address/port boost::asio::ip::udp::endpoint ep (boost::asio::ip::address_v4 (be32toh (address)), be16toh (port)); // Alice's address/port
auto session = m_Server.FindSession (ep); // find session with Alice auto session = m_Server.FindSession (ep); // find session with Alice
if (session) if (session)
session->Send (PAYLOAD_TYPE_PEER_TEST, buf1, len); // back to Alice session->Send (PAYLOAD_TYPE_PEER_TEST, buf1, len); // back to Alice
@ -797,7 +797,7 @@ namespace ssu
{ {
LogPrint ("SSU peer test from Bob. We are Charlie"); LogPrint ("SSU peer test from Bob. We are Charlie");
Send (PAYLOAD_TYPE_PEER_TEST, buf1, len); // back to Bob Send (PAYLOAD_TYPE_PEER_TEST, buf1, len); // back to Bob
SendPeerTest (nonce, be32toh (*(uint32_t *)address), be16toh (port), introKey); // to Alice SendPeerTest (nonce, be32toh (address), be16toh (port), introKey); // to Alice
} }
else else
{ {
@ -817,11 +817,19 @@ namespace ssu
uint8_t iv[16]; uint8_t iv[16];
uint8_t * payload = buf + sizeof (SSUHeader); uint8_t * payload = buf + sizeof (SSUHeader);
*(uint32_t *)payload = htobe32 (nonce); *(uint32_t *)payload = htobe32 (nonce);
payload += 4; // nonce payload += 4; // nonce
*payload = 4; if (address)
payload++; // size {
*(uint32_t *)payload = htobe32 (address); *payload = 4;
payload += 4; // address payload++; // size
*(uint32_t *)payload = htobe32 (address);
payload += 4; // address
}
else
{
*payload = 0;
payload++; //size
}
*(uint16_t *)payload = htobe16 (port); *(uint16_t *)payload = htobe16 (port);
payload += 2; // port payload += 2; // port
memcpy (payload, introKey, 32); // intro key memcpy (payload, introKey, 32); // intro key