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 {
319319 let length = inputStream!. read ( buffer, maxLength: BUFFER_MAX)
320320 if length > 0 {
321321 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) ) )
325328 }
326329 } else {
327330 var process = false
@@ -372,7 +375,6 @@ public class WebSocket : NSObject, NSStreamDelegate {
372375 if validateResponse ( buffer, bufferLen: totalSize) {
373376 dispatch_async ( queue, { [ weak self] in
374377 guard let s = self else { return }
375- s. connected = true
376378 if let connectBlock = s. onConnect {
377379 connectBlock ( )
378380 }
You can’t perform that action at this time.
0 commit comments