mirror of
https://github.com/bitcoin/bitcoin.git
synced 2025-01-29 20:47:31 -03:00
33 lines
927 B
C
33 lines
927 B
C
// Copyright (c) 2020 The Bitcoin Core developers
|
|
// Distributed under the MIT software license, see the accompanying
|
|
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
|
|
#ifndef BITCOIN_TEST_UTIL_NET_H
|
|
#define BITCOIN_TEST_UTIL_NET_H
|
|
|
|
#include <net.h>
|
|
|
|
struct ConnmanTestMsg : public CConnman {
|
|
using CConnman::CConnman;
|
|
void AddTestNode(CNode& node)
|
|
{
|
|
LOCK(cs_vNodes);
|
|
vNodes.push_back(&node);
|
|
}
|
|
void ClearTestNodes()
|
|
{
|
|
LOCK(cs_vNodes);
|
|
for (CNode* node : vNodes) {
|
|
delete node;
|
|
}
|
|
vNodes.clear();
|
|
}
|
|
|
|
void ProcessMessagesOnce(CNode& node) { m_msgproc->ProcessMessages(&node, flagInterruptMsgProc); }
|
|
|
|
void NodeReceiveMsgBytes(CNode& node, const char* pch, unsigned int nBytes, bool& complete) const;
|
|
|
|
bool ReceiveMsgFrom(CNode& node, CSerializedNetMsg& ser_msg) const;
|
|
};
|
|
|
|
#endif // BITCOIN_TEST_UTIL_NET_H
|