mirror of
https://github.com/sigp/lighthouse.git
synced 2026-03-07 02:31:45 +00:00
3913ea44c6
* persist light client updates * update beacon chain to serve light client updates * resolve todos * cache best update * extend cache parts * is better light client update * resolve merge conflict * initial api changes * add lc update db column * fmt * added tests * add sim * Merge branch 'unstable' of https://github.com/sigp/lighthouse into persist-light-client-updates * fix some weird issues with the simulator * tests * Merge branch 'unstable' of https://github.com/sigp/lighthouse into persist-light-client-updates * test changes * merge conflict * testing * started work on ef tests and some code clean up * update tests * linting * noop pre altair, were still failing on electra though * allow for zeroed light client header * Merge branch 'unstable' of https://github.com/sigp/lighthouse into persist-light-client-updates * merge unstable * remove unwraps * remove unwraps * Update light_client_update.rs * merge unstable * move functionality to helper methods * refactor is best update fn * refactor is best update fn * improve organization of light client server cache logic * fork diget calc, and only spawn as many blcoks as we need for the lc update test * fetch lc update from the cache if it exists * fmt * Fix beacon_chain tests * Add debug code to update ranking_order ef test * Fix compare code * merge conflicts * fix test * Merge branch 'persist-light-client-updates' of https://github.com/eserilev/lighthouse into persist-light-client-updates * Use blinded blocks for light client proofs * fix ef test * merge conflicts * fix lc update check * Lint * resolve merge conflict * Merge branch 'persist-light-client-updates' of https://github.com/eserilev/lighthouse into persist-light-client-updates * revert basic sim * small fix * revert sim * Review PR * resolve merge conflicts * Merge branch 'unstable' into persist-light-client-updates
144 lines
4.9 KiB
Rust
144 lines
4.9 KiB
Rust
use beacon_chain::{BeaconChain, BeaconChainTypes};
|
|
use eth2::types::{
|
|
self as api_types, ChainSpec, ForkVersionedResponse, LightClientUpdate,
|
|
LightClientUpdateResponseChunk, LightClientUpdateSszResponse, LightClientUpdatesQuery,
|
|
};
|
|
use ssz::Encode;
|
|
use std::sync::Arc;
|
|
use warp::{
|
|
hyper::{Body, Response},
|
|
reply::Reply,
|
|
Rejection,
|
|
};
|
|
|
|
use crate::version::{add_ssz_content_type_header, fork_versioned_response, V1};
|
|
|
|
const MAX_REQUEST_LIGHT_CLIENT_UPDATES: u64 = 128;
|
|
|
|
pub fn get_light_client_updates<T: BeaconChainTypes>(
|
|
chain: Arc<BeaconChain<T>>,
|
|
query: LightClientUpdatesQuery,
|
|
accept_header: Option<api_types::Accept>,
|
|
) -> Result<Response<Body>, Rejection> {
|
|
validate_light_client_updates_request(&chain, &query)?;
|
|
|
|
let light_client_updates = chain
|
|
.get_light_client_updates(query.start_period, query.count)
|
|
.map_err(|_| {
|
|
warp_utils::reject::custom_not_found("No LightClientUpdates found".to_string())
|
|
})?;
|
|
|
|
match accept_header {
|
|
Some(api_types::Accept::Ssz) => {
|
|
let response_chunks = light_client_updates
|
|
.iter()
|
|
.map(|update| map_light_client_update_to_ssz_chunk::<T>(&chain, update))
|
|
.collect::<Vec<LightClientUpdateResponseChunk>>();
|
|
|
|
let ssz_response = LightClientUpdateSszResponse {
|
|
response_chunk_len: (light_client_updates.len() as u64).to_le_bytes().to_vec(),
|
|
response_chunk: response_chunks.as_ssz_bytes(),
|
|
}
|
|
.as_ssz_bytes();
|
|
|
|
Response::builder()
|
|
.status(200)
|
|
.body(ssz_response)
|
|
.map(|res: Response<Vec<u8>>| add_ssz_content_type_header(res))
|
|
.map_err(|e| {
|
|
warp_utils::reject::custom_server_error(format!(
|
|
"failed to create response: {}",
|
|
e
|
|
))
|
|
})
|
|
}
|
|
_ => {
|
|
let fork_versioned_response = light_client_updates
|
|
.iter()
|
|
.map(|update| map_light_client_update_to_json_response::<T>(&chain, update.clone()))
|
|
.collect::<Result<Vec<ForkVersionedResponse<LightClientUpdate<T::EthSpec>>>, Rejection>>()?;
|
|
Ok(warp::reply::json(&fork_versioned_response).into_response())
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn validate_light_client_updates_request<T: BeaconChainTypes>(
|
|
chain: &BeaconChain<T>,
|
|
query: &LightClientUpdatesQuery,
|
|
) -> Result<(), Rejection> {
|
|
if query.count > MAX_REQUEST_LIGHT_CLIENT_UPDATES {
|
|
return Err(warp_utils::reject::custom_bad_request(
|
|
"Invalid count requested".to_string(),
|
|
));
|
|
}
|
|
|
|
let current_sync_period = chain
|
|
.epoch()
|
|
.map_err(|_| {
|
|
warp_utils::reject::custom_server_error("failed to get current epoch".to_string())
|
|
})?
|
|
.sync_committee_period(&chain.spec)
|
|
.map_err(|_| {
|
|
warp_utils::reject::custom_server_error(
|
|
"failed to get current sync committee period".to_string(),
|
|
)
|
|
})?;
|
|
|
|
if query.start_period > current_sync_period {
|
|
return Err(warp_utils::reject::custom_bad_request(
|
|
"Invalid sync committee period requested".to_string(),
|
|
));
|
|
}
|
|
|
|
let earliest_altair_sync_committee = chain
|
|
.spec
|
|
.altair_fork_epoch
|
|
.ok_or(warp_utils::reject::custom_server_error(
|
|
"failed to get altair fork epoch".to_string(),
|
|
))?
|
|
.sync_committee_period(&chain.spec)
|
|
.map_err(|_| {
|
|
warp_utils::reject::custom_server_error(
|
|
"failed to get earliest altair sync committee".to_string(),
|
|
)
|
|
})?;
|
|
|
|
if query.start_period < earliest_altair_sync_committee {
|
|
return Err(warp_utils::reject::custom_bad_request(
|
|
"Invalid sync committee period requested".to_string(),
|
|
));
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
fn map_light_client_update_to_ssz_chunk<T: BeaconChainTypes>(
|
|
chain: &BeaconChain<T>,
|
|
light_client_update: &LightClientUpdate<T::EthSpec>,
|
|
) -> LightClientUpdateResponseChunk {
|
|
let fork_name = chain
|
|
.spec
|
|
.fork_name_at_slot::<T::EthSpec>(*light_client_update.signature_slot());
|
|
|
|
let fork_digest = ChainSpec::compute_fork_digest(
|
|
chain.spec.fork_version_for_name(fork_name),
|
|
chain.genesis_validators_root,
|
|
);
|
|
|
|
LightClientUpdateResponseChunk {
|
|
context: fork_digest,
|
|
payload: light_client_update.as_ssz_bytes(),
|
|
}
|
|
}
|
|
|
|
fn map_light_client_update_to_json_response<T: BeaconChainTypes>(
|
|
chain: &BeaconChain<T>,
|
|
light_client_update: LightClientUpdate<T::EthSpec>,
|
|
) -> Result<ForkVersionedResponse<LightClientUpdate<T::EthSpec>>, Rejection> {
|
|
let fork_name = chain
|
|
.spec
|
|
.fork_name_at_slot::<T::EthSpec>(*light_client_update.signature_slot());
|
|
|
|
fork_versioned_response(V1, fork_name, light_client_update)
|
|
}
|