mirror of
https://github.com/slackhq/nebula.git
synced 2025-11-08 18:03:58 +01:00
b37a91cfbc
This change add more metrics around "meta" (non "message" type packets). For lighthouse packets, we also record statistics around the specific lighthouse meta type. We don't keep statistics for the "message" type so that we don't slow down the fast path (and you can just look at metrics on the tun interface to find that information).
99 lines
2.6 KiB
Go
99 lines
2.6 KiB
Go
package nebula
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
|
|
proto "github.com/golang/protobuf/proto"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestNewLhQuery(t *testing.T) {
|
|
myIp := net.ParseIP("192.1.1.1")
|
|
myIpint := ip2int(myIp)
|
|
|
|
// Generating a new lh query should work
|
|
a := NewLhQueryByInt(myIpint)
|
|
|
|
// The result should be a nebulameta protobuf
|
|
assert.IsType(t, &NebulaMeta{}, a)
|
|
|
|
// It should also Marshal fine
|
|
b, err := proto.Marshal(a)
|
|
assert.Nil(t, err)
|
|
|
|
// and then Unmarshal fine
|
|
n := &NebulaMeta{}
|
|
err = proto.Unmarshal(b, n)
|
|
assert.Nil(t, err)
|
|
|
|
}
|
|
|
|
func TestNewipandportfromudpaddr(t *testing.T) {
|
|
blah := NewUDPAddrFromString("1.2.2.3:12345")
|
|
meh := NewIpAndPortFromUDPAddr(*blah)
|
|
assert.Equal(t, uint32(16908803), meh.Ip)
|
|
assert.Equal(t, uint32(12345), meh.Port)
|
|
}
|
|
|
|
func TestNewipandportsfromudpaddrs(t *testing.T) {
|
|
blah := NewUDPAddrFromString("1.2.2.3:12345")
|
|
blah2 := NewUDPAddrFromString("9.9.9.9:47828")
|
|
group := []udpAddr{*blah, *blah2}
|
|
hah := NewIpAndPortsFromNetIps(group)
|
|
assert.IsType(t, &[]*IpAndPort{}, hah)
|
|
//t.Error(reflect.TypeOf(hah))
|
|
|
|
}
|
|
|
|
func Test_lhStaticMapping(t *testing.T) {
|
|
lh1 := "10.128.0.2"
|
|
lh1IP := net.ParseIP(lh1)
|
|
|
|
udpServer, _ := NewListener("0.0.0.0", 0, true)
|
|
|
|
meh := NewLightHouse(true, 1, []uint32{ip2int(lh1IP)}, 10, 10003, udpServer, false, 1, false)
|
|
meh.AddRemote(ip2int(lh1IP), NewUDPAddr(ip2int(lh1IP), uint16(4242)), true)
|
|
err := meh.ValidateLHStaticEntries()
|
|
assert.Nil(t, err)
|
|
|
|
lh2 := "10.128.0.3"
|
|
lh2IP := net.ParseIP(lh2)
|
|
|
|
meh = NewLightHouse(true, 1, []uint32{ip2int(lh1IP), ip2int(lh2IP)}, 10, 10003, udpServer, false, 1, false)
|
|
meh.AddRemote(ip2int(lh1IP), NewUDPAddr(ip2int(lh1IP), uint16(4242)), true)
|
|
err = meh.ValidateLHStaticEntries()
|
|
assert.EqualError(t, err, "Lighthouse 10.128.0.3 does not have a static_host_map entry")
|
|
}
|
|
|
|
//func NewLightHouse(amLighthouse bool, myIp uint32, ips []string, interval int, nebulaPort int, pc *udpConn, punchBack bool) *LightHouse {
|
|
|
|
/*
|
|
func TestLHQuery(t *testing.T) {
|
|
//n := NewLhQueryByIpString("10.128.0.3")
|
|
_, myNet, _ := net.ParseCIDR("10.128.0.0/16")
|
|
m := NewHostMap(myNet)
|
|
y, _ := net.ResolveUDPAddr("udp", "10.128.0.3:11111")
|
|
m.Add(ip2int(net.ParseIP("127.0.0.1")), y)
|
|
//t.Errorf("%s", m)
|
|
_ = m
|
|
|
|
_, n, _ := net.ParseCIDR("127.0.0.1/8")
|
|
|
|
/*udpServer, err := net.ListenUDP("udp", &net.UDPAddr{Port: 10009})
|
|
if err != nil {
|
|
t.Errorf("%s", err)
|
|
}
|
|
|
|
meh := NewLightHouse(n, m, []string{"10.128.0.2"}, false, 10, 10003, 10004)
|
|
//t.Error(m.Hosts)
|
|
meh2, err := meh.Query(ip2int(net.ParseIP("10.128.0.3")))
|
|
t.Error(err)
|
|
if err != nil {
|
|
return
|
|
}
|
|
t.Errorf("%s", meh2)
|
|
t.Errorf("%s", n)
|
|
}
|
|
*/
|