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)
// Checksum // Checksum
CDataStream& vRecv = msg.vRecv; 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) if (memcmp(hash.begin(), hdr.pchChecksum, CMessageHeader::CHECKSUM_SIZE) != 0)
{ {
LogPrintf("%s(%s, %u bytes): CHECKSUM ERROR expected %s was %s\n", __func__, 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)
vRecv.resize(std::min(hdr.nMessageSize, nDataPos + nCopy + 256 * 1024)); vRecv.resize(std::min(hdr.nMessageSize, nDataPos + nCopy + 256 * 1024));
} }
hasher.Write((const unsigned char*)pch, nCopy);
memcpy(&vRecv[nDataPos], pch, nCopy); memcpy(&vRecv[nDataPos], pch, nCopy);
nDataPos += nCopy; nDataPos += nCopy;
return 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:
class CNetMessage { class CNetMessage {
private:
mutable CHash256 hasher;
mutable uint256 data_hash;
public: public:
bool in_data; // parsing header (false) or data (true) bool in_data; // parsing header (false) or data (true)
@ -539,6 +542,8 @@ public:
return (hdr.nMessageSize == nDataPos); return (hdr.nMessageSize == nDataPos);
} }
const uint256& GetMessageHash() const;
void SetVersion(int nVersionIn) void SetVersion(int nVersionIn)
{ {
hdrbuf.SetVersion(nVersionIn); hdrbuf.SetVersion(nVersionIn);

Loading…
Cancel
Save