merge master
This commit is contained in:
commit
362f4e90b8
@ -1,6 +1,6 @@
|
|||||||
Pod::Spec.new do |s|
|
Pod::Spec.new do |s|
|
||||||
s.name = "Socket.IO-Client-Swift"
|
s.name = "Socket.IO-Client-Swift"
|
||||||
s.version = "2.3.5"
|
s.version = "2.3.6"
|
||||||
s.summary = "Socket.IO-client for iOS and OS X"
|
s.summary = "Socket.IO-client for iOS and OS X"
|
||||||
s.description = <<-DESC
|
s.description = <<-DESC
|
||||||
Socket.IO-client for iOS and OS X.
|
Socket.IO-client for iOS and OS X.
|
||||||
@ -12,7 +12,7 @@ Pod::Spec.new do |s|
|
|||||||
s.author = { "Erik" => "nuclear.ace@gmail.com" }
|
s.author = { "Erik" => "nuclear.ace@gmail.com" }
|
||||||
s.ios.deployment_target = '8.0'
|
s.ios.deployment_target = '8.0'
|
||||||
s.osx.deployment_target = '10.10'
|
s.osx.deployment_target = '10.10'
|
||||||
s.source = { :git => "https://github.com/socketio/socket.io-client-swift.git", :tag => 'v2.3.5' }
|
s.source = { :git => "https://github.com/socketio/socket.io-client-swift.git", :tag => 'v2.3.6' }
|
||||||
s.source_files = "SocketIOClientSwift/**/*.swift"
|
s.source_files = "SocketIOClientSwift/**/*.swift"
|
||||||
s.requires_arc = true
|
s.requires_arc = true
|
||||||
# s.dependency 'Starscream', '~> 0.9' # currently this repo includes Starscream swift files
|
# s.dependency 'Starscream', '~> 0.9' # currently this repo includes Starscream swift files
|
||||||
|
|||||||
@ -27,24 +27,24 @@ import Foundation
|
|||||||
public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
||||||
private typealias Probe = (msg:String, type:PacketType, data:[NSData]?)
|
private typealias Probe = (msg:String, type:PacketType, data:[NSData]?)
|
||||||
private typealias ProbeWaitQueue = [Probe]
|
private typealias ProbeWaitQueue = [Probe]
|
||||||
|
|
||||||
private let allowedCharacterSet = NSCharacterSet(charactersInString: "!*'();:@&=+$,/?%#[]\" {}").invertedSet
|
private let allowedCharacterSet = NSCharacterSet(charactersInString: "!*'();:@&=+$,/?%#[]\" {}").invertedSet
|
||||||
private let workQueue = NSOperationQueue()
|
private let workQueue = NSOperationQueue()
|
||||||
private let emitQueue = dispatch_queue_create("engineEmitQueue", DISPATCH_QUEUE_SERIAL)
|
private let emitQueue = dispatch_queue_create("engineEmitQueue", DISPATCH_QUEUE_SERIAL)
|
||||||
private let parseQueue = dispatch_queue_create("engineParseQueue", DISPATCH_QUEUE_SERIAL)
|
private let parseQueue = dispatch_queue_create("engineParseQueue", DISPATCH_QUEUE_SERIAL)
|
||||||
private let handleQueue = dispatch_queue_create("engineHandleQueue", DISPATCH_QUEUE_SERIAL)
|
private let handleQueue = dispatch_queue_create("engineHandleQueue", DISPATCH_QUEUE_SERIAL)
|
||||||
private let session:NSURLSession!
|
private let session:NSURLSession!
|
||||||
|
|
||||||
private var closed = false
|
private var closed = false
|
||||||
private var _connected = false
|
private var _connected = false
|
||||||
private var fastUpgrade = false
|
private var fastUpgrade = false
|
||||||
private var forcePolling = false
|
private var forcePolling = false
|
||||||
private var forceWebsockets = false
|
private var forceWebsockets = false
|
||||||
private var pingInterval:Int?
|
private var pingInterval:Double?
|
||||||
private var pingTimer:NSTimer?
|
private var pingTimer:NSTimer?
|
||||||
private var pingTimeout = 0 {
|
private var pingTimeout = 0.0 {
|
||||||
didSet {
|
didSet {
|
||||||
pongsMissedMax = pingTimeout / (pingInterval ?? 25)
|
pongsMissedMax = Int(pingTimeout / (pingInterval ?? 25))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
private var pongsMissed = 0
|
private var pongsMissed = 0
|
||||||
@ -57,9 +57,9 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
private var waitingForPost = false
|
private var waitingForPost = false
|
||||||
private var _websocket = false
|
private var _websocket = false
|
||||||
private var websocketConnected = false
|
private var websocketConnected = false
|
||||||
|
|
||||||
let logType = "SocketEngine"
|
let logType = "SocketEngine"
|
||||||
|
|
||||||
var connected:Bool {
|
var connected:Bool {
|
||||||
return _connected
|
return _connected
|
||||||
}
|
}
|
||||||
@ -77,7 +77,7 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
return _websocket
|
return _websocket
|
||||||
}
|
}
|
||||||
var ws:WebSocket?
|
var ws:WebSocket?
|
||||||
|
|
||||||
public enum PacketType:Int {
|
public enum PacketType:Int {
|
||||||
case OPEN = 0
|
case OPEN = 0
|
||||||
case CLOSE = 1
|
case CLOSE = 1
|
||||||
@ -86,7 +86,7 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
case MESSAGE = 4
|
case MESSAGE = 4
|
||||||
case UPGRADE = 5
|
case UPGRADE = 5
|
||||||
case NOOP = 6
|
case NOOP = 6
|
||||||
|
|
||||||
init?(str:String?) {
|
init?(str:String?) {
|
||||||
if let value = Int(str ?? ""), raw = PacketType(rawValue: value) {
|
if let value = Int(str ?? ""), raw = PacketType(rawValue: value) {
|
||||||
self = raw
|
self = raw
|
||||||
@ -95,13 +95,13 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public init(client:SocketEngineClient, sessionDelegate:NSURLSessionDelegate?) {
|
public init(client:SocketEngineClient, sessionDelegate:NSURLSessionDelegate?) {
|
||||||
self.client = client
|
self.client = client
|
||||||
self.session = NSURLSession(configuration: NSURLSessionConfiguration.ephemeralSessionConfiguration(),
|
self.session = NSURLSession(configuration: NSURLSessionConfiguration.ephemeralSessionConfiguration(),
|
||||||
delegate: sessionDelegate, delegateQueue: workQueue)
|
delegate: sessionDelegate, delegateQueue: workQueue)
|
||||||
}
|
}
|
||||||
|
|
||||||
public convenience init(client:SocketEngineClient, opts:NSDictionary?) {
|
public convenience init(client:SocketEngineClient, opts:NSDictionary?) {
|
||||||
self.init(client: client, sessionDelegate: opts?["sessionDelegate"] as? NSURLSessionDelegate)
|
self.init(client: client, sessionDelegate: opts?["sessionDelegate"] as? NSURLSessionDelegate)
|
||||||
forceWebsockets = opts?["forceWebsockets"] as? Bool ?? false
|
forceWebsockets = opts?["forceWebsockets"] as? Bool ?? false
|
||||||
@ -110,56 +110,56 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
log = opts?["log"] as? Bool ?? false
|
log = opts?["log"] as? Bool ?? false
|
||||||
socketPath = opts?["path"] as? String ?? ""
|
socketPath = opts?["path"] as? String ?? ""
|
||||||
}
|
}
|
||||||
|
|
||||||
deinit {
|
deinit {
|
||||||
SocketLogger.log("Engine is being deinit", client: self)
|
SocketLogger.log("Engine is being deinit", client: self)
|
||||||
}
|
}
|
||||||
|
|
||||||
public func close(fast fast:Bool) {
|
public func close(fast fast:Bool) {
|
||||||
SocketLogger.log("Engine is being closed. Fast: %@", client: self, args: fast)
|
SocketLogger.log("Engine is being closed. Fast: %@", client: self, args: fast)
|
||||||
|
|
||||||
pingTimer?.invalidate()
|
pingTimer?.invalidate()
|
||||||
closed = true
|
closed = true
|
||||||
|
|
||||||
ws?.disconnect()
|
ws?.disconnect()
|
||||||
|
|
||||||
if fast || polling {
|
if fast || polling {
|
||||||
write("", withType: PacketType.CLOSE, withData: nil)
|
write("", withType: PacketType.CLOSE, withData: nil)
|
||||||
client?.engineDidClose("Disconnect")
|
client?.engineDidClose("Disconnect")
|
||||||
}
|
}
|
||||||
|
|
||||||
stopPolling()
|
stopPolling()
|
||||||
}
|
}
|
||||||
|
|
||||||
private func createBinaryDataForSend(data:NSData) -> (NSData?, String?) {
|
private func createBinaryDataForSend(data:NSData) -> (NSData?, String?) {
|
||||||
if websocket {
|
if websocket {
|
||||||
var byteArray = [UInt8](count: 1, repeatedValue: 0x0)
|
var byteArray = [UInt8](count: 1, repeatedValue: 0x0)
|
||||||
byteArray[0] = 4
|
byteArray[0] = 4
|
||||||
let mutData = NSMutableData(bytes: &byteArray, length: 1)
|
let mutData = NSMutableData(bytes: &byteArray, length: 1)
|
||||||
|
|
||||||
mutData.appendData(data)
|
mutData.appendData(data)
|
||||||
|
|
||||||
return (mutData, nil)
|
return (mutData, nil)
|
||||||
} else {
|
} else {
|
||||||
var str = "b4"
|
var str = "b4"
|
||||||
str += data.base64EncodedStringWithOptions(
|
str += data.base64EncodedStringWithOptions(
|
||||||
NSDataBase64EncodingOptions.Encoding64CharacterLineLength)
|
NSDataBase64EncodingOptions.Encoding64CharacterLineLength)
|
||||||
|
|
||||||
return (nil, str)
|
return (nil, str)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private func createURLs(params:[String: AnyObject]?) -> (String?, String?) {
|
private func createURLs(params:[String: AnyObject]?) -> (String?, String?) {
|
||||||
if client == nil {
|
if client == nil {
|
||||||
return (nil, nil)
|
return (nil, nil)
|
||||||
}
|
}
|
||||||
|
|
||||||
let path = socketPath == "" ? "/socket.io" : socketPath
|
let path = socketPath == "" ? "/socket.io" : socketPath
|
||||||
|
|
||||||
let url = "\(client!.socketURL)\(path)/?transport="
|
let url = "\(client!.socketURL)\(path)/?transport="
|
||||||
var urlPolling:String
|
var urlPolling:String
|
||||||
var urlWebSocket:String
|
var urlWebSocket:String
|
||||||
|
|
||||||
if client!.secure {
|
if client!.secure {
|
||||||
urlPolling = "https://" + url + "polling"
|
urlPolling = "https://" + url + "polling"
|
||||||
urlWebSocket = "wss://" + url + "websocket"
|
urlWebSocket = "wss://" + url + "websocket"
|
||||||
@ -167,15 +167,15 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
urlPolling = "http://" + url + "polling"
|
urlPolling = "http://" + url + "polling"
|
||||||
urlWebSocket = "ws://" + url + "websocket"
|
urlWebSocket = "ws://" + url + "websocket"
|
||||||
}
|
}
|
||||||
|
|
||||||
if params != nil {
|
if params != nil {
|
||||||
|
|
||||||
for (key, value) in params! {
|
for (key, value) in params! {
|
||||||
let keyEsc = key.stringByAddingPercentEncodingWithAllowedCharacters(
|
let keyEsc = key.stringByAddingPercentEncodingWithAllowedCharacters(
|
||||||
allowedCharacterSet)!
|
allowedCharacterSet)!
|
||||||
urlPolling += "&\(keyEsc)="
|
urlPolling += "&\(keyEsc)="
|
||||||
urlWebSocket += "&\(keyEsc)="
|
urlWebSocket += "&\(keyEsc)="
|
||||||
|
|
||||||
if value is String {
|
if value is String {
|
||||||
let valueEsc = (value as! String).stringByAddingPercentEncodingWithAllowedCharacters(
|
let valueEsc = (value as! String).stringByAddingPercentEncodingWithAllowedCharacters(
|
||||||
allowedCharacterSet)!
|
allowedCharacterSet)!
|
||||||
@ -187,27 +187,27 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return (urlPolling, urlWebSocket)
|
return (urlPolling, urlWebSocket)
|
||||||
}
|
}
|
||||||
|
|
||||||
private func createWebsocket(andConnect connect:Bool) {
|
private func createWebsocket(andConnect connect:Bool) {
|
||||||
ws = WebSocket(url: NSURL(string: urlWebSocket! + "&sid=\(sid)")!,
|
ws = WebSocket(url: NSURL(string: urlWebSocket! + "&sid=\(sid)")!,
|
||||||
cookies: cookies)
|
cookies: cookies)
|
||||||
ws?.queue = handleQueue
|
ws?.queue = handleQueue
|
||||||
ws?.delegate = self
|
ws?.delegate = self
|
||||||
|
|
||||||
if connect {
|
if connect {
|
||||||
ws?.connect()
|
ws?.connect()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private func doFastUpgrade() {
|
private func doFastUpgrade() {
|
||||||
if waitingForPoll {
|
if waitingForPoll {
|
||||||
SocketLogger.err("Outstanding poll when switched to WebSockets," +
|
SocketLogger.err("Outstanding poll when switched to WebSockets," +
|
||||||
"we'll probably disconnect soon. You should report this.", client: self)
|
"we'll probably disconnect soon. You should report this.", client: self)
|
||||||
}
|
}
|
||||||
|
|
||||||
sendWebSocketMessage("", withType: PacketType.UPGRADE, datas: nil)
|
sendWebSocketMessage("", withType: PacketType.UPGRADE, datas: nil)
|
||||||
_websocket = true
|
_websocket = true
|
||||||
_polling = false
|
_polling = false
|
||||||
@ -215,32 +215,32 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
probing = false
|
probing = false
|
||||||
flushProbeWait()
|
flushProbeWait()
|
||||||
}
|
}
|
||||||
|
|
||||||
private func doPoll() {
|
private func doPoll() {
|
||||||
if websocket || waitingForPoll || !connected {
|
if websocket || waitingForPoll || !connected {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
waitingForPoll = true
|
waitingForPoll = true
|
||||||
let req = NSMutableURLRequest(URL: NSURL(string: urlPolling! + "&sid=\(sid)&b64=1")!)
|
let req = NSMutableURLRequest(URL: NSURL(string: urlPolling! + "&sid=\(sid)&b64=1")!)
|
||||||
|
|
||||||
if cookies != nil {
|
if cookies != nil {
|
||||||
let headers = NSHTTPCookie.requestHeaderFieldsWithCookies(cookies!)
|
let headers = NSHTTPCookie.requestHeaderFieldsWithCookies(cookies!)
|
||||||
req.allHTTPHeaderFields = headers
|
req.allHTTPHeaderFields = headers
|
||||||
}
|
}
|
||||||
|
|
||||||
doRequest(req)
|
doRequest(req)
|
||||||
}
|
}
|
||||||
|
|
||||||
private func doRequest(req:NSMutableURLRequest) {
|
private func doRequest(req:NSMutableURLRequest) {
|
||||||
if !polling {
|
if !polling {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
req.cachePolicy = NSURLRequestCachePolicy.ReloadIgnoringLocalAndRemoteCacheData
|
req.cachePolicy = NSURLRequestCachePolicy.ReloadIgnoringLocalAndRemoteCacheData
|
||||||
|
|
||||||
SocketLogger.log("Doing polling request", client: self)
|
SocketLogger.log("Doing polling request", client: self)
|
||||||
|
|
||||||
session.dataTaskWithRequest(req) {[weak self] data, res, err in
|
session.dataTaskWithRequest(req) {[weak self] data, res, err in
|
||||||
if let this = self {
|
if let this = self {
|
||||||
if err != nil || data == nil {
|
if err != nil || data == nil {
|
||||||
@ -251,17 +251,17 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
}
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
SocketLogger.log("Got polling response", client: this)
|
SocketLogger.log("Got polling response", client: this)
|
||||||
|
|
||||||
if let str = NSString(data: data!, encoding: NSUTF8StringEncoding) as? String {
|
if let str = NSString(data: data!, encoding: NSUTF8StringEncoding) as? String {
|
||||||
dispatch_async(this.parseQueue) {[weak this] in
|
dispatch_async(this.parseQueue) {[weak this] in
|
||||||
this?.parsePollingMessage(str)
|
this?.parsePollingMessage(str)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
this.waitingForPoll = false
|
this.waitingForPoll = false
|
||||||
|
|
||||||
if this.fastUpgrade {
|
if this.fastUpgrade {
|
||||||
this.doFastUpgrade()
|
this.doFastUpgrade()
|
||||||
} else if !this.closed && this.polling {
|
} else if !this.closed && this.polling {
|
||||||
@ -269,25 +269,25 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
}
|
}
|
||||||
}}?.resume()
|
}}?.resume()
|
||||||
}
|
}
|
||||||
|
|
||||||
private func flushProbeWait() {
|
private func flushProbeWait() {
|
||||||
SocketLogger.log("Flushing probe wait", client: self)
|
SocketLogger.log("Flushing probe wait", client: self)
|
||||||
|
|
||||||
dispatch_async(emitQueue) {[weak self] in
|
dispatch_async(emitQueue) {[weak self] in
|
||||||
if let this = self {
|
if let this = self {
|
||||||
for waiter in this.probeWait {
|
for waiter in this.probeWait {
|
||||||
this.write(waiter.msg, withType: waiter.type, withData: waiter.data)
|
this.write(waiter.msg, withType: waiter.type, withData: waiter.data)
|
||||||
}
|
}
|
||||||
|
|
||||||
this.probeWait.removeAll(keepCapacity: false)
|
this.probeWait.removeAll(keepCapacity: false)
|
||||||
|
|
||||||
if this.postWait.count != 0 {
|
if this.postWait.count != 0 {
|
||||||
this.flushWaitingForPostToWebSocket()
|
this.flushWaitingForPostToWebSocket()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private func flushWaitingForPost() {
|
private func flushWaitingForPost() {
|
||||||
if postWait.count == 0 || !connected {
|
if postWait.count == 0 || !connected {
|
||||||
return
|
return
|
||||||
@ -295,37 +295,37 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
flushWaitingForPostToWebSocket()
|
flushWaitingForPostToWebSocket()
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var postStr = ""
|
var postStr = ""
|
||||||
|
|
||||||
for packet in postWait {
|
for packet in postWait {
|
||||||
let len = packet.characters.count
|
let len = packet.characters.count
|
||||||
|
|
||||||
postStr += "\(len):\(packet)"
|
postStr += "\(len):\(packet)"
|
||||||
}
|
}
|
||||||
|
|
||||||
postWait.removeAll(keepCapacity: false)
|
postWait.removeAll(keepCapacity: false)
|
||||||
|
|
||||||
let req = NSMutableURLRequest(URL: NSURL(string: urlPolling! + "&sid=\(sid)")!)
|
let req = NSMutableURLRequest(URL: NSURL(string: urlPolling! + "&sid=\(sid)")!)
|
||||||
|
|
||||||
if cookies != nil {
|
if cookies != nil {
|
||||||
let headers = NSHTTPCookie.requestHeaderFieldsWithCookies(cookies!)
|
let headers = NSHTTPCookie.requestHeaderFieldsWithCookies(cookies!)
|
||||||
req.allHTTPHeaderFields = headers
|
req.allHTTPHeaderFields = headers
|
||||||
}
|
}
|
||||||
|
|
||||||
req.HTTPMethod = "POST"
|
req.HTTPMethod = "POST"
|
||||||
req.setValue("text/plain; charset=UTF-8", forHTTPHeaderField: "Content-Type")
|
req.setValue("text/plain; charset=UTF-8", forHTTPHeaderField: "Content-Type")
|
||||||
|
|
||||||
let postData = postStr.dataUsingEncoding(NSUTF8StringEncoding,
|
let postData = postStr.dataUsingEncoding(NSUTF8StringEncoding,
|
||||||
allowLossyConversion: false)!
|
allowLossyConversion: false)!
|
||||||
|
|
||||||
req.HTTPBody = postData
|
req.HTTPBody = postData
|
||||||
req.setValue(String(postData.length), forHTTPHeaderField: "Content-Length")
|
req.setValue(String(postData.length), forHTTPHeaderField: "Content-Length")
|
||||||
|
|
||||||
waitingForPost = true
|
waitingForPost = true
|
||||||
|
|
||||||
SocketLogger.log("POSTing: %@", client: self, args: postStr)
|
SocketLogger.log("POSTing: %@", client: self, args: postStr)
|
||||||
|
|
||||||
session.dataTaskWithRequest(req) {[weak self] data, res, err in
|
session.dataTaskWithRequest(req) {[weak self] data, res, err in
|
||||||
if let this = self {
|
if let this = self {
|
||||||
if err != nil && this.polling {
|
if err != nil && this.polling {
|
||||||
@ -335,9 +335,9 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
NSLog(err?.localizedDescription ?? "Error")
|
NSLog(err?.localizedDescription ?? "Error")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
this.waitingForPost = false
|
this.waitingForPost = false
|
||||||
|
|
||||||
dispatch_async(this.emitQueue) {[weak this] in
|
dispatch_async(this.emitQueue) {[weak this] in
|
||||||
if !(this?.fastUpgrade ?? true) {
|
if !(this?.fastUpgrade ?? true) {
|
||||||
this?.flushWaitingForPost()
|
this?.flushWaitingForPost()
|
||||||
@ -346,29 +346,29 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
}
|
}
|
||||||
}}?.resume()
|
}}?.resume()
|
||||||
}
|
}
|
||||||
|
|
||||||
// We had packets waiting for send when we upgraded
|
// We had packets waiting for send when we upgraded
|
||||||
// Send them raw
|
// Send them raw
|
||||||
private func flushWaitingForPostToWebSocket() {
|
private func flushWaitingForPostToWebSocket() {
|
||||||
for msg in postWait {
|
for msg in postWait {
|
||||||
ws?.writeString(msg)
|
ws?.writeString(msg)
|
||||||
}
|
}
|
||||||
|
|
||||||
postWait.removeAll(keepCapacity: true)
|
postWait.removeAll(keepCapacity: true)
|
||||||
}
|
}
|
||||||
|
|
||||||
private func handleClose() {
|
private func handleClose() {
|
||||||
if polling {
|
if polling {
|
||||||
client?.engineDidClose("Disconnect")
|
client?.engineDidClose("Disconnect")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private func checkIfMessageIsBase64Binary(var message:String) {
|
private func checkIfMessageIsBase64Binary(var message:String) {
|
||||||
if message.hasPrefix("b4") {
|
if message.hasPrefix("b4") {
|
||||||
// binary in base64 string
|
// binary in base64 string
|
||||||
message.removeRange(Range<String.Index>(start: message.startIndex,
|
message.removeRange(Range<String.Index>(start: message.startIndex,
|
||||||
end: advance(message.startIndex, 2)))
|
end: advance(message.startIndex, 2)))
|
||||||
|
|
||||||
if let data = NSData(base64EncodedString: message,
|
if let data = NSData(base64EncodedString: message,
|
||||||
options: NSDataBase64DecodingOptions.IgnoreUnknownCharacters), client = client {
|
options: NSDataBase64DecodingOptions.IgnoreUnknownCharacters), client = client {
|
||||||
dispatch_async(client.handleQueue) {[weak self] in
|
dispatch_async(client.handleQueue) {[weak self] in
|
||||||
@ -377,7 +377,7 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private func handleMessage(message:String) {
|
private func handleMessage(message:String) {
|
||||||
if let client = client {
|
if let client = client {
|
||||||
dispatch_async(client.handleQueue) {[weak client] in
|
dispatch_async(client.handleQueue) {[weak client] in
|
||||||
@ -385,50 +385,50 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private func handleNOOP() {
|
private func handleNOOP() {
|
||||||
doPoll()
|
doPoll()
|
||||||
}
|
}
|
||||||
|
|
||||||
private func handleOpen(openData:String) {
|
private func handleOpen(openData:String) {
|
||||||
let mesData = openData.dataUsingEncoding(NSUTF8StringEncoding, allowLossyConversion: false)!
|
let mesData = openData.dataUsingEncoding(NSUTF8StringEncoding, allowLossyConversion: false)!
|
||||||
|
|
||||||
do {
|
do {
|
||||||
let json = try NSJSONSerialization.JSONObjectWithData(mesData,
|
let json = try NSJSONSerialization.JSONObjectWithData(mesData,
|
||||||
options: NSJSONReadingOptions.AllowFragments) as? NSDictionary
|
options: NSJSONReadingOptions.AllowFragments) as? NSDictionary
|
||||||
if let sid = json?["sid"] as? String {
|
if let sid = json?["sid"] as? String {
|
||||||
self.sid = sid
|
self.sid = sid
|
||||||
_connected = true
|
_connected = true
|
||||||
|
|
||||||
if !forcePolling && !forceWebsockets {
|
if !forcePolling && !forceWebsockets {
|
||||||
createWebsocket(andConnect: true)
|
createWebsocket(andConnect: true)
|
||||||
}
|
}
|
||||||
|
|
||||||
if let pingInterval = json?["pingInterval"] as? Int, pingTimeout = json?["pingTimeout"] as? Int {
|
if let pingInterval = json?["pingInterval"] as? Double, pingTimeout = json?["pingTimeout"] as? Double {
|
||||||
self.pingInterval = pingInterval / 1000
|
self.pingInterval = pingInterval / 1000.0
|
||||||
self.pingTimeout = pingTimeout / 1000
|
self.pingTimeout = pingTimeout / 1000.0
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} catch {
|
} catch {
|
||||||
SocketLogger.err("Error parsing open packet", client: self)
|
SocketLogger.err("Error parsing open packet", client: self)
|
||||||
}
|
}
|
||||||
|
|
||||||
startPingTimer()
|
startPingTimer()
|
||||||
|
|
||||||
if !forceWebsockets {
|
if !forceWebsockets {
|
||||||
doPoll()
|
doPoll()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private func handlePong(pongMessage:String) {
|
private func handlePong(pongMessage:String) {
|
||||||
pongsMissed = 0
|
pongsMissed = 0
|
||||||
|
|
||||||
// We should upgrade
|
// We should upgrade
|
||||||
if pongMessage == "3probe" {
|
if pongMessage == "3probe" {
|
||||||
upgradeTransport()
|
upgradeTransport()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// A poll failed, tell the client about it
|
// A poll failed, tell the client about it
|
||||||
private func handlePollingFailed(reason:String) {
|
private func handlePollingFailed(reason:String) {
|
||||||
_connected = false
|
_connected = false
|
||||||
@ -436,63 +436,63 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
pingTimer?.invalidate()
|
pingTimer?.invalidate()
|
||||||
waitingForPoll = false
|
waitingForPoll = false
|
||||||
waitingForPost = false
|
waitingForPost = false
|
||||||
|
|
||||||
// If cancelled we were already closing
|
// If cancelled we were already closing
|
||||||
if client == nil || reason == "cancelled" {
|
if client == nil || reason == "cancelled" {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if !closed {
|
if !closed {
|
||||||
client?.didError(reason)
|
client?.didError(reason)
|
||||||
client?.engineDidClose(reason)
|
client?.engineDidClose(reason)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public func open(opts:[String: AnyObject]? = nil) {
|
public func open(opts:[String: AnyObject]? = nil) {
|
||||||
if connected {
|
if connected {
|
||||||
SocketLogger.err("Tried to open while connected", client: self)
|
SocketLogger.err("Tried to open while connected", client: self)
|
||||||
|
|
||||||
client?.didError("Tried to open while connected")
|
client?.didError("Tried to open while connected")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
SocketLogger.log("Starting engine", client: self)
|
SocketLogger.log("Starting engine", client: self)
|
||||||
SocketLogger.log("Handshaking", client: self)
|
SocketLogger.log("Handshaking", client: self)
|
||||||
|
|
||||||
closed = false
|
closed = false
|
||||||
|
|
||||||
(urlPolling, urlWebSocket) = createURLs(opts)
|
(urlPolling, urlWebSocket) = createURLs(opts)
|
||||||
|
|
||||||
if forceWebsockets {
|
if forceWebsockets {
|
||||||
_polling = false
|
_polling = false
|
||||||
_websocket = true
|
_websocket = true
|
||||||
createWebsocket(andConnect: true)
|
createWebsocket(andConnect: true)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
let reqPolling = NSMutableURLRequest(URL: NSURL(string: urlPolling! + "&b64=1")!)
|
let reqPolling = NSMutableURLRequest(URL: NSURL(string: urlPolling! + "&b64=1")!)
|
||||||
|
|
||||||
if cookies != nil {
|
if cookies != nil {
|
||||||
let headers = NSHTTPCookie.requestHeaderFieldsWithCookies(cookies!)
|
let headers = NSHTTPCookie.requestHeaderFieldsWithCookies(cookies!)
|
||||||
reqPolling.allHTTPHeaderFields = headers
|
reqPolling.allHTTPHeaderFields = headers
|
||||||
}
|
}
|
||||||
|
|
||||||
doRequest(reqPolling)
|
doRequest(reqPolling)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Translatation of engine.io-parser#decodePayload
|
// Translatation of engine.io-parser#decodePayload
|
||||||
private func parsePollingMessage(str:String) {
|
private func parsePollingMessage(str:String) {
|
||||||
if str.characters.count == 1 {
|
if str.characters.count == 1 {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// println(str)
|
// println(str)
|
||||||
|
|
||||||
let strArray = Array(str.characters)
|
let strArray = Array(str.characters)
|
||||||
var length = ""
|
var length = ""
|
||||||
var n = 0
|
var n = 0
|
||||||
var msg = ""
|
var msg = ""
|
||||||
|
|
||||||
func testLength(length:String, inout n:Int) -> Bool {
|
func testLength(length:String, inout n:Int) -> Bool {
|
||||||
if let num = Int(length) {
|
if let num = Int(length) {
|
||||||
n = num
|
n = num
|
||||||
@ -501,10 +501,10 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
for var i = 0, l = str.characters.count; i < l; i++ {
|
for var i = 0, l = str.characters.count; i < l; i++ {
|
||||||
let chr = String(strArray[i])
|
let chr = String(strArray[i])
|
||||||
|
|
||||||
if chr != ":" {
|
if chr != ":" {
|
||||||
length += chr
|
length += chr
|
||||||
} else {
|
} else {
|
||||||
@ -513,27 +513,27 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
handlePollingFailed("Error parsing XHR message")
|
handlePollingFailed("Error parsing XHR message")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
msg = String(strArray[i+1...i+n])
|
msg = String(strArray[i+1...i+n])
|
||||||
|
|
||||||
if let lengthInt = Int(length) where lengthInt != msg.characters.count {
|
if let lengthInt = Int(length) where lengthInt != msg.characters.count {
|
||||||
SocketLogger.err("Parsing error: %@", client: self, args: str)
|
SocketLogger.err("Parsing error: %@", client: self, args: str)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if msg.characters.count != 0 {
|
if msg.characters.count != 0 {
|
||||||
// Be sure to capture the value of the msg
|
// Be sure to capture the value of the msg
|
||||||
dispatch_async(handleQueue) {[weak self, msg] in
|
dispatch_async(handleQueue) {[weak self, msg] in
|
||||||
self?.parseEngineMessage(msg, fromPolling: true)
|
self?.parseEngineMessage(msg, fromPolling: true)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
i += n
|
i += n
|
||||||
length = ""
|
length = ""
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private func parseEngineData(data:NSData) {
|
private func parseEngineData(data:NSData) {
|
||||||
if let client = client {
|
if let client = client {
|
||||||
dispatch_async(client.handleQueue) {[weak self] in
|
dispatch_async(client.handleQueue) {[weak self] in
|
||||||
@ -541,19 +541,19 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private func parseEngineMessage(var message:String, fromPolling:Bool) {
|
private func parseEngineMessage(var message:String, fromPolling:Bool) {
|
||||||
SocketLogger.log("Got message: %@", client: self, args: message)
|
SocketLogger.log("Got message: %@", client: self, args: message)
|
||||||
|
|
||||||
if fromPolling {
|
if fromPolling {
|
||||||
fixDoubleUTF8(&message)
|
fixDoubleUTF8(&message)
|
||||||
}
|
}
|
||||||
|
|
||||||
let type = PacketType(str: (message["^(\\d)"].groups()?[1])) ?? {
|
let type = PacketType(str: (message["^(\\d)"].groups()?[1])) ?? {
|
||||||
self.checkIfMessageIsBase64Binary(message)
|
self.checkIfMessageIsBase64Binary(message)
|
||||||
return PacketType.NOOP
|
return PacketType.NOOP
|
||||||
}()
|
}()
|
||||||
|
|
||||||
switch type {
|
switch type {
|
||||||
case PacketType.MESSAGE:
|
case PacketType.MESSAGE:
|
||||||
message.removeAtIndex(message.startIndex)
|
message.removeAtIndex(message.startIndex)
|
||||||
@ -571,13 +571,13 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
SocketLogger.log("Got unknown packet type", client: self)
|
SocketLogger.log("Got unknown packet type", client: self)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private func probeWebSocket() {
|
private func probeWebSocket() {
|
||||||
if websocketConnected {
|
if websocketConnected {
|
||||||
sendWebSocketMessage("probe", withType: PacketType.PING)
|
sendWebSocketMessage("probe", withType: PacketType.PING)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Send an engine message (4)
|
/// Send an engine message (4)
|
||||||
public func send(msg:String, withData datas:[NSData]?) {
|
public func send(msg:String, withData datas:[NSData]?) {
|
||||||
if probing {
|
if probing {
|
||||||
@ -586,7 +586,7 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
write(msg, withType: PacketType.MESSAGE, withData: datas)
|
write(msg, withType: PacketType.MESSAGE, withData: datas)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@objc private func sendPing() {
|
@objc private func sendPing() {
|
||||||
//Server is not responding
|
//Server is not responding
|
||||||
if pongsMissed > pongsMissedMax {
|
if pongsMissed > pongsMissedMax {
|
||||||
@ -594,43 +594,43 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
client?.engineDidClose("Ping timeout")
|
client?.engineDidClose("Ping timeout")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
++pongsMissed
|
++pongsMissed
|
||||||
write("", withType: PacketType.PING, withData: nil)
|
write("", withType: PacketType.PING, withData: nil)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Send polling message.
|
/// Send polling message.
|
||||||
/// Only call on emitQueue
|
/// Only call on emitQueue
|
||||||
private func sendPollMessage(var msg:String, withType type:PacketType,
|
private func sendPollMessage(var msg:String, withType type:PacketType,
|
||||||
datas:[NSData]? = nil) {
|
datas:[NSData]? = nil) {
|
||||||
SocketLogger.log("Sending poll: %@ as type: %@", client: self, args: msg, type.rawValue)
|
SocketLogger.log("Sending poll: %@ as type: %@", client: self, args: msg, type.rawValue)
|
||||||
|
|
||||||
doubleEncodeUTF8(&msg)
|
doubleEncodeUTF8(&msg)
|
||||||
let strMsg = "\(type.rawValue)\(msg)"
|
let strMsg = "\(type.rawValue)\(msg)"
|
||||||
|
|
||||||
postWait.append(strMsg)
|
postWait.append(strMsg)
|
||||||
|
|
||||||
if datas != nil {
|
if datas != nil {
|
||||||
for data in datas! {
|
for data in datas! {
|
||||||
let (_, b64Data) = createBinaryDataForSend(data)
|
let (_, b64Data) = createBinaryDataForSend(data)
|
||||||
|
|
||||||
postWait.append(b64Data!)
|
postWait.append(b64Data!)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if !waitingForPost {
|
if !waitingForPost {
|
||||||
flushWaitingForPost()
|
flushWaitingForPost()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Send message on WebSockets
|
/// Send message on WebSockets
|
||||||
/// Only call on emitQueue
|
/// Only call on emitQueue
|
||||||
private func sendWebSocketMessage(str:String, withType type:PacketType,
|
private func sendWebSocketMessage(str:String, withType type:PacketType,
|
||||||
datas:[NSData]? = nil) {
|
datas:[NSData]? = nil) {
|
||||||
SocketLogger.log("Sending ws: %@ as type: %@", client: self, args: str, type.rawValue)
|
SocketLogger.log("Sending ws: %@ as type: %@", client: self, args: str, type.rawValue)
|
||||||
|
|
||||||
ws?.writeString("\(type.rawValue)\(str)")
|
ws?.writeString("\(type.rawValue)\(str)")
|
||||||
|
|
||||||
if datas != nil {
|
if datas != nil {
|
||||||
for data in datas! {
|
for data in datas! {
|
||||||
let (data, _) = createBinaryDataForSend(data)
|
let (data, _) = createBinaryDataForSend(data)
|
||||||
@ -640,37 +640,36 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Starts the ping timer
|
// Starts the ping timer
|
||||||
private func startPingTimer() {
|
private func startPingTimer() {
|
||||||
if pingInterval == nil {
|
if pingInterval == nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
pingTimer?.invalidate()
|
pingTimer?.invalidate()
|
||||||
dispatch_async(dispatch_get_main_queue()) {[weak self] in
|
dispatch_async(dispatch_get_main_queue()) {[weak self] in
|
||||||
if let this = self {
|
if let this = self {
|
||||||
this.pingTimer = NSTimer.scheduledTimerWithTimeInterval(NSTimeInterval(this.pingInterval!),
|
this.pingTimer = NSTimer.scheduledTimerWithTimeInterval(this.pingInterval!, target: this,
|
||||||
target: this,
|
|
||||||
selector: Selector("sendPing"), userInfo: nil, repeats: true)
|
selector: Selector("sendPing"), userInfo: nil, repeats: true)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func stopPolling() {
|
func stopPolling() {
|
||||||
session.invalidateAndCancel()
|
session.invalidateAndCancel()
|
||||||
}
|
}
|
||||||
|
|
||||||
private func upgradeTransport() {
|
private func upgradeTransport() {
|
||||||
if websocketConnected {
|
if websocketConnected {
|
||||||
SocketLogger.log("Upgrading transport to WebSockets", client: self)
|
SocketLogger.log("Upgrading transport to WebSockets", client: self)
|
||||||
|
|
||||||
fastUpgrade = true
|
fastUpgrade = true
|
||||||
sendPollMessage("", withType: PacketType.NOOP)
|
sendPollMessage("", withType: PacketType.NOOP)
|
||||||
// After this point, we should not send anymore polling messages
|
// After this point, we should not send anymore polling messages
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
Write a message, independent of transport.
|
Write a message, independent of transport.
|
||||||
*/
|
*/
|
||||||
@ -689,29 +688,29 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
Write a message, independent of transport. For Objective-C. withData should be an NSArray of NSData
|
Write a message, independent of transport. For Objective-C. withData should be an NSArray of NSData
|
||||||
*/
|
*/
|
||||||
public func writeObjc(msg:String, withType type:Int, withData data:NSArray?) {
|
public func writeObjc(msg:String, withType type:Int, withData data:NSArray?) {
|
||||||
if let pType = PacketType(rawValue: type) {
|
if let pType = PacketType(rawValue: type) {
|
||||||
var arr = [NSData]()
|
var arr = [NSData]()
|
||||||
|
|
||||||
if data != nil {
|
if data != nil {
|
||||||
for d in data! {
|
for d in data! {
|
||||||
arr.append(d as! NSData)
|
arr.append(d as! NSData)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
write(msg, withType: pType, withData: arr)
|
write(msg, withType: pType, withData: arr)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Delagate methods
|
// Delagate methods
|
||||||
|
|
||||||
public func websocketDidConnect(socket:WebSocket) {
|
public func websocketDidConnect(socket:WebSocket) {
|
||||||
websocketConnected = true
|
websocketConnected = true
|
||||||
|
|
||||||
if !forceWebsockets {
|
if !forceWebsockets {
|
||||||
probing = true
|
probing = true
|
||||||
probeWebSocket()
|
probeWebSocket()
|
||||||
@ -721,37 +720,37 @@ public final class SocketEngine: NSObject, WebSocketDelegate, SocketLogClient {
|
|||||||
_polling = false
|
_polling = false
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public func websocketDidDisconnect(socket:WebSocket, error:NSError?) {
|
public func websocketDidDisconnect(socket:WebSocket, error:NSError?) {
|
||||||
websocketConnected = false
|
websocketConnected = false
|
||||||
probing = false
|
probing = false
|
||||||
|
|
||||||
if closed {
|
if closed {
|
||||||
client?.engineDidClose("Disconnect")
|
client?.engineDidClose("Disconnect")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if websocket {
|
if websocket {
|
||||||
pingTimer?.invalidate()
|
pingTimer?.invalidate()
|
||||||
_connected = false
|
_connected = false
|
||||||
_websocket = false
|
_websocket = false
|
||||||
|
|
||||||
let reason = error?.localizedDescription ?? "Socket Disconnected"
|
let reason = error?.localizedDescription ?? "Socket Disconnected"
|
||||||
|
|
||||||
if error != nil {
|
if error != nil {
|
||||||
client?.didError(reason)
|
client?.didError(reason)
|
||||||
}
|
}
|
||||||
|
|
||||||
client?.engineDidClose(reason)
|
client?.engineDidClose(reason)
|
||||||
} else {
|
} else {
|
||||||
flushProbeWait()
|
flushProbeWait()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public func websocketDidReceiveMessage(socket:WebSocket, text:String) {
|
public func websocketDidReceiveMessage(socket:WebSocket, text:String) {
|
||||||
parseEngineMessage(text, fromPolling: false)
|
parseEngineMessage(text, fromPolling: false)
|
||||||
}
|
}
|
||||||
|
|
||||||
public func websocketDidReceiveData(socket:WebSocket, data:NSData) {
|
public func websocketDidReceiveData(socket:WebSocket, data:NSData) {
|
||||||
parseEngineData(data)
|
parseEngineData(data)
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user