diff options
Diffstat (limited to 'cmd')
| -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 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 { |