From ba04c4a7801e7d68a5e84035b919e5c3626eb7a7 Mon Sep 17 00:00:00 2001
From: Pieter Wuille <pieter.wuille@gmail.com>
Date: Thu, 5 Mar 2015 04:01:22 -0800
Subject: Limit message sizes before transfer

This introduces a fixed limit for the size of p2p messages, and enforces it
before download.
---
 src/net.cpp | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'src/net.cpp')

diff --git a/src/net.cpp b/src/net.cpp
index 3c3666615..d9f470b9e 100644
--- a/src/net.cpp
+++ b/src/net.cpp
@@ -523,6 +523,11 @@ bool CNode::ReceiveMsgBytes(const char *pch, unsigned int nBytes)
         if (handled < 0)
                 return false;
 
+        if (msg.in_data && msg.hdr.nMessageSize > MAX_PROTOCOL_MESSAGE_LENGTH) {
+            LogPrint("net", "Oversized message from peer=%i, disconnecting", GetId());
+            return false;
+        }
+
         pch += handled;
         nBytes -= handled;
 
-- 
cgit v1.2.3