111 lines
3 KiB
Go
111 lines
3 KiB
Go
|
// Copyright (c) 2013 Conformal Systems LLC.
|
||
|
|
// Use of this source code is governed by an ISC
|
||
|
|
// license that can be found in the LICENSE file.
|
||
|
|
|
||
|
|
package main_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/conformal/btcwire"
|
||
|
|
"net"
|
||
|
|
"opensource.conformal.com/go/btcd-internal/addrmgr"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
// naTest is used to describe a test to be perfomed against the NetAddressKey
|
||
|
|
// method.
|
||
|
|
type naTest struct {
|
||
|
|
in btcwire.NetAddress
|
||
|
|
want string
|
||
|
|
}
|
||
|
|
|
||
|
|
// naTests houses all of the tests to be performed against the NetAddressKey
|
||
|
|
// method.
|
||
|
|
var naTests = make([]naTest, 0)
|
||
|
|
|
||
|
|
func addNaTest(ip string, port uint16, want string) {
|
||
|
|
nip := net.ParseIP(ip)
|
||
|
|
na := btcwire.NetAddress{
|
||
|
|
Timestamp: time.Now(),
|
||
|
|
Services: btcwire.SFNodeNetwork,
|
||
|
|
IP: nip,
|
||
|
|
Port: port,
|
||
|
|
}
|
||
|
|
test := naTest{na, want}
|
||
|
|
naTests = append(naTests, test)
|
||
|
|
}
|
||
|
|
|
||
|
|
// addNaTests
|
||
|
|
func addNaTests() {
|
||
|
|
// IPv4
|
||
|
|
// Localhost
|
||
|
|
addNaTest("127.0.0.1", 8333, "127.0.0.1:8333")
|
||
|
|
addNaTest("127.0.0.1", 8334, "127.0.0.1:8334")
|
||
|
|
|
||
|
|
// Class A
|
||
|
|
addNaTest("1.0.0.1", 8333, "1.0.0.1:8333")
|
||
|
|
addNaTest("2.2.2.2", 8334, "2.2.2.2:8334")
|
||
|
|
addNaTest("27.253.252.251", 8335, "27.253.252.251:8335")
|
||
|
|
addNaTest("123.3.2.1", 8336, "123.3.2.1:8336")
|
||
|
|
|
||
|
|
// Private Class A
|
||
|
|
addNaTest("10.0.0.1", 8333, "10.0.0.1:8333")
|
||
|
|
addNaTest("10.1.1.1", 8334, "10.1.1.1:8334")
|
||
|
|
addNaTest("10.2.2.2", 8335, "10.2.2.2:8335")
|
||
|
|
addNaTest("10.10.10.10", 8336, "10.10.10.10:8336")
|
||
|
|
|
||
|
|
// Class B
|
||
|
|
addNaTest("128.0.0.1", 8333, "128.0.0.1:8333")
|
||
|
|
addNaTest("129.1.1.1", 8334, "129.1.1.1:8334")
|
||
|
|
addNaTest("180.2.2.2", 8335, "180.2.2.2:8335")
|
||
|
|
addNaTest("191.10.10.10", 8336, "191.10.10.10:8336")
|
||
|
|
|
||
|
|
// Private Class B
|
||
|
|
addNaTest("172.16.0.1", 8333, "172.16.0.1:8333")
|
||
|
|
addNaTest("172.16.1.1", 8334, "172.16.1.1:8334")
|
||
|
|
addNaTest("172.16.2.2", 8335, "172.16.2.2:8335")
|
||
|
|
addNaTest("172.16.172.172", 8336, "172.16.172.172:8336")
|
||
|
|
|
||
|
|
// Class C
|
||
|
|
addNaTest("193.0.0.1", 8333, "193.0.0.1:8333")
|
||
|
|
addNaTest("200.1.1.1", 8334, "200.1.1.1:8334")
|
||
|
|
addNaTest("205.2.2.2", 8335, "205.2.2.2:8335")
|
||
|
|
addNaTest("223.10.10.10", 8336, "223.10.10.10:8336")
|
||
|
|
|
||
|
|
// Private Class C
|
||
|
|
addNaTest("192.168.0.1", 8333, "192.168.0.1:8333")
|
||
|
|
addNaTest("192.168.1.1", 8334, "192.168.1.1:8334")
|
||
|
|
addNaTest("192.168.2.2", 8335, "192.168.2.2:8335")
|
||
|
|
addNaTest("192.168.192.192", 8336, "192.168.192.192:8336")
|
||
|
|
|
||
|
|
// IPv6
|
||
|
|
// Localhost
|
||
|
|
addNaTest("::1", 8333, "[::1]:8333")
|
||
|
|
addNaTest("fe80::1", 8334, "[fe80::1]:8334")
|
||
|
|
|
||
|
|
// Link-local
|
||
|
|
addNaTest("fe80::1:1", 8333, "[fe80::1:1]:8333")
|
||
|
|
addNaTest("fe91::2:2", 8334, "[fe91::2:2]:8334")
|
||
|
|
addNaTest("fea2::3:3", 8335, "[fea2::3:3]:8335")
|
||
|
|
addNaTest("feb3::4:4", 8336, "[feb3::4:4]:8336")
|
||
|
|
|
||
|
|
// Site-local
|
||
|
|
addNaTest("fec0::1:1", 8333, "[fec0::1:1]:8333")
|
||
|
|
addNaTest("fed1::2:2", 8334, "[fed1::2:2]:8334")
|
||
|
|
addNaTest("fee2::3:3", 8335, "[fee2::3:3]:8335")
|
||
|
|
addNaTest("fef3::4:4", 8336, "[fef3::4:4]:8336")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNetAddressKey(t *testing.T) {
|
||
|
|
addNaTests()
|
||
|
|
|
||
|
|
t.Logf("Running %d tests", len(naTests))
|
||
|
|
for i, test := range naTests {
|
||
|
|
key := addrmgr.NetAddressKey(&test.in)
|
||
|
|
if key != test.want {
|
||
|
|
t.Errorf("NetAddressKey #%d\n got: %s want: %s", i, key, test.want)
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|