mirror of
https://github.com/miekg/dns.git
synced 2025-10-10 17:31:01 +02:00
032fbabc82
Previously, the oob data was just stored and sent to WriteMsgUDP but it ignores the Src field when writing. Instead, now it is setting the Src to the original Dst and handling IPv4 IPs over IPv6 correctly.
69 lines
2.1 KiB
Go
69 lines
2.1 KiB
Go
// +build linux,!appengine
|
|
|
|
package dns
|
|
|
|
import (
|
|
"bytes"
|
|
"net"
|
|
"testing"
|
|
)
|
|
|
|
func TestParseUDPSocketDst(t *testing.T) {
|
|
// dst is :ffff:100.100.100.100
|
|
oob := []byte{36, 0, 0, 0, 0, 0, 0, 0, 41, 0, 0, 0, 50, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 255, 255, 100, 100, 100, 100, 2, 0, 0, 0}
|
|
dst, err := parseUDPSocketDst(oob)
|
|
if err != nil {
|
|
t.Fatalf("error parsing ipv6 oob: %v", err)
|
|
}
|
|
dst4 := dst.To4()
|
|
if dst4 == nil {
|
|
t.Errorf("failed to parse ipv4: %v", dst)
|
|
} else if dst4.String() != "100.100.100.100" {
|
|
t.Errorf("unexpected ipv4: %v", dst4)
|
|
}
|
|
|
|
// dst is 2001:db8::1
|
|
oob = []byte{36, 0, 0, 0, 0, 0, 0, 0, 41, 0, 0, 0, 50, 0, 0, 0, 32, 1, 13, 184, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0, 0, 0, 0, 0, 0}
|
|
dst, err = parseUDPSocketDst(oob)
|
|
if err != nil {
|
|
t.Fatalf("error parsing ipv6 oob: %v", err)
|
|
}
|
|
dst6 := dst.To16()
|
|
if dst6 == nil {
|
|
t.Errorf("failed to parse ipv6: %v", dst)
|
|
} else if dst6.String() != "2001:db8::1" {
|
|
t.Errorf("unexpected ipv6: %v", dst4)
|
|
}
|
|
|
|
// dst is 100.100.100.100 but was received on 10.10.10.10
|
|
oob = []byte{28, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 8, 0, 0, 0, 2, 0, 0, 0, 10, 10, 10, 10, 100, 100, 100, 100, 0, 0, 0, 0}
|
|
dst, err = parseUDPSocketDst(oob)
|
|
if err != nil {
|
|
t.Fatalf("error parsing ipv4 oob: %v", err)
|
|
}
|
|
dst4 = dst.To4()
|
|
if dst4 == nil {
|
|
t.Errorf("failed to parse ipv4: %v", dst)
|
|
} else if dst4.String() != "100.100.100.100" {
|
|
t.Errorf("unexpected ipv4: %v", dst4)
|
|
}
|
|
}
|
|
|
|
func TestMarshalUDPSocketSrc(t *testing.T) {
|
|
// src is 100.100.100.100
|
|
exoob := []byte{28, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 8, 0, 0, 0, 0, 0, 0, 0, 100, 100, 100, 100, 0, 0, 0, 0, 0, 0, 0, 0}
|
|
oob := marshalUDPSocketSrc(net.ParseIP("100.100.100.100"))
|
|
if !bytes.Equal(exoob, oob) {
|
|
t.Errorf("expected ipv4 oob:\n%v", exoob)
|
|
t.Errorf("actual ipv4 oob:\n%v", oob)
|
|
}
|
|
|
|
// src is 2001:db8::1
|
|
exoob = []byte{36, 0, 0, 0, 0, 0, 0, 0, 41, 0, 0, 0, 50, 0, 0, 0, 32, 1, 13, 184, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0}
|
|
oob = marshalUDPSocketSrc(net.ParseIP("2001:db8::1"))
|
|
if !bytes.Equal(exoob, oob) {
|
|
t.Errorf("expected ipv6 oob:\n%v", exoob)
|
|
t.Errorf("actual ipv6 oob:\n%v", oob)
|
|
}
|
|
}
|