24e941d175
* Update to spec v0.9.0 * Update to v0.9.1 * Bump spec tags for v0.9.1 * Formatting, fix CI failures * Resolve accidental KeyPair merge conflict * Document new BeaconState functions * Fix incorrect cache drops in `advance_caches` * Update fork choice for v0.9.1 * Clean up some FIXMEs * Fix a few docs/logs
39 lines
1.2 KiB
Rust
39 lines
1.2 KiB
Rust
use int_to_bytes::int_to_bytes8;
|
|
use ssz::ssz_encode;
|
|
use ssz_derive::{Decode, Encode};
|
|
use types::{AttestationData, BeaconState, ChainSpec, Domain, Epoch, EthSpec};
|
|
|
|
/// Serialized `AttestationData` augmented with a domain to encode the fork info.
|
|
#[derive(PartialEq, Eq, Clone, Hash, Debug, PartialOrd, Ord, Encode, Decode)]
|
|
pub struct AttestationId {
|
|
v: Vec<u8>,
|
|
}
|
|
|
|
/// Number of domain bytes that the end of an attestation ID is padded with.
|
|
const DOMAIN_BYTES_LEN: usize = 8;
|
|
|
|
impl AttestationId {
|
|
pub fn from_data<T: EthSpec>(
|
|
attestation: &AttestationData,
|
|
state: &BeaconState<T>,
|
|
spec: &ChainSpec,
|
|
) -> Self {
|
|
let mut bytes = ssz_encode(attestation);
|
|
let epoch = attestation.target.epoch;
|
|
bytes.extend_from_slice(&AttestationId::compute_domain_bytes(epoch, state, spec));
|
|
AttestationId { v: bytes }
|
|
}
|
|
|
|
pub fn compute_domain_bytes<T: EthSpec>(
|
|
epoch: Epoch,
|
|
state: &BeaconState<T>,
|
|
spec: &ChainSpec,
|
|
) -> Vec<u8> {
|
|
int_to_bytes8(spec.get_domain(epoch, Domain::BeaconAttester, &state.fork))
|
|
}
|
|
|
|
pub fn domain_bytes_match(&self, domain_bytes: &[u8]) -> bool {
|
|
&self.v[self.v.len() - DOMAIN_BYTES_LEN..] == domain_bytes
|
|
}
|
|
}
|