90 lines
2.1 KiB
Go
90 lines
2.1 KiB
Go
|
// +build walletrpc
|
||
|
|
|
||
|
|
package walletrpc
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"math"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/btcsuite/btcd/wire"
|
||
|
|
"github.com/btcsuite/btcutil/psbt"
|
||
|
|
"github.com/btcsuite/btcwallet/wtxmgr"
|
||
|
|
"github.com/lightningnetwork/lnd/lnwallet"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
defaultMinConf = 1
|
||
|
|
defaultMaxConf = math.MaxInt32
|
||
|
|
)
|
||
|
|
|
||
|
|
// utxoLock is a type that contains an outpoint of an UTXO and its lock lease
|
||
|
|
// information.
|
||
|
|
type utxoLock struct {
|
||
|
|
lockID wtxmgr.LockID
|
||
|
|
outpoint wire.OutPoint
|
||
|
|
expiration time.Time
|
||
|
|
}
|
||
|
|
|
||
|
|
// verifyInputsUnspent checks that all inputs are contained in the list of
|
||
|
|
// known, non-locked UTXOs given.
|
||
|
|
func verifyInputsUnspent(inputs []*wire.TxIn, utxos []*lnwallet.Utxo) error {
|
||
|
|
// TODO(guggero): Pass in UTXOs as a map to make lookup more efficient.
|
||
|
|
for idx, txIn := range inputs {
|
||
|
|
found := false
|
||
|
|
for _, u := range utxos {
|
||
|
|
if u.OutPoint == txIn.PreviousOutPoint {
|
||
|
|
found = true
|
||
|
|
break
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if !found {
|
||
|
|
return fmt.Errorf("input %d not found in list of non-"+
|
||
|
|
"locked UTXO", idx)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// lockInputs requests a lock lease for all inputs specified in a PSBT packet
|
||
|
|
// by using the internal, static lock ID of lnd's wallet.
|
||
|
|
func lockInputs(w lnwallet.WalletController, packet *psbt.Packet) ([]*utxoLock,
|
||
|
|
error) {
|
||
|
|
|
||
|
|
locks := make([]*utxoLock, len(packet.UnsignedTx.TxIn))
|
||
|
|
for idx, rawInput := range packet.UnsignedTx.TxIn {
|
||
|
|
lock := &utxoLock{
|
||
|
|
lockID: LndInternalLockID,
|
||
|
|
outpoint: rawInput.PreviousOutPoint,
|
||
|
|
}
|
||
|
|
|
||
|
|
expiration, err := w.LeaseOutput(lock.lockID, lock.outpoint)
|
||
|
|
if err != nil {
|
||
|
|
// If we run into a problem with locking one output, we
|
||
|
|
// should try to unlock those that we successfully
|
||
|
|
// locked so far. If that fails as well, there's not
|
||
|
|
// much we can do.
|
||
|
|
for i := 0; i < idx; i++ {
|
||
|
|
op := locks[i].outpoint
|
||
|
|
if err := w.ReleaseOutput(
|
||
|
|
LndInternalLockID, op,
|
||
|
|
); err != nil {
|
||
|
|
|
||
|
|
log.Errorf("could not release the "+
|
||
|
|
"lock on %v: %v", op, err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil, fmt.Errorf("could not lease a lock on "+
|
||
|
|
"UTXO: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
lock.expiration = expiration
|
||
|
|
locks[idx] = lock
|
||
|
|
}
|
||
|
|
|
||
|
|
return locks, nil
|
||
|
|
}
|