mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-09-14 13:30:11 +00:00
8a4ece1101
* [NOD-1223] Move all network stuff into a new network package. * [NOD-1223] Delete the unused package testutil. * [NOD-1223] Move infrastructure stuff into a new instrastructure package. * [NOD-1223] Move domain stuff into a new domain package.
73 lines
2.0 KiB
Go
73 lines
2.0 KiB
Go
package rpc
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/kaspanet/kaspad/network/rpc/model"
|
|
"github.com/kaspanet/kaspad/util/daghash"
|
|
)
|
|
|
|
// handleRescanBlocks implements the rescanBlocks command extension for
|
|
// websocket connections.
|
|
//
|
|
// NOTE: This extension is ported from github.com/decred/dcrd
|
|
func handleRescanBlocks(wsc *wsClient, icmd interface{}) (interface{}, error) {
|
|
cmd, ok := icmd.(*model.RescanBlocksCmd)
|
|
if !ok {
|
|
return nil, model.ErrRPCInternal
|
|
}
|
|
|
|
// Load client's transaction filter. Must exist in order to continue.
|
|
filter := wsc.FilterData()
|
|
if filter == nil {
|
|
return nil, &model.RPCError{
|
|
Code: model.ErrRPCMisc,
|
|
Message: "Transaction filter must be loaded before rescanning",
|
|
}
|
|
}
|
|
|
|
blockHashes := make([]*daghash.Hash, len(cmd.BlockHashes))
|
|
|
|
for i := range cmd.BlockHashes {
|
|
hash, err := daghash.NewHashFromStr(cmd.BlockHashes[i])
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
blockHashes[i] = hash
|
|
}
|
|
|
|
discoveredData := make([]model.RescannedBlock, 0, len(blockHashes))
|
|
|
|
// Iterate over each block in the request and rescan. When a block
|
|
// contains relevant transactions, add it to the response.
|
|
bc := wsc.server.dag
|
|
params := wsc.server.dag.Params
|
|
var lastBlockHash *daghash.Hash
|
|
for i := range blockHashes {
|
|
block, err := bc.BlockByHash(blockHashes[i])
|
|
if err != nil {
|
|
return nil, &model.RPCError{
|
|
Code: model.ErrRPCBlockNotFound,
|
|
Message: "Failed to fetch block: " + err.Error(),
|
|
}
|
|
}
|
|
if lastBlockHash != nil && !block.MsgBlock().Header.ParentHashes[0].IsEqual(lastBlockHash) { // TODO: (Stas) This is likely wrong. Modified to satisfy compilation.
|
|
return nil, &model.RPCError{
|
|
Code: model.ErrRPCInvalidParameter,
|
|
Message: fmt.Sprintf("Block %s is not a child of %s",
|
|
blockHashes[i], lastBlockHash),
|
|
}
|
|
}
|
|
lastBlockHash = blockHashes[i]
|
|
|
|
transactions := rescanBlockFilter(filter, block, params)
|
|
if len(transactions) != 0 {
|
|
discoveredData = append(discoveredData, model.RescannedBlock{
|
|
Hash: cmd.BlockHashes[i],
|
|
Transactions: transactions,
|
|
})
|
|
}
|
|
}
|
|
|
|
return &discoveredData, nil
|
|
}
|