|
|
@@ -1,6 +1,9 @@
|
|
|
package rpc
|
|
|
|
|
|
import (
|
|
|
+ "encoding/json"
|
|
|
+ "io"
|
|
|
+ "io/ioutil"
|
|
|
"net/http"
|
|
|
|
|
|
"github.com/ethereum/go-ethereum/logger"
|
|
|
@@ -16,54 +19,83 @@ const (
|
|
|
|
|
|
// JSONRPC returns a handler that implements the Ethereum JSON-RPC API.
|
|
|
func JSONRPC(pipe *xeth.XEth, dataDir string) http.Handler {
|
|
|
- var json JsonWrapper
|
|
|
api := NewEthereumApi(pipe, dataDir)
|
|
|
|
|
|
return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
|
|
|
+ // TODO this needs to be configurable
|
|
|
w.Header().Set("Access-Control-Allow-Origin", "*")
|
|
|
|
|
|
- rpchttplogger.DebugDetailln("Handling request")
|
|
|
-
|
|
|
+ // Limit request size to resist DoS
|
|
|
if req.ContentLength > maxSizeReqLength {
|
|
|
jsonerr := &RpcErrorObject{-32700, "Request too large"}
|
|
|
- json.Send(w, &RpcErrorResponse{Jsonrpc: jsonrpcver, Id: nil, Error: jsonerr})
|
|
|
+ Send(w, &RpcErrorResponse{Jsonrpc: jsonrpcver, Id: nil, Error: jsonerr})
|
|
|
return
|
|
|
}
|
|
|
|
|
|
- reqParsed, reqerr := json.ParseRequestBody(req)
|
|
|
- switch reqerr.(type) {
|
|
|
- case nil:
|
|
|
- break
|
|
|
- case *DecodeParamError, *InsufficientParamsError, *ValidationError:
|
|
|
- jsonerr := &RpcErrorObject{-32602, reqerr.Error()}
|
|
|
- json.Send(w, &RpcErrorResponse{Jsonrpc: jsonrpcver, Id: nil, Error: jsonerr})
|
|
|
- return
|
|
|
- default:
|
|
|
- jsonerr := &RpcErrorObject{-32700, "Could not parse request"}
|
|
|
- json.Send(w, &RpcErrorResponse{Jsonrpc: jsonrpcver, Id: nil, Error: jsonerr})
|
|
|
- return
|
|
|
+ // Read request body
|
|
|
+ defer req.Body.Close()
|
|
|
+ body, err := ioutil.ReadAll(req.Body)
|
|
|
+ if err != nil {
|
|
|
+ jsonerr := &RpcErrorObject{-32700, "Could not read request body"}
|
|
|
+ Send(w, &RpcErrorResponse{Jsonrpc: jsonrpcver, Id: nil, Error: jsonerr})
|
|
|
}
|
|
|
|
|
|
- var response interface{}
|
|
|
- reserr := api.GetRequestReply(&reqParsed, &response)
|
|
|
- switch reserr.(type) {
|
|
|
- case nil:
|
|
|
- break
|
|
|
- case *NotImplementedError:
|
|
|
- jsonerr := &RpcErrorObject{-32601, reserr.Error()}
|
|
|
- json.Send(w, &RpcErrorResponse{Jsonrpc: jsonrpcver, Id: reqParsed.Id, Error: jsonerr})
|
|
|
+ // Try to parse the request as a single
|
|
|
+ var reqSingle RpcRequest
|
|
|
+ if err := json.Unmarshal(body, &reqSingle); err == nil {
|
|
|
+ response := RpcResponse(api, &reqSingle)
|
|
|
+ Send(w, &response)
|
|
|
return
|
|
|
- case *DecodeParamError, *InsufficientParamsError, *ValidationError:
|
|
|
- jsonerr := &RpcErrorObject{-32602, reserr.Error()}
|
|
|
- json.Send(w, &RpcErrorResponse{Jsonrpc: jsonrpcver, Id: reqParsed.Id, Error: jsonerr})
|
|
|
- return
|
|
|
- default:
|
|
|
- jsonerr := &RpcErrorObject{-32603, reserr.Error()}
|
|
|
- json.Send(w, &RpcErrorResponse{Jsonrpc: jsonrpcver, Id: reqParsed.Id, Error: jsonerr})
|
|
|
+ }
|
|
|
+
|
|
|
+ // Try to parse the request to batch
|
|
|
+ var reqBatch []RpcRequest
|
|
|
+ if err := json.Unmarshal(body, &reqBatch); err == nil {
|
|
|
+ // Build response batch
|
|
|
+ resBatch := make([]*interface{}, len(reqBatch))
|
|
|
+ for i, request := range reqBatch {
|
|
|
+ response := RpcResponse(api, &request)
|
|
|
+ resBatch[i] = response
|
|
|
+ }
|
|
|
+ Send(w, resBatch)
|
|
|
return
|
|
|
}
|
|
|
|
|
|
- rpchttplogger.DebugDetailf("Generated response: %T %s", response, response)
|
|
|
- json.Send(w, &RpcSuccessResponse{Jsonrpc: jsonrpcver, Id: reqParsed.Id, Result: response})
|
|
|
+ // Not a batch or single request, error
|
|
|
+ jsonerr := &RpcErrorObject{-32600, "Could not decode request"}
|
|
|
+ Send(w, &RpcErrorResponse{Jsonrpc: jsonrpcver, Id: nil, Error: jsonerr})
|
|
|
})
|
|
|
}
|
|
|
+
|
|
|
+func RpcResponse(api *EthereumApi, request *RpcRequest) *interface{} {
|
|
|
+ var reply, response interface{}
|
|
|
+ reserr := api.GetRequestReply(request, &reply)
|
|
|
+ switch reserr.(type) {
|
|
|
+ case nil:
|
|
|
+ response = &RpcSuccessResponse{Jsonrpc: jsonrpcver, Id: request.Id, Result: reply}
|
|
|
+ case *NotImplementedError:
|
|
|
+ jsonerr := &RpcErrorObject{-32601, reserr.Error()}
|
|
|
+ response = &RpcErrorResponse{Jsonrpc: jsonrpcver, Id: request.Id, Error: jsonerr}
|
|
|
+ case *DecodeParamError, *InsufficientParamsError, *ValidationError:
|
|
|
+ jsonerr := &RpcErrorObject{-32602, reserr.Error()}
|
|
|
+ response = &RpcErrorResponse{Jsonrpc: jsonrpcver, Id: request.Id, Error: jsonerr}
|
|
|
+ default:
|
|
|
+ jsonerr := &RpcErrorObject{-32603, reserr.Error()}
|
|
|
+ response = &RpcErrorResponse{Jsonrpc: jsonrpcver, Id: request.Id, Error: jsonerr}
|
|
|
+ }
|
|
|
+
|
|
|
+ rpchttplogger.DebugDetailf("Generated response: %T %s", response, response)
|
|
|
+ return &response
|
|
|
+}
|
|
|
+
|
|
|
+func Send(writer io.Writer, v interface{}) (n int, err error) {
|
|
|
+ var payload []byte
|
|
|
+ payload, err = json.MarshalIndent(v, "", "\t")
|
|
|
+ if err != nil {
|
|
|
+ rpclogger.Fatalln("Error marshalling JSON", err)
|
|
|
+ return 0, err
|
|
|
+ }
|
|
|
+ rpclogger.DebugDetailf("Sending payload: %s", payload)
|
|
|
+
|
|
|
+ return writer.Write(payload)
|
|
|
+}
|