mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2024-12-25 04:57:17 +00:00
326 lines
11 KiB
Go
326 lines
11 KiB
Go
package rpcservices
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"sync/atomic"
|
|
|
|
libcommon "github.com/ledgerwatch/erigon-lib/common"
|
|
"github.com/ledgerwatch/erigon-lib/gointerfaces"
|
|
"github.com/ledgerwatch/erigon-lib/gointerfaces/remote"
|
|
types2 "github.com/ledgerwatch/erigon-lib/gointerfaces/types"
|
|
"github.com/ledgerwatch/erigon-lib/kv"
|
|
"github.com/ledgerwatch/log/v3"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/status"
|
|
"google.golang.org/protobuf/types/known/emptypb"
|
|
|
|
"github.com/ledgerwatch/erigon/core/types"
|
|
"github.com/ledgerwatch/erigon/ethdb/privateapi"
|
|
"github.com/ledgerwatch/erigon/p2p"
|
|
"github.com/ledgerwatch/erigon/rlp"
|
|
"github.com/ledgerwatch/erigon/turbo/services"
|
|
)
|
|
|
|
type RemoteBackend struct {
|
|
remoteEthBackend remote.ETHBACKENDClient
|
|
log log.Logger
|
|
version gointerfaces.Version
|
|
db kv.RoDB
|
|
blockReader services.FullBlockReader
|
|
}
|
|
|
|
func NewRemoteBackend(client remote.ETHBACKENDClient, db kv.RoDB, blockReader services.FullBlockReader) *RemoteBackend {
|
|
return &RemoteBackend{
|
|
remoteEthBackend: client,
|
|
version: gointerfaces.VersionFromProto(privateapi.EthBackendAPIVersion),
|
|
log: log.New("remote_service", "eth_backend"),
|
|
db: db,
|
|
blockReader: blockReader,
|
|
}
|
|
}
|
|
|
|
func (back *RemoteBackend) EnsureVersionCompatibility() bool {
|
|
versionReply, err := back.remoteEthBackend.Version(context.Background(), &emptypb.Empty{}, grpc.WaitForReady(true))
|
|
if err != nil {
|
|
|
|
back.log.Error("getting Version", "err", err)
|
|
return false
|
|
}
|
|
if !gointerfaces.EnsureVersion(back.version, versionReply) {
|
|
back.log.Error("incompatible interface versions", "client", back.version.String(),
|
|
"server", fmt.Sprintf("%d.%d.%d", versionReply.Major, versionReply.Minor, versionReply.Patch))
|
|
return false
|
|
}
|
|
back.log.Info("interfaces compatible", "client", back.version.String(),
|
|
"server", fmt.Sprintf("%d.%d.%d", versionReply.Major, versionReply.Minor, versionReply.Patch))
|
|
return true
|
|
}
|
|
|
|
func (back *RemoteBackend) Etherbase(ctx context.Context) (libcommon.Address, error) {
|
|
res, err := back.remoteEthBackend.Etherbase(ctx, &remote.EtherbaseRequest{})
|
|
if err != nil {
|
|
if s, ok := status.FromError(err); ok {
|
|
return libcommon.Address{}, errors.New(s.Message())
|
|
}
|
|
return libcommon.Address{}, err
|
|
}
|
|
|
|
return gointerfaces.ConvertH160toAddress(res.Address), nil
|
|
}
|
|
|
|
func (back *RemoteBackend) NetVersion(ctx context.Context) (uint64, error) {
|
|
res, err := back.remoteEthBackend.NetVersion(ctx, &remote.NetVersionRequest{})
|
|
if err != nil {
|
|
if s, ok := status.FromError(err); ok {
|
|
return 0, errors.New(s.Message())
|
|
}
|
|
return 0, err
|
|
}
|
|
|
|
return res.Id, nil
|
|
}
|
|
|
|
func (back *RemoteBackend) NetPeerCount(ctx context.Context) (uint64, error) {
|
|
res, err := back.remoteEthBackend.NetPeerCount(ctx, &remote.NetPeerCountRequest{})
|
|
if err != nil {
|
|
if s, ok := status.FromError(err); ok {
|
|
return 0, errors.New(s.Message())
|
|
}
|
|
return 0, err
|
|
}
|
|
|
|
return res.Count, nil
|
|
}
|
|
|
|
func (back *RemoteBackend) ProtocolVersion(ctx context.Context) (uint64, error) {
|
|
res, err := back.remoteEthBackend.ProtocolVersion(ctx, &remote.ProtocolVersionRequest{})
|
|
if err != nil {
|
|
if s, ok := status.FromError(err); ok {
|
|
return 0, errors.New(s.Message())
|
|
}
|
|
return 0, err
|
|
}
|
|
|
|
return res.Id, nil
|
|
}
|
|
|
|
func (back *RemoteBackend) ClientVersion(ctx context.Context) (string, error) {
|
|
res, err := back.remoteEthBackend.ClientVersion(ctx, &remote.ClientVersionRequest{})
|
|
if err != nil {
|
|
if s, ok := status.FromError(err); ok {
|
|
return "", errors.New(s.Message())
|
|
}
|
|
return "", err
|
|
}
|
|
|
|
return res.NodeName, nil
|
|
}
|
|
|
|
func (back *RemoteBackend) Subscribe(ctx context.Context, onNewEvent func(*remote.SubscribeReply)) error {
|
|
subscription, err := back.remoteEthBackend.Subscribe(ctx, &remote.SubscribeRequest{}, grpc.WaitForReady(true))
|
|
if err != nil {
|
|
if s, ok := status.FromError(err); ok {
|
|
return errors.New(s.Message())
|
|
}
|
|
return err
|
|
}
|
|
for {
|
|
event, err := subscription.Recv()
|
|
if errors.Is(err, io.EOF) {
|
|
log.Debug("rpcdaemon: the subscription channel was closed")
|
|
break
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
onNewEvent(event)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (back *RemoteBackend) SubscribeLogs(ctx context.Context, onNewLogs func(reply *remote.SubscribeLogsReply), requestor *atomic.Value) error {
|
|
subscription, err := back.remoteEthBackend.SubscribeLogs(ctx, grpc.WaitForReady(true))
|
|
if err != nil {
|
|
if s, ok := status.FromError(err); ok {
|
|
return errors.New(s.Message())
|
|
}
|
|
return err
|
|
}
|
|
requestor.Store(subscription.Send)
|
|
for {
|
|
logs, err := subscription.Recv()
|
|
if errors.Is(err, io.EOF) {
|
|
log.Info("rpcdaemon: the logs subscription channel was closed")
|
|
break
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
onNewLogs(logs)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (back *RemoteBackend) TxnLookup(ctx context.Context, tx kv.Getter, txnHash libcommon.Hash) (uint64, bool, error) {
|
|
return back.blockReader.TxnLookup(ctx, tx, txnHash)
|
|
}
|
|
func (back *RemoteBackend) BlockWithSenders(ctx context.Context, tx kv.Getter, hash libcommon.Hash, blockHeight uint64) (block *types.Block, senders []libcommon.Address, err error) {
|
|
return back.blockReader.BlockWithSenders(ctx, tx, hash, blockHeight)
|
|
}
|
|
func (back *RemoteBackend) BodyWithTransactions(ctx context.Context, tx kv.Getter, hash libcommon.Hash, blockHeight uint64) (body *types.Body, err error) {
|
|
return back.blockReader.BodyWithTransactions(ctx, tx, hash, blockHeight)
|
|
}
|
|
func (back *RemoteBackend) BodyRlp(ctx context.Context, tx kv.Getter, hash libcommon.Hash, blockHeight uint64) (bodyRlp rlp.RawValue, err error) {
|
|
return back.blockReader.BodyRlp(ctx, tx, hash, blockHeight)
|
|
}
|
|
func (back *RemoteBackend) Body(ctx context.Context, tx kv.Getter, hash libcommon.Hash, blockHeight uint64) (body *types.Body, txAmount uint32, err error) {
|
|
return back.blockReader.Body(ctx, tx, hash, blockHeight)
|
|
}
|
|
func (back *RemoteBackend) Header(ctx context.Context, tx kv.Getter, hash libcommon.Hash, blockHeight uint64) (*types.Header, error) {
|
|
return back.blockReader.Header(ctx, tx, hash, blockHeight)
|
|
}
|
|
func (back *RemoteBackend) HeaderByNumber(ctx context.Context, tx kv.Getter, blockHeight uint64) (*types.Header, error) {
|
|
return back.blockReader.HeaderByNumber(ctx, tx, blockHeight)
|
|
}
|
|
func (back *RemoteBackend) HeaderByHash(ctx context.Context, tx kv.Getter, hash libcommon.Hash) (*types.Header, error) {
|
|
return back.blockReader.HeaderByHash(ctx, tx, hash)
|
|
}
|
|
func (back *RemoteBackend) CanonicalHash(ctx context.Context, tx kv.Getter, blockHeight uint64) (libcommon.Hash, error) {
|
|
return back.blockReader.CanonicalHash(ctx, tx, blockHeight)
|
|
}
|
|
func (back *RemoteBackend) TxnByIdxInBlock(ctx context.Context, tx kv.Getter, blockNum uint64, i int) (types.Transaction, error) {
|
|
return back.blockReader.TxnByIdxInBlock(ctx, tx, blockNum, i)
|
|
}
|
|
|
|
func (back *RemoteBackend) EngineNewPayloadV1(ctx context.Context, payload *types2.ExecutionPayload) (res *remote.EnginePayloadStatus, err error) {
|
|
return back.remoteEthBackend.EngineNewPayloadV1(ctx, payload)
|
|
}
|
|
|
|
func (back *RemoteBackend) EngineNewPayloadV2(ctx context.Context, payload *types2.ExecutionPayloadV2) (res *remote.EnginePayloadStatus, err error) {
|
|
return back.remoteEthBackend.EngineNewPayloadV2(ctx, payload)
|
|
}
|
|
|
|
func (back *RemoteBackend) EngineForkchoiceUpdatedV1(ctx context.Context, request *remote.EngineForkChoiceUpdatedRequest) (*remote.EngineForkChoiceUpdatedReply, error) {
|
|
return back.remoteEthBackend.EngineForkChoiceUpdatedV1(ctx, request)
|
|
}
|
|
|
|
func (back *RemoteBackend) EngineForkchoiceUpdatedV2(ctx context.Context, request *remote.EngineForkChoiceUpdatedRequestV2) (*remote.EngineForkChoiceUpdatedReply, error) {
|
|
return back.remoteEthBackend.EngineForkChoiceUpdatedV2(ctx, request)
|
|
}
|
|
|
|
func (back *RemoteBackend) EngineGetPayloadV1(ctx context.Context, payloadId uint64) (res *types2.ExecutionPayload, err error) {
|
|
return back.remoteEthBackend.EngineGetPayloadV1(ctx, &remote.EngineGetPayloadRequest{
|
|
PayloadId: payloadId,
|
|
})
|
|
}
|
|
|
|
func (back *RemoteBackend) EngineGetPayloadV2(ctx context.Context, payloadId uint64) (res *types2.ExecutionPayloadV2, err error) {
|
|
return back.remoteEthBackend.EngineGetPayloadV2(ctx, &remote.EngineGetPayloadRequest{
|
|
PayloadId: payloadId,
|
|
})
|
|
}
|
|
|
|
func (back *RemoteBackend) NodeInfo(ctx context.Context, limit uint32) ([]p2p.NodeInfo, error) {
|
|
nodes, err := back.remoteEthBackend.NodeInfo(ctx, &remote.NodesInfoRequest{Limit: limit})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("nodes info request error: %w", err)
|
|
}
|
|
|
|
if nodes == nil || len(nodes.NodesInfo) == 0 {
|
|
return nil, errors.New("empty nodesInfo response")
|
|
}
|
|
|
|
ret := make([]p2p.NodeInfo, 0, len(nodes.NodesInfo))
|
|
for _, node := range nodes.NodesInfo {
|
|
var rawProtocols map[string]json.RawMessage
|
|
if err = json.Unmarshal(node.Protocols, &rawProtocols); err != nil {
|
|
return nil, fmt.Errorf("cannot decode protocols metadata: %w", err)
|
|
}
|
|
|
|
protocols := make(map[string]interface{}, len(rawProtocols))
|
|
for k, v := range rawProtocols {
|
|
protocols[k] = v
|
|
}
|
|
|
|
ret = append(ret, p2p.NodeInfo{
|
|
Enode: node.Enode,
|
|
ID: node.Id,
|
|
IP: node.Enode,
|
|
ENR: node.Enr,
|
|
ListenAddr: node.ListenerAddr,
|
|
Name: node.Name,
|
|
Ports: struct {
|
|
Discovery int `json:"discovery"`
|
|
Listener int `json:"listener"`
|
|
}{
|
|
Discovery: int(node.Ports.Discovery),
|
|
Listener: int(node.Ports.Listener),
|
|
},
|
|
Protocols: protocols,
|
|
})
|
|
}
|
|
|
|
return ret, nil
|
|
}
|
|
|
|
func (back *RemoteBackend) Peers(ctx context.Context) ([]*p2p.PeerInfo, error) {
|
|
rpcPeers, err := back.remoteEthBackend.Peers(ctx, &emptypb.Empty{})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("ETHBACKENDClient.Peers() error: %w", err)
|
|
}
|
|
|
|
peers := make([]*p2p.PeerInfo, 0, len(rpcPeers.Peers))
|
|
|
|
for _, rpcPeer := range rpcPeers.Peers {
|
|
peer := p2p.PeerInfo{
|
|
ENR: rpcPeer.Enr,
|
|
Enode: rpcPeer.Enode,
|
|
ID: rpcPeer.Id,
|
|
Name: rpcPeer.Name,
|
|
Caps: rpcPeer.Caps,
|
|
Network: struct {
|
|
LocalAddress string `json:"localAddress"`
|
|
RemoteAddress string `json:"remoteAddress"`
|
|
Inbound bool `json:"inbound"`
|
|
Trusted bool `json:"trusted"`
|
|
Static bool `json:"static"`
|
|
}{
|
|
LocalAddress: rpcPeer.ConnLocalAddr,
|
|
RemoteAddress: rpcPeer.ConnRemoteAddr,
|
|
Inbound: rpcPeer.ConnIsInbound,
|
|
Trusted: rpcPeer.ConnIsTrusted,
|
|
Static: rpcPeer.ConnIsStatic,
|
|
},
|
|
Protocols: nil,
|
|
}
|
|
|
|
peers = append(peers, &peer)
|
|
}
|
|
|
|
return peers, nil
|
|
}
|
|
|
|
func (back *RemoteBackend) PendingBlock(ctx context.Context) (*types.Block, error) {
|
|
blockRlp, err := back.remoteEthBackend.PendingBlock(ctx, &emptypb.Empty{})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("ETHBACKENDClient.PendingBlock() error: %w", err)
|
|
}
|
|
if blockRlp == nil {
|
|
return nil, nil
|
|
}
|
|
|
|
var block types.Block
|
|
err = rlp.Decode(bytes.NewReader(blockRlp.BlockRlp), &block)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("decoding block from %x: %w", blockRlp, err)
|
|
}
|
|
|
|
return &block, nil
|
|
}
|