123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869 |
- package controller
- import (
- "context"
- "encoding/base64"
- "encoding/json"
- "errors"
- "fmt"
- "net/http"
- "strings"
- "time"
- "github.com/go-playground/validator/v10"
- "github.com/gorilla/mux"
- "github.com/gravitl/netmaker/functions"
- "github.com/gravitl/netmaker/models"
- "github.com/gravitl/netmaker/mongoconn"
- "github.com/gravitl/netmaker/servercfg"
- "go.mongodb.org/mongo-driver/bson"
- "go.mongodb.org/mongo-driver/mongo"
- "go.mongodb.org/mongo-driver/mongo/options"
- )
- func networkHandlers(r *mux.Router) {
- r.HandleFunc("/api/networks", securityCheck(true, http.HandlerFunc(getNetworks))).Methods("GET")
- r.HandleFunc("/api/networks", securityCheck(true, http.HandlerFunc(createNetwork))).Methods("POST")
- r.HandleFunc("/api/networks/{networkname}", securityCheck(false, http.HandlerFunc(getNetwork))).Methods("GET")
- r.HandleFunc("/api/networks/{networkname}", securityCheck(false, http.HandlerFunc(updateNetwork))).Methods("PUT")
- r.HandleFunc("/api/networks/{networkname}/nodelimit", securityCheck(true, http.HandlerFunc(updateNetworkNodeLimit))).Methods("PUT")
- r.HandleFunc("/api/networks/{networkname}", securityCheck(true, http.HandlerFunc(deleteNetwork))).Methods("DELETE")
- r.HandleFunc("/api/networks/{networkname}/keyupdate", securityCheck(false, http.HandlerFunc(keyUpdate))).Methods("POST")
- r.HandleFunc("/api/networks/{networkname}/keys", securityCheck(false, http.HandlerFunc(createAccessKey))).Methods("POST")
- r.HandleFunc("/api/networks/{networkname}/keys", securityCheck(false, http.HandlerFunc(getAccessKeys))).Methods("GET")
- r.HandleFunc("/api/networks/{networkname}/signuptoken", securityCheck(false, http.HandlerFunc(getSignupToken))).Methods("GET")
- r.HandleFunc("/api/networks/{networkname}/keys/{name}", securityCheck(false, http.HandlerFunc(deleteAccessKey))).Methods("DELETE")
- }
- //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
- //or maybe some Users once implemented
- func securityCheck(reqAdmin bool, next http.Handler) http.HandlerFunc {
- return func(w http.ResponseWriter, r *http.Request) {
- var errorResponse = models.ErrorResponse{
- Code: http.StatusUnauthorized, Message: "W1R3: It's not you it's me.",
- }
- var params = mux.Vars(r)
- bearerToken := r.Header.Get("Authorization")
- err := SecurityCheck(reqAdmin, params["networkname"], bearerToken)
- if err != nil {
- if strings.Contains(err.Error(), "does not exist") {
- errorResponse.Code = http.StatusNotFound
- }
- errorResponse.Message = err.Error()
- returnErrorResponse(w, r, errorResponse)
- return
- }
- next.ServeHTTP(w, r)
- }
- }
- func SecurityCheck(reqAdmin bool, 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")
- }
- var hasBearer = true
- var tokenSplit = strings.Split(token, " ")
- var authToken = ""
- if len(tokenSplit) < 2 {
- hasBearer = false
- } else {
- authToken = tokenSplit[1]
- }
- //all endpoints here require master so not as complicated
- if !hasBearer || !authenticateMaster(authToken) {
- _, networks, isadmin, err := functions.VerifyUserToken(authToken)
- if err != nil {
- return errors.New("Error verifying user token")
- }
- if !isadmin && reqAdmin {
- return errors.New("You are unauthorized to access this endpoint")
- } else if !isadmin && netname != ""{
- if !functions.SliceContains(networks, netname){
- return errors.New("You are unauthorized to access this endpoint")
- }
- } else if !isadmin {
- return errors.New("You are unauthorized to access this endpoint")
- }
- }
- return nil
- }
- //Consider a more secure way of setting master key
- func authenticateMaster(tokenString string) bool {
- if tokenString == servercfg.GetMasterKey() {
- return true
- }
- return false
- }
- //simple get all networks function
- func getNetworks(w http.ResponseWriter, r *http.Request) {
- allnetworks, err := functions.ListNetworks()
- 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:]...)
- }
- func ValidateNetworkUpdate(network models.NetworkUpdate) error {
- v := validator.New()
- // _ = 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
- // })
- // _ = 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 {
- // return true
- // })
- // _ = 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
- }
- func ValidateNetworkCreate(network models.Network) error {
- v := validator.New()
- // _ = 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 {
- isvalid := true
- if *network.IsDualStack {
- isvalid = functions.IsIpCIDR(fl.Field().String())
- }
- return isvalid
- })
- //
- // _ = 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())
- inCharSet := functions.NameInNetworkCharSet(fl.Field().String())
- return isFieldUnique && inCharSet
- })
- //
- _ = v.RegisterValidation("displayname_valid", func(fl validator.FieldLevel) bool {
- isFieldUnique, _ := functions.IsNetworkDisplayNameUnique(fl.Field().String())
- inCharSet := functions.NameInNetworkCharSet(fl.Field().String())
- return isFieldUnique && inCharSet
- })
- err := v.Struct(network)
- if err != nil {
- for _, e := range err.(validator.ValidationErrors) {
- fmt.Println(e)
- }
- }
- return err
- }
- //Simple get network function
- func getNetwork(w http.ResponseWriter, r *http.Request) {
- // set header.
- w.Header().Set("Content-Type", "application/json")
- var params = mux.Vars(r)
- 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)
- }
- func GetNetwork(name string) (models.Network, error) {
- var network models.Network
- collection := mongoconn.Client.Database("netmaker").Collection("networks")
- ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
- filter := bson.M{"netid": name}
- err := collection.FindOne(ctx, filter, options.FindOne().SetProjection(bson.M{"_id": 0})).Decode(&network)
- defer cancel()
- if err != nil {
- return models.Network{}, err
- }
- return network, nil
- }
- func keyUpdate(w http.ResponseWriter, r *http.Request) {
- w.Header().Set("Content-Type", "application/json")
- var params = mux.Vars(r)
- netname := params["networkname"]
- network, err := KeyUpdate(netname)
- if err != nil {
- returnErrorResponse(w, r, formatError(err, "internal"))
- return
- }
- w.WriteHeader(http.StatusOK)
- json.NewEncoder(w).Encode(network)
- }
- func KeyUpdate(netname string) (models.Network, error) {
- network, err := functions.GetParentNetwork(netname)
- if err != nil {
- return models.Network{}, err
- }
- network.KeyUpdateTimeStamp = time.Now().Unix()
- collection := mongoconn.Client.Database("netmaker").Collection("networks")
- ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
- filter := bson.M{"netid": netname}
- // 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},
- {"keyupdatetimestamp", network.KeyUpdateTimeStamp},
- {"defaultsaveconfig", network.DefaultSaveConfig},
- {"defaultinterface", network.DefaultInterface},
- {"nodeslastmodified", network.NodesLastModified},
- {"networklastmodified", network.NetworkLastModified},
- {"allowmanualsignup", network.AllowManualSignUp},
- {"checkininterval", network.DefaultCheckInInterval},
- }},
- }
- err = collection.FindOneAndUpdate(ctx, filter, update).Decode(&network)
- defer cancel()
- if err != nil {
- return models.Network{}, err
- }
- return network, nil
- }
- //Update a network
- func AlertNetwork(netid string) error {
- collection := mongoconn.Client.Database("netmaker").Collection("networks")
- ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
- filter := bson.M{"netid": netid}
- var network models.Network
- network, err := functions.GetParentNetwork(netid)
- if err != nil {
- return err
- }
- updatetime := time.Now().Unix()
- update := bson.D{
- {"$set", bson.D{
- {"nodeslastmodified", updatetime},
- {"networklastmodified", updatetime},
- }},
- }
- err = collection.FindOneAndUpdate(ctx, filter, update).Decode(&network)
- defer cancel()
- return err
- }
- //Update a network
- func updateNetwork(w http.ResponseWriter, r *http.Request) {
- w.Header().Set("Content-Type", "application/json")
- var params = mux.Vars(r)
- var network models.Network
- network, err := functions.GetParentNetwork(params["networkname"])
- if err != nil {
- returnErrorResponse(w, r, formatError(err, "internal"))
- return
- }
- var networkChange models.NetworkUpdate
- _ = json.NewDecoder(r.Body).Decode(&networkChange)
- if networkChange.AddressRange == "" {
- networkChange.AddressRange = network.AddressRange
- }
- if networkChange.AddressRange6 == "" {
- networkChange.AddressRange6 = network.AddressRange6
- }
- if networkChange.NetID == "" {
- networkChange.NetID = network.NetID
- }
- err = ValidateNetworkUpdate(networkChange)
- if err != nil {
- returnErrorResponse(w, r, formatError(err, "badrequest"))
- return
- }
- returnednetwork, err := UpdateNetwork(networkChange, network)
- if err != nil {
- returnErrorResponse(w, r, formatError(err, "badrequest"))
- return
- }
- w.WriteHeader(http.StatusOK)
- json.NewEncoder(w).Encode(returnednetwork)
- }
- func updateNetworkNodeLimit(w http.ResponseWriter, r *http.Request) {
- w.Header().Set("Content-Type", "application/json")
- var params = mux.Vars(r)
- var network models.Network
- network, err := functions.GetParentNetwork(params["networkname"])
- if err != nil {
- returnErrorResponse(w, r, formatError(err, "internal"))
- return
- }
- var networkChange models.NetworkUpdate
- _ = json.NewDecoder(r.Body).Decode(&networkChange)
- collection := mongoconn.Client.Database("netmaker").Collection("networks")
- ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
- filter := bson.M{"netid": network.NetID}
- if networkChange.NodeLimit !=0 {
- update := bson.D{
- {"$set", bson.D{
- {"nodelimit", networkChange.NodeLimit},
- }},
- }
- err := collection.FindOneAndUpdate(ctx, filter, update).Decode(&network)
- defer cancel()
- if err != nil {
- returnErrorResponse(w, r, formatError(err, "badrequest"))
- return
- }
- }
- w.WriteHeader(http.StatusOK)
- json.NewEncoder(w).Encode(network)
- }
- 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.
- //DisplayName can be changed instead, which is what shows on the front end
- if networkChange.NetID != network.NetID {
- return models.Network{}, errors.New("NetID is not editable")
- }
- haschange := false
- hasrangeupdate := false
- haslocalrangeupdate := false
- if networkChange.AddressRange != "" {
- haschange = true
- hasrangeupdate = true
- network.AddressRange = networkChange.AddressRange
- }
- if networkChange.LocalRange != "" {
- haschange = true
- haslocalrangeupdate = true
- network.LocalRange = networkChange.LocalRange
- }
- if networkChange.IsLocal != nil {
- network.IsLocal = networkChange.IsLocal
- }
- if networkChange.IsDualStack != nil {
- network.IsDualStack = networkChange.IsDualStack
- }
- if networkChange.DefaultListenPort != 0 {
- network.DefaultListenPort = networkChange.DefaultListenPort
- haschange = true
- }
- if networkChange.DefaultPostDown != "" {
- network.DefaultPostDown = networkChange.DefaultPostDown
- haschange = true
- }
- if networkChange.DefaultInterface != "" {
- network.DefaultInterface = networkChange.DefaultInterface
- haschange = true
- }
- if networkChange.DefaultPostUp != "" {
- network.DefaultPostUp = networkChange.DefaultPostUp
- haschange = true
- }
- if networkChange.DefaultKeepalive != 0 {
- network.DefaultKeepalive = networkChange.DefaultKeepalive
- haschange = true
- }
- if networkChange.DisplayName != "" {
- network.DisplayName = networkChange.DisplayName
- haschange = true
- }
- if networkChange.DefaultCheckInInterval != 0 {
- network.DefaultCheckInInterval = networkChange.DefaultCheckInInterval
- haschange = true
- }
- if networkChange.AllowManualSignUp != nil {
- network.AllowManualSignUp = networkChange.AllowManualSignUp
- haschange = true
- }
- collection := mongoconn.Client.Database("netmaker").Collection("networks")
- ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
- filter := bson.M{"netid": network.NetID}
- if haschange {
- network.SetNetworkLastModified()
- }
- // 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},
- }},
- }
- err := collection.FindOneAndUpdate(ctx, filter, update).Decode(&network)
- defer cancel()
- if err != nil {
- return models.Network{}, err
- }
- //Cycles through nodes and gives them new IP's based on the new range
- //Pretty cool, but also pretty inefficient currently
- if hasrangeupdate {
- err = functions.UpdateNetworkNodeAddresses(network.NetID)
- if err != nil {
- return models.Network{}, err
- }
- }
- if haslocalrangeupdate {
- err = functions.UpdateNetworkPrivateAddresses(network.NetID)
- if err != nil {
- return models.Network{}, err
- }
- }
- returnnetwork, err := functions.GetParentNetwork(network.NetID)
- if err != nil {
- return models.Network{}, err
- }
- return returnnetwork, nil
- }
- //Delete a network
- //Will stop you if there's any nodes associated
- func deleteNetwork(w http.ResponseWriter, r *http.Request) {
- // Set header
- w.Header().Set("Content-Type", "application/json")
- var params = mux.Vars(r)
- network := params["networkname"]
- count, err := DeleteNetwork(network)
- if err != nil {
- errtype := "badrequest"
- if strings.Contains(err.Error(), "Node check failed"){
- errtype = "forbidden"
- }
- returnErrorResponse(w, r, formatError(err, errtype))
- return
- }
- w.WriteHeader(http.StatusOK)
- json.NewEncoder(w).Encode(count)
- }
- func DeleteNetwork(network string) (*mongo.DeleteResult, error) {
- none := &mongo.DeleteResult{}
- 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")
- }
- collection := mongoconn.Client.Database("netmaker").Collection("networks")
- filter := bson.M{"netid": network}
- ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
- deleteResult, err := collection.DeleteOne(ctx, filter)
- defer cancel()
- if err != nil {
- //returnErrorResponse(w, r, formatError(err, "internal"))
- return none, err
- }
- return deleteResult, nil
- }
- //Create a network
- //Pretty simple
- func createNetwork(w http.ResponseWriter, r *http.Request) {
- w.Header().Set("Content-Type", "application/json")
- var network models.Network
- // we decode our body request params
- err := json.NewDecoder(r.Body).Decode(&network)
- if err != nil {
- returnErrorResponse(w, r, formatError(err, "internal"))
- return
- }
- 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
- //Need to implement some better validation across the board
- if network.IsLocal == nil {
- falsevar := false
- network.IsLocal = &falsevar
- }
- if network.IsDualStack == nil {
- falsevar := false
- network.IsDualStack = &falsevar
- }
- err := ValidateNetworkCreate(network)
- if err != nil {
- //returnErrorResponse(w, r, formatError(err, "badrequest"))
- return err
- }
- network.SetDefaults()
- network.SetNodesLastModified()
- network.SetNetworkLastModified()
- network.KeyUpdateTimeStamp = time.Now().Unix()
- collection := mongoconn.Client.Database("netmaker").Collection("networks")
- ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
- // insert our network into the network table
- _, err = collection.InsertOne(ctx, network)
- defer cancel()
- if err != nil {
- return err
- }
- return nil
- }
- // BEGIN KEY MANAGEMENT SECTION
- //TODO: Very little error handling
- //accesskey is created as a json string inside the Network collection item in mongo
- func createAccessKey(w http.ResponseWriter, r *http.Request) {
- w.Header().Set("Content-Type", "application/json")
- var params = mux.Vars(r)
- var accesskey models.AccessKey
- //start here
- network, err := functions.GetParentNetwork(params["networkname"])
- 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
- }
- 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))
- }
- func CreateAccessKey(accesskey models.AccessKey, network models.Network) (models.AccessKey, error) {
- if accesskey.Name == "" {
- accesskey.Name = functions.GenKeyName()
- }
- if accesskey.Value == "" {
- accesskey.Value = functions.GenKey()
- }
- if accesskey.Uses == 0 {
- accesskey.Uses = 1
- }
- checkkeys, err := GetKeys(network.NetID)
- if err != nil {
- return models.AccessKey{}, errors.New("could not retrieve network keys")
- }
- for _, key := range checkkeys {
- if key.Name == accesskey.Name {
- return models.AccessKey{}, errors.New("Duplicate AccessKey Name")
- }
- }
- privAddr := ""
- if network.IsLocal != nil {
- if *network.IsLocal {
- privAddr = network.LocalRange
- }
- }
- netID := network.NetID
- var accessToken models.AccessToken
- s := servercfg.GetServerConfig()
- w := servercfg.GetWGConfig()
- servervals := models.ServerConfig{
- APIConnString: s.APIConnString,
- APIHost: s.APIHost,
- APIPort: s.APIPort,
- GRPCConnString: s.GRPCConnString,
- GRPCHost: s.GRPCHost,
- GRPCPort: s.GRPCPort,
- GRPCSSL: s.GRPCSSL,
- }
- wgvals := models.WG{
- GRPCWireGuard: w.GRPCWireGuard,
- GRPCWGAddress: w.GRPCWGAddress,
- GRPCWGPort: w.GRPCWGPort,
- GRPCWGPubKey: w.GRPCWGPubKey,
- GRPCWGEndpoint: s.APIHost,
- }
- accessToken.ServerConfig = servervals
- accessToken.WG = wgvals
- accessToken.ClientConfig.Network = netID
- accessToken.ClientConfig.Key = accesskey.Value
- accessToken.ClientConfig.LocalRange = privAddr
- tokenjson, err := json.Marshal(accessToken)
- if err != nil {
- return accesskey, err
- }
- accesskey.AccessString = base64.StdEncoding.EncodeToString([]byte(tokenjson))
- //validate accesskey
- v := validator.New()
- err = v.Struct(accesskey)
- if err != nil {
- for _, e := range err.(validator.ValidationErrors) {
- fmt.Println(e)
- }
- return models.AccessKey{}, err
- }
- network.AccessKeys = append(network.AccessKeys, accesskey)
- collection := mongoconn.Client.Database("netmaker").Collection("networks")
- ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
- // Create filter
- filter := bson.M{"netid": network.NetID}
- // 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()
- if err != nil {
- //returnErrorResponse(w, r, formatError(err, "internal"))
- return models.AccessKey{}, err
- }
- return accesskey, nil
- }
- func GetSignupToken(netID string) (models.AccessKey, error) {
- var accesskey models.AccessKey
- var accessToken models.AccessToken
- s := servercfg.GetServerConfig()
- w := servercfg.GetWGConfig()
- servervals := models.ServerConfig{
- APIConnString: s.APIConnString,
- APIHost: s.APIHost,
- APIPort: s.APIPort,
- GRPCConnString: s.GRPCConnString,
- GRPCHost: s.GRPCHost,
- GRPCPort: s.GRPCPort,
- GRPCSSL: s.GRPCSSL,
- }
- wgvals := models.WG{
- GRPCWireGuard: w.GRPCWireGuard,
- GRPCWGAddress: w.GRPCWGAddress,
- GRPCWGPort: w.GRPCWGPort,
- GRPCWGPubKey: w.GRPCWGPubKey,
- GRPCWGEndpoint: s.APIHost,
- }
- accessToken.ServerConfig = servervals
- accessToken.WG = wgvals
- tokenjson, err := json.Marshal(accessToken)
- if err != nil {
- return accesskey, err
- }
- accesskey.AccessString = base64.StdEncoding.EncodeToString([]byte(tokenjson))
- 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)
- }
- //pretty simple get
- func getAccessKeys(w http.ResponseWriter, r *http.Request) {
- w.Header().Set("Content-Type", "application/json")
- var params = mux.Vars(r)
- 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) {
- var network models.Network
- collection := mongoconn.Client.Database("netmaker").Collection("networks")
- ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
- filter := bson.M{"netid": net}
- err := collection.FindOne(ctx, filter, options.FindOne().SetProjection(bson.M{"_id": 0})).Decode(&network)
- defer cancel()
- if err != nil {
- return []models.AccessKey{}, err
- }
- return network.AccessKeys, nil
- }
- //delete key. Has to do a little funky logic since it's not a collection item
- func deleteAccessKey(w http.ResponseWriter, r *http.Request) {
- w.Header().Set("Content-Type", "application/json")
- var params = mux.Vars(r)
- keyname := params["name"]
- netname := params["networkname"]
- err := DeleteKey(keyname, netname)
- if err != nil {
- returnErrorResponse(w, r, formatError(err, "badrequest"))
- return
- }
- w.WriteHeader(http.StatusOK)
- }
- func DeleteKey(keyname, netname string) error {
- network, err := functions.GetParentNetwork(netname)
- if err != nil {
- return err
- }
- //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.
- found := false
- var updatedKeys []models.AccessKey
- for _, currentkey := range network.AccessKeys {
- if currentkey.Name == keyname {
- found = true
- } else {
- updatedKeys = append(updatedKeys, currentkey)
- }
- }
- if !found {
- return errors.New("key " + keyname + " does not exist")
- }
- collection := mongoconn.Client.Database("netmaker").Collection("networks")
- ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
- // Create filter
- filter := bson.M{"netid": netname}
- // prepare update model.
- update := bson.D{
- {"$set", bson.D{
- {"accesskeys", updatedKeys},
- }},
- }
- err = collection.FindOneAndUpdate(ctx, filter, update).Decode(&network)
- defer cancel()
- if err != nil {
- return err
- }
- return nil
- }
|