From 421cb40b2b533d24347f3fb07067c5c5a3708fba Mon Sep 17 00:00:00 2001 From: David Doan Date: Wed, 8 Nov 2023 17:33:13 -0500 Subject: repl fleuncy --- cmd/vhost/main.go | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/cmd/vhost/main.go b/cmd/vhost/main.go index 7c32df8..1603a7e 100644 --- a/cmd/vhost/main.go +++ b/cmd/vhost/main.go @@ -143,7 +143,11 @@ func main() { fmt.Println(err) continue } - conn, _ := sockets[socketID] + conn, inMap := sockets[socketID] + if !inMap { + fmt.Println("Invalid socket ID") + continue + } message := tokens[2:] messageToSend := strings.Join(message, " ") messageToSendBytes := []byte(messageToSend) @@ -168,7 +172,11 @@ func main() { fmt.Println(err) continue } - conn, _ := sockets[socketID] + conn, inMap := sockets[socketID] + if !inMap { + fmt.Println("Invalid socket ID") + continue + } // buffer := make([]byte, numberOfBytes) bytesRead, message, err := conn.VRead(numberOfBytes) if err != nil { -- cgit v1.2.3-70-g09d2