plugeth/ethereum/ethereum.go

180 lines
3.9 KiB
Go
Raw Normal View History

2014-03-22 11:03:10 +00:00
package main
import (
"encoding/hex"
2014-03-22 11:03:10 +00:00
"fmt"
"github.com/ethereum/eth-go"
"github.com/ethereum/eth-go/ethchain"
2014-03-24 09:56:42 +00:00
"github.com/ethereum/eth-go/ethminer"
2014-05-02 18:01:25 +00:00
"github.com/ethereum/eth-go/ethpub"
"github.com/ethereum/eth-go/ethrpc"
2014-03-22 11:03:10 +00:00
"github.com/ethereum/eth-go/ethutil"
"github.com/ethereum/go-ethereum/utils"
"log"
"os"
"os/signal"
"runtime"
2014-04-09 14:29:52 +00:00
"strings"
2014-03-22 11:03:10 +00:00
)
const Debug = true
// Register interrupt handlers so we can stop the ethereum
2014-04-16 14:01:22 +00:00
func RegisterInterrupts(s *eth.Ethereum) {
2014-03-22 11:03:10 +00:00
// Buffered chan of one is enough
c := make(chan os.Signal, 1)
// Notify about interrupts for now
signal.Notify(c, os.Interrupt)
go func() {
for sig := range c {
fmt.Printf("Shutting down (%v) ... \n", sig)
s.Stop()
}
}()
}
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"
}
2014-03-22 11:03:10 +00:00
func main() {
Init()
runtime.GOMAXPROCS(runtime.NumCPU())
// set logger
var logSys *log.Logger
flags := log.LstdFlags
ethutil.ReadConfig(DataDir)
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()
2014-03-22 11:03:10 +00:00
// 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.") {
2014-03-22 11:03:10 +00:00
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.")
2014-03-22 11:03:10 +00:00
}
}
os.Exit(0)
case ExportKey:
2014-03-22 11:03:10 +00:00
key := ethutil.Config.Db.GetKeys()[0]
logSys.Println(fmt.Sprintf("prvk: %x\n", key.PrivateKey))
2014-03-22 11:03:10 +00:00
os.Exit(0)
case ShowGenesis:
logSys.Println(ethereum.BlockChain().Genesis())
2014-03-22 11:03:10 +00:00
os.Exit(0)
default:
// Creates a keypair if non exists
utils.CreateKeyPair(false)
2014-03-22 11:03:10 +00:00
}
// client
logger.Infoln(fmt.Sprintf("Starting Ethereum v%s", ethutil.Config.Ver))
2014-03-22 11:03:10 +00:00
// Set the max peers
ethereum.MaxPeers = MaxPeer
// Set Mining status
ethereum.Mining = StartMining
2014-03-22 11:03:10 +00:00
if StartMining {
logger.Infoln("Miner started")
2014-03-22 11:03:10 +00:00
// Fake block mining. It broadcasts a new block every 5 seconds
go func() {
2014-03-24 09:56:42 +00:00
if StartMining {
logger.Infoln("Miner started")
2014-03-24 09:56:42 +00:00
go func() {
data, _ := ethutil.Config.Db.Get([]byte("KeyRing"))
keyRing := ethutil.NewValueFromBytes(data)
addr := keyRing.Get(1).Bytes()
pair, _ := ethchain.NewKeyPairFromSec(ethutil.FromHex(hex.EncodeToString(addr)))
miner := ethminer.NewDefaultMiner(pair.Address(), ethereum)
2014-03-24 09:56:42 +00:00
miner.Start()
}()
2014-03-22 11:03:10 +00:00
}
}()
2014-03-24 09:56:42 +00:00
2014-03-22 11:03:10 +00:00
}
if StartConsole {
err := os.Mkdir(ethutil.Config.ExecPath, os.ModePerm)
// Error is OK if the error is ErrExist
if err != nil && !os.IsExist(err) {
log.Panic("Unable to create EXECPATH:", err)
}
console := NewConsole(ethereum)
go console.Start()
}
if StartRpc {
ethereum.RpcServer, err = ethrpc.NewJsonRpcServer(ethpub.NewPEthereum(ethereum), RpcPort)
if err != nil {
logger.Infoln("Could not start RPC interface:", err)
} else {
go ethereum.RpcServer.Start()
}
}
RegisterInterrupts(ethereum)
ethereum.Start(UseSeed)
2014-03-22 11:03:10 +00:00
// Wait for shutdown
ethereum.WaitForShutdown()
}