fix bug in websocket library

nuclearace · nuclearace · commit 6ee1daeaf7df · 2015-08-17T00:09:04.000-04:00
diff --git 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
             }

Original file line number	Diff line number	Diff line change
`@@ -495,7 +495,7 @@ public class WebSocket : NSObject, NSStreamDelegate {`
`495`	`495`	`dataLength = UInt64(bytes[0].bigEndian)`
`496`	`496`	`offset += sizeof(UInt16)`
`497`	`497`	`}`
`498`		`- if bufferLen < offset {`
	`498`	`+ if bufferLen < offset \|\| UInt64(bufferLen - offset) < dataLength {`
`499`	`499`	`fragBuffer = NSData(bytes: buffer, length: bufferLen)`
`500`	`500`	`return`
`501`	`501`	`}`