From 336ee7b185b52f478c5e59407ab6940c19105029 Mon Sep 17 00:00:00 2001
From: Benoit G <benoit.georgelin@yulpa.io>
Date: Sat, 16 Nov 2019 18:38:24 -0500
Subject: [PATCH] Add c.nbrErrors checks

---
 conn.go   | 5 +++++
 server.go | 5 +++++
 2 files changed, 10 insertions(+)

diff --git a/conn.go b/conn.go
index ce228101..7620f8a9 100644
--- a/conn.go
+++ b/conn.go
@@ -99,6 +99,11 @@ func (c *Conn) handle(cmd string, arg string) {
 	}()
 
 	if cmd == "" {
+		c.nbrErrors++
+		if c.nbrErrors > 3 {
+			c.WriteResponse(500, EnhancedCode{5, 5, 2}, "Too many errors")
+			c.Close()
+		}
 		c.WriteResponse(500, EnhancedCode{5, 5, 2}, "Speak up")
 		return
 	}
diff --git a/server.go b/server.go
index 645db061..33202931 100755
--- a/server.go
+++ b/server.go
@@ -135,6 +135,11 @@ func (s *Server) handleConn(c *Conn) error {
 			cmd, arg, err := parseCmd(line)
 			if err != nil {
 				c.nbrErrors++
+				if c.nbrErrors > 3 {
+					c.WriteResponse(500, EnhancedCode{5, 5, 2}, "Too many errors")
+					c.Close()
+				}
+
 				c.WriteResponse(501, EnhancedCode{5, 5, 2}, "Bad command")
 				continue
 			}