diff options
Diffstat (limited to 'include/hotstuff/liveness.h')
| -rw-r--r-- | include/hotstuff/liveness.h | 18 |
1 files changed, 12 insertions, 6 deletions
diff --git a/include/hotstuff/liveness.h b/include/hotstuff/liveness.h index 6396fe3..59306ab 100644 --- a/include/hotstuff/liveness.h +++ b/include/hotstuff/liveness.h @@ -46,13 +46,14 @@ class PMAllParents: public virtual PaceMaker { void reg_bqc_update() { hsc->async_bqc_update().then([this](const block_t &bqc) { + const auto &pref = bqc->get_qc_ref(); for (const auto &blk: hsc->get_tails()) { block_t b; for (b = blk; - b->get_height() > bqc->get_height(); + b->get_height() > pref->get_height(); b = b->get_parents()[0]); - if (b == bqc && blk->get_height() > bqc_tail->get_height()) + if (b == pref && blk->get_height() > bqc_tail->get_height()) bqc_tail = blk; } reg_bqc_update(); @@ -61,16 +62,21 @@ class PMAllParents: public virtual PaceMaker { void reg_proposal() { hsc->async_wait_proposal().then([this](const Proposal &prop) { - if (prop.blk->get_parents()[0] == bqc_tail) - bqc_tail = prop.blk; + bqc_tail = prop.blk; reg_proposal(); }); } void reg_receive_proposal() { hsc->async_wait_receive_proposal().then([this](const Proposal &prop) { - if (prop.blk->get_parents()[0] == bqc_tail) - bqc_tail = prop.blk; + const auto &pref = hsc->get_bqc()->get_qc_ref(); + const auto &blk = prop.blk; + block_t b; + for (b = blk; + b->get_height() > pref->get_height(); + b = b->get_parents()[0]); + if (b == pref && blk->get_height() > bqc_tail->get_height()) + bqc_tail = blk; reg_receive_proposal(); }); } |