kylefang/stacks-puppet-node
1
0
Fork 0
mirror of https://github.com/alexgo-io/stacks-puppet-node.git synced 2026-01-12 22:43:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: Use a bounded reader to read the encrypted signer state from SignerDB

Browse Source
This commit is contained in:
Mårten Blankfors
2024-04-12 10:59:38 +02:00
parent 63c15f1dae
commit 7420df8522
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
libsigner/src/messages.rs
View File

@@ -38,6 +38,7 @@ use blockstack_lib::net::api::postblock_proposal::{
BlockValidateReject, BlockValidateResponse, ValidateRejectCode,
};
use blockstack_lib::util_lib::boot::boot_code_id;
use clarity::util::retry::BoundReader;
use clarity::vm::types::serialization::SerializationError;
use clarity::vm::types::QualifiedContractIdentifier;
use hashbrown::{HashMap, HashSet};
@@ -403,7 +404,12 @@ impl StacksMessageCodec for SignerMessage {
}
}
SignerMessageTypePrefix::EncryptedSignerState => {
let encrypted_state = read_next::<_, _>(fd)?;
// Typically the size of the signer state is much smaller, but in the fully degenerate case the size of dkg shares is
// (2800 + 2800) * 32 (threshold * shares * 32).
// The signer state holds some additional information, so we're adding a factor 4 to have some margin
let max_encrypted_state_size = (2800 + 2800) * 32 * 4;
let mut bound_reader = BoundReader::from_reader(fd, max_encrypted_state_size);
let encrypted_state = read_next::<_, _>(&mut bound_reader)?;
SignerMessage::EncryptedSignerState(encrypted_state)
}
};