Fix memory exhaustion

TrezarCoin · Jun 20, 2013 · 5a4c7d3 · 5a4c7d3
1 parent a651f1e
commit 5a4c7d3
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/main.cpp b/src/main.cpp
@@ -3432,6 +3432,12 @@ 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))
         {