Thus far, we have only had forwarded bytes metrics. Notably, forwarded bytes does not include bytes that were sent by the router itself, e.g. by the webserver or rsync server running on the machine. fixes https://github.com/rtr7/router7/issues/71
1014 lines
26 KiB
Go
1014 lines
26 KiB
Go
// Copyright 2018 Google Inc.
|
||
//
|
||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||
// you may not use this file except in compliance with the License.
|
||
// You may obtain a copy of the License at
|
||
//
|
||
// http://www.apache.org/licenses/LICENSE-2.0
|
||
//
|
||
// Unless required by applicable law or agreed to in writing, software
|
||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
// See the License for the specific language governing permissions and
|
||
// limitations under the License.
|
||
|
||
// Package netconfig implements network configuration (interfaces, addresses, firewall rules, …).
|
||
package netconfig
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"io/ioutil"
|
||
"net"
|
||
"os"
|
||
"path/filepath"
|
||
"regexp"
|
||
"strconv"
|
||
"strings"
|
||
"syscall"
|
||
|
||
"github.com/google/nftables"
|
||
"github.com/google/nftables/binaryutil"
|
||
"github.com/google/nftables/expr"
|
||
"github.com/google/renameio"
|
||
"github.com/vishvananda/netlink"
|
||
"golang.org/x/sys/unix"
|
||
|
||
"github.com/rtr7/router7/internal/dhcp4"
|
||
"github.com/rtr7/router7/internal/dhcp6"
|
||
"github.com/rtr7/router7/internal/notify"
|
||
"github.com/rtr7/router7/internal/teelogger"
|
||
)
|
||
|
||
var log = teelogger.NewConsole()
|
||
|
||
func subnetMaskSize(mask string) (int, error) {
|
||
parts := strings.Split(mask, ".")
|
||
if got, want := len(parts), 4; got != want {
|
||
return 0, fmt.Errorf("unexpected number of parts in subnet mask %q: got %d, want %d", mask, got, want)
|
||
}
|
||
numeric := make([]byte, len(parts))
|
||
for idx, part := range parts {
|
||
i, err := strconv.ParseUint(part, 0, 8)
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
numeric[idx] = byte(i)
|
||
}
|
||
ones, _ := net.IPv4Mask(numeric[0], numeric[1], numeric[2], numeric[3]).Size()
|
||
return ones, nil
|
||
}
|
||
|
||
func applyDhcp4(dir string) error {
|
||
b, err := ioutil.ReadFile(filepath.Join(dir, "dhcp4/wire/lease.json"))
|
||
if err != nil {
|
||
if os.IsNotExist(err) {
|
||
return nil // dhcp4 might not have obtained a lease yet
|
||
}
|
||
return err
|
||
}
|
||
var got dhcp4.Config
|
||
if err := json.Unmarshal(b, &got); err != nil {
|
||
return err
|
||
}
|
||
|
||
const linkName = "uplink0"
|
||
link, err := netlink.LinkByName(linkName)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
if got.SubnetMask == "" {
|
||
return fmt.Errorf("invalid DHCP lease: no subnet mask present")
|
||
}
|
||
|
||
subnetSize, err := subnetMaskSize(got.SubnetMask)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
gotAddr := fmt.Sprintf("%s/%d", got.ClientIP, subnetSize)
|
||
addr, err := netlink.ParseAddr(gotAddr)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
h, err := netlink.NewHandle()
|
||
if err != nil {
|
||
return fmt.Errorf("netlink.NewHandle: %v", err)
|
||
}
|
||
defer h.Delete()
|
||
log.Printf("replacing address %v on %v", addr, linkName)
|
||
if err := h.AddrReplace(link, addr); err != nil {
|
||
return fmt.Errorf("AddrReplace(%v, %v): %v", linkName, addr, err)
|
||
}
|
||
|
||
addrs, err := h.AddrList(link, netlink.FAMILY_V4)
|
||
if err != nil {
|
||
return fmt.Errorf("AddrList(%v): %v", linkName, err)
|
||
}
|
||
for _, addr := range addrs {
|
||
ipnet := addr.IPNet.String() // e.g. "85.195.199.99/25"
|
||
if ipnet == gotAddr {
|
||
continue
|
||
}
|
||
log.Printf("de-configuring old IP address %s from %v", ipnet, linkName)
|
||
if err := h.AddrDel(link, &addr); err != nil {
|
||
return fmt.Errorf("AddrDel(%v, %v): %v", linkName, addr, err)
|
||
}
|
||
}
|
||
|
||
// from include/uapi/linux/rtnetlink.h
|
||
const (
|
||
RTPROT_STATIC = 4
|
||
RTPROT_DHCP = 16
|
||
)
|
||
|
||
if err := h.RouteReplace(&netlink.Route{
|
||
LinkIndex: link.Attrs().Index,
|
||
Dst: &net.IPNet{
|
||
IP: net.ParseIP(got.Router),
|
||
Mask: net.CIDRMask(32, 32),
|
||
},
|
||
Src: net.ParseIP(got.ClientIP),
|
||
Scope: netlink.SCOPE_LINK,
|
||
Protocol: RTPROT_DHCP,
|
||
}); err != nil {
|
||
return fmt.Errorf("RouteReplace(router): %v", err)
|
||
}
|
||
|
||
if err := h.RouteReplace(&netlink.Route{
|
||
LinkIndex: link.Attrs().Index,
|
||
Dst: &net.IPNet{
|
||
IP: net.ParseIP("0.0.0.0"),
|
||
Mask: net.CIDRMask(0, 32),
|
||
},
|
||
Gw: net.ParseIP(got.Router),
|
||
Src: net.ParseIP(got.ClientIP),
|
||
Protocol: RTPROT_DHCP,
|
||
}); err != nil {
|
||
return fmt.Errorf("RouteReplace(default): %v", err)
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
func applyDhcp6(dir string) error {
|
||
b, err := ioutil.ReadFile(filepath.Join(dir, "dhcp6/wire/lease.json"))
|
||
if err != nil {
|
||
if os.IsNotExist(err) {
|
||
return nil // dhcp6 might not have obtained a lease yet
|
||
}
|
||
return err
|
||
}
|
||
var got dhcp6.Config
|
||
if err := json.Unmarshal(b, &got); err != nil {
|
||
return err
|
||
}
|
||
|
||
link, err := netlink.LinkByName("lan0")
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
for _, prefix := range got.Prefixes {
|
||
// pick the first address of the prefix, e.g. address 2a02:168:4a00::1
|
||
// for prefix 2a02:168:4a00::/48
|
||
prefix.IP[len(prefix.IP)-1] = 1
|
||
// Use the first /64 subnet within larger prefixes
|
||
if ones, bits := prefix.Mask.Size(); ones < 64 {
|
||
prefix.Mask = net.CIDRMask(64, bits)
|
||
}
|
||
addr, err := netlink.ParseAddr(prefix.String())
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
if err := netlink.AddrReplace(link, addr); err != nil {
|
||
return fmt.Errorf("AddrReplace(%v): %v", addr, err)
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
type InterfaceDetails struct {
|
||
HardwareAddr string `json:"hardware_addr"` // e.g. dc:9b:9c:ee:72:fd
|
||
SpoofHardwareAddr string `json:"spoof_hardware_addr"` // e.g. dc:9b:9c:ee:72:fd
|
||
Name string `json:"name"` // e.g. uplink0, or lan0
|
||
Addr string `json:"addr"` // e.g. 192.168.42.1/24
|
||
}
|
||
|
||
type BridgeDetails struct {
|
||
Name string `json:"name"` // e.g. br0 or lan0
|
||
InterfaceHardwareAddrs []string `json:"interface_hardware_addrs"`
|
||
}
|
||
|
||
type InterfaceConfig struct {
|
||
Interfaces []InterfaceDetails `json:"interfaces"`
|
||
Bridges []BridgeDetails `json:"bridges"`
|
||
}
|
||
|
||
// Interface returns the InterfaceDetails configured for interface ifname in
|
||
// interfaces.json.
|
||
func Interface(dir, ifname string) (InterfaceDetails, error) {
|
||
fn := filepath.Join(dir, "interfaces.json")
|
||
b, err := ioutil.ReadFile(fn)
|
||
if err != nil {
|
||
return InterfaceDetails{}, err
|
||
}
|
||
var cfg InterfaceConfig
|
||
if err := json.Unmarshal(b, &cfg); err != nil {
|
||
return InterfaceDetails{}, err
|
||
}
|
||
for _, details := range cfg.Interfaces {
|
||
if details.Name != ifname {
|
||
continue
|
||
}
|
||
return details, nil
|
||
}
|
||
return InterfaceDetails{}, fmt.Errorf("%s does not configure interface %q", fn, ifname)
|
||
}
|
||
|
||
// LinkAddress returns the IP address configured for the interface ifname in
|
||
// interfaces.json.
|
||
func LinkAddress(dir, ifname string) (net.IP, error) {
|
||
iface, err := Interface(dir, ifname)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
ip, _, err := net.ParseCIDR(iface.Addr)
|
||
return ip, err
|
||
}
|
||
|
||
func applyBridges(cfg *InterfaceConfig) error {
|
||
for _, bridge := range cfg.Bridges {
|
||
if _, err := netlink.LinkByName(bridge.Name); err != nil {
|
||
log.Printf("creating bridge %s", bridge.Name)
|
||
link := &netlink.Bridge{LinkAttrs: netlink.LinkAttrs{Name: bridge.Name}}
|
||
if err := netlink.LinkAdd(link); err != nil {
|
||
return fmt.Errorf("netlink.LinkAdd: %v", err)
|
||
}
|
||
}
|
||
interfaces := make(map[string]bool)
|
||
for _, hwaddr := range bridge.InterfaceHardwareAddrs {
|
||
interfaces[hwaddr] = true
|
||
}
|
||
bridgeLink, err := netlink.LinkByName(bridge.Name)
|
||
if err != nil {
|
||
return fmt.Errorf("LinkByName(%s): %v", bridge.Name, err)
|
||
}
|
||
|
||
links, err := netlink.LinkList()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
for _, l := range links {
|
||
attr := l.Attrs()
|
||
addr := attr.HardwareAddr.String()
|
||
if addr == "" {
|
||
continue
|
||
}
|
||
if !interfaces[addr] {
|
||
continue
|
||
}
|
||
if attr.Name == bridge.Name {
|
||
// Don’t try to add the bridge to itself: the bridge will take
|
||
// the MAC address of the first interface.
|
||
continue
|
||
}
|
||
log.Printf("adding interface %s to bridge %s", attr.Name, bridge.Name)
|
||
if err := netlink.LinkSetMaster(l, bridgeLink); err != nil {
|
||
return fmt.Errorf("LinkSetMaster(%s): %v", attr.Name, err)
|
||
}
|
||
if attr.OperState != netlink.OperUp {
|
||
log.Printf("setting interface %s up", attr.Name)
|
||
if err := netlink.LinkSetUp(l); err != nil {
|
||
return fmt.Errorf("LinkSetUp(%s): %v", attr.Name, err)
|
||
}
|
||
}
|
||
|
||
}
|
||
if attr := bridgeLink.Attrs(); attr.OperState != netlink.OperUp {
|
||
log.Printf("setting interface %s up", attr.Name)
|
||
if err := netlink.LinkSetUp(bridgeLink); err != nil {
|
||
return fmt.Errorf("LinkSetUp(%s): %v", attr.Name, err)
|
||
}
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func applyInterfaces(dir, root string) error {
|
||
b, err := ioutil.ReadFile(filepath.Join(dir, "interfaces.json"))
|
||
if err != nil {
|
||
if os.IsNotExist(err) {
|
||
return nil
|
||
}
|
||
return err
|
||
}
|
||
var cfg InterfaceConfig
|
||
if err := json.Unmarshal(b, &cfg); err != nil {
|
||
return err
|
||
}
|
||
byName := make(map[string]InterfaceDetails)
|
||
byHardwareAddr := make(map[string]InterfaceDetails)
|
||
for _, details := range cfg.Interfaces {
|
||
byHardwareAddr[details.HardwareAddr] = details
|
||
if spoof := details.SpoofHardwareAddr; spoof != "" {
|
||
byHardwareAddr[spoof] = details
|
||
}
|
||
byName[details.Name] = details
|
||
}
|
||
|
||
if err := applyBridges(&cfg); err != nil {
|
||
log.Printf("applyBridges: %v", err)
|
||
}
|
||
|
||
links, err := netlink.LinkList()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
for _, l := range links {
|
||
attr := l.Attrs()
|
||
// TODO: prefix log line with details about the interface.
|
||
// link &{LinkAttrs:{Index:2 MTU:1500 TxQLen:1000 Name:eth0 HardwareAddr:00:0d:b9:49:70:18 Flags:broadcast|multicast RawFlags:4098 ParentIndex:0 MasterIndex:0 Namespace:<nil> Alias: Statistics:0xc4200f45f8 Promisc:0 Xdp:0xc4200ca180 EncapType:ether Protinfo:<nil> OperState:down NetNsID:0 NumTxQueues:0 NumRxQueues:0 Vfs:[]}}, attr &{Index:2 MTU:1500 TxQLen:1000 Name:eth0 HardwareAddr:00:0d:b9:49:70:18 Flags:broadcast|multicast RawFlags:4098 ParentIndex:0 MasterIndex:0 Namespace:<nil> Alias: Statistics:0xc4200f45f8 Promisc:0 Xdp:0xc4200ca180 EncapType:ether Protinfo:<nil> OperState:down NetNsID:0 NumTxQueues:0 NumRxQueues:0 Vfs:[]}
|
||
|
||
var (
|
||
details InterfaceDetails
|
||
ok bool
|
||
)
|
||
addr := attr.HardwareAddr.String()
|
||
if addr == "" {
|
||
details, ok = byName[attr.Name]
|
||
if !ok {
|
||
continue // not a configurable interface (e.g. sit0)
|
||
}
|
||
} else {
|
||
details, ok = byHardwareAddr[addr]
|
||
if !ok {
|
||
details, ok = byName[attr.Name]
|
||
}
|
||
}
|
||
if !ok {
|
||
log.Printf("no config for interface %s/%s", attr.Name, addr)
|
||
continue
|
||
}
|
||
log.Printf("apply details %+v", details)
|
||
if attr.Name != details.Name {
|
||
if err := netlink.LinkSetName(l, details.Name); err != nil {
|
||
return fmt.Errorf("LinkSetName(%q): %v", details.Name, err)
|
||
}
|
||
attr.Name = details.Name
|
||
}
|
||
|
||
if spoof := details.SpoofHardwareAddr; spoof != "" {
|
||
hwaddr, err := net.ParseMAC(spoof)
|
||
if err != nil {
|
||
return fmt.Errorf("ParseMAC(%q): %v", spoof, err)
|
||
}
|
||
if err := netlink.LinkSetHardwareAddr(l, hwaddr); err != nil {
|
||
return fmt.Errorf("LinkSetHardwareAddr(%v): %v", hwaddr, err)
|
||
}
|
||
}
|
||
|
||
if attr.OperState != netlink.OperUp {
|
||
// Set the interface to up, which is required by all other configuration.
|
||
if err := netlink.LinkSetUp(l); err != nil {
|
||
return fmt.Errorf("LinkSetUp(%s): %v", attr.Name, err)
|
||
}
|
||
}
|
||
|
||
if details.Addr != "" {
|
||
addr, err := netlink.ParseAddr(details.Addr)
|
||
if err != nil {
|
||
return fmt.Errorf("ParseAddr(%q): %v", details.Addr, err)
|
||
}
|
||
|
||
if err := netlink.AddrReplace(l, addr); err != nil {
|
||
return fmt.Errorf("AddrReplace(%s, %v): %v", attr.Name, addr, err)
|
||
}
|
||
|
||
if details.Name == "lan0" {
|
||
b := []byte("nameserver " + addr.IP.String() + "\n")
|
||
fn := filepath.Join(root, "tmp", "resolv.conf")
|
||
if err := os.Remove(fn); err != nil && !os.IsNotExist(err) {
|
||
return err
|
||
}
|
||
if err := renameio.WriteFile(fn, b, 0644); err != nil {
|
||
return err
|
||
}
|
||
}
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func nfifname(n string) []byte {
|
||
b := make([]byte, 16)
|
||
copy(b, []byte(n+"\x00"))
|
||
return b
|
||
}
|
||
|
||
// matchUplinkIP is conceptually equivalent to "ip daddr <uplink0-ip>", but
|
||
// without actually using the IP address of the uplink0 interface (which would
|
||
// mean that rules need to change when the IP address changes).
|
||
//
|
||
// Instead, it uses “fib daddr type local” to match all locally-configured IP
|
||
// addresses and then excludes the loopback and LAN IP addresses.
|
||
func matchUplinkIP() []expr.Any {
|
||
return []expr.Any{
|
||
// [ payload load 4b @ network header + 16 => reg 1 ]
|
||
&expr.Payload{
|
||
DestRegister: 1,
|
||
Base: expr.PayloadBaseNetworkHeader,
|
||
Offset: 16, // TODO
|
||
Len: 4, // TODO
|
||
},
|
||
// [ bitwise reg 1 = (reg=1 & 0x000000ff ) ^ 0x00000000 ]
|
||
&expr.Bitwise{
|
||
DestRegister: 1,
|
||
SourceRegister: 1,
|
||
Len: 4,
|
||
Mask: []byte{0xff, 0x00, 0x00, 0x00}, // 255.0.0.0, i.e. /8
|
||
Xor: []byte{0x00, 0x00, 0x00, 0x00},
|
||
},
|
||
// [ cmp neq reg 1 0x0000007f ]
|
||
&expr.Cmp{
|
||
Op: expr.CmpOpNeq,
|
||
Register: 1,
|
||
Data: []byte{0x7f, 0x00, 0x00, 0x00},
|
||
},
|
||
|
||
// [ payload load 4b @ network header + 16 => reg 1 ]
|
||
&expr.Payload{
|
||
DestRegister: 1,
|
||
Base: expr.PayloadBaseNetworkHeader,
|
||
Offset: 16, // TODO
|
||
Len: 4, // TODO
|
||
},
|
||
// [ bitwise reg 1 = (reg=1 & 0x00ffffff ) ^ 0x00000000 ]
|
||
&expr.Bitwise{
|
||
DestRegister: 1,
|
||
SourceRegister: 1,
|
||
Len: 4,
|
||
Mask: []byte{0xff, 0xff, 0xff, 0x00}, // 255.255.255.0, i.e. /24
|
||
Xor: []byte{0x00, 0x00, 0x00, 0x00},
|
||
},
|
||
// [ cmp neq reg 1 0x0000000a ]
|
||
&expr.Cmp{
|
||
Op: expr.CmpOpNeq,
|
||
Register: 1,
|
||
Data: []byte{0x0a, 0x00, 0x00, 0x00},
|
||
},
|
||
|
||
// [ fib daddr type => reg 1 ]
|
||
&expr.Fib{
|
||
Register: 1,
|
||
FlagDADDR: true,
|
||
ResultADDRTYPE: true,
|
||
},
|
||
// [ cmp eq reg 1 0x00000002 ]
|
||
&expr.Cmp{
|
||
Op: expr.CmpOpEq,
|
||
Register: 1,
|
||
Data: []byte{0x02, 0x00, 0x00, 0x00},
|
||
},
|
||
}
|
||
}
|
||
|
||
func portForwardExpr(ifname string, proto uint8, portMin, portMax uint16, dest net.IP, dportMin, dportMax uint16) []expr.Any {
|
||
var cmp []expr.Any
|
||
if portMin == portMax {
|
||
cmp = []expr.Any{
|
||
// [ cmp eq reg 1 0x0000e60f ]
|
||
&expr.Cmp{
|
||
Op: expr.CmpOpEq,
|
||
Register: 1,
|
||
Data: binaryutil.BigEndian.PutUint16(portMin),
|
||
},
|
||
}
|
||
} else {
|
||
cmp = []expr.Any{
|
||
// [ cmp gte reg 1 0x0000e60f ]
|
||
&expr.Cmp{
|
||
Op: expr.CmpOpGte,
|
||
Register: 1,
|
||
Data: binaryutil.BigEndian.PutUint16(portMin),
|
||
},
|
||
// [ cmp lte reg 1 0x0000fa0f ]
|
||
&expr.Cmp{
|
||
Op: expr.CmpOpLte,
|
||
Register: 1,
|
||
Data: binaryutil.BigEndian.PutUint16(portMax),
|
||
},
|
||
}
|
||
}
|
||
ex := append(matchUplinkIP(),
|
||
// [ meta load l4proto => reg 1 ]
|
||
&expr.Meta{Key: expr.MetaKeyL4PROTO, Register: 1},
|
||
// [ cmp eq reg 1 0x00000006 ]
|
||
&expr.Cmp{
|
||
Op: expr.CmpOpEq,
|
||
Register: 1,
|
||
Data: []byte{proto},
|
||
},
|
||
|
||
// [ payload load 2b @ transport header + 2 => reg 1 ]
|
||
&expr.Payload{
|
||
DestRegister: 1,
|
||
Base: expr.PayloadBaseTransportHeader,
|
||
Offset: 2, // TODO
|
||
Len: 2, // TODO
|
||
})
|
||
ex = append(ex, cmp...)
|
||
ex = append(ex,
|
||
// [ immediate reg 1 0x0217a8c0 ]
|
||
&expr.Immediate{
|
||
Register: 1,
|
||
Data: dest.To4(),
|
||
},
|
||
)
|
||
if dportMin == dportMax {
|
||
ex = append(ex,
|
||
// [ immediate reg 2 0x0000f00f ]
|
||
&expr.Immediate{
|
||
Register: 2,
|
||
Data: binaryutil.BigEndian.PutUint16(dportMin),
|
||
},
|
||
// [ nat dnat ip addr_min reg 1 addr_max reg 0 proto_min reg 2 proto_max reg 0 ]
|
||
&expr.NAT{
|
||
Type: expr.NATTypeDestNAT,
|
||
Family: unix.NFPROTO_IPV4,
|
||
RegAddrMin: 1,
|
||
RegProtoMin: 2,
|
||
},
|
||
)
|
||
} else {
|
||
ex = append(ex,
|
||
// [ immediate reg 2 0x0000e60f ]
|
||
&expr.Immediate{
|
||
Register: 2,
|
||
Data: binaryutil.BigEndian.PutUint16(dportMin),
|
||
},
|
||
// [ immediate reg 3 0x0000fa0f ]
|
||
&expr.Immediate{
|
||
Register: 3,
|
||
Data: binaryutil.BigEndian.PutUint16(dportMax),
|
||
},
|
||
// [ nat dnat ip addr_min reg 1 addr_max reg 0 proto_min reg 2 proto_max reg 3 ]
|
||
&expr.NAT{
|
||
Type: expr.NATTypeDestNAT,
|
||
Family: unix.NFPROTO_IPV4,
|
||
RegAddrMin: 1,
|
||
RegProtoMin: 2,
|
||
RegProtoMax: 3,
|
||
},
|
||
)
|
||
}
|
||
return ex
|
||
}
|
||
|
||
type portForwarding struct {
|
||
Proto string `json:"proto"` // e.g. “tcp” (or “tcp,udp”)
|
||
Port string `json:"port"` // e.g. “8080” (or “8080-8090”)
|
||
DestAddr string `json:"dest_addr"` // e.g. “192.168.42.2”
|
||
DestPort string `json:"dest_port"` // e.g. “80” (or “80-90”)
|
||
}
|
||
|
||
type portForwardings struct {
|
||
Forwardings []portForwarding `json:"forwardings"`
|
||
}
|
||
|
||
var rangeRe = regexp.MustCompile(`^([0-9]+)(?:-([0-9]+))?$`)
|
||
|
||
func parsePort(p string) (min uint16, max uint16, _ error) {
|
||
matches := rangeRe.FindStringSubmatch(p)
|
||
if len(matches) == 0 {
|
||
return 0, 0, fmt.Errorf("malformed port %q, expected port number (e.g. 8080) or port range (e.g. 8080-8090)", p)
|
||
}
|
||
min64, err := strconv.ParseUint(matches[1], 0, 16)
|
||
if err != nil {
|
||
return 0, 0, fmt.Errorf("ParseInt(%q): %v", matches[1], err)
|
||
}
|
||
max64 := min64
|
||
if matches[2] != "" {
|
||
max64, err = strconv.ParseUint(matches[2], 0, 16)
|
||
if err != nil {
|
||
return 0, 0, fmt.Errorf("ParseInt(%q): %v", matches[2], err)
|
||
}
|
||
}
|
||
return uint16(min64), uint16(max64), nil
|
||
}
|
||
|
||
func applyPortForwardings(dir, ifname string, c *nftables.Conn, nat *nftables.Table, prerouting *nftables.Chain) error {
|
||
b, err := ioutil.ReadFile(filepath.Join(dir, "portforwardings.json"))
|
||
if err != nil {
|
||
if os.IsNotExist(err) {
|
||
return nil
|
||
}
|
||
return err
|
||
}
|
||
var cfg portForwardings
|
||
if err := json.Unmarshal(b, &cfg); err != nil {
|
||
return err
|
||
}
|
||
|
||
for _, fw := range cfg.Forwardings {
|
||
for _, proto := range strings.Split(fw.Proto, ",") {
|
||
var p uint8
|
||
switch proto {
|
||
case "", "tcp":
|
||
p = unix.IPPROTO_TCP
|
||
case "udp":
|
||
p = unix.IPPROTO_UDP
|
||
default:
|
||
return fmt.Errorf(`unknown proto %q, expected "tcp" or "udp"`, proto)
|
||
}
|
||
|
||
min, max, err := parsePort(fw.Port)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
dmin, dmax, err := parsePort(fw.DestPort)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
c.AddRule(&nftables.Rule{
|
||
Table: nat,
|
||
Chain: prerouting,
|
||
Exprs: portForwardExpr(ifname, p, min, max, net.ParseIP(fw.DestAddr), dmin, dmax),
|
||
})
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// DefaultCounterObj is overridden while testing
|
||
var DefaultCounterObj = &nftables.CounterObj{}
|
||
|
||
func getCounterObj(c *nftables.Conn, o *nftables.CounterObj) *nftables.CounterObj {
|
||
obj, err := c.GetObject(o)
|
||
if err != nil {
|
||
o.Bytes = DefaultCounterObj.Bytes
|
||
o.Packets = DefaultCounterObj.Packets
|
||
return o
|
||
}
|
||
if co, ok := obj.(*nftables.CounterObj); ok {
|
||
return co
|
||
}
|
||
o.Bytes = DefaultCounterObj.Bytes
|
||
o.Packets = DefaultCounterObj.Packets
|
||
return o
|
||
}
|
||
|
||
func hairpinDNAT() []expr.Any {
|
||
return []expr.Any{
|
||
// [ meta load oifname => reg 1 ]
|
||
&expr.Meta{Key: expr.MetaKeyIIFNAME, Register: 1},
|
||
// [ cmp eq reg 1 0x306e616c 0x00000000 0x00000000 0x00000000 ]
|
||
&expr.Cmp{
|
||
Op: expr.CmpOpEq,
|
||
Register: 1,
|
||
Data: nfifname("lan0"),
|
||
},
|
||
|
||
// [ meta load oifname => reg 1 ]
|
||
&expr.Meta{Key: expr.MetaKeyOIFNAME, Register: 1},
|
||
// [ cmp eq reg 1 0x306e616c 0x00000000 0x00000000 0x00000000 ]
|
||
&expr.Cmp{
|
||
Op: expr.CmpOpEq,
|
||
Register: 1,
|
||
Data: nfifname("lan0"),
|
||
},
|
||
|
||
// [ ct load status => reg 1 ]
|
||
&expr.Ct{
|
||
Register: 1,
|
||
SourceRegister: false,
|
||
Key: expr.CtKeySTATUS,
|
||
},
|
||
// [ bitwise reg 1 = (reg=1 & 0x00000020 ) ^ 0x00000000 ]
|
||
&expr.Bitwise{
|
||
DestRegister: 1,
|
||
SourceRegister: 1,
|
||
Len: 4,
|
||
Mask: []byte{0x20, 0x00, 0x00, 0x00},
|
||
Xor: []byte{0x00, 0x00, 0x00, 0x00},
|
||
},
|
||
|
||
// [ cmp neq reg 1 0x00000000 ]
|
||
&expr.Cmp{
|
||
Op: expr.CmpOpNeq,
|
||
Register: 1,
|
||
Data: []byte{0x00, 0x00, 0x00, 0x00},
|
||
},
|
||
// [ masq ]
|
||
&expr.Masq{},
|
||
}
|
||
}
|
||
|
||
func applyFirewall(dir, ifname string) error {
|
||
c := &nftables.Conn{}
|
||
|
||
c.FlushRuleset()
|
||
|
||
nat := c.AddTable(&nftables.Table{
|
||
Family: nftables.TableFamilyIPv4,
|
||
Name: "nat",
|
||
})
|
||
|
||
prerouting := c.AddChain(&nftables.Chain{
|
||
Name: "prerouting",
|
||
Hooknum: nftables.ChainHookPrerouting,
|
||
Priority: nftables.ChainPriorityFilter,
|
||
Table: nat,
|
||
Type: nftables.ChainTypeNAT,
|
||
})
|
||
|
||
postrouting := c.AddChain(&nftables.Chain{
|
||
Name: "postrouting",
|
||
Hooknum: nftables.ChainHookPostrouting,
|
||
Priority: nftables.ChainPriorityNATSource,
|
||
Table: nat,
|
||
Type: nftables.ChainTypeNAT,
|
||
})
|
||
|
||
c.AddRule(&nftables.Rule{
|
||
Table: nat,
|
||
Chain: postrouting,
|
||
Exprs: []expr.Any{
|
||
// meta load oifname => reg 1
|
||
&expr.Meta{Key: expr.MetaKeyOIFNAME, Register: 1},
|
||
// cmp eq reg 1 0x696c7075 0x00306b6e 0x00000000 0x00000000
|
||
&expr.Cmp{
|
||
Op: expr.CmpOpEq,
|
||
Register: 1,
|
||
Data: nfifname(ifname),
|
||
},
|
||
// masq
|
||
&expr.Masq{},
|
||
},
|
||
})
|
||
|
||
c.AddRule(&nftables.Rule{
|
||
Table: nat,
|
||
Chain: postrouting,
|
||
Exprs: hairpinDNAT(),
|
||
})
|
||
|
||
if err := applyPortForwardings(dir, ifname, c, nat, prerouting); err != nil {
|
||
return err
|
||
}
|
||
|
||
filter4 := c.AddTable(&nftables.Table{
|
||
Family: nftables.TableFamilyIPv4,
|
||
Name: "filter",
|
||
})
|
||
|
||
filter6 := c.AddTable(&nftables.Table{
|
||
Family: nftables.TableFamilyIPv6,
|
||
Name: "filter",
|
||
})
|
||
|
||
for _, filter := range []*nftables.Table{filter4, filter6} {
|
||
forward := c.AddChain(&nftables.Chain{
|
||
Name: "forward",
|
||
Hooknum: nftables.ChainHookForward,
|
||
Priority: nftables.ChainPriorityFilter,
|
||
Table: filter,
|
||
Type: nftables.ChainTypeFilter,
|
||
})
|
||
|
||
c.AddRule(&nftables.Rule{
|
||
Table: filter,
|
||
Chain: forward,
|
||
Exprs: []expr.Any{
|
||
// [ meta load oifname => reg 1 ]
|
||
&expr.Meta{Key: expr.MetaKeyOIFNAME, Register: 1},
|
||
// [ cmp eq reg 1 0x30707070 0x00000000 0x00000000 0x00000000 ]
|
||
&expr.Cmp{
|
||
Op: expr.CmpOpEq,
|
||
Register: 1,
|
||
Data: nfifname(ifname),
|
||
},
|
||
|
||
// [ meta load l4proto => reg 1 ]
|
||
&expr.Meta{Key: expr.MetaKeyL4PROTO, Register: 1},
|
||
// [ cmp eq reg 1 0x00000006 ]
|
||
&expr.Cmp{
|
||
Op: expr.CmpOpEq,
|
||
Register: 1,
|
||
Data: []byte{unix.IPPROTO_TCP},
|
||
},
|
||
|
||
// [ payload load 1b @ transport header + 13 => reg 1 ]
|
||
&expr.Payload{
|
||
DestRegister: 1,
|
||
Base: expr.PayloadBaseTransportHeader,
|
||
Offset: 13, // TODO
|
||
Len: 1, // TODO
|
||
},
|
||
// [ bitwise reg 1 = (reg=1 & 0x00000002 ) ^ 0x00000000 ]
|
||
&expr.Bitwise{
|
||
DestRegister: 1,
|
||
SourceRegister: 1,
|
||
Len: 1,
|
||
Mask: []byte{0x02},
|
||
Xor: []byte{0x00},
|
||
},
|
||
// [ cmp neq reg 1 0x00000000 ]
|
||
&expr.Cmp{
|
||
Op: expr.CmpOpNeq,
|
||
Register: 1,
|
||
Data: []byte{0x00},
|
||
},
|
||
|
||
// [ rt load tcpmss => reg 1 ]
|
||
&expr.Rt{
|
||
Register: 1,
|
||
Key: expr.RtTCPMSS,
|
||
},
|
||
// [ byteorder reg 1 = hton(reg 1, 2, 2) ]
|
||
&expr.Byteorder{
|
||
DestRegister: 1,
|
||
SourceRegister: 1,
|
||
Op: expr.ByteorderHton,
|
||
Len: 2,
|
||
Size: 2,
|
||
},
|
||
// [ exthdr write tcpopt reg 1 => 2b @ 2 + 2 ]
|
||
&expr.Exthdr{
|
||
SourceRegister: 1,
|
||
Type: 2, // TODO
|
||
Offset: 2,
|
||
Len: 2,
|
||
Op: expr.ExthdrOpTcpopt,
|
||
},
|
||
},
|
||
})
|
||
|
||
counterObj := getCounterObj(c, &nftables.CounterObj{
|
||
Table: filter,
|
||
Name: "fwded",
|
||
})
|
||
counter := c.AddObj(counterObj).(*nftables.CounterObj)
|
||
|
||
const NFT_OBJECT_COUNTER = 1 // TODO: get into x/sys/unix
|
||
c.AddRule(&nftables.Rule{
|
||
Table: filter,
|
||
Chain: forward,
|
||
Exprs: []expr.Any{
|
||
// [ counter name fwded ]
|
||
&expr.Objref{
|
||
Type: NFT_OBJECT_COUNTER,
|
||
Name: counter.Name,
|
||
},
|
||
},
|
||
})
|
||
|
||
input := c.AddChain(&nftables.Chain{
|
||
Name: "input",
|
||
Hooknum: nftables.ChainHookInput,
|
||
Priority: nftables.ChainPriorityFilter,
|
||
Table: filter,
|
||
Type: nftables.ChainTypeFilter,
|
||
})
|
||
|
||
counterObj = getCounterObj(c, &nftables.CounterObj{
|
||
Table: filter,
|
||
Name: "inputc",
|
||
})
|
||
counter = c.AddObj(counterObj).(*nftables.CounterObj)
|
||
c.AddRule(&nftables.Rule{
|
||
Table: filter,
|
||
Chain: input,
|
||
Exprs: []expr.Any{
|
||
// [ counter name input ]
|
||
&expr.Objref{
|
||
Type: NFT_OBJECT_COUNTER,
|
||
Name: counter.Name,
|
||
},
|
||
},
|
||
})
|
||
|
||
output := c.AddChain(&nftables.Chain{
|
||
Name: "output",
|
||
Hooknum: nftables.ChainHookOutput,
|
||
Priority: nftables.ChainPriorityFilter,
|
||
Table: filter,
|
||
Type: nftables.ChainTypeFilter,
|
||
})
|
||
|
||
counterObj = getCounterObj(c, &nftables.CounterObj{
|
||
Table: filter,
|
||
Name: "outputc",
|
||
})
|
||
counter = c.AddObj(counterObj).(*nftables.CounterObj)
|
||
c.AddRule(&nftables.Rule{
|
||
Table: filter,
|
||
Chain: output,
|
||
Exprs: []expr.Any{
|
||
// [ counter name output ]
|
||
&expr.Objref{
|
||
Type: NFT_OBJECT_COUNTER,
|
||
Name: counter.Name,
|
||
},
|
||
},
|
||
})
|
||
}
|
||
|
||
return c.Flush()
|
||
}
|
||
|
||
func uplinkInterface() (string, error) {
|
||
names := []string{
|
||
"uplink0", // router7
|
||
"eth0", // gokrazy
|
||
"ens3", // distri
|
||
}
|
||
for _, ifname := range names {
|
||
if _, err := net.InterfaceByName(ifname); err != nil {
|
||
continue
|
||
}
|
||
return ifname, nil
|
||
}
|
||
return "", fmt.Errorf("no uplink ethernet interface found (checked %v)", names)
|
||
}
|
||
|
||
func applySysctl(ifname string) error {
|
||
sysctls := []string{
|
||
"net.ipv4.ip_forward=1",
|
||
"net.ipv6.conf.all.forwarding=1",
|
||
}
|
||
if ifname != "" {
|
||
sysctls = append(sysctls, "net.ipv6.conf."+ifname+".accept_ra=2")
|
||
}
|
||
for _, ctl := range sysctls {
|
||
idx := strings.Index(ctl, "=")
|
||
key, val := ctl[:idx], ctl[idx+1:]
|
||
fn := strings.Replace(key, ".", "/", -1)
|
||
if err := ioutil.WriteFile("/proc/sys/"+fn, []byte(val), 0644); err != nil {
|
||
return fmt.Errorf("sysctl(%v=%v): %v", key, val, err)
|
||
}
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
func Apply(dir, root string) error {
|
||
|
||
// TODO: split into two parts: delay the up until later
|
||
if err := applyInterfaces(dir, root); err != nil {
|
||
return fmt.Errorf("interfaces: %v", err)
|
||
}
|
||
|
||
var errors []error
|
||
appendError := func(err error) {
|
||
errors = append(errors, err)
|
||
log.Println(err)
|
||
}
|
||
|
||
if err := applyDhcp4(dir); err != nil {
|
||
appendError(fmt.Errorf("dhcp4: %v", err))
|
||
}
|
||
|
||
if err := applyDhcp6(dir); err != nil {
|
||
appendError(fmt.Errorf("dhcp6: %v", err))
|
||
}
|
||
|
||
for _, process := range []string{
|
||
"dyndns", // depends on the public IPv4 address
|
||
"dnsd", // listens on private IPv4/IPv6
|
||
"diagd", // listens on private IPv4/IPv6
|
||
"backupd", // listens on private IPv4/IPv6
|
||
"captured", // listens on private IPv4/IPv6
|
||
} {
|
||
if err := notify.Process("/user/"+process, syscall.SIGUSR1); err != nil {
|
||
log.Printf("notifying %s: %v", process, err)
|
||
}
|
||
}
|
||
|
||
ifname, err := uplinkInterface()
|
||
if err != nil {
|
||
log.Printf("uplinkInterface: %v", err)
|
||
}
|
||
|
||
if err := applySysctl(ifname); err != nil {
|
||
appendError(fmt.Errorf("sysctl: %v", err))
|
||
}
|
||
|
||
if err := applyFirewall(dir, ifname); err != nil {
|
||
appendError(fmt.Errorf("firewall: %v", err))
|
||
}
|
||
|
||
if err := applyWireGuard(dir); err != nil {
|
||
appendError(fmt.Errorf("wireguard: %v", err))
|
||
}
|
||
|
||
if len(errors) > 0 {
|
||
return fmt.Errorf("%v", errors)
|
||
}
|
||
return nil
|
||
}
|