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.
87 lines
2.3 KiB
Go
87 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"github.com/guggero/chantools/btc"
|
|
"github.com/lightningnetwork/lnd/chanbackup"
|
|
"github.com/lightningnetwork/lnd/keychain"
|
|
"path"
|
|
|
|
"github.com/btcsuite/btcutil/hdkeychain"
|
|
"github.com/lightningnetwork/lnd/channeldb"
|
|
)
|
|
|
|
type chanBackupCommand struct {
|
|
RootKey string `long:"rootkey" description:"BIP32 HD root key of the wallet that should be used to create the backup. Leave empty to prompt for lnd 24 word aezeed."`
|
|
ChannelDB string `long:"channeldb" description:"The lnd channel.db file to create the backup from."`
|
|
MultiFile string `long:"multi_file" description:"The lnd channel.backup file to create."`
|
|
}
|
|
|
|
func (c *chanBackupCommand) Execute(_ []string) error {
|
|
setupChainParams(cfg)
|
|
|
|
var (
|
|
extendedKey *hdkeychain.ExtendedKey
|
|
err error
|
|
)
|
|
|
|
// Check that root key is valid or fall back to console input.
|
|
switch {
|
|
case c.RootKey != "":
|
|
extendedKey, err = hdkeychain.NewKeyFromString(c.RootKey)
|
|
|
|
default:
|
|
extendedKey, _, err = rootKeyFromConsole()
|
|
}
|
|
if err != nil {
|
|
return fmt.Errorf("error reading root key: %v", err)
|
|
}
|
|
|
|
// Check that we have a backup file.
|
|
if c.MultiFile == "" {
|
|
return fmt.Errorf("backup file is required")
|
|
}
|
|
|
|
// Check that we have a channel DB.
|
|
if c.ChannelDB == "" {
|
|
return fmt.Errorf("channel DB is required")
|
|
}
|
|
db, err := channeldb.Open(
|
|
path.Dir(c.ChannelDB), channeldb.OptionSetSyncFreelist(true),
|
|
channeldb.OptionReadOnly(true),
|
|
)
|
|
if err != nil {
|
|
return fmt.Errorf("error opening rescue DB: %v", err)
|
|
}
|
|
multiFile := chanbackup.NewMultiFile(c.MultiFile)
|
|
keyRing := &btc.HDKeyRing{
|
|
ExtendedKey: extendedKey,
|
|
ChainParams: chainParams,
|
|
}
|
|
return createChannelBackup(db, multiFile, keyRing)
|
|
}
|
|
|
|
func createChannelBackup(db *channeldb.DB, multiFile *chanbackup.MultiFile,
|
|
ring keychain.KeyRing) error {
|
|
|
|
singles, err := chanbackup.FetchStaticChanBackups(db)
|
|
if err != nil {
|
|
return fmt.Errorf("error extracting channel backup: %v", err)
|
|
}
|
|
multi := &chanbackup.Multi{
|
|
Version: chanbackup.DefaultMultiVersion,
|
|
StaticBackups: singles,
|
|
}
|
|
var b bytes.Buffer
|
|
err = multi.PackToWriter(&b, ring)
|
|
if err != nil {
|
|
return fmt.Errorf("unable to pack backup: %v", err)
|
|
}
|
|
err = multiFile.UpdateAndSwap(b.Bytes())
|
|
if err != nil {
|
|
return fmt.Errorf("unable to write backup file: %v", err)
|
|
}
|
|
return nil
|
|
}
|