mirror of https://github.com/guggero/chantools
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
220 lines
6.1 KiB
Go
220 lines
6.1 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"github.com/btcsuite/btcutil/hdkeychain"
|
|
"github.com/lightningnetwork/lnd/aezeed"
|
|
"golang.org/x/crypto/ssh/terminal"
|
|
"io/ioutil"
|
|
"os"
|
|
"strings"
|
|
"syscall"
|
|
|
|
"github.com/btcsuite/btcd/chaincfg"
|
|
"github.com/guggero/chantools/dataformat"
|
|
"github.com/jessevdk/go-flags"
|
|
"github.com/lightningnetwork/lnd/build"
|
|
"github.com/lightningnetwork/lnd/channeldb"
|
|
)
|
|
|
|
const (
|
|
defaultApiUrl = "https://blockstream.info/api"
|
|
)
|
|
|
|
type config struct {
|
|
Testnet bool `long:"testnet" description:"Set to true if testnet parameters should be used."`
|
|
ApiUrl string `long:"apiurl" description:"API URL to use (must be esplora compatible)."`
|
|
ListChannels string `long:"listchannels" description:"The channel input is in the format of lncli's listchannels format. Specify '-' to read from stdin."`
|
|
PendingChannels string `long:"pendingchannels" description:"The channel input is in the format of lncli's pendingchannels format. Specify '-' to read from stdin."`
|
|
FromSummary string `long:"fromsummary" description:"The channel input is in the format of this tool's channel summary. Specify '-' to read from stdin."`
|
|
FromChannelDB string `long:"fromchanneldb" description:"The channel input is in the format of an lnd channel.db file."`
|
|
}
|
|
|
|
var (
|
|
logWriter = build.NewRotatingLogWriter()
|
|
log = build.NewSubLogger("CHAN", logWriter.GenSubLogger)
|
|
cfg = &config{
|
|
ApiUrl: defaultApiUrl,
|
|
}
|
|
chainParams = &chaincfg.MainNetParams
|
|
)
|
|
|
|
func main() {
|
|
err := runCommandParser()
|
|
if err == nil {
|
|
return
|
|
}
|
|
|
|
_, ok := err.(*flags.Error)
|
|
if !ok {
|
|
fmt.Printf("Error running chantools: %v\n", err)
|
|
}
|
|
os.Exit(0)
|
|
}
|
|
|
|
func runCommandParser() error {
|
|
setupLogging()
|
|
|
|
// Parse command line.
|
|
parser := flags.NewParser(cfg, flags.Default)
|
|
_, _ = parser.AddCommand(
|
|
"summary", "Compile a summary about the current state of "+
|
|
"channels.", "", &summaryCommand{},
|
|
)
|
|
_, _ = parser.AddCommand(
|
|
"rescueclosed", "Try finding the private keys for funds that "+
|
|
"are in outputs of remotely force-closed channels.", "",
|
|
&rescueClosedCommand{},
|
|
)
|
|
_, _ = parser.AddCommand(
|
|
"forceclose", "Force-close the last state that is in the "+
|
|
"channel.db provided.", "", &forceCloseCommand{},
|
|
)
|
|
_, _ = parser.AddCommand(
|
|
"sweeptimelock", "Sweep the force-closed state after the time "+
|
|
"lock has expired.", "", &sweepTimeLockCommand{},
|
|
)
|
|
_, _ = parser.AddCommand(
|
|
"dumpchannels", "Dump all channel information from lnd's "+
|
|
"channel database.", "", &dumpChannelsCommand{},
|
|
)
|
|
_, _ = parser.AddCommand(
|
|
"showrootkey", "Extract and show the BIP32 HD root key from "+
|
|
"the 24 word lnd aezeed.", "", &showRootKeyCommand{},
|
|
)
|
|
_, _ = parser.AddCommand(
|
|
"dumpbackup", "Dump the content of a channel.backup file.", "",
|
|
&dumpBackupCommand{},
|
|
)
|
|
_, _ = parser.AddCommand(
|
|
"derivekey", "Derive a key with a specific derivation path "+
|
|
"from the BIP32 HD root key.", "", &deriveKeyCommand{},
|
|
)
|
|
|
|
_, err := parser.Parse()
|
|
return err
|
|
}
|
|
|
|
func parseInputType(cfg *config) ([]*dataformat.SummaryEntry, error) {
|
|
var (
|
|
content []byte
|
|
err error
|
|
target dataformat.InputFile
|
|
)
|
|
|
|
switch {
|
|
case cfg.ListChannels != "":
|
|
content, err = readInput(cfg.ListChannels)
|
|
target = &dataformat.ListChannelsFile{}
|
|
|
|
case cfg.PendingChannels != "":
|
|
content, err = readInput(cfg.PendingChannels)
|
|
target = &dataformat.PendingChannelsFile{}
|
|
|
|
case cfg.FromSummary != "":
|
|
content, err = readInput(cfg.FromSummary)
|
|
target = &dataformat.SummaryEntryFile{}
|
|
|
|
case cfg.FromChannelDB != "":
|
|
db, err := channeldb.Open(cfg.FromChannelDB)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error opening channel DB: %v",
|
|
err)
|
|
}
|
|
target = &dataformat.ChannelDBFile{DB: db}
|
|
return target.AsSummaryEntries()
|
|
|
|
default:
|
|
return nil, fmt.Errorf("an input file must be specified")
|
|
}
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
decoder := json.NewDecoder(bytes.NewReader(content))
|
|
err = decoder.Decode(&target)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return target.AsSummaryEntries()
|
|
}
|
|
|
|
func readInput(input string) ([]byte, error) {
|
|
if strings.TrimSpace(input) == "-" {
|
|
return ioutil.ReadAll(os.Stdin)
|
|
}
|
|
return ioutil.ReadFile(input)
|
|
}
|
|
|
|
func rootKeyFromConsole() (*hdkeychain.ExtendedKey, error) {
|
|
// We'll now prompt the user to enter in their 24-word mnemonic.
|
|
fmt.Printf("Input your 24-word mnemonic separated by spaces: ")
|
|
reader := bufio.NewReader(os.Stdin)
|
|
mnemonicStr, err := reader.ReadString('\n')
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// We'll trim off extra spaces, and ensure the mnemonic is all
|
|
// lower case, then populate our request.
|
|
mnemonicStr = strings.TrimSpace(mnemonicStr)
|
|
mnemonicStr = strings.ToLower(mnemonicStr)
|
|
|
|
cipherSeedMnemonic := strings.Split(mnemonicStr, " ")
|
|
|
|
fmt.Println()
|
|
|
|
if len(cipherSeedMnemonic) != 24 {
|
|
return nil, fmt.Errorf("wrong cipher seed mnemonic "+
|
|
"length: got %v words, expecting %v words",
|
|
len(cipherSeedMnemonic), 24)
|
|
}
|
|
|
|
// Additionally, the user may have a passphrase, that will also
|
|
// need to be provided so the daemon can properly decipher the
|
|
// cipher seed.
|
|
fmt.Printf("Input your cipher seed passphrase (press enter if " +
|
|
"your seed doesn't have a passphrase): ")
|
|
passphrase, err := terminal.ReadPassword(syscall.Stdin)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var mnemonic aezeed.Mnemonic
|
|
copy(mnemonic[:], cipherSeedMnemonic[:])
|
|
|
|
// If we're unable to map it back into the ciphertext, then either the
|
|
// mnemonic is wrong, or the passphrase is wrong.
|
|
cipherSeed, err := mnemonic.ToCipherSeed(passphrase)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to decrypt seed with passphrase"+
|
|
": %v", err)
|
|
}
|
|
rootKey, err := hdkeychain.NewMaster(cipherSeed.Entropy[:], chainParams)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to derive master extended key")
|
|
}
|
|
return rootKey, nil
|
|
}
|
|
|
|
func setupChainParams(cfg *config) {
|
|
if cfg.Testnet {
|
|
chainParams = &chaincfg.TestNet3Params
|
|
}
|
|
}
|
|
|
|
func setupLogging() {
|
|
logWriter.RegisterSubLogger("CHAN", log)
|
|
err := logWriter.InitLogRotator("./results/chantools.log", 10, 3)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
err = build.ParseAndSetDebugLevels("trace", logWriter)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|