net: add nSendBufferMaxSize/nReceiveFloodSize to CConnection::Options

theuni · Fuzzbawls · commit 276c94643db2 · 2020-08-25T15:25:02.000-07:00
diff --git a/src/init.cpp b/src/init.cpp
@@ -1894,6 +1894,8 @@ bool AppInit2()
     connOptions.nMaxOutbound = std::min(MAX_OUTBOUND_CONNECTIONS, connOptions.nMaxConnections);
     connOptions.nBestHeight = chainActive.Height();
     connOptions.uiInterface = &uiInterface;
+    connOptions.nSendBufferMaxSize = 1000*GetArg("-maxsendbuffer", DEFAULT_MAXSENDBUFFER);
+    connOptions.nReceiveFloodSize = 1000*GetArg("-maxreceivebuffer", DEFAULT_MAXRECEIVEBUFFER);
 
     if(!connman.Start(threadGroup, scheduler, strNodeError, connOptions))
         return UIError(strNodeError);
diff --git a/src/net.cpp b/src/net.cpp
@@ -1976,8 +1976,8 @@ bool CConnman::Start(boost::thread_group& threadGroup, CScheduler& scheduler, st
     nMaxConnections = connOptions.nMaxConnections;
     nMaxOutbound = std::min((connOptions.nMaxOutbound), nMaxConnections);
 
-    nSendBufferMaxSize = 1000*GetArg("-maxsendbuffer", DEFAULT_MAXSENDBUFFER);
-    nReceiveFloodSize = 1000*GetArg("-maxreceivebuffer", DEFAULT_MAXRECEIVEBUFFER);
+    nSendBufferMaxSize = connOptions.nSendBufferMaxSize;
+    nReceiveFloodSize = connOptions.nSendBufferMaxSize;
 
     SetBestHeight(connOptions.nBestHeight);
 
diff --git a/src/net.h b/src/net.h
@@ -121,6 +121,8 @@ class CConnman
         int nMaxOutbound = 0;
         int nBestHeight = 0;
         CClientUIInterface* uiInterface = nullptr;
+        unsigned int nSendBufferMaxSize = 0;
+        unsigned int nReceiveFloodSize = 0;
     };
     CConnman();
     ~CConnman();
