mirror of https://github.com/jetkvm/kvm.git
65 lines
1.3 KiB
Go
65 lines
1.3 KiB
Go
//go:build linux
|
|
|
|
package network
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/vishvananda/netlink"
|
|
"github.com/vishvananda/netlink/nl"
|
|
)
|
|
|
|
func (s *NetworkInterfaceState) HandleLinkUpdate(update netlink.LinkUpdate) {
|
|
if update.Link.Attrs().Name == s.interfaceName {
|
|
s.l.Info().Interface("update", update).Msg("interface link update received")
|
|
_ = s.CheckAndUpdateDhcp()
|
|
}
|
|
}
|
|
|
|
func (s *NetworkInterfaceState) Run() error {
|
|
updates := make(chan netlink.LinkUpdate)
|
|
done := make(chan struct{})
|
|
|
|
if err := netlink.LinkSubscribe(updates, done); err != nil {
|
|
s.l.Warn().Err(err).Msg("failed to subscribe to link updates")
|
|
return err
|
|
}
|
|
|
|
_ = s.setHostnameIfNotSame()
|
|
|
|
// run the dhcp client
|
|
if err := s.dhcpClient.Start(); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := s.CheckAndUpdateDhcp(); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := s.ifConfig.Apply(s); err != nil {
|
|
s.l.Error().Err(err).Msg("failed to apply network interface config")
|
|
}
|
|
|
|
go func() {
|
|
ticker := time.NewTicker(1 * time.Second)
|
|
defer ticker.Stop()
|
|
|
|
for {
|
|
select {
|
|
case update := <-updates:
|
|
s.HandleLinkUpdate(update)
|
|
case <-ticker.C:
|
|
_ = s.CheckAndUpdateDhcp()
|
|
case <-done:
|
|
return
|
|
}
|
|
}
|
|
}()
|
|
|
|
return nil
|
|
}
|
|
|
|
func netlinkAddrs(iface netlink.Link) ([]netlink.Addr, error) {
|
|
return netlink.AddrList(iface, nl.FAMILY_ALL)
|
|
}
|