slackhq.nebula/hostmap_test.go

package nebula

import (
	"net"
	"testing"

	"github.com/stretchr/testify/assert"
)

/*
func TestHostInfoDestProbe(t *testing.T) {
	a, _ := net.ResolveUDPAddr("udp", "1.0.0.1:22222")
	d := NewHostInfoDest(a)

	// 999 probes that all return should give a 100% success rate
	for i := 0; i < 999; i++ {
		meh := d.Probe()
		d.ProbeReceived(meh)
	}
	assert.Equal(t, d.Grade(), float64(1))

	// 999 probes of which only half return should give a 50% success rate
	for i := 0; i < 999; i++ {
		meh := d.Probe()
		if i%2 == 0 {
			d.ProbeReceived(meh)
		}
	}
	assert.Equal(t, d.Grade(), float64(.5))

	// 999 probes of which none return should give a 0% success rate
	for i := 0; i < 999; i++ {
		d.Probe()
	}
	assert.Equal(t, d.Grade(), float64(0))

	// 999 probes of which only 1/4 return should give a 25% success rate
	for i := 0; i < 999; i++ {
		meh := d.Probe()
		if i%4 == 0 {
			d.ProbeReceived(meh)
		}
	}
	assert.Equal(t, d.Grade(), float64(.25))

	// 999 probes of which only half return and are duplicates should give a 50% success rate
	for i := 0; i < 999; i++ {
		meh := d.Probe()
		if i%2 == 0 {
			d.ProbeReceived(meh)
			d.ProbeReceived(meh)
		}
	}
	assert.Equal(t, d.Grade(), float64(.5))

	// 999 probes of which only way old replies return should give a 0% success rate
	for i := 0; i < 999; i++ {
		meh := d.Probe()
		d.ProbeReceived(meh - 101)
	}
	assert.Equal(t, d.Grade(), float64(0))

}
*/

func TestHostmap(t *testing.T) {
	_, myNet, _ := net.ParseCIDR("10.128.0.0/16")
	_, localToMe, _ := net.ParseCIDR("192.168.1.0/24")
	myNets := []*net.IPNet{myNet}
	preferredRanges := []*net.IPNet{localToMe}

	m := NewHostMap("test", myNet, preferredRanges)

	a := NewUDPAddrFromString("10.127.0.3:11111")
	b := NewUDPAddrFromString("1.0.0.1:22222")
	y := NewUDPAddrFromString("10.128.0.3:11111")
	m.AddRemote(ip2int(net.ParseIP("127.0.0.1")), a)
	m.AddRemote(ip2int(net.ParseIP("127.0.0.1")), b)
	m.AddRemote(ip2int(net.ParseIP("127.0.0.1")), y)

	info, _ := m.QueryVpnIP(ip2int(net.ParseIP("127.0.0.1")))

	// There should be three remotes in the host map
	assert.Equal(t, 3, len(info.Remotes))

	// Adding an identical remote should not change the count
	m.AddRemote(ip2int(net.ParseIP("127.0.0.1")), y)
	assert.Equal(t, 3, len(info.Remotes))

	// Adding a fresh remote should add one
	y = NewUDPAddrFromString("10.18.0.3:11111")
	m.AddRemote(ip2int(net.ParseIP("127.0.0.1")), y)
	assert.Equal(t, 4, len(info.Remotes))

	// Query and reference remote should get the first one (and not nil)
	info, _ = m.QueryVpnIP(ip2int(net.ParseIP("127.0.0.1")))
	assert.NotNil(t, info.remote)

	// Promotion should ensure that the best remote is chosen (y)
	info.ForcePromoteBest(myNets)
	assert.True(t, myNet.Contains(udp2ip(info.remote)))

}

func TestHostmapdebug(t *testing.T) {
	_, myNet, _ := net.ParseCIDR("10.128.0.0/16")
	_, localToMe, _ := net.ParseCIDR("192.168.1.0/24")
	preferredRanges := []*net.IPNet{localToMe}
	m := NewHostMap("test", myNet, preferredRanges)

	a := NewUDPAddrFromString("10.127.0.3:11111")
	b := NewUDPAddrFromString("1.0.0.1:22222")
	y := NewUDPAddrFromString("10.128.0.3:11111")
	m.AddRemote(ip2int(net.ParseIP("127.0.0.1")), a)
	m.AddRemote(ip2int(net.ParseIP("127.0.0.1")), b)
	m.AddRemote(ip2int(net.ParseIP("127.0.0.1")), y)

	//t.Errorf("%s", m.DebugRemotes(1))
}

func TestHostMap_rotateRemote(t *testing.T) {
	h := HostInfo{}
	// 0 remotes, no panic
	h.rotateRemote()
	assert.Nil(t, h.remote)

	// 1 remote, no panic
	h.AddRemote(*NewUDPAddr(ip2int(net.IP{1, 1, 1, 1}), 0))
	h.rotateRemote()
	assert.Equal(t, udp2ipInt(h.remote), ip2int(net.IP{1, 1, 1, 1}))

	h.AddRemote(*NewUDPAddr(ip2int(net.IP{1, 1, 1, 2}), 0))
	h.AddRemote(*NewUDPAddr(ip2int(net.IP{1, 1, 1, 3}), 0))
	h.AddRemote(*NewUDPAddr(ip2int(net.IP{1, 1, 1, 4}), 0))

	// Rotate through those 3
	h.rotateRemote()
	assert.Equal(t, udp2ipInt(h.remote), ip2int(net.IP{1, 1, 1, 2}))

	h.rotateRemote()
	assert.Equal(t, udp2ipInt(h.remote), ip2int(net.IP{1, 1, 1, 3}))

	h.rotateRemote()
	assert.Equal(t, udp2ipInt(h.remote), ip2int(net.IP{1, 1, 1, 4}))

	// Finally, we should start over
	h.rotateRemote()
	assert.Equal(t, udp2ipInt(h.remote), ip2int(net.IP{1, 1, 1, 1}))
}

func BenchmarkHostmappromote2(b *testing.B) {
	for n := 0; n < b.N; n++ {
		_, myNet, _ := net.ParseCIDR("10.128.0.0/16")
		_, localToMe, _ := net.ParseCIDR("192.168.1.0/24")
		preferredRanges := []*net.IPNet{localToMe}
		m := NewHostMap("test", myNet, preferredRanges)
		y := NewUDPAddrFromString("10.128.0.3:11111")
		a := NewUDPAddrFromString("10.127.0.3:11111")
		g := NewUDPAddrFromString("1.0.0.1:22222")
		m.AddRemote(ip2int(net.ParseIP("127.0.0.1")), a)
		m.AddRemote(ip2int(net.ParseIP("127.0.0.1")), g)
		m.AddRemote(ip2int(net.ParseIP("127.0.0.1")), y)
	}
	b.Errorf("hi")

}