Merge pull request #6 from linyows/error-same-ip

Close connection if same IP
linyows · May 7, 2022 · 703fd9c · 703fd9c
2 parents f04d522 + 6cc0ccd
commit 703fd9c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/server.go b/server.go
@@ -41,6 +41,11 @@ func (s *Server) Start() error {
 			log.Printf("accept error: %#v", err)
 			continue
 		}
+		if s.Addr == conn.RemoteAddr().String() {
+			conn.Close()
+			log.Printf("closed connection due to same ip: %s", conn.RemoteAddr())
+			continue
+		}
 		go s.HandleConnection(conn)
 	}
 }