diff options
| -rw-r--r-- | cmd/vhost/main.go | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/cmd/vhost/main.go b/cmd/vhost/main.go index 6a78b97..ca4700e 100644 --- a/cmd/vhost/main.go +++ b/cmd/vhost/main.go @@ -141,7 +141,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) @@ -166,7 +170,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 { |