diff options
Diffstat (limited to 'src/netmessagemaker.h')
-rw-r--r-- | src/netmessagemaker.h | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/src/netmessagemaker.h b/src/netmessagemaker.h new file mode 100644 index 0000000000..ffb3fe2f29 --- /dev/null +++ b/src/netmessagemaker.h @@ -0,0 +1,36 @@ +// Copyright (c) 2009-2010 Satoshi Nakamoto +// Copyright (c) 2009-2018 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_NETMESSAGEMAKER_H +#define BITCOIN_NETMESSAGEMAKER_H + +#include <net.h> +#include <serialize.h> + +class CNetMsgMaker +{ +public: + explicit CNetMsgMaker(int nVersionIn) : nVersion(nVersionIn){} + + template <typename... Args> + CSerializedNetMsg Make(int nFlags, std::string msg_type, Args&&... args) const + { + CSerializedNetMsg msg; + msg.m_type = std::move(msg_type); + CVectorWriter{ SER_NETWORK, nFlags | nVersion, msg.data, 0, std::forward<Args>(args)... }; + return msg; + } + + template <typename... Args> + CSerializedNetMsg Make(std::string msg_type, Args&&... args) const + { + return Make(0, std::move(msg_type), std::forward<Args>(args)...); + } + +private: + const int nVersion; +}; + +#endif // BITCOIN_NETMESSAGEMAKER_H |