Ver código fonte

Forbid responses if unlicensed ee and not in status api

gabrielseibel1 2 anos atrás
pai
commit
ec6f6dbb7f
1 arquivos alterados com 2 adições e 3 exclusões
  1. 2 3
      ee/ee_controllers/middleware.go

+ 2 - 3
ee/ee_controllers/middleware.go

@@ -4,13 +4,12 @@ import (
 	"github.com/gravitl/netmaker/logic"
 	"github.com/gravitl/netmaker/servercfg"
 	"net/http"
-	"strings"
 )
 
 func OnlyServerAPIWhenUnlicensedMiddleware(handler http.Handler) http.Handler {
 	return http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
-		if servercfg.ErrLicenseValidation != nil && !strings.HasPrefix(request.URL.Path, "/api/server") {
-			logic.ReturnErrorResponse(writer, request, logic.FormatError(servercfg.ErrLicenseValidation, "unauthorized"))
+		if servercfg.ErrLicenseValidation != nil && request.URL.Path != "/api/server/status" {
+			logic.ReturnErrorResponse(writer, request, logic.FormatError(servercfg.ErrLicenseValidation, "forbidden"))
 			return
 		}
 		handler.ServeHTTP(writer, request)