initial commit
This commit is contained in:
tumillanino
268
internal/server/network/backend_networkd.go
Normal file
268
internal/server/network/backend_networkd.go
Normal file
@@ -0,0 +1,268 @@
|
||||
package network
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net"
|
||||
"strings"
|
||||
"sync"
|
||||
|
||||
"github.com/AvengeMedia/danklinux/internal/log"
|
||||
"github.com/godbus/dbus/v5"
|
||||
)
|
||||
|
||||
const (
|
||||
networkdBusName = "org.freedesktop.network1"
|
||||
networkdManagerPath = "/org/freedesktop/network1"
|
||||
networkdManagerIface = "org.freedesktop.network1.Manager"
|
||||
networkdLinkIface = "org.freedesktop.network1.Link"
|
||||
)
|
||||
|
||||
type linkInfo struct {
|
||||
ifindex int32
|
||||
name string
|
||||
path dbus.ObjectPath
|
||||
opState string
|
||||
}
|
||||
|
||||
type SystemdNetworkdBackend struct {
|
||||
conn *dbus.Conn
|
||||
managerPath dbus.ObjectPath
|
||||
links map[string]*linkInfo
|
||||
linksMutex sync.RWMutex
|
||||
state *BackendState
|
||||
stateMutex sync.RWMutex
|
||||
onStateChange func()
|
||||
stopChan chan struct{}
|
||||
signals chan *dbus.Signal
|
||||
sigWG sync.WaitGroup
|
||||
}
|
||||
|
||||
func NewSystemdNetworkdBackend() (*SystemdNetworkdBackend, error) {
|
||||
return &SystemdNetworkdBackend{
|
||||
managerPath: networkdManagerPath,
|
||||
links: make(map[string]*linkInfo),
|
||||
state: &BackendState{
|
||||
Backend: "networkd",
|
||||
WiFiNetworks: []WiFiNetwork{},
|
||||
},
|
||||
stopChan: make(chan struct{}),
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (b *SystemdNetworkdBackend) Initialize() error {
|
||||
c, err := dbus.ConnectSystemBus()
|
||||
if err != nil {
|
||||
return fmt.Errorf("connect bus: %w", err)
|
||||
}
|
||||
b.conn = c
|
||||
|
||||
if err := b.enumerateLinks(); err != nil {
|
||||
c.Close()
|
||||
return fmt.Errorf("enumerate links: %w", err)
|
||||
}
|
||||
|
||||
if err := b.updateState(); err != nil {
|
||||
c.Close()
|
||||
return fmt.Errorf("update initial state: %w", err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (b *SystemdNetworkdBackend) Close() {
|
||||
close(b.stopChan)
|
||||
b.StopMonitoring()
|
||||
|
||||
if b.conn != nil {
|
||||
b.conn.Close()
|
||||
}
|
||||
}
|
||||
|
||||
func (b *SystemdNetworkdBackend) enumerateLinks() error {
|
||||
obj := b.conn.Object(networkdBusName, b.managerPath)
|
||||
|
||||
var links []struct {
|
||||
Ifindex int32
|
||||
Name string
|
||||
Path dbus.ObjectPath
|
||||
}
|
||||
err := obj.Call(networkdManagerIface+".ListLinks", 0).Store(&links)
|
||||
if err != nil {
|
||||
return fmt.Errorf("ListLinks: %w", err)
|
||||
}
|
||||
|
||||
b.linksMutex.Lock()
|
||||
defer b.linksMutex.Unlock()
|
||||
|
||||
for _, l := range links {
|
||||
b.links[l.Name] = &linkInfo{
|
||||
ifindex: l.Ifindex,
|
||||
name: l.Name,
|
||||
path: l.Path,
|
||||
}
|
||||
log.Debugf("networkd: enumerated link %s (ifindex=%d, path=%s)", l.Name, l.Ifindex, l.Path)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (b *SystemdNetworkdBackend) updateState() error {
|
||||
b.linksMutex.RLock()
|
||||
defer b.linksMutex.RUnlock()
|
||||
|
||||
var wiredIface *linkInfo
|
||||
var wifiIface *linkInfo
|
||||
|
||||
for name, link := range b.links {
|
||||
if b.isVirtualInterface(name) {
|
||||
continue
|
||||
}
|
||||
|
||||
linkObj := b.conn.Object(networkdBusName, link.path)
|
||||
opStateVar, err := linkObj.GetProperty(networkdLinkIface + ".OperationalState")
|
||||
if err == nil {
|
||||
if opState, ok := opStateVar.Value().(string); ok {
|
||||
link.opState = opState
|
||||
}
|
||||
}
|
||||
|
||||
if strings.HasPrefix(name, "wlan") || strings.HasPrefix(name, "wlp") {
|
||||
if wifiIface == nil || link.opState == "routable" || link.opState == "carrier" {
|
||||
wifiIface = link
|
||||
}
|
||||
} else if !b.isVirtualInterface(name) {
|
||||
if wiredIface == nil || link.opState == "routable" || link.opState == "carrier" {
|
||||
wiredIface = link
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
var wiredConns []WiredConnection
|
||||
for name, link := range b.links {
|
||||
if b.isVirtualInterface(name) || strings.HasPrefix(name, "wlan") || strings.HasPrefix(name, "wlp") {
|
||||
continue
|
||||
}
|
||||
|
||||
active := link.opState == "routable" || link.opState == "carrier"
|
||||
wiredConns = append(wiredConns, WiredConnection{
|
||||
Path: link.path,
|
||||
ID: name,
|
||||
UUID: "wired:" + name,
|
||||
Type: "ethernet",
|
||||
IsActive: active,
|
||||
})
|
||||
}
|
||||
|
||||
b.stateMutex.Lock()
|
||||
defer b.stateMutex.Unlock()
|
||||
|
||||
b.state.NetworkStatus = StatusDisconnected
|
||||
b.state.EthernetConnected = false
|
||||
b.state.EthernetIP = ""
|
||||
b.state.WiFiConnected = false
|
||||
b.state.WiFiIP = ""
|
||||
b.state.WiredConnections = wiredConns
|
||||
|
||||
if wiredIface != nil {
|
||||
b.state.EthernetDevice = wiredIface.name
|
||||
log.Debugf("networkd: wired interface %s opState=%s", wiredIface.name, wiredIface.opState)
|
||||
if wiredIface.opState == "routable" || wiredIface.opState == "carrier" {
|
||||
b.state.EthernetConnected = true
|
||||
b.state.NetworkStatus = StatusEthernet
|
||||
|
||||
if addrs := b.getAddresses(wiredIface.name); len(addrs) > 0 {
|
||||
b.state.EthernetIP = addrs[0]
|
||||
log.Debugf("networkd: ethernet IP %s on %s", addrs[0], wiredIface.name)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if wifiIface != nil {
|
||||
b.state.WiFiDevice = wifiIface.name
|
||||
log.Debugf("networkd: wifi interface %s opState=%s", wifiIface.name, wifiIface.opState)
|
||||
if wifiIface.opState == "routable" || wifiIface.opState == "carrier" {
|
||||
b.state.WiFiConnected = true
|
||||
|
||||
if addrs := b.getAddresses(wifiIface.name); len(addrs) > 0 {
|
||||
b.state.WiFiIP = addrs[0]
|
||||
log.Debugf("networkd: wifi IP %s on %s", addrs[0], wifiIface.name)
|
||||
if b.state.NetworkStatus == StatusDisconnected {
|
||||
b.state.NetworkStatus = StatusWiFi
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (b *SystemdNetworkdBackend) isVirtualInterface(name string) bool {
|
||||
virtualPrefixes := []string{
|
||||
"lo", "docker", "veth", "virbr", "br-", "vnet", "tun", "tap",
|
||||
"vboxnet", "vmnet", "kube", "cni", "flannel", "cali",
|
||||
}
|
||||
for _, prefix := range virtualPrefixes {
|
||||
if strings.HasPrefix(name, prefix) {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func (b *SystemdNetworkdBackend) getAddresses(ifname string) []string {
|
||||
iface, err := net.InterfaceByName(ifname)
|
||||
if err != nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
addrs, err := iface.Addrs()
|
||||
if err != nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
var result []string
|
||||
for _, addr := range addrs {
|
||||
if ipnet, ok := addr.(*net.IPNet); ok {
|
||||
if ipv4 := ipnet.IP.To4(); ipv4 != nil {
|
||||
result = append(result, ipv4.String())
|
||||
}
|
||||
}
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
func (b *SystemdNetworkdBackend) GetCurrentState() (*BackendState, error) {
|
||||
b.stateMutex.RLock()
|
||||
defer b.stateMutex.RUnlock()
|
||||
s := *b.state
|
||||
return &s, nil
|
||||
}
|
||||
|
||||
func (b *SystemdNetworkdBackend) GetPromptBroker() PromptBroker {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (b *SystemdNetworkdBackend) SetPromptBroker(broker PromptBroker) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (b *SystemdNetworkdBackend) SubmitCredentials(token string, secrets map[string]string, save bool) error {
|
||||
return fmt.Errorf("credentials not needed by networkd backend")
|
||||
}
|
||||
|
||||
func (b *SystemdNetworkdBackend) CancelCredentials(token string) error {
|
||||
return fmt.Errorf("credentials not needed by networkd backend")
|
||||
}
|
||||
|
||||
func (b *SystemdNetworkdBackend) EnsureDhcpUp(ifname string) error {
|
||||
b.linksMutex.RLock()
|
||||
link, exists := b.links[ifname]
|
||||
b.linksMutex.RUnlock()
|
||||
|
||||
if !exists {
|
||||
return fmt.Errorf("interface %s not found", ifname)
|
||||
}
|
||||
|
||||
linkObj := b.conn.Object(networkdBusName, link.path)
|
||||
return linkObj.Call(networkdLinkIface+".Reconfigure", 0).Err
|
||||
}
|
||||
Reference in New Issue
Block a user