Disallow sending messages before a socket is open

beardedspice · May 12, 2016 · aaff48e · aaff48e
1 parent 11fa5fe
commit aaff48e
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/PocketSocket/PSWebSocket.m b/PocketSocket/PSWebSocket.m
@@ -201,6 +201,11 @@ - (void)open {
 - (void)send:(id)message {
     NSParameterAssert(message);
     [self executeWork:^{
+        if(!_opened || _readyState == PSWebSocketReadyStateConnecting) {
+            [NSException raise:@"Invalid State" format:@"You cannot send a PSWebSocket messages before it is finished opening."];
+            return;
+        }
+
         if([message isKindOfClass:[NSString class]]) {
             [_driver sendText:message];
         } else if([message isKindOfClass:[NSData class]]) {