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

Includ pni signatures if necessary

Browse source
This commit is contained in:
AsamK 2024-02-18 16:31:34 +01:00
parent bd792f5b7f
commit 91ed49e019
4 changed files with 83 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -299,6 +299,12 @@ public final class IncomingMessageHandler {
final var senderDeviceId = senderDeviceAddress.deviceId();
final var destination = getDestination(envelope);
if (account.getPni().equals(destination.serviceId)) {
account.getRecipientStore().markNeedsPniSignature(destination.recipientId, true);
} else if (account.getAci().equals(destination.serviceId)) {
account.getRecipientStore().markNeedsPniSignature(destination.recipientId, false);
}
if (content.getReceiptMessage().isPresent()) {
final var message = content.getReceiptMessage().get();
if (message.isDeliveryReceipt()) {

47
lib/src/main/java/org/asamk/signal/manager/helper/SendHelper.java
View file

@ -127,10 +127,10 @@ public class SendHelper {
) {
final var messageSendLogStore = account.getMessageSendLogStore();
final var result = handleSendMessage(recipientId,
(messageSender, address, unidentifiedAccess) -> messageSender.sendReceipt(address,
(messageSender, address, unidentifiedAccess, includePniSignature) -> messageSender.sendReceipt(address,
unidentifiedAccess,
receiptMessage,
false));
includePniSignature));
messageSendLogStore.insertIfPossible(receiptMessage.getWhen(), result, ContentHint.IMPLICIT, false);
handleSendMessageResult(result);
return result;
@ -144,13 +144,14 @@ public class SendHelper {
.withProfileKey(profileKey)
.build();
return handleSendMessage(recipientId,
(messageSender, address, unidentifiedAccess) -> messageSender.sendDataMessage(address,
(messageSender, address, unidentifiedAccess, includePniSignature) -> messageSender.sendDataMessage(
address,
unidentifiedAccess,
ContentHint.IMPLICIT,
message,
SignalServiceMessageSender.IndividualSendEvents.EMPTY,
false,
false));
includePniSignature));
}
public SendMessageResult sendRetryReceipt(
@ -161,7 +162,8 @@ public class SendHelper {
recipientId,
errorMessage.getDeviceId());
final var result = handleSendMessage(recipientId,
(messageSender, address, unidentifiedAccess) -> messageSender.sendRetryReceipt(address,
(messageSender, address, unidentifiedAccess, includePniSignature) -> messageSender.sendRetryReceipt(
address,
unidentifiedAccess,
groupId.map(GroupId::serialize),
errorMessage));
@ -170,7 +172,10 @@ public class SendHelper {
}
public SendMessageResult sendNullMessage(RecipientId recipientId) {
final var result = handleSendMessage(recipientId, SignalServiceMessageSender::sendNullMessage);
final var result = handleSendMessage(recipientId,
(messageSender, address, unidentifiedAccess, includePniSignature) -> messageSender.sendNullMessage(
address,
unidentifiedAccess));
handleSendMessageResult(result);
return result;
}
@ -220,10 +225,8 @@ public class SendHelper {
SignalServiceTypingMessage message, RecipientId recipientId
) {
final var result = handleSendMessage(recipientId,
(messageSender, address, unidentifiedAccess) -> messageSender.sendTyping(List.of(address),
List.of(unidentifiedAccess),
message,
null).getFirst());
(messageSender, address, unidentifiedAccess, includePniSignature) -> messageSender.sendTyping(List.of(
address), List.of(unidentifiedAccess), message, null).getFirst());
handleSendMessageResult(result);
return result;
}
@ -247,7 +250,8 @@ public class SendHelper {
logger.trace("Resending message {} to {}", timestamp, recipientId);
if (messageSendLogEntry.groupId().isEmpty()) {
return handleSendMessage(recipientId,
(messageSender, address, unidentifiedAccess) -> messageSender.resendContent(address,
(messageSender, address, unidentifiedAccess, includePniSignature) -> messageSender.resendContent(
address,
unidentifiedAccess,
timestamp,
messageSendLogEntry.content(),
@ -277,7 +281,7 @@ public class SendHelper {
.build();
final var result = handleSendMessage(recipientId,
(messageSender, address, unidentifiedAccess) -> messageSender.resendContent(address,
(messageSender, address, unidentifiedAccess, includePniSignature) -> messageSender.resendContent(address,
unidentifiedAccess,
timestamp,
contentToSend,
@ -660,17 +664,18 @@ public class SendHelper {
) {
final var messageSendLogStore = account.getMessageSendLogStore();
final var urgent = true;
final var includePniSignature = false;
final var result = handleSendMessage(recipientId,
editTargetTimestamp.isEmpty()
? (messageSender, address, unidentifiedAccess) -> messageSender.sendDataMessage(address,
? (messageSender, address, unidentifiedAccess, includePniSignature) -> messageSender.sendDataMessage(
address,
unidentifiedAccess,
ContentHint.RESENDABLE,
message,
SignalServiceMessageSender.IndividualSendEvents.EMPTY,
urgent,
includePniSignature)
: (messageSender, address, unidentifiedAccess) -> messageSender.sendEditMessage(address,
: (messageSender, address, unidentifiedAccess, includePniSignature) -> messageSender.sendEditMessage(
address,
unidentifiedAccess,
ContentHint.RESENDABLE,
message,
@ -687,8 +692,12 @@ public class SendHelper {
var address = context.getRecipientHelper().resolveSignalServiceAddress(recipientId);
try {
final boolean includePniSignature = account.getRecipientStore().needsPniSignature(recipientId);
try {
return s.send(messageSender, address, context.getUnidentifiedAccessHelper().getAccessFor(recipientId));
return s.send(messageSender,
address,
context.getUnidentifiedAccessHelper().getAccessFor(recipientId),
includePniSignature);
} catch (UnregisteredUserException e) {
final RecipientId newRecipientId;
try {
@ -699,7 +708,8 @@ public class SendHelper {
address = context.getRecipientHelper().resolveSignalServiceAddress(newRecipientId);
return s.send(messageSender,
address,
context.getUnidentifiedAccessHelper().getAccessFor(newRecipientId));
context.getUnidentifiedAccessHelper().getAccessFor(newRecipientId),
includePniSignature);
}
} catch (UnregisteredUserException e) {
return SendMessageResult.unregisteredFailure(address);
@ -774,7 +784,8 @@ public class SendHelper {
SendMessageResult send(
SignalServiceMessageSender messageSender,
SignalServiceAddress address,
Optional<UnidentifiedAccessPair> unidentifiedAccess
Optional<UnidentifiedAccessPair> unidentifiedAccess,
boolean includePniSignature
) throws IOException, UnregisteredUserException, ProofRequiredException, RateLimitException, org.whispersystems.signalservice.api.crypto.UntrustedIdentityException;
}

10
lib/src/main/java/org/asamk/signal/manager/storage/AccountDatabase.java
View file

@ -33,7 +33,7 @@ import java.util.UUID;
public class AccountDatabase extends Database {
private static final Logger logger = LoggerFactory.getLogger(AccountDatabase.class);
private static final long DATABASE_VERSION = 23;
private static final long DATABASE_VERSION = 24;
private AccountDatabase(final HikariDataSource dataSource) {
super(logger, DATABASE_VERSION, dataSource);
@ -573,6 +573,14 @@ public class AccountDatabase extends Database {
""");
}
}
if (oldVersion < 24) {
logger.debug("Updating database: Create needs_pni_signature column");
try (final var statement = connection.createStatement()) {
statement.executeUpdate("""
ALTER TABLE recipient ADD needs_pni_signature INTEGER NOT NULL DEFAULT FALSE;
""");
}
}
}
private static void createUuidMappingTable(

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

@ -66,6 +66,7 @@ public class RecipientStore implements RecipientIdCreator, RecipientResolver, Re
unregistered_timestamp INTEGER,
profile_key BLOB,
profile_key_credential BLOB,
needs_pni_signature INTEGER NOT NULL DEFAULT FALSE,
given_name TEXT,
family_name TEXT,
@ -856,6 +857,44 @@ public class RecipientStore implements RecipientIdCreator, RecipientResolver, Re
return count;
}
public void markNeedsPniSignature(final RecipientId recipientId, final boolean value) {
logger.debug("Marking {} numbers as need pni signature = {}", recipientId, value);
try (final var connection = database.getConnection()) {
final var sql = (
"""
UPDATE %s
SET needs_pni_signature = ?
WHERE _id = ?
"""
).formatted(TABLE_RECIPIENT);
try (final var statement = connection.prepareStatement(sql)) {
statement.setBoolean(1, value);
statement.setLong(2, recipientId.id());
statement.executeUpdate();
}
} catch (SQLException e) {
throw new RuntimeException("Failed update recipient store", e);
}
}
public boolean needsPniSignature(final RecipientId recipientId) {
try (final var connection = database.getConnection()) {
final var sql = (
"""
SELECT needs_pni_signature
FROM %s
WHERE _id = ?
"""
).formatted(TABLE_RECIPIENT);
try (final var statement = connection.prepareStatement(sql)) {
statement.setLong(1, recipientId.id());
return Utils.executeQuerySingleRow(statement, resultSet -> resultSet.getBoolean("needs_pni_signature"));
}
} catch (SQLException e) {
throw new RuntimeException("Failed read recipient store", e);
}
}
public void markUnregistered(final Set<String> unregisteredUsers) {
logger.debug("Marking {} numbers as unregistered", unregisteredUsers.size());
try (final var connection = database.getConnection()) {