|
@@ -7,6 +7,7 @@ import (
|
|
|
"time"
|
|
|
|
|
|
"github.com/gorilla/mux"
|
|
|
+
|
|
|
"github.com/gravitl/netmaker/auth"
|
|
|
"github.com/gravitl/netmaker/logger"
|
|
|
"github.com/gravitl/netmaker/logic"
|
|
@@ -17,10 +18,14 @@ import (
|
|
|
)
|
|
|
|
|
|
func enrollmentKeyHandlers(r *mux.Router) {
|
|
|
- r.HandleFunc("/api/v1/enrollment-keys", logic.SecurityCheck(true, http.HandlerFunc(createEnrollmentKey))).Methods(http.MethodPost)
|
|
|
- r.HandleFunc("/api/v1/enrollment-keys", logic.SecurityCheck(true, http.HandlerFunc(getEnrollmentKeys))).Methods(http.MethodGet)
|
|
|
- r.HandleFunc("/api/v1/enrollment-keys/{keyID}", logic.SecurityCheck(true, http.HandlerFunc(deleteEnrollmentKey))).Methods(http.MethodDelete)
|
|
|
- r.HandleFunc("/api/v1/host/register/{token}", http.HandlerFunc(handleHostRegister)).Methods(http.MethodPost)
|
|
|
+ r.HandleFunc("/api/v1/enrollment-keys", logic.SecurityCheck(true, http.HandlerFunc(createEnrollmentKey))).
|
|
|
+ Methods(http.MethodPost)
|
|
|
+ r.HandleFunc("/api/v1/enrollment-keys", logic.SecurityCheck(true, http.HandlerFunc(getEnrollmentKeys))).
|
|
|
+ Methods(http.MethodGet)
|
|
|
+ r.HandleFunc("/api/v1/enrollment-keys/{keyID}", logic.SecurityCheck(true, http.HandlerFunc(deleteEnrollmentKey))).
|
|
|
+ Methods(http.MethodDelete)
|
|
|
+ r.HandleFunc("/api/v1/host/register/{token}", http.HandlerFunc(handleHostRegister)).
|
|
|
+ Methods(http.MethodPost)
|
|
|
}
|
|
|
|
|
|
// swagger:route GET /api/v1/enrollment-keys enrollmentKeys getEnrollmentKeys
|
|
@@ -70,7 +75,7 @@ func getEnrollmentKeys(w http.ResponseWriter, r *http.Request) {
|
|
|
// Responses:
|
|
|
// 200: okResponse
|
|
|
func deleteEnrollmentKey(w http.ResponseWriter, r *http.Request) {
|
|
|
- var params = mux.Vars(r)
|
|
|
+ params := mux.Vars(r)
|
|
|
keyID := params["keyID"]
|
|
|
err := logic.DeleteEnrollmentKey(keyID)
|
|
|
if err != nil {
|
|
@@ -94,7 +99,6 @@ func deleteEnrollmentKey(w http.ResponseWriter, r *http.Request) {
|
|
|
// Responses:
|
|
|
// 200: EnrollmentKey
|
|
|
func createEnrollmentKey(w http.ResponseWriter, r *http.Request) {
|
|
|
-
|
|
|
var enrollmentKeyBody models.APIEnrollmentKey
|
|
|
|
|
|
err := json.NewDecoder(r.Body).Decode(&enrollmentKeyBody)
|
|
@@ -109,7 +113,13 @@ func createEnrollmentKey(w http.ResponseWriter, r *http.Request) {
|
|
|
newTime = time.Unix(enrollmentKeyBody.Expiration, 0)
|
|
|
}
|
|
|
|
|
|
- newEnrollmentKey, err := logic.CreateEnrollmentKey(enrollmentKeyBody.UsesRemaining, newTime, enrollmentKeyBody.Networks, enrollmentKeyBody.Tags, enrollmentKeyBody.Unlimited)
|
|
|
+ newEnrollmentKey, err := logic.CreateEnrollmentKey(
|
|
|
+ enrollmentKeyBody.UsesRemaining,
|
|
|
+ newTime,
|
|
|
+ enrollmentKeyBody.Networks,
|
|
|
+ enrollmentKeyBody.Tags,
|
|
|
+ enrollmentKeyBody.Unlimited,
|
|
|
+ )
|
|
|
if err != nil {
|
|
|
logger.Log(0, r.Header.Get("user"), "failed to create enrollment key:", err.Error())
|
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
@@ -138,7 +148,7 @@ func createEnrollmentKey(w http.ResponseWriter, r *http.Request) {
|
|
|
// Responses:
|
|
|
// 200: RegisterResponse
|
|
|
func handleHostRegister(w http.ResponseWriter, r *http.Request) {
|
|
|
- var params = mux.Vars(r)
|
|
|
+ params := mux.Vars(r)
|
|
|
token := params["token"]
|
|
|
logger.Log(0, "received registration attempt with token", token)
|
|
|
// check if token exists
|
|
@@ -156,7 +166,6 @@ func handleHostRegister(w http.ResponseWriter, r *http.Request) {
|
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
|
return
|
|
|
}
|
|
|
- hostExists := false
|
|
|
// re-register host with turn just in case.
|
|
|
if servercfg.IsUsingTurn() {
|
|
|
err = logic.RegisterHostWithTurn(newHost.ID.String(), newHost.HostPass)
|
|
@@ -165,9 +174,20 @@ func handleHostRegister(w http.ResponseWriter, r *http.Request) {
|
|
|
}
|
|
|
}
|
|
|
// check if host already exists
|
|
|
+ hostExists := false
|
|
|
if hostExists = logic.HostExists(&newHost); hostExists && len(enrollmentKey.Networks) == 0 {
|
|
|
- logger.Log(0, "host", newHost.ID.String(), newHost.Name, "attempted to re-register with no networks")
|
|
|
- logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("host already exists"), "badrequest"))
|
|
|
+ logger.Log(
|
|
|
+ 0,
|
|
|
+ "host",
|
|
|
+ newHost.ID.String(),
|
|
|
+ newHost.Name,
|
|
|
+ "attempted to re-register with no networks",
|
|
|
+ )
|
|
|
+ logic.ReturnErrorResponse(
|
|
|
+ w,
|
|
|
+ r,
|
|
|
+ logic.FormatError(fmt.Errorf("host already exists"), "badrequest"),
|
|
|
+ )
|
|
|
return
|
|
|
}
|
|
|
// version check
|
|
@@ -190,11 +210,16 @@ func handleHostRegister(w http.ResponseWriter, r *http.Request) {
|
|
|
// use the token
|
|
|
if ok := logic.TryToUseEnrollmentKey(enrollmentKey); !ok {
|
|
|
logger.Log(0, "host", newHost.ID.String(), newHost.Name, "failed registration")
|
|
|
- logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("invalid enrollment key"), "badrequest"))
|
|
|
+ logic.ReturnErrorResponse(
|
|
|
+ w,
|
|
|
+ r,
|
|
|
+ logic.FormatError(fmt.Errorf("invalid enrollment key"), "badrequest"),
|
|
|
+ )
|
|
|
return
|
|
|
}
|
|
|
hostPass := newHost.HostPass
|
|
|
if !hostExists {
|
|
|
+ newHost.PersistentKeepalive = models.DefaultPersistentKeepAlive
|
|
|
// register host
|
|
|
logic.CheckHostPorts(&newHost)
|
|
|
// create EMQX credentials and ACLs for host
|
|
@@ -209,14 +234,21 @@ func handleHostRegister(w http.ResponseWriter, r *http.Request) {
|
|
|
}
|
|
|
}
|
|
|
if err = logic.CreateHost(&newHost); err != nil {
|
|
|
- logger.Log(0, "host", newHost.ID.String(), newHost.Name, "failed registration -", err.Error())
|
|
|
+ logger.Log(
|
|
|
+ 0,
|
|
|
+ "host",
|
|
|
+ newHost.ID.String(),
|
|
|
+ newHost.Name,
|
|
|
+ "failed registration -",
|
|
|
+ err.Error(),
|
|
|
+ )
|
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
|
return
|
|
|
}
|
|
|
} else {
|
|
|
// need to revise the list of networks from key
|
|
|
// based on the ones host currently has
|
|
|
- var networksToAdd = []string{}
|
|
|
+ networksToAdd := []string{}
|
|
|
currentNets := logic.GetHostNetworks(newHost.ID.String())
|
|
|
for _, newNet := range enrollmentKey.Networks {
|
|
|
if !logic.StringSliceContains(currentNets, newNet) {
|