File tree Expand file tree Collapse file tree 1 file changed +6
-4
lines changed Expand file tree Collapse file tree 1 file changed +6
-4
lines changed Original file line number Diff line number Diff line change @@ -319,9 +319,12 @@ public class WebSocket : NSObject, NSStreamDelegate {
319
319
let length = inputStream!. read ( buffer, maxLength: BUFFER_MAX)
320
320
if length > 0 {
321
321
if !connected {
322
- let status = processHTTP ( buffer, bufferLen: length)
323
- if !status {
324
- doDisconnect ( errorWithDetail ( " Invalid HTTP upgrade " , code: 1 ) )
322
+ connected = processHTTP ( buffer, bufferLen: length)
323
+ if !connected {
324
+ let response = CFHTTPMessageCreateEmpty ( kCFAllocatorDefault, false ) . takeRetainedValue ( )
325
+ CFHTTPMessageAppendBytes ( response, buffer, length)
326
+ let code = CFHTTPMessageGetResponseStatusCode ( response)
327
+ doDisconnect ( errorWithDetail ( " Invalid HTTP upgrade " , code: UInt16 ( code) ) )
325
328
}
326
329
} else {
327
330
var process = false
@@ -372,7 +375,6 @@ public class WebSocket : NSObject, NSStreamDelegate {
372
375
if validateResponse ( buffer, bufferLen: totalSize) {
373
376
dispatch_async ( queue, { [ weak self] in
374
377
guard let s = self else { return }
375
- s. connected = true
376
378
if let connectBlock = s. onConnect {
377
379
connectBlock ( )
378
380
}
You can’t perform that action at this time.
0 commit comments