neo-go/pkg/rpc/server.go

package rpc

import (
	"context"
	"errors"
	"fmt"
	"net/http"

	"github.com/CityOfZion/neo-go/pkg/core"
	"github.com/CityOfZion/neo-go/pkg/network"
	"github.com/CityOfZion/neo-go/pkg/rpc/result"
	"github.com/CityOfZion/neo-go/pkg/rpc/wrappers"
	"github.com/CityOfZion/neo-go/pkg/util"
	log "github.com/sirupsen/logrus"
)

type (
	// Server represents the JSON-RPC 2.0 server.
	Server struct {
		*http.Server
		chain      core.Blockchainer
		coreServer *network.Server
	}
)

var (
	invalidBlockHeightError = func(index int, height int) error {
		return fmt.Errorf("Param at index %d should be greater than or equal to 0 and less then or equal to current block height, got: %d", index, height)
	}
)

// NewServer creates a new Server struct.
func NewServer(chain core.Blockchainer, port uint16, coreServer *network.Server) Server {
	return Server{
		Server: &http.Server{
			Addr: fmt.Sprintf(":%d", port),
		},
		chain:      chain,
		coreServer: coreServer,
	}
}

// Start creates a new JSON-RPC server
// listening on the configured port.
func (s *Server) Start(errChan chan error) {
	s.Handler = http.HandlerFunc(s.requestHandler)
	log.WithFields(log.Fields{
		"endpoint": s.Addr,
	}).Info("starting rpc-server")

	errChan <- s.ListenAndServe()
}

// Shutdown overrride the http.Server Shutdown
// method.
func (s *Server) Shutdown() error {
	log.WithFields(log.Fields{
		"endpoint": s.Addr,
	}).Info("shutting down rpc-server")
	return s.Server.Shutdown(context.Background())
}

func (s *Server) requestHandler(w http.ResponseWriter, httpRequest *http.Request) {
	req := NewRequest()

	if httpRequest.Method != "POST" {
		req.WriteErrorResponse(
			w,
			NewInvalidParamsError(
				fmt.Sprintf("Invalid method '%s', please retry with 'POST'", httpRequest.Method), nil,
			),
		)
		return
	}

	err := req.DecodeData(httpRequest.Body)
	if err != nil {
		req.WriteErrorResponse(w, NewParseError("Problem parsing JSON-RPC request body", err))
		return
	}

	reqParams, err := req.Params()
	if err != nil {
		req.WriteErrorResponse(w, NewInvalidParamsError("Problem parsing request parameters", err))
		return
	}

	s.methodHandler(w, req, *reqParams)
}

func (s *Server) methodHandler(w http.ResponseWriter, req *Request, reqParams Params) {
	log.WithFields(log.Fields{
		"method": req.Method,
		"params": fmt.Sprintf("%v", reqParams),
	}).Info("processing rpc request")

	var results interface{}
	var resultsErr *Error

Methods:
	switch req.Method {
	case "getbestblockhash":
		results = s.chain.CurrentBlockHash().String()

	case "getblock":
		var hash util.Uint256
		var err error

		param, exists := reqParams.ValueAt(0)
		if !exists {
			err = errors.New("Param at index at 0 doesn't exist")
			resultsErr = NewInvalidParamsError(err.Error(), err)
			break
		}

		switch param.Type {
		case "string":
			hash, err = util.Uint256DecodeString(param.StringVal)
			if err != nil {
				resultsErr = NewInvalidParamsError("Problem decoding block hash", err)
				break
			}
		case "number":
			if !s.validBlockHeight(param) {
				err = invalidBlockHeightError(0, param.IntVal)
				resultsErr = NewInvalidParamsError(err.Error(), err)
				break Methods
			}

			hash = s.chain.GetHeaderHash(param.IntVal)
		case "default":
			err = errors.New("Expected param at index 0 to be either string or number")
			resultsErr = NewInvalidParamsError(err.Error(), err)
			break
		}

		block, err := s.chain.GetBlock(hash)
		if err != nil {
			resultsErr = NewInternalServerError(fmt.Sprintf("Problem locating block with hash: %s", hash), err)
			break
		}

		results = wrappers.NewBlock(block, s.chain)
	case "getblockcount":
		results = s.chain.BlockHeight()

	case "getblockhash":
		if param, exists := reqParams.ValueAtAndType(0, "number"); exists && s.validBlockHeight(param) {
			results = s.chain.GetHeaderHash(param.IntVal)
		} else {
			err := invalidBlockHeightError(0, param.IntVal)
			resultsErr = NewInvalidParamsError(err.Error(), err)
			break
		}

	case "getconnectioncount":
		results = s.coreServer.PeerCount()

	case "getversion":
		results = result.Version{
			Port:      s.coreServer.ListenTCP,
			Nonce:     s.coreServer.ID(),
			UserAgent: s.coreServer.UserAgent,
		}

	case "getpeers":
		peers := result.NewPeers()
		for _, addr := range s.coreServer.UnconnectedPeers() {
			peers.AddPeer("unconnected", addr)
		}

		for _, addr := range s.coreServer.BadPeers() {
			peers.AddPeer("bad", addr)
		}

		for addr := range s.coreServer.Peers() {
			peers.AddPeer("connected", addr.Endpoint().String())
		}

		results = peers

	case "validateaddress", "getblocksysfee", "getcontractstate", "getrawmempool", "getrawtransaction", "getstorage", "submitblock", "gettxout", "invoke", "invokefunction", "invokescript", "sendrawtransaction", "getaccountstate":
		results = "TODO"

	case "getassetstate":
		var err error

		param, exists := reqParams.ValueAt(0)
		if !exists {
			err = errors.New("Param at index at 0 doesn't exist")
			resultsErr = NewInvalidParamsError(err.Error(), err)
			break
		}

		if param.Type != "string" {
			err = errors.New("Param need to be a string")
			resultsErr = NewInvalidParamsError(err.Error(), err)
			break
		}

		paramAssetID, err := util.Uint256DecodeString(param.StringVal)

		as := s.chain.GetAssetState(paramAssetID)

		if as != nil {
			results = wrappers.NewAssetState(as)
		} else {
			results = "Invalid assetid"
		}

	default:
		resultsErr = NewMethodNotFoundError(fmt.Sprintf("Method '%s' not supported", req.Method), nil)
	}

	if resultsErr != nil {
		req.WriteErrorResponse(w, resultsErr)
		return
	}

	req.WriteResponse(w, results)
}

func (s Server) validBlockHeight(param *Param) bool {
	return param.IntVal >= 0 && param.IntVal <= int(s.chain.BlockHeight())
}