diff --git a/src/seeder/bitcoin.h b/src/seeder/bitcoin.h --- a/src/seeder/bitcoin.h +++ b/src/seeder/bitcoin.h @@ -19,11 +19,16 @@ // The network magic to use. extern CMessageHeader::MessageMagic netMagic; -enum PeerMessagingState : bool { - AwaitingMessages = false, - Finished = true, +enum class PeerMessagingState : bool { + AwaitingMessages, + Finished, }; +template +constexpr typename std::underlying_type::type to_integral(E e) { + return static_cast::type>(e); +} + class CSeederNode { private: SOCKET sock; diff --git a/src/seeder/test/p2p_messaging_tests.cpp b/src/seeder/test/p2p_messaging_tests.cpp --- a/src/seeder/test/p2p_messaging_tests.cpp +++ b/src/seeder/test/p2p_messaging_tests.cpp @@ -14,6 +14,7 @@ #include #include +#include #include #include @@ -21,6 +22,11 @@ #include +std::ostream &operator<<(std::ostream &os, const PeerMessagingState &state) { + os << to_integral(state); + return os; +} + class TestCSeederNode : public CSeederNode { public: TestCSeederNode(const CService &service, std::vector *vAddrIn)