flags.go 29.6 KB
Newer Older
F
Felix Lange 已提交
1 2 3 4 5 6 7 8 9 10
// Copyright 2015 The go-ethereum Authors
// This file is part of go-ethereum.
//
// go-ethereum is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// go-ethereum is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
11
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
F
Felix Lange 已提交
12 13 14
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
15
// along with go-ethereum. If not, see <http://www.gnu.org/licenses/>.
F
Felix Lange 已提交
16

17 18 19 20
package utils

import (
	"crypto/ecdsa"
21
	"fmt"
22
	"io/ioutil"
23
	"math"
24
	"math/big"
25
	"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"
Z
zelig 已提交
33
	"github.com/ethereum/go-ethereum/common"
34
	"github.com/ethereum/go-ethereum/core"
35
	"github.com/ethereum/go-ethereum/core/state"
36 37 38 39
	"github.com/ethereum/go-ethereum/crypto"
	"github.com/ethereum/go-ethereum/eth"
	"github.com/ethereum/go-ethereum/ethdb"
	"github.com/ethereum/go-ethereum/event"
40
	"github.com/ethereum/go-ethereum/les"
41
	"github.com/ethereum/go-ethereum/logger"
O
obscuren 已提交
42
	"github.com/ethereum/go-ethereum/logger/glog"
B
Bas van Kervel 已提交
43
	"github.com/ethereum/go-ethereum/metrics"
44 45
	"github.com/ethereum/go-ethereum/node"
	"github.com/ethereum/go-ethereum/p2p/discover"
46
	"github.com/ethereum/go-ethereum/p2p/discv5"
47
	"github.com/ethereum/go-ethereum/p2p/nat"
48
	"github.com/ethereum/go-ethereum/params"
49
	"github.com/ethereum/go-ethereum/pow"
50
	"github.com/ethereum/go-ethereum/rpc"
51
	whisper "github.com/ethereum/go-ethereum/whisper/whisperv2"
52
	"gopkg.in/urfave/cli.v1"
53 54
)

55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80
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 已提交
81
// NewApp creates an app with sane defaults.
82
func NewApp(gitCommit, usage string) *cli.App {
F
Felix Lange 已提交
83
	app := cli.NewApp()
84
	app.Name = filepath.Base(os.Args[0])
F
Felix Lange 已提交
85
	app.Author = ""
O
obscuren 已提交
86
	//app.Authors = nil
F
Felix Lange 已提交
87
	app.Email = ""
88 89 90 91
	app.Version = Version
	if gitCommit != "" {
		app.Version += "-" + gitCommit[:8]
	}
F
Felix Lange 已提交
92 93 94 95
	app.Usage = usage
	return app
}

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

231 232 233 234 235 236 237 238 239 240 241 242 243 244
	VMForceJitFlag = cli.BoolFlag{
		Name:  "forcejit",
		Usage: "Force the JIT VM to take precedence",
	}
	VMJitCacheFlag = cli.IntFlag{
		Name:  "jitcache",
		Usage: "Amount of cached JIT VM programs",
		Value: 64,
	}
	VMEnableJitFlag = cli.BoolFlag{
		Name:  "jitvm",
		Usage: "Enable the JIT VM",
	}

Z
zelig 已提交
245
	// logging and debug settings
246 247
	MetricsEnabledFlag = cli.BoolFlag{
		Name:  metrics.MetricsEnabledFlag,
248
		Usage: "Enable metrics collection and reporting",
249
	}
250 251 252 253
	FakePoWFlag = cli.BoolFlag{
		Name:  "fakepow",
		Usage: "Disables proof-of-work verification",
	}
254 255 256 257

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

327 328 329
	// Network Settings
	MaxPeersFlag = cli.IntFlag{
		Name:  "maxpeers",
330
		Usage: "Maximum number of network peers (network disabled if set to 0)",
331
		Value: 25,
332
	}
333 334 335 336 337
	MaxPendingPeersFlag = cli.IntFlag{
		Name:  "maxpendpeers",
		Usage: "Maximum number of pending connection attempts (defaults used if set to 0)",
		Value: 0,
	}
338 339 340 341 342 343 344
	ListenPortFlag = cli.IntFlag{
		Name:  "port",
		Usage: "Network listening port",
		Value: 30303,
	}
	BootnodesFlag = cli.StringFlag{
		Name:  "bootnodes",
345
		Usage: "Comma separated enode URLs for P2P discovery bootstrap",
346 347 348 349 350 351 352 353 354 355 356 357
		Value: "",
	}
	NodeKeyFileFlag = cli.StringFlag{
		Name:  "nodekey",
		Usage: "P2P node key file",
	}
	NodeKeyHexFlag = cli.StringFlag{
		Name:  "nodekeyhex",
		Usage: "P2P node key as hex (for testing)",
	}
	NATFlag = cli.StringFlag{
		Name:  "nat",
358
		Usage: "NAT port mapping mechanism (any|none|upnp|pmp|extip:<IP>)",
359 360
		Value: "any",
	}
361 362 363 364
	NoDiscoverFlag = cli.BoolFlag{
		Name:  "nodiscover",
		Usage: "Disables the peer discovery mechanism (manual peer addition)",
	}
365 366 367 368
	DiscoveryV5Flag = cli.BoolFlag{
		Name:  "v5disc",
		Usage: "Enables the experimental RLPx V5 (Topic Discovery) mechanism",
	}
369 370
	WhisperEnabledFlag = cli.BoolFlag{
		Name:  "shh",
371
		Usage: "Enable Whisper",
372
	}
373
	// ATM the url is left to the user and deployment to
Z
CLI:  
zelig 已提交
374 375
	JSpathFlag = cli.StringFlag{
		Name:  "jspath",
I
Isidoro Ghezzi 已提交
376
		Usage: "JavaScript root path for `loadScript` and document root for `admin.httpGet`",
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 485 486 487
}

// MakeBootstrapNodes creates a list of bootstrap nodes from the command line
// flags, reverting to pre-configured ones if none have been specified.
func MakeBootstrapNodes(ctx *cli.Context) []*discover.Node {
	// Return pre-configured nodes if none were manually requested
	if !ctx.GlobalIsSet(BootnodesFlag.Name) {
		if ctx.GlobalBool(TestNetFlag.Name) {
488
			return params.TestnetBootnodes
489
		}
490
		return params.MainnetBootnodes
491 492 493 494 495 496 497 498 499 500 501 502 503 504 505
	}
	// Otherwise parse and use the CLI bootstrap nodes
	bootnodes := []*discover.Node{}

	for _, url := range strings.Split(ctx.GlobalString(BootnodesFlag.Name), ",") {
		node, err := discover.ParseNode(url)
		if err != nil {
			glog.V(logger.Error).Infof("Bootstrap URL %s: %v\n", url, err)
			continue
		}
		bootnodes = append(bootnodes, node)
	}
	return bootnodes
}

506 507 508 509 510
// 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 {
	// Return pre-configured nodes if none were manually requested
	if !ctx.GlobalIsSet(BootnodesFlag.Name) {
511
		return params.DiscoveryV5Bootnodes
512 513 514 515 516 517 518 519 520 521 522 523 524 525 526
	}
	// Otherwise parse and use the CLI bootstrap nodes
	bootnodes := []*discv5.Node{}

	for _, url := range strings.Split(ctx.GlobalString(BootnodesFlag.Name), ",") {
		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
}

527 528 529 530 531 532
// 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))
}

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

539 540 541
// 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))
542
	if err != nil {
543 544 545 546 547
		Fatalf("Option %s: %v", NATFlag.Name, err)
	}
	return natif
}

548 549 550 551 552 553 554 555 556 557
// 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
}

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

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

576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591
// 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
}

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

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

// 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))
632
	}
633 634 635
	return extra
}

636
// MakePasswordList reads password lines from the file specified by --password.
637
func MakePasswordList(ctx *cli.Context) []string {
638 639 640 641 642 643 644 645 646 647 648 649
	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")
650
	}
651
	return lines
652 653
}

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

661
	config := &node.Config{
662
		DataDir:           MakeDataDir(ctx),
663 664 665
		KeyStoreDir:       ctx.GlobalString(KeyStoreDirFlag.Name),
		UseLightweightKDF: ctx.GlobalBool(LightKDFFlag.Name),
		PrivateKey:        MakeNodeKey(ctx),
666 667 668
		Name:              name,
		Version:           vsn,
		UserIdent:         makeNodeUserIdent(ctx),
669 670
		NoDiscovery:       ctx.GlobalBool(NoDiscoverFlag.Name) || ctx.GlobalBool(LightModeFlag.Name),
		DiscoveryV5:       ctx.GlobalBool(DiscoveryV5Flag.Name) || ctx.GlobalBool(LightModeFlag.Name) || ctx.GlobalInt(LightServFlag.Name) > 0,
671
		DiscoveryV5Addr:   MakeDiscoveryV5Address(ctx),
672
		BootstrapNodes:    MakeBootstrapNodes(ctx),
673
		BootstrapNodesV5:  MakeBootstrapNodesV5(ctx),
674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704
		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"
	}
	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.
705
func RegisterEthService(ctx *cli.Context, stack *node.Node, extra []byte) {
706 707 708 709 710 711 712 713 714 715 716 717
	// Avoid conflicting network flags
	networks, netFlags := 0, []cli.BoolFlag{DevModeFlag, TestNetFlag, OlympicFlag}
	for _, flag := range netFlags {
		if ctx.GlobalBool(flag.Name) {
			networks++
		}
	}
	if networks > 1 {
		Fatalf("The %v flags are mutually exclusive", netFlags)
	}

	ethConf := &eth.Config{
718
		Etherbase:               MakeEtherbase(stack.AccountManager(), ctx),
719
		ChainConfig:             MakeChainConfig(ctx, stack),
720
		FastSync:                ctx.GlobalBool(FastSyncFlag.Name),
721 722 723 724
		LightMode:               ctx.GlobalBool(LightModeFlag.Name),
		LightServ:               ctx.GlobalInt(LightServFlag.Name),
		LightPeers:              ctx.GlobalInt(LightPeersFlag.Name),
		MaxPeers:                ctx.GlobalInt(MaxPeersFlag.Name),
725
		DatabaseCache:           ctx.GlobalInt(CacheFlag.Name),
726
		DatabaseHandles:         MakeDatabaseHandles(),
Z
zsfelfoldi 已提交
727 728
		NetworkId:               ctx.GlobalInt(NetworkIdFlag.Name),
		MinerThreads:            ctx.GlobalInt(MinerThreadsFlag.Name),
729
		ExtraData:               MakeMinerExtra(extra, ctx),
Z
zsfelfoldi 已提交
730
		NatSpec:                 ctx.GlobalBool(NatspecEnabledFlag.Name),
Z
zelig 已提交
731
		DocRoot:                 ctx.GlobalString(DocRootFlag.Name),
Z
zsfelfoldi 已提交
732 733 734 735 736 737 738 739 740
		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),
741
	}
742

743 744 745 746 747 748
	// Override any default configs in dev mode or the test net
	switch {
	case ctx.GlobalBool(OlympicFlag.Name):
		if !ctx.GlobalIsSet(NetworkIdFlag.Name) {
			ethConf.NetworkId = 1
		}
749
		ethConf.Genesis = core.OlympicGenesisBlock()
750

751 752
	case ctx.GlobalBool(TestNetFlag.Name):
		if !ctx.GlobalIsSet(NetworkIdFlag.Name) {
753
			ethConf.NetworkId = 3
754
		}
755
		ethConf.Genesis = core.DefaultTestnetGenesisBlock()
756

757
	case ctx.GlobalBool(DevModeFlag.Name):
758
		ethConf.Genesis = core.OlympicGenesisBlock()
759 760
		if !ctx.GlobalIsSet(GasPriceFlag.Name) {
			ethConf.GasPrice = new(big.Int)
761
		}
762
		ethConf.PowTest = true
763
	}
764 765 766 767
	// Override any global options pertaining to the Ethereum protocol
	if gen := ctx.GlobalInt(TrieCacheGenFlag.Name); gen > 0 {
		state.MaxTrieCacheGen = uint16(gen)
	}
768

769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785
	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)
		}
786
	}
787 788 789 790 791 792 793
}

// RegisterShhService configures whisper and adds it to the given node.
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)
	}
794 795
}

796 797 798 799 800 801 802 803 804 805 806 807
// SetupNetwork configures the system for either the main net or some test network.
func SetupNetwork(ctx *cli.Context) {
	switch {
	case ctx.GlobalBool(OlympicFlag.Name):
		params.DurationLimit = big.NewInt(8)
		params.GenesisGasLimit = big.NewInt(3141592)
		params.MinGasLimit = big.NewInt(125000)
		params.MaximumExtraDataSize = big.NewInt(1024)
		NetworkIdFlag.Value = 0
		core.BlockReward = big.NewInt(1.5e+18)
		core.ExpDiffPeriod = big.NewInt(math.MaxInt64)
	}
808
	params.TargetGasLimit = common.String2Big(ctx.GlobalString(TargetGasLimitFlag.Name))
809 810
}

811
// MakeChainConfig reads the chain configuration from the database in ctx.Datadir.
812
func MakeChainConfig(ctx *cli.Context, stack *node.Node) *params.ChainConfig {
813
	db := MakeChainDatabase(ctx, stack)
814 815
	defer db.Close()

816
	return MakeChainConfigFromDb(ctx, db)
817 818
}

819
// MakeChainConfigFromDb reads the chain configuration from the given database.
820
func MakeChainConfigFromDb(ctx *cli.Context, db ethdb.Database) *params.ChainConfig {
821
	// If the chain is already initialized, use any existing chain configs
822
	config := new(params.ChainConfig)
823

824 825
	genesis := core.GetBlock(db, core.GetCanonicalHash(db, 0), 0)
	if genesis != nil {
826
		storedConfig, err := core.GetChainConfig(db, genesis.Hash())
827 828 829 830 831 832
		switch err {
		case nil:
			config = storedConfig
		case core.ChainConfigNotFoundErr:
			// No configs found, use empty, will populate below
		default:
833 834 835
			Fatalf("Could not make chain configuration: %v", err)
		}
	}
J
Jeffrey Wilcke 已提交
836 837 838 839
	// set chain id in case it's zero.
	if config.ChainId == nil {
		config.ChainId = new(big.Int)
	}
840 841 842 843 844 845 846 847
	// 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 {
848
		if ctx.GlobalBool(TestNetFlag.Name) {
849
			config = params.TestnetChainConfig
850
		} else {
851
			// Homestead fork
852
			config.HomesteadBlock = params.MainNetHomesteadBlock
853
			// DAO fork
854
			config.DAOForkBlock = params.MainNetDAOForkBlock
855
			config.DAOForkSupport = true
856

857
			// DoS reprice fork
858 859
			config.EIP150Block = params.MainNetHomesteadGasRepriceBlock
			config.EIP150Hash = params.MainNetHomesteadGasRepriceHash
860 861

			// DoS state cleanup fork
862 863 864
			config.EIP155Block = params.MainNetSpuriousDragon
			config.EIP158Block = params.MainNetSpuriousDragon
			config.ChainId = params.MainNetChainID
865
		}
866
	}
867
	// Force override any existing configs if explicitly requested
868 869
	switch {
	case ctx.GlobalBool(SupportDAOFork.Name):
870
		config.DAOForkSupport = true
871
	case ctx.GlobalBool(OpposeDAOFork.Name):
872
		config.DAOForkSupport = false
873
	}
874
	return config
875 876
}

877 878 879 880 881 882 883 884
func ChainDbName(ctx *cli.Context) string {
	if ctx.GlobalBool(LightModeFlag.Name) {
		return "lightchaindata"
	} else {
		return "chaindata"
	}
}

885
// MakeChainDatabase open an LevelDB using the flags passed to the client and will hard crash if it fails.
886
func MakeChainDatabase(ctx *cli.Context, stack *node.Node) ethdb.Database {
887 888 889
	var (
		cache   = ctx.GlobalInt(CacheFlag.Name)
		handles = MakeDatabaseHandles()
890
		name    = ChainDbName(ctx)
891
	)
892

893
	chainDb, err := stack.OpenDatabase(name, cache, handles)
894
	if err != nil {
895 896
		Fatalf("Could not open database: %v", err)
	}
897 898 899 900
	return chainDb
}

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

905
	if ctx.GlobalBool(OlympicFlag.Name) {
906
		_, err := core.WriteTestNetGenesisBlock(chainDb)
907 908 909 910
		if err != nil {
			glog.Fatalln(err)
		}
	}
911
	chainConfig := MakeChainConfigFromDb(ctx, chainDb)
912

913 914 915 916 917
	pow := pow.PoW(core.FakePow{})
	if !ctx.GlobalBool(FakePoWFlag.Name) {
		pow = ethash.New()
	}
	chain, err = core.NewBlockChain(chainDb, chainConfig, pow, new(event.TypeMux))
O
obscuren 已提交
918 919 920
	if err != nil {
		Fatalf("Could not start chainmanager: %v", err)
	}
921
	return chain, chainDb
922
}
923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939

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