Gijs van Dam
07dc74e198
This commit creates the necessary endpoints for onion messages. Specifically, it adds the following: - `SendOnionMessage` endpoint to send onion messages. - `SubscribeOnionMessages` endpoint to subscribe to incoming onion messages. It uses the `msgmux` package to handle the onion messages.
81 lines
2.2 KiB
Go
81 lines
2.2 KiB
Go
package itest
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/btcsuite/btcd/btcec/v2"
|
|
"github.com/lightningnetwork/lnd/lnrpc"
|
|
"github.com/lightningnetwork/lnd/lntest"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// testOnionMessage tests sending and receiving of the onion message type.
|
|
func testOnionMessage(ht *lntest.HarnessTest) {
|
|
alice := ht.NewNode("Alice", nil)
|
|
bob := ht.NewNode("Bob", nil)
|
|
|
|
// Subscribe Alice to onion messages before we send any, so that we
|
|
// don't miss any.
|
|
msgClient, cancel := alice.RPC.SubscribeOnionMessages()
|
|
defer cancel()
|
|
|
|
// Create a channel to receive onion messages on.
|
|
messages := make(chan *lnrpc.OnionMessage)
|
|
go func() {
|
|
for {
|
|
// If we fail to receive, just exit. The test should
|
|
// fail elsewhere if it doesn't get a message that it
|
|
// was expecting.
|
|
msg, err := msgClient.Recv()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
// Deliver the message into our channel or exit if the
|
|
// test is shutting down.
|
|
select {
|
|
case messages <- msg:
|
|
case <-ht.Context().Done():
|
|
return
|
|
}
|
|
}
|
|
}()
|
|
|
|
// Connect alice and bob so that they can exchange messages.
|
|
ht.EnsureConnected(alice, bob)
|
|
|
|
// Create a random onion message.
|
|
randomPriv, err := btcec.NewPrivateKey()
|
|
require.NoError(ht.T, err)
|
|
randomPub := randomPriv.PubKey()
|
|
msgPathKey := randomPub.SerializeCompressed()
|
|
// Create a random payload. The content doesn't matter for this and
|
|
// doesn't need to be encrypted. It's also of arbitrary length, so it
|
|
// doesn't follow the BOLT 4 spec for onion message payload length of
|
|
// either 1300 or 32768 bytes. Here we just use a few bytes to keep it
|
|
// simple.
|
|
msgOnion := []byte{1, 2, 3}
|
|
|
|
// Send it from Bob to Alice.
|
|
bobMsg := &lnrpc.SendOnionMessageRequest{
|
|
Peer: alice.PubKey[:],
|
|
PathKey: msgPathKey,
|
|
Onion: msgOnion,
|
|
}
|
|
bob.RPC.SendOnionMessage(bobMsg)
|
|
|
|
// Wait for Alice to receive the message.
|
|
select {
|
|
case msg := <-messages:
|
|
// Check our type and data and (sanity) check the peer we got
|
|
// it from.
|
|
require.Equal(ht, msgOnion, msg.Onion, "msg data wrong")
|
|
require.Equal(ht, msgPathKey, msg.PathKey, "msg "+
|
|
"path key wrong")
|
|
require.Equal(ht, bob.PubKey[:], msg.Peer, "msg peer wrong")
|
|
|
|
case <-time.After(lntest.DefaultTimeout):
|
|
ht.Fatalf("alice did not receive onion message: %v", bobMsg)
|
|
}
|
|
}
|