Compare commits
1 Commits
a0d6671213
...
e816228183
Author | SHA1 | Date |
---|---|---|
|
e816228183 |
|
@ -336,8 +336,6 @@ const (
|
||||||
NF_NAT_RANGE_PERSISTENT = unix.NF_NAT_RANGE_PERSISTENT
|
NF_NAT_RANGE_PERSISTENT = unix.NF_NAT_RANGE_PERSISTENT
|
||||||
// NF_NAT_RANGE_PREFIX defines flag for a prefix masquerade
|
// NF_NAT_RANGE_PREFIX defines flag for a prefix masquerade
|
||||||
NF_NAT_RANGE_PREFIX = unix.NF_NAT_RANGE_NETMAP
|
NF_NAT_RANGE_PREFIX = unix.NF_NAT_RANGE_NETMAP
|
||||||
// NF_NAT_RANGE_PROTO_SPECIFIED defines flag for a specified range
|
|
||||||
NF_NAT_RANGE_PROTO_SPECIFIED = unix.NF_NAT_RANGE_PROTO_SPECIFIED
|
|
||||||
)
|
)
|
||||||
|
|
||||||
func (e *Masq) marshal(fam byte) ([]byte, error) {
|
func (e *Masq) marshal(fam byte) ([]byte, error) {
|
||||||
|
|
|
@ -41,7 +41,6 @@ type NAT struct {
|
||||||
FullyRandom bool
|
FullyRandom bool
|
||||||
Persistent bool
|
Persistent bool
|
||||||
Prefix bool
|
Prefix bool
|
||||||
Specified bool
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// |00048|N-|00001| |len |flags| type|
|
// |00048|N-|00001| |len |flags| type|
|
||||||
|
@ -98,9 +97,6 @@ func (e *NAT) marshalData(fam byte) ([]byte, error) {
|
||||||
if e.Prefix {
|
if e.Prefix {
|
||||||
flags |= NF_NAT_RANGE_PREFIX
|
flags |= NF_NAT_RANGE_PREFIX
|
||||||
}
|
}
|
||||||
if e.Specified {
|
|
||||||
flags |= NF_NAT_RANGE_PROTO_SPECIFIED
|
|
||||||
}
|
|
||||||
if flags != 0 {
|
if flags != 0 {
|
||||||
attrs = append(attrs, netlink.Attribute{Type: unix.NFTA_NAT_FLAGS, Data: binaryutil.BigEndian.PutUint32(flags)})
|
attrs = append(attrs, netlink.Attribute{Type: unix.NFTA_NAT_FLAGS, Data: binaryutil.BigEndian.PutUint32(flags)})
|
||||||
}
|
}
|
||||||
|
@ -134,7 +130,6 @@ func (e *NAT) unmarshal(fam byte, data []byte) error {
|
||||||
e.Random = (flags & NF_NAT_RANGE_PROTO_RANDOM) != 0
|
e.Random = (flags & NF_NAT_RANGE_PROTO_RANDOM) != 0
|
||||||
e.FullyRandom = (flags & NF_NAT_RANGE_PROTO_RANDOM_FULLY) != 0
|
e.FullyRandom = (flags & NF_NAT_RANGE_PROTO_RANDOM_FULLY) != 0
|
||||||
e.Prefix = (flags & NF_NAT_RANGE_PREFIX) != 0
|
e.Prefix = (flags & NF_NAT_RANGE_PREFIX) != 0
|
||||||
e.Specified = (flags & NF_NAT_RANGE_PROTO_SPECIFIED) != 0
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return ad.Err()
|
return ad.Err()
|
||||||
|
|
|
@ -1,66 +0,0 @@
|
||||||
package expr
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/binary"
|
|
||||||
"reflect"
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
"github.com/mdlayher/netlink"
|
|
||||||
"golang.org/x/sys/unix"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestNat(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
tests := []struct {
|
|
||||||
name string
|
|
||||||
nat NAT
|
|
||||||
}{
|
|
||||||
{
|
|
||||||
name: "Unmarshal DNAT specified case",
|
|
||||||
nat: NAT{
|
|
||||||
Type: NATTypeDestNAT,
|
|
||||||
Family: unix.NFPROTO_IPV4,
|
|
||||||
RegAddrMin: 1,
|
|
||||||
RegProtoMin: 2,
|
|
||||||
Specified: true,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "Unmarshal SNAT persistent case",
|
|
||||||
nat: NAT{
|
|
||||||
Type: NATTypeSourceNAT,
|
|
||||||
Family: unix.NFPROTO_IPV4,
|
|
||||||
RegAddrMin: 1,
|
|
||||||
RegProtoMin: 2,
|
|
||||||
Persistent: true,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, tt := range tests {
|
|
||||||
t.Run(tt.name, func(t *testing.T) {
|
|
||||||
nnat := NAT{}
|
|
||||||
data, err := tt.nat.marshal(0 /* don't care in this test */)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("marshal error: %+v", err)
|
|
||||||
|
|
||||||
}
|
|
||||||
ad, err := netlink.NewAttributeDecoder(data)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("NewAttributeDecoder() error: %+v", err)
|
|
||||||
}
|
|
||||||
ad.ByteOrder = binary.BigEndian
|
|
||||||
for ad.Next() {
|
|
||||||
if ad.Type() == unix.NFTA_EXPR_DATA {
|
|
||||||
if err := nnat.unmarshal(0, ad.Bytes()); err != nil {
|
|
||||||
t.Errorf("unmarshal error: %+v", err)
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if !reflect.DeepEqual(tt.nat, nnat) {
|
|
||||||
t.Fatalf("original %+v and recovered %+v Ct structs are different", tt.nat, nnat)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
Loading…
Reference in New Issue