Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
/.idea/
.DS_Store
12 changes: 7 additions & 5 deletions alignedbuff/alignedbuff.go
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ import (
"fmt"
"unsafe"

"github.com/google/nftables/binaryutil"
"github.com/sagernet/nftables/binaryutil"
)

// ErrEOF signals trying to read beyond the available payload information.
Expand Down Expand Up @@ -289,10 +289,12 @@ func (a *AlignedBuff) alignWrite(m int) {
}

// This is ... ugly.
var uint16AlignMask = int(unsafe.Alignof(uint16(0)) - 1)
var uint32AlignMask = int(unsafe.Alignof(uint32(0)) - 1)
var uint64AlignMask = int(unsafe.Alignof(uint64(0)) - 1)
var padding = bytes.Repeat([]byte{0}, uint64AlignMask)
var (
uint16AlignMask = int(unsafe.Alignof(uint16(0)) - 1)
uint32AlignMask = int(unsafe.Alignof(uint32(0)) - 1)
uint64AlignMask = int(unsafe.Alignof(uint64(0)) - 1)
padding = bytes.Repeat([]byte{0}, uint64AlignMask)
)

var int32AlignMask = int(unsafe.Alignof(int32(0)) - 1)

Expand Down
2 changes: 1 addition & 1 deletion chain.go
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ import (
"fmt"
"math"

"github.com/google/nftables/binaryutil"
"github.com/sagernet/nftables/binaryutil"
"github.com/mdlayher/netlink"
"golang.org/x/sys/unix"
)
Expand Down
8 changes: 5 additions & 3 deletions compat_policy.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,12 +3,14 @@ package nftables
import (
"fmt"

"github.com/google/nftables/expr"
"github.com/sagernet/nftables/expr"
"golang.org/x/sys/unix"
)

const nft_RULE_COMPAT_F_INV uint32 = (1 << 1)
const nft_RULE_COMPAT_F_MASK uint32 = nft_RULE_COMPAT_F_INV
const (
nft_RULE_COMPAT_F_INV uint32 = (1 << 1)
nft_RULE_COMPAT_F_MASK uint32 = nft_RULE_COMPAT_F_INV
)

// Used by xt match or target like xt_tcpudp to set compat policy between xtables and nftables
// https://elixir.bootlin.com/linux/v5.12/source/net/netfilter/nft_compat.c#L187
Expand Down
4 changes: 2 additions & 2 deletions compat_policy_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,8 @@ package nftables
import (
"testing"

"github.com/google/nftables/expr"
"github.com/google/nftables/xt"
"github.com/sagernet/nftables/expr"
"github.com/sagernet/nftables/xt"
"golang.org/x/sys/unix"
)

Expand Down
7 changes: 4 additions & 3 deletions conn.go
Original file line number Diff line number Diff line change
Expand Up @@ -20,10 +20,11 @@ import (
"os"
"sync"

"github.com/google/nftables/binaryutil"
"github.com/google/nftables/expr"
"github.com/sagernet/nftables/binaryutil"
"github.com/sagernet/nftables/expr"
"github.com/mdlayher/netlink"
"github.com/mdlayher/netlink/nltest"
E "github.com/sagernet/sing/common/exceptions"
"golang.org/x/sys/unix"
)

Expand Down Expand Up @@ -258,7 +259,7 @@ func (cc *Conn) Flush() error {
// Kernel will only send one permission error to user space.
return err
}
errs = errors.Join(errs, err)
errs = E.Errors(errs, err)
}
}

Expand Down
2 changes: 1 addition & 1 deletion counter.go
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
package nftables

import (
"github.com/google/nftables/binaryutil"
"github.com/sagernet/nftables/binaryutil"
"github.com/mdlayher/netlink"
"golang.org/x/sys/unix"
)
Expand Down
2 changes: 1 addition & 1 deletion expr/bitwise.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ package expr
import (
"encoding/binary"

"github.com/google/nftables/binaryutil"
"github.com/sagernet/nftables/binaryutil"
"github.com/mdlayher/netlink"
"golang.org/x/sys/unix"
)
Expand Down
1 change: 0 additions & 1 deletion expr/bitwise_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,6 @@ func TestBitwise(t *testing.T) {
data, err := tt.bw.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 {
Expand Down
2 changes: 1 addition & 1 deletion expr/byteorder.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ package expr
import (
"fmt"

"github.com/google/nftables/binaryutil"
"github.com/sagernet/nftables/binaryutil"
"github.com/mdlayher/netlink"
"golang.org/x/sys/unix"
)
Expand Down
2 changes: 1 addition & 1 deletion expr/connlimit.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ package expr
import (
"encoding/binary"

"github.com/google/nftables/binaryutil"
"github.com/sagernet/nftables/binaryutil"
"github.com/mdlayher/netlink"
"golang.org/x/sys/unix"
)
Expand Down
2 changes: 1 addition & 1 deletion expr/counter.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ package expr
import (
"encoding/binary"

"github.com/google/nftables/binaryutil"
"github.com/sagernet/nftables/binaryutil"
"github.com/mdlayher/netlink"
"golang.org/x/sys/unix"
)
Expand Down
2 changes: 1 addition & 1 deletion expr/ct.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ package expr
import (
"encoding/binary"

"github.com/google/nftables/binaryutil"
"github.com/sagernet/nftables/binaryutil"
"github.com/mdlayher/netlink"
"golang.org/x/sys/unix"
)
Expand Down
3 changes: 1 addition & 2 deletions expr/dup.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ package expr
import (
"encoding/binary"

"github.com/google/nftables/binaryutil"
"github.com/sagernet/nftables/binaryutil"
"github.com/mdlayher/netlink"
"golang.org/x/sys/unix"
)
Expand All @@ -38,7 +38,6 @@ func (e *Dup) marshal(fam byte) ([]byte, error) {
}

data, err := netlink.MarshalAttributes(attrs)

if err != nil {
return nil, err
}
Expand Down
4 changes: 2 additions & 2 deletions expr/dynset.go
Original file line number Diff line number Diff line change
Expand Up @@ -18,8 +18,8 @@ import (
"encoding/binary"
"time"

"github.com/google/nftables/binaryutil"
"github.com/google/nftables/internal/parseexprfunc"
"github.com/sagernet/nftables/binaryutil"
"github.com/sagernet/nftables/internal/parseexprfunc"
"github.com/mdlayher/netlink"
"golang.org/x/sys/unix"
)
Expand Down
13 changes: 8 additions & 5 deletions expr/expr.go
Original file line number Diff line number Diff line change
Expand Up @@ -18,8 +18,8 @@ package expr
import (
"encoding/binary"

"github.com/google/nftables/binaryutil"
"github.com/google/nftables/internal/parseexprfunc"
"github.com/sagernet/nftables/binaryutil"
"github.com/sagernet/nftables/internal/parseexprfunc"
"github.com/mdlayher/netlink"
"golang.org/x/sys/unix"
)
Expand Down Expand Up @@ -305,22 +305,25 @@ func (e *Masq) marshal(fam byte) ([]byte, error) {
}
if flags != 0 {
flagsData, err := netlink.MarshalAttributes([]netlink.Attribute{
{Type: unix.NFTA_MASQ_FLAGS, Data: binaryutil.BigEndian.PutUint32(flags)}})
{Type: unix.NFTA_MASQ_FLAGS, Data: binaryutil.BigEndian.PutUint32(flags)},
})
if err != nil {
return nil, err
}
msgData = append(msgData, flagsData...)
}
} else {
regsData, err := netlink.MarshalAttributes([]netlink.Attribute{
{Type: unix.NFTA_MASQ_REG_PROTO_MIN, Data: binaryutil.BigEndian.PutUint32(e.RegProtoMin)}})
{Type: unix.NFTA_MASQ_REG_PROTO_MIN, Data: binaryutil.BigEndian.PutUint32(e.RegProtoMin)},
})
if err != nil {
return nil, err
}
msgData = append(msgData, regsData...)
if e.RegProtoMax != 0 {
regsData, err := netlink.MarshalAttributes([]netlink.Attribute{
{Type: unix.NFTA_MASQ_REG_PROTO_MAX, Data: binaryutil.BigEndian.PutUint32(e.RegProtoMax)}})
{Type: unix.NFTA_MASQ_REG_PROTO_MAX, Data: binaryutil.BigEndian.PutUint32(e.RegProtoMax)},
})
if err != nil {
return nil, err
}
Expand Down
8 changes: 5 additions & 3 deletions expr/exthdr.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ package expr
import (
"encoding/binary"

"github.com/google/nftables/binaryutil"
"github.com/sagernet/nftables/binaryutil"
"github.com/mdlayher/netlink"
"golang.org/x/sys/unix"
)
Expand Down Expand Up @@ -46,10 +46,12 @@ func (e *Exthdr) marshal(fam byte) ([]byte, error) {
// or DestRegister is set. Mixing them results in EOPNOTSUPP.
if e.SourceRegister != 0 {
attr = []netlink.Attribute{
{Type: unix.NFTA_EXTHDR_SREG, Data: binaryutil.BigEndian.PutUint32(e.SourceRegister)}}
{Type: unix.NFTA_EXTHDR_SREG, Data: binaryutil.BigEndian.PutUint32(e.SourceRegister)},
}
} else {
attr = []netlink.Attribute{
{Type: unix.NFTA_EXTHDR_DREG, Data: binaryutil.BigEndian.PutUint32(e.DestRegister)}}
{Type: unix.NFTA_EXTHDR_DREG, Data: binaryutil.BigEndian.PutUint32(e.DestRegister)},
}
}

attr = append(attr,
Expand Down
1 change: 0 additions & 1 deletion expr/exthdr_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,6 @@ func TestExthdr(t *testing.T) {
data, err := tt.eh.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 {
Expand Down
2 changes: 1 addition & 1 deletion expr/fib.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ package expr
import (
"encoding/binary"

"github.com/google/nftables/binaryutil"
"github.com/sagernet/nftables/binaryutil"
"github.com/mdlayher/netlink"
"golang.org/x/sys/unix"
)
Expand Down
2 changes: 1 addition & 1 deletion expr/hash.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ package expr
import (
"encoding/binary"

"github.com/google/nftables/binaryutil"
"github.com/sagernet/nftables/binaryutil"
"github.com/mdlayher/netlink"
"golang.org/x/sys/unix"
)
Expand Down
2 changes: 1 addition & 1 deletion expr/immediate.go
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ import (
"encoding/binary"
"fmt"

"github.com/google/nftables/binaryutil"
"github.com/sagernet/nftables/binaryutil"
"github.com/mdlayher/netlink"
"golang.org/x/sys/unix"
)
Expand Down
2 changes: 1 addition & 1 deletion expr/limit.go
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ import (
"errors"
"fmt"

"github.com/google/nftables/binaryutil"
"github.com/sagernet/nftables/binaryutil"
"github.com/mdlayher/netlink"
"golang.org/x/sys/unix"
)
Expand Down
2 changes: 1 addition & 1 deletion expr/log.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ package expr
import (
"encoding/binary"

"github.com/google/nftables/binaryutil"
"github.com/sagernet/nftables/binaryutil"
"github.com/mdlayher/netlink"
"golang.org/x/sys/unix"
)
Expand Down
2 changes: 1 addition & 1 deletion expr/lookup.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ package expr
import (
"encoding/binary"

"github.com/google/nftables/binaryutil"
"github.com/sagernet/nftables/binaryutil"
"github.com/mdlayher/netlink"
"golang.org/x/sys/unix"
)
Expand Down
4 changes: 2 additions & 2 deletions expr/match.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,8 @@ import (
"bytes"
"encoding/binary"

"github.com/google/nftables/binaryutil"
"github.com/google/nftables/xt"
"github.com/sagernet/nftables/binaryutil"
"github.com/sagernet/nftables/xt"
"github.com/mdlayher/netlink"
"golang.org/x/sys/unix"
)
Expand Down
3 changes: 1 addition & 2 deletions expr/match_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ import (
"reflect"
"testing"

"github.com/google/nftables/xt"
"github.com/sagernet/nftables/xt"
"github.com/mdlayher/netlink"
"golang.org/x/sys/unix"
)
Expand Down Expand Up @@ -33,7 +33,6 @@ func TestMatch(t *testing.T) {
data, err := tt.mtch.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 {
Expand Down
1 change: 0 additions & 1 deletion expr/meta_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,6 @@ func TestMeta(t *testing.T) {
data, err := tt.meta.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 {
Expand Down
2 changes: 1 addition & 1 deletion expr/nat.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ package expr
import (
"encoding/binary"

"github.com/google/nftables/binaryutil"
"github.com/sagernet/nftables/binaryutil"
"github.com/mdlayher/netlink"
"golang.org/x/sys/unix"
)
Expand Down
1 change: 0 additions & 1 deletion expr/notrack.go
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,6 @@ func (e *Notrack) marshal(fam byte) ([]byte, error) {

func (e *Notrack) unmarshal(fam byte, data []byte) error {
ad, err := netlink.NewAttributeDecoder(data)

if err != nil {
return err
}
Expand Down
2 changes: 1 addition & 1 deletion expr/numgen.go
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ import (
"encoding/binary"
"fmt"

"github.com/google/nftables/binaryutil"
"github.com/sagernet/nftables/binaryutil"
"github.com/mdlayher/netlink"
"golang.org/x/sys/unix"
)
Expand Down
2 changes: 1 addition & 1 deletion expr/objref.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ package expr
import (
"encoding/binary"

"github.com/google/nftables/binaryutil"
"github.com/sagernet/nftables/binaryutil"
"github.com/mdlayher/netlink"
"golang.org/x/sys/unix"
)
Expand Down
Loading