update

main.go

@@ -204,12 +204,15 @@ func main() {
 					if err != nil {
 						log.Fatalf("Error reading from connection: %v\n", err)
 					}
+					if n == 0 {
+						break
+					}
 					str := string(buf[:n])
 					if str == "exit" {
 						break
 					}
 					log.Println("Echoing", str)
-					_, err = c.Write(buf[:n])
+					_, err = c.Write([]byte(str))
 					if err != nil {
 						log.Fatalf("Error writing to connection: %v\n", err)
 					}