Fix: rate limit check

internal/netshare/netshare_add.go

@@ -21,7 +21,6 @@ package netshare
 import (
 	"git.lcomrade.su/root/lenmonitor/internal/storage"
 	"git.lcomrade.su/root/lenmonitor/internal/tester"
-	"net"
 	"net/http"
 	"strings"
 )
@@ -35,9 +34,8 @@ func AddServer(req *http.Request, db storage.DB, rateLimit *RateLimit) (string,
 	}
 
 	// Check rate limit
-	cleanIP, _, err := net.SplitHostPort(req.RemoteAddr)
-	if err != nil {
-		return "", "", err
+	if rateLimit.CheckAndUse(GetClientAddr(req)) == false {
+		return "", "", ErrTooManyRequests
 	}
 
 	// Get domain and service
@@ -53,10 +51,6 @@ func AddServer(req *http.Request, db storage.DB, rateLimit *RateLimit) (string,
 		return "", "", ErrBadRequest
 	}
 
-	if rateLimit.CheckAndUse(cleanIP) == false {
-		return "", "", ErrTooManyRequests
-	}
-
 	// Add server
 	switch srv {
 	case "lenpaste":