91 lines
2.9 KiB
Java
91 lines
2.9 KiB
Java
package org.thoughtcrime.securesms.crypto.storage;
|
|
|
|
import android.content.Context;
|
|
import androidx.annotation.NonNull;
|
|
|
|
import org.thoughtcrime.securesms.database.DatabaseFactory;
|
|
import org.session.libsignal.libsignal.InvalidKeyIdException;
|
|
import org.session.libsignal.libsignal.state.PreKeyRecord;
|
|
import org.session.libsignal.libsignal.state.PreKeyStore;
|
|
import org.session.libsignal.libsignal.state.SignedPreKeyRecord;
|
|
import org.session.libsignal.libsignal.state.SignedPreKeyStore;
|
|
|
|
import java.util.List;
|
|
|
|
public class TextSecurePreKeyStore implements PreKeyStore, SignedPreKeyStore {
|
|
|
|
@SuppressWarnings("unused")
|
|
private static final String TAG = TextSecurePreKeyStore.class.getSimpleName();
|
|
|
|
private static final Object FILE_LOCK = new Object();
|
|
|
|
@NonNull
|
|
private final Context context;
|
|
|
|
public TextSecurePreKeyStore(@NonNull Context context) {
|
|
this.context = context;
|
|
}
|
|
|
|
@Override
|
|
public PreKeyRecord loadPreKey(int preKeyId) throws InvalidKeyIdException {
|
|
synchronized (FILE_LOCK) {
|
|
PreKeyRecord preKeyRecord = DatabaseFactory.getPreKeyDatabase(context).getPreKey(preKeyId);
|
|
|
|
if (preKeyRecord == null) throw new InvalidKeyIdException("No such key: " + preKeyId);
|
|
else return preKeyRecord;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public SignedPreKeyRecord loadSignedPreKey(int signedPreKeyId) throws InvalidKeyIdException {
|
|
synchronized (FILE_LOCK) {
|
|
SignedPreKeyRecord signedPreKeyRecord = DatabaseFactory.getSignedPreKeyDatabase(context).getSignedPreKey(signedPreKeyId);
|
|
|
|
if (signedPreKeyRecord == null) throw new InvalidKeyIdException("No such signed prekey: " + signedPreKeyId);
|
|
else return signedPreKeyRecord;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public List<SignedPreKeyRecord> loadSignedPreKeys() {
|
|
synchronized (FILE_LOCK) {
|
|
return DatabaseFactory.getSignedPreKeyDatabase(context).getAllSignedPreKeys();
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void storePreKey(int preKeyId, PreKeyRecord record) {
|
|
synchronized (FILE_LOCK) {
|
|
DatabaseFactory.getPreKeyDatabase(context).insertPreKey(preKeyId, record);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void storeSignedPreKey(int signedPreKeyId, SignedPreKeyRecord record) {
|
|
synchronized (FILE_LOCK) {
|
|
DatabaseFactory.getSignedPreKeyDatabase(context).insertSignedPreKey(signedPreKeyId, record);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public boolean containsPreKey(int preKeyId) {
|
|
return DatabaseFactory.getPreKeyDatabase(context).getPreKey(preKeyId) != null;
|
|
}
|
|
|
|
@Override
|
|
public boolean containsSignedPreKey(int signedPreKeyId) {
|
|
return DatabaseFactory.getSignedPreKeyDatabase(context).getSignedPreKey(signedPreKeyId) != null;
|
|
}
|
|
|
|
@Override
|
|
public void removePreKey(int preKeyId) {
|
|
DatabaseFactory.getPreKeyDatabase(context).removePreKey(preKeyId);
|
|
}
|
|
|
|
@Override
|
|
public void removeSignedPreKey(int signedPreKeyId) {
|
|
DatabaseFactory.getSignedPreKeyDatabase(context).removeSignedPreKey(signedPreKeyId);
|
|
}
|
|
|
|
}
|