|
| 1 | +// Copyright (c) 2026 The Dash Core developers |
| 2 | +// Distributed under the MIT software license, see the accompanying |
| 3 | +// file COPYING or http://www.opensource.org/licenses/mit-license.php. |
| 4 | + |
| 5 | +#include <governance/vote.h> |
| 6 | +#include <primitives/transaction.h> |
| 7 | +#include <serialize.h> |
| 8 | +#include <streams.h> |
| 9 | +#include <uint256.h> |
| 10 | +#include <version.h> |
| 11 | + |
| 12 | +#include <test/util/setup_common.h> |
| 13 | + |
| 14 | +#include <boost/test/unit_test.hpp> |
| 15 | + |
| 16 | +#include <ios> |
| 17 | +#include <limits> |
| 18 | +#include <vector> |
| 19 | + |
| 20 | +BOOST_FIXTURE_TEST_SUITE(governance_vote_wire_tests, BasicTestingSetup) |
| 21 | + |
| 22 | +namespace { |
| 23 | +void WriteVoteHeader(CDataStream& ss) |
| 24 | +{ |
| 25 | + ss << COutPoint{uint256::ONE, 0} << uint256::ONE |
| 26 | + << int{1} /*outcome*/ << int{1} /*signal*/ << int64_t{1'700'000'000}; |
| 27 | +} |
| 28 | + |
| 29 | +CDataStream MakeVoteWire(size_t sig_len) |
| 30 | +{ |
| 31 | + CDataStream ss(SER_NETWORK, PROTOCOL_VERSION); |
| 32 | + WriteVoteHeader(ss); |
| 33 | + ss << std::vector<unsigned char>(sig_len, 0xAA); |
| 34 | + return ss; |
| 35 | +} |
| 36 | +} // namespace |
| 37 | + |
| 38 | +// Reject invalid signature lengths, including a maximal CompactSize prefix. |
| 39 | +BOOST_AUTO_TEST_CASE(rejects_invalid_sizes) |
| 40 | +{ |
| 41 | + for (size_t bad : {size_t{0}, size_t{64}, size_t{66}, size_t{95}, size_t{97}, size_t{128}}) { |
| 42 | + CDataStream ss = MakeVoteWire(bad); |
| 43 | + CGovernanceVote vote; |
| 44 | + BOOST_CHECK_THROW(ss >> vote, std::ios_base::failure); |
| 45 | + } |
| 46 | + |
| 47 | + CDataStream ss(SER_NETWORK, PROTOCOL_VERSION); |
| 48 | + WriteVoteHeader(ss); |
| 49 | + WriteCompactSize(ss, std::numeric_limits<uint64_t>::max()); |
| 50 | + CGovernanceVote vote; |
| 51 | + BOOST_CHECK_THROW(ss >> vote, std::ios_base::failure); |
| 52 | +} |
| 53 | + |
| 54 | +// Truncated element bytes must surface as ios_base::failure so the govobjvote |
| 55 | +// handler scores the peer. |
| 56 | +BOOST_AUTO_TEST_CASE(truncated_signature_throws_ios_failure) |
| 57 | +{ |
| 58 | + CDataStream ss(SER_NETWORK, PROTOCOL_VERSION); |
| 59 | + WriteVoteHeader(ss); |
| 60 | + ss << uint8_t{CGovernanceVote::BLS_SIG_SIZE}; |
| 61 | + ss.write(MakeByteSpan(std::vector<unsigned char>(10, 0xBB))); |
| 62 | + |
| 63 | + CGovernanceVote vote; |
| 64 | + BOOST_CHECK_THROW(ss >> vote, std::ios_base::failure); |
| 65 | +} |
| 66 | + |
| 67 | +// 65-byte ECDSA and 96-byte BLS round-trip cleanly over the network. |
| 68 | +BOOST_AUTO_TEST_CASE(accepts_legitimate_boundary_sizes) |
| 69 | +{ |
| 70 | + for (size_t sig_len : {CGovernanceVote::COMPACT_SIG_SIZE, CGovernanceVote::BLS_SIG_SIZE}) { |
| 71 | + CDataStream ss = MakeVoteWire(sig_len); |
| 72 | + const size_t wire_bytes = ss.size(); |
| 73 | + |
| 74 | + CGovernanceVote vote; |
| 75 | + BOOST_REQUIRE_NO_THROW(ss >> vote); |
| 76 | + BOOST_CHECK_EQUAL(ss.size(), 0U); |
| 77 | + |
| 78 | + CDataStream out(SER_NETWORK, PROTOCOL_VERSION); |
| 79 | + out << vote; |
| 80 | + BOOST_CHECK_EQUAL(out.size(), wire_bytes); |
| 81 | + } |
| 82 | +} |
| 83 | + |
| 84 | +// SER_DISK reads stay unbounded — existing on-disk data must load unchanged. |
| 85 | +BOOST_AUTO_TEST_CASE(ser_disk_deserialization_unaffected) |
| 86 | +{ |
| 87 | + CDataStream ss(SER_DISK, PROTOCOL_VERSION); |
| 88 | + WriteVoteHeader(ss); |
| 89 | + ss << std::vector<unsigned char>(128, 0xCD); |
| 90 | + |
| 91 | + CGovernanceVote vote; |
| 92 | + BOOST_REQUIRE_NO_THROW(ss >> vote); |
| 93 | + BOOST_CHECK_EQUAL(ss.size(), 0U); |
| 94 | +} |
| 95 | + |
| 96 | +BOOST_AUTO_TEST_SUITE_END() |
0 commit comments