This commit is contained in:
Martin Holst Swende 2017-12-20 12:59:25 +00:00 committed by GitHub
commit abdd443f63
12 changed files with 221 additions and 23 deletions

View file

@ -20,6 +20,7 @@ import (
"encoding/json" "encoding/json"
"fmt" "fmt"
"io" "io"
"bytes"
) )
// The ABI holds information about a contract's context and available // The ABI holds information about a contract's context and available
@ -137,3 +138,14 @@ func (abi *ABI) UnmarshalJSON(data []byte) error {
return nil return nil
} }
// methodById looks up a method by the 4-byte id
// returns nil if none found
func (abi *ABI) MethodById(sigdata []byte) *Method {
for _, method := range abi.Methods{
if bytes.Equal(method.Id(), sigdata) {
return &method
}
}
return nil
}

View file

@ -75,7 +75,7 @@ func (e Event) tupleUnpack(v interface{}, output []byte) error {
// need to move this up because they read sequentially // need to move this up because they read sequentially
j += input.Type.Size j += input.Type.Size
} }
marshalledValue, err := toGoType((i+j)*32, input.Type, output) marshalledValue, err := ToGoType((i+j)*32, input.Type, output)
if err != nil { if err != nil {
return err return err
} }
@ -126,7 +126,7 @@ func (e Event) singleUnpack(v interface{}, output []byte) error {
value := valueOf.Elem() value := valueOf.Elem()
marshalledValue, err := toGoType(0, e.Inputs[0].Type, output) marshalledValue, err := ToGoType(0, e.Inputs[0].Type, output)
if err != nil { if err != nil {
return err return err
} }

View file

@ -99,7 +99,7 @@ func (method Method) tupleUnpack(v interface{}, output []byte) error {
// need to move this up because they read sequentially // need to move this up because they read sequentially
j += toUnpack.Type.Size j += toUnpack.Type.Size
} }
marshalledValue, err := toGoType((i+j)*32, toUnpack.Type, output) marshalledValue, err := ToGoType((i+j)*32, toUnpack.Type, output)
if err != nil { if err != nil {
return err return err
} }
@ -146,7 +146,7 @@ func (method Method) singleUnpack(v interface{}, output []byte) error {
value := valueOf.Elem() value := valueOf.Elem()
marshalledValue, err := toGoType(0, method.Outputs[0].Type, output) marshalledValue, err := ToGoType(0, method.Outputs[0].Type, output)
if err != nil { if err != nil {
return err return err
} }

View file

@ -127,7 +127,7 @@ func forEachUnpack(t Type, output []byte, start, size int) (interface{}, error)
if t.Elem.T == ArrayTy && j != 0 { if t.Elem.T == ArrayTy && j != 0 {
i = start + t.Elem.Size*32*j i = start + t.Elem.Size*32*j
} }
inter, err := toGoType(i, *t.Elem, output) inter, err := ToGoType(i, *t.Elem, output)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -139,9 +139,9 @@ func forEachUnpack(t Type, output []byte, start, size int) (interface{}, error)
return refSlice.Interface(), nil return refSlice.Interface(), nil
} }
// toGoType parses the output bytes and recursively assigns the value of these bytes // ToGoType parses the output bytes and recursively assigns the value of these bytes
// into a go type with accordance with the ABI spec. // into a go type with accordance with the ABI spec.
func toGoType(index int, t Type, output []byte) (interface{}, error) { func ToGoType(index int, t Type, output []byte) (interface{}, error) {
if index+32 > len(output) { if index+32 > len(output) {
return nil, fmt.Errorf("abi: cannot marshal in to go type: length insufficient %d require %d", len(output), index+32) return nil, fmt.Errorf("abi: cannot marshal in to go type: length insufficient %d require %d", len(output), index+32)
} }

View file

@ -127,7 +127,7 @@ func (hub *Hub) refreshWallets() {
// breaking the Ledger protocol if that is waiting for user confirmation. This // breaking the Ledger protocol if that is waiting for user confirmation. This
// is a bug acknowledged at Ledger, but it won't be fixed on old devices so we // is a bug acknowledged at Ledger, but it won't be fixed on old devices so we
// need to prevent concurrent comms ourselves. The more elegant solution would // need to prevent concurrent comms ourselves. The more elegant solution would
// be to ditch enumeration in favor of hutplug events, but that don't work yet // be to ditch enumeration in favor of hotplug events, but that don't work yet
// on Windows so if we need to hack it anyway, this is more elegant for now. // on Windows so if we need to hack it anyway, this is more elegant for now.
hub.commsLock.Lock() hub.commsLock.Lock()
if hub.commsPend > 0 { // A confirmation is pending, don't refresh if hub.commsPend > 0 { // A confirmation is pending, don't refresh

View file

@ -99,7 +99,7 @@ type wallet struct {
// //
// As such, a hardware wallet needs two locks to function correctly. A state // As such, a hardware wallet needs two locks to function correctly. A state
// lock can be used to protect the wallet's software-side internal state, which // lock can be used to protect the wallet's software-side internal state, which
// must not be held exlusively during hardware communication. A communication // must not be held exclusively during hardware communication. A communication
// lock can be used to achieve exclusive access to the device itself, this one // lock can be used to achieve exclusive access to the device itself, this one
// however should allow "skipping" waiting for operations that might want to // however should allow "skipping" waiting for operations that might want to
// use the device, but can live without too (e.g. account self-derivation). // use the device, but can live without too (e.g. account self-derivation).

View file

@ -23,6 +23,7 @@ import (
"math/rand" "math/rand"
"reflect" "reflect"
"encoding/json"
"github.com/ethereum/go-ethereum/common/hexutil" "github.com/ethereum/go-ethereum/common/hexutil"
"github.com/ethereum/go-ethereum/crypto/sha3" "github.com/ethereum/go-ethereum/crypto/sha3"
) )
@ -240,3 +241,56 @@ func (a *UnprefixedAddress) UnmarshalText(input []byte) error {
func (a UnprefixedAddress) MarshalText() ([]byte, error) { func (a UnprefixedAddress) MarshalText() ([]byte, error) {
return []byte(hex.EncodeToString(a[:])), nil return []byte(hex.EncodeToString(a[:])), nil
} }
// MixedcaseAddress retains the original string, which may or may not be
// correctly checksummed
//
// TODO! Should we really keep both addr and original, or _only_ original, and
// always calculate addr on the fly? That would reduce the possibilities for errors
// if some caller modifies the original at some point. NB: If we do so, we should still
// parse the addr in UnmarshalJSON to ensure that the format is correct, e.g. correct size and
// hex-encoded and such
type MixedcaseAddress struct {
addr Address
original string
}
// NewMixedcaseAddress constructor (mainly for testing)
func NewMixedcaseAddress(addr Address) MixedcaseAddress {
return MixedcaseAddress{addr: addr, original: addr.Hex()}
}
// UnmarshalJSON parses MixedcaseAddress
func (ma *MixedcaseAddress) UnmarshalJSON(input []byte) error {
if err := hexutil.UnmarshalFixedJSON(addressT, input, ma.addr[:]); err != nil {
return err
}
return json.Unmarshal(input, &ma.original)
}
// MarshalJSON marshals the original value
func (ma *MixedcaseAddress) MarshalJSON() ([]byte, error) {
return json.Marshal(ma.original)
}
// Address returns the address
func (ma *MixedcaseAddress) Address() Address {
return ma.addr
}
// String implements fmt.Stringer
func (ma *MixedcaseAddress) String() string {
if ma.ValidChecksum() {
return fmt.Sprintf("%s [chksum ok]", ma.original)
}
return fmt.Sprintf("%s [chksum INVALID]", ma.original)
}
// ValidChecksum returns true if the address has valid checksum
func (ma *MixedcaseAddress) ValidChecksum() bool {
return ma.original == ma.addr.Hex()
}
// Original returns the mixed-case input string
func (ma *MixedcaseAddress) Original() string {
return ma.original
}

View file

@ -18,6 +18,7 @@ package common
import ( import (
"encoding/json" "encoding/json"
"math/big" "math/big"
"strings" "strings"
"testing" "testing"
@ -149,3 +150,46 @@ func BenchmarkAddressHex(b *testing.B) {
testAddr.Hex() testAddr.Hex()
} }
} }
func TestMixedcaseAccount_Address(t *testing.T) {
// https://github.com/ethereum/EIPs/blob/master/EIPS/eip-55.md
// Note: 0X{checksum_addr} is not valid according to spec above
var res []struct {
A MixedcaseAddress
Valid bool
}
if err := json.Unmarshal([]byte(`[
{"A" : "0xae967917c465db8578ca9024c205720b1a3651A9", "Valid": false},
{"A" : "0xAe967917c465db8578ca9024c205720b1a3651A9", "Valid": true},
{"A" : "0XAe967917c465db8578ca9024c205720b1a3651A9", "Valid": false},
{"A" : "0x1111111111111111111112222222222223333323", "Valid": true}
]`), &res); err != nil {
t.Fatal(err)
}
for _, r := range res {
if got := r.A.ValidChecksum(); got != r.Valid {
t.Errorf("Expected checksum %v, got checksum %v, input %v", r.Valid, got, r.A.String())
}
}
//These should throw exceptions:
var r2 []MixedcaseAddress
for _, r := range []string{
`["0x11111111111111111111122222222222233333"]`, // Too short
`["0x111111111111111111111222222222222333332"]`, // Too short
`["0x11111111111111111111122222222222233333234"]`, // Too long
`["0x111111111111111111111222222222222333332344"]`, // Too long
`["1111111111111111111112222222222223333323"]`, // Missing 0x
`["x1111111111111111111112222222222223333323"]`, // Missing 0
`["0xG111111111111111111112222222222223333323"]`, //Non-hex
} {
if err := json.Unmarshal([]byte(r), &r2); err == nil {
t.Errorf("Expected failure, input %v", r)
}
}
}

View file

@ -33,6 +33,7 @@ import (
"time" "time"
"github.com/ethereum/go-ethereum/log" "github.com/ethereum/go-ethereum/log"
"os"
) )
var ( var (
@ -171,6 +172,8 @@ func DialContext(ctx context.Context, rawurl string) (*Client, error) {
return DialHTTP(rawurl) return DialHTTP(rawurl)
case "ws", "wss": case "ws", "wss":
return DialWebsocket(ctx, rawurl, "") return DialWebsocket(ctx, rawurl, "")
case "stdio":
return DialStdIO(ctx)
case "": case "":
return DialIPC(ctx, rawurl) return DialIPC(ctx, rawurl)
default: default:
@ -178,13 +181,51 @@ func DialContext(ctx context.Context, rawurl string) (*Client, error) {
} }
} }
type StdIOConn struct{}
func (io StdIOConn) Read(b []byte) (n int, err error) {
return os.Stdin.Read(b)
}
func (io StdIOConn) Write(b []byte) (n int, err error) {
return os.Stdout.Write(b)
}
func (io StdIOConn) Close() error {
return nil
}
func (io StdIOConn) LocalAddr() net.Addr {
return &net.UnixAddr{Name: "stdio", Net: "stdio"}
}
func (io StdIOConn) RemoteAddr() net.Addr {
return &net.UnixAddr{Name: "stdio", Net: "stdio"}
}
func (io StdIOConn) SetDeadline(t time.Time) error {
return &net.OpError{Op: "set", Net: "stdio", Source: nil, Addr: nil, Err: errors.New("deadline not supported")}
}
func (io StdIOConn) SetReadDeadline(t time.Time) error {
return &net.OpError{Op: "set", Net: "stdio", Source: nil, Addr: nil, Err: errors.New("deadline not supported")}
}
func (io StdIOConn) SetWriteDeadline(t time.Time) error {
return &net.OpError{Op: "set", Net: "stdio", Source: nil, Addr: nil, Err: errors.New("deadline not supported")}
}
func DialStdIO(ctx context.Context) (*Client, error) {
return newClient(ctx, func(_ context.Context) (net.Conn, error) {
return StdIOConn{}, nil
})
}
func newClient(initctx context.Context, connectFunc func(context.Context) (net.Conn, error)) (*Client, error) { func newClient(initctx context.Context, connectFunc func(context.Context) (net.Conn, error)) (*Client, error) {
conn, err := connectFunc(initctx) conn, err := connectFunc(initctx)
if err != nil { if err != nil {
return nil, err return nil, err
} }
_, isHTTP := conn.(*httpConn) _, isHTTP := conn.(*httpConn)
c := &Client{ c := &Client{
writeConn: conn, writeConn: conn,
isHTTP: isHTTP, isHTTP: isHTTP,
@ -524,13 +565,13 @@ func (c *Client) dispatch(conn net.Conn) {
} }
case err := <-c.readErr: case err := <-c.readErr:
log.Debug(fmt.Sprintf("<-readErr: %v", err)) log.Debug("<-readErr", "err", err)
c.closeRequestOps(err) c.closeRequestOps(err)
conn.Close() conn.Close()
reading = false reading = false
case newconn := <-c.reconnected: case newconn := <-c.reconnected:
log.Debug(fmt.Sprintf("<-reconnected: (reading=%t) %v", reading, conn.RemoteAddr())) log.Debug("<-reconnected", "reading", reading, "remote", conn.RemoteAddr())
if reading { if reading {
// Wait for the previous read loop to exit. This is a rare case. // Wait for the previous read loop to exit. This is a rare case.
conn.Close() conn.Close()
@ -587,7 +628,7 @@ func (c *Client) closeRequestOps(err error) {
func (c *Client) handleNotification(msg *jsonrpcMessage) { func (c *Client) handleNotification(msg *jsonrpcMessage) {
if !strings.HasSuffix(msg.Method, notificationMethodSuffix) { if !strings.HasSuffix(msg.Method, notificationMethodSuffix) {
log.Debug(fmt.Sprint("dropping non-subscription message: ", msg)) log.Debug("dropping non-subscription message", "msg", msg)
return return
} }
var subResult struct { var subResult struct {
@ -595,7 +636,7 @@ func (c *Client) handleNotification(msg *jsonrpcMessage) {
Result json.RawMessage `json:"result"` Result json.RawMessage `json:"result"`
} }
if err := json.Unmarshal(msg.Params, &subResult); err != nil { if err := json.Unmarshal(msg.Params, &subResult); err != nil {
log.Debug(fmt.Sprint("dropping invalid subscription message: ", msg)) log.Debug("dropping invalid subscription message", "msg", msg)
return return
} }
if c.subs[subResult.ID] != nil { if c.subs[subResult.ID] != nil {
@ -606,7 +647,7 @@ func (c *Client) handleNotification(msg *jsonrpcMessage) {
func (c *Client) handleResponse(msg *jsonrpcMessage) { func (c *Client) handleResponse(msg *jsonrpcMessage) {
op := c.respWait[string(msg.ID)] op := c.respWait[string(msg.ID)]
if op == nil { if op == nil {
log.Debug(fmt.Sprintf("unsolicited response %v", msg)) log.Debug("unsolicited response", "msg", msg)
return return
} }
delete(c.respWait, string(msg.ID)) delete(c.respWait, string(msg.ID))

View file

@ -159,11 +159,16 @@ func (srv *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
// All checks passed, create a codec that reads direct from the request body // All checks passed, create a codec that reads direct from the request body
// untilEOF and writes the response to w and order the server to process a // untilEOF and writes the response to w and order the server to process a
// single request. // single request.
ctx := context.Background()
ctx = context.WithValue(ctx, "remote", r.RemoteAddr)
ctx = context.WithValue(ctx, "scheme", r.Proto)
ctx = context.WithValue(ctx, "local", r.Host)
codec := NewJSONCodec(&httpReadWriteNopCloser{r.Body, w}) codec := NewJSONCodec(&httpReadWriteNopCloser{r.Body, w})
defer codec.Close() defer codec.Close()
w.Header().Set("content-type", contentType) w.Header().Set("content-type", contentType)
srv.ServeSingleRequest(codec, OptionMethodInvocation) srv.ServeSingleRequest(codec, OptionMethodInvocation, ctx)
} }
// validateRequest returns a non-zero response code and error message if the // validateRequest returns a non-zero response code and error message if the

View file

@ -27,6 +27,7 @@ import (
"github.com/ethereum/go-ethereum/log" "github.com/ethereum/go-ethereum/log"
"gopkg.in/fatih/set.v0" "gopkg.in/fatih/set.v0"
"time"
) )
const MetadataApi = "rpc" const MetadataApi = "rpc"
@ -42,6 +43,19 @@ const (
OptionSubscriptions = 1 << iota // support pub sub OptionSubscriptions = 1 << iota // support pub sub
) )
// RPCLogger is an interface for logging rpc requests and responses
type RPCLogger interface {
Store(record *RPCInvocationRecord)
}
// RPCInvocationRecord captures an rpc request and response, for audit logging
type RPCInvocationRecord struct {
Method string
Args []string
Time time.Time
Response string
}
// NewServer will create a new server instance with no registered handlers. // NewServer will create a new server instance with no registered handlers.
func NewServer() *Server { func NewServer() *Server {
server := &Server{ server := &Server{
@ -58,6 +72,10 @@ func NewServer() *Server {
return server return server
} }
func (s *Server) SetAuditLogger(logger RPCLogger) {
s.auditlog = &logger
}
// RPCService gives meta information about the server. // RPCService gives meta information about the server.
// e.g. gives information about the loaded modules. // e.g. gives information about the loaded modules.
type RPCService struct { type RPCService struct {
@ -125,7 +143,7 @@ func (s *Server) RegisterName(name string, rcvr interface{}) error {
// If singleShot is true it will process a single request, otherwise it will handle // If singleShot is true it will process a single request, otherwise it will handle
// requests until the codec returns an error when reading a request (in most cases // requests until the codec returns an error when reading a request (in most cases
// an EOF). It executes requests in parallel when singleShot is false. // an EOF). It executes requests in parallel when singleShot is false.
func (s *Server) serveRequest(codec ServerCodec, singleShot bool, options CodecOption) error { func (s *Server) serveRequest(codec ServerCodec, singleShot bool, options CodecOption, ctx context.Context) error {
var pend sync.WaitGroup var pend sync.WaitGroup
defer func() { defer func() {
@ -140,7 +158,8 @@ func (s *Server) serveRequest(codec ServerCodec, singleShot bool, options CodecO
s.codecsMu.Unlock() s.codecsMu.Unlock()
}() }()
ctx, cancel := context.WithCancel(context.Background()) // ctx, cancel := context.WithCancel(context.Background())
ctx, cancel := context.WithCancel(ctx)
defer cancel() defer cancel()
// if the codec supports notification include a notifier that callbacks can use // if the codec supports notification include a notifier that callbacks can use
@ -215,14 +234,14 @@ func (s *Server) serveRequest(codec ServerCodec, singleShot bool, options CodecO
// stopped. In either case the codec is closed. // stopped. In either case the codec is closed.
func (s *Server) ServeCodec(codec ServerCodec, options CodecOption) { func (s *Server) ServeCodec(codec ServerCodec, options CodecOption) {
defer codec.Close() defer codec.Close()
s.serveRequest(codec, false, options) s.serveRequest(codec, false, options, context.Background())
} }
// ServeSingleRequest reads and processes a single RPC request from the given codec. It will not // ServeSingleRequest reads and processes a single RPC request from the given codec. It will not
// close the codec unless a non-recoverable error has occurred. Note, this method will return after // close the codec unless a non-recoverable error has occurred. Note, this method will return after
// a single request has been processed! // a single request has been processed!
func (s *Server) ServeSingleRequest(codec ServerCodec, options CodecOption) { func (s *Server) ServeSingleRequest(codec ServerCodec, options CodecOption, ctx context.Context) {
s.serveRequest(codec, true, options) s.serveRequest(codec, true, options, ctx)
} }
// Stop will stop reading new requests, wait for stopPendingRequestTimeout to allow pending requests to finish, // Stop will stop reading new requests, wait for stopPendingRequestTimeout to allow pending requests to finish,
@ -253,6 +272,20 @@ func (s *Server) createSubscription(ctx context.Context, c ServerCodec, req *ser
return reply[0].Interface().(*Subscription).ID, nil return reply[0].Interface().(*Subscription).ID, nil
} }
// vstring 'resolves' values into string-values
func vstring(v reflect.Value) string {
// Resolve pointers (optional args)
if v.Kind() == reflect.Ptr && !v.IsNil() {
v = v.Elem()
}
if v.CanInterface() {
if str, ok := v.Interface().(fmt.Stringer); ok {
return str.String()
}
}
return v.String()
}
// handle executes a request and returns the response from the callback. // handle executes a request and returns the response from the callback.
func (s *Server) handle(ctx context.Context, codec ServerCodec, req *serverRequest) (interface{}, func()) { func (s *Server) handle(ctx context.Context, codec ServerCodec, req *serverRequest) (interface{}, func()) {
if req.err != nil { if req.err != nil {
@ -312,14 +345,22 @@ func (s *Server) handle(ctx context.Context, codec ServerCodec, req *serverReque
if len(reply) == 0 { if len(reply) == 0 {
return codec.CreateResponse(req.id, nil), nil return codec.CreateResponse(req.id, nil), nil
} }
record := RPCInvocationRecord{Method: req.callb.method.Name}
if s.auditlog != nil {
for _, a := range req.args {
record.Args = append(record.Args, vstring(a))
}
defer (*s.auditlog).Store(&record)
}
if req.callb.errPos >= 0 { // test if method returned an error if req.callb.errPos >= 0 { // test if method returned an error
if !reply[req.callb.errPos].IsNil() { if !reply[req.callb.errPos].IsNil() {
e := reply[req.callb.errPos].Interface().(error) e := reply[req.callb.errPos].Interface().(error)
res := codec.CreateErrorResponse(&req.id, &callbackError{e.Error()}) record.Response = e.Error()
return res, nil return codec.CreateErrorResponse(&req.id, &callbackError{e.Error()}), nil
} }
} }
record.Response = vstring(reply[0])
return codec.CreateResponse(req.id, reply[0].Interface()), nil return codec.CreateResponse(req.id, reply[0].Interface()), nil
} }

View file

@ -74,6 +74,7 @@ type Server struct {
run int32 run int32
codecsMu sync.Mutex codecsMu sync.Mutex
codecs *set.Set codecs *set.Set
auditlog *RPCLogger
} }
// rpcRequest represents a raw incoming RPC request // rpcRequest represents a raw incoming RPC request