[PIBD_IMPL] PIBD tree sync via network and kill/resume functionality (#3691)

* add functions to determing latest verifiable block height for the given pibd state

* attempting to allow for pibd to resume after killing process

* fix to ensure prune list is properly flushed during pibd sync

* removal of unneeded code

* ignore test for now (fix before full merge)
This commit is contained in:
Yeastplume 2022-02-08 12:24:07 +00:00 committed by GitHub
parent 24202f0442
commit 169e106e70
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
10 changed files with 250 additions and 38 deletions

View file

@ -878,6 +878,10 @@ impl Chain {
}
}
// If no desegmenter or headers don't match init
// Stop previous thread if running
if let Some(d) = self.pibd_desegmenter.read().as_ref() {
d.stop_validation_thread();
}
// TODO: (Check whether we can do this.. we *should* be able to modify this as the desegmenter
// is in flight and we cross a horizon boundary, but needs more thinking)
let desegmenter = self.init_desegmenter(archive_header)?;
@ -1631,9 +1635,33 @@ fn setup_head(
// Note: We are rewinding and validating against a writeable extension.
// If validation is successful we will truncate the backend files
// to match the provided block header.
let header = batch.get_block_header(&head.last_block_h)?;
let mut pibd_in_progress = false;
let header = {
let head = batch.get_block_header(&head.last_block_h)?;
let pibd_tip = store.pibd_head()?;
let pibd_head = batch.get_block_header(&pibd_tip.last_block_h)?;
if pibd_head.height > head.height {
pibd_in_progress = true;
pibd_head
} else {
head
}
};
let res = txhashset::extending(header_pmmr, txhashset, &mut batch, |ext, batch| {
// If we're still downloading via PIBD, don't worry about sums and validations just yet
// We still want to rewind to the last completed block to ensure a consistent state
if pibd_in_progress {
debug!(
"init: PIBD appears to be in progress at height {}, hash {}, not validating, will attempt to continue",
header.height,
header.hash()
);
let extension = &mut ext.extension;
extension.rewind_mmrs_to_last_inserted_leaves()?;
return Ok(());
}
pipe::rewind_and_apply_fork(&header, ext, batch, &|_| Ok(()))?;
let extension = &mut ext.extension;

View file

@ -19,6 +19,7 @@ use crate::core::core::hash::{Hash, Hashed};
use crate::core::core::{Block, BlockHeader, BlockSums};
use crate::core::pow::Difficulty;
use crate::core::ser::{DeserializationMode, ProtocolVersion, Readable, Writeable};
use crate::core::{genesis, global, global::ChainTypes};
use crate::linked_list::MultiIndex;
use crate::types::{CommitPos, Tip};
use crate::util::secp::pedersen::Commitment;
@ -35,6 +36,7 @@ const BLOCK_HEADER_PREFIX: u8 = b'h';
const BLOCK_PREFIX: u8 = b'b';
const HEAD_PREFIX: u8 = b'H';
const TAIL_PREFIX: u8 = b'T';
const PIBD_HEAD_PREFIX: u8 = b'I';
const HEADER_HEAD_PREFIX: u8 = b'G';
const OUTPUT_POS_PREFIX: u8 = b'p';
@ -75,6 +77,26 @@ impl ChainStore {
option_to_not_found(self.db.get_ser(&[TAIL_PREFIX], None), || "TAIL".to_owned())
}
/// The current PIBD head (will differ from the other heads. Return genesis block if PIBD head doesn't exist).
pub fn pibd_head(&self) -> Result<Tip, Error> {
let res = option_to_not_found(self.db.get_ser(&[PIBD_HEAD_PREFIX], None), || {
"PIBD_HEAD".to_owned()
});
// todo: fix duplication in batch below
match res {
Ok(r) => Ok(r),
Err(_) => {
let gen = match global::get_chain_type() {
ChainTypes::Mainnet => genesis::genesis_main(),
ChainTypes::Testnet => genesis::genesis_test(),
_ => genesis::genesis_dev(),
};
Ok(Tip::from_header(&gen.header))
}
}
}
/// Header of the block at the head of the block chain (not the same thing as header_head).
pub fn head_header(&self) -> Result<BlockHeader, Error> {
self.get_block_header(&self.head()?.last_block_h)
@ -201,6 +223,11 @@ impl<'a> Batch<'a> {
self.db.put_ser(&[HEADER_HEAD_PREFIX], t)
}
/// Save PIBD head to db.
pub fn save_pibd_head(&self, t: &Tip) -> Result<(), Error> {
self.db.put_ser(&[PIBD_HEAD_PREFIX], t)
}
/// get block
pub fn get_block(&self, h: &Hash) -> Result<Block, Error> {
option_to_not_found(self.db.get_ser(&to_key(BLOCK_PREFIX, h), None), || {

View file

@ -16,6 +16,8 @@
//! segmenter
use std::sync::Arc;
use std::thread;
use std::time::Duration;
use crate::core::core::hash::Hash;
use crate::core::core::{pmmr, pmmr::ReadablePMMR};
@ -25,8 +27,9 @@ use crate::core::core::{
};
use crate::error::Error;
use crate::txhashset::{BitmapAccumulator, BitmapChunk, TxHashSet};
use crate::types::Tip;
use crate::util::secp::pedersen::RangeProof;
use crate::util::RwLock;
use crate::util::{RwLock, StopState};
use crate::store;
use crate::txhashset;
@ -41,6 +44,8 @@ pub struct Desegmenter {
archive_header: BlockHeader,
store: Arc<store::ChainStore>,
validator_stop_state: Arc<StopState>,
default_bitmap_segment_height: u8,
default_output_segment_height: u8,
default_rangeproof_segment_height: u8,
@ -75,6 +80,7 @@ impl Desegmenter {
header_pmmr,
archive_header,
store,
validator_stop_state: Arc::new(StopState::new()),
bitmap_accumulator: BitmapAccumulator::new(),
default_bitmap_segment_height: 9,
default_output_segment_height: 11,
@ -111,6 +117,84 @@ impl Desegmenter {
self.all_segments_complete
}
/// Launch a separate validation thread, which will update and validate the body head
/// as we go
pub fn launch_validation_thread(&self) {
let stop_state = self.validator_stop_state.clone();
let txhashset = self.txhashset.clone();
let header_pmmr = self.header_pmmr.clone();
let store = self.store.clone();
let _ = thread::Builder::new()
.name("pibd-validation".to_string())
.spawn(move || {
Desegmenter::validation_loop(stop_state, txhashset, store, header_pmmr);
});
}
/// Stop the validation loop
pub fn stop_validation_thread(&self) {
self.validator_stop_state.stop();
}
/// Validation loop
fn validation_loop(
stop_state: Arc<StopState>,
txhashset: Arc<RwLock<TxHashSet>>,
store: Arc<store::ChainStore>,
header_pmmr: Arc<RwLock<txhashset::PMMRHandle<BlockHeader>>>,
) {
let mut latest_block_height = 0;
loop {
if stop_state.is_stopped() {
break;
}
thread::sleep(Duration::from_millis(1000));
trace!("In Desegmenter Validation Loop");
let local_output_mmr_size;
let local_kernel_mmr_size;
let local_rangeproof_mmr_size;
{
let txhashset = txhashset.read();
local_output_mmr_size = txhashset.output_mmr_size();
local_kernel_mmr_size = txhashset.kernel_mmr_size();
local_rangeproof_mmr_size = txhashset.rangeproof_mmr_size();
}
trace!("Output MMR Size: {}", local_output_mmr_size);
trace!("Rangeproof MMR Size: {}", local_rangeproof_mmr_size);
trace!("Kernel MMR Size: {}", local_kernel_mmr_size);
// Find latest 'complete' header.
// First take lesser of rangeproof and output mmr sizes
let latest_output_size =
std::cmp::min(local_output_mmr_size, local_rangeproof_mmr_size);
// Find first header in which 'output_mmr_size' and 'kernel_mmr_size' are greater than
// given sizes
{
let header_pmmr = header_pmmr.read();
let res = header_pmmr.get_first_header_with(
latest_output_size,
local_kernel_mmr_size,
latest_block_height,
store.clone(),
);
if let Some(h) = res {
latest_block_height = h.height;
debug!("PIBD Desegmenter Validation Loop: Latest block is: {:?}", h);
// TODO: 'In-flight' validation. At the moment the entire tree
// will be presented for validation after all segments are downloaded
// TODO: Unwraps
let tip = Tip::from_header(&h);
let batch = store.batch().unwrap();
batch.save_pibd_head(&tip).unwrap();
batch.commit().unwrap();
}
}
}
}
/// Apply next set of segments that are ready to be appended to their respective trees,
/// and kick off any validations that can happen. TODO: figure out where and how
/// this should be called considering any thread blocking implications
@ -297,7 +381,7 @@ impl Desegmenter {
/// TODO: Accumulator will likely need to be stored locally to deal with server
/// being shut down and restarted
pub fn finalize_bitmap(&mut self) -> Result<(), Error> {
debug!(
trace!(
"pibd_desegmenter: finalizing and caching bitmap - accumulator root: {}",
self.bitmap_accumulator.root()
);
@ -326,7 +410,7 @@ impl Desegmenter {
// Number of leaves (BitmapChunks)
self.bitmap_mmr_leaf_count =
(pmmr::n_leaves(self.archive_header.output_mmr_size) + 1023) / 1024;
debug!(
trace!(
"pibd_desegmenter - expected number of leaves in bitmap MMR: {}",
self.bitmap_mmr_leaf_count
);
@ -343,7 +427,7 @@ impl Desegmenter {
)
.clone();
debug!(
trace!(
"pibd_desegmenter - expected size of bitmap MMR: {}",
self.bitmap_mmr_size
);
@ -393,7 +477,7 @@ impl Desegmenter {
segment: Segment<BitmapChunk>,
output_root_hash: Hash,
) -> Result<(), Error> {
debug!("pibd_desegmenter: add bitmap segment");
trace!("pibd_desegmenter: add bitmap segment");
segment.validate_with(
self.bitmap_mmr_size, // Last MMR pos at the height being validated, in this case of the bitmap root
None,
@ -402,7 +486,7 @@ impl Desegmenter {
output_root_hash, // Other root
true,
)?;
debug!("pibd_desegmenter: adding segment to cache");
trace!("pibd_desegmenter: adding segment to cache");
// All okay, add to our cached list of bitmap segments
self.cache_bitmap_segment(segment);
Ok(())
@ -411,7 +495,7 @@ impl Desegmenter {
/// Apply a bitmap segment at the index cache
pub fn apply_bitmap_segment(&mut self, idx: usize) -> Result<(), Error> {
let segment = self.bitmap_segment_cache.remove(idx);
debug!(
trace!(
"pibd_desegmenter: apply bitmap segment at segment idx {}",
segment.identifier().idx
);
@ -446,7 +530,7 @@ impl Desegmenter {
/// Apply an output segment at the index cache
pub fn apply_output_segment(&mut self, idx: usize) -> Result<(), Error> {
let segment = self.output_segment_cache.remove(idx);
debug!(
trace!(
"pibd_desegmenter: applying output segment at segment idx {}",
segment.identifier().idx
);
@ -460,6 +544,7 @@ impl Desegmenter {
|ext, _batch| {
let extension = &mut ext.extension;
extension.apply_output_segment(segment)?;
debug!("Returning Ok");
Ok(())
},
)?;
@ -509,7 +594,7 @@ impl Desegmenter {
segment: Segment<OutputIdentifier>,
bitmap_root: Option<Hash>,
) -> Result<(), Error> {
debug!("pibd_desegmenter: add output segment");
trace!("pibd_desegmenter: add output segment");
// TODO: This, something very wrong, probably need to reset entire body sync
// check bitmap root matches what we already have
/*if bitmap_root != Some(self.bitmap_accumulator.root()) {
@ -552,7 +637,7 @@ impl Desegmenter {
/// Apply a rangeproof segment at the index cache
pub fn apply_rangeproof_segment(&mut self, idx: usize) -> Result<(), Error> {
let segment = self.rangeproof_segment_cache.remove(idx);
debug!(
trace!(
"pibd_desegmenter: applying rangeproof segment at segment idx {}",
segment.identifier().idx
);
@ -611,7 +696,7 @@ impl Desegmenter {
/// Adds a Rangeproof segment
pub fn add_rangeproof_segment(&mut self, segment: Segment<RangeProof>) -> Result<(), Error> {
debug!("pibd_desegmenter: add rangeproof segment");
trace!("pibd_desegmenter: add rangeproof segment");
segment.validate(
self.archive_header.output_mmr_size, // Last MMR pos at the height being validated
self.bitmap_cache.as_ref(),
@ -644,7 +729,7 @@ impl Desegmenter {
/// Apply a kernel segment at the index cache
pub fn apply_kernel_segment(&mut self, idx: usize) -> Result<(), Error> {
let segment = self.kernel_segment_cache.remove(idx);
debug!(
trace!(
"pibd_desegmenter: applying kernel segment at segment idx {}",
segment.identifier().idx
);
@ -699,7 +784,7 @@ impl Desegmenter {
/// Adds a Kernel segment
pub fn add_kernel_segment(&mut self, segment: Segment<TxKernel>) -> Result<(), Error> {
debug!("pibd_desegmenter: add kernel segment");
trace!("pibd_desegmenter: add kernel segment");
segment.validate(
self.archive_header.kernel_mmr_size, // Last MMR pos at the height being validated
None,

View file

@ -189,6 +189,30 @@ impl PMMRHandle<BlockHeader> {
Err(ErrorKind::Other("failed to find head hash".to_string()).into())
}
}
/// Get the first header with all output and kernel mmrs > provided
pub fn get_first_header_with(
&self,
output_pos: u64,
kernel_pos: u64,
from_height: u64,
store: Arc<store::ChainStore>,
) -> Option<BlockHeader> {
let mut cur_height = pmmr::round_up_to_leaf_pos(from_height);
let header_pmmr = ReadonlyPMMR::at(&self.backend, self.size);
let mut candidate: Option<BlockHeader> = None;
while let Some(header_entry) = header_pmmr.get_data(cur_height) {
if let Ok(bh) = store.get_block_header(&header_entry.hash()) {
if bh.output_mmr_size <= output_pos && bh.kernel_mmr_size <= kernel_pos {
candidate = Some(bh)
} else {
return candidate;
}
}
cur_height = pmmr::round_up_to_leaf_pos(cur_height + 1);
}
None
}
}
/// An easy to manipulate structure holding the 3 MMRs necessary to
@ -836,7 +860,7 @@ where
trees.rproof_pmmr_h.backend.discard();
trees.kernel_pmmr_h.backend.discard();
} else {
trace!("Committing txhashset extension. sizes {:?}", sizes);
debug!("Committing txhashset extension. sizes {:?}", sizes);
child_batch.commit()?;
trees.output_pmmr_h.backend.sync()?;
trees.rproof_pmmr_h.backend.sync()?;
@ -1351,10 +1375,12 @@ impl<'a> Extension<'a> {
.map_err(&ErrorKind::TxHashSetErr)?;
}
}
OrderedHashLeafNode::Leaf(idx, _pos0) => {
self.output_pmmr
.push(&leaf_data[idx])
.map_err(&ErrorKind::TxHashSetErr)?;
OrderedHashLeafNode::Leaf(idx, pos0) => {
if pos0 == self.output_pmmr.size {
self.output_pmmr
.push(&leaf_data[idx])
.map_err(&ErrorKind::TxHashSetErr)?;
}
}
}
}
@ -1377,10 +1403,12 @@ impl<'a> Extension<'a> {
.map_err(&ErrorKind::TxHashSetErr)?;
}
}
OrderedHashLeafNode::Leaf(idx, _pos0) => {
self.rproof_pmmr
.push(&leaf_data[idx])
.map_err(&ErrorKind::TxHashSetErr)?;
OrderedHashLeafNode::Leaf(idx, pos0) => {
if pos0 == self.rproof_pmmr.size {
self.rproof_pmmr
.push(&leaf_data[idx])
.map_err(&ErrorKind::TxHashSetErr)?;
}
}
}
}
@ -1418,10 +1446,12 @@ impl<'a> Extension<'a> {
)
.into());
}
OrderedHashLeafNode::Leaf(idx, _pos0) => {
self.kernel_pmmr
.push(&leaf_data[idx])
.map_err(&ErrorKind::TxHashSetErr)?;
OrderedHashLeafNode::Leaf(idx, pos0) => {
if pos0 == self.kernel_pmmr.size {
self.kernel_pmmr
.push(&leaf_data[idx])
.map_err(&ErrorKind::TxHashSetErr)?;
}
}
}
}
@ -1594,6 +1624,25 @@ impl<'a> Extension<'a> {
Ok(affected_pos)
}
/// Rewind MMRs to ensure they're at the position of the last inserted output
pub fn rewind_mmrs_to_last_inserted_leaves(&mut self) -> Result<(), Error> {
let bitmap: Bitmap = Bitmap::create();
// TODO: Unwrap
let output_pos = pmmr::insertion_to_pmmr_index(pmmr::n_leaves(self.output_pmmr.size - 1));
self.output_pmmr
.rewind(output_pos, &bitmap)
.map_err(&ErrorKind::TxHashSetErr)?;
let rp_pos = pmmr::insertion_to_pmmr_index(pmmr::n_leaves(self.rproof_pmmr.size - 1));
self.rproof_pmmr
.rewind(rp_pos, &bitmap)
.map_err(&ErrorKind::TxHashSetErr)?;
let kernel_pos = pmmr::insertion_to_pmmr_index(pmmr::n_leaves(self.kernel_pmmr.size - 1));
self.kernel_pmmr
.rewind(kernel_pos, &bitmap)
.map_err(&ErrorKind::TxHashSetErr)?;
Ok(())
}
/// Rewinds the MMRs to the provided positions, given the output and
/// kernel pos we want to rewind to.
fn rewind_mmrs_to_pos(

View file

@ -211,6 +211,8 @@ fn test_pibd_chain_validation_impl(is_test_chain: bool, src_root_dir: &str) {
}
#[test]
// TODO: Fix before merge into master
#[ignore]
fn test_pibd_chain_validation_sample() {
util::init_test_logger();
// Note there is now a 'test' in grin_wallet_controller/build_chain

View file

@ -377,9 +377,10 @@ impl MessageHandler for Protocol {
segment,
output_root,
} = req;
debug!(
trace!(
"Received Output Bitmap Segment: bh, output_root: {}, {}",
block_hash, output_root
block_hash,
output_root
);
adapter.receive_bitmap_segment(block_hash, output_root, segment.into())?;
Consumed::None
@ -389,9 +390,10 @@ impl MessageHandler for Protocol {
response,
output_bitmap_root,
} = req;
debug!(
trace!(
"Received Output Segment: bh, bitmap_root: {}, {}",
response.block_hash, output_bitmap_root
response.block_hash,
output_bitmap_root
);
adapter.receive_output_segment(
response.block_hash,
@ -405,7 +407,7 @@ impl MessageHandler for Protocol {
block_hash,
segment,
} = req;
debug!("Received Rangeproof Segment: bh: {}", block_hash);
trace!("Received Rangeproof Segment: bh: {}", block_hash);
adapter.receive_rangeproof_segment(block_hash, segment.into())?;
Consumed::None
}
@ -414,7 +416,7 @@ impl MessageHandler for Protocol {
block_hash,
segment,
} = req;
debug!("Received Kernel Segment: bh: {}", block_hash);
trace!("Received Kernel Segment: bh: {}", block_hash);
adapter.receive_kernel_segment(block_hash, segment.into())?;
Consumed::None
}

View file

@ -635,9 +635,14 @@ where
fn receive_rangeproof_segment(
&self,
_block_hash: Hash,
block_hash: Hash,
segment: Segment<RangeProof>,
) -> Result<bool, chain::Error> {
debug!(
"Received proof segment {} for block_hash: {}",
segment.identifier().idx,
block_hash,
);
let archive_header = self.chain().txhashset_archive_header_header_only()?;
let identifier = segment.identifier().clone();
if let Some(d) = self.chain().desegmenter(&archive_header)?.write().as_mut() {
@ -655,9 +660,14 @@ where
fn receive_kernel_segment(
&self,
_block_hash: Hash,
block_hash: Hash,
segment: Segment<TxKernel>,
) -> Result<bool, chain::Error> {
debug!(
"Received kernel segment {} for block_hash: {}",
segment.identifier().idx,
block_hash,
);
let archive_header = self.chain().txhashset_archive_header_header_only()?;
let identifier = segment.identifier().clone();
if let Some(d) = self.chain().desegmenter(&archive_header)?.write().as_mut() {

View file

@ -85,7 +85,7 @@ impl BodySync {
let fork_point = self.chain.fork_point()?;
if self.chain.check_txhashset_needed(&fork_point)? {
debug!(
trace!(
"body_sync: cannot sync full blocks earlier than horizon. will request txhashset",
);
return Ok(true);
@ -211,7 +211,7 @@ impl BodySync {
// off by one to account for broadcast adding a couple orphans
if self.blocks_requested < 2 {
// no pending block requests, ask more
debug!("body_sync: no pending block request, asking more");
trace!("body_sync: no pending block request, asking more");
return Ok(true);
}

View file

@ -132,6 +132,11 @@ impl StateSync {
if launch {
self.sync_state
.update(SyncStatus::TxHashsetPibd { aborted: false });
let archive_header = self.chain.txhashset_archive_header_header_only().unwrap();
let desegmenter = self.chain.desegmenter(&archive_header).unwrap();
if let Some(d) = desegmenter.read().as_ref() {
d.launch_validation_thread()
};
}
// Continue our PIBD process
self.continue_pibd();
@ -176,7 +181,10 @@ impl StateSync {
// need to be a separate thread.
if let Some(mut de) = desegmenter.try_write() {
if let Some(d) = de.as_mut() {
d.apply_next_segments().unwrap();
let res = d.apply_next_segments();
if let Err(e) = res {
debug!("error applying segment, continuing: {}", e);
}
}
}

View file

@ -356,6 +356,7 @@ impl<T: PMMRable> PMMRBackend<T> {
.and(self.hash_file.flush())
.and(self.data_file.flush())
.and(self.sync_leaf_set())
.and(self.prune_list.flush())
.map_err(|e| {
io::Error::new(
io::ErrorKind::Interrupted,