mirror of
https://github.com/gravitl/netmaker.git
synced 2024-09-20 15:26:04 +08:00
100 lines
3.5 KiB
Go
100 lines
3.5 KiB
Go
package models
|
|
|
|
import (
|
|
"net"
|
|
"time"
|
|
)
|
|
|
|
// Network Struct - contains info for a given unique network
|
|
// At some point, need to replace all instances of Name with something else like Identifier
|
|
type Network struct {
|
|
AddressRange string `json:"addressrange" bson:"addressrange" validate:"omitempty,cidrv4"`
|
|
AddressRange6 string `json:"addressrange6" bson:"addressrange6" validate:"omitempty,cidrv6"`
|
|
NetID string `json:"netid" bson:"netid" validate:"required,min=1,max=32,netid_valid"`
|
|
NodesLastModified int64 `json:"nodeslastmodified" bson:"nodeslastmodified"`
|
|
NetworkLastModified int64 `json:"networklastmodified" bson:"networklastmodified"`
|
|
DefaultInterface string `json:"defaultinterface" bson:"defaultinterface" validate:"min=1,max=35"`
|
|
DefaultListenPort int32 `json:"defaultlistenport,omitempty" bson:"defaultlistenport,omitempty" validate:"omitempty,min=1024,max=65535"`
|
|
NodeLimit int32 `json:"nodelimit" bson:"nodelimit"`
|
|
DefaultPostDown string `json:"defaultpostdown" bson:"defaultpostdown"`
|
|
DefaultKeepalive int32 `json:"defaultkeepalive" bson:"defaultkeepalive" validate:"omitempty,max=1000"`
|
|
AllowManualSignUp string `json:"allowmanualsignup" bson:"allowmanualsignup" validate:"checkyesorno"`
|
|
IsIPv4 string `json:"isipv4" bson:"isipv4" validate:"checkyesorno"`
|
|
IsIPv6 string `json:"isipv6" bson:"isipv6" validate:"checkyesorno"`
|
|
DefaultUDPHolePunch string `json:"defaultudpholepunch" bson:"defaultudpholepunch" validate:"checkyesorno"`
|
|
DefaultMTU int32 `json:"defaultmtu" bson:"defaultmtu"`
|
|
DefaultACL string `json:"defaultacl" bson:"defaultacl" yaml:"defaultacl" validate:"checkyesorno"`
|
|
}
|
|
|
|
// SaveData - sensitive fields of a network that should be kept the same
|
|
type SaveData struct { // put sensitive fields here
|
|
NetID string `json:"netid" bson:"netid" validate:"required,min=1,max=32,netid_valid"`
|
|
}
|
|
|
|
// Network.SetNodesLastModified - sets nodes last modified on network, depricated
|
|
func (network *Network) SetNodesLastModified() {
|
|
network.NodesLastModified = time.Now().Unix()
|
|
}
|
|
|
|
// Network.SetNetworkLastModified - sets network last modified time
|
|
func (network *Network) SetNetworkLastModified() {
|
|
network.NetworkLastModified = time.Now().Unix()
|
|
}
|
|
|
|
// Network.SetDefaults - sets default values for a network struct
|
|
func (network *Network) SetDefaults() {
|
|
if network.DefaultUDPHolePunch == "" {
|
|
network.DefaultUDPHolePunch = "no"
|
|
}
|
|
if network.DefaultInterface == "" {
|
|
if len(network.NetID) < 33 {
|
|
network.DefaultInterface = "nm-" + network.NetID
|
|
} else {
|
|
network.DefaultInterface = network.NetID
|
|
}
|
|
}
|
|
if network.DefaultListenPort == 0 {
|
|
network.DefaultListenPort = 51821
|
|
}
|
|
if network.NodeLimit == 0 {
|
|
network.NodeLimit = 999999999
|
|
}
|
|
if network.DefaultKeepalive == 0 {
|
|
network.DefaultKeepalive = 20
|
|
}
|
|
if network.AllowManualSignUp == "" {
|
|
network.AllowManualSignUp = "no"
|
|
}
|
|
|
|
if network.IsIPv4 == "" {
|
|
network.IsIPv4 = "yes"
|
|
}
|
|
|
|
if network.IsIPv6 == "" {
|
|
network.IsIPv6 = "no"
|
|
}
|
|
|
|
if network.DefaultMTU == 0 {
|
|
network.DefaultMTU = 1280
|
|
}
|
|
|
|
if network.DefaultACL == "" {
|
|
network.DefaultACL = "yes"
|
|
}
|
|
}
|
|
|
|
func (network *Network) GetNetworkNetworkCIDR4() *net.IPNet {
|
|
if network.AddressRange == "" {
|
|
return nil
|
|
}
|
|
_, netCidr, _ := net.ParseCIDR(network.AddressRange)
|
|
return netCidr
|
|
}
|
|
func (network *Network) GetNetworkNetworkCIDR6() *net.IPNet {
|
|
if network.AddressRange6 == "" {
|
|
return nil
|
|
}
|
|
_, netCidr, _ := net.ParseCIDR(network.AddressRange6)
|
|
return netCidr
|
|
}
|