Squashed History

src/main/java/org/whispersystems/textsecuregcm/push/APNSender.java

@@ -0,0 +1,116 @@
+/**
+ * Copyright (C) 2013 Open WhisperSystems
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+package org.whispersystems.textsecuregcm.push;
+
+import com.google.common.base.Optional;
+import com.notnoop.apns.APNS;
+import com.notnoop.apns.ApnsService;
+import com.notnoop.exceptions.NetworkIOException;
+import com.yammer.metrics.Metrics;
+import com.yammer.metrics.core.Meter;
+import org.bouncycastle.openssl.PEMReader;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.whispersystems.textsecuregcm.entities.EncryptedOutgoingMessage;
+import org.whispersystems.textsecuregcm.util.Util;
+
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.io.InputStreamReader;
+import java.net.MalformedURLException;
+import java.security.KeyPair;
+import java.security.KeyStore;
+import java.security.KeyStoreException;
+import java.security.NoSuchAlgorithmException;
+import java.security.cert.Certificate;
+import java.security.cert.CertificateException;
+import java.security.cert.X509Certificate;
+import java.util.concurrent.TimeUnit;
+
+public class APNSender {
+
+  private final Meter  success = Metrics.newMeter(APNSender.class, "sent", "success", TimeUnit.MINUTES);
+  private final Meter  failure = Metrics.newMeter(APNSender.class, "sent", "failure", TimeUnit.MINUTES);
+  private final Logger logger  = LoggerFactory.getLogger(APNSender.class);
+
+  private static final String MESSAGE_BODY = "m";
+
+  private final Optional<ApnsService> apnService;
+
+  public APNSender(String apnCertificate, String apnKey)
+      throws CertificateException, NoSuchAlgorithmException, KeyStoreException, IOException
+  {
+    if (!Util.isEmpty(apnCertificate) && !Util.isEmpty(apnKey)) {
+      byte[] keyStore = initializeKeyStore(apnCertificate, apnKey);
+      this.apnService = Optional.of(APNS.newService()
+                                        .withCert(new ByteArrayInputStream(keyStore), "insecure")
+                                        .withSandboxDestination().build());
+    } else {
+      this.apnService = Optional.absent();
+    }
+  }
+
+  public void sendMessage(String registrationId, EncryptedOutgoingMessage message)
+      throws IOException
+  {
+    try {
+      if (!apnService.isPresent()) {
+        failure.mark();
+        throw new IOException("APN access not configured!");
+      }
+
+      String payload = APNS.newPayload()
+                           .alertBody("Message!")
+                           .customField(MESSAGE_BODY, message.serialize())
+                           .build();
+
+      logger.debug("APN Payload: " + payload);
+
+      apnService.get().push(registrationId, payload);
+      success.mark();
+    } catch (MalformedURLException mue) {
+      throw new AssertionError(mue);
+    } catch (NetworkIOException nioe) {
+      logger.warn("Network Error", nioe);
+      failure.mark();
+      throw new IOException("Error sending APN");
+    }
+  }
+
+  private static byte[] initializeKeyStore(String pemCertificate, String pemKey)
+      throws KeyStoreException, CertificateException, NoSuchAlgorithmException, IOException
+  {
+    PEMReader       reader           = new PEMReader(new InputStreamReader(new ByteArrayInputStream(pemCertificate.getBytes())));
+    X509Certificate certificate      = (X509Certificate) reader.readObject();
+    Certificate[]   certificateChain = {certificate};
+
+    reader            = new PEMReader(new InputStreamReader(new ByteArrayInputStream(pemKey.getBytes())));
+    KeyPair keyPair = (KeyPair) reader.readObject();
+
+    KeyStore keyStore = KeyStore.getInstance("pkcs12");
+    keyStore.load(null);
+    keyStore.setEntry("apn",
+                      new KeyStore.PrivateKeyEntry(keyPair.getPrivate(), certificateChain),
+                      new KeyStore.PasswordProtection("insecure".toCharArray()));
+
+    ByteArrayOutputStream baos = new ByteArrayOutputStream();
+    keyStore.store(baos, "insecure".toCharArray());
+
+    return baos.toByteArray();
+  }
+}

src/main/java/org/whispersystems/textsecuregcm/push/GCMSender.java

@@ -0,0 +1,63 @@
+/**
+ * Copyright (C) 2013 Open WhisperSystems
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+package org.whispersystems.textsecuregcm.push;
+
+import com.google.android.gcm.server.Constants;
+import com.google.android.gcm.server.Message;
+import com.google.android.gcm.server.Result;
+import com.google.android.gcm.server.Sender;
+import com.yammer.metrics.Metrics;
+import com.yammer.metrics.core.Meter;
+import org.whispersystems.textsecuregcm.controllers.NoSuchUserException;
+import org.whispersystems.textsecuregcm.entities.EncryptedOutgoingMessage;
+
+import java.io.IOException;
+import java.util.concurrent.TimeUnit;
+
+public class GCMSender {
+
+  private final Meter success = Metrics.newMeter(GCMSender.class, "sent", "success", TimeUnit.MINUTES);
+  private final Meter failure = Metrics.newMeter(GCMSender.class, "sent", "failure", TimeUnit.MINUTES);
+
+  private final Sender sender;
+
+  public GCMSender(String apiKey) {
+    this.sender = new Sender(apiKey);
+  }
+
+  public String sendMessage(String gcmRegistrationId, EncryptedOutgoingMessage outgoingMessage)
+      throws IOException, NoSuchUserException
+  {
+    Message gcmMessage = new Message.Builder().addData("type", "message")
+                                              .addData("message", outgoingMessage.serialize())
+                                              .build();
+
+    Result  result = sender.send(gcmMessage, gcmRegistrationId, 5);
+
+    if (result.getMessageId() != null) {
+      success.mark();
+      return result.getCanonicalRegistrationId();
+    } else {
+      failure.mark();
+      if (result.getErrorCodeName().equals(Constants.ERROR_NOT_REGISTERED)) {
+        throw new NoSuchUserException("User no longer registered with GCM.");
+      } else {
+        throw new IOException("GCM Failed: " + result.getErrorCodeName());
+      }
+    }
+  }
+}

src/main/java/org/whispersystems/textsecuregcm/push/PushSender.java

@@ -0,0 +1,103 @@
+/**
+ * Copyright (C) 2013 Open WhisperSystems
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+package org.whispersystems.textsecuregcm.push;
+
+import com.google.common.base.Optional;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.whispersystems.textsecuregcm.configuration.ApnConfiguration;
+import org.whispersystems.textsecuregcm.configuration.GcmConfiguration;
+import org.whispersystems.textsecuregcm.controllers.NoSuchUserException;
+import org.whispersystems.textsecuregcm.entities.EncryptedOutgoingMessage;
+import org.whispersystems.textsecuregcm.entities.MessageProtos;
+import org.whispersystems.textsecuregcm.storage.Account;
+import org.whispersystems.textsecuregcm.storage.AccountsManager;
+import org.whispersystems.textsecuregcm.storage.DirectoryManager;
+
+import java.io.IOException;
+import java.security.KeyStoreException;
+import java.security.NoSuchAlgorithmException;
+import java.security.cert.CertificateException;
+
+public class PushSender {
+
+  private final Logger logger = LoggerFactory.getLogger(PushSender.class);
+
+  private final AccountsManager  accounts;
+  private final DirectoryManager directory;
+
+  private final GCMSender gcmSender;
+  private final APNSender apnSender;
+
+  public PushSender(GcmConfiguration gcmConfiguration,
+                    ApnConfiguration apnConfiguration,
+                    AccountsManager accounts,
+                    DirectoryManager directory)
+      throws CertificateException, NoSuchAlgorithmException, KeyStoreException, IOException
+  {
+    this.accounts  = accounts;
+    this.directory = directory;
+
+    this.gcmSender = new GCMSender(gcmConfiguration.getApiKey());
+    this.apnSender = new APNSender(apnConfiguration.getCertificate(), apnConfiguration.getKey());
+  }
+
+  public void sendMessage(String destination, MessageProtos.OutgoingMessageSignal outgoingMessage)
+      throws IOException, NoSuchUserException
+  {
+    Optional<Account> account = accounts.get(destination);
+
+    if (!account.isPresent()) {
+      directory.remove(destination);
+      throw new NoSuchUserException("No such local destination: " + destination);
+    }
+
+    String signalingKey              = account.get().getSignalingKey();
+    EncryptedOutgoingMessage message = new EncryptedOutgoingMessage(outgoingMessage, signalingKey);
+
+    if      (account.get().getGcmRegistrationId() != null) sendGcmMessage(account.get(), message);
+    else if (account.get().getApnRegistrationId() != null) sendApnMessage(account.get(), message);
+    else                                             throw new NoSuchUserException("No push identifier!");
+  }
+
+  private void sendGcmMessage(Account account, EncryptedOutgoingMessage outgoingMessage)
+      throws IOException, NoSuchUserException
+  {
+    try {
+      String canonicalId = gcmSender.sendMessage(account.getGcmRegistrationId(),
+                                                 outgoingMessage);
+
+      if (canonicalId != null) {
+        account.setGcmRegistrationId(canonicalId);
+        accounts.update(account);
+      }
+
+    } catch (NoSuchUserException e) {
+      logger.debug("No Such User", e);
+      account.setGcmRegistrationId(null);
+      accounts.update(account);
+      throw new NoSuchUserException("User no longer exists in GCM.");
+    }
+  }
+
+  private void sendApnMessage(Account account, EncryptedOutgoingMessage outgoingMessage)
+      throws IOException
+  {
+    apnSender.sendMessage(account.getApnRegistrationId(), outgoingMessage);
+  }
+
+}