flags.go 35.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
// Package utils contains internal helper functions for go-ethereum commands.
18 19 20 21
package utils

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

F
Felix Lange 已提交
31
	"github.com/ethereum/go-ethereum/accounts"
32
	"github.com/ethereum/go-ethereum/accounts/keystore"
Z
zelig 已提交
33
	"github.com/ethereum/go-ethereum/common"
34
	"github.com/ethereum/go-ethereum/consensus/ethash"
35
	"github.com/ethereum/go-ethereum/core"
36
	"github.com/ethereum/go-ethereum/core/state"
37
	"github.com/ethereum/go-ethereum/core/vm"
38 39
	"github.com/ethereum/go-ethereum/crypto"
	"github.com/ethereum/go-ethereum/eth"
40 41
	"github.com/ethereum/go-ethereum/eth/downloader"
	"github.com/ethereum/go-ethereum/eth/gasprice"
42
	"github.com/ethereum/go-ethereum/ethdb"
43
	"github.com/ethereum/go-ethereum/ethstats"
44
	"github.com/ethereum/go-ethereum/event"
45
	"github.com/ethereum/go-ethereum/les"
46
	"github.com/ethereum/go-ethereum/log"
B
Bas van Kervel 已提交
47
	"github.com/ethereum/go-ethereum/metrics"
48
	"github.com/ethereum/go-ethereum/node"
49
	"github.com/ethereum/go-ethereum/p2p"
50
	"github.com/ethereum/go-ethereum/p2p/discover"
51
	"github.com/ethereum/go-ethereum/p2p/discv5"
52
	"github.com/ethereum/go-ethereum/p2p/nat"
53
	"github.com/ethereum/go-ethereum/p2p/netutil"
54
	"github.com/ethereum/go-ethereum/params"
55
	whisper "github.com/ethereum/go-ethereum/whisper/whisperv5"
56
	"gopkg.in/urfave/cli.v1"
57 58
)

59 60 61 62 63 64 65 66 67 68 69 70 71
var (
	CommandHelpTemplate = `{{.cmd.Name}}{{if .cmd.Subcommands}} command{{end}}{{if .cmd.Flags}} [command options]{{end}} [arguments...]
{{if .cmd.Description}}{{.cmd.Description}}
{{end}}{{if .cmd.Subcommands}}
SUBCOMMANDS:
	{{range .cmd.Subcommands}}{{.cmd.Name}}{{with .cmd.ShortName}}, {{.cmd}}{{end}}{{ "\t" }}{{.cmd.Usage}}
	{{end}}{{end}}{{if .categorizedFlags}}
{{range $idx, $categorized := .categorizedFlags}}{{$categorized.Name}} OPTIONS:
{{range $categorized.Flags}}{{"\t"}}{{.}}
{{end}}
{{end}}{{end}}`
)

72 73 74 75 76 77 78 79 80 81 82 83 84 85
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}}
`

86
	cli.CommandHelpTemplate = CommandHelpTemplate
87 88
}

F
Felix Lange 已提交
89
// NewApp creates an app with sane defaults.
90
func NewApp(gitCommit, usage string) *cli.App {
F
Felix Lange 已提交
91
	app := cli.NewApp()
92
	app.Name = filepath.Base(os.Args[0])
F
Felix Lange 已提交
93
	app.Author = ""
O
obscuren 已提交
94
	//app.Authors = nil
F
Felix Lange 已提交
95
	app.Email = ""
96
	app.Version = params.Version
97 98 99
	if gitCommit != "" {
		app.Version += "-" + gitCommit[:8]
	}
F
Felix Lange 已提交
100 101 102 103
	app.Usage = usage
	return app
}

104 105 106 107 108 109 110 111 112
// 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
113
	DataDirFlag = DirectoryFlag{
114
		Name:  "datadir",
115
		Usage: "Data directory for the databases and keystore",
116
		Value: DirectoryString{node.DefaultDataDir()},
117
	}
K
Kobi Gurkan 已提交
118 119 120 121
	KeyStoreDirFlag = DirectoryFlag{
		Name:  "keystore",
		Usage: "Directory for the keystore (default = inside the datadir)",
	}
122 123 124 125
	NoUSBFlag = cli.BoolFlag{
		Name:  "nousb",
		Usage: "Disables monitoring for and managine USB hardware wallets",
	}
126
	NetworkIdFlag = cli.Uint64Flag{
Z
zelig 已提交
127
		Name:  "networkid",
128
		Usage: "Network identifier (integer, 1=Frontier, 2=Morden (disused), 3=Ropsten, 4=Rinkeby)",
129
		Value: eth.DefaultConfig.NetworkId,
Z
zelig 已提交
130
	}
131
	TestnetFlag = cli.BoolFlag{
132
		Name:  "testnet",
133
		Usage: "Ropsten network: pre-configured proof-of-work test network",
134
	}
135 136 137 138
	RinkebyFlag = cli.BoolFlag{
		Name:  "rinkeby",
		Usage: "Rinkeby network: pre-configured proof-of-authority test network",
	}
139 140
	DevModeFlag = cli.BoolFlag{
		Name:  "dev",
141
		Usage: "Developer mode: pre-configured private network with several debugging flags",
142
	}
143 144
	IdentityFlag = cli.StringFlag{
		Name:  "identity",
145
		Usage: "Custom node name",
146
	}
Z
zelig 已提交
147 148 149
	DocRootFlag = DirectoryFlag{
		Name:  "docroot",
		Usage: "Document Root for HTTPClient file scheme",
150
		Value: DirectoryString{homeDir()},
Z
zelig 已提交
151
	}
152 153
	FastSyncFlag = cli.BoolFlag{
		Name:  "fast",
154
		Usage: "Enable fast syncing through state downloads",
155
	}
156 157 158 159
	LightModeFlag = cli.BoolFlag{
		Name:  "light",
		Usage: "Enable light client mode",
	}
160 161 162 163 164 165 166
	defaultSyncMode = eth.DefaultConfig.SyncMode
	SyncModeFlag    = TextMarshalerFlag{
		Name:  "syncmode",
		Usage: `Blockchain sync mode ("fast", "full", or "light")`,
		Value: &defaultSyncMode,
	}

167 168 169 170 171 172 173 174 175 176
	LightServFlag = cli.IntFlag{
		Name:  "lightserv",
		Usage: "Maximum percentage of time allowed for serving LES requests (0-90)",
		Value: 0,
	}
	LightPeersFlag = cli.IntFlag{
		Name:  "lightpeers",
		Usage: "Maximum number of LES client peers",
		Value: 20,
	}
177 178
	LightKDFFlag = cli.BoolFlag{
		Name:  "lightkdf",
179
		Usage: "Reduce key-derivation RAM & CPU usage at some expense of KDF strength",
180
	}
181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246
	// Ethash settings
	EthashCacheDirFlag = DirectoryFlag{
		Name:  "ethash.cachedir",
		Usage: "Directory to store the ethash verification caches (default = inside the datadir)",
	}
	EthashCachesInMemoryFlag = cli.IntFlag{
		Name:  "ethash.cachesinmem",
		Usage: "Number of recent ethash caches to keep in memory (16MB each)",
		Value: eth.DefaultConfig.EthashCachesInMem,
	}
	EthashCachesOnDiskFlag = cli.IntFlag{
		Name:  "ethash.cachesondisk",
		Usage: "Number of recent ethash caches to keep on disk (16MB each)",
		Value: eth.DefaultConfig.EthashCachesOnDisk,
	}
	EthashDatasetDirFlag = DirectoryFlag{
		Name:  "ethash.dagdir",
		Usage: "Directory to store the ethash mining DAGs (default = inside home folder)",
		Value: DirectoryString{eth.DefaultConfig.EthashDatasetDir},
	}
	EthashDatasetsInMemoryFlag = cli.IntFlag{
		Name:  "ethash.dagsinmem",
		Usage: "Number of recent ethash mining DAGs to keep in memory (1+GB each)",
		Value: eth.DefaultConfig.EthashDatasetsInMem,
	}
	EthashDatasetsOnDiskFlag = cli.IntFlag{
		Name:  "ethash.dagsondisk",
		Usage: "Number of recent ethash mining DAGs to keep on disk (1+GB each)",
		Value: eth.DefaultConfig.EthashDatasetsOnDisk,
	}
	// Transaction pool settings
	TxPoolPriceLimitFlag = cli.Uint64Flag{
		Name:  "txpool.pricelimit",
		Usage: "Minimum gas price limit to enforce for acceptance into the pool",
		Value: eth.DefaultConfig.TxPool.PriceLimit,
	}
	TxPoolPriceBumpFlag = cli.Uint64Flag{
		Name:  "txpool.pricebump",
		Usage: "Price bump percentage to replace an already existing transaction",
		Value: eth.DefaultConfig.TxPool.PriceBump,
	}
	TxPoolAccountSlotsFlag = cli.Uint64Flag{
		Name:  "txpool.accountslots",
		Usage: "Minimum number of executable transaction slots guaranteed per account",
		Value: eth.DefaultConfig.TxPool.AccountSlots,
	}
	TxPoolGlobalSlotsFlag = cli.Uint64Flag{
		Name:  "txpool.globalslots",
		Usage: "Maximum number of executable transaction slots for all accounts",
		Value: eth.DefaultConfig.TxPool.GlobalSlots,
	}
	TxPoolAccountQueueFlag = cli.Uint64Flag{
		Name:  "txpool.accountqueue",
		Usage: "Maximum number of non-executable transaction slots permitted per account",
		Value: eth.DefaultConfig.TxPool.AccountQueue,
	}
	TxPoolGlobalQueueFlag = cli.Uint64Flag{
		Name:  "txpool.globalqueue",
		Usage: "Maximum number of non-executable transaction slots for all accounts",
		Value: eth.DefaultConfig.TxPool.GlobalQueue,
	}
	TxPoolLifetimeFlag = cli.DurationFlag{
		Name:  "txpool.lifetime",
		Usage: "Maximum amount of time non-executable transaction are queued",
		Value: eth.DefaultConfig.TxPool.Lifetime,
	}
247 248 249 250 251 252 253 254 255 256 257
	// Performance tuning settings
	CacheFlag = cli.IntFlag{
		Name:  "cache",
		Usage: "Megabytes of memory allocated to internal caching (min 16MB / database forced)",
		Value: 128,
	}
	TrieCacheGenFlag = cli.IntFlag{
		Name:  "trie-cache-gens",
		Usage: "Number of trie node generations to keep in memory",
		Value: int(state.MaxTrieCacheGen),
	}
258 259 260 261
	// Miner settings
	MiningEnabledFlag = cli.BoolFlag{
		Name:  "mine",
		Usage: "Enable mining",
262
	}
263 264
	MinerThreadsFlag = cli.IntFlag{
		Name:  "minerthreads",
265
		Usage: "Number of CPU threads to use for mining",
266 267
		Value: runtime.NumCPU(),
	}
268
	TargetGasLimitFlag = cli.Uint64Flag{
269 270
		Name:  "targetgaslimit",
		Usage: "Target gas limit sets the artificial target gas floor for the blocks to mine",
271
		Value: params.GenesisGasLimit.Uint64(),
272
	}
Z
zelig 已提交
273
	EtherbaseFlag = cli.StringFlag{
O
obscuren 已提交
274
		Name:  "etherbase",
275
		Usage: "Public address for block mining rewards (default = first account created)",
276
		Value: "0",
Z
zelig 已提交
277
	}
278
	GasPriceFlag = BigFlag{
279
		Name:  "gasprice",
280
		Usage: "Minimal gas price to accept for mining a transactions",
281
		Value: eth.DefaultConfig.GasPrice,
282
	}
Z
zelig 已提交
283 284
	ExtraDataFlag = cli.StringFlag{
		Name:  "extradata",
285
		Usage: "Block extra data set by the miner (default = client version)",
Z
zelig 已提交
286
	}
287
	// Account settings
Z
zelig 已提交
288 289
	UnlockedAccountFlag = cli.StringFlag{
		Name:  "unlock",
290
		Usage: "Comma separated list of accounts to unlock",
Z
zelig 已提交
291 292 293 294
		Value: "",
	}
	PasswordFileFlag = cli.StringFlag{
		Name:  "password",
295
		Usage: "Password file to use for non-inteactive password input",
Z
zelig 已提交
296
		Value: "",
Z
zelig 已提交
297
	}
298

299 300 301 302
	VMEnableDebugFlag = cli.BoolFlag{
		Name:  "vmdebug",
		Usage: "Record information useful for VM and contract debugging",
	}
303 304 305 306 307
	// Logging and debug settings
	EthStatsURLFlag = cli.StringFlag{
		Name:  "ethstats",
		Usage: "Reporting URL of a ethstats service (nodename:secret@host:port)",
	}
308 309
	MetricsEnabledFlag = cli.BoolFlag{
		Name:  metrics.MetricsEnabledFlag,
310
		Usage: "Enable metrics collection and reporting",
311
	}
312 313 314 315
	FakePoWFlag = cli.BoolFlag{
		Name:  "fakepow",
		Usage: "Disables proof-of-work verification",
	}
316 317 318 319
	NoCompactionFlag = cli.BoolFlag{
		Name:  "nocompaction",
		Usage: "Disables db compaction after import",
	}
320 321 322
	// RPC settings
	RPCEnabledFlag = cli.BoolFlag{
		Name:  "rpc",
323
		Usage: "Enable the HTTP-RPC server",
324 325 326
	}
	RPCListenAddrFlag = cli.StringFlag{
		Name:  "rpcaddr",
327
		Usage: "HTTP-RPC server listening interface",
328
		Value: node.DefaultHTTPHost,
329 330 331
	}
	RPCPortFlag = cli.IntFlag{
		Name:  "rpcport",
332
		Usage: "HTTP-RPC server listening port",
333
		Value: node.DefaultHTTPPort,
334
	}
335 336
	RPCCORSDomainFlag = cli.StringFlag{
		Name:  "rpccorsdomain",
337
		Usage: "Comma separated list of domains from which to accept cross origin requests (browser enforced)",
338 339
		Value: "",
	}
340
	RPCApiFlag = cli.StringFlag{
341
		Name:  "rpcapi",
342
		Usage: "API's offered over the HTTP-RPC interface",
343
		Value: "",
344
	}
B
Bas van Kervel 已提交
345 346 347 348 349 350
	IPCDisabledFlag = cli.BoolFlag{
		Name:  "ipcdisable",
		Usage: "Disable the IPC-RPC server",
	}
	IPCPathFlag = DirectoryFlag{
		Name:  "ipcpath",
351
		Usage: "Filename for IPC socket/pipe within the datadir (explicit paths escape it)",
B
Bas van Kervel 已提交
352
	}
353
	WSEnabledFlag = cli.BoolFlag{
354
		Name:  "ws",
355 356 357 358 359
		Usage: "Enable the WS-RPC server",
	}
	WSListenAddrFlag = cli.StringFlag{
		Name:  "wsaddr",
		Usage: "WS-RPC server listening interface",
360
		Value: node.DefaultWSHost,
361 362 363 364
	}
	WSPortFlag = cli.IntFlag{
		Name:  "wsport",
		Usage: "WS-RPC server listening port",
365
		Value: node.DefaultWSPort,
366 367 368 369
	}
	WSApiFlag = cli.StringFlag{
		Name:  "wsapi",
		Usage: "API's offered over the WS-RPC interface",
370
		Value: "",
371
	}
B
Bas van Kervel 已提交
372 373 374
	WSAllowedOriginsFlag = cli.StringFlag{
		Name:  "wsorigins",
		Usage: "Origins from which to accept websockets requests",
375
		Value: "",
376
	}
377 378
	ExecFlag = cli.StringFlag{
		Name:  "exec",
379
		Usage: "Execute JavaScript statement",
380
	}
381
	PreloadJSFlag = cli.StringFlag{
382 383 384
		Name:  "preload",
		Usage: "Comma separated list of JavaScript files to preload into the console",
	}
385

386 387 388
	// Network Settings
	MaxPeersFlag = cli.IntFlag{
		Name:  "maxpeers",
389
		Usage: "Maximum number of network peers (network disabled if set to 0)",
390
		Value: 25,
391
	}
392 393 394 395 396
	MaxPendingPeersFlag = cli.IntFlag{
		Name:  "maxpendpeers",
		Usage: "Maximum number of pending connection attempts (defaults used if set to 0)",
		Value: 0,
	}
397 398 399 400 401
	ListenPortFlag = cli.IntFlag{
		Name:  "port",
		Usage: "Network listening port",
		Value: 30303,
	}
402
	BootnodesFlag = cli.StringFlag{
403
		Name:  "bootnodes",
404 405 406 407 408 409 410 411 412 413 414
		Usage: "Comma separated enode URLs for P2P discovery bootstrap (set v4+v5 instead for light servers)",
		Value: "",
	}
	BootnodesV4Flag = cli.StringFlag{
		Name:  "bootnodesv4",
		Usage: "Comma separated enode URLs for P2P v4 discovery bootstrap (light server, full nodes)",
		Value: "",
	}
	BootnodesV5Flag = cli.StringFlag{
		Name:  "bootnodesv5",
		Usage: "Comma separated enode URLs for P2P v5 discovery bootstrap (light server, light nodes)",
415
		Value: "",
416 417 418 419 420 421 422 423 424 425 426
	}
	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",
427
		Usage: "NAT port mapping mechanism (any|none|upnp|pmp|extip:<IP>)",
428 429
		Value: "any",
	}
430 431 432 433
	NoDiscoverFlag = cli.BoolFlag{
		Name:  "nodiscover",
		Usage: "Disables the peer discovery mechanism (manual peer addition)",
	}
434 435 436 437
	DiscoveryV5Flag = cli.BoolFlag{
		Name:  "v5disc",
		Usage: "Enables the experimental RLPx V5 (Topic Discovery) mechanism",
	}
438 439 440 441 442
	NetrestrictFlag = cli.StringFlag{
		Name:  "netrestrict",
		Usage: "Restricts network communication to the given IP networks (CIDR masks)",
	}

443
	// ATM the url is left to the user and deployment to
Z
CLI:  
zelig 已提交
444 445
	JSpathFlag = cli.StringFlag{
		Name:  "jspath",
446
		Usage: "JavaScript root path for `loadScript`",
Z
CLI:  
zelig 已提交
447 448
		Value: ".",
	}
449 450

	// Gas price oracle settings
451 452 453
	GpoBlocksFlag = cli.IntFlag{
		Name:  "gpoblocks",
		Usage: "Number of recent blocks to check for gas prices",
454
		Value: eth.DefaultConfig.GPO.Blocks,
Z
zsfelfoldi 已提交
455
	}
456 457 458
	GpoPercentileFlag = cli.IntFlag{
		Name:  "gpopercentile",
		Usage: "Suggested gas price is the given percentile of a set of recent transaction gas prices",
459
		Value: eth.DefaultConfig.GPO.Percentile,
Z
zsfelfoldi 已提交
460
	}
461 462 463 464 465 466 467 468 469 470 471 472 473 474
	WhisperEnabledFlag = cli.BoolFlag{
		Name:  "shh",
		Usage: "Enable Whisper",
	}
	WhisperMaxMessageSizeFlag = cli.IntFlag{
		Name:  "shh.maxmessagesize",
		Usage: "Max message size accepted",
		Value: int(whisper.DefaultMaxMessageSize),
	}
	WhisperMinPOWFlag = cli.Float64Flag{
		Name:  "shh.pow",
		Usage: "Minimum POW accepted",
		Value: whisper.DefaultMinimumPoW,
	}
475 476
)

477
// MakeDataDir retrieves the currently requested data directory, terminating
478 479
// 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.
480
func MakeDataDir(ctx *cli.Context) string {
481
	if path := ctx.GlobalString(DataDirFlag.Name); path != "" {
482
		if ctx.GlobalBool(TestnetFlag.Name) {
483
			return filepath.Join(path, "testnet")
484
		}
485 486 487
		if ctx.GlobalBool(RinkebyFlag.Name) {
			return filepath.Join(path, "rinkeby")
		}
488
		return path
489
	}
490
	Fatalf("Cannot determine default data directory, please set manually (--datadir)")
491
	return ""
492 493
}

494
// setNodeKey creates a node key from set command line flags, either loading it
495 496
// 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.
497
func setNodeKey(ctx *cli.Context, cfg *p2p.Config) {
498 499 500
	var (
		hex  = ctx.GlobalString(NodeKeyHexFlag.Name)
		file = ctx.GlobalString(NodeKeyFileFlag.Name)
501 502
		key  *ecdsa.PrivateKey
		err  error
503
	)
504 505
	switch {
	case file != "" && hex != "":
506
		Fatalf("Options %q and %q are mutually exclusive", NodeKeyFileFlag.Name, NodeKeyHexFlag.Name)
507 508
	case file != "":
		if key, err = crypto.LoadECDSA(file); err != nil {
509
			Fatalf("Option %q: %v", NodeKeyFileFlag.Name, err)
510
		}
511
		cfg.PrivateKey = key
512 513
	case hex != "":
		if key, err = crypto.HexToECDSA(hex); err != nil {
514
			Fatalf("Option %q: %v", NodeKeyHexFlag.Name, err)
515
		}
516
		cfg.PrivateKey = key
517 518 519
	}
}

520 521
// setNodeUserIdent creates the user identifier from CLI flags.
func setNodeUserIdent(ctx *cli.Context, cfg *node.Config) {
522
	if identity := ctx.GlobalString(IdentityFlag.Name); len(identity) > 0 {
523
		cfg.UserIdent = identity
524 525 526
	}
}

527
// setBootstrapNodes creates a list of bootstrap nodes from the command line
528
// flags, reverting to pre-configured ones if none have been specified.
529
func setBootstrapNodes(ctx *cli.Context, cfg *p2p.Config) {
530
	urls := params.MainnetBootnodes
531
	switch {
532 533 534 535 536 537
	case ctx.GlobalIsSet(BootnodesFlag.Name) || ctx.GlobalIsSet(BootnodesV4Flag.Name):
		if ctx.GlobalIsSet(BootnodesV4Flag.Name) {
			urls = strings.Split(ctx.GlobalString(BootnodesV4Flag.Name), ",")
		} else {
			urls = strings.Split(ctx.GlobalString(BootnodesFlag.Name), ",")
		}
538
	case ctx.GlobalBool(TestnetFlag.Name):
539
		urls = params.TestnetBootnodes
540 541
	case ctx.GlobalBool(RinkebyFlag.Name):
		urls = params.RinkebyBootnodes
542 543
	}

544
	cfg.BootstrapNodes = make([]*discover.Node, 0, len(urls))
545
	for _, url := range urls {
546 547
		node, err := discover.ParseNode(url)
		if err != nil {
P
Péter Szilágyi 已提交
548
			log.Error("Bootstrap URL invalid", "enode", url, "err", err)
549 550
			continue
		}
551
		cfg.BootstrapNodes = append(cfg.BootstrapNodes, node)
552 553 554
	}
}

555
// setBootstrapNodesV5 creates a list of bootstrap nodes from the command line
556
// flags, reverting to pre-configured ones if none have been specified.
557
func setBootstrapNodesV5(ctx *cli.Context, cfg *p2p.Config) {
558
	urls := params.DiscoveryV5Bootnodes
559
	switch {
560 561 562 563 564 565
	case ctx.GlobalIsSet(BootnodesFlag.Name) || ctx.GlobalIsSet(BootnodesV5Flag.Name):
		if ctx.GlobalIsSet(BootnodesV5Flag.Name) {
			urls = strings.Split(ctx.GlobalString(BootnodesV5Flag.Name), ",")
		} else {
			urls = strings.Split(ctx.GlobalString(BootnodesFlag.Name), ",")
		}
566 567 568
	case ctx.GlobalBool(RinkebyFlag.Name):
		urls = params.RinkebyV5Bootnodes
	case cfg.BootstrapNodesV5 != nil:
569
		return // already set, don't apply defaults.
570 571
	}

572
	cfg.BootstrapNodesV5 = make([]*discv5.Node, 0, len(urls))
573
	for _, url := range urls {
574 575
		node, err := discv5.ParseNode(url)
		if err != nil {
P
Péter Szilágyi 已提交
576
			log.Error("Bootstrap URL invalid", "enode", url, "err", err)
577 578
			continue
		}
579
		cfg.BootstrapNodesV5 = append(cfg.BootstrapNodesV5, node)
580 581 582
	}
}

583
// setListenAddress creates a TCP listening address string from set command
584
// line flags.
585 586 587 588
func setListenAddress(ctx *cli.Context, cfg *p2p.Config) {
	if ctx.GlobalIsSet(ListenPortFlag.Name) {
		cfg.ListenAddr = fmt.Sprintf(":%d", ctx.GlobalInt(ListenPortFlag.Name))
	}
589 590
}

591
// setDiscoveryV5Address creates a UDP listening address string from set command
592
// line flags for the V5 discovery protocol.
593 594 595 596
func setDiscoveryV5Address(ctx *cli.Context, cfg *p2p.Config) {
	if ctx.GlobalIsSet(ListenPortFlag.Name) {
		cfg.DiscoveryV5Addr = fmt.Sprintf(":%d", ctx.GlobalInt(ListenPortFlag.Name)+1)
	}
597 598
}

599 600 601 602 603 604 605 606
// setNAT creates a port mapper from command line flags.
func setNAT(ctx *cli.Context, cfg *p2p.Config) {
	if ctx.GlobalIsSet(NATFlag.Name) {
		natif, err := nat.Parse(ctx.GlobalString(NATFlag.Name))
		if err != nil {
			Fatalf("Option %s: %v", NATFlag.Name, err)
		}
		cfg.NAT = natif
607 608 609
	}
}

610 611 612
// splitAndTrim splits input separated by a comma
// and trims excessive white space from the substrings.
func splitAndTrim(input string) []string {
613 614 615 616 617 618 619
	result := strings.Split(input, ",")
	for i, r := range result {
		result[i] = strings.TrimSpace(r)
	}
	return result
}

620
// setHTTP creates the HTTP RPC listener interface string from the set
621
// command line flags, returning empty if the HTTP endpoint is disabled.
622 623 624 625 626 627 628 629 630 631 632 633
func setHTTP(ctx *cli.Context, cfg *node.Config) {
	if ctx.GlobalBool(RPCEnabledFlag.Name) && cfg.HTTPHost == "" {
		cfg.HTTPHost = "127.0.0.1"
		if ctx.GlobalIsSet(RPCListenAddrFlag.Name) {
			cfg.HTTPHost = ctx.GlobalString(RPCListenAddrFlag.Name)
		}
	}

	if ctx.GlobalIsSet(RPCPortFlag.Name) {
		cfg.HTTPPort = ctx.GlobalInt(RPCPortFlag.Name)
	}
	if ctx.GlobalIsSet(RPCCORSDomainFlag.Name) {
634
		cfg.HTTPCors = splitAndTrim(ctx.GlobalString(RPCCORSDomainFlag.Name))
635 636
	}
	if ctx.GlobalIsSet(RPCApiFlag.Name) {
637
		cfg.HTTPModules = splitAndTrim(ctx.GlobalString(RPCApiFlag.Name))
638 639 640
	}
}

641
// setWS creates the WebSocket RPC listener interface string from the set
642
// command line flags, returning empty if the HTTP endpoint is disabled.
643 644 645 646 647 648 649 650 651 652 653 654
func setWS(ctx *cli.Context, cfg *node.Config) {
	if ctx.GlobalBool(WSEnabledFlag.Name) && cfg.WSHost == "" {
		cfg.WSHost = "127.0.0.1"
		if ctx.GlobalIsSet(WSListenAddrFlag.Name) {
			cfg.WSHost = ctx.GlobalString(WSListenAddrFlag.Name)
		}
	}

	if ctx.GlobalIsSet(WSPortFlag.Name) {
		cfg.WSPort = ctx.GlobalInt(WSPortFlag.Name)
	}
	if ctx.GlobalIsSet(WSAllowedOriginsFlag.Name) {
655
		cfg.WSOrigins = splitAndTrim(ctx.GlobalString(WSAllowedOriginsFlag.Name))
656 657
	}
	if ctx.GlobalIsSet(WSApiFlag.Name) {
658
		cfg.WSModules = splitAndTrim(ctx.GlobalString(WSApiFlag.Name))
659 660 661 662 663 664 665 666 667 668 669 670
	}
}

// setIPC creates an IPC path configuration from the set command line flags,
// returning an empty string if IPC was explicitly disabled, or the set path.
func setIPC(ctx *cli.Context, cfg *node.Config) {
	checkExclusive(ctx, IPCDisabledFlag, IPCPathFlag)
	switch {
	case ctx.GlobalBool(IPCDisabledFlag.Name):
		cfg.IPCPath = ""
	case ctx.GlobalIsSet(IPCPathFlag.Name):
		cfg.IPCPath = ctx.GlobalString(IPCPathFlag.Name)
671 672 673
	}
}

674
// makeDatabaseHandles raises out the number of allowed file handles per process
675
// for Geth and returns half of the allowance to assign to the database.
676
func makeDatabaseHandles() int {
677
	if err := raiseFdLimit(2048); err != nil {
678
		Fatalf("Failed to raise file descriptor allowance: %v", err)
679 680 681
	}
	limit, err := getFdLimit()
	if err != nil {
682
		Fatalf("Failed to retrieve file descriptor allowance: %v", err)
683 684 685 686 687 688 689
	}
	if limit > 2048 { // cap database file descriptors even if more is available
		limit = 2048
	}
	return limit / 2 // Leave half for networking and other stuff
}

690 691
// MakeAddress converts an account specified directly as a hex encoded string or
// a key index in the key store to an internal account representation.
692
func MakeAddress(ks *keystore.KeyStore, account string) (accounts.Account, error) {
693 694
	// If the specified account is a valid address, return it
	if common.IsHexAddress(account) {
F
Felix Lange 已提交
695
		return accounts.Account{Address: common.HexToAddress(account)}, nil
696 697 698
	}
	// Otherwise try to interpret the account as a keystore index
	index, err := strconv.Atoi(account)
699
	if err != nil || index < 0 {
F
Felix Lange 已提交
700
		return accounts.Account{}, fmt.Errorf("invalid account address or index %q", account)
701
	}
702 703 704 705 706
	accs := ks.Accounts()
	if len(accs) <= index {
		return accounts.Account{}, fmt.Errorf("index %d higher than number of accounts %d", index, len(accs))
	}
	return accs[index], nil
707 708
}

709
// setEtherbase retrieves the etherbase either from the directly specified
710
// command line flags or from the keystore if CLI indexed.
711 712 713 714 715 716 717 718
func setEtherbase(ctx *cli.Context, ks *keystore.KeyStore, cfg *eth.Config) {
	if ctx.GlobalIsSet(EtherbaseFlag.Name) {
		account, err := MakeAddress(ks, ctx.GlobalString(EtherbaseFlag.Name))
		if err != nil {
			Fatalf("Option %q: %v", EtherbaseFlag.Name, err)
		}
		cfg.Etherbase = account.Address
		return
719
	}
720 721 722 723 724 725 726
	accounts := ks.Accounts()
	if (cfg.Etherbase == common.Address{}) {
		if len(accounts) > 0 {
			cfg.Etherbase = accounts[0].Address
		} else {
			log.Warn("No etherbase set and no accounts found as default")
		}
727
	}
728 729
}

730
// MakePasswordList reads password lines from the file specified by the global --password flag.
731
func MakePasswordList(ctx *cli.Context) []string {
732 733 734 735 736 737
	path := ctx.GlobalString(PasswordFileFlag.Name)
	if path == "" {
		return nil
	}
	text, err := ioutil.ReadFile(path)
	if err != nil {
738
		Fatalf("Failed to read password file: %v", err)
739 740 741 742 743
	}
	lines := strings.Split(string(text), "\n")
	// Sanitise DOS line endings.
	for i := range lines {
		lines[i] = strings.TrimRight(lines[i], "\r")
744
	}
745
	return lines
746 747
}

748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763
func SetP2PConfig(ctx *cli.Context, cfg *p2p.Config) {
	setNodeKey(ctx, cfg)
	setNAT(ctx, cfg)
	setListenAddress(ctx, cfg)
	setDiscoveryV5Address(ctx, cfg)
	setBootstrapNodes(ctx, cfg)
	setBootstrapNodesV5(ctx, cfg)

	if ctx.GlobalIsSet(MaxPeersFlag.Name) {
		cfg.MaxPeers = ctx.GlobalInt(MaxPeersFlag.Name)
	}
	if ctx.GlobalIsSet(MaxPendingPeersFlag.Name) {
		cfg.MaxPendingPeers = ctx.GlobalInt(MaxPendingPeersFlag.Name)
	}
	if ctx.GlobalIsSet(NoDiscoverFlag.Name) || ctx.GlobalBool(LightModeFlag.Name) {
		cfg.NoDiscovery = true
764 765
	}

766 767 768
	// if we're running a light client or server, force enable the v5 peer discovery
	// unless it is explicitly disabled with --nodiscover note that explicitly specifying
	// --v5disc overrides --nodiscover, in which case the later only disables v4 discovery
769
	forceV5Discovery := (ctx.GlobalBool(LightModeFlag.Name) || ctx.GlobalInt(LightServFlag.Name) > 0) && !ctx.GlobalBool(NoDiscoverFlag.Name)
770 771 772 773
	if ctx.GlobalIsSet(DiscoveryV5Flag.Name) {
		cfg.DiscoveryV5 = ctx.GlobalBool(DiscoveryV5Flag.Name)
	} else if forceV5Discovery {
		cfg.DiscoveryV5 = true
774
	}
775

776 777 778
	if netrestrict := ctx.GlobalString(NetrestrictFlag.Name); netrestrict != "" {
		list, err := netutil.ParseNetlist(netrestrict)
		if err != nil {
779
			Fatalf("Option %q: %v", NetrestrictFlag.Name, err)
780
		}
781
		cfg.NetRestrict = list
782 783
	}

784 785 786 787
	if ctx.GlobalBool(DevModeFlag.Name) {
		// --dev mode can't use p2p networking.
		cfg.MaxPeers = 0
		cfg.ListenAddr = ":0"
788
		cfg.DiscoveryV5Addr = ":0"
789 790
		cfg.NoDiscovery = true
		cfg.DiscoveryV5 = false
791 792 793
	}
}

794 795 796 797 798 799 800 801 802 803 804 805 806
// SetNodeConfig applies node-related command line flags to the config.
func SetNodeConfig(ctx *cli.Context, cfg *node.Config) {
	SetP2PConfig(ctx, &cfg.P2P)
	setIPC(ctx, cfg)
	setHTTP(ctx, cfg)
	setWS(ctx, cfg)
	setNodeUserIdent(ctx, cfg)

	switch {
	case ctx.GlobalIsSet(DataDirFlag.Name):
		cfg.DataDir = ctx.GlobalString(DataDirFlag.Name)
	case ctx.GlobalBool(DevModeFlag.Name):
		cfg.DataDir = filepath.Join(os.TempDir(), "ethereum_dev_mode")
807
	case ctx.GlobalBool(TestnetFlag.Name):
808
		cfg.DataDir = filepath.Join(node.DefaultDataDir(), "testnet")
809 810
	case ctx.GlobalBool(RinkebyFlag.Name):
		cfg.DataDir = filepath.Join(node.DefaultDataDir(), "rinkeby")
811 812 813 814 815 816 817 818
	}

	if ctx.GlobalIsSet(KeyStoreDirFlag.Name) {
		cfg.KeyStoreDir = ctx.GlobalString(KeyStoreDirFlag.Name)
	}
	if ctx.GlobalIsSet(LightKDFFlag.Name) {
		cfg.UseLightweightKDF = ctx.GlobalBool(LightKDFFlag.Name)
	}
819 820 821
	if ctx.GlobalIsSet(NoUSBFlag.Name) {
		cfg.NoUSB = ctx.GlobalBool(NoUSBFlag.Name)
	}
822 823 824 825 826 827 828 829 830 831 832
}

func setGPO(ctx *cli.Context, cfg *gasprice.Config) {
	if ctx.GlobalIsSet(GpoBlocksFlag.Name) {
		cfg.Blocks = ctx.GlobalInt(GpoBlocksFlag.Name)
	}
	if ctx.GlobalIsSet(GpoPercentileFlag.Name) {
		cfg.Percentile = ctx.GlobalInt(GpoPercentileFlag.Name)
	}
}

833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856
func setTxPool(ctx *cli.Context, cfg *core.TxPoolConfig) {
	if ctx.GlobalIsSet(TxPoolPriceLimitFlag.Name) {
		cfg.PriceLimit = ctx.GlobalUint64(TxPoolPriceLimitFlag.Name)
	}
	if ctx.GlobalIsSet(TxPoolPriceBumpFlag.Name) {
		cfg.PriceBump = ctx.GlobalUint64(TxPoolPriceBumpFlag.Name)
	}
	if ctx.GlobalIsSet(TxPoolAccountSlotsFlag.Name) {
		cfg.AccountSlots = ctx.GlobalUint64(TxPoolAccountSlotsFlag.Name)
	}
	if ctx.GlobalIsSet(TxPoolGlobalSlotsFlag.Name) {
		cfg.GlobalSlots = ctx.GlobalUint64(TxPoolGlobalSlotsFlag.Name)
	}
	if ctx.GlobalIsSet(TxPoolAccountQueueFlag.Name) {
		cfg.AccountQueue = ctx.GlobalUint64(TxPoolAccountQueueFlag.Name)
	}
	if ctx.GlobalIsSet(TxPoolGlobalQueueFlag.Name) {
		cfg.GlobalQueue = ctx.GlobalUint64(TxPoolGlobalQueueFlag.Name)
	}
	if ctx.GlobalIsSet(TxPoolLifetimeFlag.Name) {
		cfg.Lifetime = ctx.GlobalDuration(TxPoolLifetimeFlag.Name)
	}
}

857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882
func setEthash(ctx *cli.Context, cfg *eth.Config) {
	if ctx.GlobalIsSet(EthashCacheDirFlag.Name) {
		cfg.EthashCacheDir = ctx.GlobalString(EthashCacheDirFlag.Name)
	}
	if ctx.GlobalIsSet(EthashDatasetDirFlag.Name) {
		cfg.EthashDatasetDir = ctx.GlobalString(EthashDatasetDirFlag.Name)
	}
	if ctx.GlobalIsSet(EthashCachesInMemoryFlag.Name) {
		cfg.EthashCachesInMem = ctx.GlobalInt(EthashCachesInMemoryFlag.Name)
	}
	if ctx.GlobalIsSet(EthashCachesOnDiskFlag.Name) {
		cfg.EthashCachesOnDisk = ctx.GlobalInt(EthashCachesOnDiskFlag.Name)
	}
	if ctx.GlobalIsSet(EthashDatasetsInMemoryFlag.Name) {
		cfg.EthashDatasetsInMem = ctx.GlobalInt(EthashDatasetsInMemoryFlag.Name)
	}
	if ctx.GlobalIsSet(EthashDatasetsOnDiskFlag.Name) {
		cfg.EthashDatasetsOnDisk = ctx.GlobalInt(EthashDatasetsOnDiskFlag.Name)
	}
}

func checkExclusive(ctx *cli.Context, flags ...cli.Flag) {
	set := make([]string, 0, 1)
	for _, flag := range flags {
		if ctx.GlobalIsSet(flag.GetName()) {
			set = append(set, "--"+flag.GetName())
883 884
		}
	}
885 886
	if len(set) > 1 {
		Fatalf("flags %v can't be used at the same time", strings.Join(set, ", "))
887
	}
888 889
}

890 891
// SetShhConfig applies shh-related command line flags to the config.
func SetShhConfig(ctx *cli.Context, stack *node.Node, cfg *whisper.Config) {
892 893
	if ctx.GlobalIsSet(WhisperMaxMessageSizeFlag.Name) {
		cfg.MaxMessageSize = uint32(ctx.GlobalUint(WhisperMaxMessageSizeFlag.Name))
894
	}
895 896
	if ctx.GlobalIsSet(WhisperMinPOWFlag.Name) {
		cfg.MinimumAcceptedPOW = ctx.GlobalFloat64(WhisperMinPOWFlag.Name)
897 898 899
	}
}

900 901 902
// SetEthConfig applies eth-related command line flags to the config.
func SetEthConfig(ctx *cli.Context, stack *node.Node, cfg *eth.Config) {
	// Avoid conflicting network flags
903
	checkExclusive(ctx, DevModeFlag, TestnetFlag, RinkebyFlag)
904 905
	checkExclusive(ctx, FastSyncFlag, LightModeFlag, SyncModeFlag)

906
	ks := stack.AccountManager().Backends(keystore.KeyStoreType)[0].(*keystore.KeyStore)
907 908
	setEtherbase(ctx, ks, cfg)
	setGPO(ctx, &cfg.GPO)
909
	setTxPool(ctx, &cfg.TxPool)
910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926
	setEthash(ctx, cfg)

	switch {
	case ctx.GlobalIsSet(SyncModeFlag.Name):
		cfg.SyncMode = *GlobalTextMarshaler(ctx, SyncModeFlag.Name).(*downloader.SyncMode)
	case ctx.GlobalBool(FastSyncFlag.Name):
		cfg.SyncMode = downloader.FastSync
	case ctx.GlobalBool(LightModeFlag.Name):
		cfg.SyncMode = downloader.LightSync
	}
	if ctx.GlobalIsSet(LightServFlag.Name) {
		cfg.LightServ = ctx.GlobalInt(LightServFlag.Name)
	}
	if ctx.GlobalIsSet(LightPeersFlag.Name) {
		cfg.LightPeers = ctx.GlobalInt(LightPeersFlag.Name)
	}
	if ctx.GlobalIsSet(NetworkIdFlag.Name) {
927
		cfg.NetworkId = ctx.GlobalUint64(NetworkIdFlag.Name)
928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954
	}

	// Ethereum needs to know maxPeers to calculate the light server peer ratio.
	// TODO(fjl): ensure Ethereum can get MaxPeers from node.
	cfg.MaxPeers = ctx.GlobalInt(MaxPeersFlag.Name)

	if ctx.GlobalIsSet(CacheFlag.Name) {
		cfg.DatabaseCache = ctx.GlobalInt(CacheFlag.Name)
	}
	cfg.DatabaseHandles = makeDatabaseHandles()

	if ctx.GlobalIsSet(MinerThreadsFlag.Name) {
		cfg.MinerThreads = ctx.GlobalInt(MinerThreadsFlag.Name)
	}
	if ctx.GlobalIsSet(DocRootFlag.Name) {
		cfg.DocRoot = ctx.GlobalString(DocRootFlag.Name)
	}
	if ctx.GlobalIsSet(ExtraDataFlag.Name) {
		cfg.ExtraData = []byte(ctx.GlobalString(ExtraDataFlag.Name))
	}
	if ctx.GlobalIsSet(GasPriceFlag.Name) {
		cfg.GasPrice = GlobalBig(ctx, GasPriceFlag.Name)
	}
	if ctx.GlobalIsSet(VMEnableDebugFlag.Name) {
		// TODO(fjl): force-enable this in --dev mode
		cfg.EnablePreimageRecording = ctx.GlobalBool(VMEnableDebugFlag.Name)
	}
955

956
	// Override any default configs for hard coded networks.
957
	switch {
958
	case ctx.GlobalBool(TestnetFlag.Name):
959
		if !ctx.GlobalIsSet(NetworkIdFlag.Name) {
960
			cfg.NetworkId = 3
961
		}
962
		cfg.Genesis = core.DefaultTestnetGenesisBlock()
963 964 965 966 967
	case ctx.GlobalBool(RinkebyFlag.Name):
		if !ctx.GlobalIsSet(NetworkIdFlag.Name) {
			cfg.NetworkId = 4
		}
		cfg.Genesis = core.DefaultRinkebyGenesisBlock()
968
	case ctx.GlobalBool(DevModeFlag.Name):
969
		cfg.Genesis = core.DevGenesisBlock()
970
		if !ctx.GlobalIsSet(GasPriceFlag.Name) {
971
			cfg.GasPrice = new(big.Int)
972
		}
973
		cfg.PowTest = true
974
	}
975 976

	// TODO(fjl): move trie cache generations into config
977 978 979
	if gen := ctx.GlobalInt(TrieCacheGenFlag.Name); gen > 0 {
		state.MaxTrieCacheGen = uint16(gen)
	}
980
}
981

982 983 984 985 986 987 988
// RegisterEthService adds an Ethereum client to the stack.
func RegisterEthService(stack *node.Node, cfg *eth.Config) {
	var err error
	if cfg.SyncMode == downloader.LightSync {
		err = stack.Register(func(ctx *node.ServiceContext) (node.Service, error) {
			return les.New(ctx, cfg)
		})
989
	} else {
990 991 992 993
		err = stack.Register(func(ctx *node.ServiceContext) (node.Service, error) {
			fullNode, err := eth.New(ctx, cfg)
			if fullNode != nil && cfg.LightServ > 0 {
				ls, _ := les.NewLesServer(fullNode, cfg)
994 995 996
				fullNode.AddLesServer(ls)
			}
			return fullNode, err
997 998 999 1000
		})
	}
	if err != nil {
		Fatalf("Failed to register the Ethereum service: %v", err)
1001
	}
1002 1003
}

1004
// RegisterShhService configures Whisper and adds it to the given node.
1005 1006 1007 1008
func RegisterShhService(stack *node.Node, cfg *whisper.Config) {
	if err := stack.Register(func(n *node.ServiceContext) (node.Service, error) {
		return whisper.New(cfg), nil
	}); err != nil {
1009
		Fatalf("Failed to register the Whisper service: %v", err)
1010
	}
1011 1012
}

1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025
// RegisterEthStatsService configures the Ethereum Stats daemon and adds it to
// th egiven node.
func RegisterEthStatsService(stack *node.Node, url string) {
	if err := stack.Register(func(ctx *node.ServiceContext) (node.Service, error) {
		// Retrieve both eth and les services
		var ethServ *eth.Ethereum
		ctx.Service(&ethServ)

		var lesServ *les.LightEthereum
		ctx.Service(&lesServ)

		return ethstats.New(url, ethServ, lesServ)
	}); err != nil {
1026
		Fatalf("Failed to register the Ethereum Stats service: %v", err)
1027 1028 1029
	}
}

1030 1031
// SetupNetwork configures the system for either the main net or some test network.
func SetupNetwork(ctx *cli.Context) {
1032
	// TODO(fjl): move target gas limit into config
1033
	params.TargetGasLimit = new(big.Int).SetUint64(ctx.GlobalUint64(TargetGasLimitFlag.Name))
1034 1035
}

1036
// MakeChainDatabase open an LevelDB using the flags passed to the client and will hard crash if it fails.
1037
func MakeChainDatabase(ctx *cli.Context, stack *node.Node) ethdb.Database {
1038 1039
	var (
		cache   = ctx.GlobalInt(CacheFlag.Name)
1040
		handles = makeDatabaseHandles()
1041
	)
1042 1043 1044 1045
	name := "chaindata"
	if ctx.GlobalBool(LightModeFlag.Name) {
		name = "lightchaindata"
	}
1046
	chainDb, err := stack.OpenDatabase(name, cache, handles)
1047
	if err != nil {
1048
		Fatalf("Could not open database: %v", err)
1049
	}
1050 1051 1052
	return chainDb
}

F
Felix Lange 已提交
1053 1054 1055
func MakeGenesis(ctx *cli.Context) *core.Genesis {
	var genesis *core.Genesis
	switch {
1056
	case ctx.GlobalBool(TestnetFlag.Name):
F
Felix Lange 已提交
1057
		genesis = core.DefaultTestnetGenesisBlock()
1058 1059
	case ctx.GlobalBool(RinkebyFlag.Name):
		genesis = core.DefaultRinkebyGenesisBlock()
F
Felix Lange 已提交
1060 1061 1062 1063 1064 1065
	case ctx.GlobalBool(DevModeFlag.Name):
		genesis = core.DevGenesisBlock()
	}
	return genesis
}

1066
// MakeChain creates a chain manager from set command line flags.
1067
func MakeChain(ctx *cli.Context, stack *node.Node) (chain *core.BlockChain, chainDb ethdb.Database) {
1068
	var err error
1069
	chainDb = MakeChainDatabase(ctx, stack)
1070

1071
	engine := ethash.NewFaker()
1072
	if !ctx.GlobalBool(FakePoWFlag.Name) {
1073
		engine = ethash.New("", 1, 0, "", 1, 0)
1074
	}
F
Felix Lange 已提交
1075 1076 1077 1078 1079
	config, _, err := core.SetupGenesisBlock(chainDb, MakeGenesis(ctx))
	if err != nil {
		Fatalf("%v", err)
	}
	vmcfg := vm.Config{EnablePreimageRecording: ctx.GlobalBool(VMEnableDebugFlag.Name)}
1080
	chain, err = core.NewBlockChain(chainDb, config, engine, new(event.TypeMux), vmcfg)
O
obscuren 已提交
1081
	if err != nil {
F
Felix Lange 已提交
1082
		Fatalf("Can't create BlockChain: %v", err)
O
obscuren 已提交
1083
	}
1084
	return chain, chainDb
1085
}
1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102

// MakeConsolePreloads retrieves the absolute paths for the console JavaScript
// scripts to preload before starting.
func MakeConsolePreloads(ctx *cli.Context) []string {
	// Skip preloading if there's nothing to preload
	if ctx.GlobalString(PreloadJSFlag.Name) == "" {
		return nil
	}
	// Otherwise resolve absolute paths and return them
	preloads := []string{}

	assets := ctx.GlobalString(JSpathFlag.Name)
	for _, file := range strings.Split(ctx.GlobalString(PreloadJSFlag.Name), ",") {
		preloads = append(preloads, common.AbsolutePath(assets, strings.TrimSpace(file)))
	}
	return preloads
}
1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126

// MigrateFlags sets the global flag from a local flag when it's set.
// This is a temporary function used for migrating old command/flags to the
// new format.
//
// e.g. geth account new --keystore /tmp/mykeystore --lightkdf
//
// is equivalent after calling this method with:
//
// geth --keystore /tmp/mykeystore --lightkdf account new
//
// This allows the use of the existing configuration functionality.
// When all flags are migrated this function can be removed and the existing
// configuration functionality must be changed that is uses local flags
func MigrateFlags(action func(ctx *cli.Context) error) func(*cli.Context) error {
	return func(ctx *cli.Context) error {
		for _, name := range ctx.FlagNames() {
			if ctx.IsSet(name) {
				ctx.GlobalSet(name, ctx.String(name))
			}
		}
		return action(ctx)
	}
}