135 lines
3.5 KiB
Go
135 lines
3.5 KiB
Go
// Copyright (c) 2014-2017 The btcsuite developers
|
|
// Use of this source code is governed by an ISC
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package main
|
|
|
|
import (
|
|
"errors"
|
|
"flag"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"net"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/lbryio/lbcd/rpcclient"
|
|
"github.com/lbryio/lbcd/wire"
|
|
"github.com/lbryio/lbcutil"
|
|
)
|
|
|
|
var (
|
|
coinid = flag.String("coinid", "1425", "Coin ID")
|
|
stratum = flag.String("stratum", "", "Stratum server")
|
|
stratumPass = flag.String("stratumpass", "", "Stratum server password")
|
|
rpcserver = flag.String("rpcserver", "localhost:9245", "LBCD RPC server")
|
|
rpcuser = flag.String("rpcuser", "rpcuser", "LBCD RPC username")
|
|
rpcpass = flag.String("rpcpass", "rpcpass", "LBCD RPC password")
|
|
notls = flag.Bool("notls", false, "Connect to LBCD with TLS disabled")
|
|
run = flag.String("run", "", "Run custom shell command")
|
|
)
|
|
|
|
func onFilteredBlockConnected(height int32, header *wire.BlockHeader, txns []*lbcutil.Tx) {
|
|
|
|
blockHash := header.BlockHash().String()
|
|
|
|
log.Printf("Block connected: %v (%d) %v", blockHash, height, header.Timestamp)
|
|
|
|
if cmd := *run; len(cmd) != 0 {
|
|
cmd = strings.ReplaceAll(cmd, "%s", blockHash)
|
|
err := execCustomCommand(cmd)
|
|
if err != nil {
|
|
log.Printf("ERROR: execCustomCommand: %s", err)
|
|
}
|
|
}
|
|
|
|
if len(*stratum) > 0 && len(*stratumPass) > 0 {
|
|
err := stratumUpdateBlock(*stratum, *stratumPass, *coinid, blockHash)
|
|
if err != nil {
|
|
log.Printf("ERROR: stratumUpdateBlock: %s", err)
|
|
}
|
|
}
|
|
}
|
|
func execCustomCommand(cmd string) error {
|
|
strs := strings.Split(cmd, " ")
|
|
path, err := exec.LookPath(strs[0])
|
|
if errors.Is(err, exec.ErrDot) {
|
|
err = nil
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
c := exec.Command(path, strs[1:]...)
|
|
c.Stdout = os.Stdout
|
|
return c.Run()
|
|
}
|
|
|
|
func stratumUpdateBlock(stratum, stratumPass, coinid, blockHash string) error {
|
|
addr, err := net.ResolveTCPAddr("tcp", stratum)
|
|
if err != nil {
|
|
return fmt.Errorf("can't resolve addr: %w", err)
|
|
}
|
|
|
|
conn, err := net.DialTCP("tcp", nil, addr)
|
|
if err != nil {
|
|
return fmt.Errorf("can't dial tcp: %w", err)
|
|
}
|
|
defer conn.Close()
|
|
|
|
msg := fmt.Sprintf(`{"id":1,"method":"mining.update_block","params":[%q,%s,%q]}`,
|
|
stratumPass, coinid, blockHash)
|
|
|
|
_, err = conn.Write([]byte(msg))
|
|
if err != nil {
|
|
return fmt.Errorf("can't write message: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
func main() {
|
|
|
|
flag.Parse()
|
|
|
|
ntfnHandlers := rpcclient.NotificationHandlers{
|
|
OnFilteredBlockConnected: onFilteredBlockConnected,
|
|
}
|
|
|
|
// Connect to local lbcd RPC server using websockets.
|
|
lbcdHomeDir := lbcutil.AppDataDir("lbcd", false)
|
|
certs, err := ioutil.ReadFile(filepath.Join(lbcdHomeDir, "rpc.cert"))
|
|
if err != nil {
|
|
log.Fatalf("can't read lbcd certificate: %s", err)
|
|
}
|
|
connCfg := &rpcclient.ConnConfig{
|
|
Host: *rpcserver,
|
|
Endpoint: "ws",
|
|
User: *rpcuser,
|
|
Pass: *rpcpass,
|
|
Certificates: certs,
|
|
DisableTLS: *notls,
|
|
}
|
|
client, err := rpcclient.New(connCfg, &ntfnHandlers)
|
|
if err != nil {
|
|
log.Fatalf("can't create rpc client: %s", err)
|
|
}
|
|
|
|
// Register for block connect and disconnect notifications.
|
|
if err = client.NotifyBlocks(); err != nil {
|
|
log.Fatalf("can't register block notification: %s", err)
|
|
}
|
|
log.Printf("NotifyBlocks: Registration Complete")
|
|
|
|
// Get the current block count.
|
|
blockCount, err := client.GetBlockCount()
|
|
if err != nil {
|
|
log.Fatalf("can't get block count: %s", err)
|
|
}
|
|
log.Printf("Block count: %d", blockCount)
|
|
|
|
// Wait until the client either shuts down gracefully (or the user
|
|
// terminates the process with Ctrl+C).
|
|
client.WaitForShutdown()
|
|
}
|