1 files changed, 12 insertions, 0 deletions
diff --git a/ws.go b/ws.go
index 39f89d0..72b78af 100644
--- a/ws.go
+++ b/ws.go
@@ -292,6 +292,10 @@ func handleConn(
 					if err != nil {
 						return fmt.Errorf("error replying to Y: %w", err)
 					}
+					err = c.Close(websocket.StatusProtocolError, "")
+					if err != nil {
+						return fmt.Errorf("error closing websocket: %w", err)
+					}
 				}
 			case "N":
 				if len(mar) != 2 {
@@ -299,6 +303,10 @@ func handleConn(
 					if err != nil {
 						return fmt.Errorf("error replying to N: %w", err)
 					}
+					err = c.Close(websocket.StatusProtocolError, "")
+					if err != nil {
+						return fmt.Errorf("error closing websocket: %w", err)
+					}
 				}
 			default:
 				err := c.Write(
@@ -309,6 +317,10 @@ func handleConn(
 				if err != nil {
 					return fmt.Errorf("error replying to unknown command: %w", err)
 				}
+				err = c.Close(websocket.StatusProtocolError, "")
+				if err != nil {
+					return fmt.Errorf("error closing websocket: %w", err)
+				}
 			}
 		}
 	}