fix bug in websocket library

2015-08-17 00:09:04 -04:00 · 2015-08-17 00:09:04 -04:00 · 9c3d991bed
commit 9c3d991bed
parent 09a26c0b1c
1 changed files with 1 additions and 1 deletions
--- a/SocketIOClientSwift/WebSocket.swift
+++ b/SocketIOClientSwift/WebSocket.swift
@ -495,7 +495,7 @@ public class WebSocket : NSObject, NSStreamDelegate {
                dataLength = UInt64(bytes[0].bigEndian)
                offset += sizeof(UInt16)
            }
-            if bufferLen < offset {
+            if bufferLen < offset || UInt64(bufferLen - offset) < dataLength {
                fragBuffer = NSData(bytes: buffer, length: bufferLen)
                return
            }