Check err

refraction-networking · Feb 13, 2024 · 4b1c669 · 4b1c669
1 parent 0b4c945
commit 4b1c669
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pkg/dtls/heartbeat.go b/pkg/dtls/heartbeat.go
@@ -75,7 +75,12 @@ func (c *hbConn) recvLoop() {
 	for {
 		buffer := make([]byte, c.maxMessageSize)
 
-		c.stream.SetReadDeadline(time.Now().Add(c.timeout))
+		err := c.stream.SetReadDeadline(time.Now().Add(c.timeout))
+		if err != nil {
+			c.Close()
+			return
+		}
+
 		n, err := c.stream.Read(buffer)
 
 		if bytes.Equal(c.hb, buffer[:n]) {