flags.go 27.0 KB
Newer Older
F
Felix Lange 已提交
1 2 3 4 5 6 7 8 9 10
// 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
11
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
F
Felix Lange 已提交
12 13 14
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
15
// along with go-ethereum. If not, see <http://www.gnu.org/licenses/>.
F
Felix Lange 已提交
16

17 18 19 20
package utils

import (
	"crypto/ecdsa"
21
	"fmt"
22
	"io/ioutil"
23
	"math"
24
	"math/big"
25
	"math/rand"
26
	"os"
27
	"path/filepath"
28
	"runtime"
29
	"strconv"
30
	"strings"
31
	"time"
32

33
	"github.com/codegangsta/cli"
34
	"github.com/ethereum/ethash"
F
Felix Lange 已提交
35
	"github.com/ethereum/go-ethereum/accounts"
Z
zelig 已提交
36
	"github.com/ethereum/go-ethereum/common"
37
	"github.com/ethereum/go-ethereum/core"
38
	"github.com/ethereum/go-ethereum/core/state"
39 40 41 42 43
	"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"
O
obscuren 已提交
44
	"github.com/ethereum/go-ethereum/logger/glog"
B
Bas van Kervel 已提交
45
	"github.com/ethereum/go-ethereum/metrics"
46 47
	"github.com/ethereum/go-ethereum/node"
	"github.com/ethereum/go-ethereum/p2p/discover"
48
	"github.com/ethereum/go-ethereum/p2p/nat"
49
	"github.com/ethereum/go-ethereum/params"
50
	"github.com/ethereum/go-ethereum/rpc"
51
	"github.com/ethereum/go-ethereum/whisper"
52 53
)

54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79
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}}
`
}

F
Felix Lange 已提交
80 81 82
// NewApp creates an app with sane defaults.
func NewApp(version, usage string) *cli.App {
	app := cli.NewApp()
83
	app.Name = filepath.Base(os.Args[0])
F
Felix Lange 已提交
84
	app.Author = ""
O
obscuren 已提交
85
	//app.Authors = nil
F
Felix Lange 已提交
86 87 88 89 90 91
	app.Email = ""
	app.Version = version
	app.Usage = usage
	return app
}

92 93 94 95 96 97 98 99 100
// 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
101
	DataDirFlag = DirectoryFlag{
102
		Name:  "datadir",
103
		Usage: "Data directory for the databases and keystore",
104
		Value: DirectoryString{common.DefaultDataDir()},
105
	}
K
Kobi Gurkan 已提交
106 107 108 109
	KeyStoreDirFlag = DirectoryFlag{
		Name:  "keystore",
		Usage: "Directory for the keystore (default = inside the datadir)",
	}
Z
zelig 已提交
110 111
	NetworkIdFlag = cli.IntFlag{
		Name:  "networkid",
112
		Usage: "Network identifier (integer, 0=Olympic, 1=Frontier, 2=Morden)",
Z
zelig 已提交
113 114
		Value: eth.NetworkId,
	}
115 116 117
	OlympicFlag = cli.BoolFlag{
		Name:  "olympic",
		Usage: "Olympic network: pre-configured pre-release test network",
O
obscuren 已提交
118
	}
119 120 121
	TestNetFlag = cli.BoolFlag{
		Name:  "testnet",
		Usage: "Morden network: pre-configured test network with modified starting nonces (replay protection)",
122
	}
123 124
	DevModeFlag = cli.BoolFlag{
		Name:  "dev",
125
		Usage: "Developer mode: pre-configured private network with several debugging flags",
126
	}
127 128 129
	GenesisFileFlag = cli.StringFlag{
		Name:  "genesis",
		Usage: "Insert/overwrite the genesis block (JSON format)",
130
	}
131 132
	IdentityFlag = cli.StringFlag{
		Name:  "identity",
133
		Usage: "Custom node name",
134
	}
135 136 137 138
	NatspecEnabledFlag = cli.BoolFlag{
		Name:  "natspec",
		Usage: "Enable NatSpec confirmation notice",
	}
Z
zelig 已提交
139 140 141 142 143
	DocRootFlag = DirectoryFlag{
		Name:  "docroot",
		Usage: "Document Root for HTTPClient file scheme",
		Value: DirectoryString{common.HomeDir()},
	}
144 145
	CacheFlag = cli.IntFlag{
		Name:  "cache",
146
		Usage: "Megabytes of memory allocated to internal caching (min 16MB / database forced)",
147
		Value: 128,
148
	}
149 150 151 152
	BlockchainVersionFlag = cli.IntFlag{
		Name:  "blockchainversion",
		Usage: "Blockchain version (integer)",
		Value: core.BlockChainVersion,
153
	}
154 155
	FastSyncFlag = cli.BoolFlag{
		Name:  "fast",
156
		Usage: "Enable fast syncing through state downloads",
157
	}
158 159
	LightKDFFlag = cli.BoolFlag{
		Name:  "lightkdf",
160
		Usage: "Reduce key-derivation RAM & CPU usage at some expense of KDF strength",
161
	}
162
	// Miner settings
163
	// TODO: refactor CPU vs GPU mining flags
164 165 166
	MiningEnabledFlag = cli.BoolFlag{
		Name:  "mine",
		Usage: "Enable mining",
167
	}
168 169
	MinerThreadsFlag = cli.IntFlag{
		Name:  "minerthreads",
170
		Usage: "Number of CPU threads to use for mining",
171 172
		Value: runtime.NumCPU(),
	}
173 174 175
	MiningGPUFlag = cli.StringFlag{
		Name:  "minergpus",
		Usage: "List of GPUs to use for mining (e.g. '0,1' will use the first two GPUs found)",
176
	}
177 178 179 180 181
	TargetGasLimitFlag = cli.StringFlag{
		Name:  "targetgaslimit",
		Usage: "Target gas limit sets the artificial target gas floor for the blocks to mine",
		Value: params.GenesisGasLimit.String(),
	}
182 183 184 185
	AutoDAGFlag = cli.BoolFlag{
		Name:  "autodag",
		Usage: "Enable automatic DAG pregeneration",
	}
Z
zelig 已提交
186
	EtherbaseFlag = cli.StringFlag{
O
obscuren 已提交
187
		Name:  "etherbase",
188
		Usage: "Public address for block mining rewards (default = first account created)",
189
		Value: "0",
Z
zelig 已提交
190
	}
191 192
	GasPriceFlag = cli.StringFlag{
		Name:  "gasprice",
193
		Usage: "Minimal gas price to accept for mining a transactions",
194
		Value: new(big.Int).Mul(big.NewInt(20), common.Shannon).String(),
195
	}
Z
zelig 已提交
196 197
	ExtraDataFlag = cli.StringFlag{
		Name:  "extradata",
198
		Usage: "Block extra data set by the miner (default = client version)",
Z
zelig 已提交
199
	}
200
	// Account settings
Z
zelig 已提交
201 202
	UnlockedAccountFlag = cli.StringFlag{
		Name:  "unlock",
203
		Usage: "Comma separated list of accounts to unlock",
Z
zelig 已提交
204 205 206 207
		Value: "",
	}
	PasswordFileFlag = cli.StringFlag{
		Name:  "password",
208
		Usage: "Password file to use for non-inteactive password input",
Z
zelig 已提交
209
		Value: "",
Z
zelig 已提交
210
	}
211

212 213 214 215 216 217 218 219 220 221 222 223 224 225
	VMForceJitFlag = cli.BoolFlag{
		Name:  "forcejit",
		Usage: "Force the JIT VM to take precedence",
	}
	VMJitCacheFlag = cli.IntFlag{
		Name:  "jitcache",
		Usage: "Amount of cached JIT VM programs",
		Value: 64,
	}
	VMEnableJitFlag = cli.BoolFlag{
		Name:  "jitvm",
		Usage: "Enable the JIT VM",
	}

Z
zelig 已提交
226
	// logging and debug settings
227 228
	MetricsEnabledFlag = cli.BoolFlag{
		Name:  metrics.MetricsEnabledFlag,
229
		Usage: "Enable metrics collection and reporting",
230
	}
231 232 233 234

	// RPC settings
	RPCEnabledFlag = cli.BoolFlag{
		Name:  "rpc",
235
		Usage: "Enable the HTTP-RPC server",
236 237 238
	}
	RPCListenAddrFlag = cli.StringFlag{
		Name:  "rpcaddr",
239
		Usage: "HTTP-RPC server listening interface",
240
		Value: common.DefaultHTTPHost,
241 242 243
	}
	RPCPortFlag = cli.IntFlag{
		Name:  "rpcport",
244
		Usage: "HTTP-RPC server listening port",
245
		Value: common.DefaultHTTPPort,
246
	}
247 248
	RPCCORSDomainFlag = cli.StringFlag{
		Name:  "rpccorsdomain",
249
		Usage: "Comma separated list of domains from which to accept cross origin requests (browser enforced)",
250 251
		Value: "",
	}
252
	RPCApiFlag = cli.StringFlag{
253
		Name:  "rpcapi",
254
		Usage: "API's offered over the HTTP-RPC interface",
255
		Value: rpc.DefaultHTTPApis,
256
	}
B
Bas van Kervel 已提交
257 258 259 260 261 262
	IPCDisabledFlag = cli.BoolFlag{
		Name:  "ipcdisable",
		Usage: "Disable the IPC-RPC server",
	}
	IPCApiFlag = cli.StringFlag{
		Name:  "ipcapi",
263
		Usage: "API's offered over the IPC-RPC interface",
264
		Value: rpc.DefaultIPCApis,
B
Bas van Kervel 已提交
265 266 267
	}
	IPCPathFlag = DirectoryFlag{
		Name:  "ipcpath",
268
		Usage: "Filename for IPC socket/pipe within the datadir (explicit paths escape it)",
269
		Value: DirectoryString{common.DefaultIPCSocket},
B
Bas van Kervel 已提交
270
	}
271
	WSEnabledFlag = cli.BoolFlag{
272
		Name:  "ws",
273 274 275 276 277
		Usage: "Enable the WS-RPC server",
	}
	WSListenAddrFlag = cli.StringFlag{
		Name:  "wsaddr",
		Usage: "WS-RPC server listening interface",
278
		Value: common.DefaultWSHost,
279 280 281 282
	}
	WSPortFlag = cli.IntFlag{
		Name:  "wsport",
		Usage: "WS-RPC server listening port",
283
		Value: common.DefaultWSPort,
284 285 286 287
	}
	WSApiFlag = cli.StringFlag{
		Name:  "wsapi",
		Usage: "API's offered over the WS-RPC interface",
288
		Value: rpc.DefaultHTTPApis,
289
	}
B
Bas van Kervel 已提交
290 291 292
	WSAllowedOriginsFlag = cli.StringFlag{
		Name:  "wsorigins",
		Usage: "Origins from which to accept websockets requests",
293
		Value: "",
294
	}
295 296
	ExecFlag = cli.StringFlag{
		Name:  "exec",
297
		Usage: "Execute JavaScript statement (only in combination with console/attach)",
298
	}
299 300 301 302
	PreLoadJSFlag = cli.StringFlag{
		Name:  "preload",
		Usage: "Comma separated list of JavaScript files to preload into the console",
	}
303

304 305 306
	// Network Settings
	MaxPeersFlag = cli.IntFlag{
		Name:  "maxpeers",
307
		Usage: "Maximum number of network peers (network disabled if set to 0)",
308
		Value: 25,
309
	}
310 311 312 313 314
	MaxPendingPeersFlag = cli.IntFlag{
		Name:  "maxpendpeers",
		Usage: "Maximum number of pending connection attempts (defaults used if set to 0)",
		Value: 0,
	}
315 316 317 318 319 320 321
	ListenPortFlag = cli.IntFlag{
		Name:  "port",
		Usage: "Network listening port",
		Value: 30303,
	}
	BootnodesFlag = cli.StringFlag{
		Name:  "bootnodes",
322
		Usage: "Comma separated enode URLs for P2P discovery bootstrap",
323 324 325 326 327 328 329 330 331 332 333 334
		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",
335
		Usage: "NAT port mapping mechanism (any|none|upnp|pmp|extip:<IP>)",
336 337
		Value: "any",
	}
338 339 340 341
	NoDiscoverFlag = cli.BoolFlag{
		Name:  "nodiscover",
		Usage: "Disables the peer discovery mechanism (manual peer addition)",
	}
342 343
	WhisperEnabledFlag = cli.BoolFlag{
		Name:  "shh",
344
		Usage: "Enable Whisper",
345
	}
346
	// ATM the url is left to the user and deployment to
Z
CLI:  
zelig 已提交
347 348
	JSpathFlag = cli.StringFlag{
		Name:  "jspath",
I
Isidoro Ghezzi 已提交
349
		Usage: "JavaScript root path for `loadScript` and document root for `admin.httpGet`",
Z
CLI:  
zelig 已提交
350 351
		Value: ".",
	}
352 353
	SolcPathFlag = cli.StringFlag{
		Name:  "solc",
354
		Usage: "Solidity compiler command to be used",
355 356
		Value: "solc",
	}
357 358

	// Gas price oracle settings
Z
zsfelfoldi 已提交
359 360 361
	GpoMinGasPriceFlag = cli.StringFlag{
		Name:  "gpomin",
		Usage: "Minimum suggested gas price",
362
		Value: new(big.Int).Mul(big.NewInt(20), common.Shannon).String(),
Z
zsfelfoldi 已提交
363 364 365 366
	}
	GpoMaxGasPriceFlag = cli.StringFlag{
		Name:  "gpomax",
		Usage: "Maximum suggested gas price",
367
		Value: new(big.Int).Mul(big.NewInt(500), common.Shannon).String(),
Z
zsfelfoldi 已提交
368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388
	}
	GpoFullBlockRatioFlag = cli.IntFlag{
		Name:  "gpofull",
		Usage: "Full block threshold for gas price calculation (%)",
		Value: 80,
	}
	GpobaseStepDownFlag = cli.IntFlag{
		Name:  "gpobasedown",
		Usage: "Suggested gas price base step down ratio (1/1000)",
		Value: 10,
	}
	GpobaseStepUpFlag = cli.IntFlag{
		Name:  "gpobaseup",
		Usage: "Suggested gas price base step up ratio (1/1000)",
		Value: 100,
	}
	GpobaseCorrectionFactorFlag = cli.IntFlag{
		Name:  "gpobasecf",
		Usage: "Suggested gas price base correction factor (%)",
		Value: 110,
	}
389 390
)

391 392 393 394 395 396 397 398 399
// MustMakeDataDir retrieves the currently requested data directory, terminating
// if none (or the empty string) is specified. If the node is starting a testnet,
// the a subdirectory of the specified datadir will be used.
func MustMakeDataDir(ctx *cli.Context) string {
	if path := ctx.GlobalString(DataDirFlag.Name); path != "" {
		if ctx.GlobalBool(TestNetFlag.Name) {
			return filepath.Join(path, "/testnet")
		}
		return path
400
	}
401 402
	Fatalf("Cannot determine default data directory, please set manually (--datadir)")
	return ""
403 404
}

K
Kobi Gurkan 已提交
405 406 407 408 409 410 411 412 413 414
// MakeKeyStoreDir resolves the folder to use for storing the account keys from the
// set command line flags, returning the explicitly requested path, or one inside
// the data directory otherwise.
func MakeKeyStoreDir(datadir string, ctx *cli.Context) string {
	if path := ctx.GlobalString(KeyStoreDirFlag.Name); path != "" {
		return path
	}
	return filepath.Join(datadir, "keystore")
}

415
// MakeIPCPath creates an IPC path configuration from the set command line flags,
416
// returning an empty string if IPC was explicitly disabled, or the set path.
417
func MakeIPCPath(ctx *cli.Context) string {
418 419 420 421 422 423
	if ctx.GlobalBool(IPCDisabledFlag.Name) {
		return ""
	}
	return ctx.GlobalString(IPCPathFlag.Name)
}

424 425 426 427 428 429 430 431 432 433 434
// MakeNodeKey creates a node key from set command line flags, either loading it
// from a file or as a specified hex value. If neither flags were provided, this
// method returns nil and an emphemeral key is to be generated.
func MakeNodeKey(ctx *cli.Context) *ecdsa.PrivateKey {
	var (
		hex  = ctx.GlobalString(NodeKeyHexFlag.Name)
		file = ctx.GlobalString(NodeKeyFileFlag.Name)

		key *ecdsa.PrivateKey
		err error
	)
435 436 437
	switch {
	case file != "" && hex != "":
		Fatalf("Options %q and %q are mutually exclusive", NodeKeyFileFlag.Name, NodeKeyHexFlag.Name)
438

439 440 441 442
	case file != "":
		if key, err = crypto.LoadECDSA(file); err != nil {
			Fatalf("Option %q: %v", NodeKeyFileFlag.Name, err)
		}
443

444 445 446 447 448 449 450 451
	case hex != "":
		if key, err = crypto.HexToECDSA(hex); err != nil {
			Fatalf("Option %q: %v", NodeKeyHexFlag.Name, err)
		}
	}
	return key
}

452 453 454 455 456 457 458 459
// MakeNodeName creates a node name from a base set and the command line flags.
func MakeNodeName(client, version string, ctx *cli.Context) string {
	name := common.MakeName(client, version)
	if identity := ctx.GlobalString(IdentityFlag.Name); len(identity) > 0 {
		name += "/" + identity
	}
	if ctx.GlobalBool(VMEnableJitFlag.Name) {
		name += "/JIT"
460
	}
461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496
	return name
}

// MakeBootstrapNodes creates a list of bootstrap nodes from the command line
// flags, reverting to pre-configured ones if none have been specified.
func MakeBootstrapNodes(ctx *cli.Context) []*discover.Node {
	// Return pre-configured nodes if none were manually requested
	if !ctx.GlobalIsSet(BootnodesFlag.Name) {
		if ctx.GlobalBool(TestNetFlag.Name) {
			return TestNetBootNodes
		}
		return FrontierBootNodes
	}
	// Otherwise parse and use the CLI bootstrap nodes
	bootnodes := []*discover.Node{}

	for _, url := range strings.Split(ctx.GlobalString(BootnodesFlag.Name), ",") {
		node, err := discover.ParseNode(url)
		if err != nil {
			glog.V(logger.Error).Infof("Bootstrap URL %s: %v\n", url, err)
			continue
		}
		bootnodes = append(bootnodes, node)
	}
	return bootnodes
}

// MakeListenAddress creates a TCP listening address string from set command
// line flags.
func MakeListenAddress(ctx *cli.Context) string {
	return fmt.Sprintf(":%d", ctx.GlobalInt(ListenPortFlag.Name))
}

// MakeNAT creates a port mapper from set command line flags.
func MakeNAT(ctx *cli.Context) nat.Interface {
	natif, err := nat.Parse(ctx.GlobalString(NATFlag.Name))
497
	if err != nil {
498 499 500 501 502
		Fatalf("Option %s: %v", NATFlag.Name, err)
	}
	return natif
}

503 504 505 506 507 508 509 510 511 512
// MakeRPCModules splits input separated by a comma and trims excessive white
// space from the substrings.
func MakeRPCModules(input string) []string {
	result := strings.Split(input, ",")
	for i, r := range result {
		result[i] = strings.TrimSpace(r)
	}
	return result
}

513
// MakeHTTPRpcHost creates the HTTP RPC listener interface string from the set
514
// command line flags, returning empty if the HTTP endpoint is disabled.
515
func MakeHTTPRpcHost(ctx *cli.Context) string {
516 517 518 519 520 521
	if !ctx.GlobalBool(RPCEnabledFlag.Name) {
		return ""
	}
	return ctx.GlobalString(RPCListenAddrFlag.Name)
}

522
// MakeWSRpcHost creates the WebSocket RPC listener interface string from the set
523
// command line flags, returning empty if the HTTP endpoint is disabled.
524
func MakeWSRpcHost(ctx *cli.Context) string {
525 526 527 528 529 530
	if !ctx.GlobalBool(WSEnabledFlag.Name) {
		return ""
	}
	return ctx.GlobalString(WSListenAddrFlag.Name)
}

531 532 533 534 535 536 537 538 539 540 541 542 543 544
// MakeGenesisBlock loads up a genesis block from an input file specified in the
// command line, or returns the empty string if none set.
func MakeGenesisBlock(ctx *cli.Context) string {
	genesis := ctx.GlobalString(GenesisFileFlag.Name)
	if genesis == "" {
		return ""
	}
	data, err := ioutil.ReadFile(genesis)
	if err != nil {
		Fatalf("Failed to load custom genesis file: %v", err)
	}
	return string(data)
}

545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560
// MakeDatabaseHandles raises out the number of allowed file handles per process
// for Geth and returns half of the allowance to assign to the database.
func MakeDatabaseHandles() int {
	if err := raiseFdLimit(2048); err != nil {
		Fatalf("Failed to raise file descriptor allowance: %v", err)
	}
	limit, err := getFdLimit()
	if err != nil {
		Fatalf("Failed to retrieve file descriptor allowance: %v", err)
	}
	if limit > 2048 { // cap database file descriptors even if more is available
		limit = 2048
	}
	return limit / 2 // Leave half for networking and other stuff
}

561 562 563
// MakeAccountManager creates an account manager from set command line flags.
func MakeAccountManager(ctx *cli.Context) *accounts.Manager {
	// Create the keystore crypto primitive, light if requested
564 565
	scryptN := accounts.StandardScryptN
	scryptP := accounts.StandardScryptP
566
	if ctx.GlobalBool(LightKDFFlag.Name) {
567 568
		scryptN = accounts.LightScryptN
		scryptP = accounts.LightScryptP
569
	}
570 571 572
	datadir := MustMakeDataDir(ctx)
	keydir := MakeKeyStoreDir(datadir, ctx)
	return accounts.NewManager(keydir, scryptN, scryptP)
573 574 575 576
}

// MakeAddress converts an account specified directly as a hex encoded string or
// a key index in the key store to an internal account representation.
F
Felix Lange 已提交
577
func MakeAddress(accman *accounts.Manager, account string) (accounts.Account, error) {
578 579
	// If the specified account is a valid address, return it
	if common.IsHexAddress(account) {
F
Felix Lange 已提交
580
		return accounts.Account{Address: common.HexToAddress(account)}, nil
581 582 583 584
	}
	// Otherwise try to interpret the account as a keystore index
	index, err := strconv.Atoi(account)
	if err != nil {
F
Felix Lange 已提交
585
		return accounts.Account{}, fmt.Errorf("invalid account address or index %q", account)
586
	}
F
Felix Lange 已提交
587
	return accman.AccountByIndex(index)
588 589 590 591 592
}

// MakeEtherbase retrieves the etherbase either from the directly specified
// command line flags or from the keystore if CLI indexed.
func MakeEtherbase(accman *accounts.Manager, ctx *cli.Context) common.Address {
F
Felix Lange 已提交
593
	accounts := accman.Accounts()
594
	if !ctx.GlobalIsSet(EtherbaseFlag.Name) && len(accounts) == 0 {
595
		glog.V(logger.Error).Infoln("WARNING: No etherbase set and no accounts found as default")
596 597
		return common.Address{}
	}
598 599 600
	etherbase := ctx.GlobalString(EtherbaseFlag.Name)
	if etherbase == "" {
		return common.Address{}
601
	}
602
	// If the specified etherbase is a valid address, return it
F
Felix Lange 已提交
603
	account, err := MakeAddress(accman, etherbase)
604
	if err != nil {
605
		Fatalf("Option %q: %v", EtherbaseFlag.Name, err)
606
	}
F
Felix Lange 已提交
607
	return account.Address
608 609 610 611 612 613 614
}

// MakeMinerExtra resolves extradata for the miner from the set command line flags
// or returns a default one composed on the client, runtime and OS metadata.
func MakeMinerExtra(extra []byte, ctx *cli.Context) []byte {
	if ctx.GlobalIsSet(ExtraDataFlag.Name) {
		return []byte(ctx.GlobalString(ExtraDataFlag.Name))
615
	}
616 617 618
	return extra
}

619
// MakePasswordList reads password lines from the file specified by --password.
620
func MakePasswordList(ctx *cli.Context) []string {
621 622 623 624 625 626 627 628 629 630 631 632
	path := ctx.GlobalString(PasswordFileFlag.Name)
	if path == "" {
		return nil
	}
	text, err := ioutil.ReadFile(path)
	if err != nil {
		Fatalf("Failed to read password file: %v", err)
	}
	lines := strings.Split(string(text), "\n")
	// Sanitise DOS line endings.
	for i := range lines {
		lines[i] = strings.TrimRight(lines[i], "\r")
633
	}
634
	return lines
635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660
}

// MakeSystemNode sets up a local node, configures the services to launch and
// assembles the P2P protocol stack.
func MakeSystemNode(name, version string, extra []byte, ctx *cli.Context) *node.Node {
	// Avoid conflicting network flags
	networks, netFlags := 0, []cli.BoolFlag{DevModeFlag, TestNetFlag, OlympicFlag}
	for _, flag := range netFlags {
		if ctx.GlobalBool(flag.Name) {
			networks++
		}
	}
	if networks > 1 {
		Fatalf("The %v flags are mutually exclusive", netFlags)
	}
	// Configure the node's service container
	stackConf := &node.Config{
		DataDir:         MustMakeDataDir(ctx),
		PrivateKey:      MakeNodeKey(ctx),
		Name:            MakeNodeName(name, version, ctx),
		NoDiscovery:     ctx.GlobalBool(NoDiscoverFlag.Name),
		BootstrapNodes:  MakeBootstrapNodes(ctx),
		ListenAddr:      MakeListenAddress(ctx),
		NAT:             MakeNAT(ctx),
		MaxPeers:        ctx.GlobalInt(MaxPeersFlag.Name),
		MaxPendingPeers: ctx.GlobalInt(MaxPendingPeersFlag.Name),
661 662 663 664
		IPCPath:         MakeIPCPath(ctx),
		HTTPHost:        MakeHTTPRpcHost(ctx),
		HTTPPort:        ctx.GlobalInt(RPCPortFlag.Name),
		HTTPCors:        ctx.GlobalString(RPCCORSDomainFlag.Name),
665
		HTTPModules:     MakeRPCModules(ctx.GlobalString(RPCApiFlag.Name)),
666 667
		WSHost:          MakeWSRpcHost(ctx),
		WSPort:          ctx.GlobalInt(WSPortFlag.Name),
B
Bas van Kervel 已提交
668
		WSOrigins:       ctx.GlobalString(WSAllowedOriginsFlag.Name),
669
		WSModules:       MakeRPCModules(ctx.GlobalString(WSApiFlag.Name)),
670 671 672 673
	}
	// Configure the Ethereum service
	accman := MakeAccountManager(ctx)

674 675 676 677 678 679 680 681 682 683
	// initialise new random number generator
	rand := rand.New(rand.NewSource(time.Now().UnixNano()))
	// get enabled jit flag
	jitEnabled := ctx.GlobalBool(VMEnableJitFlag.Name)
	// if the jit is not enabled enable it for 10 pct of the people
	if !jitEnabled && rand.Float64() < 0.1 {
		jitEnabled = true
		glog.V(logger.Info).Infoln("You're one of the lucky few that will try out the JIT VM (random). If you get a consensus failure please be so kind to report this incident with the block hash that failed. You can switch to the regular VM by setting --jitvm=false")
	}

684
	ethConf := &eth.Config{
685
		ChainConfig:             MustMakeChainConfig(ctx),
686
		Genesis:                 MakeGenesisBlock(ctx),
687
		FastSync:                ctx.GlobalBool(FastSyncFlag.Name),
Z
zsfelfoldi 已提交
688
		BlockChainVersion:       ctx.GlobalInt(BlockchainVersionFlag.Name),
689
		DatabaseCache:           ctx.GlobalInt(CacheFlag.Name),
690
		DatabaseHandles:         MakeDatabaseHandles(),
Z
zsfelfoldi 已提交
691
		NetworkId:               ctx.GlobalInt(NetworkIdFlag.Name),
692 693
		AccountManager:          accman,
		Etherbase:               MakeEtherbase(accman, ctx),
Z
zsfelfoldi 已提交
694
		MinerThreads:            ctx.GlobalInt(MinerThreadsFlag.Name),
695
		ExtraData:               MakeMinerExtra(extra, ctx),
Z
zsfelfoldi 已提交
696
		NatSpec:                 ctx.GlobalBool(NatspecEnabledFlag.Name),
Z
zelig 已提交
697
		DocRoot:                 ctx.GlobalString(DocRootFlag.Name),
698
		EnableJit:               jitEnabled,
699
		ForceJit:                ctx.GlobalBool(VMForceJitFlag.Name),
Z
zsfelfoldi 已提交
700 701 702 703 704 705 706 707 708
		GasPrice:                common.String2Big(ctx.GlobalString(GasPriceFlag.Name)),
		GpoMinGasPrice:          common.String2Big(ctx.GlobalString(GpoMinGasPriceFlag.Name)),
		GpoMaxGasPrice:          common.String2Big(ctx.GlobalString(GpoMaxGasPriceFlag.Name)),
		GpoFullBlockRatio:       ctx.GlobalInt(GpoFullBlockRatioFlag.Name),
		GpobaseStepDown:         ctx.GlobalInt(GpobaseStepDownFlag.Name),
		GpobaseStepUp:           ctx.GlobalInt(GpobaseStepUpFlag.Name),
		GpobaseCorrectionFactor: ctx.GlobalInt(GpobaseCorrectionFactorFlag.Name),
		SolcPath:                ctx.GlobalString(SolcPathFlag.Name),
		AutoDAG:                 ctx.GlobalBool(AutoDAGFlag.Name) || ctx.GlobalBool(MiningEnabledFlag.Name),
709
	}
710 711
	// Configure the Whisper service
	shhEnable := ctx.GlobalBool(WhisperEnabledFlag.Name)
712

713 714 715 716 717 718 719 720 721
	// Override any default configs in dev mode or the test net
	switch {
	case ctx.GlobalBool(OlympicFlag.Name):
		if !ctx.GlobalIsSet(NetworkIdFlag.Name) {
			ethConf.NetworkId = 1
		}
		if !ctx.GlobalIsSet(GenesisFileFlag.Name) {
			ethConf.Genesis = core.OlympicGenesisBlock()
		}
722

723 724 725 726 727 728 729 730
	case ctx.GlobalBool(TestNetFlag.Name):
		if !ctx.GlobalIsSet(NetworkIdFlag.Name) {
			ethConf.NetworkId = 2
		}
		if !ctx.GlobalIsSet(GenesisFileFlag.Name) {
			ethConf.Genesis = core.TestNetGenesisBlock()
		}
		state.StartingNonce = 1048576 // (2**20)
731

732 733 734 735
	case ctx.GlobalBool(DevModeFlag.Name):
		// Override the base network stack configs
		if !ctx.GlobalIsSet(DataDirFlag.Name) {
			stackConf.DataDir = filepath.Join(os.TempDir(), "/ethereum_dev_mode")
736 737
		}
		if !ctx.GlobalIsSet(MaxPeersFlag.Name) {
738
			stackConf.MaxPeers = 0
739 740
		}
		if !ctx.GlobalIsSet(ListenPortFlag.Name) {
741 742 743 744 745 746 747 748
			stackConf.ListenAddr = ":0"
		}
		// Override the Ethereum protocol configs
		if !ctx.GlobalIsSet(GenesisFileFlag.Name) {
			ethConf.Genesis = core.OlympicGenesisBlock()
		}
		if !ctx.GlobalIsSet(GasPriceFlag.Name) {
			ethConf.GasPrice = new(big.Int)
749 750
		}
		if !ctx.GlobalIsSet(WhisperEnabledFlag.Name) {
751
			shhEnable = true
752
		}
753
		ethConf.PowTest = true
754
	}
755 756 757 758 759
	// Assemble and return the protocol stack
	stack, err := node.New(stackConf)
	if err != nil {
		Fatalf("Failed to create the protocol stack: %v", err)
	}
760
	if err := stack.Register(func(ctx *node.ServiceContext) (node.Service, error) {
761 762 763 764 765
		return eth.New(ctx, ethConf)
	}); err != nil {
		Fatalf("Failed to register the Ethereum service: %v", err)
	}
	if shhEnable {
766
		if err := stack.Register(func(*node.ServiceContext) (node.Service, error) { return whisper.New(), nil }); err != nil {
767 768 769
			Fatalf("Failed to register the Whisper service: %v", err)
		}
	}
770

771
	return stack
772 773
}

774 775 776 777 778 779 780 781 782 783 784 785
// SetupNetwork configures the system for either the main net or some test network.
func SetupNetwork(ctx *cli.Context) {
	switch {
	case ctx.GlobalBool(OlympicFlag.Name):
		params.DurationLimit = big.NewInt(8)
		params.GenesisGasLimit = big.NewInt(3141592)
		params.MinGasLimit = big.NewInt(125000)
		params.MaximumExtraDataSize = big.NewInt(1024)
		NetworkIdFlag.Value = 0
		core.BlockReward = big.NewInt(1.5e+18)
		core.ExpDiffPeriod = big.NewInt(math.MaxInt64)
	}
786
	params.TargetGasLimit = common.String2Big(ctx.GlobalString(TargetGasLimitFlag.Name))
787 788
}

789
// MustMakeChainConfig reads the chain configuration from the database in ctx.Datadir.
790
func MustMakeChainConfig(ctx *cli.Context) *core.ChainConfig {
791
	db := MakeChainDatabase(ctx)
792 793
	defer db.Close()

794 795 796 797 798 799 800
	return MustMakeChainConfigFromDb(ctx, db)
}

// MustMakeChainConfigFromDb reads the chain configuration from the given database.
func MustMakeChainConfigFromDb(ctx *cli.Context, db ethdb.Database) *core.ChainConfig {
	genesis := core.GetBlock(db, core.GetCanonicalHash(db, 0))

801
	if genesis != nil {
802
		// Existing genesis block, use stored config if available.
803 804 805 806
		storedConfig, err := core.GetChainConfig(db, genesis.Hash())
		if err == nil {
			return storedConfig
		} else if err != core.ChainConfigNotFoundErr {
807 808 809
			Fatalf("Could not make chain configuration: %v", err)
		}
	}
810 811 812 813 814 815 816
	var homesteadBlockNo *big.Int
	if ctx.GlobalBool(TestNetFlag.Name) {
		homesteadBlockNo = params.TestNetHomesteadBlock
	} else {
		homesteadBlockNo = params.MainNetHomesteadBlock
	}
	return &core.ChainConfig{HomesteadBlock: homesteadBlockNo}
817 818
}

819 820 821 822 823 824 825
// MakeChainDatabase open an LevelDB using the flags passed to the client and will hard crash if it fails.
func MakeChainDatabase(ctx *cli.Context) ethdb.Database {
	var (
		datadir = MustMakeDataDir(ctx)
		cache   = ctx.GlobalInt(CacheFlag.Name)
		handles = MakeDatabaseHandles()
	)
826

827 828
	chainDb, err := ethdb.NewLDBDatabase(filepath.Join(datadir, "chaindata"), cache, handles)
	if err != nil {
829 830
		Fatalf("Could not open database: %v", err)
	}
831 832 833 834 835 836 837 838
	return chainDb
}

// MakeChain creates a chain manager from set command line flags.
func MakeChain(ctx *cli.Context) (chain *core.BlockChain, chainDb ethdb.Database) {
	var err error
	chainDb = MakeChainDatabase(ctx)

839
	if ctx.GlobalBool(OlympicFlag.Name) {
840
		_, err := core.WriteTestNetGenesisBlock(chainDb)
841 842 843 844
		if err != nil {
			glog.Fatalln(err)
		}
	}
845

846
	chainConfig := MustMakeChainConfigFromDb(ctx, chainDb)
847 848 849

	var eventMux event.TypeMux
	chain, err = core.NewBlockChain(chainDb, chainConfig, ethash.New(), &eventMux)
O
obscuren 已提交
850 851 852
	if err != nil {
		Fatalf("Could not start chainmanager: %v", err)
	}
853
	return chain, chainDb
854
}