d0s3t3ch/wownero
1
0
Fork 0
mirror of https://git.wownero.com/wownero/wownero.git synced 2024-08-15 01:03:23 +00:00
Code Issues Projects Releases Wiki Activity

hardfork: fix get_next_version()

Browse source
This commit is contained in:
stoffu 2018-03-20 11:09:58 +09:00
parent 7e30eadb32
commit 98cf62cc45
No known key found for this signature in database
GPG key ID: 41DAB8343A9EC012
2 changed files with 32 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/cryptonote_basic/hardfork.cpp
View file

@ -394,9 +394,9 @@ uint8_t HardFork::get_next_version() const
{ {
CRITICAL_REGION_LOCAL(lock); CRITICAL_REGION_LOCAL(lock);
uint64_t height = db.height(); uint64_t height = db.height();
for (unsigned int n = heights.size() - 1; n > 0; --n) { for (auto i = heights.rbegin(); i != heights.rend(); ++i) {
if (height >= heights[n].height) { if (height >= i->height) {
return heights[n < heights.size() - 1 ? n + 1 : n].version; return (i == heights.rbegin() ? i : (i - 1))->version;
} }
} }
return original_version; return original_version;

29
tests/unit_tests/hardfork.cpp
View file

@ -256,6 +256,35 @@ TEST(check_for_height, Success)
} }
} }
TEST(get, next_version)
{
TestDB db;
HardFork hf(db);
ASSERT_TRUE(hf.add_fork(1, 0, 0));
ASSERT_TRUE(hf.add_fork(2, 5, 1));
ASSERT_TRUE(hf.add_fork(4, 10, 2));
hf.init();
for (uint64_t h = 0; h <= 4; ++h) {
ASSERT_EQ(2, hf.get_next_version());
db.add_block(mkblock(hf, h, 1), 0, 0, 0, crypto::hash());
ASSERT_TRUE(hf.add(db.get_block_from_height(h), h));
}
for (uint64_t h = 5; h <= 9; ++h) {
ASSERT_EQ(4, hf.get_next_version());
db.add_block(mkblock(hf, h, 2), 0, 0, 0, crypto::hash());
ASSERT_TRUE(hf.add(db.get_block_from_height(h), h));
}
for (uint64_t h = 10; h <= 15; ++h) {
ASSERT_EQ(4, hf.get_next_version());
db.add_block(mkblock(hf, h, 4), 0, 0, 0, crypto::hash());
ASSERT_TRUE(hf.add(db.get_block_from_height(h), h));
}
}
TEST(states, Success) TEST(states, Success)
{ {
TestDB db; TestDB db;