cmd.go 5.45 KB
Newer Older
1 2 3 4 5 6 7 8 9 10
// Copyright 2014 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
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/>.
obscuren's avatar
obscuren committed
16

17
// Package utils contains internal helper functions for go-ethereum commands.
18 19 20
package utils

import (
zelig's avatar
zelig committed
21
	"fmt"
22
	"io"
23 24
	"os"
	"os/signal"
obscuren's avatar
obscuren committed
25
	"regexp"
26

27
	"github.com/ethereum/go-ethereum/common"
28
	"github.com/ethereum/go-ethereum/core"
obscuren's avatar
obscuren committed
29
	"github.com/ethereum/go-ethereum/core/types"
30
	"github.com/ethereum/go-ethereum/internal/debug"
obscuren's avatar
obscuren committed
31
	"github.com/ethereum/go-ethereum/logger"
obscuren's avatar
obscuren committed
32
	"github.com/ethereum/go-ethereum/logger/glog"
33
	"github.com/ethereum/go-ethereum/node"
obscuren's avatar
obscuren committed
34
	"github.com/ethereum/go-ethereum/rlp"
35 36
)

37 38 39 40
const (
	importBatchSize = 2500
)

zelig's avatar
zelig committed
41
func openLogFile(Datadir string, filename string) *os.File {
obscuren's avatar
obscuren committed
42
	path := common.AbsolutePath(Datadir, filename)
zelig's avatar
zelig committed
43 44 45 46 47 48 49
	file, err := os.OpenFile(path, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)
	if err != nil {
		panic(fmt.Sprintf("error opening log file '%s': %v", filename, err))
	}
	return file
}

50 51 52
// Fatalf formats a message to standard error and exits the program.
// The message is also printed to standard output if standard error
// is redirected to a different file.
53
func Fatalf(format string, args ...interface{}) {
54 55 56 57 58 59 60
	w := io.MultiWriter(os.Stdout, os.Stderr)
	outf, _ := os.Stdout.Stat()
	errf, _ := os.Stderr.Stat()
	if outf != nil && errf != nil && os.SameFile(outf, errf) {
		w = os.Stderr
	}
	fmt.Fprintf(w, "Fatal: "+format+"\n", args...)
61 62 63 64
	logger.Flush()
	os.Exit(1)
}

65 66 67
func StartNode(stack *node.Node) {
	if err := stack.Start(); err != nil {
		Fatalf("Error starting protocol stack: %v", err)
obscuren's avatar
obscuren committed
68
	}
69 70 71 72 73 74
	go func() {
		sigc := make(chan os.Signal, 1)
		signal.Notify(sigc, os.Interrupt)
		defer signal.Stop(sigc)
		<-sigc
		glog.V(logger.Info).Infoln("Got interrupt, shutting down...")
75
		go stack.Stop()
76 77 78
		for i := 10; i > 0; i-- {
			<-sigc
			if i > 1 {
79
				glog.V(logger.Info).Infof("Already shutting down, interrupt %d more times for panic.", i-1)
80 81
			}
		}
82
		debug.Exit() // ensure trace and CPU profile data is flushed.
83
		debug.LoudPanic("boom")
84
	}()
85 86
}

obscuren's avatar
obscuren committed
87
func FormatTransactionData(data string) []byte {
obscuren's avatar
obscuren committed
88
	d := common.StringToByteFunc(data, func(s string) (ret []byte) {
obscuren's avatar
obscuren committed
89 90
		slice := regexp.MustCompile("\\n|\\s").Split(s, 1000000000)
		for _, dataItem := range slice {
obscuren's avatar
obscuren committed
91
			d := common.FormatData(dataItem)
obscuren's avatar
obscuren committed
92 93 94 95 96 97 98 99
			ret = append(ret, d...)
		}
		return
	})

	return d
}

100
func ImportChain(chain *core.BlockChain, fn string) error {
101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122
	// Watch for Ctrl-C while the import is running.
	// If a signal is received, the import will stop at the next batch.
	interrupt := make(chan os.Signal, 1)
	stop := make(chan struct{})
	signal.Notify(interrupt, os.Interrupt)
	defer signal.Stop(interrupt)
	defer close(interrupt)
	go func() {
		if _, ok := <-interrupt; ok {
			glog.Info("caught interrupt during import, will stop at next batch")
		}
		close(stop)
	}()
	checkInterrupt := func() bool {
		select {
		case <-stop:
			return true
		default:
			return false
		}
	}

123
	glog.Infoln("Importing blockchain ", fn)
124
	fh, err := os.Open(fn)
obscuren's avatar
obscuren committed
125 126 127 128
	if err != nil {
		return err
	}
	defer fh.Close()
129
	stream := rlp.NewStream(fh, 0)
130

131
	// Run actual the import.
132
	blocks := make(types.Blocks, importBatchSize)
133
	n := 0
134
	for batch := 0; ; batch++ {
135
		// Load a batch of RLP blocks.
136 137 138
		if checkInterrupt() {
			return fmt.Errorf("interrupted")
		}
139
		i := 0
140
		for ; i < importBatchSize; i++ {
141 142 143 144 145
			var b types.Block
			if err := stream.Decode(&b); err == io.EOF {
				break
			} else if err != nil {
				return fmt.Errorf("at block %d: %v", n, err)
146
			}
147 148 149 150 151
			// don't import first block
			if b.NumberU64() == 0 {
				i--
				continue
			}
152 153
			blocks[i] = &b
			n++
154
		}
155 156 157 158
		if i == 0 {
			break
		}
		// Import the batch.
159 160 161
		if checkInterrupt() {
			return fmt.Errorf("interrupted")
		}
162 163 164 165 166
		if hasAllBlocks(chain, blocks[:i]) {
			glog.Infof("skipping batch %d, all blocks present [%x / %x]",
				batch, blocks[0].Hash().Bytes()[:4], blocks[i-1].Hash().Bytes()[:4])
			continue
		}
167

168
		if _, err := chain.InsertChain(blocks[:i]); err != nil {
169
			return fmt.Errorf("invalid block %d: %v", n, err)
170
		}
obscuren's avatar
obscuren committed
171 172 173
	}
	return nil
}
174

175
func hasAllBlocks(chain *core.BlockChain, bs []*types.Block) bool {
176 177 178 179 180 181 182 183
	for _, b := range bs {
		if !chain.HasBlock(b.Hash()) {
			return false
		}
	}
	return true
}

184
func ExportChain(blockchain *core.BlockChain, fn string) error {
185
	glog.Infoln("Exporting blockchain to ", fn)
186
	fh, err := os.OpenFile(fn, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, os.ModePerm)
187 188 189 190
	if err != nil {
		return err
	}
	defer fh.Close()
191
	if err := blockchain.Export(fh); err != nil {
192 193
		return err
	}
194
	glog.Infoln("Exported blockchain to ", fn)
195 196
	return nil
}
197

198
func ExportAppendChain(blockchain *core.BlockChain, fn string, first uint64, last uint64) error {
199
	glog.Infoln("Exporting blockchain to ", fn)
200 201 202 203 204 205
	// TODO verify mode perms
	fh, err := os.OpenFile(fn, os.O_CREATE|os.O_APPEND|os.O_WRONLY, os.ModePerm)
	if err != nil {
		return err
	}
	defer fh.Close()
206
	if err := blockchain.ExportN(fh, first, last); err != nil {
207 208
		return err
	}
209
	glog.Infoln("Exported blockchain to ", fn)
210 211
	return nil
}