path: root/cmd/gethrpctest/main.go

// Copyright 2015 The go-ethereum Authors
// This file is part of go-ethereum.
//
// go-ethereum is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// go-ethereum is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with go-ethereum. If not, see <http://www.gnu.org/licenses/>.

// gethrpctest is a command to run the external RPC tests.
package main

import (
    "flag"
    "io/ioutil"
    "log"
    "os"
    "os/signal"
    "path/filepath"
    "runtime"

    "errors"

    "github.com/ethereum/go-ethereum/accounts"
    "github.com/ethereum/go-ethereum/common"
    "github.com/ethereum/go-ethereum/crypto"
    "github.com/ethereum/go-ethereum/eth"
    "github.com/ethereum/go-ethereum/ethdb"
    "github.com/ethereum/go-ethereum/logger"
    "github.com/ethereum/go-ethereum/logger/glog"
    "github.com/ethereum/go-ethereum/node"
    "github.com/ethereum/go-ethereum/rpc"
    "github.com/ethereum/go-ethereum/tests"
    "github.com/ethereum/go-ethereum/whisper"
)

const defaultTestKey = "b71c71a67e1177ad4e901695e1b4b9ee17ae16c6668d313eac2f96dbcda3f291"

var (
    testFile = flag.String("json", "", "Path to the .json test file to load")
    testName = flag.String("test", "", "Name of the test from the .json file to run")
    testKey  = flag.String("key", defaultTestKey, "Private key of a test account to inject")
)

func main() {
    flag.Parse()

    // Load the test suite to run the RPC against
    tests, err := tests.LoadBlockTests(*testFile)
    if err != nil {
        log.Fatalf("Failed to load test suite: %v", err)
    }
    test, found := tests[*testName]
    if !found {
        log.Fatalf("Requested test (%s) not found within suite", *testName)
    }
    // Create the protocol stack to run the test with
    keydir, err := ioutil.TempDir("", "")
    if err != nil {
        log.Fatalf("Failed to create temporary keystore directory: %v", err)
    }
    defer os.RemoveAll(keydir)

    stack, err := MakeSystemNode(keydir, *testKey, test)
    if err != nil {
        log.Fatalf("Failed to assemble test stack: %v", err)
    }
    if err := stack.Start(); err != nil {
        log.Fatalf("Failed to start test node: %v", err)
    }
    defer stack.Stop()

    log.Println("Test node started...")

    // Make sure the tests contained within the suite pass
    if err := RunTest(stack, test); err != nil {
        log.Fatalf("Failed to run the pre-configured test: %v", err)
    }
    log.Println("Initial test suite passed...")

    if err := StartIPC(stack); err != nil {
        log.Fatalf("Failed to start IPC interface: %v\n", err)
    }
    log.Println("IPC Interface started, accepting requests...")

    // Start the RPC interface and wait until terminated
    if err := StartRPC(stack); err != nil {
        log.Fatalf("Failed to start RPC interface: %v", err)
    }
    log.Println("RPC Interface started, accepting requests...")

    quit := make(chan os.Signal, 1)
    signal.Notify(quit, os.Interrupt)
    <-quit
}

// MakeSystemNode configures a protocol stack for the RPC tests based on a given
// keystore path and initial pre-state.
func MakeSystemNode(keydir string, privkey string, test *tests.BlockTest) (*node.Node, error) {
    // Create a networkless protocol stack
    stack, err := node.New(&node.Config{NoDiscovery: true})
    if err != nil {
        return nil, err
    }
    // Create the keystore and inject an unlocked account if requested
    keystore := crypto.NewKeyStorePassphrase(keydir, crypto.StandardScryptN, crypto.StandardScryptP)
    accman := accounts.NewManager(keystore)

    if len(privkey) > 0 {
        key, err := crypto.HexToECDSA(privkey)
        if err != nil {
            return nil, err
        }
        if err := keystore.StoreKey(crypto.NewKeyFromECDSA(key), ""); err != nil {
            return nil, err
        }
        if err := accman.Unlock(crypto.NewKeyFromECDSA(key).Address, ""); err != nil {
            return nil, err
        }
    }
    // Initialize and register the Ethereum protocol
    db, _ := ethdb.NewMemDatabase()
    if _, err := test.InsertPreState(db, accman); err != nil {
        return nil, err
    }
    ethConf := &eth.Config{
        TestGenesisState: db,
        TestGenesisBlock: test.Genesis,
        AccountManager:   accman,
    }
    if err := stack.Register(func(ctx *node.ServiceContext) (node.Service, error) { return eth.New(ctx, ethConf) }); err != nil {
        return nil, err
    }
    // Initialize and register the Whisper protocol
    if err := stack.Register(func(*node.ServiceContext) (node.Service, error) { return whisper.New(), nil }); err != nil {
        return nil, err
    }
    return stack, nil
}

// RunTest executes the specified test against an already pre-configured protocol
// stack to ensure basic checks pass before running RPC tests.
func RunTest(stack *node.Node, test *tests.BlockTest) error {
    var ethereum *eth.Ethereum
    stack.Service(&ethereum)
    blockchain := ethereum.BlockChain()

    // Process the blocks and verify the imported headers
    blocks, err := test.TryBlocksInsert(blockchain)
    if err != nil {
        return err
    }
    if err := test.ValidateImportedHeaders(blockchain, blocks); err != nil {
        return err
    }
    // Retrieve the assembled state and validate it
    stateDb, err := blockchain.State()
    if err != nil {
        return err
    }
    if err := test.ValidatePostState(stateDb); err != nil {
        return err
    }
    return nil
}

// StartRPC initializes an RPC interface to the given protocol stack.
func StartRPC(stack *node.Node) error {
    /*
        web3 := NewPublicWeb3API(stack)
        server.RegisterName("web3", web3)
        net := NewPublicNetAPI(stack.Server(), ethereum.NetVersion())
        server.RegisterName("net", net)
    */

    for _, api := range stack.APIs() {
        if adminApi, ok := api.Service.(*node.PrivateAdminAPI); ok {
            _, err := adminApi.StartRPC("127.0.0.1", 8545, "", "admin,db,eth,debug,miner,net,shh,txpool,personal,web3")
            return err
        }
    }

    glog.V(logger.Error).Infof("Unable to start RPC-HTTP interface, could not find admin API")
    return errors.New("Unable to start RPC-HTTP interface")
}

// StartIPC initializes an IPC interface to the given protocol stack.
func StartIPC(stack *node.Node) error {
    var ethereum *eth.Ethereum
    if err := stack.Service(&ethereum); err != nil {
        return err
    }

    endpoint := `\\.\pipe\geth.ipc`
    if runtime.GOOS != "windows" {
        endpoint = filepath.Join(common.DefaultDataDir(), "geth.ipc")
    }

    listener, err := rpc.CreateIPCListener(endpoint)
    if err != nil {
        return err
    }

    server := rpc.NewServer()

    // register package API's this node provides
    offered := stack.APIs()
    for _, api := range offered {
        server.RegisterName(api.Namespace, api.Service)
        glog.V(logger.Debug).Infof("Register %T under namespace '%s' for IPC service\n", api.Service, api.Namespace)
    }

    //var ethereum *eth.Ethereum
    //if err := stack.Service(&ethereum); err != nil {
    //  return err
    //}

    go func() {
        glog.V(logger.Info).Infof("Start IPC server on %s\n", endpoint)
        for {
            conn, err := listener.Accept()
            if err != nil {
                glog.V(logger.Error).Infof("Unable to accept connection - %v\n", err)
            }

            codec := rpc.NewJSONCodec(conn)
            go server.ServeCodec(codec)
        }
    }()

    return nil
}