mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-29 10:16:45 +00:00

* [NOD-1444] Implement getHeaders RPC command * [NOD-1444] Fix tests and comments * [NOD-1444] Fix error message * [NOD-1444] Make GetHeaders propagate header serialization errors * [NOD-1444] RLock the dag on GetHeaders * [NOD-1444] Change the error field number to 1000
53 lines
1.6 KiB
Go
53 lines
1.6 KiB
Go
package rpchandlers
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/hex"
|
|
"github.com/kaspanet/kaspad/app/appmessage"
|
|
"github.com/kaspanet/kaspad/app/rpc/rpccontext"
|
|
"github.com/kaspanet/kaspad/infrastructure/network/netadapter/router"
|
|
"github.com/kaspanet/kaspad/util/daghash"
|
|
)
|
|
|
|
// HandleGetHeaders handles the respectively named RPC command
|
|
func HandleGetHeaders(context *rpccontext.Context, _ *router.Router, request appmessage.Message) (appmessage.Message, error) {
|
|
getHeadersRequest := request.(*appmessage.GetHeadersRequestMessage)
|
|
dag := context.DAG
|
|
|
|
var startHash *daghash.Hash
|
|
if getHeadersRequest.StartHash != "" {
|
|
var err error
|
|
startHash, err = daghash.NewHashFromStr(getHeadersRequest.StartHash)
|
|
if err != nil {
|
|
errorMessage := &appmessage.GetHeadersResponseMessage{}
|
|
errorMessage.Error = appmessage.RPCErrorf("Start hash could not be parsed: %s", err)
|
|
return errorMessage, nil
|
|
}
|
|
}
|
|
|
|
const getHeadersDefaultLimit uint64 = 2000
|
|
limit := getHeadersDefaultLimit
|
|
if getHeadersRequest.Limit != 0 {
|
|
limit = getHeadersRequest.Limit
|
|
}
|
|
|
|
headers, err := dag.GetHeaders(startHash, limit, getHeadersRequest.IsAscending)
|
|
if err != nil {
|
|
errorMessage := &appmessage.GetHeadersResponseMessage{}
|
|
errorMessage.Error = appmessage.RPCErrorf("Error getting the headers: %s", err)
|
|
return errorMessage, nil
|
|
}
|
|
|
|
headersHex := make([]string, len(headers))
|
|
var buf bytes.Buffer
|
|
for i, header := range headers {
|
|
err := header.Serialize(&buf)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
headersHex[i] = hex.EncodeToString(buf.Bytes())
|
|
buf.Reset()
|
|
}
|
|
return appmessage.NewGetHeadersResponseMessage(headersHex), nil
|
|
}
|