mirror of
https://github.com/guggero/chantools
synced 2024-11-11 01:10:42 +00:00
113 lines
3.0 KiB
Go
113 lines
3.0 KiB
Go
package lnd
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/btcsuite/btcd/btcec"
|
|
"github.com/btcsuite/btcd/chaincfg/chainhash"
|
|
"github.com/btcsuite/btcd/txscript"
|
|
"github.com/btcsuite/btcd/wire"
|
|
"github.com/lightningnetwork/lnd/channeldb"
|
|
"github.com/lightningnetwork/lnd/input"
|
|
)
|
|
|
|
type LightningChannel struct {
|
|
LocalChanCfg channeldb.ChannelConfig
|
|
RemoteChanCfg channeldb.ChannelConfig
|
|
SignDesc *input.SignDescriptor
|
|
ChannelState *channeldb.OpenChannel
|
|
TXSigner *Signer
|
|
}
|
|
|
|
// CreateSignDesc derives the SignDescriptor for commitment transactions from
|
|
// other fields on the LightningChannel.
|
|
func (lc *LightningChannel) CreateSignDesc() error {
|
|
localKey := lc.LocalChanCfg.MultiSigKey.PubKey.SerializeCompressed()
|
|
remoteKey := lc.RemoteChanCfg.MultiSigKey.PubKey.SerializeCompressed()
|
|
|
|
multiSigScript, err := input.GenMultiSigScript(localKey, remoteKey)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fundingPkScript, err := input.WitnessScriptHash(multiSigScript)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
lc.SignDesc = &input.SignDescriptor{
|
|
KeyDesc: lc.LocalChanCfg.MultiSigKey,
|
|
WitnessScript: multiSigScript,
|
|
Output: &wire.TxOut{
|
|
PkScript: fundingPkScript,
|
|
Value: int64(lc.ChannelState.Capacity),
|
|
},
|
|
HashType: txscript.SigHashAll,
|
|
InputIndex: 0,
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// SignedCommitTx function take the latest commitment transaction and populate
|
|
// it with witness data.
|
|
func (lc *LightningChannel) SignedCommitTx() (*wire.MsgTx, error) {
|
|
// Fetch the current commitment transaction, along with their signature
|
|
// for the transaction.
|
|
localCommit := lc.ChannelState.LocalCommitment
|
|
commitTx := localCommit.CommitTx.Copy()
|
|
theirSig, err := btcec.ParseDERSignature(
|
|
localCommit.CommitSig, btcec.S256(),
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// With this, we then generate the full witness so the caller can
|
|
// broadcast a fully signed transaction.
|
|
lc.SignDesc.SigHashes = txscript.NewTxSigHashes(commitTx)
|
|
ourSig, err := lc.TXSigner.SignOutputRaw(commitTx, lc.SignDesc)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// With the final signature generated, create the witness stack
|
|
// required to spend from the multi-sig output.
|
|
ourKey := lc.LocalChanCfg.MultiSigKey.PubKey.SerializeCompressed()
|
|
theirKey := lc.RemoteChanCfg.MultiSigKey.PubKey.SerializeCompressed()
|
|
|
|
commitTx.TxIn[0].Witness = input.SpendMultiSig(
|
|
lc.SignDesc.WitnessScript, ourKey,
|
|
ourSig, theirKey, theirSig,
|
|
)
|
|
|
|
return commitTx, nil
|
|
}
|
|
|
|
// ParseOutpoint parses a transaction outpoint in the format <txid>:<idx> into
|
|
// the wire format.
|
|
func ParseOutpoint(s string) (*wire.OutPoint, error) {
|
|
split := strings.Split(s, ":")
|
|
if len(split) != 2 {
|
|
return nil, fmt.Errorf("expecting channel point to be in " +
|
|
"format of: txid:index")
|
|
}
|
|
|
|
index, err := strconv.ParseInt(split[1], 10, 32)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("unable to decode output index: %v",
|
|
err)
|
|
}
|
|
|
|
txid, err := chainhash.NewHashFromStr(split[0])
|
|
if err != nil {
|
|
return nil, fmt.Errorf("unable to parse hex string: %v", err)
|
|
}
|
|
|
|
return &wire.OutPoint{
|
|
Hash: *txid,
|
|
Index: uint32(index),
|
|
}, nil
|
|
}
|