mirror of
https://github.com/netbirdio/netbird.git
synced 2024-11-25 17:43:38 +01:00
94 lines
2.1 KiB
Go
94 lines
2.1 KiB
Go
package iface
|
|
|
|
import (
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/vishvananda/netlink"
|
|
"os"
|
|
)
|
|
|
|
// Create Creates a new Wireguard interface, sets a given IP and brings it up.
|
|
// Will reuse an existing one.
|
|
func Create(iface string, address string) error {
|
|
attrs := netlink.NewLinkAttrs()
|
|
attrs.Name = iface
|
|
|
|
link := wgLink{
|
|
attrs: &attrs,
|
|
}
|
|
|
|
log.Debugf("adding device: %s", iface)
|
|
err := netlink.LinkAdd(&link)
|
|
if os.IsExist(err) {
|
|
log.Infof("interface %s already exists. Will reuse.", iface)
|
|
} else if err != nil {
|
|
return err
|
|
}
|
|
|
|
log.Debugf("adding address %s to interface: %s", address, iface)
|
|
addr, _ := netlink.ParseAddr(address)
|
|
err = netlink.AddrAdd(&link, addr)
|
|
if os.IsExist(err) {
|
|
log.Infof("interface %s already has the address: %s", iface, address)
|
|
} else if err != nil {
|
|
return err
|
|
}
|
|
err = assignAddr(address, iface)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// todo do a discovery
|
|
log.Debugf("setting MTU: %s", iface)
|
|
err = netlink.LinkSetMTU(&link, defaultMTU)
|
|
if err != nil {
|
|
log.Errorf("error setting MTU on interface: %s", iface)
|
|
return err
|
|
}
|
|
|
|
log.Debugf("bringing up interface: %s", iface)
|
|
err = netlink.LinkSetUp(&link)
|
|
if err != nil {
|
|
log.Errorf("error bringing up interface: %s", iface)
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// assignAddr Adds IP address to the tunnel interface
|
|
func assignAddr(address, name string) error {
|
|
var err error
|
|
attrs := netlink.NewLinkAttrs()
|
|
attrs.Name = name
|
|
|
|
link := wgLink{
|
|
attrs: &attrs,
|
|
}
|
|
|
|
log.Debugf("adding address %s to interface: %s", address, attrs.Name)
|
|
addr, _ := netlink.ParseAddr(address)
|
|
err = netlink.AddrAdd(&link, addr)
|
|
if os.IsExist(err) {
|
|
log.Infof("interface %s already has the address: %s", attrs.Name, address)
|
|
} else if err != nil {
|
|
return err
|
|
}
|
|
// On linux, the link must be brought up
|
|
err = netlink.LinkSetUp(&link)
|
|
return err
|
|
}
|
|
|
|
type wgLink struct {
|
|
attrs *netlink.LinkAttrs
|
|
}
|
|
|
|
// Attrs returns the Wireguard's default attributes
|
|
func (w *wgLink) Attrs() *netlink.LinkAttrs {
|
|
return w.attrs
|
|
}
|
|
|
|
// Type returns the interface type
|
|
func (w *wgLink) Type() string {
|
|
return "wireguard"
|
|
}
|