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

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

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

89
	cli.CommandHelpTemplate = CommandHelpTemplate
90 91
}

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

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

174 175 176 177 178 179 180 181 182 183
	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,
	}
184 185
	LightKDFFlag = cli.BoolFlag{
		Name:  "lightkdf",
186
		Usage: "Reduce key-derivation RAM & CPU usage at some expense of KDF strength",
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
	// Dashboard settings
	DashboardEnabledFlag = cli.BoolFlag{
		Name:  "dashboard",
		Usage: "Enable the dashboard",
	}
	DashboardAddrFlag = cli.StringFlag{
		Name:  "dashboard.addr",
		Usage: "Dashboard listening interface",
		Value: dashboard.DefaultConfig.Host,
	}
	DashboardPortFlag = cli.IntFlag{
		Name:  "dashboard.host",
		Usage: "Dashboard listening port",
		Value: dashboard.DefaultConfig.Port,
	}
	DashboardRefreshFlag = cli.DurationFlag{
		Name:  "dashboard.refresh",
		Usage: "Dashboard metrics collection refresh rate",
		Value: dashboard.DefaultConfig.Refresh,
	}
	DashboardAssetsFlag = cli.StringFlag{
		Name:  "dashboard.assets",
		Usage: "Developer flag to serve the dashboard from the local file system",
		Value: dashboard.DefaultConfig.Assets,
	}
213 214 215 216 217 218 219 220
	// 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)",
221
		Value: eth.DefaultConfig.Ethash.CachesInMem,
222 223 224 225
	}
	EthashCachesOnDiskFlag = cli.IntFlag{
		Name:  "ethash.cachesondisk",
		Usage: "Number of recent ethash caches to keep on disk (16MB each)",
226
		Value: eth.DefaultConfig.Ethash.CachesOnDisk,
227 228 229 230
	}
	EthashDatasetDirFlag = DirectoryFlag{
		Name:  "ethash.dagdir",
		Usage: "Directory to store the ethash mining DAGs (default = inside home folder)",
231
		Value: DirectoryString{eth.DefaultConfig.Ethash.DatasetDir},
232 233 234 235
	}
	EthashDatasetsInMemoryFlag = cli.IntFlag{
		Name:  "ethash.dagsinmem",
		Usage: "Number of recent ethash mining DAGs to keep in memory (1+GB each)",
236
		Value: eth.DefaultConfig.Ethash.DatasetsInMem,
237 238 239 240
	}
	EthashDatasetsOnDiskFlag = cli.IntFlag{
		Name:  "ethash.dagsondisk",
		Usage: "Number of recent ethash mining DAGs to keep on disk (1+GB each)",
241
		Value: eth.DefaultConfig.Ethash.DatasetsOnDisk,
242 243
	}
	// Transaction pool settings
244 245 246 247
	TxPoolNoLocalsFlag = cli.BoolFlag{
		Name:  "txpool.nolocals",
		Usage: "Disables price exemptions for locally submitted transactions",
	}
248 249 250 251 252 253 254 255 256 257
	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,
	}
258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292
	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,
	}
293 294 295 296 297 298 299 300 301 302 303
	// 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),
	}
304 305 306 307
	// Miner settings
	MiningEnabledFlag = cli.BoolFlag{
		Name:  "mine",
		Usage: "Enable mining",
308
	}
309 310
	MinerThreadsFlag = cli.IntFlag{
		Name:  "minerthreads",
311
		Usage: "Number of CPU threads to use for mining",
312 313
		Value: runtime.NumCPU(),
	}
314
	TargetGasLimitFlag = cli.Uint64Flag{
315 316
		Name:  "targetgaslimit",
		Usage: "Target gas limit sets the artificial target gas floor for the blocks to mine",
317
		Value: params.GenesisGasLimit,
318
	}
Z
zelig 已提交
319
	EtherbaseFlag = cli.StringFlag{
O
obscuren 已提交
320
		Name:  "etherbase",
321
		Usage: "Public address for block mining rewards (default = first account created)",
322
		Value: "0",
Z
zelig 已提交
323
	}
324
	GasPriceFlag = BigFlag{
325
		Name:  "gasprice",
326
		Usage: "Minimal gas price to accept for mining a transactions",
327
		Value: eth.DefaultConfig.GasPrice,
328
	}
Z
zelig 已提交
329 330
	ExtraDataFlag = cli.StringFlag{
		Name:  "extradata",
331
		Usage: "Block extra data set by the miner (default = client version)",
Z
zelig 已提交
332
	}
333
	// Account settings
Z
zelig 已提交
334 335
	UnlockedAccountFlag = cli.StringFlag{
		Name:  "unlock",
336
		Usage: "Comma separated list of accounts to unlock",
Z
zelig 已提交
337 338 339 340
		Value: "",
	}
	PasswordFileFlag = cli.StringFlag{
		Name:  "password",
A
ayeowch 已提交
341
		Usage: "Password file to use for non-interactive password input",
Z
zelig 已提交
342
		Value: "",
Z
zelig 已提交
343
	}
344

345 346 347 348
	VMEnableDebugFlag = cli.BoolFlag{
		Name:  "vmdebug",
		Usage: "Record information useful for VM and contract debugging",
	}
349 350 351 352 353
	// Logging and debug settings
	EthStatsURLFlag = cli.StringFlag{
		Name:  "ethstats",
		Usage: "Reporting URL of a ethstats service (nodename:secret@host:port)",
	}
354 355
	MetricsEnabledFlag = cli.BoolFlag{
		Name:  metrics.MetricsEnabledFlag,
356
		Usage: "Enable metrics collection and reporting",
357
	}
358 359 360 361
	FakePoWFlag = cli.BoolFlag{
		Name:  "fakepow",
		Usage: "Disables proof-of-work verification",
	}
362 363 364 365
	NoCompactionFlag = cli.BoolFlag{
		Name:  "nocompaction",
		Usage: "Disables db compaction after import",
	}
366 367 368
	// RPC settings
	RPCEnabledFlag = cli.BoolFlag{
		Name:  "rpc",
369
		Usage: "Enable the HTTP-RPC server",
370 371 372
	}
	RPCListenAddrFlag = cli.StringFlag{
		Name:  "rpcaddr",
373
		Usage: "HTTP-RPC server listening interface",
374
		Value: node.DefaultHTTPHost,
375 376 377
	}
	RPCPortFlag = cli.IntFlag{
		Name:  "rpcport",
378
		Usage: "HTTP-RPC server listening port",
379
		Value: node.DefaultHTTPPort,
380
	}
381 382
	RPCCORSDomainFlag = cli.StringFlag{
		Name:  "rpccorsdomain",
383
		Usage: "Comma separated list of domains from which to accept cross origin requests (browser enforced)",
384 385
		Value: "",
	}
386
	RPCApiFlag = cli.StringFlag{
387
		Name:  "rpcapi",
388
		Usage: "API's offered over the HTTP-RPC interface",
389
		Value: "",
390
	}
B
Bas van Kervel 已提交
391 392 393 394 395 396
	IPCDisabledFlag = cli.BoolFlag{
		Name:  "ipcdisable",
		Usage: "Disable the IPC-RPC server",
	}
	IPCPathFlag = DirectoryFlag{
		Name:  "ipcpath",
397
		Usage: "Filename for IPC socket/pipe within the datadir (explicit paths escape it)",
B
Bas van Kervel 已提交
398
	}
399
	WSEnabledFlag = cli.BoolFlag{
400
		Name:  "ws",
401 402 403 404 405
		Usage: "Enable the WS-RPC server",
	}
	WSListenAddrFlag = cli.StringFlag{
		Name:  "wsaddr",
		Usage: "WS-RPC server listening interface",
406
		Value: node.DefaultWSHost,
407 408 409 410
	}
	WSPortFlag = cli.IntFlag{
		Name:  "wsport",
		Usage: "WS-RPC server listening port",
411
		Value: node.DefaultWSPort,
412 413 414 415
	}
	WSApiFlag = cli.StringFlag{
		Name:  "wsapi",
		Usage: "API's offered over the WS-RPC interface",
416
		Value: "",
417
	}
B
Bas van Kervel 已提交
418 419 420
	WSAllowedOriginsFlag = cli.StringFlag{
		Name:  "wsorigins",
		Usage: "Origins from which to accept websockets requests",
421
		Value: "",
422
	}
423 424
	ExecFlag = cli.StringFlag{
		Name:  "exec",
425
		Usage: "Execute JavaScript statement",
426
	}
427
	PreloadJSFlag = cli.StringFlag{
428 429 430
		Name:  "preload",
		Usage: "Comma separated list of JavaScript files to preload into the console",
	}
431

432 433 434
	// Network Settings
	MaxPeersFlag = cli.IntFlag{
		Name:  "maxpeers",
435
		Usage: "Maximum number of network peers (network disabled if set to 0)",
436
		Value: 25,
437
	}
438 439 440 441 442
	MaxPendingPeersFlag = cli.IntFlag{
		Name:  "maxpendpeers",
		Usage: "Maximum number of pending connection attempts (defaults used if set to 0)",
		Value: 0,
	}
443 444 445 446 447
	ListenPortFlag = cli.IntFlag{
		Name:  "port",
		Usage: "Network listening port",
		Value: 30303,
	}
448
	BootnodesFlag = cli.StringFlag{
449
		Name:  "bootnodes",
450 451 452 453 454 455 456 457 458 459 460
		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)",
461
		Value: "",
462 463 464 465 466 467 468 469 470 471 472
	}
	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",
473
		Usage: "NAT port mapping mechanism (any|none|upnp|pmp|extip:<IP>)",
474 475
		Value: "any",
	}
476 477 478 479
	NoDiscoverFlag = cli.BoolFlag{
		Name:  "nodiscover",
		Usage: "Disables the peer discovery mechanism (manual peer addition)",
	}
480 481 482 483
	DiscoveryV5Flag = cli.BoolFlag{
		Name:  "v5disc",
		Usage: "Enables the experimental RLPx V5 (Topic Discovery) mechanism",
	}
484 485 486 487 488
	NetrestrictFlag = cli.StringFlag{
		Name:  "netrestrict",
		Usage: "Restricts network communication to the given IP networks (CIDR masks)",
	}

489
	// ATM the url is left to the user and deployment to
Z
CLI:  
zelig 已提交
490 491
	JSpathFlag = cli.StringFlag{
		Name:  "jspath",
492
		Usage: "JavaScript root path for `loadScript`",
Z
CLI:  
zelig 已提交
493 494
		Value: ".",
	}
495 496

	// Gas price oracle settings
497 498 499
	GpoBlocksFlag = cli.IntFlag{
		Name:  "gpoblocks",
		Usage: "Number of recent blocks to check for gas prices",
500
		Value: eth.DefaultConfig.GPO.Blocks,
Z
zsfelfoldi 已提交
501
	}
502 503 504
	GpoPercentileFlag = cli.IntFlag{
		Name:  "gpopercentile",
		Usage: "Suggested gas price is the given percentile of a set of recent transaction gas prices",
505
		Value: eth.DefaultConfig.GPO.Percentile,
Z
zsfelfoldi 已提交
506
	}
507 508 509 510 511 512 513 514 515 516 517 518 519 520
	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,
	}
521 522
)

523
// MakeDataDir retrieves the currently requested data directory, terminating
524 525
// 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.
526
func MakeDataDir(ctx *cli.Context) string {
527
	if path := ctx.GlobalString(DataDirFlag.Name); path != "" {
528
		if ctx.GlobalBool(TestnetFlag.Name) {
529
			return filepath.Join(path, "testnet")
530
		}
531 532 533
		if ctx.GlobalBool(RinkebyFlag.Name) {
			return filepath.Join(path, "rinkeby")
		}
534
		return path
535
	}
536
	Fatalf("Cannot determine default data directory, please set manually (--datadir)")
537
	return ""
538 539
}

540
// setNodeKey creates a node key from set command line flags, either loading it
541 542
// 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.
543
func setNodeKey(ctx *cli.Context, cfg *p2p.Config) {
544 545 546
	var (
		hex  = ctx.GlobalString(NodeKeyHexFlag.Name)
		file = ctx.GlobalString(NodeKeyFileFlag.Name)
547 548
		key  *ecdsa.PrivateKey
		err  error
549
	)
550 551
	switch {
	case file != "" && hex != "":
552
		Fatalf("Options %q and %q are mutually exclusive", NodeKeyFileFlag.Name, NodeKeyHexFlag.Name)
553 554
	case file != "":
		if key, err = crypto.LoadECDSA(file); err != nil {
555
			Fatalf("Option %q: %v", NodeKeyFileFlag.Name, err)
556
		}
557
		cfg.PrivateKey = key
558 559
	case hex != "":
		if key, err = crypto.HexToECDSA(hex); err != nil {
560
			Fatalf("Option %q: %v", NodeKeyHexFlag.Name, err)
561
		}
562
		cfg.PrivateKey = key
563 564 565
	}
}

566 567
// setNodeUserIdent creates the user identifier from CLI flags.
func setNodeUserIdent(ctx *cli.Context, cfg *node.Config) {
568
	if identity := ctx.GlobalString(IdentityFlag.Name); len(identity) > 0 {
569
		cfg.UserIdent = identity
570 571 572
	}
}

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

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

603
// setBootstrapNodesV5 creates a list of bootstrap nodes from the command line
604
// flags, reverting to pre-configured ones if none have been specified.
605
func setBootstrapNodesV5(ctx *cli.Context, cfg *p2p.Config) {
606
	urls := params.DiscoveryV5Bootnodes
607
	switch {
608 609 610 611 612 613
	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), ",")
		}
614
	case ctx.GlobalBool(RinkebyFlag.Name):
615
		urls = params.RinkebyBootnodes
616
	case cfg.BootstrapNodesV5 != nil:
617
		return // already set, don't apply defaults.
618 619
	}

620
	cfg.BootstrapNodesV5 = make([]*discv5.Node, 0, len(urls))
621
	for _, url := range urls {
622 623
		node, err := discv5.ParseNode(url)
		if err != nil {
P
Péter Szilágyi 已提交
624
			log.Error("Bootstrap URL invalid", "enode", url, "err", err)
625 626
			continue
		}
627
		cfg.BootstrapNodesV5 = append(cfg.BootstrapNodesV5, node)
628 629 630
	}
}

631
// setListenAddress creates a TCP listening address string from set command
632
// line flags.
633 634 635 636
func setListenAddress(ctx *cli.Context, cfg *p2p.Config) {
	if ctx.GlobalIsSet(ListenPortFlag.Name) {
		cfg.ListenAddr = fmt.Sprintf(":%d", ctx.GlobalInt(ListenPortFlag.Name))
	}
637 638
}

639 640 641 642 643 644 645 646
// 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
647 648 649
	}
}

650 651 652
// splitAndTrim splits input separated by a comma
// and trims excessive white space from the substrings.
func splitAndTrim(input string) []string {
653 654 655 656 657 658 659
	result := strings.Split(input, ",")
	for i, r := range result {
		result[i] = strings.TrimSpace(r)
	}
	return result
}

660
// setHTTP creates the HTTP 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 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) {
674
		cfg.HTTPCors = splitAndTrim(ctx.GlobalString(RPCCORSDomainFlag.Name))
675 676
	}
	if ctx.GlobalIsSet(RPCApiFlag.Name) {
677
		cfg.HTTPModules = splitAndTrim(ctx.GlobalString(RPCApiFlag.Name))
678 679 680
	}
}

681
// setWS creates the WebSocket RPC listener interface string from the set
682
// command line flags, returning empty if the HTTP endpoint is disabled.
683 684 685 686 687 688 689 690 691 692 693 694
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) {
695
		cfg.WSOrigins = splitAndTrim(ctx.GlobalString(WSAllowedOriginsFlag.Name))
696 697
	}
	if ctx.GlobalIsSet(WSApiFlag.Name) {
698
		cfg.WSModules = splitAndTrim(ctx.GlobalString(WSApiFlag.Name))
699 700 701 702 703 704 705 706 707 708 709 710
	}
}

// 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)
711 712 713
	}
}

714
// makeDatabaseHandles raises out the number of allowed file handles per process
715
// for Geth and returns half of the allowance to assign to the database.
716
func makeDatabaseHandles() int {
717
	limit, err := fdlimit.Current()
718
	if err != nil {
719
		Fatalf("Failed to retrieve file descriptor allowance: %v", err)
720
	}
721 722 723 724 725
	if limit < 2048 {
		if err := fdlimit.Raise(2048); err != nil {
			Fatalf("Failed to raise file descriptor allowance: %v", err)
		}
	}
726 727 728 729 730 731
	if limit > 2048 { // cap database file descriptors even if more is available
		limit = 2048
	}
	return limit / 2 // Leave half for networking and other stuff
}

732 733
// MakeAddress converts an account specified directly as a hex encoded string or
// a key index in the key store to an internal account representation.
734
func MakeAddress(ks *keystore.KeyStore, account string) (accounts.Account, error) {
735 736
	// If the specified account is a valid address, return it
	if common.IsHexAddress(account) {
F
Felix Lange 已提交
737
		return accounts.Account{Address: common.HexToAddress(account)}, nil
738 739 740
	}
	// Otherwise try to interpret the account as a keystore index
	index, err := strconv.Atoi(account)
741
	if err != nil || index < 0 {
F
Felix Lange 已提交
742
		return accounts.Account{}, fmt.Errorf("invalid account address or index %q", account)
743
	}
744 745 746 747 748 749
	log.Warn("-------------------------------------------------------------------")
	log.Warn("Referring to accounts by order in the keystore folder is dangerous!")
	log.Warn("This functionality is deprecated and will be removed in the future!")
	log.Warn("Please use explicit addresses! (can search via `geth account list`)")
	log.Warn("-------------------------------------------------------------------")

750 751 752 753 754
	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
755 756
}

757
// setEtherbase retrieves the etherbase either from the directly specified
758
// command line flags or from the keystore if CLI indexed.
759 760 761 762 763 764 765
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
766
	}
767 768
}

769
// MakePasswordList reads password lines from the file specified by the global --password flag.
770
func MakePasswordList(ctx *cli.Context) []string {
771 772 773 774 775 776
	path := ctx.GlobalString(PasswordFileFlag.Name)
	if path == "" {
		return nil
	}
	text, err := ioutil.ReadFile(path)
	if err != nil {
777
		Fatalf("Failed to read password file: %v", err)
778 779 780 781 782
	}
	lines := strings.Split(string(text), "\n")
	// Sanitise DOS line endings.
	for i := range lines {
		lines[i] = strings.TrimRight(lines[i], "\r")
783
	}
784
	return lines
785 786
}

787 788 789 790 791 792 793 794 795 796 797 798 799 800 801
func SetP2PConfig(ctx *cli.Context, cfg *p2p.Config) {
	setNodeKey(ctx, cfg)
	setNAT(ctx, cfg)
	setListenAddress(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
802 803
	}

804 805 806
	// 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
807
	forceV5Discovery := (ctx.GlobalBool(LightModeFlag.Name) || ctx.GlobalInt(LightServFlag.Name) > 0) && !ctx.GlobalBool(NoDiscoverFlag.Name)
808 809 810 811
	if ctx.GlobalIsSet(DiscoveryV5Flag.Name) {
		cfg.DiscoveryV5 = ctx.GlobalBool(DiscoveryV5Flag.Name)
	} else if forceV5Discovery {
		cfg.DiscoveryV5 = true
812
	}
813

814 815 816
	if netrestrict := ctx.GlobalString(NetrestrictFlag.Name); netrestrict != "" {
		list, err := netutil.ParseNetlist(netrestrict)
		if err != nil {
817
			Fatalf("Option %q: %v", NetrestrictFlag.Name, err)
818
		}
819
		cfg.NetRestrict = list
820 821
	}

822
	if ctx.GlobalBool(DeveloperFlag.Name) {
823 824 825 826 827
		// --dev mode can't use p2p networking.
		cfg.MaxPeers = 0
		cfg.ListenAddr = ":0"
		cfg.NoDiscovery = true
		cfg.DiscoveryV5 = false
828 829 830
	}
}

831 832 833 834 835 836 837 838 839 840 841
// 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)
842 843
	case ctx.GlobalBool(DeveloperFlag.Name):
		cfg.DataDir = "" // unless explicitly requested, use memory databases
844
	case ctx.GlobalBool(TestnetFlag.Name):
845
		cfg.DataDir = filepath.Join(node.DefaultDataDir(), "testnet")
846 847
	case ctx.GlobalBool(RinkebyFlag.Name):
		cfg.DataDir = filepath.Join(node.DefaultDataDir(), "rinkeby")
848 849 850 851 852 853 854 855
	}

	if ctx.GlobalIsSet(KeyStoreDirFlag.Name) {
		cfg.KeyStoreDir = ctx.GlobalString(KeyStoreDirFlag.Name)
	}
	if ctx.GlobalIsSet(LightKDFFlag.Name) {
		cfg.UseLightweightKDF = ctx.GlobalBool(LightKDFFlag.Name)
	}
856 857 858
	if ctx.GlobalIsSet(NoUSBFlag.Name) {
		cfg.NoUSB = ctx.GlobalBool(NoUSBFlag.Name)
	}
859 860 861 862 863 864 865 866 867 868 869
}

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

870
func setTxPool(ctx *cli.Context, cfg *core.TxPoolConfig) {
871 872 873
	if ctx.GlobalIsSet(TxPoolNoLocalsFlag.Name) {
		cfg.NoLocals = ctx.GlobalBool(TxPoolNoLocalsFlag.Name)
	}
874 875 876 877 878 879
	if ctx.GlobalIsSet(TxPoolJournalFlag.Name) {
		cfg.Journal = ctx.GlobalString(TxPoolJournalFlag.Name)
	}
	if ctx.GlobalIsSet(TxPoolRejournalFlag.Name) {
		cfg.Rejournal = ctx.GlobalDuration(TxPoolRejournalFlag.Name)
	}
880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902
	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)
	}
}

903 904
func setEthash(ctx *cli.Context, cfg *eth.Config) {
	if ctx.GlobalIsSet(EthashCacheDirFlag.Name) {
905
		cfg.Ethash.CacheDir = ctx.GlobalString(EthashCacheDirFlag.Name)
906 907
	}
	if ctx.GlobalIsSet(EthashDatasetDirFlag.Name) {
908
		cfg.Ethash.DatasetDir = ctx.GlobalString(EthashDatasetDirFlag.Name)
909 910
	}
	if ctx.GlobalIsSet(EthashCachesInMemoryFlag.Name) {
911
		cfg.Ethash.CachesInMem = ctx.GlobalInt(EthashCachesInMemoryFlag.Name)
912 913
	}
	if ctx.GlobalIsSet(EthashCachesOnDiskFlag.Name) {
914
		cfg.Ethash.CachesOnDisk = ctx.GlobalInt(EthashCachesOnDiskFlag.Name)
915 916
	}
	if ctx.GlobalIsSet(EthashDatasetsInMemoryFlag.Name) {
917
		cfg.Ethash.DatasetsInMem = ctx.GlobalInt(EthashDatasetsInMemoryFlag.Name)
918 919
	}
	if ctx.GlobalIsSet(EthashDatasetsOnDiskFlag.Name) {
920
		cfg.Ethash.DatasetsOnDisk = ctx.GlobalInt(EthashDatasetsOnDiskFlag.Name)
921 922 923
	}
}

924 925 926 927
// checkExclusive verifies that only a single isntance of the provided flags was
// set by the user. Each flag might optionally be followed by a string type to
// specialize it further.
func checkExclusive(ctx *cli.Context, args ...interface{}) {
928
	set := make([]string, 0, 1)
929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952
	for i := 0; i < len(args); i++ {
		// Make sure the next argument is a flag and skip if not set
		flag, ok := args[i].(cli.Flag)
		if !ok {
			panic(fmt.Sprintf("invalid argument, not cli.Flag type: %T", args[i]))
		}
		// Check if next arg extends current and expand its name if so
		name := flag.GetName()

		if i+1 < len(args) {
			switch option := args[i+1].(type) {
			case string:
				// Extended flag, expand the name and shift the arguments
				if ctx.GlobalString(flag.GetName()) == option {
					name += "=" + option
				}
				i++

			case cli.Flag:
			default:
				panic(fmt.Sprintf("invalid argument, not cli.Flag or string extension: %T", args[i+1]))
			}
		}
		// Mark the flag if it's set
953
		if ctx.GlobalIsSet(flag.GetName()) {
954
			set = append(set, "--"+name)
955 956
		}
	}
957
	if len(set) > 1 {
958
		Fatalf("Flags %v can't be used at the same time", strings.Join(set, ", "))
959
	}
960 961
}

962 963
// SetShhConfig applies shh-related command line flags to the config.
func SetShhConfig(ctx *cli.Context, stack *node.Node, cfg *whisper.Config) {
964 965
	if ctx.GlobalIsSet(WhisperMaxMessageSizeFlag.Name) {
		cfg.MaxMessageSize = uint32(ctx.GlobalUint(WhisperMaxMessageSizeFlag.Name))
966
	}
967 968
	if ctx.GlobalIsSet(WhisperMinPOWFlag.Name) {
		cfg.MinimumAcceptedPOW = ctx.GlobalFloat64(WhisperMinPOWFlag.Name)
969 970 971
	}
}

972 973 974
// 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
975
	checkExclusive(ctx, DeveloperFlag, TestnetFlag, RinkebyFlag)
976
	checkExclusive(ctx, FastSyncFlag, LightModeFlag, SyncModeFlag)
977 978
	checkExclusive(ctx, LightServFlag, LightModeFlag)
	checkExclusive(ctx, LightServFlag, SyncModeFlag, "light")
979

980
	ks := stack.AccountManager().Backends(keystore.KeyStoreType)[0].(*keystore.KeyStore)
981 982
	setEtherbase(ctx, ks, cfg)
	setGPO(ctx, &cfg.GPO)
983
	setTxPool(ctx, &cfg.TxPool)
984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000
	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) {
1001
		cfg.NetworkId = ctx.GlobalUint64(NetworkIdFlag.Name)
1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024
	}

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

1026
	// Override any default configs for hard coded networks.
1027
	switch {
1028
	case ctx.GlobalBool(TestnetFlag.Name):
1029
		if !ctx.GlobalIsSet(NetworkIdFlag.Name) {
1030
			cfg.NetworkId = 3
1031
		}
1032
		cfg.Genesis = core.DefaultTestnetGenesisBlock()
1033 1034 1035 1036 1037
	case ctx.GlobalBool(RinkebyFlag.Name):
		if !ctx.GlobalIsSet(NetworkIdFlag.Name) {
			cfg.NetworkId = 4
		}
		cfg.Genesis = core.DefaultRinkebyGenesisBlock()
1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057
	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)
1058
		if !ctx.GlobalIsSet(GasPriceFlag.Name) {
1059
			cfg.GasPrice = big.NewInt(1)
1060 1061
		}
	}
1062
	// TODO(fjl): move trie cache generations into config
1063 1064 1065
	if gen := ctx.GlobalInt(TrieCacheGenFlag.Name); gen > 0 {
		state.MaxTrieCacheGen = uint16(gen)
	}
1066
}
1067

1068 1069 1070 1071 1072 1073 1074 1075
// SetDashboardConfig applies dashboard related command line flags to the config.
func SetDashboardConfig(ctx *cli.Context, cfg *dashboard.Config) {
	cfg.Host = ctx.GlobalString(DashboardAddrFlag.Name)
	cfg.Port = ctx.GlobalInt(DashboardPortFlag.Name)
	cfg.Refresh = ctx.GlobalDuration(DashboardRefreshFlag.Name)
	cfg.Assets = ctx.GlobalString(DashboardAssetsFlag.Name)
}

1076 1077 1078 1079 1080 1081 1082
// 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)
		})
1083
	} else {
1084 1085 1086 1087
		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)
1088 1089 1090
				fullNode.AddLesServer(ls)
			}
			return fullNode, err
1091 1092 1093 1094
		})
	}
	if err != nil {
		Fatalf("Failed to register the Ethereum service: %v", err)
1095
	}
1096 1097
}

1098
// RegisterDashboardService adds a dashboard to the stack.
1099
func RegisterDashboardService(stack *node.Node, cfg *dashboard.Config, commit string) {
1100
	stack.Register(func(ctx *node.ServiceContext) (node.Service, error) {
1101
		return dashboard.New(cfg, commit)
1102 1103 1104
	})
}

1105
// RegisterShhService configures Whisper and adds it to the given node.
1106 1107 1108 1109
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 {
1110
		Fatalf("Failed to register the Whisper service: %v", err)
1111
	}
1112 1113
}

1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126
// 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 {
1127
		Fatalf("Failed to register the Ethereum Stats service: %v", err)
1128 1129 1130
	}
}

1131 1132
// SetupNetwork configures the system for either the main net or some test network.
func SetupNetwork(ctx *cli.Context) {
1133
	// TODO(fjl): move target gas limit into config
1134
	params.TargetGasLimit = ctx.GlobalUint64(TargetGasLimitFlag.Name)
1135 1136
}

1137
// MakeChainDatabase open an LevelDB using the flags passed to the client and will hard crash if it fails.
1138
func MakeChainDatabase(ctx *cli.Context, stack *node.Node) ethdb.Database {
1139 1140
	var (
		cache   = ctx.GlobalInt(CacheFlag.Name)
1141
		handles = makeDatabaseHandles()
1142
	)
1143 1144 1145 1146
	name := "chaindata"
	if ctx.GlobalBool(LightModeFlag.Name) {
		name = "lightchaindata"
	}
1147
	chainDb, err := stack.OpenDatabase(name, cache, handles)
1148
	if err != nil {
1149
		Fatalf("Could not open database: %v", err)
1150
	}
1151 1152 1153
	return chainDb
}

F
Felix Lange 已提交
1154 1155 1156
func MakeGenesis(ctx *cli.Context) *core.Genesis {
	var genesis *core.Genesis
	switch {
1157
	case ctx.GlobalBool(TestnetFlag.Name):
F
Felix Lange 已提交
1158
		genesis = core.DefaultTestnetGenesisBlock()
1159 1160
	case ctx.GlobalBool(RinkebyFlag.Name):
		genesis = core.DefaultRinkebyGenesisBlock()
1161 1162
	case ctx.GlobalBool(DeveloperFlag.Name):
		Fatalf("Developer chains are ephemeral")
F
Felix Lange 已提交
1163 1164 1165 1166
	}
	return genesis
}

1167
// MakeChain creates a chain manager from set command line flags.
1168
func MakeChain(ctx *cli.Context, stack *node.Node) (chain *core.BlockChain, chainDb ethdb.Database) {
1169
	var err error
1170
	chainDb = MakeChainDatabase(ctx, stack)
1171

F
Felix Lange 已提交
1172 1173 1174 1175
	config, _, err := core.SetupGenesisBlock(chainDb, MakeGenesis(ctx))
	if err != nil {
		Fatalf("%v", err)
	}
1176 1177 1178 1179 1180 1181
	var engine consensus.Engine
	if config.Clique != nil {
		engine = clique.New(config.Clique, chainDb)
	} else {
		engine = ethash.NewFaker()
		if !ctx.GlobalBool(FakePoWFlag.Name) {
1182 1183 1184 1185 1186 1187 1188 1189
			engine = ethash.New(ethash.Config{
				CacheDir:       stack.ResolvePath(eth.DefaultConfig.Ethash.CacheDir),
				CachesInMem:    eth.DefaultConfig.Ethash.CachesInMem,
				CachesOnDisk:   eth.DefaultConfig.Ethash.CachesOnDisk,
				DatasetDir:     stack.ResolvePath(eth.DefaultConfig.Ethash.DatasetDir),
				DatasetsInMem:  eth.DefaultConfig.Ethash.DatasetsInMem,
				DatasetsOnDisk: eth.DefaultConfig.Ethash.DatasetsOnDisk,
			})
1190 1191
		}
	}
F
Felix Lange 已提交
1192
	vmcfg := vm.Config{EnablePreimageRecording: ctx.GlobalBool(VMEnableDebugFlag.Name)}
1193
	chain, err = core.NewBlockChain(chainDb, config, engine, vmcfg)
O
obscuren 已提交
1194
	if err != nil {
F
Felix Lange 已提交
1195
		Fatalf("Can't create BlockChain: %v", err)
O
obscuren 已提交
1196
	}
1197
	return chain, chainDb
1198
}
1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215

// 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
}
1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239

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