mirror of
https://github.com/lightningnetwork/lnd.git
synced 2025-03-04 17:55:36 +01:00
3365461500
In preparation for moving the graph related schema structs to the graph package in an upcoming commit, we move these methods to the graph package. The structs we will move make use of these methods but we still import them from channeldb so as to keep the ReadElement and WriteElement helpers working as they do today.
149 lines
3.1 KiB
Go
149 lines
3.1 KiB
Go
package graphdb
|
|
|
|
import (
|
|
"bytes"
|
|
"net"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/lightningnetwork/lnd/tor"
|
|
)
|
|
|
|
type unknownAddrType struct{}
|
|
|
|
func (t unknownAddrType) Network() string { return "unknown" }
|
|
func (t unknownAddrType) String() string { return "unknown" }
|
|
|
|
var testIP4 = net.ParseIP("192.168.1.1")
|
|
var testIP6 = net.ParseIP("2001:0db8:0000:0000:0000:ff00:0042:8329")
|
|
|
|
var addrTests = []struct {
|
|
expAddr net.Addr
|
|
serErr string
|
|
}{
|
|
// Valid addresses.
|
|
{
|
|
expAddr: &net.TCPAddr{
|
|
IP: testIP4,
|
|
Port: 12345,
|
|
},
|
|
},
|
|
{
|
|
expAddr: &net.TCPAddr{
|
|
IP: testIP6,
|
|
Port: 65535,
|
|
},
|
|
},
|
|
{
|
|
expAddr: &tor.OnionAddr{
|
|
OnionService: "3g2upl4pq6kufc4m.onion",
|
|
Port: 9735,
|
|
},
|
|
},
|
|
{
|
|
expAddr: &tor.OnionAddr{
|
|
OnionService: "vww6ybal4bd7szmgncyruucpgfkqahzddi37ktceo3ah7ngmcopnpyyd.onion",
|
|
Port: 80,
|
|
},
|
|
},
|
|
|
|
// Invalid addresses.
|
|
{
|
|
expAddr: unknownAddrType{},
|
|
serErr: ErrUnknownAddressType.Error(),
|
|
},
|
|
{
|
|
expAddr: &net.TCPAddr{
|
|
// Remove last byte of IPv4 address.
|
|
IP: testIP4[:len(testIP4)-1],
|
|
Port: 12345,
|
|
},
|
|
serErr: "unable to encode",
|
|
},
|
|
{
|
|
expAddr: &net.TCPAddr{
|
|
// Add an extra byte of IPv4 address.
|
|
IP: append(testIP4, 0xff),
|
|
Port: 12345,
|
|
},
|
|
serErr: "unable to encode",
|
|
},
|
|
{
|
|
expAddr: &net.TCPAddr{
|
|
// Remove last byte of IPv6 address.
|
|
IP: testIP6[:len(testIP6)-1],
|
|
Port: 65535,
|
|
},
|
|
serErr: "unable to encode",
|
|
},
|
|
{
|
|
expAddr: &net.TCPAddr{
|
|
// Add an extra byte to the IPv6 address.
|
|
IP: append(testIP6, 0xff),
|
|
Port: 65535,
|
|
},
|
|
serErr: "unable to encode",
|
|
},
|
|
{
|
|
expAddr: &tor.OnionAddr{
|
|
// Invalid suffix.
|
|
OnionService: "vww6ybal4bd7szmgncyruucpgfkqahzddi37ktceo3ah7ngmcopnpyyd.inion",
|
|
Port: 80,
|
|
},
|
|
serErr: "invalid suffix",
|
|
},
|
|
{
|
|
expAddr: &tor.OnionAddr{
|
|
// Invalid length.
|
|
OnionService: "vww6ybal4bd7szmgncyruucpgfkqahzddi37ktceo3ah7ngmcopnpyy.onion",
|
|
Port: 80,
|
|
},
|
|
serErr: "unknown onion service length",
|
|
},
|
|
{
|
|
expAddr: &tor.OnionAddr{
|
|
// Invalid encoding.
|
|
OnionService: "vww6ybal4bd7szmgncyruucpgfkqahzddi37ktceo3ah7ngmcopnpyyA.onion",
|
|
Port: 80,
|
|
},
|
|
serErr: "illegal base32",
|
|
},
|
|
}
|
|
|
|
// TestAddrSerialization tests that the serialization method used by channeldb
|
|
// for net.Addr's works as intended.
|
|
func TestAddrSerialization(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
var b bytes.Buffer
|
|
for _, test := range addrTests {
|
|
err := SerializeAddr(&b, test.expAddr)
|
|
switch {
|
|
case err == nil && test.serErr != "":
|
|
t.Fatalf("expected serialization err for addr %v",
|
|
test.expAddr)
|
|
|
|
case err != nil && test.serErr == "":
|
|
t.Fatalf("unexpected serialization err for addr %v: %v",
|
|
test.expAddr, err)
|
|
|
|
case err != nil && !strings.Contains(err.Error(), test.serErr):
|
|
t.Fatalf("unexpected serialization err for addr %v, "+
|
|
"want: %v, got %v", test.expAddr, test.serErr,
|
|
err)
|
|
|
|
case err != nil:
|
|
continue
|
|
}
|
|
|
|
addr, err := DeserializeAddr(&b)
|
|
if err != nil {
|
|
t.Fatalf("unable to deserialize address: %v", err)
|
|
}
|
|
|
|
if addr.String() != test.expAddr.String() {
|
|
t.Fatalf("expected address %v after serialization, "+
|
|
"got %v", addr, test.expAddr)
|
|
}
|
|
}
|
|
}
|