nmode's Git Repositories - signal-cli/blob - src/main/java/org/asamk/signal/JsonIdentityKeyStore.java

   1 package org.asamk.signal;
   2
   3 import com.fasterxml.jackson.core.JsonGenerator;
   4 import com.fasterxml.jackson.core.JsonParser;
   5 import com.fasterxml.jackson.core.JsonProcessingException;
   6 import com.fasterxml.jackson.databind.*;
   7 import org.whispersystems.libsignal.IdentityKey;
   8 import org.whispersystems.libsignal.IdentityKeyPair;
   9 import org.whispersystems.libsignal.InvalidKeyException;
  10 import org.whispersystems.libsignal.state.IdentityKeyStore;
  11
  12 import java.io.IOException;
  13 import java.util.*;
  14
  15 class JsonIdentityKeyStore implements IdentityKeyStore {
  16
  17     private final Map<String, List<Identity>> trustedKeys = new HashMap<>();
  18
  19     private final IdentityKeyPair identityKeyPair;
  20     private final int localRegistrationId;
  21
  22
  23     public JsonIdentityKeyStore(IdentityKeyPair identityKeyPair, int localRegistrationId) {
  24         this.identityKeyPair = identityKeyPair;
  25         this.localRegistrationId = localRegistrationId;
  26     }
  27
  28     @Override
  29     public IdentityKeyPair getIdentityKeyPair() {
  30         return identityKeyPair;
  31     }
  32
  33     @Override
  34     public int getLocalRegistrationId() {
  35         return localRegistrationId;
  36     }
  37
  38     @Override
  39     public void saveIdentity(String name, IdentityKey identityKey) {
  40         saveIdentity(name, identityKey, TrustLevel.TRUSTED_UNVERIFIED, null);
  41     }
  42
  43     /**
  44      * Adds or updates the given identityKey for the user name and sets the trustLevel and added timestamp.
  45      *
  46      * @param name        User name, i.e. phone number
  47      * @param identityKey The user's public key
  48      * @param trustLevel
  49      * @param added       Added timestamp, if null and the key is newly added, the current time is used.
  50      */
  51     public void saveIdentity(String name, IdentityKey identityKey, TrustLevel trustLevel, Date added) {
  52         List<Identity> identities = trustedKeys.get(name);
  53         if (identities == null) {
  54             identities = new ArrayList<>();
  55             trustedKeys.put(name, identities);
  56         } else {
  57             for (Identity id : identities) {
  58                 if (!id.identityKey.equals(identityKey))
  59                     continue;
  60
  61                 if (id.trustLevel.compareTo(trustLevel) < 0) {
  62                     id.trustLevel = trustLevel;
  63                 }
  64                 if (added != null) {
  65                     id.added = added;
  66                 }
  67                 return;
  68             }
  69         }
  70         identities.add(new Identity(identityKey, trustLevel, added != null ? added : new Date()));
  71     }
  72
  73     @Override
  74     public boolean isTrustedIdentity(String name, IdentityKey identityKey) {
  75         List<Identity> identities = trustedKeys.get(name);
  76         if (identities == null) {
  77             // Trust on first use
  78             return true;
  79         }
  80
  81         for (Identity id : identities) {
  82             if (id.identityKey.equals(identityKey)) {
  83                 return id.isTrusted();
  84             }
  85         }
  86
  87         return false;
  88     }
  89
  90     public Map<String, List<Identity>> getIdentities() {
  91         // TODO deep copy
  92         return trustedKeys;
  93     }
  94
  95     public List<Identity> getIdentities(String name) {
  96         // TODO deep copy
  97         return trustedKeys.get(name);
  98     }
  99
 100     public static class JsonIdentityKeyStoreDeserializer extends JsonDeserializer<JsonIdentityKeyStore> {
 101
 102         @Override
 103         public JsonIdentityKeyStore deserialize(JsonParser jsonParser, DeserializationContext deserializationContext) throws IOException, JsonProcessingException {
 104             JsonNode node = jsonParser.getCodec().readTree(jsonParser);
 105
 106             try {
 107                 int localRegistrationId = node.get("registrationId").asInt();
 108                 IdentityKeyPair identityKeyPair = new IdentityKeyPair(Base64.decode(node.get("identityKey").asText()));
 109
 110
 111                 JsonIdentityKeyStore keyStore = new JsonIdentityKeyStore(identityKeyPair, localRegistrationId);
 112
 113                 JsonNode trustedKeysNode = node.get("trustedKeys");
 114                 if (trustedKeysNode.isArray()) {
 115                     for (JsonNode trustedKey : trustedKeysNode) {
 116                         String trustedKeyName = trustedKey.get("name").asText();
 117                         try {
 118                             IdentityKey id = new IdentityKey(Base64.decode(trustedKey.get("identityKey").asText()), 0);
 119                             TrustLevel trustLevel = trustedKey.has("trustLevel") ? TrustLevel.fromInt(trustedKey.get("trustLevel").asInt()) : TrustLevel.TRUSTED_UNVERIFIED;
 120                             Date added = trustedKey.has("addedTimestamp") ? new Date(trustedKey.get("addedTimestamp").asLong()) : new Date();
 121                             keyStore.saveIdentity(trustedKeyName, id, trustLevel, added);
 122                         } catch (InvalidKeyException | IOException e) {
 123                             System.out.println(String.format("Error while decoding key for: %s", trustedKeyName));
 124                         }
 125                     }
 126                 }
 127
 128                 return keyStore;
 129             } catch (InvalidKeyException e) {
 130                 throw new IOException(e);
 131             }
 132         }
 133     }
 134
 135     public static class JsonIdentityKeyStoreSerializer extends JsonSerializer<JsonIdentityKeyStore> {
 136
 137         @Override
 138         public void serialize(JsonIdentityKeyStore jsonIdentityKeyStore, JsonGenerator json, SerializerProvider serializerProvider) throws IOException, JsonProcessingException {
 139             json.writeStartObject();
 140             json.writeNumberField("registrationId", jsonIdentityKeyStore.getLocalRegistrationId());
 141             json.writeStringField("identityKey", Base64.encodeBytes(jsonIdentityKeyStore.getIdentityKeyPair().serialize()));
 142             json.writeArrayFieldStart("trustedKeys");
 143             for (Map.Entry<String, List<Identity>> trustedKey : jsonIdentityKeyStore.trustedKeys.entrySet()) {
 144                 for (Identity id : trustedKey.getValue()) {
 145                     json.writeStartObject();
 146                     json.writeStringField("name", trustedKey.getKey());
 147                     json.writeStringField("identityKey", Base64.encodeBytes(id.identityKey.serialize()));
 148                     json.writeNumberField("trustLevel", id.trustLevel.ordinal());
 149                     json.writeNumberField("addedTimestamp", id.added.getTime());
 150                     json.writeEndObject();
 151                 }
 152             }
 153             json.writeEndArray();
 154             json.writeEndObject();
 155         }
 156     }
 157
 158     public class Identity {
 159         IdentityKey identityKey;
 160         TrustLevel trustLevel;
 161         Date added;
 162
 163         public Identity(IdentityKey identityKey, TrustLevel trustLevel) {
 164             this.identityKey = identityKey;
 165             this.trustLevel = trustLevel;
 166             this.added = new Date();
 167         }
 168
 169         public Identity(IdentityKey identityKey, TrustLevel trustLevel, Date added) {
 170             this.identityKey = identityKey;
 171             this.trustLevel = trustLevel;
 172             this.added = added;
 173         }
 174
 175         public boolean isTrusted() {
 176             return trustLevel == TrustLevel.TRUSTED_UNVERIFIED ||
 177                     trustLevel == TrustLevel.TRUSTED_VERIFIED;
 178         }
 179
 180         public byte[] getFingerprint() {
 181             return identityKey.getPublicKey().serialize();
 182         }
 183     }
 184 }