Add websocket handlers to the RPC server for wallet connections.
This change adds additional http listeners for websocket connections on "/wallet". Websockets are used to provide asynchronous messaging between wallet daemons (i.e. btcwallet) and btcd as they allow an easy way for btcd to provide instant notifications (instead of a wallet polling for updates) and multiple replies to a single request. Standard RPC commands sent over a websocket connection are handled just like RPC, returning the same results, the only difference being that the connection is async. In cases where the standard RPC commands fall short of wallet daemons requests, and to request notifications for addresses and events, extension JSON methods are used. Multiple wallets can be connected to the same btcd, and replies to websocket requests and notifications are properly routed back to the original requesting wallet. Due to the nature of turning a synchronous protocol asynchronous, it is highly recommended to use the JSON id field as a type of sequence number, so replies from btcd can be routed back to the proper handler in a wallet daemon.
This commit is contained in:
parent
44aafcf742
commit
a80f9da6c3
2 changed files with 734 additions and 34 deletions
|
@ -627,6 +627,12 @@ func (b *blockManager) handleNotifyMsg(notification *btcchain.Notification) {
|
|||
b.server.txMemPool.removeTransaction(tx)
|
||||
}
|
||||
|
||||
// Notify frontends
|
||||
if r := b.server.rpcServer; r != nil {
|
||||
go r.NotifyBlockConnected(block)
|
||||
go r.NotifyNewTxListeners(b.server.db, block)
|
||||
}
|
||||
|
||||
// A block has been disconnected from the main block chain.
|
||||
case btcchain.NTBlockDisconnected:
|
||||
block, ok := notification.Data.(*btcutil.Block)
|
||||
|
@ -646,6 +652,11 @@ func (b *blockManager) handleNotifyMsg(notification *btcchain.Notification) {
|
|||
b.server.txMemPool.removeTransaction(tx)
|
||||
}
|
||||
}
|
||||
|
||||
// Notify frontends
|
||||
if r := b.server.rpcServer; r != nil {
|
||||
go r.NotifyBlockDisconnected(block)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
|
755
rpcserver.go
755
rpcserver.go
|
@ -5,10 +5,16 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"code.google.com/p/go.crypto/ripemd160"
|
||||
"code.google.com/p/go.net/websocket"
|
||||
"encoding/base64"
|
||||
"encoding/hex"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"github.com/conformal/btcchain"
|
||||
"github.com/conformal/btcdb"
|
||||
"github.com/conformal/btcjson"
|
||||
"github.com/conformal/btcscript"
|
||||
"github.com/conformal/btcutil"
|
||||
|
@ -23,17 +29,63 @@ import (
|
|||
"sync/atomic"
|
||||
)
|
||||
|
||||
// Errors
|
||||
var (
|
||||
ErrBadParamsField = errors.New("bad params field")
|
||||
)
|
||||
|
||||
// rpcServer holds the items the rpc server may need to access (config,
|
||||
// shutdown, main server, etc.)
|
||||
type rpcServer struct {
|
||||
started int32
|
||||
shutdown int32
|
||||
server *server
|
||||
ws wsContext
|
||||
wg sync.WaitGroup
|
||||
rpcport string
|
||||
username string
|
||||
password string
|
||||
listeners []net.Listener
|
||||
quit chan int
|
||||
}
|
||||
|
||||
// wsContext holds the items the RPC server needs to handle websocket
|
||||
// connections for wallets.
|
||||
type wsContext struct {
|
||||
// txRequests maps between a 160-byte pubkey hash and slice of contexts
|
||||
// to route replies back to the original requesting wallets.
|
||||
txRequests struct {
|
||||
sync.RWMutex
|
||||
m map[addressHash][]requesterContext
|
||||
}
|
||||
|
||||
// spentRequests maps between the Outpoint of an unspent transaction
|
||||
// output and a slice of contexts to route notifications back to the
|
||||
// original requesting wallets.
|
||||
spentRequests struct {
|
||||
sync.RWMutex
|
||||
m map[btcwire.OutPoint][]requesterContext
|
||||
}
|
||||
|
||||
// Channel to add a wallet listener.
|
||||
addWalletListener chan (chan []byte)
|
||||
|
||||
// Channel to removes a wallet listener.
|
||||
removeWalletListener chan (chan []byte)
|
||||
|
||||
// Any chain notifications meant to be received by every connected
|
||||
// wallet are sent across this channel.
|
||||
walletNotificationMaster chan []byte
|
||||
}
|
||||
|
||||
type addressHash [ripemd160.Size]byte
|
||||
|
||||
// requesterContext holds a slice of reply channels for wallets
|
||||
// requesting information about some address, and the id of the original
|
||||
// request so notifications can be routed back to the appropiate handler.
|
||||
type requesterContext struct {
|
||||
c chan *btcjson.Reply
|
||||
id interface{}
|
||||
}
|
||||
|
||||
// Start is used by server.go to start the rpc listener.
|
||||
|
@ -54,6 +106,10 @@ func (s *rpcServer) Start() {
|
|||
jsonAuthFail(w, r, s)
|
||||
}
|
||||
})
|
||||
go s.walletListenerDuplicator()
|
||||
http.Handle("/wallet", websocket.Handler(func(ws *websocket.Conn) {
|
||||
s.walletReqsNotifications(ws)
|
||||
}))
|
||||
httpServer := &http.Server{}
|
||||
for _, listener := range s.listeners {
|
||||
s.wg.Add(1)
|
||||
|
@ -82,6 +138,7 @@ func (s *rpcServer) Stop() error {
|
|||
}
|
||||
log.Infof("RPCS: RPC server shutdown complete")
|
||||
s.wg.Wait()
|
||||
close(s.quit)
|
||||
return nil
|
||||
}
|
||||
|
||||
|
@ -95,6 +152,13 @@ func newRPCServer(s *server) (*rpcServer, error) {
|
|||
rpc.username = cfg.RPCUser
|
||||
rpc.password = cfg.RPCPass
|
||||
|
||||
// initialize memory for websocket connections
|
||||
rpc.ws.txRequests.m = make(map[addressHash][]requesterContext)
|
||||
rpc.ws.spentRequests.m = make(map[btcwire.OutPoint][]requesterContext)
|
||||
rpc.ws.addWalletListener = make(chan (chan []byte))
|
||||
rpc.ws.removeWalletListener = make(chan (chan []byte))
|
||||
rpc.ws.walletNotificationMaster = make(chan []byte)
|
||||
|
||||
// IPv4 listener.
|
||||
var listeners []net.Listener
|
||||
listenAddr4 := net.JoinHostPort("127.0.0.1", rpc.rpcport)
|
||||
|
@ -115,6 +179,7 @@ func newRPCServer(s *server) (*rpcServer, error) {
|
|||
listeners = append(listeners, listener6)
|
||||
|
||||
rpc.listeners = listeners
|
||||
|
||||
return &rpc, err
|
||||
}
|
||||
|
||||
|
@ -123,46 +188,74 @@ func jsonAuthFail(w http.ResponseWriter, r *http.Request, s *rpcServer) {
|
|||
fmt.Fprint(w, "401 Unauthorized.\n")
|
||||
}
|
||||
|
||||
// jsonRPCRead is the main function that handles reading messages, getting
|
||||
// the data the message requests, and writing the reply.
|
||||
// jsonRPCRead is the RPC wrapper around the jsonRead function to handles
|
||||
// reading and responding to RPC messages.
|
||||
func jsonRPCRead(w http.ResponseWriter, r *http.Request, s *rpcServer) {
|
||||
_ = spew.Dump
|
||||
r.Close = true
|
||||
if atomic.LoadInt32(&s.shutdown) != 0 {
|
||||
return
|
||||
}
|
||||
var rawReply btcjson.Reply
|
||||
body, err := btcjson.GetRaw(r.Body)
|
||||
spew.Dump(body)
|
||||
if err != nil {
|
||||
log.Errorf("RPCS: Error getting json message: %v", err)
|
||||
return
|
||||
}
|
||||
var message btcjson.Message
|
||||
err = json.Unmarshal(body, &message)
|
||||
if err != nil {
|
||||
log.Errorf("RPCS: Error unmarshalling json message: %v", err)
|
||||
jsonError := btcjson.Error{
|
||||
Code: -32700,
|
||||
Message: "Parse error",
|
||||
}
|
||||
|
||||
rawReply = btcjson.Reply{
|
||||
Result: nil,
|
||||
Error: &jsonError,
|
||||
Id: nil,
|
||||
replychan := make(chan *btcjson.Reply)
|
||||
if err = jsonRead(replychan, body, s); err != nil {
|
||||
log.Error(err)
|
||||
return
|
||||
}
|
||||
log.Tracef("RPCS: reply: %v", rawReply)
|
||||
msg, err := btcjson.MarshallAndSend(rawReply, w)
|
||||
reply := <-replychan
|
||||
|
||||
if reply != nil {
|
||||
log.Tracef("[RPCS] reply: %v", *reply)
|
||||
msg, err := btcjson.MarshallAndSend(*reply, w)
|
||||
if err != nil {
|
||||
log.Errorf(msg)
|
||||
return
|
||||
}
|
||||
log.Debugf(msg)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
// jsonRead abstracts the JSON unmarshalling and reply handling,
|
||||
// returning replies across a channel. A channel is used as some websocket
|
||||
// method extensions require multiple replies.
|
||||
func jsonRead(replychan chan *btcjson.Reply, body []byte, s *rpcServer) (err error) {
|
||||
var message btcjson.Message
|
||||
err = json.Unmarshal(body, &message)
|
||||
if err != nil {
|
||||
jsonError := btcjson.Error{
|
||||
Code: -32700,
|
||||
Message: "Parse error",
|
||||
}
|
||||
|
||||
reply := btcjson.Reply{
|
||||
Result: nil,
|
||||
Error: &jsonError,
|
||||
Id: nil,
|
||||
}
|
||||
|
||||
log.Tracef("RPCS: reply: %v", reply)
|
||||
|
||||
replychan <- &reply
|
||||
return fmt.Errorf("RPCS: Error unmarshalling json message: %v", err)
|
||||
}
|
||||
log.Tracef("RPCS: received: %v", message)
|
||||
|
||||
var rawReply btcjson.Reply
|
||||
requester := false
|
||||
|
||||
defer func() {
|
||||
replychan <- &rawReply
|
||||
if !requester {
|
||||
close(replychan)
|
||||
}
|
||||
}()
|
||||
|
||||
// Deal with commands
|
||||
switch message.Method {
|
||||
case "stop":
|
||||
|
@ -382,8 +475,7 @@ func jsonRPCRead(w http.ResponseWriter, r *http.Request, s *rpcServer) {
|
|||
|
||||
_, maxidx, err := s.server.db.NewestSha()
|
||||
if err != nil {
|
||||
log.Errorf("RPCS: Cannot get newest sha: %v", err)
|
||||
return
|
||||
return fmt.Errorf("RPCS: Cannot get newest sha: %v", err)
|
||||
}
|
||||
|
||||
blockHeader := &blk.MsgBlock().Header
|
||||
|
@ -439,10 +531,7 @@ func jsonRPCRead(w http.ResponseWriter, r *http.Request, s *rpcServer) {
|
|||
}
|
||||
}
|
||||
|
||||
if int(verbose) != 1 {
|
||||
// Don't return details
|
||||
// not used yet
|
||||
} else {
|
||||
if int(verbose) != 0 {
|
||||
txSha, _ := btcwire.NewShaHashFromStr(tx)
|
||||
var txS *btcwire.MsgTx
|
||||
txList, err := s.server.db.FetchTxBySha(txSha)
|
||||
|
@ -516,8 +605,7 @@ func jsonRPCRead(w http.ResponseWriter, r *http.Request, s *rpcServer) {
|
|||
|
||||
_, maxidx, err := s.server.db.NewestSha()
|
||||
if err != nil {
|
||||
log.Errorf("RPCS: Cannot get newest sha: %v", err)
|
||||
return
|
||||
return fmt.Errorf("RPCS: Cannot get newest sha: %v", err)
|
||||
}
|
||||
confirmations := uint64(1 + maxidx - idx)
|
||||
|
||||
|
@ -540,6 +628,9 @@ func jsonRPCRead(w http.ResponseWriter, r *http.Request, s *rpcServer) {
|
|||
Error: nil,
|
||||
Id: &message.Id,
|
||||
}
|
||||
} else {
|
||||
// Don't return details
|
||||
// not used yet
|
||||
}
|
||||
case "decoderawtransaction":
|
||||
var f interface{}
|
||||
|
@ -562,6 +653,295 @@ func jsonRPCRead(w http.ResponseWriter, r *http.Request, s *rpcServer) {
|
|||
Error: nil,
|
||||
Id: &message.Id,
|
||||
}
|
||||
case "sendrawtransaction":
|
||||
params, ok := message.Params.([]interface{})
|
||||
if !ok || len(params) != 1 {
|
||||
jsonError := btcjson.Error{
|
||||
Code: -32602,
|
||||
Message: "Invalid parameters",
|
||||
}
|
||||
rawReply = btcjson.Reply{
|
||||
Result: nil,
|
||||
Error: &jsonError,
|
||||
Id: &message.Id,
|
||||
}
|
||||
return ErrBadParamsField
|
||||
}
|
||||
serializedtxhex, ok := params[0].(string)
|
||||
if !ok {
|
||||
jsonError := btcjson.Error{
|
||||
Code: -32602,
|
||||
Message: "Raw tx is not a string",
|
||||
}
|
||||
rawReply = btcjson.Reply{
|
||||
Result: nil,
|
||||
Error: &jsonError,
|
||||
Id: &message.Id,
|
||||
}
|
||||
return ErrBadParamsField
|
||||
}
|
||||
|
||||
// Deserialize and send off to tx relay
|
||||
serializedtx, err := hex.DecodeString(serializedtxhex)
|
||||
if err != nil {
|
||||
jsonError := btcjson.Error{
|
||||
Code: -22,
|
||||
Message: "Unable to decode hex string",
|
||||
}
|
||||
rawReply = btcjson.Reply{
|
||||
Result: nil,
|
||||
Error: &jsonError,
|
||||
Id: &message.Id,
|
||||
}
|
||||
return err
|
||||
}
|
||||
msgtx := btcwire.NewMsgTx()
|
||||
err = msgtx.Deserialize(bytes.NewBuffer(serializedtx))
|
||||
if err != nil {
|
||||
jsonError := btcjson.Error{
|
||||
Code: -22,
|
||||
Message: "Unable to deserialize raw tx",
|
||||
}
|
||||
rawReply = btcjson.Reply{
|
||||
Result: nil,
|
||||
Error: &jsonError,
|
||||
Id: &message.Id,
|
||||
}
|
||||
return err
|
||||
}
|
||||
err = s.server.txMemPool.ProcessTransaction(msgtx)
|
||||
if err != nil {
|
||||
log.Errorf("RPCS: Failed to process transaction: %v", err)
|
||||
jsonError := btcjson.Error{
|
||||
Code: -22,
|
||||
Message: "Failed to process transaction",
|
||||
}
|
||||
rawReply = btcjson.Reply{
|
||||
Result: nil,
|
||||
Error: &jsonError,
|
||||
Id: &message.Id,
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
var result interface{}
|
||||
txsha, err := msgtx.TxSha()
|
||||
if err == nil {
|
||||
result = txsha.String()
|
||||
}
|
||||
rawReply = btcjson.Reply{
|
||||
Result: result,
|
||||
Error: nil,
|
||||
Id: &message.Id,
|
||||
}
|
||||
|
||||
// Extensions
|
||||
case "getcurrentnet":
|
||||
var net btcwire.BitcoinNet
|
||||
if cfg.TestNet3 {
|
||||
net = btcwire.TestNet3
|
||||
} else {
|
||||
net = btcwire.MainNet
|
||||
}
|
||||
rawReply = btcjson.Reply{
|
||||
Result: float64(net),
|
||||
Id: &message.Id,
|
||||
}
|
||||
|
||||
case "rescan":
|
||||
var addr string
|
||||
minblock, maxblock := int64(0), btcdb.AllShas
|
||||
params, ok := message.Params.([]interface{})
|
||||
if !ok {
|
||||
return ErrBadParamsField
|
||||
}
|
||||
for i, v := range params {
|
||||
switch v.(type) {
|
||||
case string:
|
||||
if i == 0 {
|
||||
addr = v.(string)
|
||||
}
|
||||
case float64:
|
||||
if i == 1 {
|
||||
minblock = int64(v.(float64))
|
||||
} else if i == 2 {
|
||||
maxblock = int64(v.(float64))
|
||||
}
|
||||
}
|
||||
}
|
||||
addrhash, _, err := btcutil.DecodeAddress(addr)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// FetchHeightRange may not return a complete list of block shas for
|
||||
// the given range, so fetch range as many times as necessary.
|
||||
for {
|
||||
blkshalist, err := s.server.db.FetchHeightRange(minblock, maxblock)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if len(blkshalist) == 0 {
|
||||
break
|
||||
}
|
||||
|
||||
for i := range blkshalist {
|
||||
blk, err := s.server.db.FetchBlockBySha(&blkshalist[i])
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
txShaList, err := blk.TxShas()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
txList := s.server.db.FetchTxByShaList(txShaList)
|
||||
for j := range txList {
|
||||
for _, txout := range txList[j].Tx.TxOut {
|
||||
_, txaddrhash, err := btcscript.ScriptToAddrHash(txout.PkScript)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if !bytes.Equal(addrhash, txaddrhash) {
|
||||
reply := btcjson.Reply{
|
||||
Result: txList[j].Sha,
|
||||
Error: nil,
|
||||
Id: &message.Id,
|
||||
}
|
||||
replychan <- &reply
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if maxblock-minblock > int64(len(blkshalist)) {
|
||||
minblock += int64(len(blkshalist))
|
||||
} else {
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
rawReply = btcjson.Reply{
|
||||
Result: nil,
|
||||
Error: nil,
|
||||
Id: &message.Id,
|
||||
}
|
||||
|
||||
case "notifynewtxs":
|
||||
params, ok := message.Params.([]interface{})
|
||||
if !ok || len(params) != 1 {
|
||||
jsonError := btcjson.Error{
|
||||
Code: -32602,
|
||||
Message: "Invalid parameters",
|
||||
}
|
||||
rawReply = btcjson.Reply{
|
||||
Result: nil,
|
||||
Error: &jsonError,
|
||||
Id: &message.Id,
|
||||
}
|
||||
return ErrBadParamsField
|
||||
}
|
||||
addr, ok := params[0].(string)
|
||||
if !ok {
|
||||
jsonError := btcjson.Error{
|
||||
Code: -32602,
|
||||
Message: "Invalid parameters",
|
||||
}
|
||||
rawReply = btcjson.Reply{
|
||||
Result: nil,
|
||||
Error: &jsonError,
|
||||
Id: &message.Id,
|
||||
}
|
||||
return ErrBadParamsField
|
||||
}
|
||||
addrhash, _, err := btcutil.DecodeAddress(addr)
|
||||
if err != nil {
|
||||
jsonError := btcjson.Error{
|
||||
Code: -32602,
|
||||
Message: "Cannot decode address",
|
||||
}
|
||||
rawReply = btcjson.Reply{
|
||||
Result: nil,
|
||||
Error: &jsonError,
|
||||
Id: &message.Id,
|
||||
}
|
||||
return ErrBadParamsField
|
||||
}
|
||||
var hash addressHash
|
||||
copy(hash[:], addrhash)
|
||||
s.ws.txRequests.Lock()
|
||||
cxts := s.ws.txRequests.m[hash]
|
||||
cxt := requesterContext{
|
||||
c: replychan,
|
||||
id: message.Id,
|
||||
}
|
||||
s.ws.txRequests.m[hash] = append(cxts, cxt)
|
||||
s.ws.txRequests.Unlock()
|
||||
|
||||
rawReply = btcjson.Reply{
|
||||
Result: nil,
|
||||
Error: nil,
|
||||
Id: &message.Id,
|
||||
}
|
||||
requester = true
|
||||
|
||||
case "notifyspent":
|
||||
params, ok := message.Params.([]interface{})
|
||||
if !ok || len(params) != 2 {
|
||||
jsonError := btcjson.Error{
|
||||
Code: -32602,
|
||||
Message: "Invalid parameters",
|
||||
}
|
||||
rawReply = btcjson.Reply{
|
||||
Result: nil,
|
||||
Error: &jsonError,
|
||||
Id: &message.Id,
|
||||
}
|
||||
return ErrBadParamsField
|
||||
}
|
||||
hashBE, ok1 := params[0].(string)
|
||||
index, ok2 := params[1].(float64)
|
||||
if !ok1 || !ok2 {
|
||||
jsonError := btcjson.Error{
|
||||
Code: -32602,
|
||||
Message: "Invalid parameters",
|
||||
}
|
||||
rawReply = btcjson.Reply{
|
||||
Result: nil,
|
||||
Error: &jsonError,
|
||||
Id: &message.Id,
|
||||
}
|
||||
return ErrBadParamsField
|
||||
}
|
||||
s.ws.spentRequests.Lock()
|
||||
hash, err := btcwire.NewShaHashFromStr(hashBE)
|
||||
if err != nil {
|
||||
jsonError := btcjson.Error{
|
||||
Code: -32602,
|
||||
Message: "Hash string cannot be parsed.",
|
||||
}
|
||||
rawReply = btcjson.Reply{
|
||||
Result: nil,
|
||||
Error: &jsonError,
|
||||
Id: &message.Id,
|
||||
}
|
||||
return ErrBadParamsField
|
||||
}
|
||||
op := btcwire.NewOutPoint(hash, uint32(index))
|
||||
cxts := s.ws.spentRequests.m[*op]
|
||||
cxt := requesterContext{
|
||||
c: replychan,
|
||||
id: message.Id,
|
||||
}
|
||||
s.ws.spentRequests.m[*op] = append(cxts, cxt)
|
||||
s.ws.spentRequests.Unlock()
|
||||
|
||||
rawReply = btcjson.Reply{
|
||||
Result: nil,
|
||||
Error: nil,
|
||||
Id: &message.Id,
|
||||
}
|
||||
requester = true
|
||||
|
||||
default:
|
||||
jsonError := btcjson.Error{
|
||||
Code: -32601,
|
||||
|
@ -574,13 +954,7 @@ func jsonRPCRead(w http.ResponseWriter, r *http.Request, s *rpcServer) {
|
|||
}
|
||||
}
|
||||
|
||||
msg, err := btcjson.MarshallAndSend(rawReply, w)
|
||||
if err != nil {
|
||||
log.Errorf(msg)
|
||||
return
|
||||
}
|
||||
log.Debugf(msg)
|
||||
return
|
||||
return nil
|
||||
}
|
||||
|
||||
// getDifficultyRatio returns the proof-of-work difficulty as a multiple of the
|
||||
|
@ -602,3 +976,318 @@ func getDifficultyRatio(bits uint32) float64 {
|
|||
}
|
||||
return diff
|
||||
}
|
||||
|
||||
// AddWalletListener adds a channel to listen for new messages from a
|
||||
// wallet.
|
||||
func (s *rpcServer) AddWalletListener(c chan []byte) {
|
||||
s.ws.addWalletListener <- c
|
||||
}
|
||||
|
||||
// RemoveWalletListener removes a wallet listener channel.
|
||||
func (s *rpcServer) RemoveWalletListener(c chan []byte) {
|
||||
s.ws.removeWalletListener <- c
|
||||
}
|
||||
|
||||
// walletListenerDuplicator listens for new wallet listener channels
|
||||
// and duplicates messages sent to walletNotificationMaster to all
|
||||
// connected listeners.
|
||||
func (s *rpcServer) walletListenerDuplicator() {
|
||||
// walletListeners is a map holding each currently connected wallet
|
||||
// listener as the key. The value is ignored, as this is only used as
|
||||
// a set.
|
||||
walletListeners := make(map[chan []byte]bool)
|
||||
|
||||
// Don't want to add or delete a wallet listener while iterating
|
||||
// through each to propigate to every attached wallet. Use a mutex to
|
||||
// prevent this.
|
||||
var mtx sync.Mutex
|
||||
|
||||
// Check for listener channels to add or remove from set.
|
||||
go func() {
|
||||
for {
|
||||
select {
|
||||
case c := <-s.ws.addWalletListener:
|
||||
mtx.Lock()
|
||||
walletListeners[c] = true
|
||||
mtx.Unlock()
|
||||
|
||||
case c := <-s.ws.removeWalletListener:
|
||||
mtx.Lock()
|
||||
delete(walletListeners, c)
|
||||
mtx.Unlock()
|
||||
|
||||
case <-s.quit:
|
||||
return
|
||||
}
|
||||
}
|
||||
}()
|
||||
|
||||
// Duplicate all messages sent across walletNotificationMaster to each
|
||||
// listening wallet.
|
||||
for {
|
||||
select {
|
||||
case ntfn := <-s.ws.walletNotificationMaster:
|
||||
mtx.Lock()
|
||||
for c := range walletListeners {
|
||||
c <- ntfn
|
||||
}
|
||||
mtx.Unlock()
|
||||
|
||||
case <-s.quit:
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// walletReqsNotifications is the handler function for websocket
|
||||
// connections from a btcwallet instance. It reads messages from wallet and
|
||||
// sends back replies, as well as notififying wallets of chain updates.
|
||||
func (s *rpcServer) walletReqsNotifications(ws *websocket.Conn) {
|
||||
// Add wallet notification channel so this handler receives btcd chain
|
||||
// notifications.
|
||||
c := make(chan []byte)
|
||||
s.AddWalletListener(c)
|
||||
defer s.RemoveWalletListener(c)
|
||||
|
||||
// msgs is a channel for all messages received over the websocket.
|
||||
msgs := make(chan []byte)
|
||||
|
||||
// Receive messages from websocket and send across reqs until the
|
||||
// connection is lost.
|
||||
go func() {
|
||||
for {
|
||||
select {
|
||||
case <-s.quit:
|
||||
close(msgs)
|
||||
return
|
||||
default:
|
||||
var m []byte
|
||||
if err := websocket.Message.Receive(ws, &m); err != nil {
|
||||
close(msgs)
|
||||
return
|
||||
}
|
||||
msgs <- m
|
||||
}
|
||||
}
|
||||
}()
|
||||
|
||||
for {
|
||||
select {
|
||||
case m, ok := <-msgs:
|
||||
if !ok {
|
||||
// Wallet disconnected.
|
||||
return
|
||||
}
|
||||
// Handle request here.
|
||||
go s.websocketJSONHandler(c, m)
|
||||
case ntfn, _ := <-c:
|
||||
// Send btcd notification to btcwallet instance over
|
||||
// websocket.
|
||||
if err := websocket.Message.Send(ws, ntfn); err != nil {
|
||||
// Wallet disconnected.
|
||||
return
|
||||
}
|
||||
case <-s.quit:
|
||||
// Server closed.
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// websocketJSONHandler parses and handles a marshalled json message,
|
||||
// sending the marshalled reply to a wallet notification channel.
|
||||
func (s *rpcServer) websocketJSONHandler(walletNotification chan []byte, msg []byte) {
|
||||
replychan := make(chan *btcjson.Reply)
|
||||
|
||||
go func() {
|
||||
for {
|
||||
select {
|
||||
case reply, ok := <-replychan:
|
||||
if !ok {
|
||||
// jsonRead() function called below has finished.
|
||||
return
|
||||
}
|
||||
if reply == nil {
|
||||
continue
|
||||
}
|
||||
log.Tracef("[RPCS] reply: %v", *reply)
|
||||
replyBytes, err := json.Marshal(reply)
|
||||
if err != nil {
|
||||
log.Errorf("[RPCS] Error Marshalling reply: %v", err)
|
||||
return
|
||||
}
|
||||
walletNotification <- replyBytes
|
||||
|
||||
case <-s.quit:
|
||||
return
|
||||
}
|
||||
}
|
||||
}()
|
||||
|
||||
s.wg.Add(1)
|
||||
err := jsonRead(replychan, msg, s)
|
||||
s.wg.Done()
|
||||
if err != nil {
|
||||
log.Error(err)
|
||||
}
|
||||
}
|
||||
|
||||
// NotifyBlockConnected creates and marshalls a JSON message to notify
|
||||
// of a new block connected to the main chain. The notification is sent
|
||||
// to each connected wallet.
|
||||
func (s *rpcServer) NotifyBlockConnected(block *btcutil.Block) {
|
||||
var id interface{} = "btcd:blockconnected"
|
||||
hash, err := block.Sha()
|
||||
if err != nil {
|
||||
log.Error("Bad block; connected block notification dropped.")
|
||||
return
|
||||
}
|
||||
ntfn := btcjson.Reply{
|
||||
Result: struct {
|
||||
Hash string `json:"hash"`
|
||||
Height int64 `json:"height"`
|
||||
}{
|
||||
Hash: hash.String(),
|
||||
Height: block.Height(),
|
||||
},
|
||||
Id: &id,
|
||||
}
|
||||
m, _ := json.Marshal(ntfn)
|
||||
s.ws.walletNotificationMaster <- m
|
||||
}
|
||||
|
||||
// NotifyBlockDisconnected creates and marshalls a JSON message to notify
|
||||
// of a new block disconnected from the main chain. The notification is sent
|
||||
// to each connected wallet.
|
||||
func (s *rpcServer) NotifyBlockDisconnected(block *btcutil.Block) {
|
||||
var id interface{} = "btcd:blockdisconnected"
|
||||
hash, err := block.Sha()
|
||||
if err != nil {
|
||||
log.Error("Bad block; connected block notification dropped.")
|
||||
return
|
||||
}
|
||||
ntfn := btcjson.Reply{
|
||||
Result: struct {
|
||||
Hash string `json:"hash"`
|
||||
Height int64 `json:"height"`
|
||||
}{
|
||||
Hash: hash.String(),
|
||||
Height: block.Height(),
|
||||
},
|
||||
Id: &id,
|
||||
}
|
||||
m, _ := json.Marshal(ntfn)
|
||||
s.ws.walletNotificationMaster <- m
|
||||
}
|
||||
|
||||
// NotifyNewTxListeners creates and marshals a JSON message to notify wallets
|
||||
// of new transactions (with both spent and unspent outputs) for a watched
|
||||
// address.
|
||||
func (s *rpcServer) NotifyNewTxListeners(db btcdb.Db, block *btcutil.Block) {
|
||||
txShaList, err := block.TxShas()
|
||||
if err != nil {
|
||||
log.Error("Bad block; All notifications for block dropped.")
|
||||
return
|
||||
}
|
||||
txList := db.FetchTxByShaList(txShaList)
|
||||
for _, tx := range txList {
|
||||
go s.newBlockNotifyCheckTxIn(tx.Tx.TxIn)
|
||||
go s.newBlockNotifyCheckTxOut(db, block, tx)
|
||||
}
|
||||
}
|
||||
|
||||
// newBlockNotifyCheckTxIn is a helper function to iterate through
|
||||
// each transaction input of a new block and perform any checks and
|
||||
// notify listening frontends when necessary.
|
||||
func (s *rpcServer) newBlockNotifyCheckTxIn(txins []*btcwire.TxIn) {
|
||||
for _, txin := range txins {
|
||||
s.ws.spentRequests.RLock()
|
||||
for out, cxts := range s.ws.spentRequests.m {
|
||||
if txin.PreviousOutpoint != out {
|
||||
continue
|
||||
}
|
||||
|
||||
reply := &btcjson.Reply{
|
||||
Result: struct {
|
||||
TxHash string `json:"txhash"`
|
||||
Index uint32 `json:"index"`
|
||||
}{
|
||||
TxHash: out.Hash.String(),
|
||||
Index: uint32(out.Index),
|
||||
},
|
||||
Error: nil,
|
||||
// Id is set for each requester separately below.
|
||||
}
|
||||
for _, cxt := range cxts {
|
||||
reply.Id = &cxt.id
|
||||
cxt.c <- reply
|
||||
}
|
||||
|
||||
s.ws.spentRequests.RUnlock()
|
||||
s.ws.spentRequests.Lock()
|
||||
delete(s.ws.spentRequests.m, out)
|
||||
s.ws.spentRequests.Unlock()
|
||||
s.ws.spentRequests.RLock()
|
||||
}
|
||||
s.ws.spentRequests.RUnlock()
|
||||
}
|
||||
}
|
||||
|
||||
// newBlockNotifyCheckTxOut is a helper function to iterate through
|
||||
// each transaction output of a new block and perform any checks and
|
||||
// notify listening frontends when necessary.
|
||||
func (s *rpcServer) newBlockNotifyCheckTxOut(db btcdb.Db, block *btcutil.Block, tx *btcdb.TxListReply) {
|
||||
for i, txout := range tx.Tx.TxOut {
|
||||
_, txaddrhash, err := btcscript.ScriptToAddrHash(txout.PkScript)
|
||||
if err != nil {
|
||||
log.Error("Error getting payment address from tx; dropping any Tx notifications.")
|
||||
break
|
||||
}
|
||||
s.ws.txRequests.RLock()
|
||||
for addr, cxts := range s.ws.txRequests.m {
|
||||
if !bytes.Equal(addr[:], txaddrhash) {
|
||||
continue
|
||||
}
|
||||
blkhash, err := block.Sha()
|
||||
if err != nil {
|
||||
log.Error("Error getting block sha; dropping Tx notification.")
|
||||
break
|
||||
}
|
||||
txaddr, err := btcutil.EncodeAddress(txaddrhash, s.server.btcnet)
|
||||
if err != nil {
|
||||
log.Error("Error encoding address; dropping Tx notification.")
|
||||
break
|
||||
}
|
||||
reply := &btcjson.Reply{
|
||||
Result: struct {
|
||||
Sender string `json:"sender"`
|
||||
Receiver string `json:"receiver"`
|
||||
BlockHash string `json:"blockhash"`
|
||||
Height int64 `json:"height"`
|
||||
TxHash string `json:"txhash"`
|
||||
Index uint32 `json:"index"`
|
||||
Amount int64 `json:"amount"`
|
||||
PkScript string `json:"pkscript"`
|
||||
Spent bool `json:"spent"`
|
||||
}{
|
||||
Sender: "Unknown", // TODO(jrick)
|
||||
Receiver: txaddr,
|
||||
BlockHash: blkhash.String(),
|
||||
Height: block.Height(),
|
||||
TxHash: tx.Sha.String(),
|
||||
Index: uint32(i),
|
||||
Amount: txout.Value,
|
||||
PkScript: btcutil.Base58Encode(txout.PkScript),
|
||||
Spent: tx.TxSpent[i],
|
||||
},
|
||||
Error: nil,
|
||||
// Id is set for each requester separately below.
|
||||
}
|
||||
for _, cxt := range cxts {
|
||||
reply.Id = &cxt.id
|
||||
cxt.c <- reply
|
||||
}
|
||||
}
|
||||
s.ws.txRequests.RUnlock()
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue