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

Implement simple handling of incoming decryption error message

Browse source
This commit is contained in:
AsamK 2021-12-20 14:01:24 +01:00
parent 1f48ce1f39
commit baed2b7ead
4 changed files with 29 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

10
lib/src/main/java/org/asamk/signal/manager/helper/IncomingMessageHandler.java
View file

@ -233,6 +233,16 @@ public final class IncomingMessageHandler {
if (content.getDecryptionErrorMessage().isPresent()) {
var message = content.getDecryptionErrorMessage().get();
logger.debug("Received a decryption error message (resend request for {})", message.getTimestamp());
if (message.getRatchetKey().isPresent()) {
if (message.getDeviceId() == account.getDeviceId() && account.getSessionStore()
.isCurrentRatchetKey(sender, senderDeviceId, message.getRatchetKey().get())) {
logger.debug("Renewing the session with sender");
actions.add(new RenewSessionAction(sender));
}
} else {
logger.debug("Reset shared sender keys with this recipient");
account.getSenderKeyStore().deleteSharedWith(sender);
}
}
if (content.getDataMessage().isPresent()) {

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

@ -75,7 +75,9 @@ public class SenderKeyRecordStore implements org.whispersystems.libsignal.groups
cachedSenderKeys.clear();
final var keys = getKeysLocked(recipientId);
for (var key : keys) {
if (key.distributionId.equals(distributionId)) deleteSenderKeyLocked(key);
if (key.distributionId.equals(distributionId)) {
deleteSenderKeyLocked(key);
}
}
}
}

5
lib/src/main/java/org/asamk/signal/manager/storage/senderKeys/SenderKeySharedStore.java
View file

@ -163,8 +163,9 @@ public class SenderKeySharedStore {
public void deleteAllFor(final DistributionId distributionId) {
synchronized (sharedSenderKeys) {
sharedSenderKeys.remove(distributionId.asUuid());
saveLocked();
if (sharedSenderKeys.remove(distributionId.asUuid()) != null) {
saveLocked();
}
}
}

13
lib/src/main/java/org/asamk/signal/manager/storage/sessions/SessionStore.java
View file

@ -7,6 +7,7 @@ import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.whispersystems.libsignal.NoSessionException;
import org.whispersystems.libsignal.SignalProtocolAddress;
import org.whispersystems.libsignal.ecc.ECPublicKey;
import org.whispersystems.libsignal.protocol.CiphertextMessage;
import org.whispersystems.libsignal.state.SessionRecord;
import org.whispersystems.signalservice.api.SignalServiceSessionStore;
@ -91,6 +92,18 @@ public class SessionStore implements SignalServiceSessionStore {
}
}
public boolean isCurrentRatchetKey(RecipientId recipientId, int deviceId, ECPublicKey ratchetKey) {
final var key = new Key(recipientId, deviceId);
synchronized (cachedSessions) {
final var session = loadSessionLocked(key);
if (session == null) {
return false;
}
return session.currentRatchetKeyMatches(ratchetKey);
}
}
@Override
public void storeSession(SignalProtocolAddress address, SessionRecord session) {
final var key = getKey(address);