131 lines
3.6 KiB
Nim
131 lines
3.6 KiB
Nim
# Nimbus
|
|
# Copyright (c) 2021 Status Research & Development GmbH
|
|
# Licensed under either of
|
|
# * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE) or
|
|
# http://www.apache.org/licenses/LICENSE-2.0)
|
|
# * MIT license ([LICENSE-MIT](LICENSE-MIT) or
|
|
# http://opensource.org/licenses/MIT)
|
|
# at your option. This file may not be copied, modified, or distributed
|
|
# except according to those terms.
|
|
|
|
{.push raises: [].}
|
|
|
|
import
|
|
chronos,
|
|
eth/[common, p2p],
|
|
stew/byteutils,
|
|
"../../.."/[protocol, protocol/trace_config, types],
|
|
../../worker_desc,
|
|
./get_error
|
|
|
|
logScope:
|
|
topics = "snap-get"
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Public functions
|
|
# ------------------------------------------------------------------------------
|
|
|
|
proc getBlockHeader*(
|
|
buddy: SnapBuddyRef;
|
|
num: BlockNumber;
|
|
): Future[Result[BlockHeader,GetError]]
|
|
{.async.} =
|
|
## Get single block header
|
|
let
|
|
peer = buddy.peer
|
|
reqLen = 1u
|
|
hdrReq = BlocksRequest(
|
|
startBlock: HashOrNum(
|
|
isHash: false,
|
|
number: num),
|
|
maxResults: reqLen,
|
|
skip: 0,
|
|
reverse: false)
|
|
|
|
when trSnapTracePacketsOk:
|
|
trace trEthSendSendingGetBlockHeaders, peer, header=num.toStr, reqLen
|
|
|
|
var hdrResp: Option[blockHeadersObj]
|
|
try:
|
|
hdrResp = await peer.getBlockHeaders(hdrReq)
|
|
except CatchableError as e:
|
|
when trSnapTracePacketsOk:
|
|
trace trSnapRecvError & "waiting for GetByteCodes reply", peer,
|
|
error=e.msg
|
|
return err(GetNetworkProblem)
|
|
|
|
var hdrRespLen = 0
|
|
if hdrResp.isSome:
|
|
hdrRespLen = hdrResp.get.headers.len
|
|
if hdrRespLen == 0:
|
|
when trSnapTracePacketsOk:
|
|
trace trEthRecvReceivedBlockHeaders, peer, reqLen, respose="n/a"
|
|
return err(GetNoHeaderAvailable)
|
|
|
|
if hdrRespLen == 1:
|
|
let
|
|
header = hdrResp.get.headers[0]
|
|
blockNumber = header.blockNumber
|
|
when trSnapTracePacketsOk:
|
|
trace trEthRecvReceivedBlockHeaders, peer, hdrRespLen, blockNumber
|
|
return ok(header)
|
|
|
|
when trSnapTracePacketsOk:
|
|
trace trEthRecvReceivedBlockHeaders, peer, reqLen, hdrRespLen
|
|
return err(GetTooManyHeaders)
|
|
|
|
|
|
proc getBlockHeader*(
|
|
buddy: SnapBuddyRef;
|
|
hash: Hash256;
|
|
): Future[Result[BlockHeader,GetError]]
|
|
{.async.} =
|
|
## Get single block header
|
|
let
|
|
peer = buddy.peer
|
|
reqLen = 1u
|
|
hdrReq = BlocksRequest(
|
|
startBlock: HashOrNum(
|
|
isHash: true,
|
|
hash: hash),
|
|
maxResults: reqLen,
|
|
skip: 0,
|
|
reverse: false)
|
|
|
|
when trSnapTracePacketsOk:
|
|
trace trEthSendSendingGetBlockHeaders, peer,
|
|
header=hash.data.toHex, reqLen
|
|
|
|
var hdrResp: Option[blockHeadersObj]
|
|
try:
|
|
hdrResp = await peer.getBlockHeaders(hdrReq)
|
|
except CatchableError as e:
|
|
when trSnapTracePacketsOk:
|
|
trace trSnapRecvError & "waiting for GetByteCodes reply", peer,
|
|
error=e.msg
|
|
return err(GetNetworkProblem)
|
|
|
|
var hdrRespLen = 0
|
|
if hdrResp.isSome:
|
|
hdrRespLen = hdrResp.get.headers.len
|
|
if hdrRespLen == 0:
|
|
when trSnapTracePacketsOk:
|
|
trace trEthRecvReceivedBlockHeaders, peer, reqLen, respose="n/a"
|
|
return err(GetNoHeaderAvailable)
|
|
|
|
if hdrRespLen == 1:
|
|
let
|
|
header = hdrResp.get.headers[0]
|
|
blockNumber = header.blockNumber
|
|
when trSnapTracePacketsOk:
|
|
trace trEthRecvReceivedBlockHeaders, peer, hdrRespLen, blockNumber
|
|
return ok(header)
|
|
|
|
when trSnapTracePacketsOk:
|
|
trace trEthRecvReceivedBlockHeaders, peer, reqLen, hdrRespLen
|
|
return err(GetTooManyHeaders)
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# End
|
|
# ------------------------------------------------------------------------------
|