#ifndef BITCOIN_NET_H
#define BITCOIN_NET_H
-#include <limits>
-#include <deque>
-#ifndef Q_MOC_RUN
-#include <boost/array.hpp>
-#endif
-#include <openssl/rand.h>
+#include "mruset.h"
+#include "netbase.h"
+#include "addrman.h"
+#include "hash.h"
+#include "streams.h"
#ifndef WIN32
#include <arpa/inet.h>
#endif
-#include "mruset.h"
-#include "netbase.h"
-#include "addrman.h"
-#include "hash.h"
+#include <limits>
+#include <deque>
+
-class CRequestTracker;
class CNode;
class CBlockIndex;
extern int nBestHeight;
MSG_BLOCK
};
-class CRequestTracker
-{
-public:
- void (*fn)(void*, CDataStream&);
- void* param1;
-
- explicit CRequestTracker(void (*fnIn)(void*, CDataStream&)=NULL, void* param1In=NULL)
- {
- fn = fnIn;
- param1 = param1In;
- }
-
- bool IsNull()
- {
- return fn == NULL;
- }
-};
-
/** Thread types */
enum threadId
extern uint64_t nLocalServices;
extern uint64_t nLocalHostNonce;
extern CAddress addrSeenByPeer;
-extern boost::array<int, THREAD_MAX> vnThreadsRunning;
+extern std::array<int, THREAD_MAX> vnThreadsRunning;
extern CAddrMan addrman;
extern std::vector<CNode*> vNodes;
public:
int64_t nReleaseTime;
- std::map<uint256, CRequestTracker> mapRequests;
- CCriticalSection cs_mapRequests;
uint256 hashContinue;
CBlockIndex* pindexLastGetBlocksBegin;
uint256 hashLastGetBlocksEnd;
mapAskFor.insert(std::make_pair(nRequestTime, inv));
}
-
-
void BeginMessage(const char* pszCommand)
{
ENTER_CRITICAL_SECTION(cs_vSend);
printf("(aborted)\n");
}
- void EndMessage()
- {
- if (mapArgs.count("-dropmessagestest") && GetRand(atoi(mapArgs["-dropmessagestest"])) == 0)
- {
- printf("dropmessages DROPPING SEND MESSAGE\n");
- AbortMessage();
- return;
- }
-
- if (nHeaderStart < 0) {
- LEAVE_CRITICAL_SECTION(cs_vSend);
- return;
- }
-
- // Set the size
- uint32_t nSize = (uint32_t) vSend.size() - nMessageStart;
- memcpy((char*)&vSend[nHeaderStart] + CMessageHeader::MESSAGE_SIZE_OFFSET, &nSize, sizeof(nSize));
-
- // Set the checksum
- uint256 hash = Hash(vSend.begin() + nMessageStart, vSend.end());
- uint32_t nChecksum = 0;
- memcpy(&nChecksum, &hash, sizeof(nChecksum));
- assert(nMessageStart - nHeaderStart >= CMessageHeader::CHECKSUM_OFFSET + sizeof(nChecksum));
- memcpy((char*)&vSend[nHeaderStart] + CMessageHeader::CHECKSUM_OFFSET, &nChecksum, sizeof(nChecksum));
-
- if (fDebug) {
- printf("(%d bytes)\n", nSize);
- }
-
- nHeaderStart = -1;
- nMessageStart = std::numeric_limits<uint32_t>::max();
- LEAVE_CRITICAL_SECTION(cs_vSend);
- }
+ void EndMessage();
void EndMessageAbortIfEmpty()
{
AbortMessage();
}
-
-
void PushVersion();
-
- void PushMessage(const char* pszCommand)
- {
- try
- {
- BeginMessage(pszCommand);
- EndMessage();
- }
- catch (...)
- {
- AbortMessage();
- throw;
- }
- }
-
template<typename ...Args>
void PushMessage(const char* pszCommand, const Args&... args)
{
try
{
BeginMessage(pszCommand);
- (vSend << ... << args);
+ if constexpr (sizeof...(Args) > 0)
+ (vSend << ... << args);
EndMessage();
}
catch (...)
}
}
- void PushRequest(const char* pszCommand,
- void (*fn)(void*, CDataStream&), void* param1)
- {
- uint256 hashReply;
- RAND_bytes((unsigned char*)&hashReply, sizeof(hashReply));
-
- {
- LOCK(cs_mapRequests);
- mapRequests[hashReply] = CRequestTracker(fn, param1);
- }
-
- PushMessage(pszCommand, hashReply);
- }
-
- template<typename T1>
- void PushRequest(const char* pszCommand, const T1& a1,
- void (*fn)(void*, CDataStream&), void* param1)
- {
- uint256 hashReply;
- RAND_bytes((unsigned char*)&hashReply, sizeof(hashReply));
-
- {
- LOCK(cs_mapRequests);
- mapRequests[hashReply] = CRequestTracker(fn, param1);
- }
-
- PushMessage(pszCommand, hashReply, a1);
- }
-
- template<typename T1, typename T2>
- void PushRequest(const char* pszCommand, const T1& a1, const T2& a2,
- void (*fn)(void*, CDataStream&), void* param1)
- {
- uint256 hashReply;
- RAND_bytes((unsigned char*)&hashReply, sizeof(hashReply));
-
- {
- LOCK(cs_mapRequests);
- mapRequests[hashReply] = CRequestTracker(fn, param1);
- }
-
- PushMessage(pszCommand, hashReply, a1, a2);
- }
-
-
-
void PushGetBlocks(CBlockIndex* pindexBegin, uint256 hashEnd);
bool IsSubscribed(unsigned int nChannel);
void Subscribe(unsigned int nChannel, unsigned int nHops=0);
static uint64_t GetTotalBytesSent();
};
-inline void RelayInventory(const CInv& inv)
-{
- // Put on lists to offer to the other nodes
- {
- LOCK(cs_vNodes);
- for (CNode* pnode : vNodes)
- pnode->PushInventory(inv);
- }
-}
-
class CTransaction;
void RelayTransaction(const CTransaction& tx, const uint256& hash);
void RelayTransaction(const CTransaction& tx, const uint256& hash, const CDataStream& ss);