98 lines
2.0 KiB
Go
98 lines
2.0 KiB
Go
package xt
|
|
|
|
import (
|
|
"net"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"golang.org/x/sys/unix"
|
|
)
|
|
|
|
func TestTargetDNAT(t *testing.T) {
|
|
t.Parallel()
|
|
tests := []struct {
|
|
name string
|
|
fam byte
|
|
rev uint32
|
|
info InfoAny
|
|
empty InfoAny
|
|
}{
|
|
{
|
|
name: "un/marshal NatRange IPv4 round-trip",
|
|
fam: unix.NFPROTO_IPV4,
|
|
rev: 0,
|
|
info: &NatRange{
|
|
Flags: 0x1234,
|
|
MinIP: net.ParseIP("12.23.34.45").To4(),
|
|
MaxIP: net.ParseIP("21.32.43.54").To4(),
|
|
MinPort: 0x5678,
|
|
MaxPort: 0xabcd,
|
|
},
|
|
empty: &NatRange{},
|
|
},
|
|
{
|
|
name: "un/marshal NatRange IPv6 round-trip",
|
|
fam: unix.NFPROTO_IPV6,
|
|
rev: 0,
|
|
info: &NatRange{
|
|
Flags: 0x1234,
|
|
MinIP: net.ParseIP("fe80::dead:beef"),
|
|
MaxIP: net.ParseIP("fe80::c001:cafe"),
|
|
MinPort: 0x5678,
|
|
MaxPort: 0xabcd,
|
|
},
|
|
empty: &NatRange{},
|
|
},
|
|
{
|
|
name: "un/marshal NatRange2 IPv4 round-trip",
|
|
fam: unix.NFPROTO_IPV4,
|
|
rev: 0,
|
|
info: &NatRange2{
|
|
NatRange: NatRange{
|
|
Flags: 0x1234,
|
|
MinIP: net.ParseIP("12.23.34.45").To4(),
|
|
MaxIP: net.ParseIP("21.32.43.54").To4(),
|
|
MinPort: 0x5678,
|
|
MaxPort: 0xabcd,
|
|
},
|
|
BasePort: 0xfedc,
|
|
},
|
|
empty: &NatRange2{},
|
|
},
|
|
{
|
|
name: "un/marshal NatRange2 IPv6 round-trip",
|
|
fam: unix.NFPROTO_IPV6,
|
|
rev: 0,
|
|
info: &NatRange2{
|
|
NatRange: NatRange{
|
|
Flags: 0x1234,
|
|
MinIP: net.ParseIP("fe80::dead:beef"),
|
|
MaxIP: net.ParseIP("fe80::c001:cafe"),
|
|
MinPort: 0x5678,
|
|
MaxPort: 0xabcd,
|
|
},
|
|
BasePort: 0xfedc,
|
|
},
|
|
empty: &NatRange2{},
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
data, err := tt.info.marshal(TableFamily(tt.fam), tt.rev)
|
|
if err != nil {
|
|
t.Fatalf("marshal error: %+v", err)
|
|
|
|
}
|
|
var recoveredInfo InfoAny = tt.empty
|
|
err = recoveredInfo.unmarshal(TableFamily(tt.fam), tt.rev, data)
|
|
if err != nil {
|
|
t.Fatalf("unmarshal error: %+v", err)
|
|
}
|
|
if !reflect.DeepEqual(tt.info, recoveredInfo) {
|
|
t.Fatalf("original %+v and recovered %+v are different", tt.info, recoveredInfo)
|
|
}
|
|
})
|
|
}
|
|
}
|