2015-07-06 17:54:22 -07:00
|
|
|
// 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
|
2015-07-22 09:48:40 -07:00
|
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
2015-07-06 17:54:22 -07:00
|
|
|
// GNU General Public License for more details.
|
|
|
|
//
|
|
|
|
// You should have received a copy of the GNU General Public License
|
2015-07-22 09:48:40 -07:00
|
|
|
// along with go-ethereum. If not, see <http://www.gnu.org/licenses/>.
|
2015-07-06 17:54:22 -07:00
|
|
|
|
2015-05-27 04:43:49 -07:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
"os"
|
|
|
|
"path/filepath"
|
|
|
|
"strconv"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
"github.com/codegangsta/cli"
|
|
|
|
"github.com/ethereum/go-ethereum/cmd/utils"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
|
|
"github.com/ethereum/go-ethereum/core"
|
|
|
|
"github.com/ethereum/go-ethereum/core/state"
|
|
|
|
"github.com/ethereum/go-ethereum/core/types"
|
2015-09-14 00:35:57 -07:00
|
|
|
"github.com/ethereum/go-ethereum/ethdb"
|
2015-05-27 04:43:49 -07:00
|
|
|
"github.com/ethereum/go-ethereum/logger/glog"
|
|
|
|
)
|
|
|
|
|
|
|
|
var (
|
|
|
|
importCommand = cli.Command{
|
|
|
|
Action: importChain,
|
|
|
|
Name: "import",
|
|
|
|
Usage: `import a blockchain file`,
|
|
|
|
}
|
|
|
|
exportCommand = cli.Command{
|
|
|
|
Action: exportChain,
|
|
|
|
Name: "export",
|
|
|
|
Usage: `export blockchain into file`,
|
2015-06-05 21:02:32 -07:00
|
|
|
Description: `
|
|
|
|
Requires a first argument of the file to write to.
|
|
|
|
Optional second and third arguments control the first and
|
|
|
|
last block to write. In this mode, the file will be appended
|
|
|
|
if already existing.
|
|
|
|
`,
|
2015-05-27 04:43:49 -07:00
|
|
|
}
|
|
|
|
upgradedbCommand = cli.Command{
|
|
|
|
Action: upgradeDB,
|
|
|
|
Name: "upgradedb",
|
|
|
|
Usage: "upgrade chainblock database",
|
|
|
|
}
|
|
|
|
removedbCommand = cli.Command{
|
|
|
|
Action: removeDB,
|
|
|
|
Name: "removedb",
|
|
|
|
Usage: "Remove blockchain and state databases",
|
|
|
|
}
|
|
|
|
dumpCommand = cli.Command{
|
|
|
|
Action: dump,
|
|
|
|
Name: "dump",
|
|
|
|
Usage: `dump a specific block from storage`,
|
|
|
|
Description: `
|
|
|
|
The arguments are interpreted as block numbers or hashes.
|
|
|
|
Use "ethereum dump 0" to dump the genesis block.
|
|
|
|
`,
|
|
|
|
}
|
|
|
|
)
|
|
|
|
|
|
|
|
func importChain(ctx *cli.Context) {
|
|
|
|
if len(ctx.Args()) != 1 {
|
|
|
|
utils.Fatalf("This command requires an argument.")
|
|
|
|
}
|
2015-08-06 10:57:39 -07:00
|
|
|
chain, chainDb := utils.MakeChain(ctx)
|
2015-05-27 04:43:49 -07:00
|
|
|
start := time.Now()
|
2015-05-27 07:02:08 -07:00
|
|
|
err := utils.ImportChain(chain, ctx.Args().First())
|
2015-08-06 10:57:39 -07:00
|
|
|
chainDb.Close()
|
2015-05-27 07:02:08 -07:00
|
|
|
if err != nil {
|
|
|
|
utils.Fatalf("Import error: %v", err)
|
|
|
|
}
|
2015-05-27 04:43:49 -07:00
|
|
|
fmt.Printf("Import done in %v", time.Since(start))
|
|
|
|
}
|
|
|
|
|
|
|
|
func exportChain(ctx *cli.Context) {
|
2015-06-06 06:50:23 -07:00
|
|
|
if len(ctx.Args()) < 1 {
|
2015-05-27 04:43:49 -07:00
|
|
|
utils.Fatalf("This command requires an argument.")
|
|
|
|
}
|
2015-08-06 10:57:39 -07:00
|
|
|
chain, _ := utils.MakeChain(ctx)
|
2015-05-27 04:43:49 -07:00
|
|
|
start := time.Now()
|
2015-06-05 21:02:32 -07:00
|
|
|
|
|
|
|
var err error
|
2015-06-06 07:04:13 -07:00
|
|
|
fp := ctx.Args().First()
|
2015-06-05 21:02:32 -07:00
|
|
|
if len(ctx.Args()) < 3 {
|
2015-06-06 07:04:13 -07:00
|
|
|
err = utils.ExportChain(chain, fp)
|
2015-06-05 21:02:32 -07:00
|
|
|
} else {
|
|
|
|
// This can be improved to allow for numbers larger than 9223372036854775807
|
|
|
|
first, ferr := strconv.ParseInt(ctx.Args().Get(1), 10, 64)
|
|
|
|
last, lerr := strconv.ParseInt(ctx.Args().Get(2), 10, 64)
|
|
|
|
if ferr != nil || lerr != nil {
|
2015-06-06 07:04:13 -07:00
|
|
|
utils.Fatalf("Export error in parsing parameters: block number not an integer\n")
|
2015-06-05 21:02:32 -07:00
|
|
|
}
|
2015-06-06 07:04:13 -07:00
|
|
|
if first < 0 || last < 0 {
|
|
|
|
utils.Fatalf("Export error: block number must be greater than 0\n")
|
|
|
|
}
|
|
|
|
err = utils.ExportAppendChain(chain, fp, uint64(first), uint64(last))
|
2015-06-05 21:02:32 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
if err != nil {
|
2015-05-27 04:43:49 -07:00
|
|
|
utils.Fatalf("Export error: %v\n", err)
|
|
|
|
}
|
|
|
|
fmt.Printf("Export done in %v", time.Since(start))
|
|
|
|
}
|
|
|
|
|
|
|
|
func removeDB(ctx *cli.Context) {
|
2015-08-18 06:56:37 -07:00
|
|
|
confirm, err := utils.PromptConfirm("Remove local database?")
|
2015-05-27 04:43:49 -07:00
|
|
|
if err != nil {
|
|
|
|
utils.Fatalf("%v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if confirm {
|
2015-08-18 06:56:37 -07:00
|
|
|
fmt.Println("Removing chaindata...")
|
2015-05-27 04:43:49 -07:00
|
|
|
start := time.Now()
|
|
|
|
|
2015-08-18 06:56:37 -07:00
|
|
|
os.RemoveAll(filepath.Join(ctx.GlobalString(utils.DataDirFlag.Name), "chaindata"))
|
2015-05-27 04:43:49 -07:00
|
|
|
|
|
|
|
fmt.Printf("Removed in %v\n", time.Since(start))
|
|
|
|
} else {
|
|
|
|
fmt.Println("Operation aborted")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func upgradeDB(ctx *cli.Context) {
|
|
|
|
glog.Infoln("Upgrading blockchain database")
|
|
|
|
|
2015-08-06 10:57:39 -07:00
|
|
|
chain, chainDb := utils.MakeChain(ctx)
|
2015-12-10 16:33:45 -08:00
|
|
|
bcVersion := core.GetBlockChainVersion(chainDb)
|
2015-05-27 04:43:49 -07:00
|
|
|
if bcVersion == 0 {
|
|
|
|
bcVersion = core.BlockChainVersion
|
|
|
|
}
|
|
|
|
|
|
|
|
// Export the current chain.
|
|
|
|
filename := fmt.Sprintf("blockchain_%d_%s.chain", bcVersion, time.Now().Format("20060102_150405"))
|
|
|
|
exportFile := filepath.Join(ctx.GlobalString(utils.DataDirFlag.Name), filename)
|
|
|
|
if err := utils.ExportChain(chain, exportFile); err != nil {
|
2015-05-27 07:02:08 -07:00
|
|
|
utils.Fatalf("Unable to export chain for reimport %s", err)
|
2015-05-27 04:43:49 -07:00
|
|
|
}
|
2015-08-06 10:57:39 -07:00
|
|
|
chainDb.Close()
|
|
|
|
os.RemoveAll(filepath.Join(ctx.GlobalString(utils.DataDirFlag.Name), "chaindata"))
|
2015-05-27 04:43:49 -07:00
|
|
|
|
|
|
|
// Import the chain file.
|
2015-08-06 10:57:39 -07:00
|
|
|
chain, chainDb = utils.MakeChain(ctx)
|
2015-12-10 16:33:45 -08:00
|
|
|
core.WriteBlockChainVersion(chainDb, core.BlockChainVersion)
|
2015-05-27 04:43:49 -07:00
|
|
|
err := utils.ImportChain(chain, exportFile)
|
2015-08-06 10:57:39 -07:00
|
|
|
chainDb.Close()
|
2015-05-27 04:43:49 -07:00
|
|
|
if err != nil {
|
2015-05-27 07:02:08 -07:00
|
|
|
utils.Fatalf("Import error %v (a backup is made in %s, use the import command to import it)", err, exportFile)
|
2015-05-27 04:43:49 -07:00
|
|
|
} else {
|
|
|
|
os.Remove(exportFile)
|
|
|
|
glog.Infoln("Import finished")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func dump(ctx *cli.Context) {
|
2015-08-06 10:57:39 -07:00
|
|
|
chain, chainDb := utils.MakeChain(ctx)
|
2015-05-27 04:43:49 -07:00
|
|
|
for _, arg := range ctx.Args() {
|
|
|
|
var block *types.Block
|
|
|
|
if hashish(arg) {
|
|
|
|
block = chain.GetBlock(common.HexToHash(arg))
|
|
|
|
} else {
|
|
|
|
num, _ := strconv.Atoi(arg)
|
|
|
|
block = chain.GetBlockByNumber(uint64(num))
|
|
|
|
}
|
|
|
|
if block == nil {
|
|
|
|
fmt.Println("{}")
|
|
|
|
utils.Fatalf("block not found")
|
|
|
|
} else {
|
2015-10-06 07:35:55 -07:00
|
|
|
state, err := state.New(block.Root(), chainDb)
|
|
|
|
if err != nil {
|
|
|
|
utils.Fatalf("could not create new state: %v", err)
|
|
|
|
return
|
|
|
|
}
|
2015-05-27 04:43:49 -07:00
|
|
|
fmt.Printf("%s\n", state.Dump())
|
|
|
|
}
|
|
|
|
}
|
2015-08-06 10:57:39 -07:00
|
|
|
chainDb.Close()
|
2015-05-27 04:43:49 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
// hashish returns true for strings that look like hashes.
|
|
|
|
func hashish(x string) bool {
|
|
|
|
_, err := strconv.Atoi(x)
|
|
|
|
return err != nil
|
|
|
|
}
|
|
|
|
|
2015-09-14 00:35:57 -07:00
|
|
|
func closeAll(dbs ...ethdb.Database) {
|
2015-05-27 04:43:49 -07:00
|
|
|
for _, db := range dbs {
|
|
|
|
db.Close()
|
|
|
|
}
|
|
|
|
}
|