go-ethereum/ethereum/ethereum.go

package main

import (
	"fmt"
	"github.com/ethereum/eth-go"
	"github.com/ethereum/eth-go/ethchain"
	"github.com/ethereum/eth-go/ethutil"
	"github.com/ethereum/go-ethereum/utils"
	"io/ioutil"
	"log"
	"os"
	"os/signal"
	"runtime"
	"strings"
)

const Debug = true

func RegisterInterrupt(cb func(os.Signal)) {
	go func() {
		// Buffered chan of one is enough
		c := make(chan os.Signal, 1)
		// Notify about interrupts for now
		signal.Notify(c, os.Interrupt)

		for sig := range c {
			cb(sig)
		}
	}()
}

func confirm(message string) bool {
	fmt.Println(message, "Are you sure? (y/n)")
	var r string
	fmt.Scanln(&r)
	for ; ; fmt.Scanln(&r) {
		if r == "n" || r == "y" {
			break
		} else {
			fmt.Printf("Yes or no?", r)
		}
	}
	return r == "y"
}

func main() {
	Init()

	runtime.GOMAXPROCS(runtime.NumCPU())

	// set logger
	var logSys *log.Logger
	flags := log.LstdFlags

	if StartJsConsole || len(InputFile) > 0 {
		ethutil.ReadConfig(DataDir, ethutil.LogFile)
	} else {
		ethutil.ReadConfig(DataDir, ethutil.LogFile|ethutil.LogStd)
	}

	logger := ethutil.Config.Log

	if LogFile != "" {
		logfile, err := os.OpenFile(LogFile, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)
		if err != nil {
			panic(fmt.Sprintf("error opening log file '%s': %v", LogFile, err))
		}
		defer logfile.Close()
		log.SetOutput(logfile)
		logSys = log.New(logfile, "", flags)
		logger.AddLogSystem(logSys)
	} else {
		logSys = log.New(os.Stdout, "", flags)
	}

	ethchain.InitFees()

	// Instantiated a eth stack
	ethereum, err := eth.New(eth.CapDefault, UseUPnP)
	if err != nil {
		log.Println("eth start err:", err)
		return
	}
	ethereum.Port = OutboundPort

	// bookkeeping tasks
	switch {
	case GenAddr:
		if NonInteractive || confirm("This action overwrites your old private key.") {
			utils.CreateKeyPair(true)
		}
		os.Exit(0)
	case len(ImportKey) > 0:
		if NonInteractive || confirm("This action overwrites your old private key.") {
			mnemonic := strings.Split(ImportKey, " ")
			if len(mnemonic) == 24 {
				logSys.Println("Got mnemonic key, importing.")
				key := ethutil.MnemonicDecode(mnemonic)
				utils.ImportPrivateKey(key)
			} else if len(mnemonic) == 1 {
				logSys.Println("Got hex key, importing.")
				utils.ImportPrivateKey(ImportKey)
			} else {
				logSys.Println("Did not recognise format, exiting.")
			}
		}
		os.Exit(0)
	case ExportKey:
		keyPair := ethutil.GetKeyRing().Get(0)
		fmt.Printf(`
Generating new address and keypair.
Please keep your keys somewhere save.

++++++++++++++++ KeyRing +++++++++++++++++++
addr: %x
prvk: %x
pubk: %x
++++++++++++++++++++++++++++++++++++++++++++
save these words so you can restore your account later: %s
`, keyPair.Address(), keyPair.PrivateKey, keyPair.PublicKey)

		os.Exit(0)
	case ShowGenesis:
		logSys.Println(ethereum.BlockChain().Genesis())
		os.Exit(0)
	default:
		// Creates a keypair if non exists
		utils.CreateKeyPair(false)
	}

	// client
	logger.Infoln(fmt.Sprintf("Starting Ethereum v%s", ethutil.Config.Ver))

	// Set the max peers
	ethereum.MaxPeers = MaxPeer

	// Set Mining status
	ethereum.Mining = StartMining

	if StartMining {
		utils.DoMining(ethereum)
	}

	if StartJsConsole {
		repl := NewJSRepl(ethereum)

		go repl.Start()

		RegisterInterrupt(func(os.Signal) {
			repl.Stop()
		})
	} else if len(InputFile) > 0 {
		file, err := os.Open(InputFile)
		if err != nil {
			ethutil.Config.Log.Fatal(err)
		}

		content, err := ioutil.ReadAll(file)
		if err != nil {
			ethutil.Config.Log.Fatal(err)
		}

		re := NewJSRE(ethereum)
		RegisterInterrupt(func(os.Signal) {
			re.Stop()
		})
		re.Run(string(content))
	}

	if StartRpc {
		utils.DoRpc(ethereum, RpcPort)
	}

	RegisterInterrupt(func(sig os.Signal) {
		fmt.Printf("Shutting down (%v) ... \n", sig)
		ethereum.Stop()
	})

	ethereum.Start(UseSeed)

	// Wait for shutdown
	ethereum.WaitForShutdown()
}