Initial pre-alpha support for sender key.

2026-04-20 00:29:11 +01:00 · 2021-05-14 14:03:35 -04:00
parent c54f016213
commit 57c0b8fd0f
124 changed files with 3668 additions and 444 deletions
--- a/app/src/main/java/org/thoughtcrime/securesms/service/PendingRetryReceiptManager.java
+++ b/app/src/main/java/org/thoughtcrime/securesms/service/PendingRetryReceiptManager.java
@@ -0,0 +1,88 @@
+package org.thoughtcrime.securesms.service;
+
+import android.app.Application;
+import android.content.BroadcastReceiver;
+import android.content.Context;
+import android.content.Intent;
+
+import androidx.annotation.AnyThread;
+import androidx.annotation.NonNull;
+import androidx.annotation.Nullable;
+import androidx.annotation.WorkerThread;
+
+import org.signal.core.util.logging.Log;
+import org.thoughtcrime.securesms.database.DatabaseFactory;
+import org.thoughtcrime.securesms.database.MessageDatabase;
+import org.thoughtcrime.securesms.database.PendingRetryReceiptDatabase;
+import org.thoughtcrime.securesms.database.model.PendingRetryReceiptModel;
+import org.thoughtcrime.securesms.dependencies.ApplicationDependencies;
+import org.thoughtcrime.securesms.util.FeatureFlags;
+
+
+/**
+ * Manages the time-based creation of error messages for retries that are pending for messages we couldn't decrypt.
+ */
+public final class PendingRetryReceiptManager extends TimedEventManager<PendingRetryReceiptModel> {
+
+  private static final String TAG = Log.tag(PendingRetryReceiptManager.class);
+
+  private final PendingRetryReceiptDatabase pendingDatabase;
+  private final MessageDatabase             messageDatabase;
+
+  public PendingRetryReceiptManager(@NonNull Application application) {
+    super(application, "PendingRetryReceiptManager");
+
+    this.pendingDatabase = DatabaseFactory.getPendingRetryReceiptDatabase(application);
+    this.messageDatabase = DatabaseFactory.getSmsDatabase(application);
+
+    scheduleIfNecessary();
+  }
+
+  @WorkerThread
+  @Override
+  protected @Nullable PendingRetryReceiptModel getNextClosestEvent() {
+    PendingRetryReceiptModel model = pendingDatabase.getOldest();
+
+    if (model != null) {
+      Log.i(TAG, "Next closest expiration is in " + getDelayForEvent(model) + " ms for timestamp " + model.getSentTimestamp() + ".");
+    } else {
+      Log.d(TAG, "No pending receipts to schedule.");
+    }
+
+    return model;
+  }
+
+  @WorkerThread
+  @Override
+  protected void executeEvent(@NonNull PendingRetryReceiptModel event) {
+    Log.w(TAG, "It's been " + (System.currentTimeMillis() - event.getReceivedTimestamp()) + " ms since this retry receipt was received. Showing an error.");
+    messageDatabase.insertBadDecryptMessage(event.getAuthor(), event.getAuthorDevice(), event.getSentTimestamp(), event.getReceivedTimestamp(), event.getThreadId());
+    pendingDatabase.delete(event.getId());
+  }
+
+  @WorkerThread
+  @Override
+  protected long getDelayForEvent(@NonNull PendingRetryReceiptModel event) {
+    long expiresAt = event.getReceivedTimestamp() + FeatureFlags.retryReceiptLifespan();
+    long timeLeft  = expiresAt - System.currentTimeMillis();
+
+    return Math.max(0, timeLeft);
+  }
+
+  @AnyThread
+  @Override
+  protected void scheduleAlarm(@NonNull Application application, long delay) {
+    setAlarm(application, delay, PendingRetryReceiptAlarm.class);
+  }
+
+  public static class PendingRetryReceiptAlarm extends BroadcastReceiver {
+
+    private static final String TAG = Log.tag(PendingRetryReceiptAlarm.class);
+
+    @Override
+    public void onReceive(Context context, Intent intent) {
+      Log.d(TAG, "onReceive()");
+      ApplicationDependencies.getPendingRetryReceiptManager().scheduleIfNecessary();
+    }
+  }
+}