nmode's Git Repositories - signal-cli/blob - lib/src/main/java/org/asamk/signal/manager/storage/protocol/JsonSignalProtocolStore.java

   1 package org.asamk.signal.manager.storage.protocol;
   2
   3 import com.fasterxml.jackson.annotation.JsonProperty;
   4 import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
   5 import com.fasterxml.jackson.databind.annotation.JsonSerialize;
   6
   7 import org.asamk.signal.manager.TrustLevel;
   8 import org.whispersystems.libsignal.IdentityKey;
   9 import org.whispersystems.libsignal.IdentityKeyPair;
  10 import org.whispersystems.libsignal.InvalidKeyIdException;
  11 import org.whispersystems.libsignal.SignalProtocolAddress;
  12 import org.whispersystems.libsignal.state.PreKeyRecord;
  13 import org.whispersystems.libsignal.state.SessionRecord;
  14 import org.whispersystems.libsignal.state.SignalProtocolStore;
  15 import org.whispersystems.libsignal.state.SignedPreKeyRecord;
  16 import org.whispersystems.signalservice.api.push.SignalServiceAddress;
  17
  18 import java.util.List;
  19
  20 public class JsonSignalProtocolStore implements SignalProtocolStore {
  21
  22     @JsonProperty("preKeys")
  23     @JsonDeserialize(using = JsonPreKeyStore.JsonPreKeyStoreDeserializer.class)
  24     @JsonSerialize(using = JsonPreKeyStore.JsonPreKeyStoreSerializer.class)
  25     private JsonPreKeyStore preKeyStore;
  26
  27     @JsonProperty("sessionStore")
  28     @JsonDeserialize(using = JsonSessionStore.JsonSessionStoreDeserializer.class)
  29     @JsonSerialize(using = JsonSessionStore.JsonSessionStoreSerializer.class)
  30     private JsonSessionStore sessionStore;
  31
  32     @JsonProperty("signedPreKeyStore")
  33     @JsonDeserialize(using = JsonSignedPreKeyStore.JsonSignedPreKeyStoreDeserializer.class)
  34     @JsonSerialize(using = JsonSignedPreKeyStore.JsonSignedPreKeyStoreSerializer.class)
  35     private JsonSignedPreKeyStore signedPreKeyStore;
  36
  37     @JsonProperty("identityKeyStore")
  38     @JsonDeserialize(using = JsonIdentityKeyStore.JsonIdentityKeyStoreDeserializer.class)
  39     @JsonSerialize(using = JsonIdentityKeyStore.JsonIdentityKeyStoreSerializer.class)
  40     private JsonIdentityKeyStore identityKeyStore;
  41
  42     public JsonSignalProtocolStore() {
  43     }
  44
  45     public JsonSignalProtocolStore(
  46             JsonPreKeyStore preKeyStore,
  47             JsonSessionStore sessionStore,
  48             JsonSignedPreKeyStore signedPreKeyStore,
  49             JsonIdentityKeyStore identityKeyStore
  50     ) {
  51         this.preKeyStore = preKeyStore;
  52         this.sessionStore = sessionStore;
  53         this.signedPreKeyStore = signedPreKeyStore;
  54         this.identityKeyStore = identityKeyStore;
  55     }
  56
  57     public JsonSignalProtocolStore(IdentityKeyPair identityKeyPair, int registrationId) {
  58         preKeyStore = new JsonPreKeyStore();
  59         sessionStore = new JsonSessionStore();
  60         signedPreKeyStore = new JsonSignedPreKeyStore();
  61         this.identityKeyStore = new JsonIdentityKeyStore(identityKeyPair, registrationId);
  62     }
  63
  64     public void setResolver(final SignalServiceAddressResolver resolver) {
  65         sessionStore.setResolver(resolver);
  66         identityKeyStore.setResolver(resolver);
  67     }
  68
  69     @Override
  70     public IdentityKeyPair getIdentityKeyPair() {
  71         return identityKeyStore.getIdentityKeyPair();
  72     }
  73
  74     @Override
  75     public int getLocalRegistrationId() {
  76         return identityKeyStore.getLocalRegistrationId();
  77     }
  78
  79     @Override
  80     public boolean saveIdentity(SignalProtocolAddress address, IdentityKey identityKey) {
  81         return identityKeyStore.saveIdentity(address, identityKey);
  82     }
  83
  84     public void saveIdentity(SignalServiceAddress serviceAddress, IdentityKey identityKey, TrustLevel trustLevel) {
  85         identityKeyStore.saveIdentity(serviceAddress, identityKey, trustLevel, null);
  86     }
  87
  88     public void setIdentityTrustLevel(
  89             SignalServiceAddress serviceAddress, IdentityKey identityKey, TrustLevel trustLevel
  90     ) {
  91         identityKeyStore.setIdentityTrustLevel(serviceAddress, identityKey, trustLevel);
  92     }
  93
  94     public List<IdentityInfo> getIdentities() {
  95         return identityKeyStore.getIdentities();
  96     }
  97
  98     public List<IdentityInfo> getIdentities(SignalServiceAddress serviceAddress) {
  99         return identityKeyStore.getIdentities(serviceAddress);
 100     }
 101
 102     @Override
 103     public boolean isTrustedIdentity(SignalProtocolAddress address, IdentityKey identityKey, Direction direction) {
 104         return identityKeyStore.isTrustedIdentity(address, identityKey, direction);
 105     }
 106
 107     @Override
 108     public IdentityKey getIdentity(SignalProtocolAddress address) {
 109         return identityKeyStore.getIdentity(address);
 110     }
 111
 112     public IdentityInfo getIdentity(SignalServiceAddress serviceAddress) {
 113         return identityKeyStore.getIdentity(serviceAddress);
 114     }
 115
 116     @Override
 117     public PreKeyRecord loadPreKey(int preKeyId) throws InvalidKeyIdException {
 118         return preKeyStore.loadPreKey(preKeyId);
 119     }
 120
 121     @Override
 122     public void storePreKey(int preKeyId, PreKeyRecord record) {
 123         preKeyStore.storePreKey(preKeyId, record);
 124     }
 125
 126     @Override
 127     public boolean containsPreKey(int preKeyId) {
 128         return preKeyStore.containsPreKey(preKeyId);
 129     }
 130
 131     @Override
 132     public void removePreKey(int preKeyId) {
 133         preKeyStore.removePreKey(preKeyId);
 134     }
 135
 136     @Override
 137     public SessionRecord loadSession(SignalProtocolAddress address) {
 138         return sessionStore.loadSession(address);
 139     }
 140
 141     public List<SessionInfo> getSessions() {
 142         return sessionStore.getSessions();
 143     }
 144
 145     @Override
 146     public List<Integer> getSubDeviceSessions(String name) {
 147         return sessionStore.getSubDeviceSessions(name);
 148     }
 149
 150     @Override
 151     public void storeSession(SignalProtocolAddress address, SessionRecord record) {
 152         sessionStore.storeSession(address, record);
 153     }
 154
 155     @Override
 156     public boolean containsSession(SignalProtocolAddress address) {
 157         return sessionStore.containsSession(address);
 158     }
 159
 160     @Override
 161     public void deleteSession(SignalProtocolAddress address) {
 162         sessionStore.deleteSession(address);
 163     }
 164
 165     @Override
 166     public void deleteAllSessions(String name) {
 167         sessionStore.deleteAllSessions(name);
 168     }
 169
 170     public void deleteAllSessions(SignalServiceAddress serviceAddress) {
 171         sessionStore.deleteAllSessions(serviceAddress);
 172     }
 173
 174     @Override
 175     public SignedPreKeyRecord loadSignedPreKey(int signedPreKeyId) throws InvalidKeyIdException {
 176         return signedPreKeyStore.loadSignedPreKey(signedPreKeyId);
 177     }
 178
 179     @Override
 180     public List<SignedPreKeyRecord> loadSignedPreKeys() {
 181         return signedPreKeyStore.loadSignedPreKeys();
 182     }
 183
 184     @Override
 185     public void storeSignedPreKey(int signedPreKeyId, SignedPreKeyRecord record) {
 186         signedPreKeyStore.storeSignedPreKey(signedPreKeyId, record);
 187     }
 188
 189     @Override
 190     public boolean containsSignedPreKey(int signedPreKeyId) {
 191         return signedPreKeyStore.containsSignedPreKey(signedPreKeyId);
 192     }
 193
 194     @Override
 195     public void removeSignedPreKey(int signedPreKeyId) {
 196         signedPreKeyStore.removeSignedPreKey(signedPreKeyId);
 197     }
 198 }