webmaster/signal-cli
1
0
Fork 0
mirror of https://github.com/AsamK/signal-cli synced 2025-08-29 18:40:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix issue with prekey update

Browse source
This commit is contained in:
AsamK 2024-02-18 19:26:21 +01:00
parent 7206b4da25
commit 2c0ad7feb7
3 changed files with 47 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

4
lib/src/main/java/org/asamk/signal/manager/helper/AccountHelper.java
View file

@ -90,7 +90,11 @@ public class AccountHelper {
}
try {
updateAccountAttributes();
if (account.getPreviousStorageVersion() < 9) {
context.getPreKeyHelper().forceRefreshPreKeys();
} else {
context.getPreKeyHelper().refreshPreKeysIfNecessary();
}
if (account.getAci() == null || account.getPni() == null) {
checkWhoAmiI();
}

50
lib/src/main/java/org/asamk/signal/manager/helper/PreKeyHelper.java
View file

@ -41,6 +41,11 @@ public class PreKeyHelper {
refreshPreKeysIfNecessary(ServiceIdType.PNI);
}
public void forceRefreshPreKeys() throws IOException {
forceRefreshPreKeys(ServiceIdType.ACI);
forceRefreshPreKeys(ServiceIdType.PNI);
}
public void refreshPreKeysIfNecessary(ServiceIdType serviceIdType) throws IOException {
final var identityKeyPair = account.getIdentityKeyPair(serviceIdType);
if (identityKeyPair == null) {
@ -56,6 +61,22 @@ public class PreKeyHelper {
}
}
public void forceRefreshPreKeys(ServiceIdType serviceIdType) throws IOException {
final var identityKeyPair = account.getIdentityKeyPair(serviceIdType);
if (identityKeyPair == null) {
return;
}
final var accountId = account.getAccountId(serviceIdType);
if (accountId == null) {
return;
}
final var counts = new OneTimePreKeyCounts(0, 0);
if (refreshPreKeysIfNecessary(serviceIdType, identityKeyPair, counts, true)) {
refreshPreKeysIfNecessary(serviceIdType, identityKeyPair, counts, true);
}
}
private boolean refreshPreKeysIfNecessary(
final ServiceIdType serviceIdType, final IdentityKeyPair identityKeyPair
) throws IOException {
@ -67,8 +88,17 @@ public class PreKeyHelper {
preKeyCounts = new OneTimePreKeyCounts(0, 0);
}
return refreshPreKeysIfNecessary(serviceIdType, identityKeyPair, preKeyCounts, false);
}
private boolean refreshPreKeysIfNecessary(
final ServiceIdType serviceIdType,
final IdentityKeyPair identityKeyPair,
final OneTimePreKeyCounts preKeyCounts,
final boolean force
) throws IOException {
List<PreKeyRecord> preKeyRecords = null;
if (preKeyCounts.getEcCount() < ServiceConfig.PREKEY_MINIMUM_COUNT) {
if (force || preKeyCounts.getEcCount() < ServiceConfig.PREKEY_MINIMUM_COUNT) {
logger.debug("Refreshing {} ec pre keys, because only {} of min {} pre keys remain",
serviceIdType,
preKeyCounts.getEcCount(),
@ -77,13 +107,13 @@ public class PreKeyHelper {
}
SignedPreKeyRecord signedPreKeyRecord = null;
if (signedPreKeyNeedsRefresh(serviceIdType)) {
if (force || signedPreKeyNeedsRefresh(serviceIdType)) {
logger.debug("Refreshing {} signed pre key.", serviceIdType);
signedPreKeyRecord = generateSignedPreKey(serviceIdType, identityKeyPair);
}
List<KyberPreKeyRecord> kyberPreKeyRecords = null;
if (preKeyCounts.getKyberCount() < ServiceConfig.PREKEY_MINIMUM_COUNT) {
if (force || preKeyCounts.getKyberCount() < ServiceConfig.PREKEY_MINIMUM_COUNT) {
logger.debug("Refreshing {} kyber pre keys, because only {} of min {} pre keys remain",
serviceIdType,
preKeyCounts.getKyberCount(),
@ -92,9 +122,11 @@ public class PreKeyHelper {
}
KyberPreKeyRecord lastResortKyberPreKeyRecord = null;
if (lastResortKyberPreKeyNeedsRefresh(serviceIdType)) {
if (force || lastResortKyberPreKeyNeedsRefresh(serviceIdType)) {
logger.debug("Refreshing {} last resort kyber pre key.", serviceIdType);
lastResortKyberPreKeyRecord = generateLastResortKyberPreKey(serviceIdType, identityKeyPair);
lastResortKyberPreKeyRecord = generateLastResortKyberPreKey(serviceIdType,
identityKeyPair,
kyberPreKeyRecords == null ? 0 : kyberPreKeyRecords.size());
}
if (signedPreKeyRecord == null
@ -157,9 +189,7 @@ public class PreKeyHelper {
final var accountData = account.getAccountData(serviceIdType);
final var offset = accountData.getPreKeyMetadata().getNextPreKeyId();
var records = KeyUtils.generatePreKeyRecords(offset);
return records;
return KeyUtils.generatePreKeyRecords(offset);
}
private boolean signedPreKeyNeedsRefresh(ServiceIdType serviceIdType) {
@ -210,10 +240,10 @@ public class PreKeyHelper {
}
private KyberPreKeyRecord generateLastResortKyberPreKey(
ServiceIdType serviceIdType, IdentityKeyPair identityKeyPair
ServiceIdType serviceIdType, IdentityKeyPair identityKeyPair, final int offset
) {
final var accountData = account.getAccountData(serviceIdType);
final var signedPreKeyId = accountData.getPreKeyMetadata().getNextKyberPreKeyId();
final var signedPreKeyId = accountData.getPreKeyMetadata().getNextKyberPreKeyId() + offset;
return KeyUtils.generateKyberPreKeyRecord(signedPreKeyId, identityKeyPair.getPrivateKey());
}

4
lib/src/main/java/org/asamk/signal/manager/storage/SignalAccount.java
View file

@ -114,7 +114,7 @@ public class SignalAccount implements Closeable {
private static final Logger logger = LoggerFactory.getLogger(SignalAccount.class);
private static final int MINIMUM_STORAGE_VERSION = 1;
private static final int CURRENT_STORAGE_VERSION = 8;
private static final int CURRENT_STORAGE_VERSION = 9;
private final Object LOCK = new Object();
@ -1111,7 +1111,7 @@ public class SignalAccount implements Closeable {
serviceIdType,
preKeyMetadata.nextKyberPreKeyId);
accountData.getSignalServiceAccountDataStore()
.markAllOneTimeEcPreKeysStaleIfNecessary(System.currentTimeMillis());
.markAllOneTimeKyberPreKeysStaleIfNecessary(System.currentTimeMillis());
for (var record : records) {
if (preKeyMetadata.nextKyberPreKeyId != record.getId()) {
logger.error("Invalid kyber pre key id {}, expected {}",