Add new tool to rewind the database to a specific height or sha
Initial code for a block show/dump tool. Initial code for a simple block inserter.
This commit is contained in:
Dale Rahn
parent
a10da1ed6c
commit
7659a134c9
3 changed files with 744 additions and 0 deletions
264
util/addblock/addblock.go
Normal file
264
util/addblock/addblock.go
Normal file
|
|
@ -0,0 +1,264 @@
|
|||
/*
|
||||
* Copyright (c) 2013 Conformal Systems LLC.
|
||||
*/
|
||||
|
||||
package main
|
||||
|
||||
import (
|
||||
"encoding/binary"
|
||||
"flag"
|
||||
"fmt"
|
||||
"github.com/conformal/btcdb"
|
||||
"github.com/conformal/btcutil"
|
||||
_ "github.com/conformal/btcdb/sqlite3"
|
||||
_ "github.com/conformal/btcdb/ldb"
|
||||
"github.com/conformal/btcwire"
|
||||
"github.com/conformal/seelog"
|
||||
"io"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"runtime"
|
||||
"runtime/pprof"
|
||||
)
|
||||
|
||||
type ShaHash btcwire.ShaHash
|
||||
|
||||
var log seelog.LoggerInterface
|
||||
|
||||
const (
|
||||
ArgSha = iota
|
||||
ArgHeight
|
||||
)
|
||||
|
||||
type bufQueue struct {
|
||||
height int64
|
||||
blkbuf []byte
|
||||
}
|
||||
|
||||
type blkQueue struct {
|
||||
complete chan bool
|
||||
height int64
|
||||
blk *btcutil.Block
|
||||
}
|
||||
|
||||
func main() {
|
||||
var err error
|
||||
var dbType string
|
||||
var datadir string
|
||||
var infile string
|
||||
var progress int
|
||||
flag.StringVar(&dbType, "dbtype", "", "Database backend to use for the Block Chain")
|
||||
flag.StringVar(&datadir, "datadir", "", "Directory to store data")
|
||||
flag.StringVar(&infile, "i", "", "infile")
|
||||
flag.IntVar(&progress, "p", 0, "show progress")
|
||||
|
||||
flag.Parse()
|
||||
|
||||
runtime.GOMAXPROCS(runtime.NumCPU())
|
||||
|
||||
if len (infile) == 0 {
|
||||
fmt.Printf("Must specify inputfile")
|
||||
return
|
||||
}
|
||||
|
||||
log, err = seelog.LoggerFromWriterWithMinLevel(os.Stdout,
|
||||
seelog.InfoLvl)
|
||||
if err != nil {
|
||||
fmt.Fprintf(os.Stderr, "failed to create logger: %v", err)
|
||||
return
|
||||
}
|
||||
defer log.Flush()
|
||||
btcdb.UseLogger(log)
|
||||
|
||||
if len(dbType) == 0 {
|
||||
dbType = "sqlite"
|
||||
}
|
||||
|
||||
if len(datadir) == 0 {
|
||||
datadir = filepath.Join(btcdHomeDir(), "data")
|
||||
}
|
||||
datadir = filepath.Join(datadir, "mainnet")
|
||||
|
||||
err = os.MkdirAll(datadir, 0700)
|
||||
if err != nil {
|
||||
fmt.Printf("unable to create db repo area %v, %v", datadir, err)
|
||||
}
|
||||
|
||||
|
||||
blockDbNamePrefix := "blocks"
|
||||
dbName := blockDbNamePrefix + "_" + dbType
|
||||
if dbType == "sqlite" {
|
||||
dbName = dbName + ".db"
|
||||
}
|
||||
dbPath := filepath.Join(datadir, dbName)
|
||||
|
||||
log.Infof("loading db")
|
||||
db, err := btcdb.CreateDB(dbType, dbPath)
|
||||
if err != nil {
|
||||
log.Warnf("db open failed: %v", err)
|
||||
return
|
||||
}
|
||||
defer db.Close()
|
||||
log.Infof("db created")
|
||||
|
||||
|
||||
var fi io.ReadCloser
|
||||
|
||||
fi, err = os.Open(infile)
|
||||
if err != nil {
|
||||
log.Warnf("failed to open file %v, err %v", infile, err)
|
||||
}
|
||||
defer func() {
|
||||
if err := fi.Close(); err != nil {
|
||||
log.Warn("failed to close file %v %v", infile, err)
|
||||
}
|
||||
}()
|
||||
|
||||
bufqueue := make(chan *bufQueue, 2)
|
||||
blkqueue := make(chan *blkQueue, 2)
|
||||
|
||||
for i := 0; i < runtime.NumCPU(); i++ {
|
||||
go processBuf(i, bufqueue, blkqueue)
|
||||
}
|
||||
go processBuf(0, bufqueue, blkqueue)
|
||||
|
||||
go readBlocks(fi, bufqueue)
|
||||
|
||||
var eheight int64
|
||||
doneMap := map [int64] *blkQueue {}
|
||||
for {
|
||||
|
||||
select {
|
||||
case blkM := <- blkqueue:
|
||||
doneMap[blkM.height] = blkM
|
||||
|
||||
for {
|
||||
if blkP, ok := doneMap[eheight]; ok {
|
||||
delete(doneMap, eheight)
|
||||
blkP.complete <- true
|
||||
db.InsertBlock(blkP.blk)
|
||||
|
||||
if progress != 0 && eheight%int64(progress) == 0 {
|
||||
log.Infof("Processing block %v", eheight)
|
||||
}
|
||||
eheight++
|
||||
|
||||
if eheight % 2000 == 0 {
|
||||
f, err := os.Create(fmt.Sprintf("profile.%d", eheight))
|
||||
if err == nil {
|
||||
pprof.WriteHeapProfile(f)
|
||||
f.Close()
|
||||
} else {
|
||||
log.Warnf("profile failed %v", err)
|
||||
}
|
||||
}
|
||||
} else {
|
||||
break
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func processBuf(idx int, bufqueue chan *bufQueue, blkqueue chan *blkQueue) {
|
||||
complete := make (chan bool)
|
||||
for {
|
||||
select {
|
||||
case bq := <- bufqueue:
|
||||
var blkmsg blkQueue
|
||||
|
||||
blkmsg.height = bq.height
|
||||
|
||||
if len(bq.blkbuf) == 0 {
|
||||
// we are done
|
||||
blkqueue <- &blkmsg
|
||||
}
|
||||
|
||||
blk, err := btcutil.NewBlockFromBytes(bq.blkbuf)
|
||||
if err != nil {
|
||||
fmt.Printf("failed to parse block %v", bq.height)
|
||||
return
|
||||
}
|
||||
blkmsg.blk = blk
|
||||
blkmsg.complete = complete
|
||||
blkqueue <- &blkmsg
|
||||
select {
|
||||
case <- complete:
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func readBlocks(fi io.Reader, bufqueue chan *bufQueue) {
|
||||
var height int64
|
||||
for {
|
||||
var net, blen uint32
|
||||
|
||||
var bufM bufQueue
|
||||
bufM.height = height
|
||||
|
||||
// generate and write header values
|
||||
err := binary.Read(fi, binary.LittleEndian, &net)
|
||||
if err != nil {
|
||||
break
|
||||
bufqueue <- &bufM
|
||||
}
|
||||
if net != uint32(btcwire.MainNet) {
|
||||
fmt.Printf("network mismatch %v %v",
|
||||
net, uint32(btcwire.MainNet))
|
||||
|
||||
bufqueue <- &bufM
|
||||
}
|
||||
err = binary.Read(fi, binary.LittleEndian, &blen)
|
||||
if err != nil {
|
||||
bufqueue <- &bufM
|
||||
}
|
||||
blkbuf := make([]byte, blen)
|
||||
err = binary.Read(fi, binary.LittleEndian, blkbuf)
|
||||
bufM.blkbuf = blkbuf
|
||||
bufqueue <- &bufM
|
||||
height++
|
||||
}
|
||||
}
|
||||
|
||||
// newLogger creates a new seelog logger using the provided logging level and
|
||||
// log message prefix.
|
||||
func newLogger(level string, prefix string) seelog.LoggerInterface {
|
||||
fmtstring := `
|
||||
<seelog type="adaptive" mininterval="2000000" maxinterval="100000000"
|
||||
critmsgcount="500" minlevel="%s">
|
||||
<outputs formatid="all">
|
||||
<console/>
|
||||
</outputs>
|
||||
<formats>
|
||||
<format id="all" format="[%%Time %%Date] [%%LEV] [%s] %%Msg%%n" />
|
||||
</formats>
|
||||
</seelog>`
|
||||
config := fmt.Sprintf(fmtstring, level, prefix)
|
||||
|
||||
logger, err := seelog.LoggerFromConfigAsString(config)
|
||||
if err != nil {
|
||||
fmt.Fprintf(os.Stderr, "failed to create logger: %v", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
return logger
|
||||
}
|
||||
|
||||
// btcdHomeDir returns an OS appropriate home directory for btcd.
|
||||
func btcdHomeDir() string {
|
||||
// Search for Windows APPDATA first. This won't exist on POSIX OSes.
|
||||
appData := os.Getenv("APPDATA")
|
||||
if appData != "" {
|
||||
return filepath.Join(appData, "btcd")
|
||||
}
|
||||
|
||||
// Fall back to standard HOME directory that works for most POSIX OSes.
|
||||
home := os.Getenv("HOME")
|
||||
if home != "" {
|
||||
return filepath.Join(home, ".btcd")
|
||||
}
|
||||
|
||||
// In the worst case, use the current directory.
|
||||
return "."
|
||||
}
|
||||
197
util/dropafter/dropafter.go
Normal file
197
util/dropafter/dropafter.go
Normal file
|
|
@ -0,0 +1,197 @@
|
|||
/*
|
||||
t f
|
||||
* Copyright (c) 2013 Conformal Systems LLC.
|
||||
*/
|
||||
|
||||
package main
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"flag"
|
||||
"fmt"
|
||||
"github.com/conformal/btcdb"
|
||||
_ "github.com/conformal/btcdb/ldb"
|
||||
"github.com/conformal/btcwire"
|
||||
"github.com/conformal/seelog"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
type ShaHash btcwire.ShaHash
|
||||
|
||||
var log seelog.LoggerInterface
|
||||
|
||||
const (
|
||||
ArgSha = iota
|
||||
ArgHeight
|
||||
)
|
||||
|
||||
func main() {
|
||||
var err error
|
||||
var dbType string
|
||||
var datadir string
|
||||
var shastring string
|
||||
flag.StringVar(&dbType, "dbtype", "", "Database backend to use for the Block Chain")
|
||||
flag.StringVar(&datadir, "datadir", "", "Directory to store data")
|
||||
flag.StringVar(&shastring, "s", "", "Block sha to process")
|
||||
|
||||
flag.Parse()
|
||||
|
||||
log, err = seelog.LoggerFromWriterWithMinLevel(os.Stdout,
|
||||
seelog.TraceLvl)
|
||||
if err != nil {
|
||||
fmt.Fprintf(os.Stderr, "failed to create logger: %v", err)
|
||||
return
|
||||
}
|
||||
defer log.Flush()
|
||||
btcdb.UseLogger(log)
|
||||
|
||||
if len(dbType) == 0 {
|
||||
dbType = "sqlite"
|
||||
}
|
||||
|
||||
if len(datadir) == 0 {
|
||||
datadir = filepath.Join(btcdHomeDir(), "data")
|
||||
}
|
||||
datadir = filepath.Join(datadir, "mainnet")
|
||||
|
||||
blockDbNamePrefix := "blocks"
|
||||
dbName := blockDbNamePrefix + "_" + dbType
|
||||
if dbType == "sqlite" {
|
||||
dbName = dbName + ".db"
|
||||
}
|
||||
dbPath := filepath.Join(datadir, dbName)
|
||||
|
||||
log.Infof("loading db")
|
||||
db, err := btcdb.OpenDB(dbType, dbPath)
|
||||
if err != nil {
|
||||
log.Warnf("db open failed: %v", err)
|
||||
return
|
||||
}
|
||||
defer db.Close()
|
||||
log.Infof("db load complete")
|
||||
|
||||
_, height, err := db.NewestSha()
|
||||
log.Infof("loaded block height %v", height)
|
||||
|
||||
sha, err := getSha(db, shastring)
|
||||
if err != nil {
|
||||
log.Infof("Invalid block %v", shastring)
|
||||
return
|
||||
}
|
||||
|
||||
err = db.DropAfterBlockBySha(&sha)
|
||||
if err != nil {
|
||||
log.Warnf("failed %v", err)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func getSha(db btcdb.Db, str string) (btcwire.ShaHash, error) {
|
||||
argtype, idx, sha, err := parsesha(str)
|
||||
if err != nil {
|
||||
log.Warnf("unable to decode [%v] %v", str, err)
|
||||
return btcwire.ShaHash{}, err
|
||||
}
|
||||
|
||||
switch argtype {
|
||||
case ArgSha:
|
||||
// nothing to do
|
||||
case ArgHeight:
|
||||
sha, err = db.FetchBlockShaByHeight(idx)
|
||||
if err != nil {
|
||||
return btcwire.ShaHash{}, err
|
||||
}
|
||||
}
|
||||
if sha == nil {
|
||||
fmt.Printf("wtf sha is nil but err is %v", err)
|
||||
}
|
||||
return *sha, nil
|
||||
}
|
||||
|
||||
|
||||
var ntxcnt int64
|
||||
var txspendcnt int64
|
||||
var txgivecnt int64
|
||||
|
||||
var ErrBadShaPrefix = errors.New("invalid prefix")
|
||||
var ErrBadShaLen = errors.New("invalid len")
|
||||
var ErrBadShaChar = errors.New("invalid character")
|
||||
|
||||
func parsesha(argstr string) (argtype int, height int64, psha *btcwire.ShaHash, err error) {
|
||||
var sha btcwire.ShaHash
|
||||
|
||||
var hashbuf string
|
||||
|
||||
switch len(argstr) {
|
||||
case 64:
|
||||
hashbuf = argstr
|
||||
case 66:
|
||||
if argstr[0:2] != "0x" {
|
||||
log.Infof("prefix is %v", argstr[0:2])
|
||||
err = ErrBadShaPrefix
|
||||
return
|
||||
}
|
||||
hashbuf = argstr[2:]
|
||||
default:
|
||||
if len(argstr) <= 16 {
|
||||
// assume value is height
|
||||
argtype = ArgHeight
|
||||
var h int
|
||||
h, err = strconv.Atoi(argstr)
|
||||
if err == nil {
|
||||
height = int64(h)
|
||||
return
|
||||
}
|
||||
log.Infof("Unable to parse height %v, err %v", height, err)
|
||||
}
|
||||
err = ErrBadShaLen
|
||||
return
|
||||
}
|
||||
|
||||
var buf [32]byte
|
||||
for idx, ch := range hashbuf {
|
||||
var val rune
|
||||
|
||||
switch {
|
||||
case ch >= '0' && ch <= '9':
|
||||
val = ch - '0'
|
||||
case ch >= 'a' && ch <= 'f':
|
||||
val = ch - 'a' + rune(10)
|
||||
case ch >= 'A' && ch <= 'F':
|
||||
val = ch - 'A' + rune(10)
|
||||
default:
|
||||
err = ErrBadShaChar
|
||||
return
|
||||
}
|
||||
b := buf[31-idx/2]
|
||||
if idx&1 == 1 {
|
||||
b |= byte(val)
|
||||
} else {
|
||||
b |= (byte(val) << 4)
|
||||
}
|
||||
buf[31-idx/2] = b
|
||||
}
|
||||
sha.SetBytes(buf[0:32])
|
||||
psha = &sha
|
||||
return
|
||||
}
|
||||
|
||||
// btcdHomeDir returns an OS appropriate home directory for btcd.
|
||||
func btcdHomeDir() string {
|
||||
// Search for Windows APPDATA first. This won't exist on POSIX OSes.
|
||||
appData := os.Getenv("APPDATA")
|
||||
if appData != "" {
|
||||
return filepath.Join(appData, "btcd")
|
||||
}
|
||||
|
||||
// Fall back to standard HOME directory that works for most POSIX OSes.
|
||||
home := os.Getenv("HOME")
|
||||
if home != "" {
|
||||
return filepath.Join(home, ".btcd")
|
||||
}
|
||||
|
||||
// In the worst case, use the current directory.
|
||||
return "."
|
||||
}
|
||||
283
util/showblock/showblock.go
Normal file
283
util/showblock/showblock.go
Normal file
|
|
@ -0,0 +1,283 @@
|
|||
/*
|
||||
* Copyright (c) 2013 Conformal Systems LLC.
|
||||
*/
|
||||
|
||||
package main
|
||||
|
||||
import (
|
||||
"encoding/binary"
|
||||
"errors"
|
||||
"flag"
|
||||
"fmt"
|
||||
"github.com/conformal/btcdb"
|
||||
_ "github.com/conformal/btcdb/sqlite3"
|
||||
_ "github.com/conformal/btcdb/ldb"
|
||||
"github.com/conformal/btcwire"
|
||||
"github.com/conformal/seelog"
|
||||
"github.com/davecgh/go-spew/spew"
|
||||
"io"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
type ShaHash btcwire.ShaHash
|
||||
|
||||
var log seelog.LoggerInterface
|
||||
|
||||
const (
|
||||
ArgSha = iota
|
||||
ArgHeight
|
||||
)
|
||||
|
||||
func main() {
|
||||
var err error
|
||||
var dbType string
|
||||
var datadir string
|
||||
var shastring, eshastring, outfile string
|
||||
var rflag, fflag, tflag bool
|
||||
var progress int
|
||||
end := int64(-1)
|
||||
flag.StringVar(&dbType, "dbtype", "", "Database backend to use for the Block Chain")
|
||||
flag.StringVar(&datadir, "datadir", ".", "Directory to store data")
|
||||
|
||||
flag.StringVar(&shastring, "s", "", "Block sha to process")
|
||||
flag.StringVar(&eshastring, "e", "", "Block sha to process")
|
||||
flag.StringVar(&outfile, "o", "", "outfile")
|
||||
flag.BoolVar(&rflag, "r", false, "raw block")
|
||||
flag.BoolVar(&fflag, "f", false, "fmt block")
|
||||
flag.BoolVar(&tflag, "t", false, "show transactions")
|
||||
flag.IntVar(&progress, "p", 0, "show progress")
|
||||
|
||||
flag.Parse()
|
||||
|
||||
log, err = seelog.LoggerFromWriterWithMinLevel(os.Stdout,
|
||||
seelog.InfoLvl)
|
||||
if err != nil {
|
||||
fmt.Fprintf(os.Stderr, "failed to create logger: %v", err)
|
||||
return
|
||||
}
|
||||
defer log.Flush()
|
||||
btcdb.UseLogger(log)
|
||||
|
||||
if len(dbType) == 0 {
|
||||
dbType = "sqlite"
|
||||
}
|
||||
|
||||
if len(datadir) == 0 {
|
||||
datadir = filepath.Join(btcdHomeDir(), "data")
|
||||
}
|
||||
datadir = filepath.Join(datadir, "mainnet")
|
||||
|
||||
blockDbNamePrefix := "blocks"
|
||||
dbName := blockDbNamePrefix + "_" + dbType
|
||||
if dbType == "sqlite" {
|
||||
dbName = dbName + ".db"
|
||||
}
|
||||
dbPath := filepath.Join(datadir, dbName)
|
||||
|
||||
|
||||
log.Infof("loading db %v", dbType)
|
||||
db, err := btcdb.OpenDB(dbType, dbPath)
|
||||
if err != nil {
|
||||
log.Warnf("db open failed: %v", err)
|
||||
return
|
||||
}
|
||||
defer db.Close()
|
||||
log.Infof("db load complete")
|
||||
|
||||
height, err := getHeight(db, shastring)
|
||||
if err != nil {
|
||||
log.Infof("Invalid block %v", shastring)
|
||||
return
|
||||
}
|
||||
if eshastring != "" {
|
||||
end, err = getHeight(db, eshastring)
|
||||
if err != nil {
|
||||
log.Infof("Invalid end block %v", eshastring)
|
||||
return
|
||||
}
|
||||
} else {
|
||||
end = height + 1
|
||||
}
|
||||
|
||||
log.Infof("height %v end %v", height, end)
|
||||
|
||||
var fo io.WriteCloser
|
||||
if outfile != "" {
|
||||
fo, err = os.Create(outfile)
|
||||
if err != nil {
|
||||
log.Warnf("failed to open file %v, err %v", outfile, err)
|
||||
}
|
||||
defer func() {
|
||||
if err := fo.Close(); err != nil {
|
||||
log.Warn("failed to close file %v %v", outfile, err)
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
for ; height < end; height++ {
|
||||
if progress != 0 && height%int64(progress) == 0 {
|
||||
log.Infof("Processing block %v", height)
|
||||
}
|
||||
err = DumpBlock(db, height, fo, rflag, fflag, tflag)
|
||||
if err != nil {
|
||||
break
|
||||
}
|
||||
}
|
||||
if progress != 0 {
|
||||
height--
|
||||
log.Infof("Processing block %v", height)
|
||||
}
|
||||
}
|
||||
|
||||
func getHeight(db btcdb.Db, str string) (int64, error) {
|
||||
argtype, idx, sha, err := parsesha(str)
|
||||
if err != nil {
|
||||
log.Warnf("unable to decode [%v] %v", str, err)
|
||||
return 0, err
|
||||
}
|
||||
|
||||
switch argtype {
|
||||
case ArgSha:
|
||||
// nothing to do
|
||||
blk, err := db.FetchBlockBySha(sha)
|
||||
if err != nil {
|
||||
log.Warnf("unable to locate block sha %v err %v",
|
||||
sha, err)
|
||||
return 0, err
|
||||
}
|
||||
idx = blk.Height()
|
||||
case ArgHeight:
|
||||
}
|
||||
return idx, nil
|
||||
}
|
||||
|
||||
func DumpBlock(db btcdb.Db, height int64, fo io.Writer, rflag bool, fflag bool, tflag bool) error {
|
||||
sha, err := db.FetchBlockShaByHeight(height)
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
blk, err := db.FetchBlockBySha(sha)
|
||||
if err != nil {
|
||||
log.Warnf("Failed to fetch block %v, err %v", sha, err)
|
||||
return err
|
||||
}
|
||||
rblk, err := blk.Bytes()
|
||||
blkid := blk.Height()
|
||||
|
||||
if rflag {
|
||||
log.Infof("Block %v depth %v %v", sha, blkid, spew.Sdump(rblk))
|
||||
}
|
||||
|
||||
mblk := blk.MsgBlock()
|
||||
if fflag {
|
||||
log.Infof("Block %v depth %v %v", sha, blkid, spew.Sdump(mblk))
|
||||
}
|
||||
if tflag {
|
||||
log.Infof("Num transactions %v", len(mblk.Transactions))
|
||||
for i, tx := range mblk.Transactions {
|
||||
txsha, err := tx.TxSha()
|
||||
if err != nil {
|
||||
continue
|
||||
}
|
||||
log.Infof("tx %v: %v", i, &txsha)
|
||||
|
||||
}
|
||||
}
|
||||
if fo != nil {
|
||||
// generate and write header values
|
||||
binary.Write(fo, binary.LittleEndian, uint32(btcwire.MainNet))
|
||||
binary.Write(fo, binary.LittleEndian, uint32(len(rblk)))
|
||||
|
||||
// write block
|
||||
fo.Write(rblk)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
var ntxcnt int64
|
||||
var txspendcnt int64
|
||||
var txgivecnt int64
|
||||
|
||||
var ErrBadShaPrefix = errors.New("invalid prefix")
|
||||
var ErrBadShaLen = errors.New("invalid len")
|
||||
var ErrBadShaChar = errors.New("invalid character")
|
||||
|
||||
func parsesha(argstr string) (argtype int, height int64, psha *btcwire.ShaHash, err error) {
|
||||
var sha btcwire.ShaHash
|
||||
|
||||
var hashbuf string
|
||||
|
||||
switch len(argstr) {
|
||||
case 64:
|
||||
hashbuf = argstr
|
||||
case 66:
|
||||
if argstr[0:2] != "0x" {
|
||||
log.Infof("prefix is %v", argstr[0:2])
|
||||
err = ErrBadShaPrefix
|
||||
return
|
||||
}
|
||||
hashbuf = argstr[2:]
|
||||
default:
|
||||
if len(argstr) <= 16 {
|
||||
// assume value is height
|
||||
argtype = ArgHeight
|
||||
var h int
|
||||
h, err = strconv.Atoi(argstr)
|
||||
if err == nil {
|
||||
height = int64(h)
|
||||
return
|
||||
}
|
||||
log.Infof("Unable to parse height %v, err %v", height, err)
|
||||
}
|
||||
err = ErrBadShaLen
|
||||
return
|
||||
}
|
||||
|
||||
var buf [32]byte
|
||||
for idx, ch := range hashbuf {
|
||||
var val rune
|
||||
|
||||
switch {
|
||||
case ch >= '0' && ch <= '9':
|
||||
val = ch - '0'
|
||||
case ch >= 'a' && ch <= 'f':
|
||||
val = ch - 'a' + rune(10)
|
||||
case ch >= 'A' && ch <= 'F':
|
||||
val = ch - 'A' + rune(10)
|
||||
default:
|
||||
err = ErrBadShaChar
|
||||
return
|
||||
}
|
||||
b := buf[31-idx/2]
|
||||
if idx&1 == 1 {
|
||||
b |= byte(val)
|
||||
} else {
|
||||
b |= (byte(val) << 4)
|
||||
}
|
||||
buf[31-idx/2] = b
|
||||
}
|
||||
sha.SetBytes(buf[0:32])
|
||||
psha = &sha
|
||||
return
|
||||
}
|
||||
|
||||
// btcdHomeDir returns an OS appropriate home directory for btcd.
|
||||
func btcdHomeDir() string {
|
||||
// Search for Windows APPDATA first. This won't exist on POSIX OSes.
|
||||
appData := os.Getenv("APPDATA")
|
||||
if appData != "" {
|
||||
return filepath.Join(appData, "btcd")
|
||||
}
|
||||
|
||||
// Fall back to standard HOME directory that works for most POSIX OSes.
|
||||
home := os.Getenv("HOME")
|
||||
if home != "" {
|
||||
return filepath.Join(home, ".btcd")
|
||||
}
|
||||
|
||||
// In the worst case, use the current directory.
|
||||
return "."
|
||||
}
|
||||
Loading…
Reference in a new issue