netmaker/controllers/networkHttpController.go

778 lines
23 KiB
Go
Raw Normal View History

2021-03-26 00:17:52 +08:00
package controller
import (
2021-04-16 02:48:10 +08:00
"context"
"encoding/base64"
"encoding/json"
"errors"
"fmt"
"net"
2021-04-16 02:48:10 +08:00
"net/http"
"strings"
"time"
2021-05-01 11:07:25 +08:00
"github.com/go-playground/validator/v10"
2021-04-16 02:48:10 +08:00
"github.com/gorilla/mux"
"github.com/gravitl/netmaker/functions"
"github.com/gravitl/netmaker/models"
"github.com/gravitl/netmaker/mongoconn"
"github.com/gravitl/netmaker/servercfg"
2021-04-16 02:48:10 +08:00
"go.mongodb.org/mongo-driver/bson"
2021-05-01 11:07:25 +08:00
"go.mongodb.org/mongo-driver/mongo"
2021-04-16 02:48:10 +08:00
"go.mongodb.org/mongo-driver/mongo/options"
2021-03-26 00:17:52 +08:00
)
func networkHandlers(r *mux.Router) {
2021-04-16 02:48:10 +08:00
r.HandleFunc("/api/networks", securityCheck(http.HandlerFunc(getNetworks))).Methods("GET")
r.HandleFunc("/api/networks", securityCheck(http.HandlerFunc(createNetwork))).Methods("POST")
r.HandleFunc("/api/networks/{networkname}", securityCheck(http.HandlerFunc(getNetwork))).Methods("GET")
r.HandleFunc("/api/networks/{networkname}", securityCheck(http.HandlerFunc(updateNetwork))).Methods("PUT")
r.HandleFunc("/api/networks/{networkname}", securityCheck(http.HandlerFunc(deleteNetwork))).Methods("DELETE")
r.HandleFunc("/api/networks/{networkname}/keyupdate", securityCheck(http.HandlerFunc(keyUpdate))).Methods("POST")
r.HandleFunc("/api/networks/{networkname}/keys", securityCheck(http.HandlerFunc(createAccessKey))).Methods("POST")
r.HandleFunc("/api/networks/{networkname}/keys", securityCheck(http.HandlerFunc(getAccessKeys))).Methods("GET")
2021-05-26 00:48:04 +08:00
r.HandleFunc("/api/networks/{networkname}/signuptoken", securityCheck(http.HandlerFunc(getSignupToken))).Methods("GET")
2021-04-16 02:48:10 +08:00
r.HandleFunc("/api/networks/{networkname}/keys/{name}", securityCheck(http.HandlerFunc(deleteAccessKey))).Methods("DELETE")
2021-03-26 00:17:52 +08:00
}
//Security check is middleware for every function and just checks to make sure that its the master calling
//Only admin should have access to all these network-level actions
2021-03-26 00:17:52 +08:00
//or maybe some Users once implemented
func securityCheck(next http.Handler) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
var errorResponse = models.ErrorResponse{
2021-05-03 03:54:12 +08:00
Code: http.StatusUnauthorized, Message: "W1R3: It's not you it's me.",
2021-03-26 00:17:52 +08:00
}
var params = mux.Vars(r)
2021-05-03 03:54:12 +08:00
bearerToken := r.Header.Get("Authorization")
err := SecurityCheck(params["networkname"], bearerToken)
2021-04-16 02:48:10 +08:00
if err != nil {
2021-05-28 22:04:07 +08:00
if strings.Contains(err.Error(), "does not exist") {
errorResponse.Code = http.StatusNotFound
}
2021-05-03 03:54:12 +08:00
errorResponse.Message = err.Error()
2021-03-26 00:17:52 +08:00
returnErrorResponse(w, r, errorResponse)
2021-04-13 11:19:01 +08:00
return
2021-05-03 03:54:12 +08:00
}
next.ServeHTTP(w, r)
}
}
func SecurityCheck(netname, token string) error {
hasnetwork := netname != ""
networkexists, err := functions.NetworkExists(netname)
if err != nil {
return err
}
if hasnetwork && !networkexists {
return errors.New("This network does not exist")
}
2021-04-16 02:48:10 +08:00
2021-05-03 03:54:12 +08:00
var hasBearer = true
var tokenSplit = strings.Split(token, " ")
var authToken = ""
2021-04-16 02:48:10 +08:00
2021-05-03 03:54:12 +08:00
if len(tokenSplit) < 2 {
hasBearer = false
} else {
authToken = tokenSplit[1]
2021-03-26 00:17:52 +08:00
}
2021-05-03 03:54:12 +08:00
//all endpoints here require master so not as complicated
if !hasBearer || !authenticateMaster(authToken) {
2021-07-02 12:03:46 +08:00
_, networks, isadmin, err := functions.VerifyUserToken(authToken)
if err != nil {
return errors.New("Error verifying user token")
}
if !isadmin && netname != ""{
if !functions.SliceContains(networks, netname){
return errors.New("You are unauthorized to access this endpoint")
}
} else if !isadmin {
2021-05-30 04:18:22 +08:00
return errors.New("You are unauthorized to access this endpoint")
}
}
2021-05-03 03:54:12 +08:00
return nil
2021-03-26 00:17:52 +08:00
}
2021-04-16 02:48:10 +08:00
2021-03-26 00:17:52 +08:00
//Consider a more secure way of setting master key
func authenticateMaster(tokenString string) bool {
if tokenString == servercfg.GetMasterKey() {
2021-04-16 02:48:10 +08:00
return true
}
return false
2021-03-26 00:17:52 +08:00
}
//simple get all networks function
func getNetworks(w http.ResponseWriter, r *http.Request) {
2021-03-26 00:17:52 +08:00
allnetworks, err := functions.ListNetworks()
2021-04-13 11:19:01 +08:00
if err != nil {
returnErrorResponse(w, r, formatError(err, "internal"))
return
}
networks := RemoveComms(allnetworks)
w.WriteHeader(http.StatusOK)
json.NewEncoder(w).Encode(networks)
return
}
func RemoveComms(networks []models.Network) []models.Network {
var index int = 100000001
for ind, net := range networks {
if net.NetID == "comms" {
index = ind
}
}
if index == 100000001 {
return networks
}
returnable := make([]models.Network, 0)
returnable = append(returnable, networks[:index]...)
return append(returnable, networks[index+1:]...)
2021-03-26 00:17:52 +08:00
}
2021-05-03 19:13:12 +08:00
func ValidateNetworkUpdate(network models.NetworkUpdate) error {
v := validator.New()
2021-05-03 19:13:12 +08:00
// _ = v.RegisterValidation("addressrange_valid", func(fl validator.FieldLevel) bool {
// isvalid := fl.Field().String() == "" || functions.IsIpCIDR(fl.Field().String())
// return isvalid
// })
//_ = v.RegisterValidation("addressrange6_valid", func(fl validator.FieldLevel) bool {
// isvalid := fl.Field().String() == "" || functions.IsIpCIDR(fl.Field().String())
// return isvalid
// })
2021-05-03 19:13:12 +08:00
// _ = v.RegisterValidation("localrange_valid", func(fl validator.FieldLevel) bool {
// isvalid := fl.Field().String() == "" || functions.IsIpCIDR(fl.Field().String())
// return isvalid
// })
2021-05-03 19:13:12 +08:00
// _ = v.RegisterValidation("netid_valid", func(fl validator.FieldLevel) bool {
// return true
// })
2021-05-03 19:13:12 +08:00
// _ = v.RegisterValidation("displayname_unique", func(fl validator.FieldLevel) bool {
// return true
// })
err := v.Struct(network)
if err != nil {
for _, e := range err.(validator.ValidationErrors) {
fmt.Println(e)
}
}
return err
}
2021-05-01 11:07:25 +08:00
func ValidateNetworkCreate(network models.Network) error {
2021-03-26 00:17:52 +08:00
2021-04-16 02:48:10 +08:00
v := validator.New()
2021-03-26 00:17:52 +08:00
2021-05-01 11:07:25 +08:00
// _ = v.RegisterValidation("addressrange_valid", func(fl validator.FieldLevel) bool {
// isvalid := functions.IsIpCIDR(fl.Field().String())
// return isvalid
// })
_ = v.RegisterValidation("addressrange6_valid", func(fl validator.FieldLevel) bool {
2021-04-28 11:22:52 +08:00
isvalid := true
if *network.IsDualStack {
isvalid = functions.IsIpCIDR(fl.Field().String())
}
return isvalid
})
2021-05-01 11:07:25 +08:00
//
// _ = v.RegisterValidation("localrange_valid", func(fl validator.FieldLevel) bool {
// isvalid := fl.Field().String() == "" || functions.IsIpCIDR(fl.Field().String())
// return isvalid
// })
//
_ = v.RegisterValidation("netid_valid", func(fl validator.FieldLevel) bool {
isFieldUnique, _ := functions.IsNetworkNameUnique(fl.Field().String())
2021-05-01 11:07:25 +08:00
// inCharSet := functions.NameInNetworkCharSet(fl.Field().String())
return isFieldUnique
2021-04-16 02:48:10 +08:00
})
2021-05-01 11:07:25 +08:00
//
_ = v.RegisterValidation("displayname_unique", func(fl validator.FieldLevel) bool {
isFieldUnique, _ := functions.IsNetworkDisplayNameUnique(fl.Field().String())
return isFieldUnique
})
2021-03-26 00:17:52 +08:00
2021-04-16 02:48:10 +08:00
err := v.Struct(network)
2021-03-26 00:17:52 +08:00
2021-04-16 02:48:10 +08:00
if err != nil {
for _, e := range err.(validator.ValidationErrors) {
fmt.Println(e)
}
}
return err
2021-03-26 00:17:52 +08:00
}
//Simple get network function
func getNetwork(w http.ResponseWriter, r *http.Request) {
2021-04-16 02:48:10 +08:00
// set header.
w.Header().Set("Content-Type", "application/json")
var params = mux.Vars(r)
2021-05-01 19:57:49 +08:00
netname := params["networkname"]
network, err := GetNetwork(netname)
if err != nil {
returnErrorResponse(w, r, formatError(err, "internal"))
return
}
w.WriteHeader(http.StatusOK)
json.NewEncoder(w).Encode(network)
}
2021-03-26 00:17:52 +08:00
2021-05-01 19:57:49 +08:00
func GetNetwork(name string) (models.Network, error) {
2021-04-16 02:48:10 +08:00
var network models.Network
collection := mongoconn.Client.Database("netmaker").Collection("networks")
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
2021-05-01 19:57:49 +08:00
filter := bson.M{"netid": name}
2021-04-16 02:48:10 +08:00
err := collection.FindOne(ctx, filter, options.FindOne().SetProjection(bson.M{"_id": 0})).Decode(&network)
defer cancel()
if err != nil {
2021-05-01 19:57:49 +08:00
return models.Network{}, err
2021-04-16 02:48:10 +08:00
}
2021-05-01 19:57:49 +08:00
return network, nil
2021-03-26 00:17:52 +08:00
}
2021-04-06 06:09:21 +08:00
func keyUpdate(w http.ResponseWriter, r *http.Request) {
2021-04-16 02:48:10 +08:00
w.Header().Set("Content-Type", "application/json")
var params = mux.Vars(r)
2021-05-01 19:57:49 +08:00
netname := params["networkname"]
network, err := KeyUpdate(netname)
2021-04-16 02:48:10 +08:00
if err != nil {
returnErrorResponse(w, r, formatError(err, "internal"))
return
2021-04-13 11:19:01 +08:00
}
2021-05-01 19:57:49 +08:00
w.WriteHeader(http.StatusOK)
json.NewEncoder(w).Encode(network)
}
2021-04-13 11:19:01 +08:00
2021-05-01 19:57:49 +08:00
func KeyUpdate(netname string) (models.Network, error) {
network, err := functions.GetParentNetwork(netname)
if err != nil {
return models.Network{}, err
}
network.KeyUpdateTimeStamp = time.Now().Unix()
2021-04-16 02:48:10 +08:00
collection := mongoconn.Client.Database("netmaker").Collection("networks")
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
2021-05-01 19:57:49 +08:00
filter := bson.M{"netid": netname}
2021-05-01 11:07:25 +08:00
// prepare update model.
update := bson.D{
{"$set", bson.D{
{"addressrange", network.AddressRange},
{"addressrange6", network.AddressRange6},
{"displayname", network.DisplayName},
{"defaultlistenport", network.DefaultListenPort},
{"defaultpostup", network.DefaultPostUp},
{"defaultpostdown", network.DefaultPostDown},
2021-04-28 11:17:59 +08:00
{"defaultkeepalive", network.DefaultKeepalive},
{"keyupdatetimestamp", network.KeyUpdateTimeStamp},
{"defaultsaveconfig", network.DefaultSaveConfig},
{"defaultinterface", network.DefaultInterface},
{"nodeslastmodified", network.NodesLastModified},
{"networklastmodified", network.NetworkLastModified},
{"allowmanualsignup", network.AllowManualSignUp},
{"checkininterval", network.DefaultCheckInInterval},
2021-05-01 11:07:25 +08:00
}},
}
2021-04-16 02:48:10 +08:00
err = collection.FindOneAndUpdate(ctx, filter, update).Decode(&network)
defer cancel()
if err != nil {
2021-05-01 19:57:49 +08:00
return models.Network{}, err
2021-04-16 02:48:10 +08:00
}
2021-05-01 19:57:49 +08:00
return network, nil
2021-04-06 06:09:21 +08:00
}
2021-04-13 14:55:49 +08:00
//Update a network
2021-04-16 02:48:10 +08:00
func AlertNetwork(netid string) error {
2021-04-13 14:55:49 +08:00
2021-04-16 02:48:10 +08:00
collection := mongoconn.Client.Database("netmaker").Collection("networks")
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
filter := bson.M{"netid": netid}
2021-04-13 14:55:49 +08:00
2021-04-16 02:48:10 +08:00
var network models.Network
2021-04-13 14:55:49 +08:00
2021-04-16 02:48:10 +08:00
network, err := functions.GetParentNetwork(netid)
if err != nil {
return err
}
2021-04-13 14:55:49 +08:00
updatetime := time.Now().Unix()
2021-04-16 02:48:10 +08:00
update := bson.D{
{"$set", bson.D{
{"nodeslastmodified", updatetime},
{"networklastmodified", updatetime},
}},
}
2021-04-13 14:55:49 +08:00
2021-04-16 02:48:10 +08:00
err = collection.FindOneAndUpdate(ctx, filter, update).Decode(&network)
defer cancel()
2021-04-13 14:55:49 +08:00
2021-04-16 02:48:10 +08:00
return err
2021-04-13 14:55:49 +08:00
}
//Update a network
func updateNetwork(w http.ResponseWriter, r *http.Request) {
2021-04-16 02:48:10 +08:00
w.Header().Set("Content-Type", "application/json")
var params = mux.Vars(r)
var network models.Network
network, err := functions.GetParentNetwork(params["networkname"])
2021-04-16 02:48:10 +08:00
if err != nil {
returnErrorResponse(w, r, formatError(err, "internal"))
return
}
2021-03-26 00:17:52 +08:00
2021-05-03 19:13:12 +08:00
var networkChange models.NetworkUpdate
2021-03-26 00:17:52 +08:00
_ = json.NewDecoder(r.Body).Decode(&networkChange)
if networkChange.AddressRange == "" {
networkChange.AddressRange = network.AddressRange
2021-03-26 00:17:52 +08:00
}
2021-05-01 11:07:25 +08:00
if networkChange.AddressRange6 == "" {
networkChange.AddressRange6 = network.AddressRange6
}
if networkChange.NetID == "" {
2021-04-16 02:48:10 +08:00
networkChange.NetID = network.NetID
2021-03-26 00:17:52 +08:00
}
2021-05-01 11:07:25 +08:00
err = ValidateNetworkUpdate(networkChange)
if err != nil {
returnErrorResponse(w, r, formatError(err, "badrequest"))
return
}
2021-05-04 05:51:38 +08:00
returnednetwork, err := UpdateNetwork(networkChange, network)
if err != nil {
returnErrorResponse(w, r, formatError(err, "badrequest"))
return
}
2021-03-26 00:17:52 +08:00
2021-05-04 05:51:38 +08:00
w.WriteHeader(http.StatusOK)
json.NewEncoder(w).Encode(returnednetwork)
}
func UpdateNetwork(networkChange models.NetworkUpdate, network models.Network) (models.Network, error) {
//NOTE: Network.NetID is intentionally NOT editable. It acts as a static ID for the network.
2021-04-13 11:19:01 +08:00
//DisplayName can be changed instead, which is what shows on the front end
2021-04-16 02:48:10 +08:00
if networkChange.NetID != network.NetID {
2021-05-04 05:51:38 +08:00
return models.Network{}, errors.New("NetID is not editable")
2021-04-16 02:48:10 +08:00
}
2021-03-26 00:17:52 +08:00
2021-05-04 05:51:38 +08:00
haschange := false
hasrangeupdate := false
haslocalrangeupdate := false
2021-03-26 00:17:52 +08:00
2021-05-04 05:51:38 +08:00
if networkChange.AddressRange != "" {
2021-04-16 02:48:10 +08:00
haschange = true
hasrangeupdate = true
2021-05-04 05:51:38 +08:00
network.AddressRange = networkChange.AddressRange
2021-04-16 02:48:10 +08:00
}
2021-04-13 22:04:31 +08:00
if networkChange.LocalRange != "" {
2021-04-16 02:48:10 +08:00
haschange = true
haslocalrangeupdate = true
2021-05-04 05:51:38 +08:00
network.LocalRange = networkChange.LocalRange
2021-04-13 11:19:01 +08:00
}
2021-04-13 22:04:31 +08:00
if networkChange.IsLocal != nil {
network.IsLocal = networkChange.IsLocal
2021-04-13 11:19:01 +08:00
}
2021-05-01 11:07:25 +08:00
if networkChange.IsDualStack != nil {
network.IsDualStack = networkChange.IsDualStack
}
if networkChange.DefaultListenPort != 0 {
network.DefaultListenPort = networkChange.DefaultListenPort
2021-04-13 11:19:01 +08:00
haschange = true
2021-04-16 02:48:10 +08:00
}
if networkChange.DefaultPostDown != "" {
2021-04-13 14:55:49 +08:00
network.DefaultPostDown = networkChange.DefaultPostDown
2021-04-13 11:19:01 +08:00
haschange = true
2021-04-16 02:48:10 +08:00
}
if networkChange.DefaultInterface != "" {
network.DefaultInterface = networkChange.DefaultInterface
2021-04-13 11:19:01 +08:00
haschange = true
2021-04-16 02:48:10 +08:00
}
if networkChange.DefaultPostUp != "" {
network.DefaultPostUp = networkChange.DefaultPostUp
2021-04-13 11:19:01 +08:00
haschange = true
2021-04-16 02:48:10 +08:00
}
if networkChange.DefaultKeepalive != 0 {
network.DefaultKeepalive = networkChange.DefaultKeepalive
2021-04-13 11:19:01 +08:00
haschange = true
2021-04-16 02:48:10 +08:00
}
if networkChange.DisplayName != "" {
network.DisplayName = networkChange.DisplayName
2021-04-13 11:19:01 +08:00
haschange = true
2021-04-16 02:48:10 +08:00
}
if networkChange.DefaultCheckInInterval != 0 {
network.DefaultCheckInInterval = networkChange.DefaultCheckInInterval
2021-04-13 11:19:01 +08:00
haschange = true
2021-04-16 02:48:10 +08:00
}
if networkChange.AllowManualSignUp != nil {
network.AllowManualSignUp = networkChange.AllowManualSignUp
2021-04-13 11:19:01 +08:00
haschange = true
2021-04-16 02:48:10 +08:00
}
2021-03-26 00:17:52 +08:00
2021-04-16 02:48:10 +08:00
collection := mongoconn.Client.Database("netmaker").Collection("networks")
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
2021-05-04 05:51:38 +08:00
filter := bson.M{"netid": network.NetID}
2021-03-26 00:17:52 +08:00
if haschange {
network.SetNetworkLastModified()
2021-03-26 00:17:52 +08:00
}
2021-05-01 11:07:25 +08:00
// prepare update model.
update := bson.D{
{"$set", bson.D{
{"addressrange", network.AddressRange},
{"addressrange6", network.AddressRange6},
{"displayname", network.DisplayName},
{"defaultlistenport", network.DefaultListenPort},
{"defaultpostup", network.DefaultPostUp},
{"defaultpostdown", network.DefaultPostDown},
{"defaultkeepalive", network.DefaultKeepalive},
{"defaultsaveconfig", network.DefaultSaveConfig},
{"defaultinterface", network.DefaultInterface},
{"nodeslastmodified", network.NodesLastModified},
{"networklastmodified", network.NetworkLastModified},
{"allowmanualsignup", network.AllowManualSignUp},
{"localrange", network.LocalRange},
{"islocal", network.IsLocal},
{"isdualstack", network.IsDualStack},
{"checkininterval", network.DefaultCheckInInterval},
}},
2021-04-16 02:48:10 +08:00
}
2021-03-26 00:17:52 +08:00
2021-05-04 05:51:38 +08:00
err := collection.FindOneAndUpdate(ctx, filter, update).Decode(&network)
2021-04-16 02:48:10 +08:00
defer cancel()
2021-03-26 00:17:52 +08:00
2021-04-16 02:48:10 +08:00
if err != nil {
2021-05-04 05:51:38 +08:00
return models.Network{}, err
2021-04-16 02:48:10 +08:00
}
2021-03-26 00:17:52 +08:00
//Cycles through nodes and gives them new IP's based on the new range
//Pretty cool, but also pretty inefficient currently
2021-04-16 02:48:10 +08:00
if hasrangeupdate {
2021-05-04 05:51:38 +08:00
err = functions.UpdateNetworkNodeAddresses(network.NetID)
2021-04-13 11:19:01 +08:00
if err != nil {
2021-05-04 05:51:38 +08:00
return models.Network{}, err
2021-04-13 11:19:01 +08:00
}
2021-03-26 00:17:52 +08:00
}
2021-04-13 22:04:31 +08:00
if haslocalrangeupdate {
2021-05-04 05:51:38 +08:00
err = functions.UpdateNetworkPrivateAddresses(network.NetID)
2021-04-16 02:48:10 +08:00
if err != nil {
2021-05-04 05:51:38 +08:00
return models.Network{}, err
2021-04-16 02:48:10 +08:00
}
2021-04-13 11:19:01 +08:00
}
returnnetwork, err := functions.GetParentNetwork(network.NetID)
2021-04-16 02:48:10 +08:00
if err != nil {
2021-05-04 05:51:38 +08:00
return models.Network{}, err
2021-04-16 02:48:10 +08:00
}
2021-05-04 05:51:38 +08:00
return returnnetwork, nil
2021-03-26 00:17:52 +08:00
}
//Delete a network
2021-03-26 00:17:52 +08:00
//Will stop you if there's any nodes associated
func deleteNetwork(w http.ResponseWriter, r *http.Request) {
2021-04-16 02:48:10 +08:00
// Set header
w.Header().Set("Content-Type", "application/json")
2021-03-26 00:17:52 +08:00
2021-04-16 02:48:10 +08:00
var params = mux.Vars(r)
2021-05-01 11:07:25 +08:00
network := params["networkname"]
count, err := DeleteNetwork(network)
2021-03-26 00:17:52 +08:00
2021-04-13 11:19:01 +08:00
if err != nil {
2021-05-28 22:04:07 +08:00
errtype := "badrequest"
if strings.Contains(err.Error(), "Node check failed"){
errtype = "forbidden"
}
returnErrorResponse(w, r, formatError(err, errtype))
2021-04-16 02:48:10 +08:00
return
}
2021-05-01 11:07:25 +08:00
w.WriteHeader(http.StatusOK)
json.NewEncoder(w).Encode(count)
}
2021-03-26 00:17:52 +08:00
2021-05-01 11:07:25 +08:00
func DeleteNetwork(network string) (*mongo.DeleteResult, error) {
none := &mongo.DeleteResult{}
2021-03-26 00:17:52 +08:00
2021-05-01 11:07:25 +08:00
nodecount, err := functions.GetNetworkNodeNumber(network)
if err != nil {
//returnErrorResponse(w, r, formatError(err, "internal"))
return none, err
} else if nodecount > 0 {
//errorResponse := models.ErrorResponse{
// Code: http.StatusForbidden, Message: "W1R3: Node check failed. All nodes must be deleted before deleting network.",
//}
//returnErrorResponse(w, r, errorResponse)
return none, errors.New("Node check failed. All nodes must be deleted before deleting network")
}
2021-03-26 00:17:52 +08:00
2021-05-01 11:07:25 +08:00
collection := mongoconn.Client.Database("netmaker").Collection("networks")
filter := bson.M{"netid": network}
2021-04-16 02:48:10 +08:00
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
deleteResult, err := collection.DeleteOne(ctx, filter)
defer cancel()
if err != nil {
2021-05-01 11:07:25 +08:00
//returnErrorResponse(w, r, formatError(err, "internal"))
return none, err
2021-04-16 02:48:10 +08:00
}
2021-05-01 11:07:25 +08:00
return deleteResult, nil
2021-03-26 00:17:52 +08:00
}
//Create a network
2021-03-26 00:17:52 +08:00
//Pretty simple
func createNetwork(w http.ResponseWriter, r *http.Request) {
2021-03-26 00:17:52 +08:00
2021-04-16 02:48:10 +08:00
w.Header().Set("Content-Type", "application/json")
2021-03-26 00:17:52 +08:00
2021-04-16 02:48:10 +08:00
var network models.Network
2021-03-26 00:17:52 +08:00
2021-04-16 02:48:10 +08:00
// we decode our body request params
err := json.NewDecoder(r.Body).Decode(&network)
2021-04-16 02:48:10 +08:00
if err != nil {
returnErrorResponse(w, r, formatError(err, "internal"))
return
}
2021-03-26 00:17:52 +08:00
2021-05-01 11:07:25 +08:00
err = CreateNetwork(network)
if err != nil {
returnErrorResponse(w, r, formatError(err, "badrequest"))
return
}
w.WriteHeader(http.StatusOK)
//json.NewEncoder(w).Encode(result)
}
func CreateNetwork(network models.Network) error {
//TODO: Not really doing good validation here. Same as createNode, updateNode, and updateNetwork
2021-03-26 00:17:52 +08:00
//Need to implement some better validation across the board
2021-04-16 19:34:48 +08:00
if network.IsLocal == nil {
falsevar := false
network.IsLocal = &falsevar
}
2021-05-01 11:07:25 +08:00
if network.IsDualStack == nil {
falsevar := false
network.IsDualStack = &falsevar
}
2021-03-26 00:17:52 +08:00
2021-05-01 11:07:25 +08:00
err := ValidateNetworkCreate(network)
2021-04-16 02:48:10 +08:00
if err != nil {
2021-05-01 11:07:25 +08:00
//returnErrorResponse(w, r, formatError(err, "badrequest"))
return err
2021-04-16 02:48:10 +08:00
}
network.SetDefaults()
network.SetNodesLastModified()
network.SetNetworkLastModified()
network.KeyUpdateTimeStamp = time.Now().Unix()
2021-03-26 00:17:52 +08:00
2021-04-16 02:48:10 +08:00
collection := mongoconn.Client.Database("netmaker").Collection("networks")
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
2021-03-26 00:17:52 +08:00
2021-04-16 02:48:10 +08:00
// insert our network into the network table
2021-05-01 19:57:49 +08:00
_, err = collection.InsertOne(ctx, network)
2021-04-16 02:48:10 +08:00
defer cancel()
if err != nil {
2021-05-01 11:07:25 +08:00
return err
2021-04-16 02:48:10 +08:00
}
2021-05-01 11:07:25 +08:00
return nil
2021-03-26 00:17:52 +08:00
}
// BEGIN KEY MANAGEMENT SECTION
//TODO: Very little error handling
//accesskey is created as a json string inside the Network collection item in mongo
2021-03-26 00:17:52 +08:00
func createAccessKey(w http.ResponseWriter, r *http.Request) {
2021-04-16 02:48:10 +08:00
w.Header().Set("Content-Type", "application/json")
var params = mux.Vars(r)
var accesskey models.AccessKey
//start here
network, err := functions.GetParentNetwork(params["networkname"])
2021-04-16 02:48:10 +08:00
if err != nil {
returnErrorResponse(w, r, formatError(err, "internal"))
return
}
err = json.NewDecoder(r.Body).Decode(&accesskey)
if err != nil {
returnErrorResponse(w, r, formatError(err, "internal"))
return
}
2021-05-01 19:57:49 +08:00
key, err := CreateAccessKey(accesskey, network)
if err != nil {
returnErrorResponse(w, r, formatError(err, "badrequest"))
return
}
w.WriteHeader(http.StatusOK)
json.NewEncoder(w).Encode(key)
//w.Write([]byte(accesskey.AccessString))
}
2021-03-26 00:17:52 +08:00
2021-05-01 19:57:49 +08:00
func CreateAccessKey(accesskey models.AccessKey, network models.Network) (models.AccessKey, error) {
2021-05-28 22:04:07 +08:00
2021-03-26 00:17:52 +08:00
if accesskey.Name == "" {
2021-04-16 02:48:10 +08:00
accesskey.Name = functions.GenKeyName()
}
2021-05-28 22:04:07 +08:00
2021-03-26 00:17:52 +08:00
if accesskey.Value == "" {
accesskey.Value = functions.GenKey()
}
if accesskey.Uses == 0 {
accesskey.Uses = 1
}
2021-05-28 22:04:07 +08:00
checkkeys, err := GetKeys(network.NetID)
if err != nil {
return models.AccessKey{}, errors.New("could not retrieve network keys")
}
for _, key := range checkkeys {
2021-05-01 19:57:49 +08:00
if key.Name == accesskey.Name {
return models.AccessKey{}, errors.New("Duplicate AccessKey Name")
}
}
privAddr := ""
2021-05-01 19:57:49 +08:00
if network.IsLocal != nil {
if *network.IsLocal {
privAddr = network.LocalRange
}
}
2021-05-10 01:31:50 +08:00
netID := network.NetID
grpcaddress := net.JoinHostPort(servercfg.GetGRPCHost(), servercfg.GetGRPCPort())
apiaddress := net.JoinHostPort(servercfg.GetAPIHost(), servercfg.GetAPIPort())
2021-06-02 03:32:20 +08:00
wgport := servercfg.GetGRPCWGPort()
2021-06-02 03:28:01 +08:00
accessstringdec := wgport + "|" +grpcaddress + "|" + apiaddress + "|" + netID + "|" + accesskey.Value + "|" + privAddr
accesskey.AccessString = base64.StdEncoding.EncodeToString([]byte(accessstringdec))
2021-05-01 19:57:49 +08:00
//validate accesskey
v := validator.New()
2021-05-28 22:04:07 +08:00
err = v.Struct(accesskey)
2021-05-01 19:57:49 +08:00
if err != nil {
for _, e := range err.(validator.ValidationErrors) {
fmt.Println(e)
}
return models.AccessKey{}, err
}
network.AccessKeys = append(network.AccessKeys, accesskey)
2021-04-16 02:48:10 +08:00
collection := mongoconn.Client.Database("netmaker").Collection("networks")
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
// Create filter
2021-05-01 19:57:49 +08:00
filter := bson.M{"netid": network.NetID}
2021-04-16 02:48:10 +08:00
// Read update model from body request
// prepare update model.
update := bson.D{
{"$set", bson.D{
{"accesskeys", network.AccessKeys},
}},
}
err = collection.FindOneAndUpdate(ctx, filter, update).Decode(&network)
defer cancel()
2021-04-13 11:19:01 +08:00
if err != nil {
2021-05-01 19:57:49 +08:00
//returnErrorResponse(w, r, formatError(err, "internal"))
return models.AccessKey{}, err
2021-04-16 02:48:10 +08:00
}
2021-05-01 19:57:49 +08:00
return accesskey, nil
2021-03-26 00:17:52 +08:00
}
2021-05-26 00:48:04 +08:00
func GetSignupToken(netID string) (models.AccessKey, error) {
var accesskey models.AccessKey
address := net.JoinHostPort(servercfg.GetGRPCHost(), servercfg.GetGRPCPort())
2021-05-26 00:48:04 +08:00
accessstringdec := address + "|" + netID + "|" + "" + "|"
accesskey.AccessString = base64.StdEncoding.EncodeToString([]byte(accessstringdec))
return accesskey, nil
}
func getSignupToken(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "application/json")
var params = mux.Vars(r)
netID := params["networkname"]
token, err := GetSignupToken(netID)
if err != nil {
returnErrorResponse(w, r, formatError(err, "internal"))
return
}
w.WriteHeader(http.StatusOK)
json.NewEncoder(w).Encode(token)
}
2021-03-26 00:17:52 +08:00
//pretty simple get
func getAccessKeys(w http.ResponseWriter, r *http.Request) {
2021-04-16 02:48:10 +08:00
w.Header().Set("Content-Type", "application/json")
var params = mux.Vars(r)
2021-05-01 19:57:49 +08:00
network := params["networkname"]
keys, err := GetKeys(network)
if err != nil {
returnErrorResponse(w, r, formatError(err, "internal"))
return
}
w.WriteHeader(http.StatusOK)
json.NewEncoder(w).Encode(keys)
}
func GetKeys(net string) ([]models.AccessKey, error) {
2021-03-26 00:17:52 +08:00
2021-04-16 02:48:10 +08:00
var network models.Network
collection := mongoconn.Client.Database("netmaker").Collection("networks")
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
2021-05-01 19:57:49 +08:00
filter := bson.M{"netid": net}
2021-04-16 02:48:10 +08:00
err := collection.FindOne(ctx, filter, options.FindOne().SetProjection(bson.M{"_id": 0})).Decode(&network)
defer cancel()
if err != nil {
2021-05-01 19:57:49 +08:00
return []models.AccessKey{}, err
2021-04-16 02:48:10 +08:00
}
2021-05-01 19:57:49 +08:00
return network.AccessKeys, nil
2021-03-26 00:17:52 +08:00
}
//delete key. Has to do a little funky logic since it's not a collection item
func deleteAccessKey(w http.ResponseWriter, r *http.Request) {
2021-04-16 02:48:10 +08:00
w.Header().Set("Content-Type", "application/json")
var params = mux.Vars(r)
2021-03-26 00:17:52 +08:00
keyname := params["name"]
2021-05-01 19:57:49 +08:00
netname := params["networkname"]
err := DeleteKey(keyname, netname)
2021-04-16 02:48:10 +08:00
if err != nil {
2021-05-01 19:57:49 +08:00
returnErrorResponse(w, r, formatError(err, "badrequest"))
2021-04-16 02:48:10 +08:00
return
}
2021-05-01 19:57:49 +08:00
w.WriteHeader(http.StatusOK)
}
func DeleteKey(keyname, netname string) error {
network, err := functions.GetParentNetwork(netname)
if err != nil {
return err
}
2021-03-26 00:17:52 +08:00
//basically, turn the list of access keys into the list of access keys before and after the item
//have not done any error handling for if there's like...1 item. I think it works? need to test.
2021-04-18 04:52:53 +08:00
found := false
2021-05-01 19:57:49 +08:00
var updatedKeys []models.AccessKey
for _, currentkey := range network.AccessKeys {
2021-03-26 00:17:52 +08:00
if currentkey.Name == keyname {
2021-04-18 04:52:53 +08:00
found = true
2021-05-01 19:57:49 +08:00
} else {
updatedKeys = append(updatedKeys, currentkey)
2021-03-26 00:17:52 +08:00
}
}
2021-04-18 04:52:53 +08:00
if !found {
2021-05-01 19:57:49 +08:00
return errors.New("key " + keyname + " does not exist")
2021-04-18 04:52:53 +08:00
}
2021-03-26 00:17:52 +08:00
2021-04-16 02:48:10 +08:00
collection := mongoconn.Client.Database("netmaker").Collection("networks")
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
// Create filter
2021-05-01 19:57:49 +08:00
filter := bson.M{"netid": netname}
2021-04-16 02:48:10 +08:00
// prepare update model.
update := bson.D{
{"$set", bson.D{
2021-05-01 19:57:49 +08:00
{"accesskeys", updatedKeys},
2021-04-16 02:48:10 +08:00
}},
}
err = collection.FindOneAndUpdate(ctx, filter, update).Decode(&network)
defer cancel()
if err != nil {
2021-05-01 19:57:49 +08:00
return err
2021-04-16 02:48:10 +08:00
}
2021-05-01 19:57:49 +08:00
return nil
2021-03-26 00:17:52 +08:00
}