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

Fix potential dead lock in recipient store

Browse source
This commit is contained in:
AsamK 2024-02-09 17:51:15 +01:00
parent 4e61f2b2e5
commit be699cbd85
1 changed files with 67 additions and 84 deletions
Download patch file Download diff file Expand all files Collapse all files

23
lib/src/main/java/org/asamk/signal/manager/storage/recipients/RecipientStore.java
View file

@ -47,7 +47,6 @@ public class RecipientStore implements RecipientIdCreator, RecipientResolver, Re
private final SelfProfileKeyProvider selfProfileKeyProvider;
private final Database database;
private final Object recipientsLock = new Object();
private final Map<Long, Long> recipientsMerged = new HashMap<>();
private final Map<ServiceId, RecipientWithAddress> recipientAddressCache = new HashMap<>();
@ -164,7 +163,6 @@ public class RecipientStore implements RecipientIdCreator, RecipientResolver, Re
}
private RecipientId resolveRecipientByNumber(final String number) {
synchronized (recipientsLock) {
final RecipientId recipientId;
try (final var connection = database.getConnection()) {
connection.setAutoCommit(false);
@ -175,17 +173,15 @@ public class RecipientStore implements RecipientIdCreator, RecipientResolver, Re
}
return recipientId;
}
}
@Override
public RecipientId resolveRecipient(final ServiceId serviceId) {
synchronized (recipientsLock) {
try (final var connection = database.getConnection()) {
connection.setAutoCommit(false);
final var recipientWithAddress = recipientAddressCache.get(serviceId);
if (recipientWithAddress != null) {
return recipientWithAddress.id();
}
try (final var connection = database.getConnection()) {
connection.setAutoCommit(false);
final var recipientId = resolveRecipientLocked(connection, serviceId);
connection.commit();
return recipientId;
@ -193,7 +189,6 @@ public class RecipientStore implements RecipientIdCreator, RecipientResolver, Re
throw new RuntimeException("Failed read recipient store", e);
}
}
}
/**
* Should only be used for recipientIds from the database.
@ -258,7 +253,6 @@ public class RecipientStore implements RecipientIdCreator, RecipientResolver, Re
}
public RecipientId resolveRecipient(RecipientAddress address) {
synchronized (recipientsLock) {
final RecipientId recipientId;
try (final var connection = database.getConnection()) {
connection.setAutoCommit(false);
@ -269,7 +263,6 @@ public class RecipientStore implements RecipientIdCreator, RecipientResolver, Re
}
return recipientId;
}
}
public RecipientId resolveRecipient(Connection connection, RecipientAddress address) throws SQLException {
return resolveRecipientLocked(connection, address);
@ -550,10 +543,9 @@ public class RecipientStore implements RecipientIdCreator, RecipientResolver, Re
public void deleteRecipientData(RecipientId recipientId) {
logger.debug("Deleting recipient data for {}", recipientId);
synchronized (recipientsLock) {
recipientAddressCache.entrySet().removeIf(e -> e.getValue().id().equals(recipientId));
try (final var connection = database.getConnection()) {
connection.setAutoCommit(false);
recipientAddressCache.entrySet().removeIf(e -> e.getValue().id().equals(recipientId));
storeContact(connection, recipientId, null);
storeProfile(connection, recipientId, null);
storeProfileKey(connection, recipientId, null, false);
@ -564,7 +556,6 @@ public class RecipientStore implements RecipientIdCreator, RecipientResolver, Re
throw new RuntimeException("Failed update recipient store", e);
}
}
}
@Override
public Profile getProfile(final RecipientId recipientId) {
@ -1026,7 +1017,6 @@ public class RecipientStore implements RecipientIdCreator, RecipientResolver, Re
private RecipientId resolveRecipientTrusted(RecipientAddress address, boolean isSelf) {
final Pair<RecipientId, List<RecipientId>> pair;
synchronized (recipientsLock) {
try (final var connection = database.getConnection()) {
connection.setAutoCommit(false);
pair = resolveRecipientTrustedLocked(connection, address, isSelf);
@ -1034,7 +1024,6 @@ public class RecipientStore implements RecipientIdCreator, RecipientResolver, Re
} catch (SQLException e) {
throw new RuntimeException("Failed update recipient store", e);
}
}
if (!pair.second().isEmpty()) {
logger.debug("Resolved address {}, merging {} other recipients", address, pair.second().size());
@ -1073,11 +1062,9 @@ public class RecipientStore implements RecipientIdCreator, RecipientResolver, Re
for (final var toBeMergedRecipientId : toBeMergedRecipientIds) {
recipientMergeHandler.mergeRecipients(connection, recipientId, toBeMergedRecipientId);
deleteRecipient(connection, toBeMergedRecipientId);
synchronized (recipientsLock) {
recipientAddressCache.entrySet().removeIf(e -> e.getValue().id().equals(toBeMergedRecipientId));
}
}
}
private RecipientId resolveRecipientLocked(
Connection connection, RecipientAddress address
@ -1164,7 +1151,6 @@ public class RecipientStore implements RecipientIdCreator, RecipientResolver, Re
}
private void removeRecipientAddress(Connection connection, RecipientId recipientId) throws SQLException {
synchronized (recipientsLock) {
recipientAddressCache.entrySet().removeIf(e -> e.getValue().id().equals(recipientId));
final var sql = (
"""
@ -1178,12 +1164,10 @@ public class RecipientStore implements RecipientIdCreator, RecipientResolver, Re
statement.executeUpdate();
}
}
}
private void updateRecipientAddress(
Connection connection, RecipientId recipientId, final RecipientAddress address
) throws SQLException {
synchronized (recipientsLock) {
recipientAddressCache.entrySet().removeIf(e -> e.getValue().id().equals(recipientId));
final var sql = (
"""
@ -1202,7 +1186,6 @@ public class RecipientStore implements RecipientIdCreator, RecipientResolver, Re
}
rotateStorageId(connection, recipientId);
}
}
private void deleteRecipient(final Connection connection, final RecipientId recipientId) throws SQLException {
final var sql = (