Compare commits

...

3 Commits

Author SHA1 Message Date
Artem Smorodin d4b2d6e2d6
Merge e01db4c5b9 into ef45dd3322 2023-12-31 14:58:25 -07:00
Francesco Cheinasso ef45dd3322
NAT: prefix support (#251) 2023-12-22 08:33:18 +01:00
Artem Smorodin e01db4c5b9 [expr] Conntrack SourceRegister field unmarshalling 2023-05-17 22:35:24 +03:00
3 changed files with 10 additions and 0 deletions

View File

@ -109,6 +109,9 @@ func (e *Ct) unmarshal(fam byte, data []byte) error {
e.Key = CtKey(ad.Uint32())
case unix.NFTA_CT_DREG:
e.Register = ad.Uint32()
case unix.NFTA_CT_SREG:
e.Register = ad.Uint32()
e.SourceRegister = true
}
}
return ad.Err()

View File

@ -287,6 +287,8 @@ const (
NF_NAT_RANGE_PROTO_RANDOM_FULLY = 0x10
// NF_NAT_RANGE_PERSISTENT defines flag for a persistent masquerade
NF_NAT_RANGE_PERSISTENT = 0x8
// NF_NAT_RANGE_PREFIX defines flag for a prefix masquerade
NF_NAT_RANGE_PREFIX = 0x40
)
func (e *Masq) marshal(fam byte) ([]byte, error) {

View File

@ -40,6 +40,7 @@ type NAT struct {
Random bool
FullyRandom bool
Persistent bool
Prefix bool
}
// |00048|N-|00001| |len |flags| type|
@ -82,6 +83,9 @@ func (e *NAT) marshal(fam byte) ([]byte, error) {
if e.Persistent {
flags |= NF_NAT_RANGE_PERSISTENT
}
if e.Prefix {
flags |= NF_NAT_RANGE_PREFIX
}
if flags != 0 {
attrs = append(attrs, netlink.Attribute{Type: unix.NFTA_NAT_FLAGS, Data: binaryutil.BigEndian.PutUint32(flags)})
}
@ -121,6 +125,7 @@ func (e *NAT) unmarshal(fam byte, data []byte) error {
e.Persistent = (flags & NF_NAT_RANGE_PERSISTENT) != 0
e.Random = (flags & NF_NAT_RANGE_PROTO_RANDOM) != 0
e.FullyRandom = (flags & NF_NAT_RANGE_PROTO_RANDOM_FULLY) != 0
e.Prefix = (flags & NF_NAT_RANGE_PREFIX) != 0
}
}
return ad.Err()