lnwallet+funding: include node net address in reservation workflow

This commit modifies the existing channel reservation workflow slightly
to thread through the IP address that we were able to reach the node
at, or the one which the node reached us via. Additionally, rather than
using OpenChannel.FullSync() at the end of the reservation workflow, we
now use OpenChannel.FullSyncWithAddr() in order to create the
relationship in the database between the channel, and the p2p node we
created the channel with.

All tests, as well as a portion of the fundingManager have been updated
accordingly,

fundingmanager.go

@@ -340,7 +340,7 @@ func (f *fundingManager) handleFundingRequest(fmsg *fundingRequestMsg) {
 	// channel ourselves.
 	// TODO(roasbeef): passing num confs 1 is irrelevant here, make signed?
 	reservation, err := f.wallet.InitChannelReservation(amt, 0,
-		fmsg.peer.identityPub, 1, delay)
+		fmsg.peer.identityPub, fmsg.peer.lightningAddr.NetAddr, 1, delay)
 	if err != nil {
 		// TODO(roasbeef): push ErrorGeneric message
 		fndgLog.Errorf("Unable to initialize reservation: %v", err)
@@ -723,13 +723,14 @@ func (f *fundingManager) handleInitFundingMsg(msg *initFundingMsg) {
 	)
 
 	fndgLog.Infof("Initiating fundingRequest(localAmt=%v, remoteAmt=%v, "+
-		"capacity=%v, numConfs=%v)", localAmt, remoteAmt, capacity, numConfs)
+		"capacity=%v, numConfs=%v, addr=%v)", localAmt, remoteAmt,
+		capacity, numConfs, msg.peer.lightningAddr.NetAddr)
 
 	// Initialize a funding reservation with the local wallet. If the
 	// wallet doesn't have enough funds to commit to this channel, then
 	// the request will fail, and be aborted.
 	reservation, err := f.wallet.InitChannelReservation(capacity, localAmt,
-		nodeID, uint16(numConfs), 4)
+		nodeID, msg.peer.lightningAddr.NetAddr, uint16(numConfs), 4)
 	if err != nil {
 		msg.err <- err
 		return

lnwallet/interface_test.go

@@ -5,6 +5,7 @@ import (
 	"encoding/hex"
 	"fmt"
 	"io/ioutil"
+	"net"
 	"os"
 	"path/filepath"
 	"testing"
@@ -59,6 +60,8 @@ var (
 	// The number of confirmations required to consider any created channel
 	// open.
 	numReqConfs = uint16(1)
+
+	bobAddr, _ = net.ResolveTCPAddr("tcp", "10.0.0.2:9000")
 )
 
 // assertProperBalance asserts than the total value of the unspent outputs
@@ -355,7 +358,7 @@ func testDualFundingReservationWorkflow(miner *rpctest.Harness, wallet *lnwallet
 	// Bob initiates a channel funded with 5 BTC for each side, so 10
 	// BTC total. He also generates 2 BTC in change.
 	chanReservation, err := wallet.InitChannelReservation(fundingAmount*2,
-		fundingAmount, bobNode.id, numReqConfs, 4)
+		fundingAmount, bobNode.id, bobAddr, numReqConfs, 4)
 	if err != nil {
 		t.Fatalf("unable to initialize funding reservation: %v", err)
 	}
@@ -514,7 +517,7 @@ func testFundingTransactionLockedOutputs(miner *rpctest.Harness,
 	// Create a single channel asking for 16 BTC total.
 	fundingAmount := btcutil.Amount(8 * 1e8)
 	_, err := wallet.InitChannelReservation(fundingAmount, fundingAmount,
-		testPub, numReqConfs, 4)
+		testPub, bobAddr, numReqConfs, 4)
 	if err != nil {
 		t.Fatalf("unable to initialize funding reservation 1: %v", err)
 	}
@@ -524,7 +527,7 @@ func testFundingTransactionLockedOutputs(miner *rpctest.Harness,
 	// that aren't locked, so this should fail.
 	amt := btcutil.Amount(900 * 1e8)
 	failedReservation, err := wallet.InitChannelReservation(amt, amt,
-		testPub, numReqConfs, 4)
+		testPub, bobAddr, numReqConfs, 4)
 	if err == nil {
 		t.Fatalf("not error returned, should fail on coin selection")
 	}
@@ -544,14 +547,14 @@ func testFundingCancellationNotEnoughFunds(miner *rpctest.Harness,
 	// Create a reservation for 44 BTC.
 	fundingAmount := btcutil.Amount(44 * 1e8)
 	chanReservation, err := wallet.InitChannelReservation(fundingAmount,
-		fundingAmount, testPub, numReqConfs, 4)
+		fundingAmount, testPub, bobAddr, numReqConfs, 4)
 	if err != nil {
 		t.Fatalf("unable to initialize funding reservation: %v", err)
 	}
 
 	// Attempt to create another channel with 44 BTC, this should fail.
 	_, err = wallet.InitChannelReservation(fundingAmount,
-		fundingAmount, testPub, numReqConfs, 4)
+		fundingAmount, testPub, bobAddr, numReqConfs, 4)
 	if _, ok := err.(*lnwallet.ErrInsufficientFunds); !ok {
 		t.Fatalf("coin selection succeded should have insufficient funds: %v",
 			err)
@@ -581,7 +584,7 @@ func testFundingCancellationNotEnoughFunds(miner *rpctest.Harness,
 
 	// Request to fund a new channel should now succeeed.
 	_, err = wallet.InitChannelReservation(fundingAmount, fundingAmount,
-		testPub, numReqConfs, 4)
+		testPub, bobAddr, numReqConfs, 4)
 	if err != nil {
 		t.Fatalf("unable to initialize funding reservation: %v", err)
 	}
@@ -620,7 +623,7 @@ func testSingleFunderReservationWorkflowInitiator(miner *rpctest.Harness,
 	// Initialize a reservation for a channel with 4 BTC funded solely by us.
 	fundingAmt := btcutil.Amount(4 * 1e8)
 	chanReservation, err := lnwallet.InitChannelReservation(fundingAmt,
-		fundingAmt, bobNode.id, numReqConfs, 4)
+		fundingAmt, bobNode.id, bobAddr, numReqConfs, 4)
 	if err != nil {
 		t.Fatalf("unable to init channel reservation: %v", err)
 	}
@@ -754,7 +757,7 @@ func testSingleFunderReservationWorkflowResponder(miner *rpctest.Harness,
 	// contribution and the necessary resources.
 	fundingAmt := btcutil.Amount(0)
 	chanReservation, err := wallet.InitChannelReservation(capacity,
-		fundingAmt, bobNode.id, numReqConfs, 4)
+		fundingAmt, bobNode.id, bobAddr, numReqConfs, 4)
 	if err != nil {
 		t.Fatalf("unable to init channel reservation: %v", err)
 	}

lnwallet/reservation.go

@@ -1,6 +1,7 @@
 package lnwallet
 
 import (
+	"net"
 	"sync"
 
 	"github.com/lightningnetwork/lnd/channeldb"
@@ -114,6 +115,7 @@ type ChannelReservation struct {
 	theirContribution *ChannelContribution
 
 	partialState *channeldb.OpenChannel
+	nodeAddr     *net.TCPAddr
 
 	// The ID of this reservation, used to uniquely track the reservation
 	// throughout its lifetime.

lnwallet/wallet.go

@@ -2,6 +2,7 @@ package lnwallet
 
 import (
 	"fmt"
+	"net"
 	"sync"
 	"sync/atomic"
 
@@ -69,10 +70,17 @@ func (e *ErrInsufficientFunds) Error() string {
 // selected will be 'locked', making them unavailable, for any other pending
 // reservations. Therefore, all channels in reservation limbo will be periodically
 // after a timeout period in order to avoid "exhaustion" attacks.
-// NOTE: The workflow currently assumes fully balanced symmetric channels.
-// Meaning both parties must encumber the same amount of funds.
+//
 // TODO(roasbeef): zombie reservation sweeper goroutine.
 type initFundingReserveMsg struct {
+	// The ID of the remote node we would like to open a channel with.
+	nodeID *btcec.PublicKey
+
+	// The IP address plus port that we used to either establish or accept
+	// the connection which led to the negotiation of this funding
+	// workflow.
+	nodeAddr *net.TCPAddr
+
 	// The number of confirmations required before the channel is considered
 	// open.
 	numConfs uint16
@@ -91,9 +99,6 @@ type initFundingReserveMsg struct {
 	// TODO(roasbeef): integrate fee estimation project...
 	minFeeRate btcutil.Amount
 
-	// The ID of the remote node we would like to open a channel with.
-	nodeID *btcec.PublicKey
-
 	// The delay on the "pay-to-self" output(s) of the commitment transaction.
 	csvDelay uint32
 
@@ -473,7 +478,8 @@ out:
 // commitment transaction is valid.
 func (l *LightningWallet) InitChannelReservation(capacity,
 	ourFundAmt btcutil.Amount, theirID *btcec.PublicKey,
-	numConfs uint16, csvDelay uint32) (*ChannelReservation, error) {
+	theirAddr *net.TCPAddr, numConfs uint16,
+	csvDelay uint32) (*ChannelReservation, error) {
 
 	errChan := make(chan error, 1)
 	respChan := make(chan *ChannelReservation, 1)
@@ -512,6 +518,7 @@ func (l *LightningWallet) handleFundingReserveRequest(req *initFundingReserveMsg
 	defer reservation.Unlock()
 
 	reservation.partialState.IdentityPub = req.nodeID
+	reservation.nodeAddr = req.nodeAddr
 	ourContribution := reservation.ourContribution
 	ourContribution.CsvDelay = req.csvDelay
 	reservation.partialState.LocalCsvDelay = req.csvDelay
@@ -903,7 +910,7 @@ func (l *LightningWallet) handleSingleContribution(req *addSingleContributionMsg
 // our signature.
 func (l *LightningWallet) handleFundingCounterPartySigs(msg *addCounterPartySigsMsg) {
 	l.limboMtx.RLock()
-	pendingReservation, ok := l.fundingLimbo[msg.pendingFundingID]
+	res, ok := l.fundingLimbo[msg.pendingFundingID]
 	l.limboMtx.RUnlock()
 	if !ok {
 		msg.err <- fmt.Errorf("attempted to update non-existant funding state")
@@ -911,14 +918,14 @@ func (l *LightningWallet) handleFundingCounterPartySigs(msg *addCounterPartySigs
 	}
 
 	// Grab the mutex on the ChannelReservation to ensure thead-safety
-	pendingReservation.Lock()
-	defer pendingReservation.Unlock()
+	res.Lock()
+	defer res.Unlock()
 
 	// Now we can complete the funding transaction by adding their
 	// signatures to their inputs.
-	pendingReservation.theirFundingInputScripts = msg.theirFundingInputScripts
+	res.theirFundingInputScripts = msg.theirFundingInputScripts
 	inputScripts := msg.theirFundingInputScripts
-	fundingTx := pendingReservation.fundingTx
+	fundingTx := res.fundingTx
 	sigIndex := 0
 	fundingHashCache := txscript.NewTxSigHashes(fundingTx)
 	for i, txin := range fundingTx.TxIn {
@@ -956,19 +963,19 @@ func (l *LightningWallet) handleFundingCounterPartySigs(msg *addCounterPartySigs
 
 	// At this point, we can also record and verify their signature for our
 	// commitment transaction.
-	pendingReservation.theirCommitmentSig = msg.theirCommitmentSig
-	commitTx := pendingReservation.partialState.OurCommitTx
-	theirKey := pendingReservation.theirContribution.MultiSigKey
+	res.theirCommitmentSig = msg.theirCommitmentSig
+	commitTx := res.partialState.OurCommitTx
+	theirKey := res.theirContribution.MultiSigKey
 
 	// Re-generate both the witnessScript and p2sh output. We sign the
 	// witnessScript script, but include the p2sh output as the subscript
 	// for verification.
-	witnessScript := pendingReservation.partialState.FundingWitnessScript
+	witnessScript := res.partialState.FundingWitnessScript
 
 	// Next, create the spending scriptSig, and then verify that the script
 	// is complete, allowing us to spend from the funding transaction.
 	theirCommitSig := msg.theirCommitmentSig
-	channelValue := int64(pendingReservation.partialState.Capacity)
+	channelValue := int64(res.partialState.Capacity)
 	hashCache := txscript.NewTxSigHashes(commitTx)
 	sigHash, err := txscript.CalcWitnessSigHash(witnessScript, hashCache,
 		txscript.SigHashAll, commitTx, 0, channelValue)
@@ -987,19 +994,15 @@ func (l *LightningWallet) handleFundingCounterPartySigs(msg *addCounterPartySigs
 		msg.err <- fmt.Errorf("counterparty's commitment signature is invalid")
 		return
 	}
-	pendingReservation.partialState.OurCommitSig = theirCommitSig
+	res.partialState.OurCommitSig = theirCommitSig
 
 	// Funding complete, this entry can be removed from limbo.
 	l.limboMtx.Lock()
-	delete(l.fundingLimbo, pendingReservation.reservationID)
-	// TODO(roasbeef): unlock outputs here, Store.InsertTx will handle marking
-	// input in unconfirmed tx, so future coin selects don't pick it up
-	//  * also record location of change address so can use AddCredit
+	delete(l.fundingLimbo, res.reservationID)
 	l.limboMtx.Unlock()
 
 	walletLog.Infof("Broadcasting funding tx for ChannelPoint(%v): %v",
-		pendingReservation.partialState.FundingOutpoint,
-		spew.Sdump(fundingTx))
+		res.partialState.FundingOutpoint, spew.Sdump(fundingTx))
 
 	// Broacast the finalized funding transaction to the network.
 	if err := l.PublishTransaction(fundingTx); err != nil {
@@ -1009,14 +1012,16 @@ func (l *LightningWallet) handleFundingCounterPartySigs(msg *addCounterPartySigs
 
 	// Add the complete funding transaction to the DB, in it's open bucket
 	// which will be used for the lifetime of this channel.
-	if err := pendingReservation.partialState.FullSync(); err != nil {
+	// TODO(roasbeef): revisit faul-tolerance of this flow
+	nodeAddr := res.nodeAddr
+	if err := res.partialState.FullSyncWithAddr(nodeAddr); err != nil {
 		msg.err <- err
 		return
 	}
 
 	// Create a goroutine to watch the chain so we can open the channel once
 	// the funding tx has enough confirmations.
-	go l.openChannelAfterConfirmations(pendingReservation)
+	go l.openChannelAfterConfirmations(res)
 
 	msg.err <- nil
 }
@@ -1136,7 +1141,8 @@ func (l *LightningWallet) handleChannelOpen(req *channelOpenMsg) {
 	res, ok := l.fundingLimbo[req.pendingFundingID]
 	l.limboMtx.RUnlock()
 	if !ok {
-		req.err <- fmt.Errorf("attempted to update non-existant funding state")
+		req.err <- fmt.Errorf("attempted to update non-existant " +
+			"funding state")
 		res.chanOpen <- nil
 		return
 	}
@@ -1152,7 +1158,7 @@ func (l *LightningWallet) handleChannelOpen(req *channelOpenMsg) {
 
 	// Add the complete funding transaction to the DB, in it's open bucket
 	// which will be used for the lifetime of this channel.
-	if err := res.partialState.FullSync(); err != nil {
+	if err := res.partialState.FullSyncWithAddr(res.nodeAddr); err != nil {
 		req.err <- err
 		res.chanOpen <- nil
 		return
@@ -1160,7 +1166,8 @@ func (l *LightningWallet) handleChannelOpen(req *channelOpenMsg) {
 
 	// Finally, create and officially open the payment channel!
 	// TODO(roasbeef): CreationTime once tx is 'open'
-	channel, _ := NewLightningChannel(l.Signer, l.chainIO, l.chainNotifier, res.partialState)
+	channel, _ := NewLightningChannel(l.Signer, l.chainIO, l.chainNotifier,
+		res.partialState)
 
 	res.chanOpen <- channel
 	req.err <- nil