126 lines
4.1 KiB
Go
126 lines
4.1 KiB
Go
// Copyright (c) 2013 Conformal Systems LLC.
|
|
// Use of this source code is governed by an ISC
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/conformal/btcdb"
|
|
_ "github.com/conformal/btcdb/ldb"
|
|
"github.com/conformal/btcnet"
|
|
"github.com/conformal/btcutil"
|
|
"github.com/conformal/btcwire"
|
|
"github.com/conformal/go-flags"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
const (
|
|
minCandidates = 1
|
|
maxCandidates = 20
|
|
defaultNumCandidates = 5
|
|
defaultDbType = "leveldb"
|
|
)
|
|
|
|
var (
|
|
btcdHomeDir = btcutil.AppDataDir("btcd", false)
|
|
defaultDataDir = filepath.Join(btcdHomeDir, "data")
|
|
knownDbTypes = btcdb.SupportedDBs()
|
|
activeNetwork = &btcnet.MainNetParams
|
|
)
|
|
|
|
// config defines the configuration options for findcheckpoint.
|
|
//
|
|
// See loadConfig for details on the configuration load process.
|
|
type config struct {
|
|
DataDir string `short:"b" long:"datadir" description:"Location of the btcd data directory"`
|
|
DbType string `long:"dbtype" description:"Database backend to use for the Block Chain"`
|
|
TestNet3 bool `long:"testnet" description:"Use the test network"`
|
|
NumCandidates int `short:"n" long:"numcandidates" description:"Max num of checkpoint candidates to show {1-20}"`
|
|
UseGoOutput bool `short:"g" long:"gooutput" description:"Display the candidates using Go syntax that is ready to insert into the btcchain checkpoint list"`
|
|
}
|
|
|
|
// validDbType returns whether or not dbType is a supported database type.
|
|
func validDbType(dbType string) bool {
|
|
for _, knownType := range knownDbTypes {
|
|
if dbType == knownType {
|
|
return true
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
// netName returns the name used when referring to a bitcoin network. At the
|
|
// time of writing, btcd currently places blocks for testnet version 3 in the
|
|
// data and log directory "testnet", which does not match the Name field of the
|
|
// btcnet parameters. This function can be used to override this directory name
|
|
// as "testnet" when the passed active network matches btcwire.TestNet3.
|
|
//
|
|
// A proper upgrade to move the data and log directories for this network to
|
|
// "testnet3" is planned for the future, at which point this function can be
|
|
// removed and the network parameter's name used instead.
|
|
func netName(netParams *btcnet.Params) string {
|
|
switch netParams.Net {
|
|
case btcwire.TestNet3:
|
|
return "testnet"
|
|
default:
|
|
return netParams.Name
|
|
}
|
|
}
|
|
|
|
// loadConfig initializes and parses the config using command line options.
|
|
func loadConfig() (*config, []string, error) {
|
|
// Default config.
|
|
cfg := config{
|
|
DataDir: defaultDataDir,
|
|
DbType: defaultDbType,
|
|
NumCandidates: defaultNumCandidates,
|
|
}
|
|
|
|
// Parse command line options.
|
|
parser := flags.NewParser(&cfg, flags.Default)
|
|
remainingArgs, err := parser.Parse()
|
|
if err != nil {
|
|
if e, ok := err.(*flags.Error); !ok || e.Type != flags.ErrHelp {
|
|
parser.WriteHelp(os.Stderr)
|
|
}
|
|
return nil, nil, err
|
|
}
|
|
|
|
// Choose the active network based on the flags.
|
|
if cfg.TestNet3 {
|
|
activeNetwork = &btcnet.TestNet3Params
|
|
}
|
|
|
|
// Validate database type.
|
|
if !validDbType(cfg.DbType) {
|
|
str := "%s: The specified database type [%v] is invalid -- " +
|
|
"supported types %v"
|
|
err := fmt.Errorf(str, "loadConfig", cfg.DbType, knownDbTypes)
|
|
fmt.Fprintln(os.Stderr, err)
|
|
parser.WriteHelp(os.Stderr)
|
|
return nil, nil, err
|
|
}
|
|
|
|
// Append the network type to the data directory so it is "namespaced"
|
|
// per network. In addition to the block database, there are other
|
|
// pieces of data that are saved to disk such as address manager state.
|
|
// All data is specific to a network, so namespacing the data directory
|
|
// means each individual piece of serialized data does not have to
|
|
// worry about changing names per network and such.
|
|
cfg.DataDir = filepath.Join(cfg.DataDir, netName(activeNetwork))
|
|
|
|
// Validate the number of candidates.
|
|
if cfg.NumCandidates < minCandidates || cfg.NumCandidates > maxCandidates {
|
|
str := "%s: The specified number of candidates is out of " +
|
|
"range -- parsed [%v]"
|
|
err = fmt.Errorf(str, "loadConfig", cfg.NumCandidates)
|
|
fmt.Fprintln(os.Stderr, err)
|
|
parser.WriteHelp(os.Stderr)
|
|
return nil, nil, err
|
|
}
|
|
|
|
return &cfg, remainingArgs, nil
|
|
}
|