Browse Source

Hash P2P messages as they are received instead of at process-time

0.14
Matt Corallo 8 years ago
parent
commit
fe1dc62cef
  1. 2
      src/main.cpp
  2. 9
      src/net.cpp
  3. 5
      src/net.h

2
src/main.cpp

@ -6350,7 +6350,7 @@ bool ProcessMessages(CNode* pfrom, CConnman& connman) @@ -6350,7 +6350,7 @@ bool ProcessMessages(CNode* pfrom, CConnman& connman)
// Checksum
CDataStream& vRecv = msg.vRecv;
uint256 hash = Hash(vRecv.begin(), vRecv.begin() + nMessageSize);
const uint256& hash = msg.GetMessageHash();
if (memcmp(hash.begin(), hdr.pchChecksum, CMessageHeader::CHECKSUM_SIZE) != 0)
{
LogPrintf("%s(%s, %u bytes): CHECKSUM ERROR expected %s was %s\n", __func__,

9
src/net.cpp

@ -758,12 +758,21 @@ int CNetMessage::readData(const char *pch, unsigned int nBytes) @@ -758,12 +758,21 @@ int CNetMessage::readData(const char *pch, unsigned int nBytes)
vRecv.resize(std::min(hdr.nMessageSize, nDataPos + nCopy + 256 * 1024));
}
hasher.Write((const unsigned char*)pch, nCopy);
memcpy(&vRecv[nDataPos], pch, nCopy);
nDataPos += nCopy;
return nCopy;
}
const uint256& CNetMessage::GetMessageHash() const
{
assert(complete());
if (data_hash.IsNull())
hasher.Finalize(data_hash.begin());
return data_hash;
}

5
src/net.h

@ -512,6 +512,9 @@ public: @@ -512,6 +512,9 @@ public:
class CNetMessage {
private:
mutable CHash256 hasher;
mutable uint256 data_hash;
public:
bool in_data; // parsing header (false) or data (true)
@ -539,6 +542,8 @@ public: @@ -539,6 +542,8 @@ public:
return (hdr.nMessageSize == nDataPos);
}
const uint256& GetMessageHash() const;
void SetVersion(int nVersionIn)
{
hdrbuf.SetVersion(nVersionIn);

Loading…
Cancel
Save