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

Update handle_rpc_response blobs match arms to be consistent with block v2 protocols.

Browse Source
This commit is contained in:
Jimmy Chen 2023-06-28 16:56:52 +10:00
parent 68140fa036
commit 03a17a84da
No known key found for this signature in database
GPG Key ID: 7AAEE02659DCF690
2 changed files with 34 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

74
beacon_node/lighthouse_network/src/rpc/codec/ssz_snappy.rs
View File

@ -530,7 +530,7 @@ fn handle_rpc_request<T: EthSpec>(
fn handle_rpc_response<T: EthSpec>( fn handle_rpc_response<T: EthSpec>(
versioned_protocol: SupportedProtocol, versioned_protocol: SupportedProtocol,
decoded_buffer: &[u8], decoded_buffer: &[u8],
mut fork_name: Option<ForkName>, fork_name: Option<ForkName>,
) -> Result<Option<RPCResponse<T>>, RPCError> { ) -> Result<Option<RPCResponse<T>>, RPCError> {
match versioned_protocol { match versioned_protocol {
SupportedProtocol::StatusV1 => Ok(Some(RPCResponse::Status( SupportedProtocol::StatusV1 => Ok(Some(RPCResponse::Status(
@ -546,46 +546,38 @@ fn handle_rpc_response<T: EthSpec>(
SupportedProtocol::BlocksByRootV1 => Ok(Some(RPCResponse::BlocksByRoot(Arc::new( SupportedProtocol::BlocksByRootV1 => Ok(Some(RPCResponse::BlocksByRoot(Arc::new(
SignedBeaconBlock::Base(SignedBeaconBlockBase::from_ssz_bytes(decoded_buffer)?), SignedBeaconBlock::Base(SignedBeaconBlockBase::from_ssz_bytes(decoded_buffer)?),
)))), )))),
SupportedProtocol::BlobsByRangeV1 => { SupportedProtocol::BlobsByRangeV1 => match fork_name {
let fork_name = fork_name.take().ok_or_else(|| { Some(ForkName::Deneb) => Ok(Some(RPCResponse::BlobsByRange(Arc::new(
RPCError::ErrorResponse( BlobSidecar::from_ssz_bytes(decoded_buffer)?,
RPCResponseErrorCode::InvalidRequest, )))),
format!( Some(_) => Err(RPCError::ErrorResponse(
"No context bytes provided for {:?} response", RPCResponseErrorCode::InvalidRequest,
versioned_protocol "Invalid fork name for blobs by range".to_string(),
), )),
) None => Err(RPCError::ErrorResponse(
})?; RPCResponseErrorCode::InvalidRequest,
match fork_name { format!(
ForkName::Deneb => Ok(Some(RPCResponse::BlobsByRange(Arc::new( "No context bytes provided for {:?} response",
BlobSidecar::from_ssz_bytes(decoded_buffer)?, versioned_protocol
)))), ),
_ => Err(RPCError::ErrorResponse( )),
RPCResponseErrorCode::InvalidRequest, },
"Invalid fork name for blobs by range".to_string(), SupportedProtocol::BlobsByRootV1 => match fork_name {
)), Some(ForkName::Deneb) => Ok(Some(RPCResponse::SidecarByRoot(Arc::new(
} BlobSidecar::from_ssz_bytes(decoded_buffer)?,
} )))),
SupportedProtocol::BlobsByRootV1 => { Some(_) => Err(RPCError::ErrorResponse(
let fork_name = fork_name.take().ok_or_else(|| { RPCResponseErrorCode::InvalidRequest,
RPCError::ErrorResponse( "Invalid fork name for blobs by root".to_string(),
RPCResponseErrorCode::InvalidRequest, )),
format!( None => Err(RPCError::ErrorResponse(
"No context bytes provided for {:?} response", RPCResponseErrorCode::InvalidRequest,
versioned_protocol format!(
), "No context bytes provided for {:?} response",
) versioned_protocol
})?; ),
match fork_name { )),
ForkName::Deneb => Ok(Some(RPCResponse::SidecarByRoot(Arc::new( },
BlobSidecar::from_ssz_bytes(decoded_buffer)?,
)))),
_ => Err(RPCError::ErrorResponse(
RPCResponseErrorCode::InvalidRequest,
"Invalid fork name for block and blobs by root".to_string(),
)),
}
}
SupportedProtocol::PingV1 => Ok(Some(RPCResponse::Pong(Ping { SupportedProtocol::PingV1 => Ok(Some(RPCResponse::Pong(Ping {
data: u64::from_ssz_bytes(decoded_buffer)?, data: u64::from_ssz_bytes(decoded_buffer)?,
}))), }))),

2
beacon_node/network/src/beacon_processor/worker/rpc_methods.rs
View File

@ -375,7 +375,7 @@ impl<T: BeaconChainTypes> Worker<T> {
send_on_drop: SendOnDrop, send_on_drop: SendOnDrop,
peer_id: PeerId, peer_id: PeerId,
request_id: PeerRequestId, request_id: PeerRequestId,
mut req: BlocksByRangeRequest, req: BlocksByRangeRequest,
) { ) {
debug!(self.log, "Received BlocksByRange Request"; debug!(self.log, "Received BlocksByRange Request";
"peer_id" => %peer_id, "peer_id" => %peer_id,