flags.go 26.7 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
// MakeHTTPRpcHost creates the HTTP RPC listener interface string from the set
504
// command line flags, returning empty if the HTTP endpoint is disabled.
505
func MakeHTTPRpcHost(ctx *cli.Context) string {
506 507 508 509 510 511
	if !ctx.GlobalBool(RPCEnabledFlag.Name) {
		return ""
	}
	return ctx.GlobalString(RPCListenAddrFlag.Name)
}

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

521 522 523 524 525 526 527 528 529 530 531 532 533 534
// 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)
}

535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550
// 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
}

551 552 553
// 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
554 555
	scryptN := accounts.StandardScryptN
	scryptP := accounts.StandardScryptP
556
	if ctx.GlobalBool(LightKDFFlag.Name) {
557 558
		scryptN = accounts.LightScryptN
		scryptP = accounts.LightScryptP
559
	}
560 561 562
	datadir := MustMakeDataDir(ctx)
	keydir := MakeKeyStoreDir(datadir, ctx)
	return accounts.NewManager(keydir, scryptN, scryptP)
563 564 565 566
}

// MakeAddress converts an account specified directly as a hex encoded string or
// a key index in the key store to an internal account representation.
567
func MakeAddress(accman *accounts.Manager, account string) (a common.Address, err error) {
568 569
	// If the specified account is a valid address, return it
	if common.IsHexAddress(account) {
570
		return common.HexToAddress(account), nil
571 572 573 574
	}
	// Otherwise try to interpret the account as a keystore index
	index, err := strconv.Atoi(account)
	if err != nil {
575
		return a, fmt.Errorf("invalid account address or index %q", account)
576 577 578
	}
	hex, err := accman.AddressByIndex(index)
	if err != nil {
579
		return a, fmt.Errorf("can't get account #%d (%v)", index, err)
580
	}
581
	return common.HexToAddress(hex), nil
582 583 584 585 586 587
}

// 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 {
	accounts, _ := accman.Accounts()
588
	if !ctx.GlobalIsSet(EtherbaseFlag.Name) && len(accounts) == 0 {
589
		glog.V(logger.Error).Infoln("WARNING: No etherbase set and no accounts found as default")
590 591
		return common.Address{}
	}
592 593 594
	etherbase := ctx.GlobalString(EtherbaseFlag.Name)
	if etherbase == "" {
		return common.Address{}
595
	}
596 597
	// If the specified etherbase is a valid address, return it
	addr, err := MakeAddress(accman, etherbase)
598
	if err != nil {
599
		Fatalf("Option %q: %v", EtherbaseFlag.Name, err)
600
	}
601
	return addr
602 603 604 605 606 607 608
}

// 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))
609
	}
610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649
	return extra
}

// MakePasswordList loads up a list of password from a file specified by the
// command line flags.
func MakePasswordList(ctx *cli.Context) []string {
	if path := ctx.GlobalString(PasswordFileFlag.Name); path != "" {
		blob, err := ioutil.ReadFile(path)
		if err != nil {
			Fatalf("Failed to read password file: %v", err)
		}
		return strings.Split(string(blob), "\n")
	}
	return nil
}

// 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),
650 651 652 653 654 655 656
		IPCPath:         MakeIPCPath(ctx),
		HTTPHost:        MakeHTTPRpcHost(ctx),
		HTTPPort:        ctx.GlobalInt(RPCPortFlag.Name),
		HTTPCors:        ctx.GlobalString(RPCCORSDomainFlag.Name),
		HTTPModules:     strings.Split(ctx.GlobalString(RPCApiFlag.Name), ","),
		WSHost:          MakeWSRpcHost(ctx),
		WSPort:          ctx.GlobalInt(WSPortFlag.Name),
B
Bas van Kervel 已提交
657
		WSOrigins:       ctx.GlobalString(WSAllowedOriginsFlag.Name),
658
		WSModules:       strings.Split(ctx.GlobalString(WSApiFlag.Name), ","),
659 660 661 662
	}
	// Configure the Ethereum service
	accman := MakeAccountManager(ctx)

663 664 665 666 667 668 669 670 671 672
	// 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")
	}

673
	ethConf := &eth.Config{
674
		ChainConfig:             MustMakeChainConfig(ctx),
675
		Genesis:                 MakeGenesisBlock(ctx),
676
		FastSync:                ctx.GlobalBool(FastSyncFlag.Name),
Z
zsfelfoldi 已提交
677
		BlockChainVersion:       ctx.GlobalInt(BlockchainVersionFlag.Name),
678
		DatabaseCache:           ctx.GlobalInt(CacheFlag.Name),
679
		DatabaseHandles:         MakeDatabaseHandles(),
Z
zsfelfoldi 已提交
680
		NetworkId:               ctx.GlobalInt(NetworkIdFlag.Name),
681 682
		AccountManager:          accman,
		Etherbase:               MakeEtherbase(accman, ctx),
Z
zsfelfoldi 已提交
683
		MinerThreads:            ctx.GlobalInt(MinerThreadsFlag.Name),
684
		ExtraData:               MakeMinerExtra(extra, ctx),
Z
zsfelfoldi 已提交
685
		NatSpec:                 ctx.GlobalBool(NatspecEnabledFlag.Name),
Z
zelig 已提交
686
		DocRoot:                 ctx.GlobalString(DocRootFlag.Name),
687
		EnableJit:               jitEnabled,
688
		ForceJit:                ctx.GlobalBool(VMForceJitFlag.Name),
Z
zsfelfoldi 已提交
689 690 691 692 693 694 695 696 697
		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),
698
	}
699 700
	// Configure the Whisper service
	shhEnable := ctx.GlobalBool(WhisperEnabledFlag.Name)
701

702 703 704 705 706 707 708 709 710
	// 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()
		}
711

712 713 714 715 716 717 718 719
	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)
720

721 722 723 724
	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")
725 726
		}
		if !ctx.GlobalIsSet(MaxPeersFlag.Name) {
727
			stackConf.MaxPeers = 0
728 729
		}
		if !ctx.GlobalIsSet(ListenPortFlag.Name) {
730 731 732 733 734 735 736 737
			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)
738 739
		}
		if !ctx.GlobalIsSet(WhisperEnabledFlag.Name) {
740
			shhEnable = true
741
		}
742
		ethConf.PowTest = true
743
	}
744 745 746 747 748
	// Assemble and return the protocol stack
	stack, err := node.New(stackConf)
	if err != nil {
		Fatalf("Failed to create the protocol stack: %v", err)
	}
749
	if err := stack.Register(func(ctx *node.ServiceContext) (node.Service, error) {
750 751 752 753 754
		return eth.New(ctx, ethConf)
	}); err != nil {
		Fatalf("Failed to register the Ethereum service: %v", err)
	}
	if shhEnable {
755
		if err := stack.Register(func(*node.ServiceContext) (node.Service, error) { return whisper.New(), nil }); err != nil {
756 757 758
			Fatalf("Failed to register the Whisper service: %v", err)
		}
	}
759

760
	return stack
761 762
}

763 764 765 766 767 768 769 770 771 772 773 774
// 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)
	}
775
	params.TargetGasLimit = common.String2Big(ctx.GlobalString(TargetGasLimitFlag.Name))
776 777
}

778
// MustMakeChainConfig reads the chain configuration from the database in ctx.Datadir.
779
func MustMakeChainConfig(ctx *cli.Context) *core.ChainConfig {
780
	db := MakeChainDatabase(ctx)
781 782
	defer db.Close()

783 784 785 786 787 788 789
	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))

790
	if genesis != nil {
791
		// Existing genesis block, use stored config if available.
792 793 794 795
		storedConfig, err := core.GetChainConfig(db, genesis.Hash())
		if err == nil {
			return storedConfig
		} else if err != core.ChainConfigNotFoundErr {
796 797 798
			Fatalf("Could not make chain configuration: %v", err)
		}
	}
799 800 801 802 803 804 805
	var homesteadBlockNo *big.Int
	if ctx.GlobalBool(TestNetFlag.Name) {
		homesteadBlockNo = params.TestNetHomesteadBlock
	} else {
		homesteadBlockNo = params.MainNetHomesteadBlock
	}
	return &core.ChainConfig{HomesteadBlock: homesteadBlockNo}
806 807
}

808 809 810 811 812 813 814
// 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()
	)
815

816 817
	chainDb, err := ethdb.NewLDBDatabase(filepath.Join(datadir, "chaindata"), cache, handles)
	if err != nil {
818 819
		Fatalf("Could not open database: %v", err)
	}
820 821 822 823 824 825 826 827
	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)

828
	if ctx.GlobalBool(OlympicFlag.Name) {
829
		_, err := core.WriteTestNetGenesisBlock(chainDb)
830 831 832 833
		if err != nil {
			glog.Fatalln(err)
		}
	}
834

835
	chainConfig := MustMakeChainConfigFromDb(ctx, chainDb)
836 837 838

	var eventMux event.TypeMux
	chain, err = core.NewBlockChain(chainDb, chainConfig, ethash.New(), &eventMux)
O
obscuren 已提交
839 840 841
	if err != nil {
		Fatalf("Could not start chainmanager: %v", err)
	}
842
	return chain, chainDb
843
}