fix bug in websocket library

2015-08-17 00:09:04 -04:00 · 2015-08-17 00:09:04 -04:00 · 6ee1daeaf7
commit 6ee1daeaf7
parent 0fd8a9540b
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
            }