WIP RPC interface
This commit is contained in:
		
							
								
								
									
										20
									
								
								rpc/json.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										20
									
								
								rpc/json.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,20 @@ | ||||
| package rpc | ||||
|  | ||||
| import ( | ||||
| 	"encoding/json" | ||||
| 	"io" | ||||
| ) | ||||
|  | ||||
| type jsonWrapper struct{} | ||||
|  | ||||
| func (self jsonWrapper) Send(writer io.Writer, v interface{}) (n int, err error) { | ||||
| 	var payload []byte | ||||
| 	payload, err = json.Marshal(v) | ||||
| 	if err != nil { | ||||
| 		return 0, err | ||||
| 	} | ||||
|  | ||||
| 	return writer.Write(payload) | ||||
| } | ||||
|  | ||||
| var JSON jsonWrapper | ||||
							
								
								
									
										14
									
								
								rpc/message.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								rpc/message.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,14 @@ | ||||
| package rpc | ||||
|  | ||||
| import "github.com/ethereum/eth-go/ethutil" | ||||
|  | ||||
| type Message struct { | ||||
| 	Call string        `json:"call"` | ||||
| 	Args []interface{} `json:"args"` | ||||
| 	Id   int           `json:"_id"` | ||||
| 	Data interface{}   `json:"data"` | ||||
| } | ||||
|  | ||||
| func (self *Message) Arguments() *ethutil.Value { | ||||
| 	return ethutil.NewValue(self.Args) | ||||
| } | ||||
							
								
								
									
										311
									
								
								rpc/packages.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										311
									
								
								rpc/packages.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,311 @@ | ||||
| package rpc | ||||
|  | ||||
| import ( | ||||
| 	"encoding/json" | ||||
| 	"errors" | ||||
| 	"math/big" | ||||
| 	"strings" | ||||
|  | ||||
| 	"github.com/ethereum/eth-go/ethpipe" | ||||
| 	"github.com/ethereum/eth-go/ethutil" | ||||
| ) | ||||
|  | ||||
| type EthereumApi struct { | ||||
| 	pipe *ethpipe.JSPipe | ||||
| } | ||||
|  | ||||
| type JsonArgs interface { | ||||
| 	requirements() error | ||||
| } | ||||
|  | ||||
| type BlockResponse struct { | ||||
| 	JsonResponse | ||||
| } | ||||
| type GetBlockArgs struct { | ||||
| 	BlockNumber int | ||||
| 	Hash        string | ||||
| } | ||||
|  | ||||
| type ErrorResponse struct { | ||||
| 	Error     bool   `json:"error"` | ||||
| 	ErrorText string `json:"errorText"` | ||||
| } | ||||
|  | ||||
| type JsonResponse interface { | ||||
| } | ||||
|  | ||||
| type SuccessRes struct { | ||||
| 	Error  bool         `json:"error"` | ||||
| 	Result JsonResponse `json:"result"` | ||||
| } | ||||
|  | ||||
| func NewSuccessRes(object JsonResponse) string { | ||||
| 	e := SuccessRes{Error: false, Result: object} | ||||
| 	res, err := json.Marshal(e) | ||||
| 	if err != nil { | ||||
| 		// This should never happen | ||||
| 		panic("Creating json error response failed, help") | ||||
| 	} | ||||
| 	success := string(res) | ||||
| 	return success | ||||
| } | ||||
|  | ||||
| func NewErrorResponse(msg string) error { | ||||
| 	e := ErrorResponse{Error: true, ErrorText: msg} | ||||
| 	res, err := json.Marshal(e) | ||||
| 	if err != nil { | ||||
| 		// This should never happen | ||||
| 		panic("Creating json error response failed, help") | ||||
| 	} | ||||
| 	newErr := errors.New(string(res)) | ||||
| 	return newErr | ||||
| } | ||||
|  | ||||
| func (b *GetBlockArgs) requirements() error { | ||||
| 	if b.BlockNumber == 0 && b.Hash == "" { | ||||
| 		return NewErrorResponse("GetBlock requires either a block 'number' or a block 'hash' as argument") | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (p *EthereumApi) GetBlock(args *GetBlockArgs, reply *string) error { | ||||
| 	err := args.requirements() | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	block := p.pipe.BlockByHash(args.Hash) | ||||
| 	*reply = NewSuccessRes(block) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| type NewTxArgs struct { | ||||
| 	Sec       string | ||||
| 	Recipient string | ||||
| 	Value     string | ||||
| 	Gas       string | ||||
| 	GasPrice  string | ||||
| 	Init      string | ||||
| 	Body      string | ||||
| } | ||||
| type TxResponse struct { | ||||
| 	Hash string | ||||
| } | ||||
|  | ||||
| func (a *NewTxArgs) requirements() error { | ||||
| 	if a.Recipient == "" { | ||||
| 		return NewErrorResponse("Transact requires a 'recipient' address as argument") | ||||
| 	} | ||||
| 	if a.Value == "" { | ||||
| 		return NewErrorResponse("Transact requires a 'value' as argument") | ||||
| 	} | ||||
| 	if a.Gas == "" { | ||||
| 		return NewErrorResponse("Transact requires a 'gas' value as argument") | ||||
| 	} | ||||
| 	if a.GasPrice == "" { | ||||
| 		return NewErrorResponse("Transact requires a 'gasprice' value as argument") | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (a *NewTxArgs) requirementsContract() error { | ||||
| 	if a.Value == "" { | ||||
| 		return NewErrorResponse("Create requires a 'value' as argument") | ||||
| 	} | ||||
| 	if a.Gas == "" { | ||||
| 		return NewErrorResponse("Create requires a 'gas' value as argument") | ||||
| 	} | ||||
| 	if a.GasPrice == "" { | ||||
| 		return NewErrorResponse("Create requires a 'gasprice' value as argument") | ||||
| 	} | ||||
| 	if a.Body == "" { | ||||
| 		return NewErrorResponse("Create requires a 'body' value as argument") | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (p *EthereumApi) Transact(args *NewTxArgs, reply *string) error { | ||||
| 	err := args.requirements() | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	result, _ := p.pipe.Transact(p.pipe.Key().PrivateKey, args.Recipient, args.Value, args.Gas, args.GasPrice, args.Body) | ||||
| 	*reply = NewSuccessRes(result) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (p *EthereumApi) Create(args *NewTxArgs, reply *string) error { | ||||
| 	err := args.requirementsContract() | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	result, _ := p.pipe.Transact(p.pipe.Key().PrivateKey, "", args.Value, args.Gas, args.GasPrice, args.Body) | ||||
| 	*reply = NewSuccessRes(result) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| type PushTxArgs struct { | ||||
| 	Tx string | ||||
| } | ||||
|  | ||||
| func (a *PushTxArgs) requirementsPushTx() error { | ||||
| 	if a.Tx == "" { | ||||
| 		return NewErrorResponse("PushTx requires a 'tx' as argument") | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (p *EthereumApi) PushTx(args *PushTxArgs, reply *string) error { | ||||
| 	err := args.requirementsPushTx() | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	result, _ := p.pipe.PushTx(args.Tx) | ||||
| 	*reply = NewSuccessRes(result) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (p *EthereumApi) GetKey(args interface{}, reply *string) error { | ||||
| 	*reply = NewSuccessRes(p.pipe.Key()) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| type GetStorageArgs struct { | ||||
| 	Address string | ||||
| 	Key     string | ||||
| } | ||||
|  | ||||
| func (a *GetStorageArgs) requirements() error { | ||||
| 	if a.Address == "" { | ||||
| 		return NewErrorResponse("GetStorageAt requires an 'address' value as argument") | ||||
| 	} | ||||
| 	if a.Key == "" { | ||||
| 		return NewErrorResponse("GetStorageAt requires an 'key' value as argument") | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| type GetStorageAtRes struct { | ||||
| 	Key     string `json:"key"` | ||||
| 	Value   string `json:"value"` | ||||
| 	Address string `json:"address"` | ||||
| } | ||||
|  | ||||
| func (p *EthereumApi) GetStorageAt(args *GetStorageArgs, reply *string) error { | ||||
| 	err := args.requirements() | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	state := p.pipe.World().SafeGet(ethutil.Hex2Bytes(args.Address)) | ||||
|  | ||||
| 	var hx string | ||||
| 	if strings.Index(args.Key, "0x") == 0 { | ||||
| 		hx = string([]byte(args.Key)[2:]) | ||||
| 	} else { | ||||
| 		// Convert the incoming string (which is a bigint) into hex | ||||
| 		i, _ := new(big.Int).SetString(args.Key, 10) | ||||
| 		hx = ethutil.Bytes2Hex(i.Bytes()) | ||||
| 	} | ||||
| 	logger.Debugf("GetStorageAt(%s, %s)\n", args.Address, hx) | ||||
| 	value := state.Storage(ethutil.Hex2Bytes(hx)) | ||||
| 	*reply = NewSuccessRes(GetStorageAtRes{Address: args.Address, Key: args.Key, Value: value.Str()}) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| type GetTxCountArgs struct { | ||||
| 	Address string `json:"address"` | ||||
| } | ||||
| type GetTxCountRes struct { | ||||
| 	Nonce int `json:"nonce"` | ||||
| } | ||||
|  | ||||
| func (a *GetTxCountArgs) requirements() error { | ||||
| 	if a.Address == "" { | ||||
| 		return NewErrorResponse("GetTxCountAt requires an 'address' value as argument") | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| type GetPeerCountRes struct { | ||||
| 	PeerCount int `json:"peerCount"` | ||||
| } | ||||
|  | ||||
| func (p *EthereumApi) GetPeerCount(args *interface{}, reply *string) error { | ||||
| 	*reply = NewSuccessRes(GetPeerCountRes{PeerCount: p.pipe.PeerCount()}) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| type GetListeningRes struct { | ||||
| 	IsListening bool `json:"isListening"` | ||||
| } | ||||
|  | ||||
| func (p *EthereumApi) GetIsListening(args *interface{}, reply *string) error { | ||||
| 	*reply = NewSuccessRes(GetListeningRes{IsListening: p.pipe.IsListening()}) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| type GetCoinbaseRes struct { | ||||
| 	Coinbase string `json:"coinbase"` | ||||
| } | ||||
|  | ||||
| func (p *EthereumApi) GetCoinbase(args *interface{}, reply *string) error { | ||||
| 	*reply = NewSuccessRes(GetCoinbaseRes{Coinbase: p.pipe.CoinBase()}) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| type GetMiningRes struct { | ||||
| 	IsMining bool `json:"isMining"` | ||||
| } | ||||
|  | ||||
| func (p *EthereumApi) GetIsMining(args *interface{}, reply *string) error { | ||||
| 	*reply = NewSuccessRes(GetMiningRes{IsMining: p.pipe.IsMining()}) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (p *EthereumApi) GetTxCountAt(args *GetTxCountArgs, reply *string) error { | ||||
| 	err := args.requirements() | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	state := p.pipe.TxCountAt(args.Address) | ||||
| 	*reply = NewSuccessRes(GetTxCountRes{Nonce: state}) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| type GetBalanceArgs struct { | ||||
| 	Address string | ||||
| } | ||||
|  | ||||
| func (a *GetBalanceArgs) requirements() error { | ||||
| 	if a.Address == "" { | ||||
| 		return NewErrorResponse("GetBalanceAt requires an 'address' value as argument") | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| type BalanceRes struct { | ||||
| 	Balance string `json:"balance"` | ||||
| 	Address string `json:"address"` | ||||
| } | ||||
|  | ||||
| func (p *EthereumApi) GetBalanceAt(args *GetBalanceArgs, reply *string) error { | ||||
| 	err := args.requirements() | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	state := p.pipe.World().SafeGet(ethutil.Hex2Bytes(args.Address)) | ||||
| 	*reply = NewSuccessRes(BalanceRes{Balance: state.Balance.String(), Address: args.Address}) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| type TestRes struct { | ||||
| 	JsonResponse `json:"-"` | ||||
| 	Answer       int `json:"answer"` | ||||
| } | ||||
|  | ||||
| func (p *EthereumApi) Test(args *GetBlockArgs, reply *string) error { | ||||
| 	*reply = NewSuccessRes(TestRes{Answer: 15}) | ||||
| 	return nil | ||||
| } | ||||
							
								
								
									
										67
									
								
								rpc/server.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										67
									
								
								rpc/server.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,67 @@ | ||||
| package rpc | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"net" | ||||
| 	"net/rpc" | ||||
| 	"net/rpc/jsonrpc" | ||||
|  | ||||
| 	"github.com/ethereum/eth-go/ethlog" | ||||
| 	"github.com/ethereum/eth-go/ethpipe" | ||||
| ) | ||||
|  | ||||
| var logger = ethlog.NewLogger("JSON") | ||||
|  | ||||
| type JsonRpcServer struct { | ||||
| 	quit     chan bool | ||||
| 	listener net.Listener | ||||
| 	pipe     *ethpipe.JSPipe | ||||
| } | ||||
|  | ||||
| func (s *JsonRpcServer) exitHandler() { | ||||
| out: | ||||
| 	for { | ||||
| 		select { | ||||
| 		case <-s.quit: | ||||
| 			s.listener.Close() | ||||
| 			break out | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	logger.Infoln("Shutdown JSON-RPC server") | ||||
| } | ||||
|  | ||||
| func (s *JsonRpcServer) Stop() { | ||||
| 	close(s.quit) | ||||
| } | ||||
|  | ||||
| func (s *JsonRpcServer) Start() { | ||||
| 	logger.Infoln("Starting JSON-RPC server") | ||||
| 	go s.exitHandler() | ||||
| 	rpc.Register(&EthereumApi{pipe: s.pipe}) | ||||
| 	rpc.HandleHTTP() | ||||
|  | ||||
| 	for { | ||||
| 		conn, err := s.listener.Accept() | ||||
| 		if err != nil { | ||||
| 			logger.Infoln("Error starting JSON-RPC:", err) | ||||
| 			break | ||||
| 		} | ||||
| 		logger.Debugln("Incoming request.") | ||||
| 		go jsonrpc.ServeConn(conn) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func NewJsonRpcServer(pipe *ethpipe.JSPipe, port int) (*JsonRpcServer, error) { | ||||
| 	sport := fmt.Sprintf(":%d", port) | ||||
| 	l, err := net.Listen("tcp", sport) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
|  | ||||
| 	return &JsonRpcServer{ | ||||
| 		listener: l, | ||||
| 		quit:     make(chan bool), | ||||
| 		pipe:     pipe, | ||||
| 	}, nil | ||||
| } | ||||
							
								
								
									
										75
									
								
								rpc/writer.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										75
									
								
								rpc/writer.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,75 @@ | ||||
| package rpc | ||||
|  | ||||
| /* | ||||
| func pack(id int, v ...interface{}) Message { | ||||
| 	return Message{Data: v, Id: id} | ||||
| } | ||||
|  | ||||
| func WriteOn(msg *Message, writer io.Writer) { | ||||
| 	//msg := &Message{Seed: seed, Data: data} | ||||
|  | ||||
| 	switch msg.Call { | ||||
| 	case "compile": | ||||
| 		data := ethutil.NewValue(msg.Args) | ||||
| 		bcode, err := ethutil.Compile(data.Get(0).Str(), false) | ||||
| 		if err != nil { | ||||
| 			JSON.Send(writer, pack(msg.Id, err.Error())) | ||||
| 		} | ||||
|  | ||||
| 		code := ethutil.Bytes2Hex(bcode) | ||||
|  | ||||
| 		JSON.Send(writer, pack(msg.Id, code, nil)) | ||||
| 	case "block": | ||||
| 		args := msg.Arguments() | ||||
|  | ||||
| 		block := pipe.BlockByNumber(int32(args.Get(0).Uint())) | ||||
|  | ||||
| 		JSON.Send(writer, pack(msg.Id, block)) | ||||
| 	case "transact": | ||||
| 		if mp, ok := msg.Args[0].(map[string]interface{}); ok { | ||||
| 			object := mapToTxParams(mp) | ||||
| 			JSON.Send( | ||||
| 				writer, | ||||
| 				pack(msg.Id, args(pipe.Transact(object["from"], object["to"], object["value"], object["gas"], object["gasPrice"], object["data"]))), | ||||
| 			) | ||||
|  | ||||
| 		} | ||||
| 	case "coinbase": | ||||
| 		JSON.Send(writer, pack(msg.Id, pipe.CoinBase(), msg.Seed)) | ||||
|  | ||||
| 	case "listening": | ||||
| 		JSON.Send(writer, pack(msg.Id, pipe.IsListening())) | ||||
|  | ||||
| 	case "mining": | ||||
| 		JSON.Send(writer, pack(msg.Id, pipe.IsMining())) | ||||
|  | ||||
| 	case "peerCoint": | ||||
| 		JSON.Send(writer, pack(msg.Id, pipe.PeerCount())) | ||||
|  | ||||
| 	case "countAt": | ||||
| 		args := msg.Arguments() | ||||
|  | ||||
| 		JSON.Send(writer, pack(msg.Id, pipe.TxCountAt(args.Get(0).Str()))) | ||||
|  | ||||
| 	case "codeAt": | ||||
| 		args := msg.Arguments() | ||||
|  | ||||
| 		JSON.Send(writer, pack(msg.Id, len(pipe.CodeAt(args.Get(0).Str())))) | ||||
|  | ||||
| 	case "stateAt": | ||||
| 		args := msg.Arguments() | ||||
|  | ||||
| 		JSON.Send(writer, pack(msg.Id, pipe.StorageAt(args.Get(0).Str(), args.Get(1).Str()))) | ||||
|  | ||||
| 	case "balanceAt": | ||||
| 		args := msg.Arguments() | ||||
|  | ||||
| 		JSON.Send(writer, pack(msg.Id, pipe.BalanceAt(args.Get(0).Str()))) | ||||
|  | ||||
| 	case "newFilter": | ||||
| 	case "newFilterString": | ||||
| 	case "messages": | ||||
| 		// TODO | ||||
| 	} | ||||
| } | ||||
| */ | ||||
		Reference in New Issue
	
	Block a user