TheArchive/session-android
1
0
Fork 0
mirror of https://github.com/oxen-io/session-android.git synced 2025-06-03 03:38:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Enforce max envelope size in more places.

Browse Source
This commit is contained in:
Greyson Parrelli 2020-11-06 15:58:10 -05:00 committed by Cody Henthorne
parent 0068d62122
commit f4c1e34402
2 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
app/src/main/java/org/thoughtcrime/securesms/util/FeatureFlags.java
View File

@ -107,7 +107,8 @@ public final class FeatureFlags {
private static final Set<String> HOT_SWAPPABLE = Sets.newHashSet(
GROUPS_V2_JOIN_VERSION,
VERIFY_V2,
CLIENT_EXPIRATION
CLIENT_EXPIRATION,
MAX_ENVELOPE_SIZE
);
/**

19
libsignal/service/src/main/java/org/whispersystems/signalservice/api/SignalServiceMessageSender.java
View File

@ -701,13 +701,7 @@ public class SignalServiceMessageSender {
builder.setTimestamp(message.getTimestamp());
byte[] content = container.setDataMessage(builder).build().toByteArray();
if (maxEnvelopeSize > 0 && content.length > maxEnvelopeSize) {
throw new ContentTooLargeException(content.length);
}
return content;
return enforceMaxContentSize(container.setDataMessage(builder).build().toByteArray());
}
private byte[] createCallContent(SignalServiceCallMessage callMessage) {
@ -1261,6 +1255,8 @@ public class SignalServiceMessageSender {
CancelationSignal cancelationSignal)
throws IOException
{
enforceMaxContentSize(content);
long startTime = System.currentTimeMillis();
List<Future<SendMessageResult>> futureResults = new LinkedList<>();
Iterator<SignalServiceAddress> recipientIterator = recipients.iterator();
@ -1325,6 +1321,8 @@ public class SignalServiceMessageSender {
CancelationSignal cancelationSignal)
throws UntrustedIdentityException, IOException
{
enforceMaxContentSize(content);
long startTime = System.currentTimeMillis();
for (int i = 0; i < RETRY_COUNT; i++) {
@ -1591,6 +1589,13 @@ public class SignalServiceMessageSender {
return results;
}
private byte[] enforceMaxContentSize(byte[] content) {
if (maxEnvelopeSize > 0 && content.length > maxEnvelopeSize) {
throw new ContentTooLargeException(content.length);
}
return content;
}
public static interface EventListener {
public void onSecurityEvent(SignalServiceAddress address);
}