From eb4cc7f4e5d2b8af7eb46419a467e6f2bf89e592 Mon Sep 17 00:00:00 2001
From: Moxie Marlinspike <moxie@thoughtcrime.org>
Date: Sat, 12 Oct 2013 03:57:17 -0700
Subject: [PATCH] Add incoming message sources to active token dir.

---
 src/org/thoughtcrime/securesms/gcm/GcmIntentService.java | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/org/thoughtcrime/securesms/gcm/GcmIntentService.java b/src/org/thoughtcrime/securesms/gcm/GcmIntentService.java
index 397e34a529..2b34a66018 100644
--- a/src/org/thoughtcrime/securesms/gcm/GcmIntentService.java
+++ b/src/org/thoughtcrime/securesms/gcm/GcmIntentService.java
@@ -9,6 +9,7 @@ import org.thoughtcrime.securesms.service.RegistrationService;
 import org.thoughtcrime.securesms.service.SendReceiveService;
 import org.thoughtcrime.securesms.util.TextSecurePreferences;
 import org.whispersystems.textsecure.crypto.InvalidVersionException;
+import org.whispersystems.textsecure.directory.Directory;
 import org.whispersystems.textsecure.push.IncomingEncryptedPushMessage;
 import org.whispersystems.textsecure.push.IncomingPushMessage;
 import org.whispersystems.textsecure.push.PushServiceSocket;
@@ -62,6 +63,9 @@ public class GcmIntentService extends GCMBaseIntentService {
       service.setAction(SendReceiveService.RECEIVE_PUSH_ACTION);
       service.putExtra("message", message);
 
+      Directory directory = Directory.getInstance(context);
+      directory.setToken(directory.getToken(message.getSource()), true);
+
       context.startService(service);
     } catch (IOException e) {
       Log.w("GcmIntentService", e);