From 06dbe57ab216e52f7cf7e38edbfc86ce794a766c Mon Sep 17 00:00:00 2001
From: Lan Tian <xuyh0120@outlook.com>
Date: Tue, 27 Oct 2020 23:09:55 +0800
Subject: [PATCH] proxy: fix IP limit handling

---
 proxy/main.go | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/proxy/main.go b/proxy/main.go
index 15a5b6b..4ae5129 100644
--- a/proxy/main.go
+++ b/proxy/main.go
@@ -25,7 +25,7 @@ func accessHandler(next http.Handler) http.Handler {
 	return http.HandlerFunc(func(httpW http.ResponseWriter, httpR *http.Request) {
 
 		// setting.allowedIPs will always have at least one element because of how it's defined
-		if setting.allowedIPs[0] == ""  {
+		if setting.allowedIPs[0] == "" {
 			next.ServeHTTP(httpW, httpR)
 		}
 
@@ -39,6 +39,7 @@ func accessHandler(next http.Handler) http.Handler {
 		for _, allowedIP := range setting.allowedIPs {
 			if requestIp == allowedIP {
 				next.ServeHTTP(httpW, httpR)
+				return
 			}
 		}
 
@@ -51,7 +52,7 @@ type settingType struct {
 	birdSocket  string
 	bird6Socket string
 	listen      string
-	allowedIPs	[]string
+	allowedIPs  []string
 }
 
 var setting settingType
@@ -99,4 +100,3 @@ func main() {
 	http.HandleFunc("/traceroute6", tracerouteIPv6Wrapper)
 	http.ListenAndServe(*listenParam, handlers.LoggingHandler(os.Stdout, accessHandler(http.DefaultServeMux)))
 }
-