mirror of
https://github.com/oxen-io/session-android.git
synced 2024-12-27 02:07:42 +00:00
100 lines
3.4 KiB
Java
100 lines
3.4 KiB
Java
package org.thoughtcrime.securesms.jobs;
|
|
|
|
import android.content.Context;
|
|
import android.support.annotation.NonNull;
|
|
|
|
import org.thoughtcrime.securesms.ApplicationContext;
|
|
import org.thoughtcrime.securesms.crypto.IdentityKeyUtil;
|
|
import org.thoughtcrime.securesms.crypto.MasterSecret;
|
|
import org.thoughtcrime.securesms.crypto.PreKeyUtil;
|
|
import org.thoughtcrime.securesms.dependencies.InjectableType;
|
|
import org.thoughtcrime.securesms.jobmanager.JobParameters;
|
|
import org.thoughtcrime.securesms.jobmanager.SafeData;
|
|
import org.thoughtcrime.securesms.logging.Log;
|
|
import org.thoughtcrime.securesms.util.TextSecurePreferences;
|
|
import org.whispersystems.libsignal.IdentityKeyPair;
|
|
import org.whispersystems.libsignal.state.PreKeyRecord;
|
|
import org.whispersystems.libsignal.state.SignedPreKeyRecord;
|
|
import org.whispersystems.signalservice.api.SignalServiceAccountManager;
|
|
import org.whispersystems.signalservice.api.push.exceptions.NonSuccessfulResponseCodeException;
|
|
import org.whispersystems.signalservice.api.push.exceptions.PushNetworkException;
|
|
|
|
import java.io.IOException;
|
|
import java.util.List;
|
|
|
|
import javax.inject.Inject;
|
|
|
|
import androidx.work.Data;
|
|
import androidx.work.WorkerParameters;
|
|
|
|
public class RefreshPreKeysJob extends ContextJob implements InjectableType {
|
|
|
|
private static final String TAG = RefreshPreKeysJob.class.getSimpleName();
|
|
|
|
private static final int PREKEY_MINIMUM = 10;
|
|
|
|
@Inject transient SignalServiceAccountManager accountManager;
|
|
|
|
public RefreshPreKeysJob(@NonNull Context context, @NonNull WorkerParameters workerParameters) {
|
|
super(context, workerParameters);
|
|
}
|
|
|
|
public RefreshPreKeysJob(Context context) {
|
|
super(context, JobParameters.newBuilder()
|
|
.withGroupId(RefreshPreKeysJob.class.getSimpleName())
|
|
.withNetworkRequirement()
|
|
.withRetryCount(5)
|
|
.create());
|
|
}
|
|
|
|
@Override
|
|
protected void initialize(@NonNull SafeData data) {
|
|
}
|
|
|
|
@Override
|
|
protected @NonNull Data serialize(@NonNull Data.Builder dataBuilder) {
|
|
return dataBuilder.build();
|
|
}
|
|
|
|
@Override
|
|
public void onRun() throws IOException {
|
|
if (!TextSecurePreferences.isPushRegistered(context)) return;
|
|
|
|
int availableKeys = accountManager.getPreKeysCount();
|
|
|
|
if (availableKeys >= PREKEY_MINIMUM && TextSecurePreferences.isSignedPreKeyRegistered(context)) {
|
|
Log.i(TAG, "Available keys sufficient: " + availableKeys);
|
|
return;
|
|
}
|
|
|
|
List<PreKeyRecord> preKeyRecords = PreKeyUtil.generatePreKeys(context);
|
|
IdentityKeyPair identityKey = IdentityKeyUtil.getIdentityKeyPair(context);
|
|
SignedPreKeyRecord signedPreKeyRecord = PreKeyUtil.generateSignedPreKey(context, identityKey, false);
|
|
|
|
Log.i(TAG, "Registering new prekeys...");
|
|
|
|
accountManager.setPreKeys(identityKey.getPublicKey(), signedPreKeyRecord, preKeyRecords);
|
|
|
|
PreKeyUtil.setActiveSignedPreKeyId(context, signedPreKeyRecord.getId());
|
|
TextSecurePreferences.setSignedPreKeyRegistered(context, true);
|
|
|
|
ApplicationContext.getInstance(context)
|
|
.getJobManager()
|
|
.add(new CleanPreKeysJob(context));
|
|
}
|
|
|
|
@Override
|
|
public boolean onShouldRetry(Exception exception) {
|
|
if (exception instanceof NonSuccessfulResponseCodeException) return false;
|
|
if (exception instanceof PushNetworkException) return true;
|
|
|
|
return false;
|
|
}
|
|
|
|
@Override
|
|
public void onCanceled() {
|
|
|
|
}
|
|
|
|
}
|