flags.go 30.2 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

31
	"github.com/ethereum/ethash"
F
Felix Lange 已提交
32
	"github.com/ethereum/go-ethereum/accounts"
33
	"github.com/ethereum/go-ethereum/accounts/keystore"
Z
zelig 已提交
34
	"github.com/ethereum/go-ethereum/common"
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 40
	"github.com/ethereum/go-ethereum/crypto"
	"github.com/ethereum/go-ethereum/eth"
	"github.com/ethereum/go-ethereum/ethdb"
41
	"github.com/ethereum/go-ethereum/ethstats"
42
	"github.com/ethereum/go-ethereum/event"
43
	"github.com/ethereum/go-ethereum/les"
44
	"github.com/ethereum/go-ethereum/logger"
O
obscuren 已提交
45
	"github.com/ethereum/go-ethereum/logger/glog"
B
Bas van Kervel 已提交
46
	"github.com/ethereum/go-ethereum/metrics"
47 48
	"github.com/ethereum/go-ethereum/node"
	"github.com/ethereum/go-ethereum/p2p/discover"
49
	"github.com/ethereum/go-ethereum/p2p/discv5"
50
	"github.com/ethereum/go-ethereum/p2p/nat"
51
	"github.com/ethereum/go-ethereum/p2p/netutil"
52
	"github.com/ethereum/go-ethereum/params"
53
	"github.com/ethereum/go-ethereum/pow"
54
	"github.com/ethereum/go-ethereum/rpc"
55
	whisper "github.com/ethereum/go-ethereum/whisper/whisperv2"
56
	"gopkg.in/urfave/cli.v1"
57 58
)

59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84
func init() {
	cli.AppHelpTemplate = `{{.Name}} {{if .Flags}}[global options] {{end}}command{{if .Flags}} [command options]{{end}} [arguments...]

VERSION:
   {{.Version}}

COMMANDS:
   {{range .Commands}}{{.Name}}{{with .ShortName}}, {{.}}{{end}}{{ "\t" }}{{.Usage}}
   {{end}}{{if .Flags}}
GLOBAL OPTIONS:
   {{range .Flags}}{{.}}
   {{end}}{{end}}
`

	cli.CommandHelpTemplate = `{{.Name}}{{if .Subcommands}} command{{end}}{{if .Flags}} [command options]{{end}} [arguments...]
{{if .Description}}{{.Description}}
{{end}}{{if .Subcommands}}
SUBCOMMANDS:
	{{range .Subcommands}}{{.Name}}{{with .ShortName}}, {{.}}{{end}}{{ "\t" }}{{.Usage}}
	{{end}}{{end}}{{if .Flags}}
OPTIONS:
	{{range .Flags}}{{.}}
	{{end}}{{end}}
`
}

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

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

218 219 220 221 222 223 224 225 226 227 228 229 230
	VMForceJitFlag = cli.BoolFlag{
		Name:  "forcejit",
		Usage: "Force the JIT VM to take precedence",
	}
	VMJitCacheFlag = cli.IntFlag{
		Name:  "jitcache",
		Usage: "Amount of cached JIT VM programs",
		Value: 64,
	}
	VMEnableJitFlag = cli.BoolFlag{
		Name:  "jitvm",
		Usage: "Enable the JIT VM",
	}
231 232 233 234
	VMEnableDebugFlag = cli.BoolFlag{
		Name:  "vmdebug",
		Usage: "Record information useful for VM and contract debugging",
	}
235 236 237 238 239
	// Logging and debug settings
	EthStatsURLFlag = cli.StringFlag{
		Name:  "ethstats",
		Usage: "Reporting URL of a ethstats service (nodename:secret@host:port)",
	}
240 241
	MetricsEnabledFlag = cli.BoolFlag{
		Name:  metrics.MetricsEnabledFlag,
242
		Usage: "Enable metrics collection and reporting",
243
	}
244 245 246 247
	FakePoWFlag = cli.BoolFlag{
		Name:  "fakepow",
		Usage: "Disables proof-of-work verification",
	}
248 249 250 251

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

321 322 323
	// Network Settings
	MaxPeersFlag = cli.IntFlag{
		Name:  "maxpeers",
324
		Usage: "Maximum number of network peers (network disabled if set to 0)",
325
		Value: 25,
326
	}
327 328 329 330 331
	MaxPendingPeersFlag = cli.IntFlag{
		Name:  "maxpendpeers",
		Usage: "Maximum number of pending connection attempts (defaults used if set to 0)",
		Value: 0,
	}
332 333 334 335 336
	ListenPortFlag = cli.IntFlag{
		Name:  "port",
		Usage: "Network listening port",
		Value: 30303,
	}
337
	BootnodesFlag = cli.StringFlag{
338
		Name:  "bootnodes",
339
		Usage: "Comma separated enode URLs for P2P discovery bootstrap",
340
		Value: "",
341 342 343 344 345 346 347 348 349 350 351
	}
	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",
352
		Usage: "NAT port mapping mechanism (any|none|upnp|pmp|extip:<IP>)",
353 354
		Value: "any",
	}
355 356 357 358
	NoDiscoverFlag = cli.BoolFlag{
		Name:  "nodiscover",
		Usage: "Disables the peer discovery mechanism (manual peer addition)",
	}
359 360 361 362
	DiscoveryV5Flag = cli.BoolFlag{
		Name:  "v5disc",
		Usage: "Enables the experimental RLPx V5 (Topic Discovery) mechanism",
	}
363 364 365 366 367
	NetrestrictFlag = cli.StringFlag{
		Name:  "netrestrict",
		Usage: "Restricts network communication to the given IP networks (CIDR masks)",
	}

368 369
	WhisperEnabledFlag = cli.BoolFlag{
		Name:  "shh",
370
		Usage: "Enable Whisper",
371
	}
372

373
	// ATM the url is left to the user and deployment to
Z
CLI:  
zelig 已提交
374 375
	JSpathFlag = cli.StringFlag{
		Name:  "jspath",
376
		Usage: "JavaScript root path for `loadScript`",
Z
CLI:  
zelig 已提交
377 378
		Value: ".",
	}
379 380
	SolcPathFlag = cli.StringFlag{
		Name:  "solc",
381
		Usage: "Solidity compiler command to be used",
382 383
		Value: "solc",
	}
384 385

	// Gas price oracle settings
Z
zsfelfoldi 已提交
386 387 388
	GpoMinGasPriceFlag = cli.StringFlag{
		Name:  "gpomin",
		Usage: "Minimum suggested gas price",
389
		Value: new(big.Int).Mul(big.NewInt(20), common.Shannon).String(),
Z
zsfelfoldi 已提交
390 391 392 393
	}
	GpoMaxGasPriceFlag = cli.StringFlag{
		Name:  "gpomax",
		Usage: "Maximum suggested gas price",
394
		Value: new(big.Int).Mul(big.NewInt(500), common.Shannon).String(),
Z
zsfelfoldi 已提交
395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415
	}
	GpoFullBlockRatioFlag = cli.IntFlag{
		Name:  "gpofull",
		Usage: "Full block threshold for gas price calculation (%)",
		Value: 80,
	}
	GpobaseStepDownFlag = cli.IntFlag{
		Name:  "gpobasedown",
		Usage: "Suggested gas price base step down ratio (1/1000)",
		Value: 10,
	}
	GpobaseStepUpFlag = cli.IntFlag{
		Name:  "gpobaseup",
		Usage: "Suggested gas price base step up ratio (1/1000)",
		Value: 100,
	}
	GpobaseCorrectionFactorFlag = cli.IntFlag{
		Name:  "gpobasecf",
		Usage: "Suggested gas price base correction factor (%)",
		Value: 110,
	}
416 417
)

418
// MakeDataDir retrieves the currently requested data directory, terminating
419 420
// 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.
421
func MakeDataDir(ctx *cli.Context) string {
422
	if path := ctx.GlobalString(DataDirFlag.Name); path != "" {
423
		// TODO: choose a different location outside of the regular datadir.
424
		if ctx.GlobalBool(TestNetFlag.Name) {
425
			return filepath.Join(path, "testnet")
426 427
		}
		return path
428
	}
429 430
	Fatalf("Cannot determine default data directory, please set manually (--datadir)")
	return ""
431 432
}

433
// MakeIPCPath creates an IPC path configuration from the set command line flags,
434
// returning an empty string if IPC was explicitly disabled, or the set path.
435
func MakeIPCPath(ctx *cli.Context) string {
436 437 438 439 440 441
	if ctx.GlobalBool(IPCDisabledFlag.Name) {
		return ""
	}
	return ctx.GlobalString(IPCPathFlag.Name)
}

442 443 444 445 446 447 448 449 450 451 452
// MakeNodeKey creates a node key from set command line flags, either loading it
// from a file or as a specified hex value. If neither flags were provided, this
// method returns nil and an emphemeral key is to be generated.
func MakeNodeKey(ctx *cli.Context) *ecdsa.PrivateKey {
	var (
		hex  = ctx.GlobalString(NodeKeyHexFlag.Name)
		file = ctx.GlobalString(NodeKeyFileFlag.Name)

		key *ecdsa.PrivateKey
		err error
	)
453 454 455
	switch {
	case file != "" && hex != "":
		Fatalf("Options %q and %q are mutually exclusive", NodeKeyFileFlag.Name, NodeKeyHexFlag.Name)
456

457 458 459 460
	case file != "":
		if key, err = crypto.LoadECDSA(file); err != nil {
			Fatalf("Option %q: %v", NodeKeyFileFlag.Name, err)
		}
461

462 463 464 465 466 467 468 469
	case hex != "":
		if key, err = crypto.HexToECDSA(hex); err != nil {
			Fatalf("Option %q: %v", NodeKeyHexFlag.Name, err)
		}
	}
	return key
}

470 471 472
// makeNodeUserIdent creates the user identifier from CLI flags.
func makeNodeUserIdent(ctx *cli.Context) string {
	var comps []string
473
	if identity := ctx.GlobalString(IdentityFlag.Name); len(identity) > 0 {
474
		comps = append(comps, identity)
475 476
	}
	if ctx.GlobalBool(VMEnableJitFlag.Name) {
477
		comps = append(comps, "JIT")
478
	}
479
	return strings.Join(comps, "/")
480 481 482 483 484
}

// MakeBootstrapNodes creates a list of bootstrap nodes from the command line
// flags, reverting to pre-configured ones if none have been specified.
func MakeBootstrapNodes(ctx *cli.Context) []*discover.Node {
485 486
	urls := params.MainnetBootnodes
	if ctx.GlobalIsSet(BootnodesFlag.Name) {
487
		urls = strings.Split(ctx.GlobalString(BootnodesFlag.Name), ",")
488 489
	} else if ctx.GlobalBool(TestNetFlag.Name) {
		urls = params.TestnetBootnodes
490 491
	}

492 493
	bootnodes := make([]*discover.Node, 0, len(urls))
	for _, url := range urls {
494 495 496 497 498 499 500 501 502 503
		node, err := discover.ParseNode(url)
		if err != nil {
			glog.V(logger.Error).Infof("Bootstrap URL %s: %v\n", url, err)
			continue
		}
		bootnodes = append(bootnodes, node)
	}
	return bootnodes
}

504 505 506
// MakeBootstrapNodesV5 creates a list of bootstrap nodes from the command line
// flags, reverting to pre-configured ones if none have been specified.
func MakeBootstrapNodesV5(ctx *cli.Context) []*discv5.Node {
507 508
	urls := params.DiscoveryV5Bootnodes
	if ctx.GlobalIsSet(BootnodesFlag.Name) {
509
		urls = strings.Split(ctx.GlobalString(BootnodesFlag.Name), ",")
510 511
	}

512 513
	bootnodes := make([]*discv5.Node, 0, len(urls))
	for _, url := range urls {
514 515 516 517 518 519 520 521 522 523
		node, err := discv5.ParseNode(url)
		if err != nil {
			glog.V(logger.Error).Infof("Bootstrap URL %s: %v\n", url, err)
			continue
		}
		bootnodes = append(bootnodes, node)
	}
	return bootnodes
}

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

530 531 532
// MakeDiscoveryV5Address creates a UDP listening address string from set command
// line flags for the V5 discovery protocol.
func MakeDiscoveryV5Address(ctx *cli.Context) string {
533 534 535
	return fmt.Sprintf(":%d", ctx.GlobalInt(ListenPortFlag.Name)+1)
}

536 537 538
// MakeNAT creates a port mapper from set command line flags.
func MakeNAT(ctx *cli.Context) nat.Interface {
	natif, err := nat.Parse(ctx.GlobalString(NATFlag.Name))
539
	if err != nil {
540 541 542 543 544
		Fatalf("Option %s: %v", NATFlag.Name, err)
	}
	return natif
}

545 546 547 548 549 550 551 552 553 554
// MakeRPCModules splits input separated by a comma and trims excessive white
// space from the substrings.
func MakeRPCModules(input string) []string {
	result := strings.Split(input, ",")
	for i, r := range result {
		result[i] = strings.TrimSpace(r)
	}
	return result
}

555
// MakeHTTPRpcHost creates the HTTP RPC listener interface string from the set
556
// command line flags, returning empty if the HTTP endpoint is disabled.
557
func MakeHTTPRpcHost(ctx *cli.Context) string {
558 559 560 561 562 563
	if !ctx.GlobalBool(RPCEnabledFlag.Name) {
		return ""
	}
	return ctx.GlobalString(RPCListenAddrFlag.Name)
}

564
// MakeWSRpcHost creates the WebSocket RPC listener interface string from the set
565
// command line flags, returning empty if the HTTP endpoint is disabled.
566
func MakeWSRpcHost(ctx *cli.Context) string {
567 568 569 570 571 572
	if !ctx.GlobalBool(WSEnabledFlag.Name) {
		return ""
	}
	return ctx.GlobalString(WSListenAddrFlag.Name)
}

573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588
// MakeDatabaseHandles raises out the number of allowed file handles per process
// for Geth and returns half of the allowance to assign to the database.
func MakeDatabaseHandles() int {
	if err := raiseFdLimit(2048); err != nil {
		Fatalf("Failed to raise file descriptor allowance: %v", err)
	}
	limit, err := getFdLimit()
	if err != nil {
		Fatalf("Failed to retrieve file descriptor allowance: %v", err)
	}
	if limit > 2048 { // cap database file descriptors even if more is available
		limit = 2048
	}
	return limit / 2 // Leave half for networking and other stuff
}

589 590
// MakeAddress converts an account specified directly as a hex encoded string or
// a key index in the key store to an internal account representation.
591
func MakeAddress(ks *keystore.KeyStore, account string) (accounts.Account, error) {
592 593
	// If the specified account is a valid address, return it
	if common.IsHexAddress(account) {
F
Felix Lange 已提交
594
		return accounts.Account{Address: common.HexToAddress(account)}, nil
595 596 597
	}
	// Otherwise try to interpret the account as a keystore index
	index, err := strconv.Atoi(account)
598
	if err != nil || index < 0 {
F
Felix Lange 已提交
599
		return accounts.Account{}, fmt.Errorf("invalid account address or index %q", account)
600
	}
601 602 603 604 605
	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
606 607 608 609
}

// MakeEtherbase retrieves the etherbase either from the directly specified
// command line flags or from the keystore if CLI indexed.
610 611
func MakeEtherbase(ks *keystore.KeyStore, ctx *cli.Context) common.Address {
	accounts := ks.Accounts()
612
	if !ctx.GlobalIsSet(EtherbaseFlag.Name) && len(accounts) == 0 {
613
		glog.V(logger.Error).Infoln("WARNING: No etherbase set and no accounts found as default")
614 615
		return common.Address{}
	}
616 617 618
	etherbase := ctx.GlobalString(EtherbaseFlag.Name)
	if etherbase == "" {
		return common.Address{}
619
	}
620
	// If the specified etherbase is a valid address, return it
621
	account, err := MakeAddress(ks, etherbase)
622
	if err != nil {
623
		Fatalf("Option %q: %v", EtherbaseFlag.Name, err)
624
	}
F
Felix Lange 已提交
625
	return account.Address
626 627 628 629 630 631 632
}

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

637
// MakePasswordList reads password lines from the file specified by --password.
638
func MakePasswordList(ctx *cli.Context) []string {
639 640 641 642 643 644 645 646 647 648 649 650
	path := ctx.GlobalString(PasswordFileFlag.Name)
	if path == "" {
		return nil
	}
	text, err := ioutil.ReadFile(path)
	if err != nil {
		Fatalf("Failed to read password file: %v", err)
	}
	lines := strings.Split(string(text), "\n")
	// Sanitise DOS line endings.
	for i := range lines {
		lines[i] = strings.TrimRight(lines[i], "\r")
651
	}
652
	return lines
653 654
}

655
// MakeNode configures a node with no services from command line flags.
656
func MakeNode(ctx *cli.Context, name, gitCommit string) *node.Node {
657
	vsn := params.Version
658 659 660 661
	if gitCommit != "" {
		vsn += "-" + gitCommit[:8]
	}

662 663 664 665
	// 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
	forceV5Discovery := (ctx.GlobalBool(LightModeFlag.Name) || ctx.GlobalInt(LightServFlag.Name) > 0) && !ctx.GlobalBool(NoDiscoverFlag.Name)

666
	config := &node.Config{
667
		DataDir:           MakeDataDir(ctx),
668 669 670
		KeyStoreDir:       ctx.GlobalString(KeyStoreDirFlag.Name),
		UseLightweightKDF: ctx.GlobalBool(LightKDFFlag.Name),
		PrivateKey:        MakeNodeKey(ctx),
671 672 673
		Name:              name,
		Version:           vsn,
		UserIdent:         makeNodeUserIdent(ctx),
674 675
		NoDiscovery:       ctx.GlobalBool(NoDiscoverFlag.Name) || ctx.GlobalBool(LightModeFlag.Name), // always disable v4 discovery in light client mode
		DiscoveryV5:       ctx.GlobalBool(DiscoveryV5Flag.Name) || forceV5Discovery,
676
		DiscoveryV5Addr:   MakeDiscoveryV5Address(ctx),
677
		BootstrapNodes:    MakeBootstrapNodes(ctx),
678
		BootstrapNodesV5:  MakeBootstrapNodesV5(ctx),
679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700
		ListenAddr:        MakeListenAddress(ctx),
		NAT:               MakeNAT(ctx),
		MaxPeers:          ctx.GlobalInt(MaxPeersFlag.Name),
		MaxPendingPeers:   ctx.GlobalInt(MaxPendingPeersFlag.Name),
		IPCPath:           MakeIPCPath(ctx),
		HTTPHost:          MakeHTTPRpcHost(ctx),
		HTTPPort:          ctx.GlobalInt(RPCPortFlag.Name),
		HTTPCors:          ctx.GlobalString(RPCCORSDomainFlag.Name),
		HTTPModules:       MakeRPCModules(ctx.GlobalString(RPCApiFlag.Name)),
		WSHost:            MakeWSRpcHost(ctx),
		WSPort:            ctx.GlobalInt(WSPortFlag.Name),
		WSOrigins:         ctx.GlobalString(WSAllowedOriginsFlag.Name),
		WSModules:         MakeRPCModules(ctx.GlobalString(WSApiFlag.Name)),
	}
	if ctx.GlobalBool(DevModeFlag.Name) {
		if !ctx.GlobalIsSet(DataDirFlag.Name) {
			config.DataDir = filepath.Join(os.TempDir(), "/ethereum_dev_mode")
		}
		// --dev mode does not need p2p networking.
		config.MaxPeers = 0
		config.ListenAddr = ":0"
	}
701 702 703 704 705 706 707 708
	if netrestrict := ctx.GlobalString(NetrestrictFlag.Name); netrestrict != "" {
		list, err := netutil.ParseNetlist(netrestrict)
		if err != nil {
			Fatalf("Option %q: %v", NetrestrictFlag.Name, err)
		}
		config.NetRestrict = list
	}

709 710 711 712 713 714 715 716 717
	stack, err := node.New(config)
	if err != nil {
		Fatalf("Failed to create the protocol stack: %v", err)
	}
	return stack
}

// RegisterEthService configures eth.Ethereum from command line flags and adds it to the
// given node.
718
func RegisterEthService(ctx *cli.Context, stack *node.Node, extra []byte) {
719
	// Avoid conflicting network flags
720
	networks, netFlags := 0, []cli.BoolFlag{DevModeFlag, TestNetFlag}
721 722 723 724 725 726 727 728
	for _, flag := range netFlags {
		if ctx.GlobalBool(flag.Name) {
			networks++
		}
	}
	if networks > 1 {
		Fatalf("The %v flags are mutually exclusive", netFlags)
	}
729
	ks := stack.AccountManager().Backends(keystore.KeyStoreType)[0].(*keystore.KeyStore)
730 731

	ethConf := &eth.Config{
732
		Etherbase:               MakeEtherbase(ks, ctx),
733
		ChainConfig:             MakeChainConfig(ctx, stack),
734
		FastSync:                ctx.GlobalBool(FastSyncFlag.Name),
735 736 737 738
		LightMode:               ctx.GlobalBool(LightModeFlag.Name),
		LightServ:               ctx.GlobalInt(LightServFlag.Name),
		LightPeers:              ctx.GlobalInt(LightPeersFlag.Name),
		MaxPeers:                ctx.GlobalInt(MaxPeersFlag.Name),
739
		DatabaseCache:           ctx.GlobalInt(CacheFlag.Name),
740
		DatabaseHandles:         MakeDatabaseHandles(),
Z
zsfelfoldi 已提交
741 742
		NetworkId:               ctx.GlobalInt(NetworkIdFlag.Name),
		MinerThreads:            ctx.GlobalInt(MinerThreadsFlag.Name),
743
		ExtraData:               MakeMinerExtra(extra, ctx),
Z
zelig 已提交
744
		DocRoot:                 ctx.GlobalString(DocRootFlag.Name),
Z
zsfelfoldi 已提交
745 746 747 748 749 750 751 752 753
		GasPrice:                common.String2Big(ctx.GlobalString(GasPriceFlag.Name)),
		GpoMinGasPrice:          common.String2Big(ctx.GlobalString(GpoMinGasPriceFlag.Name)),
		GpoMaxGasPrice:          common.String2Big(ctx.GlobalString(GpoMaxGasPriceFlag.Name)),
		GpoFullBlockRatio:       ctx.GlobalInt(GpoFullBlockRatioFlag.Name),
		GpobaseStepDown:         ctx.GlobalInt(GpobaseStepDownFlag.Name),
		GpobaseStepUp:           ctx.GlobalInt(GpobaseStepUpFlag.Name),
		GpobaseCorrectionFactor: ctx.GlobalInt(GpobaseCorrectionFactorFlag.Name),
		SolcPath:                ctx.GlobalString(SolcPathFlag.Name),
		AutoDAG:                 ctx.GlobalBool(AutoDAGFlag.Name) || ctx.GlobalBool(MiningEnabledFlag.Name),
754
		EnablePreimageRecording: ctx.GlobalBool(VMEnableDebugFlag.Name),
755
	}
756

757 758 759 760
	// Override any default configs in dev mode or the test net
	switch {
	case ctx.GlobalBool(TestNetFlag.Name):
		if !ctx.GlobalIsSet(NetworkIdFlag.Name) {
761
			ethConf.NetworkId = 3
762
		}
763
		ethConf.Genesis = core.DefaultTestnetGenesisBlock()
764

765
	case ctx.GlobalBool(DevModeFlag.Name):
766
		ethConf.Genesis = core.DevGenesisBlock()
767 768
		if !ctx.GlobalIsSet(GasPriceFlag.Name) {
			ethConf.GasPrice = new(big.Int)
769
		}
770
		ethConf.PowTest = true
771
	}
772 773 774 775
	// Override any global options pertaining to the Ethereum protocol
	if gen := ctx.GlobalInt(TrieCacheGenFlag.Name); gen > 0 {
		state.MaxTrieCacheGen = uint16(gen)
	}
776

777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793
	if ethConf.LightMode {
		if err := stack.Register(func(ctx *node.ServiceContext) (node.Service, error) {
			return les.New(ctx, ethConf)
		}); err != nil {
			Fatalf("Failed to register the Ethereum light node service: %v", err)
		}
	} else {
		if err := stack.Register(func(ctx *node.ServiceContext) (node.Service, error) {
			fullNode, err := eth.New(ctx, ethConf)
			if fullNode != nil && ethConf.LightServ > 0 {
				ls, _ := les.NewLesServer(fullNode, ethConf)
				fullNode.AddLesServer(ls)
			}
			return fullNode, err
		}); err != nil {
			Fatalf("Failed to register the Ethereum full node service: %v", err)
		}
794
	}
795 796
}

797
// RegisterShhService configures Whisper and adds it to the given node.
798 799 800 801
func RegisterShhService(stack *node.Node) {
	if err := stack.Register(func(*node.ServiceContext) (node.Service, error) { return whisper.New(), nil }); err != nil {
		Fatalf("Failed to register the Whisper service: %v", err)
	}
802 803
}

804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820
// 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 {
		Fatalf("Failed to register the Ethereum Stats service: %v", err)
	}
}

821 822
// SetupNetwork configures the system for either the main net or some test network.
func SetupNetwork(ctx *cli.Context) {
823
	params.TargetGasLimit = common.String2Big(ctx.GlobalString(TargetGasLimitFlag.Name))
824 825
}

826
// MakeChainConfig reads the chain configuration from the database in ctx.Datadir.
827
func MakeChainConfig(ctx *cli.Context, stack *node.Node) *params.ChainConfig {
828
	db := MakeChainDatabase(ctx, stack)
829 830
	defer db.Close()

831
	return MakeChainConfigFromDb(ctx, db)
832 833
}

834
// MakeChainConfigFromDb reads the chain configuration from the given database.
835
func MakeChainConfigFromDb(ctx *cli.Context, db ethdb.Database) *params.ChainConfig {
836
	// If the chain is already initialized, use any existing chain configs
837
	config := new(params.ChainConfig)
838

839 840
	genesis := core.GetBlock(db, core.GetCanonicalHash(db, 0), 0)
	if genesis != nil {
841
		storedConfig, err := core.GetChainConfig(db, genesis.Hash())
842 843 844 845 846 847
		switch err {
		case nil:
			config = storedConfig
		case core.ChainConfigNotFoundErr:
			// No configs found, use empty, will populate below
		default:
848 849 850
			Fatalf("Could not make chain configuration: %v", err)
		}
	}
J
Jeffrey Wilcke 已提交
851 852 853 854
	// set chain id in case it's zero.
	if config.ChainId == nil {
		config.ChainId = new(big.Int)
	}
855 856 857 858 859 860 861 862
	// Check whether we are allowed to set default config params or not:
	//  - If no genesis is set, we're running either mainnet or testnet (private nets use `geth init`)
	//  - If a genesis is already set, ensure we have a configuration for it (mainnet or testnet)
	defaults := genesis == nil ||
		(genesis.Hash() == params.MainNetGenesisHash && !ctx.GlobalBool(TestNetFlag.Name)) ||
		(genesis.Hash() == params.TestNetGenesisHash && ctx.GlobalBool(TestNetFlag.Name))

	if defaults {
863
		if ctx.GlobalBool(TestNetFlag.Name) {
864
			config = params.TestnetChainConfig
865
		} else {
866
			// Homestead fork
867
			config.HomesteadBlock = params.MainNetHomesteadBlock
868
			// DAO fork
869
			config.DAOForkBlock = params.MainNetDAOForkBlock
870
			config.DAOForkSupport = true
871

872
			// DoS reprice fork
873 874
			config.EIP150Block = params.MainNetHomesteadGasRepriceBlock
			config.EIP150Hash = params.MainNetHomesteadGasRepriceHash
875 876

			// DoS state cleanup fork
877 878 879
			config.EIP155Block = params.MainNetSpuriousDragon
			config.EIP158Block = params.MainNetSpuriousDragon
			config.ChainId = params.MainNetChainID
880
		}
881
	}
882
	return config
883 884
}

885 886 887 888 889 890 891 892
func ChainDbName(ctx *cli.Context) string {
	if ctx.GlobalBool(LightModeFlag.Name) {
		return "lightchaindata"
	} else {
		return "chaindata"
	}
}

893
// MakeChainDatabase open an LevelDB using the flags passed to the client and will hard crash if it fails.
894
func MakeChainDatabase(ctx *cli.Context, stack *node.Node) ethdb.Database {
895 896 897
	var (
		cache   = ctx.GlobalInt(CacheFlag.Name)
		handles = MakeDatabaseHandles()
898
		name    = ChainDbName(ctx)
899
	)
900

901
	chainDb, err := stack.OpenDatabase(name, cache, handles)
902
	if err != nil {
903 904
		Fatalf("Could not open database: %v", err)
	}
905 906 907 908
	return chainDb
}

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

913
	if ctx.GlobalBool(TestNetFlag.Name) {
914
		_, err := core.WriteTestNetGenesisBlock(chainDb)
915 916 917 918
		if err != nil {
			glog.Fatalln(err)
		}
	}
919

920
	chainConfig := MakeChainConfigFromDb(ctx, chainDb)
921

922 923 924 925
	pow := pow.PoW(core.FakePow{})
	if !ctx.GlobalBool(FakePoWFlag.Name) {
		pow = ethash.New()
	}
926
	chain, err = core.NewBlockChain(chainDb, chainConfig, pow, new(event.TypeMux), vm.Config{EnablePreimageRecording: ctx.GlobalBool(VMEnableDebugFlag.Name)})
O
obscuren 已提交
927 928 929
	if err != nil {
		Fatalf("Could not start chainmanager: %v", err)
	}
930
	return chain, chainDb
931
}
932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948

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