websocket: handle possible half write on Send()

When sending large payload, there is a possibility that a single Write
may not send all packet due to buffer or size limit on TCP.

1 files changed, 9 insertions, 2 deletions

diff --git a/lib/websocket/funcs.go b/lib/websocket/funcs.go
index a7cb2f93..ce215018 100644
--- a/lib/websocket/funcs.go
+++ b/lib/websocket/funcs.go
@@ -43,9 +43,16 @@ func Recv(fd int) (packet []byte, err error) {
 // Send the packet through web socket file descriptor `fd`.
 //
 func Send(fd int, packet []byte) (err error) {
-	_, err = unix.Write(fd, packet)
+	var n int
+	for len(packet) > 0 {
+		n, err = unix.Write(fd, packet)
+		if err != nil {
+			break
+		}
+		packet = packet[n:]
+	}
 
-	return
+	return err
 }
 
 //