muun-recovery/recovery_tool.go

145 lines
2.9 KiB
Go

package main
import (
"bytes"
"encoding/hex"
"github.com/btcsuite/btcd/chaincfg/chainhash"
"github.com/btcsuite/btcd/wire"
"github.com/btcsuite/btcutil"
"github.com/muun/libwallet"
"github.com/muun/libwallet/btcsuitew/txscriptw"
"github.com/muun/recovery/scanner"
)
func buildSweepTx(utxos []*scanner.Utxo, sweepAddress btcutil.Address, fee int64) ([]byte, error) {
tx := wire.NewMsgTx(2)
value := int64(0)
for _, utxo := range utxos {
chainHash, err := chainhash.NewHashFromStr(utxo.TxID)
if err != nil {
return nil, err
}
outpoint := wire.OutPoint{
Hash: *chainHash,
Index: uint32(utxo.OutputIndex),
}
tx.AddTxIn(wire.NewTxIn(&outpoint, []byte{}, [][]byte{}))
value += utxo.Amount
}
value -= fee
script, err := txscriptw.PayToAddrScript(sweepAddress)
if err != nil {
return nil, err
}
tx.AddTxOut(wire.NewTxOut(value, script))
writer := &bytes.Buffer{}
err = tx.Serialize(writer)
if err != nil {
return nil, err
}
if fee != 0 {
readConfirmation(value, fee, sweepAddress.String())
}
return writer.Bytes(), nil
}
func buildSignedTx(utxos []*scanner.Utxo, sweepTx []byte, userKey *libwallet.HDPrivateKey,
muunKey *libwallet.HDPrivateKey) (*wire.MsgTx, error) {
inputList := &libwallet.InputList{}
for _, utxo := range utxos {
inputList.Add(&input{
utxo,
[]byte{},
})
}
nonces := libwallet.GenerateMusigNonces(len(utxos))
pstx, err := libwallet.NewPartiallySignedTransaction(inputList, sweepTx, nonces)
if err != nil {
return nil, err
}
signedTx, err := pstx.FullySign(userKey, muunKey)
if err != nil {
return nil, err
}
wireTx := wire.NewMsgTx(0)
wireTx.BtcDecode(bytes.NewReader(signedTx.Bytes), 0, wire.WitnessEncoding)
return wireTx, nil
}
// input is a minimal type that implements libwallet.Input
type input struct {
utxo *scanner.Utxo
muunSignature []byte
}
func (i *input) OutPoint() libwallet.Outpoint {
return &outpoint{utxo: i.utxo}
}
func (i *input) Address() libwallet.MuunAddress {
return i.utxo.Address
}
func (i *input) UserSignature() []byte {
return []byte{}
}
func (i *input) MuunSignature() []byte {
return i.muunSignature
}
func (i *input) SubmarineSwapV1() libwallet.InputSubmarineSwapV1 {
return nil
}
func (i *input) SubmarineSwapV2() libwallet.InputSubmarineSwapV2 {
return nil
}
func (i *input) IncomingSwap() libwallet.InputIncomingSwap {
return nil
}
func (i *input) MuunPublicNonce() []byte {
// Will always be nil in the context of the tool
// Look at coinV5.signFirstWith for the reason why.
return nil
}
// outpoint is a minimal type that implements libwallet.Outpoint
type outpoint struct {
utxo *scanner.Utxo
}
func (o *outpoint) TxId() []byte {
raw, err := hex.DecodeString(o.utxo.TxID)
if err != nil {
panic(err) // we wrote this hex value ourselves, no input from anywhere else
}
return raw
}
func (o *outpoint) Index() int {
return o.utxo.OutputIndex
}
func (o *outpoint) Amount() int64 {
return o.utxo.Amount
}