flags.go 37.5 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 35
	"github.com/ethereum/go-ethereum/consensus"
	"github.com/ethereum/go-ethereum/consensus/clique"
36
	"github.com/ethereum/go-ethereum/consensus/ethash"
37
	"github.com/ethereum/go-ethereum/core"
38
	"github.com/ethereum/go-ethereum/core/state"
39
	"github.com/ethereum/go-ethereum/core/vm"
40 41
	"github.com/ethereum/go-ethereum/crypto"
	"github.com/ethereum/go-ethereum/eth"
42 43
	"github.com/ethereum/go-ethereum/eth/downloader"
	"github.com/ethereum/go-ethereum/eth/gasprice"
44
	"github.com/ethereum/go-ethereum/ethdb"
45
	"github.com/ethereum/go-ethereum/ethstats"
46
	"github.com/ethereum/go-ethereum/les"
47
	"github.com/ethereum/go-ethereum/log"
B
Bas van Kervel 已提交
48
	"github.com/ethereum/go-ethereum/metrics"
49
	"github.com/ethereum/go-ethereum/node"
50
	"github.com/ethereum/go-ethereum/p2p"
51
	"github.com/ethereum/go-ethereum/p2p/discover"
52
	"github.com/ethereum/go-ethereum/p2p/discv5"
53
	"github.com/ethereum/go-ethereum/p2p/nat"
54
	"github.com/ethereum/go-ethereum/p2p/netutil"
55
	"github.com/ethereum/go-ethereum/params"
56
	whisper "github.com/ethereum/go-ethereum/whisper/whisperv5"
57
	"gopkg.in/urfave/cli.v1"
58 59
)

60 61 62 63 64 65 66 67 68 69 70 71 72
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}}`
)

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

87
	cli.CommandHelpTemplate = CommandHelpTemplate
88 89
}

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

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

172 173 174 175 176 177 178 179 180 181
	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,
	}
182 183
	LightKDFFlag = cli.BoolFlag{
		Name:  "lightkdf",
184
		Usage: "Reduce key-derivation RAM & CPU usage at some expense of KDF strength",
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
	// 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
217 218 219 220
	TxPoolNoLocalsFlag = cli.BoolFlag{
		Name:  "txpool.nolocals",
		Usage: "Disables price exemptions for locally submitted transactions",
	}
221 222 223 224 225 226 227 228 229 230
	TxPoolJournalFlag = cli.StringFlag{
		Name:  "txpool.journal",
		Usage: "Disk journal for local transaction to survive node restarts",
		Value: core.DefaultTxPoolConfig.Journal,
	}
	TxPoolRejournalFlag = cli.DurationFlag{
		Name:  "txpool.rejournal",
		Usage: "Time interval to regenerate the local transaction journal",
		Value: core.DefaultTxPoolConfig.Rejournal,
	}
231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265
	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,
	}
266 267 268 269 270 271 272 273 274 275 276
	// 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),
	}
277 278 279 280
	// Miner settings
	MiningEnabledFlag = cli.BoolFlag{
		Name:  "mine",
		Usage: "Enable mining",
281
	}
282 283
	MinerThreadsFlag = cli.IntFlag{
		Name:  "minerthreads",
284
		Usage: "Number of CPU threads to use for mining",
285 286
		Value: runtime.NumCPU(),
	}
287
	TargetGasLimitFlag = cli.Uint64Flag{
288 289
		Name:  "targetgaslimit",
		Usage: "Target gas limit sets the artificial target gas floor for the blocks to mine",
290
		Value: params.GenesisGasLimit.Uint64(),
291
	}
Z
zelig 已提交
292
	EtherbaseFlag = cli.StringFlag{
O
obscuren 已提交
293
		Name:  "etherbase",
294
		Usage: "Public address for block mining rewards (default = first account created)",
295
		Value: "0",
Z
zelig 已提交
296
	}
297
	GasPriceFlag = BigFlag{
298
		Name:  "gasprice",
299
		Usage: "Minimal gas price to accept for mining a transactions",
300
		Value: eth.DefaultConfig.GasPrice,
301
	}
Z
zelig 已提交
302 303
	ExtraDataFlag = cli.StringFlag{
		Name:  "extradata",
304
		Usage: "Block extra data set by the miner (default = client version)",
Z
zelig 已提交
305
	}
306
	// Account settings
Z
zelig 已提交
307 308
	UnlockedAccountFlag = cli.StringFlag{
		Name:  "unlock",
309
		Usage: "Comma separated list of accounts to unlock",
Z
zelig 已提交
310 311 312 313
		Value: "",
	}
	PasswordFileFlag = cli.StringFlag{
		Name:  "password",
A
ayeowch 已提交
314
		Usage: "Password file to use for non-interactive password input",
Z
zelig 已提交
315
		Value: "",
Z
zelig 已提交
316
	}
317

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

405 406 407
	// Network Settings
	MaxPeersFlag = cli.IntFlag{
		Name:  "maxpeers",
408
		Usage: "Maximum number of network peers (network disabled if set to 0)",
409
		Value: 25,
410
	}
411 412 413 414 415
	MaxPendingPeersFlag = cli.IntFlag{
		Name:  "maxpendpeers",
		Usage: "Maximum number of pending connection attempts (defaults used if set to 0)",
		Value: 0,
	}
416 417 418 419 420
	ListenPortFlag = cli.IntFlag{
		Name:  "port",
		Usage: "Network listening port",
		Value: 30303,
	}
421
	BootnodesFlag = cli.StringFlag{
422
		Name:  "bootnodes",
423 424 425 426 427 428 429 430 431 432 433
		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)",
434
		Value: "",
435 436 437 438 439 440 441 442 443 444 445
	}
	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",
446
		Usage: "NAT port mapping mechanism (any|none|upnp|pmp|extip:<IP>)",
447 448
		Value: "any",
	}
449 450 451 452
	NoDiscoverFlag = cli.BoolFlag{
		Name:  "nodiscover",
		Usage: "Disables the peer discovery mechanism (manual peer addition)",
	}
453 454 455 456
	DiscoveryV5Flag = cli.BoolFlag{
		Name:  "v5disc",
		Usage: "Enables the experimental RLPx V5 (Topic Discovery) mechanism",
	}
457 458 459 460 461
	NetrestrictFlag = cli.StringFlag{
		Name:  "netrestrict",
		Usage: "Restricts network communication to the given IP networks (CIDR masks)",
	}

462
	// ATM the url is left to the user and deployment to
Z
CLI:  
zelig 已提交
463 464
	JSpathFlag = cli.StringFlag{
		Name:  "jspath",
465
		Usage: "JavaScript root path for `loadScript`",
Z
CLI:  
zelig 已提交
466 467
		Value: ".",
	}
468 469

	// Gas price oracle settings
470 471 472
	GpoBlocksFlag = cli.IntFlag{
		Name:  "gpoblocks",
		Usage: "Number of recent blocks to check for gas prices",
473
		Value: eth.DefaultConfig.GPO.Blocks,
Z
zsfelfoldi 已提交
474
	}
475 476 477
	GpoPercentileFlag = cli.IntFlag{
		Name:  "gpopercentile",
		Usage: "Suggested gas price is the given percentile of a set of recent transaction gas prices",
478
		Value: eth.DefaultConfig.GPO.Percentile,
Z
zsfelfoldi 已提交
479
	}
480 481 482 483 484 485 486 487 488 489 490 491 492 493
	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,
	}
494 495
)

496
// MakeDataDir retrieves the currently requested data directory, terminating
497 498
// 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.
499
func MakeDataDir(ctx *cli.Context) string {
500
	if path := ctx.GlobalString(DataDirFlag.Name); path != "" {
501
		if ctx.GlobalBool(TestnetFlag.Name) {
502
			return filepath.Join(path, "testnet")
503
		}
504 505 506
		if ctx.GlobalBool(RinkebyFlag.Name) {
			return filepath.Join(path, "rinkeby")
		}
507
		return path
508
	}
509
	Fatalf("Cannot determine default data directory, please set manually (--datadir)")
510
	return ""
511 512
}

513
// setNodeKey creates a node key from set command line flags, either loading it
514 515
// 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.
516
func setNodeKey(ctx *cli.Context, cfg *p2p.Config) {
517 518 519
	var (
		hex  = ctx.GlobalString(NodeKeyHexFlag.Name)
		file = ctx.GlobalString(NodeKeyFileFlag.Name)
520 521
		key  *ecdsa.PrivateKey
		err  error
522
	)
523 524
	switch {
	case file != "" && hex != "":
525
		Fatalf("Options %q and %q are mutually exclusive", NodeKeyFileFlag.Name, NodeKeyHexFlag.Name)
526 527
	case file != "":
		if key, err = crypto.LoadECDSA(file); err != nil {
528
			Fatalf("Option %q: %v", NodeKeyFileFlag.Name, err)
529
		}
530
		cfg.PrivateKey = key
531 532
	case hex != "":
		if key, err = crypto.HexToECDSA(hex); err != nil {
533
			Fatalf("Option %q: %v", NodeKeyHexFlag.Name, err)
534
		}
535
		cfg.PrivateKey = key
536 537 538
	}
}

539 540
// setNodeUserIdent creates the user identifier from CLI flags.
func setNodeUserIdent(ctx *cli.Context, cfg *node.Config) {
541
	if identity := ctx.GlobalString(IdentityFlag.Name); len(identity) > 0 {
542
		cfg.UserIdent = identity
543 544 545
	}
}

546
// setBootstrapNodes creates a list of bootstrap nodes from the command line
547
// flags, reverting to pre-configured ones if none have been specified.
548
func setBootstrapNodes(ctx *cli.Context, cfg *p2p.Config) {
549
	urls := params.MainnetBootnodes
550
	switch {
551 552 553 554 555 556
	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), ",")
		}
557
	case ctx.GlobalBool(TestnetFlag.Name):
558
		urls = params.TestnetBootnodes
559 560
	case ctx.GlobalBool(RinkebyFlag.Name):
		urls = params.RinkebyBootnodes
561 562
	}

563
	cfg.BootstrapNodes = make([]*discover.Node, 0, len(urls))
564
	for _, url := range urls {
565 566
		node, err := discover.ParseNode(url)
		if err != nil {
P
Péter Szilágyi 已提交
567
			log.Error("Bootstrap URL invalid", "enode", url, "err", err)
568 569
			continue
		}
570
		cfg.BootstrapNodes = append(cfg.BootstrapNodes, node)
571 572 573
	}
}

574
// setBootstrapNodesV5 creates a list of bootstrap nodes from the command line
575
// flags, reverting to pre-configured ones if none have been specified.
576
func setBootstrapNodesV5(ctx *cli.Context, cfg *p2p.Config) {
577
	urls := params.DiscoveryV5Bootnodes
578
	switch {
579 580 581 582 583 584
	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), ",")
		}
585 586 587
	case ctx.GlobalBool(RinkebyFlag.Name):
		urls = params.RinkebyV5Bootnodes
	case cfg.BootstrapNodesV5 != nil:
588
		return // already set, don't apply defaults.
589 590
	}

591
	cfg.BootstrapNodesV5 = make([]*discv5.Node, 0, len(urls))
592
	for _, url := range urls {
593 594
		node, err := discv5.ParseNode(url)
		if err != nil {
P
Péter Szilágyi 已提交
595
			log.Error("Bootstrap URL invalid", "enode", url, "err", err)
596 597
			continue
		}
598
		cfg.BootstrapNodesV5 = append(cfg.BootstrapNodesV5, node)
599 600 601
	}
}

602
// setListenAddress creates a TCP listening address string from set command
603
// line flags.
604 605 606 607
func setListenAddress(ctx *cli.Context, cfg *p2p.Config) {
	if ctx.GlobalIsSet(ListenPortFlag.Name) {
		cfg.ListenAddr = fmt.Sprintf(":%d", ctx.GlobalInt(ListenPortFlag.Name))
	}
608 609
}

610
// setDiscoveryV5Address creates a UDP listening address string from set command
611
// line flags for the V5 discovery protocol.
612 613 614 615
func setDiscoveryV5Address(ctx *cli.Context, cfg *p2p.Config) {
	if ctx.GlobalIsSet(ListenPortFlag.Name) {
		cfg.DiscoveryV5Addr = fmt.Sprintf(":%d", ctx.GlobalInt(ListenPortFlag.Name)+1)
	}
616 617
}

618 619 620 621 622 623 624 625
// 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
626 627 628
	}
}

629 630 631
// splitAndTrim splits input separated by a comma
// and trims excessive white space from the substrings.
func splitAndTrim(input string) []string {
632 633 634 635 636 637 638
	result := strings.Split(input, ",")
	for i, r := range result {
		result[i] = strings.TrimSpace(r)
	}
	return result
}

639
// setHTTP creates the HTTP RPC listener interface string from the set
640
// command line flags, returning empty if the HTTP endpoint is disabled.
641 642 643 644 645 646 647 648 649 650 651 652
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) {
653
		cfg.HTTPCors = splitAndTrim(ctx.GlobalString(RPCCORSDomainFlag.Name))
654 655
	}
	if ctx.GlobalIsSet(RPCApiFlag.Name) {
656
		cfg.HTTPModules = splitAndTrim(ctx.GlobalString(RPCApiFlag.Name))
657 658 659
	}
}

660
// setWS creates the WebSocket RPC listener interface string from the set
661
// command line flags, returning empty if the HTTP endpoint is disabled.
662 663 664 665 666 667 668 669 670 671 672 673
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) {
674
		cfg.WSOrigins = splitAndTrim(ctx.GlobalString(WSAllowedOriginsFlag.Name))
675 676
	}
	if ctx.GlobalIsSet(WSApiFlag.Name) {
677
		cfg.WSModules = splitAndTrim(ctx.GlobalString(WSApiFlag.Name))
678 679 680 681 682 683 684 685 686 687 688 689
	}
}

// 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)
690 691 692
	}
}

693
// makeDatabaseHandles raises out the number of allowed file handles per process
694
// for Geth and returns half of the allowance to assign to the database.
695
func makeDatabaseHandles() int {
696
	if err := raiseFdLimit(2048); err != nil {
697
		Fatalf("Failed to raise file descriptor allowance: %v", err)
698 699 700
	}
	limit, err := getFdLimit()
	if err != nil {
701
		Fatalf("Failed to retrieve file descriptor allowance: %v", err)
702 703 704 705 706 707 708
	}
	if limit > 2048 { // cap database file descriptors even if more is available
		limit = 2048
	}
	return limit / 2 // Leave half for networking and other stuff
}

709 710
// MakeAddress converts an account specified directly as a hex encoded string or
// a key index in the key store to an internal account representation.
711
func MakeAddress(ks *keystore.KeyStore, account string) (accounts.Account, error) {
712 713
	// If the specified account is a valid address, return it
	if common.IsHexAddress(account) {
F
Felix Lange 已提交
714
		return accounts.Account{Address: common.HexToAddress(account)}, nil
715 716 717
	}
	// Otherwise try to interpret the account as a keystore index
	index, err := strconv.Atoi(account)
718
	if err != nil || index < 0 {
F
Felix Lange 已提交
719
		return accounts.Account{}, fmt.Errorf("invalid account address or index %q", account)
720
	}
721 722 723 724 725
	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
726 727
}

728
// setEtherbase retrieves the etherbase either from the directly specified
729
// command line flags or from the keystore if CLI indexed.
730 731 732 733 734 735 736 737
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
738
	}
739 740 741 742 743 744 745
	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")
		}
746
	}
747 748
}

749
// MakePasswordList reads password lines from the file specified by the global --password flag.
750
func MakePasswordList(ctx *cli.Context) []string {
751 752 753 754 755 756
	path := ctx.GlobalString(PasswordFileFlag.Name)
	if path == "" {
		return nil
	}
	text, err := ioutil.ReadFile(path)
	if err != nil {
757
		Fatalf("Failed to read password file: %v", err)
758 759 760 761 762
	}
	lines := strings.Split(string(text), "\n")
	// Sanitise DOS line endings.
	for i := range lines {
		lines[i] = strings.TrimRight(lines[i], "\r")
763
	}
764
	return lines
765 766
}

767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782
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
783 784
	}

785 786 787
	// 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
788
	forceV5Discovery := (ctx.GlobalBool(LightModeFlag.Name) || ctx.GlobalInt(LightServFlag.Name) > 0) && !ctx.GlobalBool(NoDiscoverFlag.Name)
789 790 791 792
	if ctx.GlobalIsSet(DiscoveryV5Flag.Name) {
		cfg.DiscoveryV5 = ctx.GlobalBool(DiscoveryV5Flag.Name)
	} else if forceV5Discovery {
		cfg.DiscoveryV5 = true
793
	}
794

795 796 797
	if netrestrict := ctx.GlobalString(NetrestrictFlag.Name); netrestrict != "" {
		list, err := netutil.ParseNetlist(netrestrict)
		if err != nil {
798
			Fatalf("Option %q: %v", NetrestrictFlag.Name, err)
799
		}
800
		cfg.NetRestrict = list
801 802
	}

803
	if ctx.GlobalBool(DeveloperFlag.Name) {
804 805 806
		// --dev mode can't use p2p networking.
		cfg.MaxPeers = 0
		cfg.ListenAddr = ":0"
807
		cfg.DiscoveryV5Addr = ":0"
808 809
		cfg.NoDiscovery = true
		cfg.DiscoveryV5 = false
810 811 812
	}
}

813 814 815 816 817 818 819 820 821 822 823
// 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)
824 825
	case ctx.GlobalBool(DeveloperFlag.Name):
		cfg.DataDir = "" // unless explicitly requested, use memory databases
826
	case ctx.GlobalBool(TestnetFlag.Name):
827
		cfg.DataDir = filepath.Join(node.DefaultDataDir(), "testnet")
828 829
	case ctx.GlobalBool(RinkebyFlag.Name):
		cfg.DataDir = filepath.Join(node.DefaultDataDir(), "rinkeby")
830 831 832 833 834 835 836 837
	}

	if ctx.GlobalIsSet(KeyStoreDirFlag.Name) {
		cfg.KeyStoreDir = ctx.GlobalString(KeyStoreDirFlag.Name)
	}
	if ctx.GlobalIsSet(LightKDFFlag.Name) {
		cfg.UseLightweightKDF = ctx.GlobalBool(LightKDFFlag.Name)
	}
838 839 840
	if ctx.GlobalIsSet(NoUSBFlag.Name) {
		cfg.NoUSB = ctx.GlobalBool(NoUSBFlag.Name)
	}
841 842 843 844 845 846 847 848 849 850 851
}

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)
	}
}

852
func setTxPool(ctx *cli.Context, cfg *core.TxPoolConfig) {
853 854 855
	if ctx.GlobalIsSet(TxPoolNoLocalsFlag.Name) {
		cfg.NoLocals = ctx.GlobalBool(TxPoolNoLocalsFlag.Name)
	}
856 857 858 859 860 861
	if ctx.GlobalIsSet(TxPoolJournalFlag.Name) {
		cfg.Journal = ctx.GlobalString(TxPoolJournalFlag.Name)
	}
	if ctx.GlobalIsSet(TxPoolRejournalFlag.Name) {
		cfg.Rejournal = ctx.GlobalDuration(TxPoolRejournalFlag.Name)
	}
862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884
	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)
	}
}

885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910
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())
911 912
		}
	}
913 914
	if len(set) > 1 {
		Fatalf("flags %v can't be used at the same time", strings.Join(set, ", "))
915
	}
916 917
}

918 919
// SetShhConfig applies shh-related command line flags to the config.
func SetShhConfig(ctx *cli.Context, stack *node.Node, cfg *whisper.Config) {
920 921
	if ctx.GlobalIsSet(WhisperMaxMessageSizeFlag.Name) {
		cfg.MaxMessageSize = uint32(ctx.GlobalUint(WhisperMaxMessageSizeFlag.Name))
922
	}
923 924
	if ctx.GlobalIsSet(WhisperMinPOWFlag.Name) {
		cfg.MinimumAcceptedPOW = ctx.GlobalFloat64(WhisperMinPOWFlag.Name)
925 926 927
	}
}

928 929 930
// 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
931
	checkExclusive(ctx, DeveloperFlag, TestnetFlag, RinkebyFlag)
932 933
	checkExclusive(ctx, FastSyncFlag, LightModeFlag, SyncModeFlag)

934
	ks := stack.AccountManager().Backends(keystore.KeyStoreType)[0].(*keystore.KeyStore)
935 936
	setEtherbase(ctx, ks, cfg)
	setGPO(ctx, &cfg.GPO)
937
	setTxPool(ctx, &cfg.TxPool)
938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954
	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) {
955
		cfg.NetworkId = ctx.GlobalUint64(NetworkIdFlag.Name)
956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978
	}

	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)
	}
979

980
	// Override any default configs for hard coded networks.
981
	switch {
982
	case ctx.GlobalBool(TestnetFlag.Name):
983
		if !ctx.GlobalIsSet(NetworkIdFlag.Name) {
984
			cfg.NetworkId = 3
985
		}
986
		cfg.Genesis = core.DefaultTestnetGenesisBlock()
987 988 989 990 991
	case ctx.GlobalBool(RinkebyFlag.Name):
		if !ctx.GlobalIsSet(NetworkIdFlag.Name) {
			cfg.NetworkId = 4
		}
		cfg.Genesis = core.DefaultRinkebyGenesisBlock()
992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011
	case ctx.GlobalBool(DeveloperFlag.Name):
		// Create new developer account or reuse existing one
		var (
			developer accounts.Account
			err       error
		)
		if accs := ks.Accounts(); len(accs) > 0 {
			developer = ks.Accounts()[0]
		} else {
			developer, err = ks.NewAccount("")
			if err != nil {
				Fatalf("Failed to create developer account: %v", err)
			}
		}
		if err := ks.Unlock(developer, ""); err != nil {
			Fatalf("Failed to unlock developer account: %v", err)
		}
		log.Info("Using developer account", "address", developer.Address)

		cfg.Genesis = core.DeveloperGenesisBlock(uint64(ctx.GlobalInt(DeveloperPeriodFlag.Name)), developer.Address)
1012
		if !ctx.GlobalIsSet(GasPriceFlag.Name) {
1013
			cfg.GasPrice = big.NewInt(1)
1014 1015
		}
	}
1016
	// TODO(fjl): move trie cache generations into config
1017 1018 1019
	if gen := ctx.GlobalInt(TrieCacheGenFlag.Name); gen > 0 {
		state.MaxTrieCacheGen = uint16(gen)
	}
1020
}
1021

1022 1023 1024 1025 1026 1027 1028
// 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)
		})
1029
	} else {
1030 1031 1032 1033
		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)
1034 1035 1036
				fullNode.AddLesServer(ls)
			}
			return fullNode, err
1037 1038 1039 1040
		})
	}
	if err != nil {
		Fatalf("Failed to register the Ethereum service: %v", err)
1041
	}
1042 1043
}

1044
// RegisterShhService configures Whisper and adds it to the given node.
1045 1046 1047 1048
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 {
1049
		Fatalf("Failed to register the Whisper service: %v", err)
1050
	}
1051 1052
}

1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065
// 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 {
1066
		Fatalf("Failed to register the Ethereum Stats service: %v", err)
1067 1068 1069
	}
}

1070 1071
// SetupNetwork configures the system for either the main net or some test network.
func SetupNetwork(ctx *cli.Context) {
1072
	// TODO(fjl): move target gas limit into config
1073
	params.TargetGasLimit = new(big.Int).SetUint64(ctx.GlobalUint64(TargetGasLimitFlag.Name))
1074 1075
}

1076
// MakeChainDatabase open an LevelDB using the flags passed to the client and will hard crash if it fails.
1077
func MakeChainDatabase(ctx *cli.Context, stack *node.Node) ethdb.Database {
1078 1079
	var (
		cache   = ctx.GlobalInt(CacheFlag.Name)
1080
		handles = makeDatabaseHandles()
1081
	)
1082 1083 1084 1085
	name := "chaindata"
	if ctx.GlobalBool(LightModeFlag.Name) {
		name = "lightchaindata"
	}
1086
	chainDb, err := stack.OpenDatabase(name, cache, handles)
1087
	if err != nil {
1088
		Fatalf("Could not open database: %v", err)
1089
	}
1090 1091 1092
	return chainDb
}

F
Felix Lange 已提交
1093 1094 1095
func MakeGenesis(ctx *cli.Context) *core.Genesis {
	var genesis *core.Genesis
	switch {
1096
	case ctx.GlobalBool(TestnetFlag.Name):
F
Felix Lange 已提交
1097
		genesis = core.DefaultTestnetGenesisBlock()
1098 1099
	case ctx.GlobalBool(RinkebyFlag.Name):
		genesis = core.DefaultRinkebyGenesisBlock()
1100 1101
	case ctx.GlobalBool(DeveloperFlag.Name):
		Fatalf("Developer chains are ephemeral")
F
Felix Lange 已提交
1102 1103 1104 1105
	}
	return genesis
}

1106
// MakeChain creates a chain manager from set command line flags.
1107
func MakeChain(ctx *cli.Context, stack *node.Node) (chain *core.BlockChain, chainDb ethdb.Database) {
1108
	var err error
1109
	chainDb = MakeChainDatabase(ctx, stack)
1110

F
Felix Lange 已提交
1111 1112 1113 1114
	config, _, err := core.SetupGenesisBlock(chainDb, MakeGenesis(ctx))
	if err != nil {
		Fatalf("%v", err)
	}
1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126
	var engine consensus.Engine
	if config.Clique != nil {
		engine = clique.New(config.Clique, chainDb)
	} else {
		engine = ethash.NewFaker()
		if !ctx.GlobalBool(FakePoWFlag.Name) {
			engine = ethash.New(
				stack.ResolvePath(eth.DefaultConfig.EthashCacheDir), eth.DefaultConfig.EthashCachesInMem, eth.DefaultConfig.EthashCachesOnDisk,
				stack.ResolvePath(eth.DefaultConfig.EthashDatasetDir), eth.DefaultConfig.EthashDatasetsInMem, eth.DefaultConfig.EthashDatasetsOnDisk,
			)
		}
	}
F
Felix Lange 已提交
1127
	vmcfg := vm.Config{EnablePreimageRecording: ctx.GlobalBool(VMEnableDebugFlag.Name)}
1128
	chain, err = core.NewBlockChain(chainDb, config, engine, vmcfg)
O
obscuren 已提交
1129
	if err != nil {
F
Felix Lange 已提交
1130
		Fatalf("Can't create BlockChain: %v", err)
O
obscuren 已提交
1131
	}
1132
	return chain, chainDb
1133
}
1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150

// 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
}
1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174

// 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)
	}
}