integrationradvd: unflake test

This commit is contained in:
Michael Stapelberg 2018-06-08 16:51:33 +02:00
parent 4baac19e13
commit c37ddf4f6e
2 changed files with 20 additions and 10 deletions

View File

@ -30,6 +30,8 @@ func TestRouterAdvertisement(t *testing.T) {
exec.Command("ip", "netns", "exec", ns, "ip", "addr", "add", "192.168.23.1/24", "dev", "veth0b"), exec.Command("ip", "netns", "exec", ns, "ip", "addr", "add", "192.168.23.1/24", "dev", "veth0b"),
exec.Command("ip", "netns", "exec", ns, "ip", "link", "set", "veth0b", "up"), exec.Command("ip", "netns", "exec", ns, "ip", "link", "set", "veth0b", "up"),
exec.Command("ip", "netns", "exec", ns, "ip", "link", "set", "veth0b"), exec.Command("ip", "netns", "exec", ns, "ip", "link", "set", "veth0b"),
exec.Command("/bin/sh", "-c", "echo 1 > /proc/sys/net/ipv6/conf/veth0a/forwarding"),
} }
for _, cmd := range nsSetup { for _, cmd := range nsSetup {
@ -52,8 +54,12 @@ func TestRouterAdvertisement(t *testing.T) {
srv.SetPrefixes([]net.IPNet{ srv.SetPrefixes([]net.IPNet{
net.IPNet{IP: net.ParseIP("2a02:168:4a00::"), Mask: net.CIDRMask(64, 128)}, net.IPNet{IP: net.ParseIP("2a02:168:4a00::"), Mask: net.CIDRMask(64, 128)},
}) })
conn, err := net.ListenIP("ip6:ipv6-icmp", &net.IPAddr{net.IPv6unspecified, ""})
if err != nil {
t.Fatal(err)
}
go func() { go func() {
if err := srv.ListenAndServe("veth0a"); err != nil { if err := srv.Serve("veth0a", conn); err != nil {
t.Fatal(err) t.Fatal(err)
} }
}() }()
@ -66,7 +72,7 @@ func TestRouterAdvertisement(t *testing.T) {
rdisc6.Stderr = os.Stderr rdisc6.Stderr = os.Stderr
b, err := rdisc6.Output() b, err := rdisc6.Output()
if err != nil { if err != nil {
t.Fatal(err) t.Fatalf("%v: %v (output: %v)", rdisc6.Args, err, string(b))
} }
t.Logf("b = %s", string(b)) t.Logf("b = %s", string(b))

View File

@ -36,20 +36,13 @@ func (s *Server) SetPrefixes(prefixes []net.IPNet) {
} }
} }
func (s *Server) ListenAndServe(ifname string) error { func (s *Server) Serve(ifname string, conn net.PacketConn) error {
var err error var err error
s.iface, err = net.InterfaceByName(ifname) s.iface, err = net.InterfaceByName(ifname)
if err != nil { if err != nil {
return err return err
} }
// TODO(correctness): would it be better to listen on
// net.IPv6linklocalallrouters? Just specifying that results in an error,
// though.
conn, err := net.ListenIP("ip6:ipv6-icmp", &net.IPAddr{net.IPv6unspecified, ""})
if err != nil {
return err
}
defer conn.Close() defer conn.Close()
s.pc = ipv6.NewPacketConn(conn) s.pc = ipv6.NewPacketConn(conn)
s.pc.SetHopLimit(255) // as per RFC 4861, section 4.1 s.pc.SetHopLimit(255) // as per RFC 4861, section 4.1
@ -89,6 +82,17 @@ func (s *Server) ListenAndServe(ifname string) error {
return nil return nil
} }
func (s *Server) ListenAndServe(ifname string) error {
// TODO(correctness): would it be better to listen on
// net.IPv6linklocalallrouters? Just specifying that results in an error,
// though.
conn, err := net.ListenIP("ip6:ipv6-icmp", &net.IPAddr{net.IPv6unspecified, ""})
if err != nil {
return err
}
return s.Serve(ifname, conn)
}
type sourceLinkLayerAddress struct { type sourceLinkLayerAddress struct {
address net.HardwareAddr address net.HardwareAddr
} }