kelvin
/
shrooms-server
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Write newlines to trigger read error out on a closed connection

This commit is contained in:
Kelvin Ly 2023-05-17 07:22:44 -04:00
parent 738c52984e
commit 811ed951e1
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
shroom_internals/tcp_server.go
View File

@ -250,13 +250,19 @@ func InitTcpServer(db *sql.DB, state *ShroomState) {
left = buf left = buf
} }
_, err := conn.Write([]byte{}) _, err := conn.Write([]byte("\n"))
if err != nil { if err != nil {
log.Println("tcp write error: ", err) log.Println("tcp write error: ", err)
_ = conn.Close() _ = conn.Close()
log.Println("disconnected from client") log.Println("disconnected from client")
break break
} }
err = conn.SetReadDeadline(time.Now().Add(time.Second))
if err != nil {
log.Println("unable to set read deadline ", err)
_ = conn.Close()
break
}
num_read, err := conn.Read(left) num_read, err := conn.Read(left)
left = left[num_read:] left = left[num_read:]