2015-03-06 02:00:41 +00:00
package utils
import (
"crypto/ecdsa"
2015-04-20 15:45:37 +00:00
"fmt"
"log"
"net/http"
2015-03-09 22:00:27 +00:00
"os"
2015-03-06 02:00:41 +00:00
"path"
"runtime"
"github.com/codegangsta/cli"
2015-04-13 08:13:52 +00:00
"github.com/ethereum/ethash"
2015-03-07 11:39:52 +00:00
"github.com/ethereum/go-ethereum/accounts"
2015-03-18 07:44:58 +00:00
"github.com/ethereum/go-ethereum/common"
2015-03-06 02:00:41 +00:00
"github.com/ethereum/go-ethereum/core"
"github.com/ethereum/go-ethereum/crypto"
"github.com/ethereum/go-ethereum/eth"
"github.com/ethereum/go-ethereum/ethdb"
"github.com/ethereum/go-ethereum/event"
"github.com/ethereum/go-ethereum/logger"
2015-04-03 15:09:01 +00:00
"github.com/ethereum/go-ethereum/logger/glog"
2015-03-06 02:00:41 +00:00
"github.com/ethereum/go-ethereum/p2p/nat"
2015-03-09 22:00:27 +00:00
"github.com/ethereum/go-ethereum/rpc"
"github.com/ethereum/go-ethereum/xeth"
2015-03-06 02:00:41 +00:00
)
2015-03-10 15:44:48 +00:00
func init ( ) {
cli . AppHelpTemplate = ` { { . Name } } { { if . Flags } } [ global options ] { { end } } command { { if . Flags } } [ command options ] { { end } } [ arguments ... ]
VERSION :
{ { . Version } }
COMMANDS :
{ { range . Commands } } { { . Name } } { { with . ShortName } } , { { . } } { { end } } { { "\t" } } { { . Usage } }
{ { end } } { { if . Flags } }
GLOBAL OPTIONS :
{ { range . Flags } } { { . } }
{ { end } } { { end } }
`
cli . CommandHelpTemplate = ` { { . Name } } { { if . Subcommands } } command { { end } } { { if . Flags } } [ command options ] { { end } } [ arguments ... ]
{ { if . Description } } { { . Description } }
{ { end } } { { if . Subcommands } }
SUBCOMMANDS :
{ { range . Subcommands } } { { . Name } } { { with . ShortName } } , { { . } } { { end } } { { "\t" } } { { . Usage } }
{ { end } } { { end } } { { if . Flags } }
OPTIONS :
{ { range . Flags } } { { . } }
{ { end } } { { end } }
`
}
2015-03-09 21:51:50 +00:00
// NewApp creates an app with sane defaults.
func NewApp ( version , usage string ) * cli . App {
app := cli . NewApp ( )
app . Name = path . Base ( os . Args [ 0 ] )
app . Author = ""
2015-03-26 00:03:03 +00:00
//app.Authors = nil
2015-03-09 21:51:50 +00:00
app . Email = ""
app . Version = version
app . Usage = usage
return app
}
2015-03-06 02:00:41 +00:00
// These are all the command line flags we support.
// If you add to this list, please remember to include the
// flag in the appropriate command definition.
//
// The flags are defined here so their names and help texts
// are the same for all commands.
var (
// General settings
2015-04-08 13:43:55 +00:00
DataDirFlag = DirectoryFlag {
2015-03-06 02:00:41 +00:00
Name : "datadir" ,
Usage : "Data directory to be used" ,
2015-04-08 13:43:55 +00:00
Value : DirectoryString { common . DefaultDataDir ( ) } ,
2015-03-06 02:00:41 +00:00
}
2015-03-18 07:44:58 +00:00
ProtocolVersionFlag = cli . IntFlag {
Name : "protocolversion" ,
2015-04-21 23:41:34 +00:00
Usage : "ETH protocol version (integer)" ,
2015-03-18 07:44:58 +00:00
Value : eth . ProtocolVersion ,
}
NetworkIdFlag = cli . IntFlag {
Name : "networkid" ,
2015-04-21 23:41:34 +00:00
Usage : "Network Id (integer)" ,
2015-03-18 07:44:58 +00:00
Value : eth . NetworkId ,
}
2015-04-13 08:13:52 +00:00
BlockchainVersionFlag = cli . IntFlag {
Name : "blockchainversion" ,
2015-04-21 23:41:34 +00:00
Usage : "Blockchain version (integer)" ,
2015-04-13 08:13:52 +00:00
Value : core . BlockChainVersion ,
}
2015-04-18 21:53:30 +00:00
IdentityFlag = cli . StringFlag {
Name : "identity" ,
2015-04-21 23:41:34 +00:00
Usage : "Custom node name" ,
2015-04-18 21:53:30 +00:00
}
2015-04-08 11:22:31 +00:00
NatspecEnabledFlag = cli . BoolFlag {
Name : "natspec" ,
Usage : "Enable NatSpec confirmation notice" ,
}
2015-03-18 07:44:58 +00:00
// miner settings
2015-03-06 02:00:41 +00:00
MinerThreadsFlag = cli . IntFlag {
Name : "minerthreads" ,
Usage : "Number of miner threads" ,
Value : runtime . NumCPU ( ) ,
}
MiningEnabledFlag = cli . BoolFlag {
Name : "mine" ,
Usage : "Enable mining" ,
}
2015-03-26 21:49:22 +00:00
EtherbaseFlag = cli . StringFlag {
2015-03-27 11:14:00 +00:00
Name : "etherbase" ,
2015-04-21 23:41:34 +00:00
Usage : "Public address for block mining rewards. By default the address of your primary account is used" ,
2015-03-26 21:49:22 +00:00
Value : "primary" ,
}
2015-03-18 07:44:58 +00:00
UnlockedAccountFlag = cli . StringFlag {
Name : "unlock" ,
2015-04-21 23:41:34 +00:00
Usage : "Unlock the account given until this program exits (prompts for password). '--unlock primary' unlocks the primary account" ,
2015-03-23 13:00:06 +00:00
Value : "" ,
}
PasswordFileFlag = cli . StringFlag {
Name : "password" ,
2015-04-21 23:41:34 +00:00
Usage : "Path to password file to use with options and subcommands needing a password" ,
2015-03-23 13:00:06 +00:00
Value : "" ,
2015-03-18 07:44:58 +00:00
}
2015-03-06 02:00:41 +00:00
2015-03-18 07:44:58 +00:00
// logging and debug settings
2015-03-06 02:00:41 +00:00
LogFileFlag = cli . StringFlag {
Name : "logfile" ,
Usage : "Send log output to a file" ,
}
LogLevelFlag = cli . IntFlag {
Name : "loglevel" ,
2015-04-21 23:41:34 +00:00
Usage : "Logging verbosity: 0-6 (0=silent, 1=error, 2=warn, 3=info, 4=core, 5=debug, 6=debug detail)" ,
2015-03-06 02:00:41 +00:00
Value : int ( logger . InfoLevel ) ,
}
2015-03-21 09:20:47 +00:00
LogJSONFlag = cli . StringFlag {
Name : "logjson" ,
Usage : "Send json structured log output to a file or '-' for standard output (default: no json output)" ,
Value : "" ,
2015-03-06 02:00:41 +00:00
}
2015-04-20 15:45:37 +00:00
LogToStdErrFlag = cli . BoolFlag {
Name : "logtostderr" ,
Usage : "Logs are written to standard error instead of to files." ,
}
LogVModuleFlag = cli . GenericFlag {
Name : "vmodule" ,
2015-04-21 23:41:34 +00:00
Usage : "The syntax of the argument is a comma-separated list of pattern=N, where pattern is a literal file name (minus the \".go\" suffix) or \"glob\" pattern and N is a log verbosity level." ,
2015-04-20 15:45:37 +00:00
Value : glog . GetVModule ( ) ,
}
2015-03-18 07:44:58 +00:00
VMDebugFlag = cli . BoolFlag {
Name : "vmdebug" ,
Usage : "Virtual Machine debug output" ,
}
2015-04-20 15:45:37 +00:00
BacktraceAtFlag = cli . GenericFlag {
Name : "backtrace_at" ,
2015-04-21 23:41:34 +00:00
Usage : "If set to a file and line number (e.g., \"block.go:271\") holding a logging statement, a stack trace will be logged" ,
2015-04-20 15:45:37 +00:00
Value : glog . GetTraceLocation ( ) ,
}
2015-04-20 16:14:49 +00:00
PProfEanbledFlag = cli . BoolFlag {
Name : "pprof" ,
2015-04-21 23:41:34 +00:00
Usage : "Enable the profiling server on localhost" ,
2015-04-20 15:45:37 +00:00
}
PProfPortFlag = cli . IntFlag {
Name : "pprofport" ,
Usage : "Port on which the profiler should listen" ,
Value : 6060 ,
}
2015-03-06 02:00:41 +00:00
// RPC settings
RPCEnabledFlag = cli . BoolFlag {
Name : "rpc" ,
2015-04-21 23:41:34 +00:00
Usage : "Enable the JSON-RPC server" ,
2015-03-06 02:00:41 +00:00
}
RPCListenAddrFlag = cli . StringFlag {
Name : "rpcaddr" ,
Usage : "Listening address for the JSON-RPC server" ,
Value : "127.0.0.1" ,
}
RPCPortFlag = cli . IntFlag {
Name : "rpcport" ,
Usage : "Port on which the JSON-RPC server should listen" ,
Value : 8545 ,
}
2015-03-29 19:21:14 +00:00
RPCCORSDomainFlag = cli . StringFlag {
Name : "rpccorsdomain" ,
Usage : "Domain on which to send Access-Control-Allow-Origin header" ,
Value : "" ,
}
2015-03-06 02:00:41 +00:00
// Network Settings
MaxPeersFlag = cli . IntFlag {
Name : "maxpeers" ,
2015-04-21 23:41:34 +00:00
Usage : "Maximum number of network peers (network disabled if set to 0)" ,
2015-03-06 02:00:41 +00:00
Value : 16 ,
}
ListenPortFlag = cli . IntFlag {
Name : "port" ,
Usage : "Network listening port" ,
Value : 30303 ,
}
BootnodesFlag = cli . StringFlag {
Name : "bootnodes" ,
2015-04-21 23:41:34 +00:00
Usage : "Space-separated enode URLs of initial peers to connect to" ,
2015-03-06 02:00:41 +00:00
Value : "" ,
}
NodeKeyFileFlag = cli . StringFlag {
Name : "nodekey" ,
Usage : "P2P node key file" ,
}
NodeKeyHexFlag = cli . StringFlag {
Name : "nodekeyhex" ,
Usage : "P2P node key as hex (for testing)" ,
}
NATFlag = cli . StringFlag {
Name : "nat" ,
2015-04-21 23:41:34 +00:00
Usage : "NAT port mapping mechanism (any|none|upnp|pmp|extip:<IP>)" ,
2015-03-06 02:00:41 +00:00
Value : "any" ,
}
2015-04-20 15:45:37 +00:00
WhisperEnabledFlag = cli . BoolFlag {
Name : "shh" ,
2015-04-21 23:41:34 +00:00
Usage : "Enable whisper" ,
2015-04-20 15:45:37 +00:00
}
2015-03-15 06:31:40 +00:00
JSpathFlag = cli . StringFlag {
Name : "jspath" ,
Usage : "JS library path to be used with console and js subcommands" ,
Value : "." ,
}
2015-03-06 02:00:41 +00:00
)
func GetNAT ( ctx * cli . Context ) nat . Interface {
natif , err := nat . Parse ( ctx . GlobalString ( NATFlag . Name ) )
if err != nil {
Fatalf ( "Option %s: %v" , NATFlag . Name , err )
}
return natif
}
func GetNodeKey ( ctx * cli . Context ) ( key * ecdsa . PrivateKey ) {
hex , file := ctx . GlobalString ( NodeKeyHexFlag . Name ) , ctx . GlobalString ( NodeKeyFileFlag . Name )
var err error
switch {
case file != "" && hex != "" :
Fatalf ( "Options %q and %q are mutually exclusive" , NodeKeyFileFlag . Name , NodeKeyHexFlag . Name )
case file != "" :
if key , err = crypto . LoadECDSA ( file ) ; err != nil {
Fatalf ( "Option %q: %v" , NodeKeyFileFlag . Name , err )
}
case hex != "" :
if key , err = crypto . HexToECDSA ( hex ) ; err != nil {
Fatalf ( "Option %q: %v" , NodeKeyHexFlag . Name , err )
}
}
return key
}
2015-03-13 17:30:45 +00:00
func MakeEthConfig ( clientID , version string , ctx * cli . Context ) * eth . Config {
2015-04-03 15:09:01 +00:00
// Set verbosity on glog
glog . SetV ( ctx . GlobalInt ( LogLevelFlag . Name ) )
// Set the log type
2015-04-08 20:18:16 +00:00
//glog.SetToStderr(ctx.GlobalBool(LogToStdErrFlag.Name))
glog . SetToStderr ( true )
2015-04-07 12:57:04 +00:00
// Set the log dir
glog . SetLogDir ( ctx . GlobalString ( LogFileFlag . Name ) )
2015-04-03 15:09:01 +00:00
2015-04-18 21:53:30 +00:00
customName := ctx . GlobalString ( IdentityFlag . Name )
if len ( customName ) > 0 {
clientID += "/" + customName
}
2015-03-13 17:30:45 +00:00
return & eth . Config {
2015-04-14 14:12:35 +00:00
Name : common . MakeName ( clientID , version ) ,
DataDir : ctx . GlobalString ( DataDirFlag . Name ) ,
ProtocolVersion : ctx . GlobalInt ( ProtocolVersionFlag . Name ) ,
BlockChainVersion : ctx . GlobalInt ( BlockchainVersionFlag . Name ) ,
SkipBcVersionCheck : false ,
NetworkId : ctx . GlobalInt ( NetworkIdFlag . Name ) ,
LogFile : ctx . GlobalString ( LogFileFlag . Name ) ,
LogLevel : ctx . GlobalInt ( LogLevelFlag . Name ) ,
LogJSON : ctx . GlobalString ( LogJSONFlag . Name ) ,
Etherbase : ctx . GlobalString ( EtherbaseFlag . Name ) ,
MinerThreads : ctx . GlobalInt ( MinerThreadsFlag . Name ) ,
AccountManager : GetAccountManager ( ctx ) ,
VmDebug : ctx . GlobalBool ( VMDebugFlag . Name ) ,
MaxPeers : ctx . GlobalInt ( MaxPeersFlag . Name ) ,
Port : ctx . GlobalString ( ListenPortFlag . Name ) ,
NAT : GetNAT ( ctx ) ,
NatSpec : ctx . GlobalBool ( NatspecEnabledFlag . Name ) ,
NodeKey : GetNodeKey ( ctx ) ,
2015-04-20 15:45:37 +00:00
Shh : ctx . GlobalBool ( WhisperEnabledFlag . Name ) ,
2015-04-14 14:12:35 +00:00
Dial : true ,
BootNodes : ctx . GlobalString ( BootnodesFlag . Name ) ,
2015-03-13 17:30:45 +00:00
}
2015-03-06 02:00:41 +00:00
}
2015-03-16 10:27:38 +00:00
func GetChain ( ctx * cli . Context ) ( * core . ChainManager , common . Database , common . Database ) {
2015-03-06 02:00:41 +00:00
dataDir := ctx . GlobalString ( DataDirFlag . Name )
2015-04-13 08:13:52 +00:00
2015-03-06 17:26:16 +00:00
blockDb , err := ethdb . NewLDBDatabase ( path . Join ( dataDir , "blockchain" ) )
2015-03-06 02:00:41 +00:00
if err != nil {
Fatalf ( "Could not open database: %v" , err )
}
2015-03-06 17:26:16 +00:00
stateDb , err := ethdb . NewLDBDatabase ( path . Join ( dataDir , "state" ) )
if err != nil {
Fatalf ( "Could not open database: %v" , err )
}
2015-04-13 08:13:52 +00:00
extraDb , err := ethdb . NewLDBDatabase ( path . Join ( dataDir , "extra" ) )
if err != nil {
Fatalf ( "Could not open database: %v" , err )
}
eventMux := new ( event . TypeMux )
chainManager := core . NewChainManager ( blockDb , stateDb , eventMux )
pow := ethash . New ( chainManager )
txPool := core . NewTxPool ( eventMux , chainManager . State )
blockProcessor := core . NewBlockProcessor ( stateDb , extraDb , pow , txPool , chainManager , eventMux )
chainManager . SetProcessor ( blockProcessor )
return chainManager , blockDb , stateDb
2015-03-06 02:00:41 +00:00
}
2015-03-07 11:39:52 +00:00
2015-03-08 00:52:49 +00:00
func GetAccountManager ( ctx * cli . Context ) * accounts . Manager {
2015-03-07 11:39:52 +00:00
dataDir := ctx . GlobalString ( DataDirFlag . Name )
2015-03-26 18:55:39 +00:00
ks := crypto . NewKeyStorePassphrase ( path . Join ( dataDir , "keys" ) )
2015-03-09 23:09:39 +00:00
return accounts . NewManager ( ks )
2015-03-07 11:39:52 +00:00
}
2015-03-09 22:00:27 +00:00
func StartRPC ( eth * eth . Ethereum , ctx * cli . Context ) {
2015-03-29 19:26:47 +00:00
config := rpc . RpcConfig {
ListenAddress : ctx . GlobalString ( RPCListenAddrFlag . Name ) ,
ListenPort : uint ( ctx . GlobalInt ( RPCPortFlag . Name ) ) ,
CorsDomain : ctx . GlobalString ( RPCCORSDomainFlag . Name ) ,
2015-03-09 22:00:27 +00:00
}
2015-03-29 19:26:47 +00:00
xeth := xeth . New ( eth , nil )
_ = rpc . Start ( xeth , config )
2015-03-09 22:00:27 +00:00
}
2015-04-20 15:45:37 +00:00
func StartPProf ( ctx * cli . Context ) {
address := fmt . Sprintf ( "localhost:%d" , ctx . GlobalInt ( PProfPortFlag . Name ) )
go func ( ) {
log . Println ( http . ListenAndServe ( address , nil ) )
} ( )
}