-
Notifications
You must be signed in to change notification settings - Fork 2.1k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #8390 from carlaKC/7883-experimental-endorsement
Add Experimental Endorsement Signalling
- Loading branch information
Showing
20 changed files
with
380 additions
and
40 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,124 @@ | ||
package itest | ||
|
||
import ( | ||
"math" | ||
|
||
"github.com/btcsuite/btcd/btcutil" | ||
"github.com/lightningnetwork/lnd/lnrpc" | ||
"github.com/lightningnetwork/lnd/lnrpc/routerrpc" | ||
"github.com/lightningnetwork/lnd/lntest" | ||
"github.com/lightningnetwork/lnd/lntest/rpc" | ||
"github.com/lightningnetwork/lnd/lntest/wait" | ||
"github.com/lightningnetwork/lnd/lntypes" | ||
"github.com/lightningnetwork/lnd/lnwire" | ||
"github.com/stretchr/testify/require" | ||
) | ||
|
||
// testExperimentalEndorsement tests setting of positive and negative | ||
// experimental endorsement signals. | ||
func testExperimentalEndorsement(ht *lntest.HarnessTest) { | ||
testEndorsement(ht, true) | ||
testEndorsement(ht, false) | ||
} | ||
|
||
// testEndorsement sets up a 5 hop network and tests propagation of | ||
// experimental endorsement signals. | ||
func testEndorsement(ht *lntest.HarnessTest, aliceEndorse bool) { | ||
alice, bob := ht.Alice, ht.Bob | ||
carol := ht.NewNode( | ||
"carol", []string{"--protocol.no-experimental-endorsement"}, | ||
) | ||
dave := ht.NewNode("dave", nil) | ||
eve := ht.NewNode("eve", nil) | ||
|
||
ht.EnsureConnected(alice, bob) | ||
ht.EnsureConnected(bob, carol) | ||
ht.EnsureConnected(carol, dave) | ||
ht.EnsureConnected(dave, eve) | ||
|
||
ht.FundCoins(btcutil.SatoshiPerBitcoin, carol) | ||
ht.FundCoins(btcutil.SatoshiPerBitcoin, dave) | ||
// Open and wait for channels. | ||
const chanAmt = btcutil.Amount(300000) | ||
p := lntest.OpenChannelParams{Amt: chanAmt} | ||
reqs := []*lntest.OpenChannelRequest{ | ||
{Local: alice, Remote: bob, Param: p}, | ||
{Local: bob, Remote: carol, Param: p}, | ||
{Local: carol, Remote: dave, Param: p}, | ||
{Local: dave, Remote: eve, Param: p}, | ||
} | ||
resp := ht.OpenMultiChannelsAsync(reqs) | ||
cpAB, cpBC, cpCD, cpDE := resp[0], resp[1], resp[2], resp[3] | ||
|
||
// Make sure Alice is aware of Bob=>Carol=>Dave=>Eve channels. | ||
ht.AssertChannelInGraph(alice, cpBC) | ||
ht.AssertChannelInGraph(alice, cpCD) | ||
ht.AssertChannelInGraph(alice, cpDE) | ||
|
||
bobIntercept, cancelBob := bob.RPC.HtlcInterceptor() | ||
defer cancelBob() | ||
|
||
carolIntercept, cancelCarol := carol.RPC.HtlcInterceptor() | ||
defer cancelCarol() | ||
|
||
daveIntercept, cancelDave := dave.RPC.HtlcInterceptor() | ||
defer cancelDave() | ||
|
||
req := &lnrpc.Invoice{ValueMsat: 1000} | ||
addResponse := eve.RPC.AddInvoice(req) | ||
invoice := eve.RPC.LookupInvoice(addResponse.RHash) | ||
|
||
sendReq := &routerrpc.SendPaymentRequest{ | ||
PaymentRequest: invoice.PaymentRequest, | ||
TimeoutSeconds: int32(wait.PaymentTimeout.Seconds()), | ||
FeeLimitMsat: math.MaxInt64, | ||
} | ||
|
||
expectedValue := []byte{lnwire.ExperimentalUnendorsed} | ||
if aliceEndorse { | ||
expectedValue = []byte{lnwire.ExperimentalEndorsed} | ||
t := uint64(lnwire.ExperimentalEndorsementType) | ||
sendReq.FirstHopCustomRecords = map[uint64][]byte{ | ||
t: expectedValue, | ||
} | ||
} | ||
|
||
_ = alice.RPC.SendPayment(sendReq) | ||
|
||
// Validate that our signal (positive or zero) propagates until carol | ||
// and then is dropped because she has disabled the feature. | ||
validateEndorsedAndResume(ht, bobIntercept, true, expectedValue) | ||
validateEndorsedAndResume(ht, carolIntercept, true, expectedValue) | ||
validateEndorsedAndResume(ht, daveIntercept, false, nil) | ||
|
||
var preimage lntypes.Preimage | ||
copy(preimage[:], invoice.RPreimage) | ||
ht.AssertPaymentStatus(alice, preimage, lnrpc.Payment_SUCCEEDED) | ||
|
||
ht.CloseChannel(alice, cpAB) | ||
ht.CloseChannel(bob, cpBC) | ||
ht.CloseChannel(carol, cpCD) | ||
ht.CloseChannel(dave, cpDE) | ||
} | ||
|
||
func validateEndorsedAndResume(ht *lntest.HarnessTest, | ||
interceptor rpc.InterceptorClient, hasEndorsement bool, | ||
expectedValue []byte) { | ||
|
||
packet := ht.ReceiveHtlcInterceptor(interceptor) | ||
|
||
var expectedRecords map[uint64][]byte | ||
if hasEndorsement { | ||
u64Type := uint64(lnwire.ExperimentalEndorsementType) | ||
expectedRecords = map[uint64][]byte{ | ||
u64Type: expectedValue, | ||
} | ||
} | ||
require.Equal(ht, expectedRecords, packet.InWireCustomRecords) | ||
|
||
err := interceptor.Send(&routerrpc.ForwardHtlcInterceptResponse{ | ||
IncomingCircuitKey: packet.IncomingCircuitKey, | ||
Action: routerrpc.ResolveHoldForwardAction_RESUME, | ||
}) | ||
require.NoError(ht, err) | ||
} |
Oops, something went wrong.