diff options
| author | David Doan <daviddoan@Davids-MacBook-Pro-70.local> | 2023-11-08 17:33:13 -0500 |
|---|---|---|
| committer | David Doan <daviddoan@Davids-MacBook-Pro-70.local> | 2023-11-08 17:33:13 -0500 |
| commit | 421cb40b2b533d24347f3fb07067c5c5a3708fba (patch) | |
| tree | d0a747bb85288f02377a5987de745ab5e1f33900 | |
| parent | f8ea46a281d473f5db097b25755321d87e9686e4 (diff) | |
repl fleuncy
| -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 { |