wormhole/node/pkg/terra/watcher.go

304 lines
9.5 KiB
Go

package terra
import (
"context"
"encoding/hex"
"fmt"
"github.com/certusone/wormhole/node/pkg/p2p"
gossipv1 "github.com/certusone/wormhole/node/pkg/proto/gossip/v1"
"github.com/prometheus/client_golang/prometheus/promauto"
"io/ioutil"
"net/http"
"time"
"github.com/prometheus/client_golang/prometheus"
eth_common "github.com/ethereum/go-ethereum/common"
"github.com/certusone/wormhole/node/pkg/common"
"github.com/certusone/wormhole/node/pkg/readiness"
"github.com/certusone/wormhole/node/pkg/supervisor"
"github.com/certusone/wormhole/node/pkg/vaa"
"github.com/gorilla/websocket"
"github.com/tidwall/gjson"
"go.uber.org/zap"
)
type (
// BridgeWatcher is responsible for looking over Terra blockchain and reporting new transactions to the bridge
BridgeWatcher struct {
urlWS string
urlLCD string
bridge string
msgChan chan *common.MessagePublication
setChan chan *common.GuardianSet
}
)
var (
terraConnectionErrors = promauto.NewCounterVec(
prometheus.CounterOpts{
Name: "wormhole_terra_connection_errors_total",
Help: "Total number of Terra connection errors",
}, []string{"reason"})
terraMessagesConfirmed = promauto.NewCounter(
prometheus.CounterOpts{
Name: "wormhole_terra_messages_confirmed_total",
Help: "Total number of verified terra messages found",
})
currentTerraHeight = promauto.NewGauge(
prometheus.GaugeOpts{
Name: "wormhole_terra_current_height",
Help: "Current terra slot height (at default commitment level, not the level used for observations)",
})
queryLatency = promauto.NewHistogramVec(
prometheus.HistogramOpts{
Name: "wormhole_terra_query_latency",
Help: "Latency histogram for terra RPC calls",
}, []string{"operation"})
)
type clientRequest struct {
JSONRPC string `json:"jsonrpc"`
// A String containing the name of the method to be invoked.
Method string `json:"method"`
// Object to pass as request parameter to the method.
Params [1]string `json:"params"`
// The request id. This can be of any type. It is used to match the
// response with the request that it is replying to.
ID uint64 `json:"id"`
}
// NewTerraBridgeWatcher creates a new terra bridge watcher
func NewTerraBridgeWatcher(urlWS string, urlLCD string, bridge string, lockEvents chan *common.MessagePublication, setEvents chan *common.GuardianSet) *BridgeWatcher {
return &BridgeWatcher{urlWS: urlWS, urlLCD: urlLCD, bridge: bridge, msgChan: lockEvents, setChan: setEvents}
}
// Run is the main Terra Bridge run cycle
func (e *BridgeWatcher) Run(ctx context.Context) error {
p2p.DefaultRegistry.SetNetworkStats(vaa.ChainIDTerra, &gossipv1.Heartbeat_Network{
BridgeAddress: e.bridge,
})
errC := make(chan error)
logger := supervisor.Logger(ctx)
logger.Info("connecting to websocket", zap.String("url", e.urlWS))
c, _, err := websocket.DefaultDialer.DialContext(ctx, e.urlWS, nil)
if err != nil {
p2p.DefaultRegistry.AddErrorCount(vaa.ChainIDTerra, 1)
terraConnectionErrors.WithLabelValues("websocket_dial_error").Inc()
return fmt.Errorf("websocket dial failed: %w", err)
}
defer c.Close()
// Subscribe to smart contract transactions
params := [...]string{fmt.Sprintf("tm.event='Tx' AND execute_contract.contract_address='%s'", e.bridge)}
command := &clientRequest{
JSONRPC: "2.0",
Method: "subscribe",
Params: params,
ID: 1,
}
err = c.WriteJSON(command)
if err != nil {
p2p.DefaultRegistry.AddErrorCount(vaa.ChainIDTerra, 1)
terraConnectionErrors.WithLabelValues("websocket_subscription_error").Inc()
return fmt.Errorf("websocket subscription failed: %w", err)
}
// Wait for the success response
_, _, err = c.ReadMessage()
if err != nil {
p2p.DefaultRegistry.AddErrorCount(vaa.ChainIDTerra, 1)
terraConnectionErrors.WithLabelValues("event_subscription_error").Inc()
return fmt.Errorf("event subscription failed: %w", err)
}
logger.Info("subscribed to new transaction events")
readiness.SetReady(common.ReadinessTerraSyncing)
go func() {
t := time.NewTicker(5 * time.Second)
client := &http.Client{
Timeout: time.Second * 5,
}
for {
<-t.C
// Query and report height and set currentTerraHeight
resp, err := client.Get(fmt.Sprintf("%s/blocks/latest", e.urlLCD))
if err != nil {
logger.Error("query latest block response error", zap.Error(err))
continue
}
blocksBody, err := ioutil.ReadAll(resp.Body)
if err != nil {
logger.Error("query guardian set error", zap.Error(err))
errC <- err
resp.Body.Close()
continue
}
resp.Body.Close()
blockJSON := string(blocksBody)
latestBlock := gjson.Get(blockJSON, "block.header.height")
logger.Info("current Terra height", zap.Int64("block", latestBlock.Int()))
currentTerraHeight.Set(float64(latestBlock.Int()))
p2p.DefaultRegistry.SetNetworkStats(vaa.ChainIDTerra, &gossipv1.Heartbeat_Network{
Height: latestBlock.Int(),
BridgeAddress: e.bridge,
})
}
}()
go func() {
defer close(errC)
for {
_, message, err := c.ReadMessage()
if err != nil {
p2p.DefaultRegistry.AddErrorCount(vaa.ChainIDTerra, 1)
terraConnectionErrors.WithLabelValues("channel_read_error").Inc()
logger.Error("error reading channel", zap.Error(err))
errC <- err
return
}
// Received a message from the blockchain
json := string(message)
payload := gjson.Get(json, "result.events.from_contract\\.message\\.message.0")
sender := gjson.Get(json, "result.events.from_contract\\.message\\.sender.0")
chainId := gjson.Get(json, "result.events.from_contract\\.message\\.chain_id.0")
nonce := gjson.Get(json, "result.events.from_contract\\.message\\.nonce.0")
sequence := gjson.Get(json, "result.events.from_contract\\.message\\.sequence.0")
blockTime := gjson.Get(json, "result.events.from_contract\\.message\\.block_time.0")
txHash := gjson.Get(json, "result.events.tx\\.hash.0")
if payload.Exists() && sender.Exists() && chainId.Exists() && nonce.Exists() && sequence.Exists() &&
blockTime.Exists() && txHash.Exists() {
logger.Info("new message detected on terra",
zap.String("chainId", chainId.String()),
zap.String("txHash", txHash.String()),
zap.String("sender", sender.String()),
zap.String("nonce", nonce.String()),
zap.String("sequence", sequence.String()),
zap.String("blockTime", blockTime.String()),
)
senderAddress, err := StringToAddress(sender.String())
if err != nil {
logger.Error("cannot decode emitter hex", zap.String("value", sender.String()))
continue
}
txHashValue, err := StringToHash(txHash.String())
if err != nil {
logger.Error("cannot decode tx hash hex", zap.String("value", txHash.String()))
continue
}
payloadValue, err := hex.DecodeString(payload.String())
if err != nil {
logger.Error("cannot decode payload", zap.String("value", payload.String()))
continue
}
messagePublication := &common.MessagePublication{
TxHash: txHashValue,
Timestamp: time.Unix(blockTime.Int(), 0),
Nonce: uint32(nonce.Uint()),
Sequence: sequence.Uint(),
EmitterChain: vaa.ChainIDTerra,
EmitterAddress: senderAddress,
Payload: payloadValue,
ConsistencyLevel: 0, // Instant finality
}
e.msgChan <- messagePublication
terraMessagesConfirmed.Inc()
}
client := &http.Client{
Timeout: time.Second * 15,
}
// Query and report guardian set status
requestURL := fmt.Sprintf("%s/wasm/contracts/%s/store?query_msg={\"guardian_set_info\":{}}", e.urlLCD, e.bridge)
req, err := http.NewRequestWithContext(ctx, http.MethodGet, requestURL, nil)
if err != nil {
p2p.DefaultRegistry.AddErrorCount(vaa.ChainIDTerra, 1)
terraConnectionErrors.WithLabelValues("guardian_set_req_error").Inc()
logger.Error("query guardian set request error", zap.Error(err))
errC <- err
return
}
msm := time.Now()
resp, err := client.Do(req)
if err != nil {
p2p.DefaultRegistry.AddErrorCount(vaa.ChainIDTerra, 1)
logger.Error("query guardian set response error", zap.Error(err))
errC <- err
return
}
body, err := ioutil.ReadAll(resp.Body)
queryLatency.WithLabelValues("guardian_set_info").Observe(time.Since(msm).Seconds())
if err != nil {
p2p.DefaultRegistry.AddErrorCount(vaa.ChainIDTerra, 1)
logger.Error("query guardian set error", zap.Error(err))
errC <- err
resp.Body.Close()
return
}
json = string(body)
guardianSetIndex := gjson.Get(json, "result.guardian_set_index")
addresses := gjson.Get(json, "result.addresses.#.bytes")
logger.Debug("current guardian set on Terra",
zap.Any("guardianSetIndex", guardianSetIndex),
zap.Any("addresses", addresses))
resp.Body.Close()
// We do not send guardian changes to the processor - ETH guardians are the source of truth.
}
}()
select {
case <-ctx.Done():
err := c.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, ""))
if err != nil {
logger.Error("error on closing socket ", zap.Error(err))
}
return ctx.Err()
case err := <-errC:
return err
}
}
// StringToAddress convert string into address
func StringToAddress(value string) (vaa.Address, error) {
var address vaa.Address
res, err := hex.DecodeString(value)
if err != nil {
return address, err
}
copy(address[:], res)
return address, nil
}
// StringToHash convert string into transaction hash
func StringToHash(value string) (eth_common.Hash, error) {
var hash eth_common.Hash
res, err := hex.DecodeString(value)
if err != nil {
return hash, err
}
copy(hash[:], res)
return hash, nil
}