mirror of
https://github.com/sigp/lighthouse.git
synced 2026-03-14 10:22:38 +00:00
Persist light client updates (#5545)
* 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
This commit is contained in:
Eitan Seri-Levi
143
beacon_node/http_api/src/light_client.rs
Normal file
143
beacon_node/http_api/src/light_client.rs
Normal file
@@ -0,0 +1,143 @@
|
||||
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)
|
||||
}
|
||||
Reference in New Issue
Block a user