|
@@ -1,20 +1,21 @@
|
|
|
package controller
|
|
|
|
|
|
import (
|
|
|
- "github.com/gravitl/netmaker/models"
|
|
|
- "github.com/gravitl/netmaker/functions"
|
|
|
- "github.com/gravitl/netmaker/serverctl"
|
|
|
- "github.com/gravitl/netmaker/servercfg"
|
|
|
- "encoding/json"
|
|
|
- "strings"
|
|
|
- "net/http"
|
|
|
- "github.com/gorilla/mux"
|
|
|
+ "encoding/json"
|
|
|
+ "net/http"
|
|
|
+ "strings"
|
|
|
+
|
|
|
+ "github.com/gorilla/mux"
|
|
|
+ "github.com/gravitl/netmaker/functions"
|
|
|
+ "github.com/gravitl/netmaker/models"
|
|
|
+ "github.com/gravitl/netmaker/servercfg"
|
|
|
+ "github.com/gravitl/netmaker/serverctl"
|
|
|
)
|
|
|
|
|
|
func serverHandlers(r *mux.Router) {
|
|
|
- r.HandleFunc("/api/server/addnetwork/{network}", securityCheckServer(true, http.HandlerFunc(addNetwork))).Methods("POST")
|
|
|
- r.HandleFunc("/api/server/getconfig", securityCheckServer(false, http.HandlerFunc(getConfig))).Methods("GET")
|
|
|
- r.HandleFunc("/api/server/removenetwork/{network}", securityCheckServer(true, http.HandlerFunc(removeNetwork))).Methods("DELETE")
|
|
|
+ r.HandleFunc("/api/server/addnetwork/{network}", securityCheckServer(true, http.HandlerFunc(addNetwork))).Methods("POST")
|
|
|
+ r.HandleFunc("/api/server/getconfig", securityCheckServer(false, http.HandlerFunc(getConfig))).Methods("GET")
|
|
|
+ r.HandleFunc("/api/server/removenetwork/{network}", securityCheckServer(true, http.HandlerFunc(removeNetwork))).Methods("DELETE")
|
|
|
}
|
|
|
|
|
|
//Security check is middleware for every function and just checks to make sure that its the master calling
|
|
@@ -29,93 +30,83 @@ func securityCheckServer(adminonly bool, next http.Handler) http.HandlerFunc {
|
|
|
bearerToken := r.Header.Get("Authorization")
|
|
|
|
|
|
var tokenSplit = strings.Split(bearerToken, " ")
|
|
|
- var authToken = ""
|
|
|
+ var authToken = ""
|
|
|
if len(tokenSplit) < 2 {
|
|
|
- errorResponse = models.ErrorResponse{
|
|
|
- Code: http.StatusUnauthorized, Message: "W1R3: You are unauthorized to access this endpoint.",
|
|
|
- }
|
|
|
- returnErrorResponse(w, r, errorResponse)
|
|
|
- return
|
|
|
- } else {
|
|
|
+ errorResponse = models.ErrorResponse{
|
|
|
+ Code: http.StatusUnauthorized, Message: "W1R3: You are unauthorized to access this endpoint.",
|
|
|
+ }
|
|
|
+ returnErrorResponse(w, r, errorResponse)
|
|
|
+ return
|
|
|
+ } else {
|
|
|
authToken = tokenSplit[1]
|
|
|
}
|
|
|
//all endpoints here require master so not as complicated
|
|
|
//still might not be a good way of doing this
|
|
|
- user, _, isadmin, err := functions.VerifyUserToken(authToken)
|
|
|
- errorResponse = models.ErrorResponse{
|
|
|
- Code: http.StatusUnauthorized, Message: "W1R3: You are unauthorized to access this endpoint.",
|
|
|
- }
|
|
|
- if !adminonly && (err != nil || user == "") {
|
|
|
- returnErrorResponse(w, r, errorResponse)
|
|
|
- }
|
|
|
- if !isadmin && !authenticateMasterServer(authToken) {
|
|
|
- returnErrorResponse(w, r, errorResponse)
|
|
|
- }
|
|
|
+ user, _, isadmin, err := functions.VerifyUserToken(authToken)
|
|
|
+ errorResponse = models.ErrorResponse{
|
|
|
+ Code: http.StatusUnauthorized, Message: "W1R3: You are unauthorized to access this endpoint.",
|
|
|
+ }
|
|
|
+ if !adminonly && (err != nil || user == "") {
|
|
|
+ returnErrorResponse(w, r, errorResponse)
|
|
|
+ return
|
|
|
+ }
|
|
|
+ if adminonly && !isadmin && !authenticateMasterServer(authToken) {
|
|
|
+ returnErrorResponse(w, r, errorResponse)
|
|
|
+ return
|
|
|
+ }
|
|
|
next.ServeHTTP(w, r)
|
|
|
}
|
|
|
}
|
|
|
+
|
|
|
//Consider a more secure way of setting master key
|
|
|
func authenticateMasterServer(tokenString string) bool {
|
|
|
- if tokenString == servercfg.GetMasterKey() {
|
|
|
- return true
|
|
|
- }
|
|
|
- return false
|
|
|
+ if tokenString == servercfg.GetMasterKey() {
|
|
|
+ return true
|
|
|
+ }
|
|
|
+ return false
|
|
|
}
|
|
|
|
|
|
func removeNetwork(w http.ResponseWriter, r *http.Request) {
|
|
|
- // Set header
|
|
|
- w.Header().Set("Content-Type", "application/json")
|
|
|
+ // Set header
|
|
|
+ w.Header().Set("Content-Type", "application/json")
|
|
|
|
|
|
- // get params
|
|
|
- var params = mux.Vars(r)
|
|
|
+ // get params
|
|
|
+ var params = mux.Vars(r)
|
|
|
|
|
|
- success, err := serverctl.RemoveNetwork(params["network"])
|
|
|
+ success, err := serverctl.RemoveNetwork(params["network"])
|
|
|
|
|
|
- if err != nil || !success {
|
|
|
- json.NewEncoder(w).Encode("Could not remove server from network " + params["network"])
|
|
|
- return
|
|
|
- }
|
|
|
+ if err != nil || !success {
|
|
|
+ json.NewEncoder(w).Encode("Could not remove server from network " + params["network"])
|
|
|
+ return
|
|
|
+ }
|
|
|
|
|
|
- json.NewEncoder(w).Encode("Server removed from network " + params["network"])
|
|
|
+ json.NewEncoder(w).Encode("Server removed from network " + params["network"])
|
|
|
}
|
|
|
|
|
|
func getConfig(w http.ResponseWriter, r *http.Request) {
|
|
|
// Set header
|
|
|
- w.Header().Set("Content-Type", "application/json")
|
|
|
+ w.Header().Set("Content-Type", "application/json")
|
|
|
|
|
|
- // get params
|
|
|
+ // get params
|
|
|
|
|
|
- scfg := servercfg.GetServerConfig()
|
|
|
- w.WriteHeader(http.StatusOK)
|
|
|
- json.NewEncoder(w).Encode(scfg)
|
|
|
+ scfg := servercfg.GetServerConfig()
|
|
|
+ json.NewEncoder(w).Encode(scfg)
|
|
|
+ //w.WriteHeader(http.StatusOK)
|
|
|
}
|
|
|
|
|
|
-/*
|
|
|
-func getMongoConfig(w http.ResponseWriter, r *http.Request) {
|
|
|
- // Set header
|
|
|
- w.Header().Set("Content-Type", "application/json")
|
|
|
-
|
|
|
- // get params
|
|
|
-
|
|
|
- mcfg := servercfg.GetMongoConfig()
|
|
|
- w.WriteHeader(http.StatusOK)
|
|
|
- json.NewEncoder(w).Encode(mcfg)
|
|
|
-}
|
|
|
-*/
|
|
|
-
|
|
|
func addNetwork(w http.ResponseWriter, r *http.Request) {
|
|
|
- // Set header
|
|
|
- w.Header().Set("Content-Type", "application/json")
|
|
|
+ // Set header
|
|
|
+ w.Header().Set("Content-Type", "application/json")
|
|
|
|
|
|
- // get params
|
|
|
- var params = mux.Vars(r)
|
|
|
+ // get params
|
|
|
+ var params = mux.Vars(r)
|
|
|
|
|
|
- success, err := serverctl.AddNetwork(params["network"])
|
|
|
+ success, err := serverctl.AddNetwork(params["network"])
|
|
|
|
|
|
- if err != nil || !success {
|
|
|
- json.NewEncoder(w).Encode("Could not add server to network " + params["network"])
|
|
|
- return
|
|
|
- }
|
|
|
+ if err != nil || !success {
|
|
|
+ json.NewEncoder(w).Encode("Could not add server to network " + params["network"])
|
|
|
+ return
|
|
|
+ }
|
|
|
|
|
|
- json.NewEncoder(w).Encode("Server added to network " + params["network"])
|
|
|
+ json.NewEncoder(w).Encode("Server added to network " + params["network"])
|
|
|
}
|