1 package org
.asamk
.signal
.manager
.storage
;
3 import com
.fasterxml
.jackson
.databind
.JsonNode
;
4 import com
.fasterxml
.jackson
.databind
.ObjectMapper
;
6 import org
.asamk
.signal
.manager
.Settings
;
7 import org
.asamk
.signal
.manager
.api
.Pair
;
8 import org
.asamk
.signal
.manager
.api
.TrustLevel
;
9 import org
.asamk
.signal
.manager
.config
.ServiceEnvironment
;
10 import org
.asamk
.signal
.manager
.groups
.GroupId
;
11 import org
.asamk
.signal
.manager
.helper
.RecipientAddressResolver
;
12 import org
.asamk
.signal
.manager
.storage
.configuration
.ConfigurationStore
;
13 import org
.asamk
.signal
.manager
.storage
.contacts
.ContactsStore
;
14 import org
.asamk
.signal
.manager
.storage
.contacts
.LegacyJsonContactsStore
;
15 import org
.asamk
.signal
.manager
.storage
.groups
.GroupInfoV1
;
16 import org
.asamk
.signal
.manager
.storage
.groups
.GroupStore
;
17 import org
.asamk
.signal
.manager
.storage
.groups
.LegacyGroupStore
;
18 import org
.asamk
.signal
.manager
.storage
.identities
.IdentityKeyStore
;
19 import org
.asamk
.signal
.manager
.storage
.identities
.LegacyIdentityKeyStore
;
20 import org
.asamk
.signal
.manager
.storage
.identities
.SignalIdentityKeyStore
;
21 import org
.asamk
.signal
.manager
.storage
.messageCache
.MessageCache
;
22 import org
.asamk
.signal
.manager
.storage
.prekeys
.LegacyPreKeyStore
;
23 import org
.asamk
.signal
.manager
.storage
.prekeys
.LegacySignedPreKeyStore
;
24 import org
.asamk
.signal
.manager
.storage
.prekeys
.PreKeyStore
;
25 import org
.asamk
.signal
.manager
.storage
.prekeys
.SignedPreKeyStore
;
26 import org
.asamk
.signal
.manager
.storage
.profiles
.LegacyProfileStore
;
27 import org
.asamk
.signal
.manager
.storage
.profiles
.ProfileStore
;
28 import org
.asamk
.signal
.manager
.storage
.protocol
.LegacyJsonSignalProtocolStore
;
29 import org
.asamk
.signal
.manager
.storage
.protocol
.SignalProtocolStore
;
30 import org
.asamk
.signal
.manager
.storage
.recipients
.Contact
;
31 import org
.asamk
.signal
.manager
.storage
.recipients
.LegacyRecipientStore
;
32 import org
.asamk
.signal
.manager
.storage
.recipients
.LegacyRecipientStore2
;
33 import org
.asamk
.signal
.manager
.storage
.recipients
.Profile
;
34 import org
.asamk
.signal
.manager
.storage
.recipients
.RecipientAddress
;
35 import org
.asamk
.signal
.manager
.storage
.recipients
.RecipientId
;
36 import org
.asamk
.signal
.manager
.storage
.recipients
.RecipientIdCreator
;
37 import org
.asamk
.signal
.manager
.storage
.recipients
.RecipientResolver
;
38 import org
.asamk
.signal
.manager
.storage
.recipients
.RecipientStore
;
39 import org
.asamk
.signal
.manager
.storage
.recipients
.RecipientTrustedResolver
;
40 import org
.asamk
.signal
.manager
.storage
.sendLog
.MessageSendLogStore
;
41 import org
.asamk
.signal
.manager
.storage
.senderKeys
.LegacySenderKeyRecordStore
;
42 import org
.asamk
.signal
.manager
.storage
.senderKeys
.LegacySenderKeySharedStore
;
43 import org
.asamk
.signal
.manager
.storage
.senderKeys
.SenderKeyStore
;
44 import org
.asamk
.signal
.manager
.storage
.sessions
.LegacySessionStore
;
45 import org
.asamk
.signal
.manager
.storage
.sessions
.SessionStore
;
46 import org
.asamk
.signal
.manager
.storage
.stickers
.LegacyStickerStore
;
47 import org
.asamk
.signal
.manager
.storage
.stickers
.StickerStore
;
48 import org
.asamk
.signal
.manager
.storage
.threads
.LegacyJsonThreadStore
;
49 import org
.asamk
.signal
.manager
.util
.IOUtils
;
50 import org
.asamk
.signal
.manager
.util
.KeyUtils
;
51 import org
.signal
.libsignal
.protocol
.IdentityKeyPair
;
52 import org
.signal
.libsignal
.protocol
.InvalidMessageException
;
53 import org
.signal
.libsignal
.protocol
.SignalProtocolAddress
;
54 import org
.signal
.libsignal
.protocol
.state
.PreKeyRecord
;
55 import org
.signal
.libsignal
.protocol
.state
.SessionRecord
;
56 import org
.signal
.libsignal
.protocol
.state
.SignedPreKeyRecord
;
57 import org
.signal
.libsignal
.protocol
.util
.KeyHelper
;
58 import org
.signal
.libsignal
.protocol
.util
.Medium
;
59 import org
.signal
.libsignal
.zkgroup
.InvalidInputException
;
60 import org
.signal
.libsignal
.zkgroup
.profiles
.ProfileKey
;
61 import org
.slf4j
.Logger
;
62 import org
.slf4j
.LoggerFactory
;
63 import org
.whispersystems
.signalservice
.api
.SignalServiceAccountDataStore
;
64 import org
.whispersystems
.signalservice
.api
.SignalServiceDataStore
;
65 import org
.whispersystems
.signalservice
.api
.crypto
.UnidentifiedAccess
;
66 import org
.whispersystems
.signalservice
.api
.kbs
.MasterKey
;
67 import org
.whispersystems
.signalservice
.api
.push
.ACI
;
68 import org
.whispersystems
.signalservice
.api
.push
.PNI
;
69 import org
.whispersystems
.signalservice
.api
.push
.ServiceId
;
70 import org
.whispersystems
.signalservice
.api
.push
.ServiceIdType
;
71 import org
.whispersystems
.signalservice
.api
.push
.SignalServiceAddress
;
72 import org
.whispersystems
.signalservice
.api
.storage
.SignalStorageManifest
;
73 import org
.whispersystems
.signalservice
.api
.storage
.StorageKey
;
74 import org
.whispersystems
.signalservice
.api
.util
.CredentialsProvider
;
75 import org
.whispersystems
.signalservice
.api
.util
.UuidUtil
;
77 import java
.io
.ByteArrayInputStream
;
78 import java
.io
.ByteArrayOutputStream
;
79 import java
.io
.Closeable
;
81 import java
.io
.FileInputStream
;
82 import java
.io
.FileOutputStream
;
83 import java
.io
.IOException
;
84 import java
.io
.RandomAccessFile
;
85 import java
.nio
.channels
.Channels
;
86 import java
.nio
.channels
.ClosedChannelException
;
87 import java
.nio
.channels
.FileChannel
;
88 import java
.nio
.channels
.FileLock
;
89 import java
.nio
.file
.Files
;
90 import java
.security
.SecureRandom
;
91 import java
.sql
.Connection
;
92 import java
.sql
.SQLException
;
93 import java
.util
.Base64
;
94 import java
.util
.Comparator
;
95 import java
.util
.HashSet
;
96 import java
.util
.List
;
97 import java
.util
.Optional
;
98 import java
.util
.function
.Supplier
;
100 public class SignalAccount
implements Closeable
{
102 private final static Logger logger
= LoggerFactory
.getLogger(SignalAccount
.class);
104 private static final int MINIMUM_STORAGE_VERSION
= 1;
105 private static final int CURRENT_STORAGE_VERSION
= 6;
107 private final Object LOCK
= new Object();
109 private final ObjectMapper jsonProcessor
= Utils
.createStorageObjectMapper();
111 private final FileChannel fileChannel
;
112 private final FileLock lock
;
114 private int previousStorageVersion
;
116 private File dataPath
;
117 private String accountPath
;
118 private ServiceEnvironment serviceEnvironment
;
119 private String number
;
122 private String encryptedDeviceName
;
123 private int deviceId
= SignalServiceAddress
.DEFAULT_DEVICE_ID
;
124 private boolean isMultiDevice
= false;
125 private String password
;
126 private String registrationLockPin
;
127 private MasterKey pinMasterKey
;
128 private StorageKey storageKey
;
129 private long storageManifestVersion
= -1;
130 private ProfileKey profileKey
;
131 private int aciPreKeyIdOffset
= 1;
132 private int aciNextSignedPreKeyId
= 1;
133 private int pniPreKeyIdOffset
= 1;
134 private int pniNextSignedPreKeyId
= 1;
135 private IdentityKeyPair aciIdentityKeyPair
;
136 private IdentityKeyPair pniIdentityKeyPair
;
137 private int localRegistrationId
;
138 private int localPniRegistrationId
;
139 private Settings settings
;
140 private long lastReceiveTimestamp
= 0;
142 private boolean registered
= false;
144 private SignalProtocolStore aciSignalProtocolStore
;
145 private SignalProtocolStore pniSignalProtocolStore
;
146 private PreKeyStore aciPreKeyStore
;
147 private SignedPreKeyStore aciSignedPreKeyStore
;
148 private PreKeyStore pniPreKeyStore
;
149 private SignedPreKeyStore pniSignedPreKeyStore
;
150 private SessionStore aciSessionStore
;
151 private SessionStore pniSessionStore
;
152 private IdentityKeyStore identityKeyStore
;
153 private SignalIdentityKeyStore aciIdentityKeyStore
;
154 private SignalIdentityKeyStore pniIdentityKeyStore
;
155 private SenderKeyStore senderKeyStore
;
156 private GroupStore groupStore
;
157 private RecipientStore recipientStore
;
158 private StickerStore stickerStore
;
159 private ConfigurationStore configurationStore
;
160 private ConfigurationStore
.Storage configurationStoreStorage
;
162 private MessageCache messageCache
;
163 private MessageSendLogStore messageSendLogStore
;
165 private AccountDatabase accountDatabase
;
167 private SignalAccount(final FileChannel fileChannel
, final FileLock lock
) {
168 this.fileChannel
= fileChannel
;
172 public static SignalAccount
load(
173 File dataPath
, String accountPath
, boolean waitForLock
, final Settings settings
174 ) throws IOException
{
175 logger
.trace("Opening account file");
176 final var fileName
= getFileName(dataPath
, accountPath
);
177 final var pair
= openFileChannel(fileName
, waitForLock
);
179 var signalAccount
= new SignalAccount(pair
.first(), pair
.second());
180 logger
.trace("Loading account file");
181 signalAccount
.load(dataPath
, accountPath
, settings
);
182 logger
.trace("Migrating legacy parts of account file");
183 signalAccount
.migrateLegacyConfigs();
185 return signalAccount
;
186 } catch (Throwable e
) {
187 pair
.second().close();
188 pair
.first().close();
193 public static SignalAccount
create(
197 ServiceEnvironment serviceEnvironment
,
198 IdentityKeyPair aciIdentityKey
,
199 IdentityKeyPair pniIdentityKey
,
201 int pniRegistrationId
,
202 ProfileKey profileKey
,
203 final Settings settings
204 ) throws IOException
{
205 IOUtils
.createPrivateDirectories(dataPath
);
206 var fileName
= getFileName(dataPath
, accountPath
);
207 if (!fileName
.exists()) {
208 IOUtils
.createPrivateFile(fileName
);
211 final var pair
= openFileChannel(fileName
, true);
212 var signalAccount
= new SignalAccount(pair
.first(), pair
.second());
214 signalAccount
.accountPath
= accountPath
;
215 signalAccount
.number
= number
;
216 signalAccount
.serviceEnvironment
= serviceEnvironment
;
217 signalAccount
.profileKey
= profileKey
;
219 signalAccount
.dataPath
= dataPath
;
220 signalAccount
.aciIdentityKeyPair
= aciIdentityKey
;
221 signalAccount
.pniIdentityKeyPair
= pniIdentityKey
;
222 signalAccount
.localRegistrationId
= registrationId
;
223 signalAccount
.localPniRegistrationId
= pniRegistrationId
;
224 signalAccount
.settings
= settings
;
225 signalAccount
.configurationStore
= new ConfigurationStore(signalAccount
::saveConfigurationStore
);
227 signalAccount
.registered
= false;
229 signalAccount
.previousStorageVersion
= CURRENT_STORAGE_VERSION
;
230 signalAccount
.migrateLegacyConfigs();
231 signalAccount
.save();
233 return signalAccount
;
236 public static SignalAccount
createOrUpdateLinkedAccount(
240 ServiceEnvironment serviceEnvironment
,
244 String encryptedDeviceName
,
246 IdentityKeyPair aciIdentityKey
,
247 IdentityKeyPair pniIdentityKey
,
249 int pniRegistrationId
,
250 ProfileKey profileKey
,
251 final Settings settings
252 ) throws IOException
{
253 IOUtils
.createPrivateDirectories(dataPath
);
254 var fileName
= getFileName(dataPath
, accountPath
);
255 if (!fileName
.exists()) {
256 return createLinkedAccount(dataPath
,
273 final var signalAccount
= load(dataPath
, accountPath
, true, settings
);
274 signalAccount
.setProvisioningData(number
,
283 signalAccount
.getRecipientTrustedResolver()
284 .resolveSelfRecipientTrusted(signalAccount
.getSelfRecipientAddress());
285 signalAccount
.getAciSessionStore().archiveAllSessions();
286 signalAccount
.getPniSessionStore().archiveAllSessions();
287 signalAccount
.getSenderKeyStore().deleteAll();
288 signalAccount
.clearAllPreKeys();
289 return signalAccount
;
292 public void initDatabase() {
293 getAccountDatabase();
296 private void clearAllPreKeys() {
297 resetPreKeyOffsets(ServiceIdType
.ACI
);
298 resetPreKeyOffsets(ServiceIdType
.PNI
);
299 this.getAciPreKeyStore().removeAllPreKeys();
300 this.getAciSignedPreKeyStore().removeAllSignedPreKeys();
301 this.getPniPreKeyStore().removeAllPreKeys();
302 this.getPniSignedPreKeyStore().removeAllSignedPreKeys();
306 private static SignalAccount
createLinkedAccount(
310 ServiceEnvironment serviceEnvironment
,
314 String encryptedDeviceName
,
316 IdentityKeyPair aciIdentityKey
,
317 IdentityKeyPair pniIdentityKey
,
319 int pniRegistrationId
,
320 ProfileKey profileKey
,
321 final Settings settings
322 ) throws IOException
{
323 var fileName
= getFileName(dataPath
, accountPath
);
324 IOUtils
.createPrivateFile(fileName
);
326 final var pair
= openFileChannel(fileName
, true);
327 var signalAccount
= new SignalAccount(pair
.first(), pair
.second());
329 signalAccount
.dataPath
= dataPath
;
330 signalAccount
.accountPath
= accountPath
;
331 signalAccount
.serviceEnvironment
= serviceEnvironment
;
332 signalAccount
.localRegistrationId
= registrationId
;
333 signalAccount
.localPniRegistrationId
= pniRegistrationId
;
334 signalAccount
.settings
= settings
;
335 signalAccount
.setProvisioningData(number
,
345 signalAccount
.configurationStore
= new ConfigurationStore(signalAccount
::saveConfigurationStore
);
347 signalAccount
.getRecipientTrustedResolver()
348 .resolveSelfRecipientTrusted(signalAccount
.getSelfRecipientAddress());
349 signalAccount
.previousStorageVersion
= CURRENT_STORAGE_VERSION
;
350 signalAccount
.migrateLegacyConfigs();
351 signalAccount
.clearAllPreKeys();
352 signalAccount
.save();
354 return signalAccount
;
357 private void setProvisioningData(
361 final String password
,
362 final String encryptedDeviceName
,
364 final IdentityKeyPair aciIdentity
,
365 final IdentityKeyPair pniIdentity
,
366 final ProfileKey profileKey
368 this.number
= number
;
371 this.password
= password
;
372 this.profileKey
= profileKey
;
373 getProfileStore().storeSelfProfileKey(getSelfRecipientId(), getProfileKey());
374 this.encryptedDeviceName
= encryptedDeviceName
;
375 this.deviceId
= deviceId
;
376 this.aciIdentityKeyPair
= aciIdentity
;
377 this.pniIdentityKeyPair
= pniIdentity
;
378 this.registered
= true;
379 this.isMultiDevice
= true;
380 this.lastReceiveTimestamp
= 0;
381 this.pinMasterKey
= null;
382 this.storageManifestVersion
= -1;
383 this.setStorageManifest(null);
384 this.storageKey
= null;
385 final var aciPublicKey
= getAciIdentityKeyPair().getPublicKey();
386 getIdentityKeyStore().saveIdentity(getAci(), aciPublicKey
);
387 getIdentityKeyStore().setIdentityTrustLevel(getAci(), aciPublicKey
, TrustLevel
.TRUSTED_VERIFIED
);
388 if (getPniIdentityKeyPair() != null) {
389 final var pniPublicKey
= getPniIdentityKeyPair().getPublicKey();
390 getIdentityKeyStore().saveIdentity(getPni(), pniPublicKey
);
391 getIdentityKeyStore().setIdentityTrustLevel(getPni(), pniPublicKey
, TrustLevel
.TRUSTED_VERIFIED
);
395 private void migrateLegacyConfigs() {
396 if (getPassword() == null) {
397 setPassword(KeyUtils
.createPassword());
400 if (getProfileKey() == null) {
401 // Old config file, creating new profile key
402 setProfileKey(KeyUtils
.createProfileKey());
404 getProfileStore().storeProfileKey(getSelfRecipientId(), getProfileKey());
405 if (isPrimaryDevice() && getPniIdentityKeyPair() == null && getPni() != null) {
406 setPniIdentityKeyPair(KeyUtils
.generateIdentityKeyPair());
410 private void mergeRecipients(
411 final Connection connection
, RecipientId recipientId
, RecipientId toBeMergedRecipientId
412 ) throws SQLException
{
413 getMessageCache().mergeRecipients(recipientId
, toBeMergedRecipientId
);
414 getGroupStore().mergeRecipients(connection
, recipientId
, toBeMergedRecipientId
);
417 public void removeRecipient(final RecipientId recipientId
) {
418 getRecipientStore().deleteRecipientData(recipientId
);
419 getMessageCache().deleteMessages(recipientId
);
420 final var recipientAddress
= getRecipientStore().resolveRecipientAddress(recipientId
);
421 if (recipientAddress
.serviceId().isPresent()) {
422 final var serviceId
= recipientAddress
.serviceId().get();
423 getAciSessionStore().deleteAllSessions(serviceId
);
424 getPniSessionStore().deleteAllSessions(serviceId
);
425 getIdentityKeyStore().deleteIdentity(serviceId
);
426 getSenderKeyStore().deleteAll(serviceId
);
430 public static File
getFileName(File dataPath
, String account
) {
431 return new File(dataPath
, account
);
434 private static File
getUserPath(final File dataPath
, final String account
) {
435 final var path
= new File(dataPath
, account
+ ".d");
437 IOUtils
.createPrivateDirectories(path
);
438 } catch (IOException e
) {
439 throw new AssertionError("Failed to create user path", e
);
444 private static File
getMessageCachePath(File dataPath
, String account
) {
445 return new File(getUserPath(dataPath
, account
), "msg-cache");
448 private static File
getGroupCachePath(File dataPath
, String account
) {
449 return new File(getUserPath(dataPath
, account
), "group-cache");
452 private static File
getAciPreKeysPath(File dataPath
, String account
) {
453 return new File(getUserPath(dataPath
, account
), "pre-keys");
456 private static File
getAciSignedPreKeysPath(File dataPath
, String account
) {
457 return new File(getUserPath(dataPath
, account
), "signed-pre-keys");
460 private static File
getPniPreKeysPath(File dataPath
, String account
) {
461 return new File(getUserPath(dataPath
, account
), "pre-keys-pni");
464 private static File
getPniSignedPreKeysPath(File dataPath
, String account
) {
465 return new File(getUserPath(dataPath
, account
), "signed-pre-keys-pni");
468 private static File
getIdentitiesPath(File dataPath
, String account
) {
469 return new File(getUserPath(dataPath
, account
), "identities");
472 private static File
getSessionsPath(File dataPath
, String account
) {
473 return new File(getUserPath(dataPath
, account
), "sessions");
476 private static File
getSenderKeysPath(File dataPath
, String account
) {
477 return new File(getUserPath(dataPath
, account
), "sender-keys");
480 private static File
getSharedSenderKeysFile(File dataPath
, String account
) {
481 return new File(getUserPath(dataPath
, account
), "shared-sender-keys-store");
484 private static File
getRecipientsStoreFile(File dataPath
, String account
) {
485 return new File(getUserPath(dataPath
, account
), "recipients-store");
488 private static File
getStorageManifestFile(File dataPath
, String account
) {
489 return new File(getUserPath(dataPath
, account
), "storage-manifest");
492 private static File
getDatabaseFile(File dataPath
, String account
) {
493 return new File(getUserPath(dataPath
, account
), "account.db");
496 public static boolean accountFileExists(File dataPath
, String account
) {
497 if (account
== null) {
500 var f
= getFileName(dataPath
, account
);
501 return !(!f
.exists() || f
.isDirectory());
505 File dataPath
, String accountPath
, final Settings settings
506 ) throws IOException
{
507 this.dataPath
= dataPath
;
508 this.accountPath
= accountPath
;
509 final JsonNode rootNode
;
510 synchronized (fileChannel
) {
511 fileChannel
.position(0);
512 rootNode
= jsonProcessor
.readTree(Channels
.newInputStream(fileChannel
));
515 var migratedLegacyConfig
= false;
517 if (rootNode
.hasNonNull("version")) {
518 var accountVersion
= rootNode
.get("version").asInt(1);
519 if (accountVersion
> CURRENT_STORAGE_VERSION
) {
520 throw new IOException("Config file was created by a more recent version: " + accountVersion
);
521 } else if (accountVersion
< MINIMUM_STORAGE_VERSION
) {
522 throw new IOException("Config file was created by a no longer supported older version: "
525 previousStorageVersion
= accountVersion
;
526 if (accountVersion
< CURRENT_STORAGE_VERSION
) {
527 migratedLegacyConfig
= true;
531 number
= Utils
.getNotNullNode(rootNode
, "username").asText();
532 if (rootNode
.hasNonNull("password")) {
533 password
= rootNode
.get("password").asText();
535 if (rootNode
.hasNonNull("serviceEnvironment")) {
536 serviceEnvironment
= ServiceEnvironment
.valueOf(rootNode
.get("serviceEnvironment").asText());
538 registered
= Utils
.getNotNullNode(rootNode
, "registered").asBoolean();
539 if (rootNode
.hasNonNull("uuid")) {
541 aci
= ACI
.parseOrThrow(rootNode
.get("uuid").asText());
542 } catch (IllegalArgumentException e
) {
543 throw new IOException("Config file contains an invalid aci/uuid, needs to be a valid UUID", e
);
546 if (rootNode
.hasNonNull("pni")) {
548 pni
= PNI
.parseOrThrow(rootNode
.get("pni").asText());
549 } catch (IllegalArgumentException e
) {
550 throw new IOException("Config file contains an invalid pni, needs to be a valid UUID", e
);
553 if (rootNode
.hasNonNull("deviceName")) {
554 encryptedDeviceName
= rootNode
.get("deviceName").asText();
556 if (rootNode
.hasNonNull("deviceId")) {
557 deviceId
= rootNode
.get("deviceId").asInt();
559 if (rootNode
.hasNonNull("isMultiDevice")) {
560 isMultiDevice
= rootNode
.get("isMultiDevice").asBoolean();
562 if (rootNode
.hasNonNull("lastReceiveTimestamp")) {
563 lastReceiveTimestamp
= rootNode
.get("lastReceiveTimestamp").asLong();
565 int registrationId
= 0;
566 if (rootNode
.hasNonNull("registrationId")) {
567 registrationId
= rootNode
.get("registrationId").asInt();
569 if (rootNode
.hasNonNull("pniRegistrationId")) {
570 localPniRegistrationId
= rootNode
.get("pniRegistrationId").asInt();
572 localPniRegistrationId
= KeyHelper
.generateRegistrationId(false);
574 IdentityKeyPair aciIdentityKeyPair
= null;
575 if (rootNode
.hasNonNull("identityPrivateKey") && rootNode
.hasNonNull("identityKey")) {
576 final var publicKeyBytes
= Base64
.getDecoder().decode(rootNode
.get("identityKey").asText());
577 final var privateKeyBytes
= Base64
.getDecoder().decode(rootNode
.get("identityPrivateKey").asText());
578 aciIdentityKeyPair
= KeyUtils
.getIdentityKeyPair(publicKeyBytes
, privateKeyBytes
);
580 if (rootNode
.hasNonNull("pniIdentityPrivateKey") && rootNode
.hasNonNull("pniIdentityKey")) {
581 final var publicKeyBytes
= Base64
.getDecoder().decode(rootNode
.get("pniIdentityKey").asText());
582 final var privateKeyBytes
= Base64
.getDecoder().decode(rootNode
.get("pniIdentityPrivateKey").asText());
583 pniIdentityKeyPair
= KeyUtils
.getIdentityKeyPair(publicKeyBytes
, privateKeyBytes
);
586 if (rootNode
.hasNonNull("registrationLockPin")) {
587 registrationLockPin
= rootNode
.get("registrationLockPin").asText();
589 if (rootNode
.hasNonNull("pinMasterKey")) {
590 pinMasterKey
= new MasterKey(Base64
.getDecoder().decode(rootNode
.get("pinMasterKey").asText()));
592 if (rootNode
.hasNonNull("storageKey")) {
593 storageKey
= new StorageKey(Base64
.getDecoder().decode(rootNode
.get("storageKey").asText()));
595 if (rootNode
.hasNonNull("storageManifestVersion")) {
596 storageManifestVersion
= rootNode
.get("storageManifestVersion").asLong();
598 if (rootNode
.hasNonNull("preKeyIdOffset")) {
599 aciPreKeyIdOffset
= rootNode
.get("preKeyIdOffset").asInt(1);
601 aciPreKeyIdOffset
= 1;
603 if (rootNode
.hasNonNull("nextSignedPreKeyId")) {
604 aciNextSignedPreKeyId
= rootNode
.get("nextSignedPreKeyId").asInt(1);
606 aciNextSignedPreKeyId
= 1;
608 if (rootNode
.hasNonNull("pniPreKeyIdOffset")) {
609 pniPreKeyIdOffset
= rootNode
.get("pniPreKeyIdOffset").asInt(1);
611 pniPreKeyIdOffset
= 1;
613 if (rootNode
.hasNonNull("pniNextSignedPreKeyId")) {
614 pniNextSignedPreKeyId
= rootNode
.get("pniNextSignedPreKeyId").asInt(1);
616 pniNextSignedPreKeyId
= 1;
618 if (rootNode
.hasNonNull("profileKey")) {
620 profileKey
= new ProfileKey(Base64
.getDecoder().decode(rootNode
.get("profileKey").asText()));
621 } catch (InvalidInputException e
) {
622 throw new IOException(
623 "Config file contains an invalid profileKey, needs to be base64 encoded array of 32 bytes",
628 if (previousStorageVersion
< 5) {
629 final var legacyRecipientsStoreFile
= getRecipientsStoreFile(dataPath
, accountPath
);
630 if (legacyRecipientsStoreFile
.exists()) {
631 LegacyRecipientStore2
.migrate(legacyRecipientsStoreFile
, getRecipientStore());
632 // Ensure our profile key is stored in profile store
633 getProfileStore().storeSelfProfileKey(getSelfRecipientId(), getProfileKey());
634 migratedLegacyConfig
= true;
637 if (previousStorageVersion
< 6) {
638 getRecipientTrustedResolver().resolveSelfRecipientTrusted(getSelfRecipientAddress());
640 final var legacyAciPreKeysPath
= getAciPreKeysPath(dataPath
, accountPath
);
641 if (legacyAciPreKeysPath
.exists()) {
642 LegacyPreKeyStore
.migrate(legacyAciPreKeysPath
, getAciPreKeyStore());
643 migratedLegacyConfig
= true;
645 final var legacyPniPreKeysPath
= getPniPreKeysPath(dataPath
, accountPath
);
646 if (legacyPniPreKeysPath
.exists()) {
647 LegacyPreKeyStore
.migrate(legacyPniPreKeysPath
, getPniPreKeyStore());
648 migratedLegacyConfig
= true;
650 final var legacyAciSignedPreKeysPath
= getAciSignedPreKeysPath(dataPath
, accountPath
);
651 if (legacyAciSignedPreKeysPath
.exists()) {
652 LegacySignedPreKeyStore
.migrate(legacyAciSignedPreKeysPath
, getAciSignedPreKeyStore());
653 migratedLegacyConfig
= true;
655 final var legacyPniSignedPreKeysPath
= getPniSignedPreKeysPath(dataPath
, accountPath
);
656 if (legacyPniSignedPreKeysPath
.exists()) {
657 LegacySignedPreKeyStore
.migrate(legacyPniSignedPreKeysPath
, getPniSignedPreKeyStore());
658 migratedLegacyConfig
= true;
660 final var legacySessionsPath
= getSessionsPath(dataPath
, accountPath
);
661 if (legacySessionsPath
.exists()) {
662 LegacySessionStore
.migrate(legacySessionsPath
,
663 getRecipientResolver(),
664 getRecipientAddressResolver(),
665 getAciSessionStore());
666 migratedLegacyConfig
= true;
668 final var legacyIdentitiesPath
= getIdentitiesPath(dataPath
, accountPath
);
669 if (legacyIdentitiesPath
.exists()) {
670 LegacyIdentityKeyStore
.migrate(legacyIdentitiesPath
,
671 getRecipientResolver(),
672 getRecipientAddressResolver(),
673 getIdentityKeyStore());
674 migratedLegacyConfig
= true;
676 final var legacySignalProtocolStore
= rootNode
.hasNonNull("axolotlStore")
677 ? jsonProcessor
.convertValue(Utils
.getNotNullNode(rootNode
, "axolotlStore"),
678 LegacyJsonSignalProtocolStore
.class)
680 if (legacySignalProtocolStore
!= null && legacySignalProtocolStore
.getLegacyIdentityKeyStore() != null) {
681 aciIdentityKeyPair
= legacySignalProtocolStore
.getLegacyIdentityKeyStore().getIdentityKeyPair();
682 registrationId
= legacySignalProtocolStore
.getLegacyIdentityKeyStore().getLocalRegistrationId();
683 migratedLegacyConfig
= true;
686 this.aciIdentityKeyPair
= aciIdentityKeyPair
;
687 this.localRegistrationId
= registrationId
;
688 this.settings
= settings
;
690 migratedLegacyConfig
= loadLegacyStores(rootNode
, legacySignalProtocolStore
) || migratedLegacyConfig
;
692 final var legacySenderKeysPath
= getSenderKeysPath(dataPath
, accountPath
);
693 if (legacySenderKeysPath
.exists()) {
694 LegacySenderKeyRecordStore
.migrate(legacySenderKeysPath
,
695 getRecipientResolver(),
696 getRecipientAddressResolver(),
697 getSenderKeyStore());
698 migratedLegacyConfig
= true;
700 final var legacySenderKeysSharedPath
= getSharedSenderKeysFile(dataPath
, accountPath
);
701 if (legacySenderKeysSharedPath
.exists()) {
702 LegacySenderKeySharedStore
.migrate(legacySenderKeysSharedPath
,
703 getRecipientResolver(),
704 getRecipientAddressResolver(),
705 getSenderKeyStore());
706 migratedLegacyConfig
= true;
708 if (rootNode
.hasNonNull("groupStore")) {
709 final var groupStoreStorage
= jsonProcessor
.convertValue(rootNode
.get("groupStore"),
710 LegacyGroupStore
.Storage
.class);
711 LegacyGroupStore
.migrate(groupStoreStorage
,
712 getGroupCachePath(dataPath
, accountPath
),
713 getRecipientResolver(),
715 migratedLegacyConfig
= true;
718 if (rootNode
.hasNonNull("stickerStore")) {
719 final var storage
= jsonProcessor
.convertValue(rootNode
.get("stickerStore"),
720 LegacyStickerStore
.Storage
.class);
721 LegacyStickerStore
.migrate(storage
, getStickerStore());
722 migratedLegacyConfig
= true;
725 if (rootNode
.hasNonNull("configurationStore")) {
726 configurationStoreStorage
= jsonProcessor
.convertValue(rootNode
.get("configurationStore"),
727 ConfigurationStore
.Storage
.class);
728 configurationStore
= ConfigurationStore
.fromStorage(configurationStoreStorage
,
729 this::saveConfigurationStore
);
731 configurationStore
= new ConfigurationStore(this::saveConfigurationStore
);
734 migratedLegacyConfig
= loadLegacyThreadStore(rootNode
) || migratedLegacyConfig
;
736 if (migratedLegacyConfig
) {
741 private boolean loadLegacyStores(
742 final JsonNode rootNode
, final LegacyJsonSignalProtocolStore legacySignalProtocolStore
744 var migrated
= false;
745 var legacyRecipientStoreNode
= rootNode
.get("recipientStore");
746 if (legacyRecipientStoreNode
!= null) {
747 logger
.debug("Migrating legacy recipient store.");
748 var legacyRecipientStore
= jsonProcessor
.convertValue(legacyRecipientStoreNode
, LegacyRecipientStore
.class);
749 if (legacyRecipientStore
!= null) {
750 legacyRecipientStore
.getAddresses()
751 .forEach(recipient
-> getRecipientStore().resolveRecipientTrusted(recipient
));
753 getRecipientTrustedResolver().resolveSelfRecipientTrusted(getSelfRecipientAddress());
757 if (legacySignalProtocolStore
!= null && legacySignalProtocolStore
.getLegacyPreKeyStore() != null) {
758 logger
.debug("Migrating legacy pre key store.");
759 for (var entry
: legacySignalProtocolStore
.getLegacyPreKeyStore().getPreKeys().entrySet()) {
761 getAciPreKeyStore().storePreKey(entry
.getKey(), new PreKeyRecord(entry
.getValue()));
762 } catch (InvalidMessageException e
) {
763 logger
.warn("Failed to migrate pre key, ignoring", e
);
769 if (legacySignalProtocolStore
!= null && legacySignalProtocolStore
.getLegacySignedPreKeyStore() != null) {
770 logger
.debug("Migrating legacy signed pre key store.");
771 for (var entry
: legacySignalProtocolStore
.getLegacySignedPreKeyStore().getSignedPreKeys().entrySet()) {
773 getAciSignedPreKeyStore().storeSignedPreKey(entry
.getKey(),
774 new SignedPreKeyRecord(entry
.getValue()));
775 } catch (InvalidMessageException e
) {
776 logger
.warn("Failed to migrate signed pre key, ignoring", e
);
782 if (legacySignalProtocolStore
!= null && legacySignalProtocolStore
.getLegacySessionStore() != null) {
783 logger
.debug("Migrating legacy session store.");
784 for (var session
: legacySignalProtocolStore
.getLegacySessionStore().getSessions()) {
786 getAciSessionStore().storeSession(new SignalProtocolAddress(session
.address
.getIdentifier(),
787 session
.deviceId
), new SessionRecord(session
.sessionRecord
));
788 } catch (Exception e
) {
789 logger
.warn("Failed to migrate session, ignoring", e
);
795 if (legacySignalProtocolStore
!= null && legacySignalProtocolStore
.getLegacyIdentityKeyStore() != null) {
796 logger
.debug("Migrating legacy identity session store.");
797 for (var identity
: legacySignalProtocolStore
.getLegacyIdentityKeyStore().getIdentities()) {
798 if (identity
.getAddress().serviceId().isEmpty()) {
801 final var serviceId
= identity
.getAddress().getServiceId();
802 getIdentityKeyStore().saveIdentity(serviceId
, identity
.getIdentityKey());
803 getIdentityKeyStore().setIdentityTrustLevel(serviceId
,
804 identity
.getIdentityKey(),
805 identity
.getTrustLevel());
810 if (rootNode
.hasNonNull("contactStore")) {
811 logger
.debug("Migrating legacy contact store.");
812 final var contactStoreNode
= rootNode
.get("contactStore");
813 final var contactStore
= jsonProcessor
.convertValue(contactStoreNode
, LegacyJsonContactsStore
.class);
814 for (var contact
: contactStore
.getContacts()) {
815 final var recipientId
= getRecipientStore().resolveRecipientTrusted(contact
.getAddress());
816 getContactStore().storeContact(recipientId
,
817 new Contact(contact
.name
,
820 contact
.messageExpirationTime
,
825 // Store profile keys only in profile store
826 var profileKeyString
= contact
.profileKey
;
827 if (profileKeyString
!= null) {
828 final ProfileKey profileKey
;
830 profileKey
= new ProfileKey(Base64
.getDecoder().decode(profileKeyString
));
831 getProfileStore().storeProfileKey(recipientId
, profileKey
);
832 } catch (InvalidInputException e
) {
833 logger
.warn("Failed to parse legacy contact profile key: {}", e
.getMessage());
840 if (rootNode
.hasNonNull("profileStore")) {
841 logger
.debug("Migrating legacy profile store.");
842 var profileStoreNode
= rootNode
.get("profileStore");
843 final var legacyProfileStore
= jsonProcessor
.convertValue(profileStoreNode
, LegacyProfileStore
.class);
844 for (var profileEntry
: legacyProfileStore
.getProfileEntries()) {
845 var recipientId
= getRecipientResolver().resolveRecipient(profileEntry
.getAddress());
846 // Not migrating profile key credential here, it was changed to expiring profile key credentials
847 getProfileStore().storeProfileKey(recipientId
, profileEntry
.getProfileKey());
848 final var profile
= profileEntry
.getProfile();
849 if (profile
!= null) {
850 final var capabilities
= new HashSet
<Profile
.Capability
>();
851 if (profile
.getCapabilities() != null) {
852 if (profile
.getCapabilities().gv1Migration
) {
853 capabilities
.add(Profile
.Capability
.gv1Migration
);
855 if (profile
.getCapabilities().storage
) {
856 capabilities
.add(Profile
.Capability
.storage
);
859 final var newProfile
= new Profile(profileEntry
.getLastUpdateTimestamp(),
860 profile
.getGivenName(),
861 profile
.getFamilyName(),
863 profile
.getAboutEmoji(),
866 profile
.isUnrestrictedUnidentifiedAccess()
867 ? Profile
.UnidentifiedAccessMode
.UNRESTRICTED
868 : profile
.getUnidentifiedAccess() != null
869 ? Profile
.UnidentifiedAccessMode
.ENABLED
870 : Profile
.UnidentifiedAccessMode
.DISABLED
,
872 getProfileStore().storeProfile(recipientId
, newProfile
);
880 private boolean loadLegacyThreadStore(final JsonNode rootNode
) {
881 var threadStoreNode
= rootNode
.get("threadStore");
882 if (threadStoreNode
!= null && !threadStoreNode
.isNull()) {
883 var threadStore
= jsonProcessor
.convertValue(threadStoreNode
, LegacyJsonThreadStore
.class);
884 // Migrate thread info to group and contact store
885 for (var thread
: threadStore
.getThreads()) {
886 if (thread
.id
== null || thread
.id
.isEmpty()) {
890 if (UuidUtil
.isUuid(thread
.id
) || thread
.id
.startsWith("+")) {
891 final var recipientId
= getRecipientResolver().resolveRecipient(thread
.id
);
892 var contact
= getContactStore().getContact(recipientId
);
893 if (contact
!= null) {
894 getContactStore().storeContact(recipientId
,
895 Contact
.newBuilder(contact
)
896 .withMessageExpirationTime(thread
.messageExpirationTime
)
900 var groupInfo
= getGroupStore().getGroup(GroupId
.fromBase64(thread
.id
));
901 if (groupInfo
instanceof GroupInfoV1
) {
902 ((GroupInfoV1
) groupInfo
).messageExpirationTime
= thread
.messageExpirationTime
;
903 getGroupStore().updateGroup(groupInfo
);
906 } catch (Exception e
) {
907 logger
.warn("Failed to read legacy thread info: {}", e
.getMessage());
916 private void saveConfigurationStore(ConfigurationStore
.Storage storage
) {
917 this.configurationStoreStorage
= storage
;
921 private void save() {
922 synchronized (fileChannel
) {
923 var rootNode
= jsonProcessor
.createObjectNode();
924 rootNode
.put("version", CURRENT_STORAGE_VERSION
)
925 .put("username", number
)
926 .put("serviceEnvironment", serviceEnvironment
== null ?
null : serviceEnvironment
.name())
927 .put("uuid", aci
== null ?
null : aci
.toString())
928 .put("pni", pni
== null ?
null : pni
.toString())
929 .put("deviceName", encryptedDeviceName
)
930 .put("deviceId", deviceId
)
931 .put("isMultiDevice", isMultiDevice
)
932 .put("lastReceiveTimestamp", lastReceiveTimestamp
)
933 .put("password", password
)
934 .put("registrationId", localRegistrationId
)
935 .put("pniRegistrationId", localPniRegistrationId
)
936 .put("identityPrivateKey",
937 Base64
.getEncoder().encodeToString(aciIdentityKeyPair
.getPrivateKey().serialize()))
939 Base64
.getEncoder().encodeToString(aciIdentityKeyPair
.getPublicKey().serialize()))
940 .put("pniIdentityPrivateKey",
941 pniIdentityKeyPair
== null
943 : Base64
.getEncoder()
944 .encodeToString(pniIdentityKeyPair
.getPrivateKey().serialize()))
945 .put("pniIdentityKey",
946 pniIdentityKeyPair
== null
948 : Base64
.getEncoder().encodeToString(pniIdentityKeyPair
.getPublicKey().serialize()))
949 .put("registrationLockPin", registrationLockPin
)
951 pinMasterKey
== null ?
null : Base64
.getEncoder().encodeToString(pinMasterKey
.serialize()))
953 storageKey
== null ?
null : Base64
.getEncoder().encodeToString(storageKey
.serialize()))
954 .put("storageManifestVersion", storageManifestVersion
== -1 ?
null : storageManifestVersion
)
955 .put("preKeyIdOffset", aciPreKeyIdOffset
)
956 .put("nextSignedPreKeyId", aciNextSignedPreKeyId
)
957 .put("pniPreKeyIdOffset", pniPreKeyIdOffset
)
958 .put("pniNextSignedPreKeyId", pniNextSignedPreKeyId
)
960 profileKey
== null ?
null : Base64
.getEncoder().encodeToString(profileKey
.serialize()))
961 .put("registered", registered
)
962 .putPOJO("configurationStore", configurationStoreStorage
);
964 try (var output
= new ByteArrayOutputStream()) {
965 // Write to memory first to prevent corrupting the file in case of serialization errors
966 jsonProcessor
.writeValue(output
, rootNode
);
967 var input
= new ByteArrayInputStream(output
.toByteArray());
968 fileChannel
.position(0);
969 input
.transferTo(Channels
.newOutputStream(fileChannel
));
970 fileChannel
.truncate(fileChannel
.position());
971 fileChannel
.force(false);
973 } catch (Exception e
) {
974 logger
.error("Error saving file: {}", e
.getMessage(), e
);
979 private static Pair
<FileChannel
, FileLock
> openFileChannel(File fileName
, boolean waitForLock
) throws IOException
{
980 var fileChannel
= new RandomAccessFile(fileName
, "rw").getChannel();
982 var lock
= fileChannel
.tryLock();
985 logger
.debug("Config file is in use by another instance.");
986 throw new IOException("Config file is in use by another instance.");
988 logger
.info("Config file is in use by another instance, waiting…");
989 lock
= fileChannel
.lock();
990 logger
.info("Config file lock acquired.");
992 final var result
= new Pair
<>(fileChannel
, lock
);
996 if (fileChannel
!= null) {
1002 public void resetPreKeyOffsets(final ServiceIdType serviceIdType
) {
1003 if (serviceIdType
.equals(ServiceIdType
.ACI
)) {
1004 this.aciPreKeyIdOffset
= new SecureRandom().nextInt(Medium
.MAX_VALUE
);
1005 this.aciNextSignedPreKeyId
= new SecureRandom().nextInt(Medium
.MAX_VALUE
);
1007 this.pniPreKeyIdOffset
= new SecureRandom().nextInt(Medium
.MAX_VALUE
);
1008 this.pniNextSignedPreKeyId
= new SecureRandom().nextInt(Medium
.MAX_VALUE
);
1013 public void addPreKeys(ServiceIdType serviceIdType
, List
<PreKeyRecord
> records
) {
1014 if (serviceIdType
.equals(ServiceIdType
.ACI
)) {
1015 addAciPreKeys(records
);
1017 addPniPreKeys(records
);
1021 public void addAciPreKeys(List
<PreKeyRecord
> records
) {
1022 for (var record : records
) {
1023 if (aciPreKeyIdOffset
!= record.getId()) {
1024 logger
.error("Invalid pre key id {}, expected {}", record.getId(), aciPreKeyIdOffset
);
1025 throw new AssertionError("Invalid pre key id");
1027 getAciPreKeyStore().storePreKey(record.getId(), record);
1028 aciPreKeyIdOffset
= (aciPreKeyIdOffset
+ 1) % Medium
.MAX_VALUE
;
1033 public void addPniPreKeys(List
<PreKeyRecord
> records
) {
1034 for (var record : records
) {
1035 if (pniPreKeyIdOffset
!= record.getId()) {
1036 logger
.error("Invalid pre key id {}, expected {}", record.getId(), pniPreKeyIdOffset
);
1037 throw new AssertionError("Invalid pre key id");
1039 getPniPreKeyStore().storePreKey(record.getId(), record);
1040 pniPreKeyIdOffset
= (pniPreKeyIdOffset
+ 1) % Medium
.MAX_VALUE
;
1045 public void addSignedPreKey(ServiceIdType serviceIdType
, SignedPreKeyRecord
record) {
1046 if (serviceIdType
.equals(ServiceIdType
.ACI
)) {
1047 addAciSignedPreKey(record);
1049 addPniSignedPreKey(record);
1053 public void addAciSignedPreKey(SignedPreKeyRecord
record) {
1054 if (aciNextSignedPreKeyId
!= record.getId()) {
1055 logger
.error("Invalid signed pre key id {}, expected {}", record.getId(), aciNextSignedPreKeyId
);
1056 throw new AssertionError("Invalid signed pre key id");
1058 getAciSignedPreKeyStore().storeSignedPreKey(record.getId(), record);
1059 aciNextSignedPreKeyId
= (aciNextSignedPreKeyId
+ 1) % Medium
.MAX_VALUE
;
1063 public void addPniSignedPreKey(SignedPreKeyRecord
record) {
1064 if (pniNextSignedPreKeyId
!= record.getId()) {
1065 logger
.error("Invalid signed pre key id {}, expected {}", record.getId(), pniNextSignedPreKeyId
);
1066 throw new AssertionError("Invalid signed pre key id");
1068 getPniSignedPreKeyStore().storeSignedPreKey(record.getId(), record);
1069 pniNextSignedPreKeyId
= (pniNextSignedPreKeyId
+ 1) % Medium
.MAX_VALUE
;
1073 public int getPreviousStorageVersion() {
1074 return previousStorageVersion
;
1077 public SignalServiceDataStore
getSignalServiceDataStore() {
1078 return new SignalServiceDataStore() {
1080 public SignalServiceAccountDataStore
get(final ServiceId accountIdentifier
) {
1081 if (accountIdentifier
.equals(aci
)) {
1083 } else if (accountIdentifier
.equals(pni
)) {
1086 throw new IllegalArgumentException("No matching store found for " + accountIdentifier
);
1091 public SignalServiceAccountDataStore
aci() {
1092 return getAciSignalServiceAccountDataStore();
1096 public SignalServiceAccountDataStore
pni() {
1097 return getPniSignalServiceAccountDataStore();
1101 public boolean isMultiDevice() {
1102 return SignalAccount
.this.isMultiDevice();
1107 private SignalServiceAccountDataStore
getAciSignalServiceAccountDataStore() {
1108 return getOrCreate(() -> aciSignalProtocolStore
,
1109 () -> aciSignalProtocolStore
= new SignalProtocolStore(getAciPreKeyStore(),
1110 getAciSignedPreKeyStore(),
1111 getAciSessionStore(),
1112 getAciIdentityKeyStore(),
1113 getSenderKeyStore(),
1114 this::isMultiDevice
));
1117 private SignalServiceAccountDataStore
getPniSignalServiceAccountDataStore() {
1118 return getOrCreate(() -> pniSignalProtocolStore
,
1119 () -> pniSignalProtocolStore
= new SignalProtocolStore(getPniPreKeyStore(),
1120 getPniSignedPreKeyStore(),
1121 getPniSessionStore(),
1122 getPniIdentityKeyStore(),
1123 getSenderKeyStore(),
1124 this::isMultiDevice
));
1127 private PreKeyStore
getAciPreKeyStore() {
1128 return getOrCreate(() -> aciPreKeyStore
,
1129 () -> aciPreKeyStore
= new PreKeyStore(getAccountDatabase(), ServiceIdType
.ACI
));
1132 private SignedPreKeyStore
getAciSignedPreKeyStore() {
1133 return getOrCreate(() -> aciSignedPreKeyStore
,
1134 () -> aciSignedPreKeyStore
= new SignedPreKeyStore(getAccountDatabase(), ServiceIdType
.ACI
));
1137 private PreKeyStore
getPniPreKeyStore() {
1138 return getOrCreate(() -> pniPreKeyStore
,
1139 () -> pniPreKeyStore
= new PreKeyStore(getAccountDatabase(), ServiceIdType
.PNI
));
1142 private SignedPreKeyStore
getPniSignedPreKeyStore() {
1143 return getOrCreate(() -> pniSignedPreKeyStore
,
1144 () -> pniSignedPreKeyStore
= new SignedPreKeyStore(getAccountDatabase(), ServiceIdType
.PNI
));
1147 public SessionStore
getAciSessionStore() {
1148 return getOrCreate(() -> aciSessionStore
,
1149 () -> aciSessionStore
= new SessionStore(getAccountDatabase(), ServiceIdType
.ACI
));
1152 public SessionStore
getPniSessionStore() {
1153 return getOrCreate(() -> pniSessionStore
,
1154 () -> pniSessionStore
= new SessionStore(getAccountDatabase(), ServiceIdType
.PNI
));
1157 public IdentityKeyStore
getIdentityKeyStore() {
1158 return getOrCreate(() -> identityKeyStore
,
1159 () -> identityKeyStore
= new IdentityKeyStore(getAccountDatabase(), settings
.trustNewIdentity()));
1162 public SignalIdentityKeyStore
getAciIdentityKeyStore() {
1163 return getOrCreate(() -> aciIdentityKeyStore
,
1164 () -> aciIdentityKeyStore
= new SignalIdentityKeyStore(getRecipientResolver(),
1165 () -> aciIdentityKeyPair
,
1166 localRegistrationId
,
1167 getIdentityKeyStore()));
1170 public SignalIdentityKeyStore
getPniIdentityKeyStore() {
1171 return getOrCreate(() -> pniIdentityKeyStore
,
1172 () -> pniIdentityKeyStore
= new SignalIdentityKeyStore(getRecipientResolver(),
1173 () -> pniIdentityKeyPair
,
1174 localRegistrationId
,
1175 getIdentityKeyStore()));
1178 public GroupStore
getGroupStore() {
1179 return getOrCreate(() -> groupStore
,
1180 () -> groupStore
= new GroupStore(getAccountDatabase(),
1181 getRecipientResolver(),
1182 getRecipientIdCreator()));
1185 public ContactsStore
getContactStore() {
1186 return getRecipientStore();
1189 private RecipientIdCreator
getRecipientIdCreator() {
1190 return recipientId
-> getRecipientStore().create(recipientId
);
1193 public RecipientResolver
getRecipientResolver() {
1194 return new RecipientResolver
.RecipientResolverWrapper(this::getRecipientStore
);
1197 public RecipientTrustedResolver
getRecipientTrustedResolver() {
1198 return new RecipientTrustedResolver
.RecipientTrustedResolverWrapper(this::getRecipientStore
);
1201 public RecipientAddressResolver
getRecipientAddressResolver() {
1202 return recipientId
-> getRecipientStore().resolveRecipientAddress(recipientId
);
1205 public RecipientStore
getRecipientStore() {
1206 return getOrCreate(() -> recipientStore
,
1207 () -> recipientStore
= new RecipientStore(this::mergeRecipients
,
1208 this::getSelfRecipientAddress
,
1209 getAccountDatabase()));
1212 public ProfileStore
getProfileStore() {
1213 return getRecipientStore();
1216 public StickerStore
getStickerStore() {
1217 return getOrCreate(() -> stickerStore
, () -> stickerStore
= new StickerStore(getAccountDatabase()));
1220 public SenderKeyStore
getSenderKeyStore() {
1221 return getOrCreate(() -> senderKeyStore
, () -> senderKeyStore
= new SenderKeyStore(getAccountDatabase()));
1224 public ConfigurationStore
getConfigurationStore() {
1225 return configurationStore
;
1228 public MessageCache
getMessageCache() {
1229 return getOrCreate(() -> messageCache
,
1230 () -> messageCache
= new MessageCache(getMessageCachePath(dataPath
, accountPath
)));
1233 public AccountDatabase
getAccountDatabase() {
1234 return getOrCreate(() -> accountDatabase
, () -> {
1236 accountDatabase
= AccountDatabase
.init(getDatabaseFile(dataPath
, accountPath
));
1237 } catch (SQLException e
) {
1238 throw new RuntimeException(e
);
1243 public MessageSendLogStore
getMessageSendLogStore() {
1244 return getOrCreate(() -> messageSendLogStore
,
1245 () -> messageSendLogStore
= new MessageSendLogStore(getAccountDatabase(),
1246 settings
.disableMessageSendLog()));
1249 public CredentialsProvider
getCredentialsProvider() {
1250 return new CredentialsProvider() {
1252 public ACI
getAci() {
1257 public PNI
getPni() {
1262 public String
getE164() {
1267 public String
getPassword() {
1272 public int getDeviceId() {
1278 public String
getNumber() {
1282 public void setNumber(final String number
) {
1283 this.number
= number
;
1287 public ServiceEnvironment
getServiceEnvironment() {
1288 return serviceEnvironment
;
1291 public void setServiceEnvironment(final ServiceEnvironment serviceEnvironment
) {
1292 this.serviceEnvironment
= serviceEnvironment
;
1296 public ServiceId
getAccountId(ServiceIdType serviceIdType
) {
1297 return serviceIdType
.equals(ServiceIdType
.ACI
) ? aci
: pni
;
1300 public ACI
getAci() {
1304 public void setAci(final ACI aci
) {
1309 public PNI
getPni() {
1313 public void setPni(final PNI updatedPni
) {
1314 if (this.pni
!= null && !this.pni
.equals(updatedPni
)) {
1315 // Clear data for old PNI
1316 identityKeyStore
.deleteIdentity(this.pni
);
1317 getPniPreKeyStore().removeAllPreKeys();
1318 getPniSignedPreKeyStore().removeAllSignedPreKeys();
1321 this.pni
= updatedPni
;
1326 final PNI updatedPni
,
1327 final IdentityKeyPair pniIdentityKeyPair
,
1328 final SignedPreKeyRecord pniSignedPreKey
,
1329 final int localPniRegistrationId
1333 setPniIdentityKeyPair(pniIdentityKeyPair
);
1334 addPniSignedPreKey(pniSignedPreKey
);
1335 setLocalPniRegistrationId(localPniRegistrationId
);
1338 public SignalServiceAddress
getSelfAddress() {
1339 return new SignalServiceAddress(aci
, number
);
1342 public RecipientAddress
getSelfRecipientAddress() {
1343 return new RecipientAddress(aci
, pni
, number
);
1346 public RecipientId
getSelfRecipientId() {
1347 return getRecipientResolver().resolveRecipient(getSelfRecipientAddress());
1350 public byte[] getEncryptedDeviceName() {
1351 return encryptedDeviceName
== null ?
null : Base64
.getDecoder().decode(encryptedDeviceName
);
1354 public void setEncryptedDeviceName(final String encryptedDeviceName
) {
1355 this.encryptedDeviceName
= encryptedDeviceName
;
1359 public int getDeviceId() {
1363 public boolean isPrimaryDevice() {
1364 return deviceId
== SignalServiceAddress
.DEFAULT_DEVICE_ID
;
1367 public IdentityKeyPair
getIdentityKeyPair(ServiceIdType serviceIdType
) {
1368 return serviceIdType
.equals(ServiceIdType
.ACI
) ? aciIdentityKeyPair
: pniIdentityKeyPair
;
1371 public IdentityKeyPair
getAciIdentityKeyPair() {
1372 return aciIdentityKeyPair
;
1375 public IdentityKeyPair
getPniIdentityKeyPair() {
1376 return pniIdentityKeyPair
;
1379 public void setPniIdentityKeyPair(final IdentityKeyPair identityKeyPair
) {
1380 pniIdentityKeyPair
= identityKeyPair
;
1381 final var pniPublicKey
= getPniIdentityKeyPair().getPublicKey();
1382 getIdentityKeyStore().saveIdentity(getPni(), pniPublicKey
);
1383 getIdentityKeyStore().setIdentityTrustLevel(getPni(), pniPublicKey
, TrustLevel
.TRUSTED_VERIFIED
);
1387 public int getLocalRegistrationId() {
1388 return localRegistrationId
;
1391 public int getLocalPniRegistrationId() {
1392 return localPniRegistrationId
;
1395 public void setLocalPniRegistrationId(final int localPniRegistrationId
) {
1396 this.localPniRegistrationId
= localPniRegistrationId
;
1400 public String
getPassword() {
1404 private void setPassword(final String password
) {
1405 this.password
= password
;
1409 public void setRegistrationLockPin(final String registrationLockPin
) {
1410 this.registrationLockPin
= registrationLockPin
;
1414 public String
getRegistrationLockPin() {
1415 return registrationLockPin
;
1418 public String
getRegistrationLock() {
1419 final var masterKey
= getPinBackedMasterKey();
1420 if (masterKey
== null) {
1423 return masterKey
.deriveRegistrationLock();
1426 public MasterKey
getPinBackedMasterKey() {
1427 if (registrationLockPin
== null) {
1430 return pinMasterKey
;
1433 public MasterKey
getOrCreatePinMasterKey() {
1434 if (pinMasterKey
== null) {
1435 pinMasterKey
= KeyUtils
.createMasterKey();
1438 return pinMasterKey
;
1441 public StorageKey
getStorageKey() {
1442 if (pinMasterKey
!= null) {
1443 return pinMasterKey
.deriveStorageServiceKey();
1448 public StorageKey
getOrCreateStorageKey() {
1449 if (isPrimaryDevice()) {
1450 return getOrCreatePinMasterKey().deriveStorageServiceKey();
1455 public void setStorageKey(final StorageKey storageKey
) {
1456 if (storageKey
.equals(this.storageKey
)) {
1459 this.storageKey
= storageKey
;
1463 public long getStorageManifestVersion() {
1464 return this.storageManifestVersion
;
1467 public void setStorageManifestVersion(final long storageManifestVersion
) {
1468 if (storageManifestVersion
== this.storageManifestVersion
) {
1471 this.storageManifestVersion
= storageManifestVersion
;
1475 public Optional
<SignalStorageManifest
> getStorageManifest() {
1476 final var storageManifestFile
= getStorageManifestFile(dataPath
, accountPath
);
1477 if (!storageManifestFile
.exists()) {
1478 return Optional
.empty();
1480 try (var inputStream
= new FileInputStream(storageManifestFile
)) {
1481 return Optional
.of(SignalStorageManifest
.deserialize(inputStream
.readAllBytes()));
1482 } catch (IOException e
) {
1483 logger
.warn("Failed to read local storage manifest.", e
);
1484 return Optional
.empty();
1488 public void setStorageManifest(SignalStorageManifest manifest
) {
1489 final var storageManifestFile
= getStorageManifestFile(dataPath
, accountPath
);
1490 if (manifest
== null) {
1491 if (storageManifestFile
.exists()) {
1493 Files
.delete(storageManifestFile
.toPath());
1494 } catch (IOException e
) {
1495 logger
.error("Failed to delete local storage manifest.", e
);
1501 final var manifestBytes
= manifest
.serialize();
1502 try (var outputStream
= new FileOutputStream(storageManifestFile
)) {
1503 outputStream
.write(manifestBytes
);
1504 } catch (IOException e
) {
1505 logger
.error("Failed to store local storage manifest.", e
);
1509 public ProfileKey
getProfileKey() {
1513 public void setProfileKey(final ProfileKey profileKey
) {
1514 if (profileKey
.equals(this.profileKey
)) {
1517 this.profileKey
= profileKey
;
1519 getProfileStore().storeSelfProfileKey(getSelfRecipientId(), getProfileKey());
1522 public byte[] getSelfUnidentifiedAccessKey() {
1523 return UnidentifiedAccess
.deriveAccessKeyFrom(getProfileKey());
1526 public int getPreKeyIdOffset(ServiceIdType serviceIdType
) {
1527 return serviceIdType
.equals(ServiceIdType
.ACI
) ? aciPreKeyIdOffset
: pniPreKeyIdOffset
;
1530 public int getNextSignedPreKeyId(ServiceIdType serviceIdType
) {
1531 return serviceIdType
.equals(ServiceIdType
.ACI
) ? aciNextSignedPreKeyId
: pniNextSignedPreKeyId
;
1534 public boolean isRegistered() {
1538 public void setRegistered(final boolean registered
) {
1539 this.registered
= registered
;
1543 public boolean isMultiDevice() {
1544 return isMultiDevice
;
1547 public void setMultiDevice(final boolean multiDevice
) {
1548 if (isMultiDevice
== multiDevice
) {
1551 isMultiDevice
= multiDevice
;
1555 public long getLastReceiveTimestamp() {
1556 return lastReceiveTimestamp
;
1559 public void setLastReceiveTimestamp(final long lastReceiveTimestamp
) {
1560 this.lastReceiveTimestamp
= lastReceiveTimestamp
;
1564 public boolean isUnrestrictedUnidentifiedAccess() {
1565 final var profile
= getProfileStore().getProfile(getSelfRecipientId());
1566 return profile
!= null && profile
.getUnidentifiedAccessMode() == Profile
.UnidentifiedAccessMode
.UNRESTRICTED
;
1569 public boolean isDiscoverableByPhoneNumber() {
1570 final var phoneNumberUnlisted
= configurationStore
.getPhoneNumberUnlisted();
1571 return phoneNumberUnlisted
== null || !phoneNumberUnlisted
;
1574 public void finishRegistration(final ACI aci
, final PNI pni
, final MasterKey masterKey
, final String pin
) {
1575 this.pinMasterKey
= masterKey
;
1576 this.storageManifestVersion
= -1;
1577 this.setStorageManifest(null);
1578 this.storageKey
= null;
1579 this.encryptedDeviceName
= null;
1580 this.deviceId
= SignalServiceAddress
.DEFAULT_DEVICE_ID
;
1581 this.isMultiDevice
= false;
1582 this.registered
= true;
1585 this.registrationLockPin
= pin
;
1586 this.lastReceiveTimestamp
= 0;
1590 getAciSessionStore().archiveAllSessions();
1591 getPniSessionStore().archiveAllSessions();
1592 getSenderKeyStore().deleteAll();
1593 getRecipientTrustedResolver().resolveSelfRecipientTrusted(getSelfRecipientAddress());
1594 final var aciPublicKey
= getAciIdentityKeyPair().getPublicKey();
1595 getIdentityKeyStore().saveIdentity(getAci(), aciPublicKey
);
1596 getIdentityKeyStore().setIdentityTrustLevel(getAci(), aciPublicKey
, TrustLevel
.TRUSTED_VERIFIED
);
1597 final var pniPublicKey
= getPniIdentityKeyPair().getPublicKey();
1598 getIdentityKeyStore().saveIdentity(getPni(), pniPublicKey
);
1599 getIdentityKeyStore().setIdentityTrustLevel(getPni(), pniPublicKey
, TrustLevel
.TRUSTED_VERIFIED
);
1602 public void deleteAccountData() throws IOException
{
1604 try (final var files
= Files
.walk(getUserPath(dataPath
, accountPath
).toPath())
1605 .sorted(Comparator
.reverseOrder())) {
1606 for (final var file
= files
.iterator(); file
.hasNext(); ) {
1607 Files
.delete(file
.next());
1610 Files
.delete(getFileName(dataPath
, accountPath
).toPath());
1614 public void close() {
1615 synchronized (fileChannel
) {
1616 if (accountDatabase
!= null) {
1618 accountDatabase
.close();
1619 } catch (SQLException e
) {
1620 logger
.warn("Failed to close account database: {}", e
.getMessage(), e
);
1623 if (messageSendLogStore
!= null) {
1624 messageSendLogStore
.close();
1629 } catch (ClosedChannelException ignored
) {
1631 fileChannel
.close();
1632 } catch (IOException e
) {
1633 logger
.warn("Failed to close account: {}", e
.getMessage(), e
);
1638 private <T
> T
getOrCreate(Supplier
<T
> supplier
, Callable creator
) {
1639 var value
= supplier
.get();
1640 if (value
!= null) {
1644 synchronized (LOCK
) {
1645 value
= supplier
.get();
1646 if (value
!= null) {
1650 return supplier
.get();
1654 private interface Callable
{