Socket Management

Signal/src/AppDelegate.m

@@ -208,6 +208,7 @@
 }
 
 -(void) applicationDidBecomeActive:(UIApplication *)application {
+    [TSSocketManager becomeActive];
     [AppAudioManager.sharedInstance awake];
     
     // Hacky way to clear notification center after processed push
@@ -231,8 +232,8 @@
     completionHandler();
 }
 
-- (void)applicationWillResignActive:(UIApplication *)application{
-    [self protectScreen];
+- (void)applicationDidEnterBackground:(UIApplication *)application{
+    [TSSocketManager resignActivity];
 }
 
 - (void)prepareScreenshotProtection{

Signal/src/textsecure/Network/WebSockets/TSSocketManager.m

@@ -163,7 +163,12 @@ NSString * const SocketConnectingNotification = @"SocketConnectingNotification";
     [message setResponse:response.build];
     [message setType:WebSocketMessageTypeResponse];
     
-    [self.websocket send:message.build.data];
+    @try {
+        [self.websocket send:message.build.data];
+    }
+    @catch (NSException *exception) {
+        DDLogWarn(@"Caught exception while trying to write on the socket %@", exception.debugDescription);
+    }
 }
 
 - (void)webSocket:(SRWebSocket *)webSocket didCloseWithCode:(NSInteger)code reason:(NSString *)reason wasClean:(BOOL)wasClean {
@@ -174,7 +179,12 @@ NSString * const SocketConnectingNotification = @"SocketConnectingNotification";
 }
 
 - (void)webSocketHeartBeat {
-    [self.websocket sendPing:nil];
+    @try {
+        [self.websocket sendPing:nil];
+    }
+    @catch (NSException *exception) {
+        DDLogWarn(@"Caught exception while trying to write on the socket %@", exception.debugDescription);
+    }
 }
 
 - (NSString*)webSocketAuthenticationString{

Signal/src/util/protocols/PreKeyBundle+jsonDict.m

@@ -23,7 +23,6 @@
     }
     
     NSData *identityKey = [NSData dataFromBase64String:identityKeyString];
-    NSLog(@"IDentityKey %@", identityKey);
     
     for (NSDictionary *deviceDict in devicesArray) {
         NSNumber *registrationIdString = [deviceDict objectForKey:@"registrationId"];