cerc-io/lighthouse
16
0
Fork 0
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

Rename AttestationRecord -> Attestation

Browse Source
This commit is contained in:
Paul Hauner 2018-12-12 12:16:11 +11:00
parent 7259e1d7b6
commit 4d0641636e
No known key found for this signature in database
GPG Key ID: 303E4494BB28068C
7 changed files with 57 additions and 51 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
beacon_chain/types/src/active_state.rs
View File

@ -1,9 +1,9 @@
use super::Hash256; use super::Hash256;
use super::{AttestationRecord, SpecialRecord}; use super::{Attestation, SpecialRecord};
#[derive(Debug, PartialEq)] #[derive(Debug, PartialEq)]
pub struct ActiveState { pub struct ActiveState {
pub pending_attestations: Vec<AttestationRecord>, pub pending_attestations: Vec<Attestation>,
pub pending_specials: Vec<SpecialRecord>, pub pending_specials: Vec<SpecialRecord>,
pub recent_block_hashes: Vec<Hash256>, pub recent_block_hashes: Vec<Hash256>,
pub randao_mix: Hash256, pub randao_mix: Hash256,

14
beacon_chain/types/src/attestation_record.rs → beacon_chain/types/src/attestation.rs
View File

@ -11,14 +11,14 @@ pub const MIN_SSZ_ATTESTION_RECORD_LENGTH: usize = {
}; };
#[derive(Debug, Clone, PartialEq)] #[derive(Debug, Clone, PartialEq)]
pub struct AttestationRecord { pub struct Attestation {
pub data: AttestationData, pub data: AttestationData,
pub participation_bitfield: Bitfield, pub participation_bitfield: Bitfield,
pub custody_bitfield: Bitfield, pub custody_bitfield: Bitfield,
pub aggregate_sig: AggregateSignature, pub aggregate_sig: AggregateSignature,
} }
impl Encodable for AttestationRecord { impl Encodable for Attestation {
fn ssz_append(&self, s: &mut SszStream) { fn ssz_append(&self, s: &mut SszStream) {
s.append(&self.data); s.append(&self.data);
s.append(&self.participation_bitfield); s.append(&self.participation_bitfield);
@ -27,7 +27,7 @@ impl Encodable for AttestationRecord {
} }
} }
impl Decodable for AttestationRecord { impl Decodable for Attestation {
fn ssz_decode(bytes: &[u8], i: usize) -> Result<(Self, usize), DecodeError> { fn ssz_decode(bytes: &[u8], i: usize) -> Result<(Self, usize), DecodeError> {
let (data, i) = AttestationData::ssz_decode(bytes, i)?; let (data, i) = AttestationData::ssz_decode(bytes, i)?;
let (participation_bitfield, i) = Bitfield::ssz_decode(bytes, i)?; let (participation_bitfield, i) = Bitfield::ssz_decode(bytes, i)?;
@ -46,7 +46,7 @@ impl Decodable for AttestationRecord {
} }
} }
impl AttestationRecord { impl Attestation {
pub fn zero() -> Self { pub fn zero() -> Self {
Self { Self {
data: AttestationData::zero(), data: AttestationData::zero(),
@ -64,7 +64,7 @@ mod tests {
#[test] #[test]
pub fn test_attestation_record_min_ssz_length() { pub fn test_attestation_record_min_ssz_length() {
let ar = AttestationRecord::zero(); let ar = Attestation::zero();
let ssz = ssz_encode(&ar); let ssz = ssz_encode(&ar);
assert_eq!(ssz.len(), MIN_SSZ_ATTESTION_RECORD_LENGTH); assert_eq!(ssz.len(), MIN_SSZ_ATTESTION_RECORD_LENGTH);
@ -72,7 +72,7 @@ mod tests {
#[test] #[test]
pub fn test_attestation_record_ssz_round_trip() { pub fn test_attestation_record_ssz_round_trip() {
let original = AttestationRecord { let original = Attestation {
data: AttestationData::zero(), data: AttestationData::zero(),
participation_bitfield: Bitfield::from_bytes(&vec![17; 42][..]), participation_bitfield: Bitfield::from_bytes(&vec![17; 42][..]),
custody_bitfield: Bitfield::from_bytes(&vec![18; 12][..]), custody_bitfield: Bitfield::from_bytes(&vec![18; 12][..]),
@ -80,7 +80,7 @@ mod tests {
}; };
let ssz = ssz_encode(&original); let ssz = ssz_encode(&original);
let (decoded, _) = AttestationRecord::ssz_decode(&ssz, 0).unwrap(); let (decoded, _) = Attestation::ssz_decode(&ssz, 0).unwrap();
assert_eq!(original, decoded); assert_eq!(original, decoded);
} }

4
beacon_chain/types/src/beacon_block.rs
View File

@ -1,4 +1,4 @@
use super::attestation_record::AttestationRecord; use super::attestation::Attestation;
use super::special_record::SpecialRecord; use super::special_record::SpecialRecord;
use super::ssz::{Decodable, DecodeError, Encodable, SszStream}; use super::ssz::{Decodable, DecodeError, Encodable, SszStream};
use super::Hash256; use super::Hash256;
@ -23,7 +23,7 @@ pub struct BeaconBlock {
pub ancestor_hashes: Vec<Hash256>, pub ancestor_hashes: Vec<Hash256>,
pub active_state_root: Hash256, pub active_state_root: Hash256,
pub crystallized_state_root: Hash256, pub crystallized_state_root: Hash256,
pub attestations: Vec<AttestationRecord>, pub attestations: Vec<Attestation>,
pub specials: Vec<SpecialRecord>, pub specials: Vec<SpecialRecord>,
} }

4
beacon_chain/types/src/lib.rs
View File

@ -5,7 +5,7 @@ extern crate ssz;
pub mod active_state; pub mod active_state;
pub mod attestation_data; pub mod attestation_data;
pub mod attestation_record; pub mod attestation;
pub mod beacon_block; pub mod beacon_block;
pub mod beacon_state; pub mod beacon_state;
pub mod candidate_pow_receipt_root_record; pub mod candidate_pow_receipt_root_record;
@ -25,7 +25,7 @@ use std::collections::HashMap;
pub use active_state::ActiveState; pub use active_state::ActiveState;
pub use attestation_data::AttestationData; pub use attestation_data::AttestationData;
pub use attestation_record::AttestationRecord; pub use attestation::Attestation;
pub use beacon_block::BeaconBlock; pub use beacon_block::BeaconBlock;
pub use beacon_state::BeaconState; pub use beacon_state::BeaconState;
pub use chain_config::ChainConfig; pub use chain_config::ChainConfig;

30
beacon_chain/utils/ssz_helpers/src/attestation_ssz_splitter.rs
View File

@ -1,15 +1,15 @@
use super::bls::BLS_AGG_SIG_BYTE_SIZE; use super::bls::BLS_AGG_SIG_BYTE_SIZE;
use super::ssz::decode::decode_length; use super::ssz::decode::decode_length;
use super::ssz::LENGTH_BYTES; use super::ssz::LENGTH_BYTES;
use super::types::attestation::MIN_SSZ_ATTESTION_RECORD_LENGTH;
use super::types::attestation_data::SSZ_ATTESTION_DATA_LENGTH; use super::types::attestation_data::SSZ_ATTESTION_DATA_LENGTH;
use super::types::attestation_record::MIN_SSZ_ATTESTION_RECORD_LENGTH;
#[derive(Debug, PartialEq)] #[derive(Debug, PartialEq)]
pub enum AttestationSplitError { pub enum AttestationSplitError {
TooShort, TooShort,
} }
/// Given some ssz slice, find the bounds of each serialized AttestationRecord and return a vec of /// Given some ssz slice, find the bounds of each serialized Attestation and return a vec of
/// slices point to each. /// slices point to each.
pub fn split_all_attestations<'a>( pub fn split_all_attestations<'a>(
full_ssz: &'a [u8], full_ssz: &'a [u8],
@ -25,7 +25,7 @@ pub fn split_all_attestations<'a>(
Ok(v) Ok(v)
} }
/// Given some ssz slice, find the bounds of one serialized AttestationRecord /// Given some ssz slice, find the bounds of one serialized Attestation
/// and return a slice pointing to that. /// and return a slice pointing to that.
pub fn split_one_attestation( pub fn split_one_attestation(
full_ssz: &[u8], full_ssz: &[u8],
@ -43,11 +43,11 @@ pub fn split_one_attestation(
} }
} }
/// Given some SSZ, assume that a serialized `AttestationRecord` begins at the `index` position and /// Given some SSZ, assume that a serialized `Attestation` begins at the `index` position and
/// attempt to find the length (in bytes) of that serialized `AttestationRecord`. /// attempt to find the length (in bytes) of that serialized `Attestation`.
/// ///
/// This function does not perform validation on the `AttestationRecord`. It is very likely that /// This function does not perform validation on the `Attestation`. It is very likely that
/// given some sufficiently long non-`AttestationRecord` bytes it will not raise an error. /// given some sufficiently long non-`Attestation` bytes it will not raise an error.
fn determine_ssz_attestation_len( fn determine_ssz_attestation_len(
full_ssz: &[u8], full_ssz: &[u8],
index: usize, index: usize,
@ -68,7 +68,7 @@ fn determine_ssz_attestation_len(
.map_err(|_| AttestationSplitError::TooShort)?; .map_err(|_| AttestationSplitError::TooShort)?;
let custody_bitfield_end = participation_bitfield_end + LENGTH_BYTES + custody_bitfield_len; let custody_bitfield_end = participation_bitfield_end + LENGTH_BYTES + custody_bitfield_len;
// Determine the very end of the AttestationRecord. // Determine the very end of the Attestation.
let agg_sig_end = custody_bitfield_end + LENGTH_BYTES + BLS_AGG_SIG_BYTE_SIZE; let agg_sig_end = custody_bitfield_end + LENGTH_BYTES + BLS_AGG_SIG_BYTE_SIZE;
if agg_sig_end > full_ssz.len() { if agg_sig_end > full_ssz.len() {
@ -82,11 +82,11 @@ fn determine_ssz_attestation_len(
mod tests { mod tests {
use super::super::bls::AggregateSignature; use super::super::bls::AggregateSignature;
use super::super::ssz::{Decodable, SszStream}; use super::super::ssz::{Decodable, SszStream};
use super::super::types::{AttestationData, AttestationRecord, Bitfield, Hash256}; use super::super::types::{Attestation, AttestationData, Bitfield, Hash256};
use super::*; use super::*;
fn get_two_records() -> Vec<AttestationRecord> { fn get_two_records() -> Vec<Attestation> {
let a = AttestationRecord { let a = Attestation {
data: AttestationData { data: AttestationData {
slot: 7, slot: 7,
shard: 9, shard: 9,
@ -101,7 +101,7 @@ mod tests {
custody_bitfield: Bitfield::from_bytes(&vec![255; 12][..]), custody_bitfield: Bitfield::from_bytes(&vec![255; 12][..]),
aggregate_sig: AggregateSignature::new(), aggregate_sig: AggregateSignature::new(),
}; };
let b = AttestationRecord { let b = Attestation {
data: AttestationData { data: AttestationData {
slot: 9, slot: 9,
shard: 7, shard: 7,
@ -133,7 +133,7 @@ mod tests {
let ssz = ssz_stream.drain(); let ssz = ssz_stream.drain();
let (a_ssz, i) = split_one_attestation(&ssz, 0).unwrap(); let (a_ssz, i) = split_one_attestation(&ssz, 0).unwrap();
assert_eq!(i, ssz.len()); assert_eq!(i, ssz.len());
let (decoded_a, _) = AttestationRecord::ssz_decode(a_ssz, 0).unwrap(); let (decoded_a, _) = Attestation::ssz_decode(a_ssz, 0).unwrap();
assert_eq!(a, decoded_a); assert_eq!(a, decoded_a);
/* /*
@ -144,8 +144,8 @@ mod tests {
ssz_stream.append(&b); ssz_stream.append(&b);
let ssz = ssz_stream.drain(); let ssz = ssz_stream.drain();
let ssz_vec = split_all_attestations(&ssz, 0).unwrap(); let ssz_vec = split_all_attestations(&ssz, 0).unwrap();
let (decoded_a, _) = AttestationRecord::ssz_decode(ssz_vec[0], 0).unwrap(); let (decoded_a, _) = Attestation::ssz_decode(ssz_vec[0], 0).unwrap();
let (decoded_b, _) = AttestationRecord::ssz_decode(ssz_vec[1], 0).unwrap(); let (decoded_b, _) = Attestation::ssz_decode(ssz_vec[1], 0).unwrap();
assert_eq!(a, decoded_a); assert_eq!(a, decoded_a);
assert_eq!(b, decoded_b); assert_eq!(b, decoded_b);

38
beacon_chain/utils/ssz_helpers/src/ssz_beacon_block.rs
View File

@ -217,7 +217,7 @@ mod tests {
use super::super::ssz::encode::encode_length; use super::super::ssz::encode::encode_length;
use super::super::ssz::SszStream; use super::super::ssz::SszStream;
use super::super::types::Hash256; use super::super::types::Hash256;
use super::super::types::{AttestationRecord, BeaconBlock, SpecialRecord}; use super::super::types::{Attestation, BeaconBlock, SpecialRecord};
use super::*; use super::*;
fn get_block_ssz(b: &BeaconBlock) -> Vec<u8> { fn get_block_ssz(b: &BeaconBlock) -> Vec<u8> {
@ -232,7 +232,7 @@ mod tests {
ssz_stream.drain() ssz_stream.drain()
} }
fn get_attestation_record_ssz(ar: &AttestationRecord) -> Vec<u8> { fn get_attestation_record_ssz(ar: &Attestation) -> Vec<u8> {
let mut ssz_stream = SszStream::new(); let mut ssz_stream = SszStream::new();
ssz_stream.append(ar); ssz_stream.append(ar);
ssz_stream.drain() ssz_stream.drain()
@ -250,7 +250,7 @@ mod tests {
#[test] #[test]
fn test_ssz_block_single_attestation_record_one_byte_short() { fn test_ssz_block_single_attestation_record_one_byte_short() {
let mut b = BeaconBlock::zero(); let mut b = BeaconBlock::zero();
b.attestations = vec![AttestationRecord::zero()]; b.attestations = vec![Attestation::zero()];
let ssz = get_block_ssz(&b); let ssz = get_block_ssz(&b);
assert_eq!( assert_eq!(
@ -262,7 +262,7 @@ mod tests {
#[test] #[test]
fn test_ssz_block_single_attestation_record_one_byte_long() { fn test_ssz_block_single_attestation_record_one_byte_long() {
let mut b = BeaconBlock::zero(); let mut b = BeaconBlock::zero();
b.attestations = vec![AttestationRecord::zero()]; b.attestations = vec![Attestation::zero()];
let mut ssz = get_block_ssz(&b); let mut ssz = get_block_ssz(&b);
let original_len = ssz.len(); let original_len = ssz.len();
ssz.push(42); ssz.push(42);
@ -275,7 +275,7 @@ mod tests {
#[test] #[test]
fn test_ssz_block_single_attestation_record() { fn test_ssz_block_single_attestation_record() {
let mut b = BeaconBlock::zero(); let mut b = BeaconBlock::zero();
b.attestations = vec![AttestationRecord::zero()]; b.attestations = vec![Attestation::zero()];
let ssz = get_block_ssz(&b); let ssz = get_block_ssz(&b);
assert!(SszBeaconBlock::from_slice(&ssz[..]).is_ok()); assert!(SszBeaconBlock::from_slice(&ssz[..]).is_ok());
@ -284,7 +284,7 @@ mod tests {
#[test] #[test]
fn test_ssz_block_block_hash() { fn test_ssz_block_block_hash() {
let mut block = BeaconBlock::zero(); let mut block = BeaconBlock::zero();
block.attestations.push(AttestationRecord::zero()); block.attestations.push(Attestation::zero());
let serialized = get_block_ssz(&block); let serialized = get_block_ssz(&block);
let ssz_block = SszBeaconBlock::from_slice(&serialized).unwrap(); let ssz_block = SszBeaconBlock::from_slice(&serialized).unwrap();
let hash = ssz_block.block_hash(); let hash = ssz_block.block_hash();
@ -311,7 +311,7 @@ mod tests {
#[test] #[test]
fn test_ssz_block_slot() { fn test_ssz_block_slot() {
let mut block = BeaconBlock::zero(); let mut block = BeaconBlock::zero();
block.attestations.push(AttestationRecord::zero()); block.attestations.push(Attestation::zero());
block.slot = 42; block.slot = 42;
let serialized = get_block_ssz(&block); let serialized = get_block_ssz(&block);
@ -323,7 +323,7 @@ mod tests {
#[test] #[test]
fn test_ssz_block_randao_reveal() { fn test_ssz_block_randao_reveal() {
let mut block = BeaconBlock::zero(); let mut block = BeaconBlock::zero();
block.attestations.push(AttestationRecord::zero()); block.attestations.push(Attestation::zero());
let reference_hash = Hash256::from([42_u8; 32]); let reference_hash = Hash256::from([42_u8; 32]);
block.randao_reveal = reference_hash.clone(); block.randao_reveal = reference_hash.clone();
@ -404,14 +404,14 @@ mod tests {
#[test] #[test]
fn test_ssz_block_attestations() { fn test_ssz_block_attestations() {
/* /*
* Single AttestationRecord * Single Attestation
*/ */
let mut block = BeaconBlock::zero(); let mut block = BeaconBlock::zero();
block.attestations.push(AttestationRecord::zero()); block.attestations.push(Attestation::zero());
let serialized = get_block_ssz(&block); let serialized = get_block_ssz(&block);
let ssz_block = SszBeaconBlock::from_slice(&serialized).unwrap(); let ssz_block = SszBeaconBlock::from_slice(&serialized).unwrap();
let ssz_ar = get_attestation_record_ssz(&AttestationRecord::zero()); let ssz_ar = get_attestation_record_ssz(&Attestation::zero());
let mut expected = encode_length(ssz_ar.len(), LENGTH_PREFIX_BYTES); let mut expected = encode_length(ssz_ar.len(), LENGTH_PREFIX_BYTES);
expected.append(&mut ssz_ar.to_vec()); expected.append(&mut ssz_ar.to_vec());
@ -419,16 +419,16 @@ mod tests {
assert_eq!(ssz_block.attestations(), &expected[..]); assert_eq!(ssz_block.attestations(), &expected[..]);
/* /*
* Multiple AttestationRecords * Multiple Attestations
*/ */
let mut block = BeaconBlock::zero(); let mut block = BeaconBlock::zero();
block.attestations.push(AttestationRecord::zero()); block.attestations.push(Attestation::zero());
block.attestations.push(AttestationRecord::zero()); block.attestations.push(Attestation::zero());
let serialized = get_block_ssz(&block); let serialized = get_block_ssz(&block);
let ssz_block = SszBeaconBlock::from_slice(&serialized).unwrap(); let ssz_block = SszBeaconBlock::from_slice(&serialized).unwrap();
let mut ssz_ar = get_attestation_record_ssz(&AttestationRecord::zero()); let mut ssz_ar = get_attestation_record_ssz(&Attestation::zero());
ssz_ar.append(&mut get_attestation_record_ssz(&AttestationRecord::zero())); ssz_ar.append(&mut get_attestation_record_ssz(&Attestation::zero()));
let mut expected = encode_length(ssz_ar.len(), LENGTH_PREFIX_BYTES); let mut expected = encode_length(ssz_ar.len(), LENGTH_PREFIX_BYTES);
expected.append(&mut ssz_ar.to_vec()); expected.append(&mut ssz_ar.to_vec());
@ -439,7 +439,7 @@ mod tests {
#[test] #[test]
fn test_ssz_block_pow_chain_reference() { fn test_ssz_block_pow_chain_reference() {
let mut block = BeaconBlock::zero(); let mut block = BeaconBlock::zero();
block.attestations.push(AttestationRecord::zero()); block.attestations.push(Attestation::zero());
let reference_hash = Hash256::from([42_u8; 32]); let reference_hash = Hash256::from([42_u8; 32]);
block.pow_chain_reference = reference_hash.clone(); block.pow_chain_reference = reference_hash.clone();
@ -455,7 +455,7 @@ mod tests {
#[test] #[test]
fn test_ssz_block_act_state_root() { fn test_ssz_block_act_state_root() {
let mut block = BeaconBlock::zero(); let mut block = BeaconBlock::zero();
block.attestations.push(AttestationRecord::zero()); block.attestations.push(Attestation::zero());
let reference_hash = Hash256::from([42_u8; 32]); let reference_hash = Hash256::from([42_u8; 32]);
block.active_state_root = reference_hash.clone(); block.active_state_root = reference_hash.clone();
@ -468,7 +468,7 @@ mod tests {
#[test] #[test]
fn test_ssz_block_cry_state_root() { fn test_ssz_block_cry_state_root() {
let mut block = BeaconBlock::zero(); let mut block = BeaconBlock::zero();
block.attestations.push(AttestationRecord::zero()); block.attestations.push(Attestation::zero());
let reference_hash = Hash256::from([42_u8; 32]); let reference_hash = Hash256::from([42_u8; 32]);
block.crystallized_state_root = reference_hash.clone(); block.crystallized_state_root = reference_hash.clone();

14
lighthouse/db/src/stores/beacon_block_store.rs
View File

@ -87,7 +87,7 @@ mod tests {
extern crate types; extern crate types;
use self::ssz::SszStream; use self::ssz::SszStream;
use self::types::attestation_record::AttestationRecord; use self::types::attestation::Attestation;
use self::types::beacon_block::BeaconBlock; use self::types::beacon_block::BeaconBlock;
use self::types::Hash256; use self::types::Hash256;
@ -182,7 +182,10 @@ mod tests {
let hash = &Hash256::from("some hash".as_bytes()).to_vec(); let hash = &Hash256::from("some hash".as_bytes()).to_vec();
db.put(DB_COLUMN, hash, ssz).unwrap(); db.put(DB_COLUMN, hash, ssz).unwrap();
assert_eq!(store.block_at_slot(hash, 42), Err(BeaconBlockAtSlotError::InvalidBeaconBlock)); assert_eq!(
store.block_at_slot(hash, 42),
Err(BeaconBlockAtSlotError::InvalidBeaconBlock)
);
} }
#[test] #[test]
@ -195,7 +198,10 @@ mod tests {
let other_hash = &Hash256::from("another hash".as_bytes()).to_vec(); let other_hash = &Hash256::from("another hash".as_bytes()).to_vec();
db.put(DB_COLUMN, hash, ssz).unwrap(); db.put(DB_COLUMN, hash, ssz).unwrap();
assert_eq!(store.block_at_slot(other_hash, 42), Err(BeaconBlockAtSlotError::UnknownBeaconBlock)); assert_eq!(
store.block_at_slot(other_hash, 42),
Err(BeaconBlockAtSlotError::UnknownBeaconBlock)
);
} }
#[test] #[test]
@ -244,7 +250,7 @@ mod tests {
let blocks = (0..5).into_iter().map(|_| { let blocks = (0..5).into_iter().map(|_| {
let mut block = BeaconBlock::zero(); let mut block = BeaconBlock::zero();
let ar = AttestationRecord::zero(); let ar = Attestation::zero();
block.attestations.push(ar); block.attestations.push(ar);
block block
}); });