From b1debdf91204a6254d4780edda8ef764be9995b6 Mon Sep 17 00:00:00 2001
From: Erik <nuclear.ace@gmail.com>
Date: Sun, 18 Dec 2016 20:06:12 -0500
Subject: [PATCH] bump ws, fix #598

---
 Source/SSLSecurity.swift | 2 +-
 Source/WebSocket.swift   | 5 +++++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/Source/SSLSecurity.swift b/Source/SSLSecurity.swift
index d2a2b78..4d2c60a 100644
--- a/Source/SSLSecurity.swift
+++ b/Source/SSLSecurity.swift
@@ -26,7 +26,7 @@ public protocol SSLTrustValidator {
     func isValid(_ trust: SecTrust, domain: String?) -> Bool
 }
 
-open class SSLCert {
+open class SSLCert : NSObject {
     var certData: Data?
     var key: SecKey?
     
diff --git a/Source/WebSocket.swift b/Source/WebSocket.swift
index 108592d..3bdff34 100644
--- a/Source/WebSocket.swift
+++ b/Source/WebSocket.swift
@@ -162,6 +162,11 @@ open class WebSocket : NSObject, StreamDelegate {
     public init(url: URL, protocols: [String]? = nil) {
         self.url = url
         self.origin = url.absoluteString
+        if let hostUrl = URL (string: "/", relativeTo: url) {
+            var origin = hostUrl.absoluteString
+            origin.remove(at: origin.index(before: origin.endIndex))
+            self.origin = origin
+        }
         writeQueue.maxConcurrentOperationCount = 1
         optionalProtocols = protocols
     }