116 lines
3.4 KiB
Java
116 lines
3.4 KiB
Java
package org.whispersystems.test;
|
|
|
|
import org.whispersystems.libaxolotl.IdentityKey;
|
|
import org.whispersystems.libaxolotl.IdentityKeyPair;
|
|
import org.whispersystems.libaxolotl.InvalidKeyIdException;
|
|
import org.whispersystems.libaxolotl.state.AxolotlStore;
|
|
import org.whispersystems.libaxolotl.state.PreKeyRecord;
|
|
import org.whispersystems.libaxolotl.state.SessionRecord;
|
|
import org.whispersystems.libaxolotl.state.SignedPreKeyRecord;
|
|
|
|
import java.util.List;
|
|
|
|
public class InMemoryAxolotlStore implements AxolotlStore {
|
|
|
|
private final InMemoryIdentityKeyStore identityKeyStore = new InMemoryIdentityKeyStore();
|
|
private final InMemoryPreKeyStore preKeyStore = new InMemoryPreKeyStore();
|
|
private final InMemorySessionStore sessionStore = new InMemorySessionStore();
|
|
private final InMemorySignedPreKeyStore signedPreKeyStore = new InMemorySignedPreKeyStore();
|
|
|
|
|
|
@Override
|
|
public IdentityKeyPair getIdentityKeyPair() {
|
|
return identityKeyStore.getIdentityKeyPair();
|
|
}
|
|
|
|
@Override
|
|
public int getLocalRegistrationId() {
|
|
return identityKeyStore.getLocalRegistrationId();
|
|
}
|
|
|
|
@Override
|
|
public void saveIdentity(long recipientId, IdentityKey identityKey) {
|
|
identityKeyStore.saveIdentity(recipientId, identityKey);
|
|
}
|
|
|
|
@Override
|
|
public boolean isTrustedIdentity(long recipientId, IdentityKey identityKey) {
|
|
return identityKeyStore.isTrustedIdentity(recipientId, identityKey);
|
|
}
|
|
|
|
@Override
|
|
public PreKeyRecord loadPreKey(int preKeyId) throws InvalidKeyIdException {
|
|
return preKeyStore.loadPreKey(preKeyId);
|
|
}
|
|
|
|
@Override
|
|
public void storePreKey(int preKeyId, PreKeyRecord record) {
|
|
preKeyStore.storePreKey(preKeyId, record);
|
|
}
|
|
|
|
@Override
|
|
public boolean containsPreKey(int preKeyId) {
|
|
return preKeyStore.containsPreKey(preKeyId);
|
|
}
|
|
|
|
@Override
|
|
public void removePreKey(int preKeyId) {
|
|
preKeyStore.removePreKey(preKeyId);
|
|
}
|
|
|
|
@Override
|
|
public SessionRecord loadSession(long recipientId, int deviceId) {
|
|
return sessionStore.loadSession(recipientId, deviceId);
|
|
}
|
|
|
|
@Override
|
|
public List<Integer> getSubDeviceSessions(long recipientId) {
|
|
return sessionStore.getSubDeviceSessions(recipientId);
|
|
}
|
|
|
|
@Override
|
|
public void storeSession(long recipientId, int deviceId, SessionRecord record) {
|
|
sessionStore.storeSession(recipientId, deviceId, record);
|
|
}
|
|
|
|
@Override
|
|
public boolean containsSession(long recipientId, int deviceId) {
|
|
return sessionStore.containsSession(recipientId, deviceId);
|
|
}
|
|
|
|
@Override
|
|
public void deleteSession(long recipientId, int deviceId) {
|
|
sessionStore.deleteSession(recipientId, deviceId);
|
|
}
|
|
|
|
@Override
|
|
public void deleteAllSessions(long recipientId) {
|
|
sessionStore.deleteAllSessions(recipientId);
|
|
}
|
|
|
|
@Override
|
|
public SignedPreKeyRecord loadSignedPreKey(int signedPreKeyId) throws InvalidKeyIdException {
|
|
return signedPreKeyStore.loadSignedPreKey(signedPreKeyId);
|
|
}
|
|
|
|
@Override
|
|
public List<SignedPreKeyRecord> loadSignedPreKeys() {
|
|
return signedPreKeyStore.loadSignedPreKeys();
|
|
}
|
|
|
|
@Override
|
|
public void storeSignedPreKey(int signedPreKeyId, SignedPreKeyRecord record) {
|
|
signedPreKeyStore.storeSignedPreKey(signedPreKeyId, record);
|
|
}
|
|
|
|
@Override
|
|
public boolean containsSignedPreKey(int signedPreKeyId) {
|
|
return signedPreKeyStore.containsSignedPreKey(signedPreKeyId);
|
|
}
|
|
|
|
@Override
|
|
public void removeSignedPreKey(int signedPreKeyId) {
|
|
signedPreKeyStore.removeSignedPreKey(signedPreKeyId);
|
|
}
|
|
}
|