dione/node/network_service.go

package node

import (
	"context"
	"errors"
	"fmt"

	"github.com/Secured-Finance/dione/blockchain"

	gorpc "github.com/libp2p/go-libp2p-gorpc"

	"github.com/sirupsen/logrus"

	"github.com/Secured-Finance/dione/consensus/policy"
	"github.com/Secured-Finance/dione/node/wire"

	"github.com/Secured-Finance/dione/blockchain/pool"
)

type NetworkService struct {
	blockpool *blockchain.BlockChain
	mempool   *pool.Mempool
	rpcClient *gorpc.Client
}

func NewNetworkService(bp *blockchain.BlockChain) *NetworkService {
	return &NetworkService{
		blockpool: bp,
	}
}

func (s *NetworkService) LastBlockHeight(ctx context.Context, arg struct{}, reply *wire.LastBlockHeightReply) {
	height, err := s.blockpool.GetLatestBlockHeight()
	if err != nil {
		reply.Error = err
		return
	}
	reply.Height = height
}

func (s *NetworkService) GetRangeOfBlocks(ctx context.Context, arg wire.GetRangeOfBlocksArg, reply *wire.GetRangeOfBlocksReply) {
	if arg.From > arg.To {
		errText := "incorrect arguments: from > to"
		reply.Error = &errText
		return
	}
	if arg.To-arg.From > policy.MaxBlockCountForRetrieving {
		errText := "incorrect arguments: count of block for retrieving is exceeded the limit"
		reply.Error = &errText
		return
	}
	for i := arg.From; i <= arg.To; i++ {
		block, err := s.blockpool.FetchBlockByHeight(i)
		if err != nil {
			logrus.Warnf("failed to retrieve block from blockpool with height %d", i)
			reply.FailedBlockHeights = append(reply.FailedBlockHeights, i)
			continue
		}
		reply.Blocks = append(reply.Blocks, *block)
	}
}

func (s *NetworkService) Mempool(ctx context.Context, arg struct{}, reply *wire.InvMessage) {
	txs := s.mempool.GetAllTransactions()

	// extract hashes of txs
	for _, v := range txs {
		reply.Inventory = append(reply.Inventory, wire.InvItem{
			Type: wire.TxInvType,
			Hash: v.Hash,
		})
	}
}

func (s *NetworkService) GetMempoolTxs(ctx context.Context, arg wire.GetMempoolTxsArg, reply *wire.GetMempoolTxsReply) {
	if len(arg.Items) > MaxTransactionCountForRetrieving {
		pid, _ := gorpc.GetRequestSender(ctx)
		logrus.Warnf("Max tx count limit exceeded for GetMempoolTxs request of node %s", pid)
		reply.Error = fmt.Errorf("max tx count limit exceeded")
		return
	}

	for _, v := range arg.Items {
		tx, err := s.mempool.GetTransaction(v)
		if err != nil {
			if errors.Is(err, pool.ErrTxNotFound) {
				reply.NotFoundTxs = append(reply.NotFoundTxs, v)
			} else {
				reply.Error = err
				return
			}
		}
		reply.Transactions = append(reply.Transactions, *tx)
	}
}