diff --git a/src/protocol.cpp b/src/protocol.cpp --- a/src/protocol.cpp +++ b/src/protocol.cpp @@ -227,6 +227,8 @@ return cmd.append(NetMsgType::MERKLEBLOCK); case MSG_CMPCT_BLOCK: return cmd.append(NetMsgType::CMPCTBLOCK); + case MSG_AVA_PROOF: + return cmd.append(NetMsgType::AVAPROOF); default: throw std::out_of_range( strprintf("CInv::GetCommand(): type=%d unknown type", type)); diff --git a/src/test/inv_tests.cpp b/src/test/inv_tests.cpp --- a/src/test/inv_tests.cpp +++ b/src/test/inv_tests.cpp @@ -24,6 +24,8 @@ false, true); CheckType(GetDataMsg::MSG_CMPCT_BLOCK, GetDataMsg::MSG_CMPCT_BLOCK, false, true); + CheckType(GetDataMsg::MSG_AVA_PROOF, GetDataMsg::MSG_AVA_PROOF, false, + false); } static void CheckCommand(int type, std::string expected) { @@ -37,6 +39,7 @@ CheckCommand(GetDataMsg::MSG_BLOCK, "block"); CheckCommand(GetDataMsg::MSG_FILTERED_BLOCK, "merkleblock"); CheckCommand(GetDataMsg::MSG_CMPCT_BLOCK, "cmpctblock"); + CheckCommand(GetDataMsg::MSG_AVA_PROOF, "avaproof"); } BOOST_AUTO_TEST_SUITE_END()