mirror of https://github.com/jetkvm/kvm.git
111 lines
3.7 KiB
Go
111 lines
3.7 KiB
Go
package network
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"time"
|
|
|
|
"github.com/guregu/null/v6"
|
|
"github.com/jetkvm/kvm/internal/mdns"
|
|
"golang.org/x/net/idna"
|
|
)
|
|
|
|
type IPv6Address struct {
|
|
Address net.IP `json:"address"`
|
|
Prefix net.IPNet `json:"prefix"`
|
|
ValidLifetime *time.Time `json:"valid_lifetime"`
|
|
PreferredLifetime *time.Time `json:"preferred_lifetime"`
|
|
Scope int `json:"scope"`
|
|
}
|
|
|
|
type IPv4StaticConfig struct {
|
|
Address null.String `json:"address,omitempty" validate_type:"ipv4" required:"true"`
|
|
Netmask null.String `json:"netmask,omitempty" validate_type:"ipv4" required:"true"`
|
|
Gateway null.String `json:"gateway,omitempty" validate_type:"ipv4" required:"true"`
|
|
DNS []string `json:"dns,omitempty" validate_type:"ipv4" required:"true"`
|
|
}
|
|
|
|
type IPv6StaticConfig struct {
|
|
Address null.String `json:"address,omitempty" validate_type:"ipv6" required:"true"`
|
|
Prefix null.String `json:"prefix,omitempty" validate_type:"ipv6" required:"true"`
|
|
Gateway null.String `json:"gateway,omitempty" validate_type:"ipv6" required:"true"`
|
|
DNS []string `json:"dns,omitempty" validate_type:"ipv6" required:"true"`
|
|
}
|
|
type NetworkConfig struct {
|
|
Hostname null.String `json:"hostname,omitempty" validate_type:"hostname"`
|
|
Domain null.String `json:"domain,omitempty" validate_type:"hostname"`
|
|
|
|
IPv4Mode null.String `json:"ipv4_mode,omitempty" one_of:"dhcp,static,disabled" default:"dhcp"`
|
|
IPv4Static *IPv4StaticConfig `json:"ipv4_static,omitempty" required_if:"IPv4Mode=static"`
|
|
|
|
IPv6Mode null.String `json:"ipv6_mode,omitempty" one_of:"slaac,dhcpv6,slaac_and_dhcpv6,static,link_local,disabled" default:"slaac"`
|
|
IPv6Static *IPv6StaticConfig `json:"ipv6_static,omitempty" required_if:"IPv6Mode=static"`
|
|
|
|
LLDPMode null.String `json:"lldp_mode,omitempty" one_of:"disabled,basic,all" default:"basic"`
|
|
LLDPTxTLVs []string `json:"lldp_tx_tlvs,omitempty" one_of:"chassis,port,system,vlan" default:"chassis,port,system,vlan"`
|
|
MDNSMode null.String `json:"mdns_mode,omitempty" one_of:"disabled,auto,ipv4_only,ipv6_only" default:"auto"`
|
|
TimeSyncMode null.String `json:"time_sync_mode,omitempty" one_of:"ntp_only,ntp_and_http,http_only,custom" default:"ntp_and_http"`
|
|
TimeSyncOrdering []string `json:"time_sync_ordering,omitempty" one_of:"http,ntp,ntp_dhcp,ntp_user_provided,ntp_fallback" default:"ntp,http"`
|
|
TimeSyncDisableFallback null.Bool `json:"time_sync_disable_fallback,omitempty" default:"false"`
|
|
TimeSyncParallel null.Int `json:"time_sync_parallel,omitempty" default:"4"`
|
|
}
|
|
|
|
func (c *NetworkConfig) GetMDNSMode() *mdns.MDNSListenOptions {
|
|
mode := c.MDNSMode.String
|
|
listenOptions := &mdns.MDNSListenOptions{
|
|
IPv4: true,
|
|
IPv6: true,
|
|
}
|
|
|
|
switch mode {
|
|
case "ipv4_only":
|
|
listenOptions.IPv6 = false
|
|
case "ipv6_only":
|
|
listenOptions.IPv4 = false
|
|
case "disabled":
|
|
listenOptions.IPv4 = false
|
|
listenOptions.IPv6 = false
|
|
}
|
|
|
|
return listenOptions
|
|
}
|
|
func (s *NetworkInterfaceState) GetHostname() string {
|
|
hostname := ToValidHostname(s.config.Hostname.String)
|
|
|
|
if hostname == "" {
|
|
return s.defaultHostname
|
|
}
|
|
|
|
return hostname
|
|
}
|
|
|
|
func ToValidDomain(domain string) string {
|
|
ascii, err := idna.Lookup.ToASCII(domain)
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
|
|
return ascii
|
|
}
|
|
|
|
func (s *NetworkInterfaceState) GetDomain() string {
|
|
domain := ToValidDomain(s.config.Domain.String)
|
|
|
|
if domain == "" {
|
|
lease := s.dhcpClient.GetLease()
|
|
if lease != nil && lease.Domain != "" {
|
|
domain = ToValidDomain(lease.Domain)
|
|
}
|
|
}
|
|
|
|
if domain == "" {
|
|
return "local"
|
|
}
|
|
|
|
return domain
|
|
}
|
|
|
|
func (s *NetworkInterfaceState) GetFQDN() string {
|
|
return fmt.Sprintf("%s.%s", s.GetHostname(), s.GetDomain())
|
|
}
|