Fix memory exhaustion

Truncate messages to the size of the tx in them
Tranz5 · May 20, 2014 · 45a18e4 · 45a18e4
1 parent da048f5
commit 45a18e4
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/main.cpp b/src/main.cpp
@@ -3377,6 +3377,11 @@ bool static ProcessMessage(CNode* pfrom, string strCommand, CDataStream& vRecv)
         CInv inv(MSG_TX, tx.GetHash());
         pfrom->AddInventoryKnown(inv);
 
+        // Truncate messages to the size of the tx in them
+        unsigned int nSize = ::GetSerializeSize(tx,SER_NETWORK, PROTOCOL_VERSION);
+        if (nSize < vMsg.size())
+            vMsg.resize(nSize);
+
         bool fMissingInputs = false;
         if (tx.AcceptToMemoryPool(txdb, true, &fMissingInputs))
         {