mirror of
https://github.com/gravitl/netmaker.git
synced 2024-11-11 01:54:34 +08:00
209 lines
8.8 KiB
Go
209 lines
8.8 KiB
Go
package models
|
|
|
|
import (
|
|
"context"
|
|
"math/rand"
|
|
"net"
|
|
"time"
|
|
|
|
"github.com/gravitl/netmaker/mongoconn"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
|
)
|
|
|
|
const charset = "abcdefghijklmnopqrstuvwxyz" + "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
|
|
|
|
var seededRand *rand.Rand = rand.New(
|
|
rand.NewSource(time.Now().UnixNano()))
|
|
|
|
//node struct
|
|
type Node struct {
|
|
ID primitive.ObjectID `json:"_id,omitempty" bson:"_id,omitempty"`
|
|
Address string `json:"address" bson:"address" validate:"omitempty,ipv4"`
|
|
Address6 string `json:"address6" bson:"address6" validate:"omitempty,ipv6"`
|
|
LocalAddress string `json:"localaddress" bson:"localaddress" validate:"omitempty,ip"`
|
|
Name string `json:"name" bson:"name" validate:"omitempty,max=12"`
|
|
ListenPort int32 `json:"listenport" bson:"listenport" validate:"omitempty,numeric,min=1024,max=65535"`
|
|
PublicKey string `json:"publickey" bson:"publickey" validate:"required,base64"`
|
|
Endpoint string `json:"endpoint" bson:"endpoint" validate:"required,ip"`
|
|
PostUp string `json:"postup" bson:"postup"`
|
|
PostDown string `json:"postdown" bson:"postdown"`
|
|
AllowedIPs string `json:"allowedips" bson:"allowedips"`
|
|
PersistentKeepalive int32 `json:"persistentkeepalive" bson:"persistentkeepalive" validate:"omitempty,numeric,max=1000"`
|
|
SaveConfig *bool `json:"saveconfig" bson:"saveconfig"`
|
|
AccessKey string `json:"accesskey" bson:"accesskey"`
|
|
Interface string `json:"interface" bson:"interface"`
|
|
LastModified int64 `json:"lastmodified" bson:"lastmodified"`
|
|
KeyUpdateTimeStamp int64 `json:"keyupdatetimestamp" bson:"keyupdatetimestamp"`
|
|
ExpirationDateTime int64 `json:"expdatetime" bson:"expdatetime"`
|
|
LastPeerUpdate int64 `json:"lastpeerupdate" bson:"lastpeerupdate"`
|
|
LastCheckIn int64 `json:"lastcheckin" bson:"lastcheckin"`
|
|
MacAddress string `json:"macaddress" bson:"macaddress" validate:"required,mac,macaddress_unique"`
|
|
CheckInInterval int32 `json:"checkininterval" bson:"checkininterval"`
|
|
Password string `json:"password" bson:"password" validate:"required,min=6"`
|
|
Network string `json:"network" bson:"network" validate:"network_exists"`
|
|
IsPending bool `json:"ispending" bson:"ispending"`
|
|
IsEgressGateway bool `json:"isegressgateway" bson:"isegressgateway"`
|
|
IsIngressGateway bool `json:"isingressgateway" bson:"isingressgateway"`
|
|
EgressGatewayRange string `json:"gatewayrange" bson:"gatewayrange"`
|
|
PostChanges string `json:"postchanges" bson:"postchanges"`
|
|
}
|
|
|
|
//node update struct --- only validations are different
|
|
type NodeUpdate struct {
|
|
ID primitive.ObjectID `json:"_id,omitempty" bson:"_id,omitempty"`
|
|
Address string `json:"address" bson:"address" validate:"omitempty,ip"`
|
|
Address6 string `json:"address6" bson:"address6" validate:"omitempty,ipv6"`
|
|
LocalAddress string `json:"localaddress" bson:"localaddress" validate:"omitempty,ip"`
|
|
Name string `json:"name" bson:"name" validate:"omitempty,max=12"`
|
|
ListenPort int32 `json:"listenport" bson:"listenport" validate:"omitempty,numeric,min=1024,max=65535"`
|
|
PublicKey string `json:"publickey" bson:"publickey" validate:"omitempty,base64"`
|
|
Endpoint string `json:"endpoint" bson:"endpoint" validate:"omitempty,ip"`
|
|
PostUp string `json:"postup" bson:"postup"`
|
|
PostDown string `json:"postdown" bson:"postdown"`
|
|
AllowedIPs string `json:"allowedips" bson:"allowedips"`
|
|
PersistentKeepalive int32 `json:"persistentkeepalive" bson:"persistentkeepalive" validate:"omitempty,numeric,max=1000"`
|
|
SaveConfig *bool `json:"saveconfig" bson:"saveconfig"`
|
|
AccessKey string `json:"accesskey" bson:"accesskey"`
|
|
Interface string `json:"interface" bson:"interface"`
|
|
LastModified int64 `json:"lastmodified" bson:"lastmodified"`
|
|
KeyUpdateTimeStamp int64 `json:"keyupdatetimestamp" bson:"keyupdatetimestamp"`
|
|
ExpirationDateTime int64 `json:"expdatetime" bson:"expdatetime"`
|
|
LastPeerUpdate int64 `json:"lastpeerupdate" bson:"lastpeerupdate"`
|
|
LastCheckIn int64 `json:"lastcheckin" bson:"lastcheckin"`
|
|
MacAddress string `json:"macaddress" bson:"macaddress" validate:"required,mac"`
|
|
CheckInInterval int32 `json:"checkininterval" bson:"checkininterval"`
|
|
Password string `json:"password" bson:"password" validate:"omitempty,min=5"`
|
|
Network string `json:"network" bson:"network" validate:"network_exists"`
|
|
IsPending bool `json:"ispending" bson:"ispending"`
|
|
IsIngressGateway bool `json:"isingressgateway" bson:"isingressgateway"`
|
|
IsEgressGateway bool `json:"isegressgateway" bson:"isegressgateway"`
|
|
EgressGatewayRange string `json:"gatewayrange" bson:"gatewayrange"`
|
|
PostChanges string `json:"postchanges" bson:"postchanges"`
|
|
}
|
|
|
|
//Duplicated function for NodeUpdates
|
|
func (node *NodeUpdate) GetNetwork() (Network, error) {
|
|
|
|
var network Network
|
|
|
|
collection := mongoconn.NetworkDB
|
|
//collection := mongoconn.Client.Database("netmaker").Collection("networks")
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
|
|
filter := bson.M{"netid": node.Network}
|
|
err := collection.FindOne(ctx, filter).Decode(&network)
|
|
|
|
defer cancel()
|
|
|
|
if err != nil {
|
|
//log.Fatal(err)
|
|
return network, err
|
|
}
|
|
|
|
return network, err
|
|
}
|
|
|
|
//TODO: Contains a fatal error return. Need to change
|
|
//Used in contexts where it's not the Parent network.
|
|
func (node *Node) GetNetwork() (Network, error) {
|
|
|
|
var network Network
|
|
|
|
//collection := mongoconn.NetworkDB
|
|
collection := mongoconn.Client.Database("netmaker").Collection("networks")
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
|
|
filter := bson.M{"netid": node.Network}
|
|
err := collection.FindOne(ctx, filter).Decode(&network)
|
|
|
|
defer cancel()
|
|
|
|
if err != nil {
|
|
//log.Fatal(err)
|
|
return network, err
|
|
}
|
|
|
|
return network, err
|
|
}
|
|
|
|
//TODO:
|
|
//Not sure if below two methods are necessary. May want to revisit
|
|
func (node *Node) SetLastModified() {
|
|
node.LastModified = time.Now().Unix()
|
|
}
|
|
|
|
func (node *Node) SetLastCheckIn() {
|
|
node.LastCheckIn = time.Now().Unix()
|
|
}
|
|
|
|
func (node *Node) SetLastPeerUpdate() {
|
|
node.LastPeerUpdate = time.Now().Unix()
|
|
}
|
|
|
|
func (node *Node) SetExpirationDateTime() {
|
|
node.ExpirationDateTime = time.Unix(33174902665, 0).Unix()
|
|
}
|
|
|
|
func (node *Node) SetDefaultName() {
|
|
if node.Name == "" {
|
|
nodeid := StringWithCharset(5, charset)
|
|
nodename := "node-" + nodeid
|
|
node.Name = nodename
|
|
}
|
|
}
|
|
|
|
//TODO: I dont know why this exists
|
|
//This should exist on the node.go struct. I'm sure there was a reason?
|
|
func (node *Node) SetDefaults() {
|
|
|
|
//TODO: Maybe I should make Network a part of the node struct. Then we can just query the Network object for stuff.
|
|
parentNetwork, _ := node.GetNetwork()
|
|
|
|
node.ExpirationDateTime = time.Unix(33174902665, 0).Unix()
|
|
|
|
if node.ListenPort == 0 {
|
|
node.ListenPort = parentNetwork.DefaultListenPort
|
|
}
|
|
if node.PostDown == "" {
|
|
//Empty because we dont set it
|
|
//may want to set it to something in the future
|
|
}
|
|
//TODO: This is dumb and doesn't work
|
|
//Need to change
|
|
if node.SaveConfig == nil {
|
|
if parentNetwork.DefaultSaveConfig != nil {
|
|
defaultsave := *parentNetwork.DefaultSaveConfig
|
|
node.SaveConfig = &defaultsave
|
|
}
|
|
}
|
|
if node.Interface == "" {
|
|
node.Interface = parentNetwork.DefaultInterface
|
|
}
|
|
if node.PersistentKeepalive == 0 {
|
|
node.PersistentKeepalive = parentNetwork.DefaultKeepalive
|
|
}
|
|
if node.PostUp == "" {
|
|
postup := parentNetwork.DefaultPostUp
|
|
node.PostUp = postup
|
|
}
|
|
node.CheckInInterval = parentNetwork.DefaultCheckInInterval
|
|
|
|
}
|
|
|
|
func StringWithCharset(length int, charset string) string {
|
|
b := make([]byte, length)
|
|
for i := range b {
|
|
b[i] = charset[seededRand.Intn(len(charset))]
|
|
}
|
|
return string(b)
|
|
}
|
|
|
|
//Check for valid IPv4 address
|
|
//Note: We dont handle IPv6 AT ALL!!!!! This definitely is needed at some point
|
|
//But for iteration 1, lets just stick to IPv4. Keep it simple stupid.
|
|
func IsIpv4Net(host string) bool {
|
|
return net.ParseIP(host) != nil
|
|
}
|