]> nmode's Git Repositories - signal-cli/blob - lib/src/main/java/org/asamk/signal/manager/internal/ManagerImpl.java
Implement remote storage sync
[signal-cli] / lib / src / main / java / org / asamk / signal / manager / internal / ManagerImpl.java
1 /*
2 Copyright (C) 2015-2022 AsamK and contributors
3
4 This program is free software: you can redistribute it and/or modify
5 it under the terms of the GNU General Public License as published by
6 the Free Software Foundation, either version 3 of the License, or
7 (at your option) any later version.
8
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
13
14 You should have received a copy of the GNU General Public License
15 along with this program. If not, see <http://www.gnu.org/licenses/>.
16 */
17 package org.asamk.signal.manager.internal;
18
19 import org.asamk.signal.manager.Manager;
20 import org.asamk.signal.manager.api.AlreadyReceivingException;
21 import org.asamk.signal.manager.api.AttachmentInvalidException;
22 import org.asamk.signal.manager.api.CaptchaRequiredException;
23 import org.asamk.signal.manager.api.Configuration;
24 import org.asamk.signal.manager.api.Device;
25 import org.asamk.signal.manager.api.DeviceLinkUrl;
26 import org.asamk.signal.manager.api.Group;
27 import org.asamk.signal.manager.api.GroupId;
28 import org.asamk.signal.manager.api.GroupInviteLinkUrl;
29 import org.asamk.signal.manager.api.GroupNotFoundException;
30 import org.asamk.signal.manager.api.GroupSendingNotAllowedException;
31 import org.asamk.signal.manager.api.Identity;
32 import org.asamk.signal.manager.api.IdentityVerificationCode;
33 import org.asamk.signal.manager.api.InactiveGroupLinkException;
34 import org.asamk.signal.manager.api.IncorrectPinException;
35 import org.asamk.signal.manager.api.InvalidDeviceLinkException;
36 import org.asamk.signal.manager.api.InvalidStickerException;
37 import org.asamk.signal.manager.api.InvalidUsernameException;
38 import org.asamk.signal.manager.api.LastGroupAdminException;
39 import org.asamk.signal.manager.api.Message;
40 import org.asamk.signal.manager.api.MessageEnvelope;
41 import org.asamk.signal.manager.api.NonNormalizedPhoneNumberException;
42 import org.asamk.signal.manager.api.NotAGroupMemberException;
43 import org.asamk.signal.manager.api.NotPrimaryDeviceException;
44 import org.asamk.signal.manager.api.Pair;
45 import org.asamk.signal.manager.api.PendingAdminApprovalException;
46 import org.asamk.signal.manager.api.PinLockedException;
47 import org.asamk.signal.manager.api.Profile;
48 import org.asamk.signal.manager.api.RateLimitException;
49 import org.asamk.signal.manager.api.ReceiveConfig;
50 import org.asamk.signal.manager.api.Recipient;
51 import org.asamk.signal.manager.api.RecipientIdentifier;
52 import org.asamk.signal.manager.api.SendGroupMessageResults;
53 import org.asamk.signal.manager.api.SendMessageResult;
54 import org.asamk.signal.manager.api.SendMessageResults;
55 import org.asamk.signal.manager.api.StickerPackId;
56 import org.asamk.signal.manager.api.StickerPackInvalidException;
57 import org.asamk.signal.manager.api.StickerPackUrl;
58 import org.asamk.signal.manager.api.TextStyle;
59 import org.asamk.signal.manager.api.TypingAction;
60 import org.asamk.signal.manager.api.UnregisteredRecipientException;
61 import org.asamk.signal.manager.api.UpdateGroup;
62 import org.asamk.signal.manager.api.UpdateProfile;
63 import org.asamk.signal.manager.api.UserStatus;
64 import org.asamk.signal.manager.api.UsernameLinkUrl;
65 import org.asamk.signal.manager.config.ServiceEnvironmentConfig;
66 import org.asamk.signal.manager.helper.AccountFileUpdater;
67 import org.asamk.signal.manager.helper.Context;
68 import org.asamk.signal.manager.helper.RecipientHelper.RegisteredUser;
69 import org.asamk.signal.manager.jobs.SyncStorageJob;
70 import org.asamk.signal.manager.storage.AttachmentStore;
71 import org.asamk.signal.manager.storage.AvatarStore;
72 import org.asamk.signal.manager.storage.SignalAccount;
73 import org.asamk.signal.manager.storage.groups.GroupInfo;
74 import org.asamk.signal.manager.storage.identities.IdentityInfo;
75 import org.asamk.signal.manager.storage.recipients.RecipientAddress;
76 import org.asamk.signal.manager.storage.recipients.RecipientId;
77 import org.asamk.signal.manager.storage.stickerPacks.JsonStickerPack;
78 import org.asamk.signal.manager.storage.stickerPacks.StickerPackStore;
79 import org.asamk.signal.manager.storage.stickers.StickerPack;
80 import org.asamk.signal.manager.util.AttachmentUtils;
81 import org.asamk.signal.manager.util.KeyUtils;
82 import org.asamk.signal.manager.util.MimeUtils;
83 import org.asamk.signal.manager.util.StickerUtils;
84 import org.signal.libsignal.protocol.InvalidMessageException;
85 import org.signal.libsignal.usernames.BaseUsernameException;
86 import org.slf4j.Logger;
87 import org.slf4j.LoggerFactory;
88 import org.whispersystems.signalservice.api.SignalSessionLock;
89 import org.whispersystems.signalservice.api.messages.SignalServiceDataMessage;
90 import org.whispersystems.signalservice.api.messages.SignalServicePreview;
91 import org.whispersystems.signalservice.api.messages.SignalServiceReceiptMessage;
92 import org.whispersystems.signalservice.api.messages.SignalServiceTypingMessage;
93 import org.whispersystems.signalservice.api.push.ServiceId;
94 import org.whispersystems.signalservice.api.push.ServiceId.ACI;
95 import org.whispersystems.signalservice.api.push.ServiceIdType;
96 import org.whispersystems.signalservice.api.push.exceptions.CdsiResourceExhaustedException;
97 import org.whispersystems.signalservice.api.util.DeviceNameUtil;
98 import org.whispersystems.signalservice.api.util.InvalidNumberException;
99 import org.whispersystems.signalservice.api.util.PhoneNumberFormatter;
100 import org.whispersystems.signalservice.api.util.StreamDetails;
101 import org.whispersystems.signalservice.internal.util.Hex;
102 import org.whispersystems.signalservice.internal.util.Util;
103
104 import java.io.ByteArrayInputStream;
105 import java.io.File;
106 import java.io.IOException;
107 import java.io.InputStream;
108 import java.nio.charset.StandardCharsets;
109 import java.time.Duration;
110 import java.util.ArrayList;
111 import java.util.Collection;
112 import java.util.HashMap;
113 import java.util.HashSet;
114 import java.util.List;
115 import java.util.Map;
116 import java.util.Objects;
117 import java.util.Optional;
118 import java.util.Set;
119 import java.util.concurrent.ExecutorService;
120 import java.util.concurrent.Executors;
121 import java.util.concurrent.TimeUnit;
122 import java.util.concurrent.atomic.AtomicInteger;
123 import java.util.concurrent.locks.ReentrantLock;
124 import java.util.function.Function;
125 import java.util.stream.Collectors;
126 import java.util.stream.Stream;
127
128 import io.reactivex.rxjava3.disposables.CompositeDisposable;
129 import io.reactivex.rxjava3.schedulers.Schedulers;
130
131 public class ManagerImpl implements Manager {
132
133 private static final Logger logger = LoggerFactory.getLogger(ManagerImpl.class);
134
135 private SignalAccount account;
136 private final SignalDependencies dependencies;
137 private final Context context;
138
139 private final ExecutorService executor = Executors.newCachedThreadPool();
140
141 private Thread receiveThread;
142 private boolean isReceivingSynchronous;
143 private final Set<ReceiveMessageHandler> weakHandlers = new HashSet<>();
144 private final Set<ReceiveMessageHandler> messageHandlers = new HashSet<>();
145 private final List<Runnable> closedListeners = new ArrayList<>();
146 private final List<Runnable> addressChangedListeners = new ArrayList<>();
147 private final CompositeDisposable disposable = new CompositeDisposable();
148
149 public ManagerImpl(
150 SignalAccount account,
151 PathConfig pathConfig,
152 AccountFileUpdater accountFileUpdater,
153 ServiceEnvironmentConfig serviceEnvironmentConfig,
154 String userAgent
155 ) {
156 this.account = account;
157
158 final var sessionLock = new SignalSessionLock() {
159 private final ReentrantLock LEGACY_LOCK = new ReentrantLock();
160
161 @Override
162 public Lock acquire() {
163 LEGACY_LOCK.lock();
164 return LEGACY_LOCK::unlock;
165 }
166 };
167 this.dependencies = new SignalDependencies(serviceEnvironmentConfig,
168 userAgent,
169 account.getCredentialsProvider(),
170 account.getSignalServiceDataStore(),
171 executor,
172 sessionLock);
173 final var avatarStore = new AvatarStore(pathConfig.avatarsPath());
174 final var attachmentStore = new AttachmentStore(pathConfig.attachmentsPath());
175 final var stickerPackStore = new StickerPackStore(pathConfig.stickerPacksPath());
176
177 this.context = new Context(account, new AccountFileUpdater() {
178 @Override
179 public void updateAccountIdentifiers(final String number, final ACI aci) {
180 accountFileUpdater.updateAccountIdentifiers(number, aci);
181 synchronized (addressChangedListeners) {
182 addressChangedListeners.forEach(Runnable::run);
183 }
184 }
185
186 @Override
187 public void removeAccount() {
188 accountFileUpdater.removeAccount();
189 }
190 }, dependencies, avatarStore, attachmentStore, stickerPackStore);
191 this.context.getAccountHelper().setUnregisteredListener(this::close);
192 this.context.getReceiveHelper().setAuthenticationFailureListener(this::close);
193 this.context.getReceiveHelper().setCaughtUpWithOldMessagesListener(() -> {
194 synchronized (this) {
195 this.notifyAll();
196 }
197 });
198 disposable.add(account.getIdentityKeyStore()
199 .getIdentityChanges()
200 .observeOn(Schedulers.from(executor))
201 .subscribe(serviceId -> {
202 logger.trace("Archiving old sessions for {}", serviceId);
203 account.getAccountData(ServiceIdType.ACI).getSessionStore().archiveSessions(serviceId);
204 account.getAccountData(ServiceIdType.PNI).getSessionStore().archiveSessions(serviceId);
205 account.getSenderKeyStore().deleteSharedWith(serviceId);
206 final var recipientId = account.getRecipientResolver().resolveRecipient(serviceId);
207 final var profile = account.getProfileStore().getProfile(recipientId);
208 if (profile != null) {
209 account.getProfileStore()
210 .storeProfile(recipientId,
211 Profile.newBuilder(profile)
212 .withUnidentifiedAccessMode(Profile.UnidentifiedAccessMode.UNKNOWN)
213 .withLastUpdateTimestamp(0)
214 .build());
215 }
216 }));
217 }
218
219 @Override
220 public String getSelfNumber() {
221 return account.getNumber();
222 }
223
224 public void checkAccountState() throws IOException {
225 context.getAccountHelper().checkAccountState();
226 final var lastRecipientsRefresh = account.getLastRecipientsRefresh();
227 if (lastRecipientsRefresh == null
228 || lastRecipientsRefresh < System.currentTimeMillis() - TimeUnit.DAYS.toMillis(1)) {
229 try {
230 context.getRecipientHelper().refreshUsers();
231 } catch (Exception e) {
232 logger.warn("Full CDSI recipients refresh failed, ignoring: {} ({})",
233 e.getMessage(),
234 e.getClass().getSimpleName());
235 logger.debug("Full CDSI refresh failed", e);
236 }
237 context.getAccountHelper().checkWhoAmiI();
238 }
239 }
240
241 @Override
242 public Map<String, UserStatus> getUserStatus(Set<String> numbers) throws IOException, RateLimitException {
243 final var canonicalizedNumbers = numbers.stream().collect(Collectors.toMap(n -> n, n -> {
244 try {
245 final var canonicalizedNumber = PhoneNumberFormatter.formatNumber(n, account.getNumber());
246 if (!canonicalizedNumber.equals(n)) {
247 logger.debug("Normalized number {} to {}.", n, canonicalizedNumber);
248 }
249 return canonicalizedNumber;
250 } catch (InvalidNumberException e) {
251 return "";
252 }
253 }));
254
255 // Note "registeredUsers" has no optionals. It only gives us info on users who are registered
256 final var canonicalizedNumbersSet = canonicalizedNumbers.values()
257 .stream()
258 .filter(s -> !s.isEmpty())
259 .collect(Collectors.toSet());
260
261 final Map<String, RegisteredUser> registeredUsers;
262 try {
263 registeredUsers = context.getRecipientHelper().getRegisteredUsers(canonicalizedNumbersSet);
264 } catch (CdsiResourceExhaustedException e) {
265 logger.debug("CDSI resource exhausted: {}", e.getMessage());
266 throw new RateLimitException(System.currentTimeMillis() + e.getRetryAfterSeconds() * 1000L);
267 }
268
269 return numbers.stream().collect(Collectors.toMap(n -> n, n -> {
270 final var number = canonicalizedNumbers.get(n);
271 final var user = registeredUsers.get(number);
272 final var serviceId = user == null ? null : user.getServiceId();
273 final var profile = serviceId == null
274 ? null
275 : context.getProfileHelper()
276 .getRecipientProfile(account.getRecipientResolver().resolveRecipient(serviceId));
277 return new UserStatus(number.isEmpty() ? null : number,
278 serviceId == null ? null : serviceId.getRawUuid(),
279 profile != null
280 && profile.getUnidentifiedAccessMode() == Profile.UnidentifiedAccessMode.UNRESTRICTED);
281 }));
282 }
283
284 @Override
285 public void updateAccountAttributes(String deviceName, Boolean unrestrictedUnidentifiedSender) throws IOException {
286 if (deviceName != null) {
287 context.getAccountHelper().setDeviceName(deviceName);
288 }
289 if (unrestrictedUnidentifiedSender != null) {
290 account.setUnrestrictedUnidentifiedAccess(unrestrictedUnidentifiedSender);
291 }
292 context.getAccountHelper().updateAccountAttributes();
293 context.getAccountHelper().checkWhoAmiI();
294 }
295
296 @Override
297 public Configuration getConfiguration() {
298 final var configurationStore = account.getConfigurationStore();
299 return Configuration.from(configurationStore);
300 }
301
302 @Override
303 public void updateConfiguration(Configuration configuration) {
304 final var configurationStore = account.getConfigurationStore();
305 if (configuration.readReceipts().isPresent()) {
306 configurationStore.setReadReceipts(configuration.readReceipts().get());
307 }
308 if (configuration.unidentifiedDeliveryIndicators().isPresent()) {
309 configurationStore.setUnidentifiedDeliveryIndicators(configuration.unidentifiedDeliveryIndicators().get());
310 }
311 if (configuration.typingIndicators().isPresent()) {
312 configurationStore.setTypingIndicators(configuration.typingIndicators().get());
313 }
314 if (configuration.linkPreviews().isPresent()) {
315 configurationStore.setLinkPreviews(configuration.linkPreviews().get());
316 }
317 context.getSyncHelper().sendConfigurationMessage();
318 syncRemoteStorage();
319 }
320
321 @Override
322 public void updateProfile(UpdateProfile updateProfile) throws IOException {
323 context.getProfileHelper()
324 .setProfile(updateProfile.getGivenName(),
325 updateProfile.getFamilyName(),
326 updateProfile.getAbout(),
327 updateProfile.getAboutEmoji(),
328 updateProfile.isDeleteAvatar()
329 ? Optional.empty()
330 : updateProfile.getAvatar() == null ? null : Optional.of(updateProfile.getAvatar()),
331 updateProfile.getMobileCoinAddress());
332 context.getSyncHelper().sendSyncFetchProfileMessage();
333 }
334
335 void refreshCurrentUsername() throws IOException, BaseUsernameException {
336 context.getAccountHelper().refreshCurrentUsername();
337 }
338
339 @Override
340 public String getUsername() {
341 return account.getUsername();
342 }
343
344 @Override
345 public UsernameLinkUrl getUsernameLink() {
346 return new UsernameLinkUrl(account.getUsernameLink());
347 }
348
349 @Override
350 public void setUsername(final String username) throws IOException, InvalidUsernameException {
351 try {
352 context.getAccountHelper().reserveUsername(username);
353 } catch (BaseUsernameException e) {
354 throw new InvalidUsernameException(e.getMessage() + " (" + e.getClass().getSimpleName() + ")", e);
355 }
356 }
357
358 @Override
359 public void deleteUsername() throws IOException {
360 context.getAccountHelper().deleteUsername();
361 }
362
363 @Override
364 public void startChangeNumber(
365 String newNumber, boolean voiceVerification, String captcha
366 ) throws RateLimitException, IOException, CaptchaRequiredException, NonNormalizedPhoneNumberException, NotPrimaryDeviceException {
367 if (!account.isPrimaryDevice()) {
368 throw new NotPrimaryDeviceException();
369 }
370 context.getAccountHelper().startChangeNumber(newNumber, voiceVerification, captcha);
371 }
372
373 @Override
374 public void finishChangeNumber(
375 String newNumber, String verificationCode, String pin
376 ) throws IncorrectPinException, PinLockedException, IOException, NotPrimaryDeviceException {
377 if (!account.isPrimaryDevice()) {
378 throw new NotPrimaryDeviceException();
379 }
380 context.getAccountHelper().finishChangeNumber(newNumber, verificationCode, pin);
381 }
382
383 @Override
384 public void unregister() throws IOException {
385 context.getAccountHelper().unregister();
386 }
387
388 @Override
389 public void deleteAccount() throws IOException {
390 context.getAccountHelper().deleteAccount();
391 }
392
393 @Override
394 public void submitRateLimitRecaptchaChallenge(String challenge, String captcha) throws IOException {
395 captcha = captcha == null ? null : captcha.replace("signalcaptcha://", "");
396
397 dependencies.getAccountManager().submitRateLimitRecaptchaChallenge(challenge, captcha);
398 }
399
400 @Override
401 public List<Device> getLinkedDevices() throws IOException {
402 var devices = dependencies.getAccountManager().getDevices();
403 account.setMultiDevice(devices.size() > 1);
404 var identityKey = account.getAciIdentityKeyPair().getPrivateKey();
405 return devices.stream().map(d -> {
406 String deviceName = d.getName();
407 if (deviceName != null) {
408 try {
409 deviceName = DeviceNameUtil.decryptDeviceName(deviceName, identityKey);
410 } catch (IOException e) {
411 logger.debug("Failed to decrypt device name, maybe plain text?", e);
412 }
413 }
414 return new Device(d.getId(),
415 deviceName,
416 d.getCreated(),
417 d.getLastSeen(),
418 d.getId() == account.getDeviceId());
419 }).toList();
420 }
421
422 @Override
423 public void removeLinkedDevices(int deviceId) throws IOException {
424 context.getAccountHelper().removeLinkedDevices(deviceId);
425 }
426
427 @Override
428 public void addDeviceLink(DeviceLinkUrl linkUrl) throws IOException, InvalidDeviceLinkException, NotPrimaryDeviceException {
429 if (!account.isPrimaryDevice()) {
430 throw new NotPrimaryDeviceException();
431 }
432 context.getAccountHelper().addDevice(linkUrl);
433 }
434
435 @Override
436 public void setRegistrationLockPin(Optional<String> pin) throws IOException, NotPrimaryDeviceException {
437 if (!account.isPrimaryDevice()) {
438 throw new NotPrimaryDeviceException();
439 }
440 if (pin.isPresent()) {
441 context.getAccountHelper().setRegistrationPin(pin.get());
442 } else {
443 context.getAccountHelper().removeRegistrationPin();
444 }
445 }
446
447 void refreshPreKeys() throws IOException {
448 context.getPreKeyHelper().refreshPreKeysIfNecessary();
449 }
450
451 @Override
452 public List<Group> getGroups() {
453 return account.getGroupStore().getGroups().stream().map(this::toGroup).toList();
454 }
455
456 private Group toGroup(final GroupInfo groupInfo) {
457 if (groupInfo == null) {
458 return null;
459 }
460
461 return Group.from(groupInfo, account.getRecipientAddressResolver(), account.getSelfRecipientId());
462 }
463
464 @Override
465 public SendGroupMessageResults quitGroup(
466 GroupId groupId, Set<RecipientIdentifier.Single> groupAdmins
467 ) throws GroupNotFoundException, IOException, NotAGroupMemberException, LastGroupAdminException, UnregisteredRecipientException {
468 final var newAdmins = context.getRecipientHelper().resolveRecipients(groupAdmins);
469 return context.getGroupHelper().quitGroup(groupId, newAdmins);
470 }
471
472 @Override
473 public void deleteGroup(GroupId groupId) throws IOException {
474 final var group = context.getGroupHelper().getGroup(groupId);
475 if (group.isMember(account.getSelfRecipientId())) {
476 throw new IOException(
477 "The local group information cannot be removed, as the user is still a member of the group");
478 }
479 context.getGroupHelper().deleteGroup(groupId);
480 }
481
482 @Override
483 public Pair<GroupId, SendGroupMessageResults> createGroup(
484 String name, Set<RecipientIdentifier.Single> members, String avatarFile
485 ) throws IOException, AttachmentInvalidException, UnregisteredRecipientException {
486 return context.getGroupHelper()
487 .createGroup(name,
488 members == null ? null : context.getRecipientHelper().resolveRecipients(members),
489 avatarFile);
490 }
491
492 @Override
493 public SendGroupMessageResults updateGroup(
494 final GroupId groupId, final UpdateGroup updateGroup
495 ) throws IOException, GroupNotFoundException, AttachmentInvalidException, NotAGroupMemberException, GroupSendingNotAllowedException, UnregisteredRecipientException {
496 return context.getGroupHelper()
497 .updateGroup(groupId,
498 updateGroup.getName(),
499 updateGroup.getDescription(),
500 updateGroup.getMembers() == null
501 ? null
502 : context.getRecipientHelper().resolveRecipients(updateGroup.getMembers()),
503 updateGroup.getRemoveMembers() == null
504 ? null
505 : context.getRecipientHelper().resolveRecipients(updateGroup.getRemoveMembers()),
506 updateGroup.getAdmins() == null
507 ? null
508 : context.getRecipientHelper().resolveRecipients(updateGroup.getAdmins()),
509 updateGroup.getRemoveAdmins() == null
510 ? null
511 : context.getRecipientHelper().resolveRecipients(updateGroup.getRemoveAdmins()),
512 updateGroup.getBanMembers() == null
513 ? null
514 : context.getRecipientHelper().resolveRecipients(updateGroup.getBanMembers()),
515 updateGroup.getUnbanMembers() == null
516 ? null
517 : context.getRecipientHelper().resolveRecipients(updateGroup.getUnbanMembers()),
518 updateGroup.isResetGroupLink(),
519 updateGroup.getGroupLinkState(),
520 updateGroup.getAddMemberPermission(),
521 updateGroup.getEditDetailsPermission(),
522 updateGroup.getAvatarFile(),
523 updateGroup.getExpirationTimer(),
524 updateGroup.getIsAnnouncementGroup());
525 }
526
527 @Override
528 public Pair<GroupId, SendGroupMessageResults> joinGroup(
529 GroupInviteLinkUrl inviteLinkUrl
530 ) throws IOException, InactiveGroupLinkException, PendingAdminApprovalException {
531 return context.getGroupHelper().joinGroup(inviteLinkUrl);
532 }
533
534 private SendMessageResults sendMessage(
535 SignalServiceDataMessage.Builder messageBuilder, Set<RecipientIdentifier> recipients, boolean notifySelf
536 ) throws IOException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
537 return sendMessage(messageBuilder, recipients, notifySelf, Optional.empty());
538 }
539
540 private SendMessageResults sendMessage(
541 SignalServiceDataMessage.Builder messageBuilder,
542 Set<RecipientIdentifier> recipients,
543 boolean notifySelf,
544 Optional<Long> editTargetTimestamp
545 ) throws IOException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
546 var results = new HashMap<RecipientIdentifier, List<SendMessageResult>>();
547 long timestamp = System.currentTimeMillis();
548 messageBuilder.withTimestamp(timestamp);
549 for (final var recipient : recipients) {
550 if (recipient instanceof RecipientIdentifier.NoteToSelf || (
551 recipient instanceof RecipientIdentifier.Single single
552 && new RecipientAddress(single.toPartialRecipientAddress()).matches(account.getSelfRecipientAddress())
553 )) {
554 final var result = notifySelf
555 ? context.getSendHelper()
556 .sendMessage(messageBuilder, account.getSelfRecipientId(), editTargetTimestamp)
557 : context.getSendHelper().sendSelfMessage(messageBuilder, editTargetTimestamp);
558 results.put(recipient, List.of(toSendMessageResult(result)));
559 } else if (recipient instanceof RecipientIdentifier.Single single) {
560 try {
561 final var recipientId = context.getRecipientHelper().resolveRecipient(single);
562 final var result = context.getSendHelper()
563 .sendMessage(messageBuilder, recipientId, editTargetTimestamp);
564 results.put(recipient, List.of(toSendMessageResult(result)));
565 } catch (UnregisteredRecipientException e) {
566 results.put(recipient,
567 List.of(SendMessageResult.unregisteredFailure(single.toPartialRecipientAddress())));
568 }
569 } else if (recipient instanceof RecipientIdentifier.Group group) {
570 final var result = context.getSendHelper()
571 .sendAsGroupMessage(messageBuilder, group.groupId(), notifySelf, editTargetTimestamp);
572 results.put(recipient, result.stream().map(this::toSendMessageResult).toList());
573 }
574 }
575 return new SendMessageResults(timestamp, results);
576 }
577
578 private SendMessageResult toSendMessageResult(final org.whispersystems.signalservice.api.messages.SendMessageResult result) {
579 return SendMessageResult.from(result, account.getRecipientResolver(), account.getRecipientAddressResolver());
580 }
581
582 private SendMessageResults sendTypingMessage(
583 SignalServiceTypingMessage.Action action, Set<RecipientIdentifier> recipients
584 ) throws IOException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
585 var results = new HashMap<RecipientIdentifier, List<SendMessageResult>>();
586 final var timestamp = System.currentTimeMillis();
587 for (var recipient : recipients) {
588 if (recipient instanceof RecipientIdentifier.Single single) {
589 final var message = new SignalServiceTypingMessage(action, timestamp, Optional.empty());
590 try {
591 final var recipientId = context.getRecipientHelper().resolveRecipient(single);
592 final var result = context.getSendHelper().sendTypingMessage(message, recipientId);
593 results.put(recipient, List.of(toSendMessageResult(result)));
594 } catch (UnregisteredRecipientException e) {
595 results.put(recipient,
596 List.of(SendMessageResult.unregisteredFailure(single.toPartialRecipientAddress())));
597 }
598 } else if (recipient instanceof RecipientIdentifier.Group) {
599 final var groupId = ((RecipientIdentifier.Group) recipient).groupId();
600 final var message = new SignalServiceTypingMessage(action, timestamp, Optional.of(groupId.serialize()));
601 final var result = context.getSendHelper().sendGroupTypingMessage(message, groupId);
602 results.put(recipient, result.stream().map(this::toSendMessageResult).toList());
603 }
604 }
605 return new SendMessageResults(timestamp, results);
606 }
607
608 @Override
609 public SendMessageResults sendTypingMessage(
610 TypingAction action, Set<RecipientIdentifier> recipients
611 ) throws IOException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
612 return sendTypingMessage(action.toSignalService(), recipients);
613 }
614
615 @Override
616 public SendMessageResults sendReadReceipt(
617 RecipientIdentifier.Single sender, List<Long> messageIds
618 ) {
619 final var timestamp = System.currentTimeMillis();
620 var receiptMessage = new SignalServiceReceiptMessage(SignalServiceReceiptMessage.Type.READ,
621 messageIds,
622 timestamp);
623
624 return sendReceiptMessage(sender, timestamp, receiptMessage);
625 }
626
627 @Override
628 public SendMessageResults sendViewedReceipt(
629 RecipientIdentifier.Single sender, List<Long> messageIds
630 ) {
631 final var timestamp = System.currentTimeMillis();
632 var receiptMessage = new SignalServiceReceiptMessage(SignalServiceReceiptMessage.Type.VIEWED,
633 messageIds,
634 timestamp);
635
636 return sendReceiptMessage(sender, timestamp, receiptMessage);
637 }
638
639 private SendMessageResults sendReceiptMessage(
640 final RecipientIdentifier.Single sender,
641 final long timestamp,
642 final SignalServiceReceiptMessage receiptMessage
643 ) {
644 try {
645 final var result = context.getSendHelper()
646 .sendReceiptMessage(receiptMessage, context.getRecipientHelper().resolveRecipient(sender));
647 return new SendMessageResults(timestamp, Map.of(sender, List.of(toSendMessageResult(result))));
648 } catch (UnregisteredRecipientException e) {
649 return new SendMessageResults(timestamp,
650 Map.of(sender, List.of(SendMessageResult.unregisteredFailure(sender.toPartialRecipientAddress()))));
651 }
652 }
653
654 @Override
655 public SendMessageResults sendMessage(
656 Message message, Set<RecipientIdentifier> recipients, boolean notifySelf
657 ) throws IOException, AttachmentInvalidException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException, UnregisteredRecipientException, InvalidStickerException {
658 final var selfProfile = context.getProfileHelper().getSelfProfile();
659 if (selfProfile == null || selfProfile.getDisplayName().isEmpty()) {
660 logger.warn(
661 "No profile name set. When sending a message it's recommended to set a profile name with the updateProfile command. This may become mandatory in the future.");
662 }
663 final var messageBuilder = SignalServiceDataMessage.newBuilder();
664 applyMessage(messageBuilder, message);
665 return sendMessage(messageBuilder, recipients, notifySelf);
666 }
667
668 @Override
669 public SendMessageResults sendEditMessage(
670 Message message, Set<RecipientIdentifier> recipients, long editTargetTimestamp
671 ) throws IOException, AttachmentInvalidException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException, UnregisteredRecipientException, InvalidStickerException {
672 final var messageBuilder = SignalServiceDataMessage.newBuilder();
673 applyMessage(messageBuilder, message);
674 return sendMessage(messageBuilder, recipients, false, Optional.of(editTargetTimestamp));
675 }
676
677 private void applyMessage(
678 final SignalServiceDataMessage.Builder messageBuilder, final Message message
679 ) throws AttachmentInvalidException, IOException, UnregisteredRecipientException, InvalidStickerException {
680 if (message.messageText().length() > 2000) {
681 final var messageBytes = message.messageText().getBytes(StandardCharsets.UTF_8);
682 final var textAttachment = AttachmentUtils.createAttachmentStream(new StreamDetails(new ByteArrayInputStream(
683 messageBytes), MimeUtils.LONG_TEXT, messageBytes.length), Optional.empty());
684 messageBuilder.withBody(message.messageText().substring(0, 2000));
685 messageBuilder.withAttachment(context.getAttachmentHelper().uploadAttachment(textAttachment));
686 } else {
687 messageBuilder.withBody(message.messageText());
688 }
689 if (!message.attachments().isEmpty()) {
690 messageBuilder.withAttachments(context.getAttachmentHelper().uploadAttachments(message.attachments()));
691 }
692 if (!message.mentions().isEmpty()) {
693 messageBuilder.withMentions(resolveMentions(message.mentions()));
694 }
695 if (!message.textStyles().isEmpty()) {
696 messageBuilder.withBodyRanges(message.textStyles().stream().map(TextStyle::toBodyRange).toList());
697 }
698 if (message.quote().isPresent()) {
699 final var quote = message.quote().get();
700 final var quotedAttachments = new ArrayList<SignalServiceDataMessage.Quote.QuotedAttachment>();
701 for (final var a : quote.attachments()) {
702 final var quotedAttachment = new SignalServiceDataMessage.Quote.QuotedAttachment(a.contentType(),
703 a.filename(),
704 a.preview() == null ? null : context.getAttachmentHelper().uploadAttachment(a.preview()));
705 quotedAttachments.add(quotedAttachment);
706 }
707 messageBuilder.withQuote(new SignalServiceDataMessage.Quote(quote.timestamp(),
708 context.getRecipientHelper()
709 .resolveSignalServiceAddress(context.getRecipientHelper().resolveRecipient(quote.author()))
710 .getServiceId(),
711 quote.message(),
712 quotedAttachments,
713 resolveMentions(quote.mentions()),
714 SignalServiceDataMessage.Quote.Type.NORMAL,
715 quote.textStyles().stream().map(TextStyle::toBodyRange).toList()));
716 }
717 if (message.sticker().isPresent()) {
718 final var sticker = message.sticker().get();
719 final var packId = StickerPackId.deserialize(sticker.packId());
720 final var stickerId = sticker.stickerId();
721
722 final var stickerPack = context.getAccount().getStickerStore().getStickerPack(packId);
723 if (stickerPack == null) {
724 throw new InvalidStickerException("Sticker pack not found");
725 }
726 final var manifest = context.getStickerHelper().getOrRetrieveStickerPack(packId, stickerPack.packKey());
727 if (manifest.stickers().size() <= stickerId) {
728 throw new InvalidStickerException("Sticker id not part of this pack");
729 }
730 final var manifestSticker = manifest.stickers().get(stickerId);
731 final var streamDetails = context.getStickerPackStore().retrieveSticker(packId, stickerId);
732 if (streamDetails == null) {
733 throw new InvalidStickerException("Missing local sticker file");
734 }
735 messageBuilder.withSticker(new SignalServiceDataMessage.Sticker(packId.serialize(),
736 stickerPack.packKey(),
737 stickerId,
738 manifestSticker.emoji(),
739 AttachmentUtils.createAttachmentStream(streamDetails, Optional.empty())));
740 }
741 if (!message.previews().isEmpty()) {
742 final var previews = new ArrayList<SignalServicePreview>(message.previews().size());
743 for (final var p : message.previews()) {
744 final var image = p.image().isPresent() ? context.getAttachmentHelper()
745 .uploadAttachment(p.image().get()) : null;
746 previews.add(new SignalServicePreview(p.url(),
747 p.title(),
748 p.description(),
749 0,
750 Optional.ofNullable(image)));
751 }
752 messageBuilder.withPreviews(previews);
753 }
754 if (message.storyReply().isPresent()) {
755 final var storyReply = message.storyReply().get();
756 final var authorServiceId = context.getRecipientHelper()
757 .resolveSignalServiceAddress(context.getRecipientHelper().resolveRecipient(storyReply.author()))
758 .getServiceId();
759 messageBuilder.withStoryContext(new SignalServiceDataMessage.StoryContext(authorServiceId,
760 storyReply.timestamp()));
761 }
762 }
763
764 private ArrayList<SignalServiceDataMessage.Mention> resolveMentions(final List<Message.Mention> mentionList) throws UnregisteredRecipientException {
765 final var mentions = new ArrayList<SignalServiceDataMessage.Mention>();
766 for (final var m : mentionList) {
767 final var recipientId = context.getRecipientHelper().resolveRecipient(m.recipient());
768 mentions.add(new SignalServiceDataMessage.Mention(context.getRecipientHelper()
769 .resolveSignalServiceAddress(recipientId)
770 .getServiceId(), m.start(), m.length()));
771 }
772 return mentions;
773 }
774
775 @Override
776 public SendMessageResults sendRemoteDeleteMessage(
777 long targetSentTimestamp, Set<RecipientIdentifier> recipients
778 ) throws IOException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
779 var delete = new SignalServiceDataMessage.RemoteDelete(targetSentTimestamp);
780 final var messageBuilder = SignalServiceDataMessage.newBuilder().withRemoteDelete(delete);
781 for (final var recipient : recipients) {
782 if (recipient instanceof RecipientIdentifier.Uuid u) {
783 account.getMessageSendLogStore()
784 .deleteEntryForRecipientNonGroup(targetSentTimestamp, ACI.from(u.uuid()));
785 } else if (recipient instanceof RecipientIdentifier.Single r) {
786 try {
787 final var recipientId = context.getRecipientHelper().resolveRecipient(r);
788 final var address = account.getRecipientAddressResolver().resolveRecipientAddress(recipientId);
789 if (address.serviceId().isPresent()) {
790 account.getMessageSendLogStore()
791 .deleteEntryForRecipientNonGroup(targetSentTimestamp, address.serviceId().get());
792 }
793 } catch (UnregisteredRecipientException ignored) {
794 }
795 } else if (recipient instanceof RecipientIdentifier.Group r) {
796 account.getMessageSendLogStore().deleteEntryForGroup(targetSentTimestamp, r.groupId());
797 }
798 }
799 return sendMessage(messageBuilder, recipients, false);
800 }
801
802 @Override
803 public SendMessageResults sendMessageReaction(
804 String emoji,
805 boolean remove,
806 RecipientIdentifier.Single targetAuthor,
807 long targetSentTimestamp,
808 Set<RecipientIdentifier> recipients,
809 final boolean isStory
810 ) throws IOException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException, UnregisteredRecipientException {
811 var targetAuthorRecipientId = context.getRecipientHelper().resolveRecipient(targetAuthor);
812 final var authorServiceId = context.getRecipientHelper()
813 .resolveSignalServiceAddress(targetAuthorRecipientId)
814 .getServiceId();
815 var reaction = new SignalServiceDataMessage.Reaction(emoji, remove, authorServiceId, targetSentTimestamp);
816 final var messageBuilder = SignalServiceDataMessage.newBuilder().withReaction(reaction);
817 if (isStory) {
818 messageBuilder.withStoryContext(new SignalServiceDataMessage.StoryContext(authorServiceId,
819 targetSentTimestamp));
820 }
821 return sendMessage(messageBuilder, recipients, false);
822 }
823
824 @Override
825 public SendMessageResults sendPaymentNotificationMessage(
826 byte[] receipt, String note, RecipientIdentifier.Single recipient
827 ) throws IOException {
828 final var paymentNotification = new SignalServiceDataMessage.PaymentNotification(receipt, note);
829 final var payment = new SignalServiceDataMessage.Payment(paymentNotification, null);
830 final var messageBuilder = SignalServiceDataMessage.newBuilder().withPayment(payment);
831 try {
832 return sendMessage(messageBuilder, Set.of(recipient), false);
833 } catch (NotAGroupMemberException | GroupNotFoundException | GroupSendingNotAllowedException e) {
834 throw new AssertionError(e);
835 }
836 }
837
838 @Override
839 public SendMessageResults sendEndSessionMessage(Set<RecipientIdentifier.Single> recipients) throws IOException {
840 var messageBuilder = SignalServiceDataMessage.newBuilder().asEndSessionMessage();
841
842 try {
843 return sendMessage(messageBuilder,
844 recipients.stream().map(RecipientIdentifier.class::cast).collect(Collectors.toSet()),
845 false);
846 } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
847 throw new AssertionError(e);
848 } finally {
849 for (var recipient : recipients) {
850 final RecipientId recipientId;
851 try {
852 recipientId = context.getRecipientHelper().resolveRecipient(recipient);
853 } catch (UnregisteredRecipientException e) {
854 continue;
855 }
856 final var serviceId = context.getAccount()
857 .getRecipientAddressResolver()
858 .resolveRecipientAddress(recipientId)
859 .serviceId();
860 if (serviceId.isPresent()) {
861 account.getAccountData(ServiceIdType.ACI).getSessionStore().deleteAllSessions(serviceId.get());
862 }
863 }
864 }
865 }
866
867 @Override
868 public void hideRecipient(final RecipientIdentifier.Single recipient) {
869 final var recipientIdOptional = context.getRecipientHelper().resolveRecipientOptional(recipient);
870 if (recipientIdOptional.isPresent()) {
871 context.getContactHelper().setContactHidden(recipientIdOptional.get(), true);
872 account.removeRecipient(recipientIdOptional.get());
873 syncRemoteStorage();
874 }
875 }
876
877 @Override
878 public void deleteRecipient(final RecipientIdentifier.Single recipient) {
879 final var recipientIdOptional = context.getRecipientHelper().resolveRecipientOptional(recipient);
880 if (recipientIdOptional.isPresent()) {
881 account.removeRecipient(recipientIdOptional.get());
882 syncRemoteStorage();
883 }
884 }
885
886 @Override
887 public void deleteContact(final RecipientIdentifier.Single recipient) {
888 final var recipientIdOptional = context.getRecipientHelper().resolveRecipientOptional(recipient);
889 if (recipientIdOptional.isPresent()) {
890 account.getContactStore().deleteContact(recipientIdOptional.get());
891 syncRemoteStorage();
892 }
893 }
894
895 @Override
896 public void setContactName(
897 RecipientIdentifier.Single recipient, String givenName, final String familyName
898 ) throws NotPrimaryDeviceException, UnregisteredRecipientException {
899 if (!account.isPrimaryDevice()) {
900 throw new NotPrimaryDeviceException();
901 }
902 context.getContactHelper()
903 .setContactName(context.getRecipientHelper().resolveRecipient(recipient), givenName, familyName);
904 syncRemoteStorage();
905 }
906
907 @Override
908 public void setContactsBlocked(
909 Collection<RecipientIdentifier.Single> recipients, boolean blocked
910 ) throws IOException, UnregisteredRecipientException {
911 if (recipients.isEmpty()) {
912 return;
913 }
914 final var recipientIds = context.getRecipientHelper().resolveRecipients(recipients);
915 final var selfRecipientId = account.getSelfRecipientId();
916 boolean shouldRotateProfileKey = false;
917 for (final var recipientId : recipientIds) {
918 if (context.getContactHelper().isContactBlocked(recipientId) == blocked) {
919 continue;
920 }
921 context.getContactHelper().setContactBlocked(recipientId, blocked);
922 // if we don't have a common group with the blocked contact we need to rotate the profile key
923 shouldRotateProfileKey = blocked && (
924 shouldRotateProfileKey || account.getGroupStore()
925 .getGroups()
926 .stream()
927 .noneMatch(g -> g.isMember(selfRecipientId) && g.isMember(recipientId))
928 );
929 }
930 if (shouldRotateProfileKey) {
931 context.getProfileHelper().rotateProfileKey();
932 }
933 context.getSyncHelper().sendBlockedList();
934 syncRemoteStorage();
935 }
936
937 @Override
938 public void setGroupsBlocked(
939 final Collection<GroupId> groupIds, final boolean blocked
940 ) throws GroupNotFoundException, IOException {
941 if (groupIds.isEmpty()) {
942 return;
943 }
944 boolean shouldRotateProfileKey = false;
945 for (final var groupId : groupIds) {
946 if (context.getGroupHelper().isGroupBlocked(groupId) == blocked) {
947 continue;
948 }
949 context.getGroupHelper().setGroupBlocked(groupId, blocked);
950 shouldRotateProfileKey = blocked;
951 }
952 if (shouldRotateProfileKey) {
953 context.getProfileHelper().rotateProfileKey();
954 }
955 context.getSyncHelper().sendBlockedList();
956 syncRemoteStorage();
957 }
958
959 @Override
960 public void setExpirationTimer(
961 RecipientIdentifier.Single recipient, int messageExpirationTimer
962 ) throws IOException, UnregisteredRecipientException {
963 var recipientId = context.getRecipientHelper().resolveRecipient(recipient);
964 context.getContactHelper().setExpirationTimer(recipientId, messageExpirationTimer);
965 final var messageBuilder = SignalServiceDataMessage.newBuilder().asExpirationUpdate();
966 try {
967 sendMessage(messageBuilder, Set.of(recipient), false);
968 } catch (NotAGroupMemberException | GroupNotFoundException | GroupSendingNotAllowedException e) {
969 throw new AssertionError(e);
970 }
971 syncRemoteStorage();
972 }
973
974 @Override
975 public StickerPackUrl uploadStickerPack(File path) throws IOException, StickerPackInvalidException {
976 var manifest = StickerUtils.getSignalServiceStickerManifestUpload(path);
977
978 var messageSender = dependencies.getMessageSender();
979
980 var packKey = KeyUtils.createStickerUploadKey();
981 var packIdString = messageSender.uploadStickerManifest(manifest, packKey);
982 var packId = StickerPackId.deserialize(Hex.fromStringCondensed(packIdString));
983
984 var sticker = new StickerPack(packId, packKey);
985 account.getStickerStore().addStickerPack(sticker);
986 context.getSyncHelper().sendStickerOperationsMessage(List.of(sticker), List.of());
987
988 return new StickerPackUrl(packId, packKey);
989 }
990
991 @Override
992 public void installStickerPack(StickerPackUrl url) throws IOException {
993 final var packId = url.packId();
994 final var packKey = url.packKey();
995 try {
996 context.getStickerHelper().retrieveStickerPack(packId, packKey);
997 } catch (InvalidMessageException e) {
998 throw new IOException(e);
999 }
1000
1001 final var sticker = context.getStickerHelper().addOrUpdateStickerPack(packId, packKey, true);
1002 context.getSyncHelper().sendStickerOperationsMessage(List.of(sticker), List.of());
1003 }
1004
1005 @Override
1006 public List<org.asamk.signal.manager.api.StickerPack> getStickerPacks() {
1007 final var stickerPackStore = context.getStickerPackStore();
1008 return account.getStickerStore().getStickerPacks().stream().map(pack -> {
1009 if (stickerPackStore.existsStickerPack(pack.packId())) {
1010 try {
1011 final var manifest = stickerPackStore.retrieveManifest(pack.packId());
1012 return new org.asamk.signal.manager.api.StickerPack(pack.packId(),
1013 new StickerPackUrl(pack.packId(), pack.packKey()),
1014 pack.isInstalled(),
1015 manifest.title(),
1016 manifest.author(),
1017 Optional.ofNullable(manifest.cover() == null ? null : manifest.cover().toApi()),
1018 manifest.stickers().stream().map(JsonStickerPack.JsonSticker::toApi).toList());
1019 } catch (Exception e) {
1020 logger.warn("Failed to read local sticker pack manifest: {}", e.getMessage(), e);
1021 }
1022 }
1023
1024 return new org.asamk.signal.manager.api.StickerPack(pack.packId(), pack.packKey(), pack.isInstalled());
1025 }).toList();
1026 }
1027
1028 @Override
1029 public void requestAllSyncData() {
1030 context.getSyncHelper().requestAllSyncData();
1031 syncRemoteStorage();
1032 }
1033
1034 void syncRemoteStorage() {
1035 context.getJobExecutor().enqueueJob(new SyncStorageJob());
1036 }
1037
1038 @Override
1039 public void addReceiveHandler(final ReceiveMessageHandler handler, final boolean isWeakListener) {
1040 synchronized (messageHandlers) {
1041 if (isWeakListener) {
1042 weakHandlers.add(handler);
1043 } else {
1044 messageHandlers.add(handler);
1045 startReceiveThreadIfRequired();
1046 }
1047 }
1048 }
1049
1050 private static final AtomicInteger threadNumber = new AtomicInteger(0);
1051
1052 private void startReceiveThreadIfRequired() {
1053 if (receiveThread != null || isReceivingSynchronous) {
1054 return;
1055 }
1056 receiveThread = Thread.ofPlatform().name("receive-" + threadNumber.getAndIncrement()).start(() -> {
1057 logger.debug("Starting receiving messages");
1058 context.getReceiveHelper().receiveMessagesContinuously(this::passReceivedMessageToHandlers);
1059 logger.debug("Finished receiving messages");
1060 synchronized (messageHandlers) {
1061 receiveThread = null;
1062
1063 // Check if in the meantime another handler has been registered
1064 if (!messageHandlers.isEmpty()) {
1065 logger.debug("Another handler has been registered, starting receive thread again");
1066 startReceiveThreadIfRequired();
1067 }
1068 }
1069 });
1070 }
1071
1072 private void passReceivedMessageToHandlers(MessageEnvelope envelope, Throwable e) {
1073 synchronized (messageHandlers) {
1074 Stream.concat(messageHandlers.stream(), weakHandlers.stream()).forEach(h -> {
1075 try {
1076 h.handleMessage(envelope, e);
1077 } catch (Throwable ex) {
1078 logger.warn("Message handler failed, ignoring", ex);
1079 }
1080 });
1081 }
1082 }
1083
1084 @Override
1085 public void removeReceiveHandler(final ReceiveMessageHandler handler) {
1086 final Thread thread;
1087 synchronized (messageHandlers) {
1088 weakHandlers.remove(handler);
1089 messageHandlers.remove(handler);
1090 if (!messageHandlers.isEmpty() || receiveThread == null || isReceivingSynchronous) {
1091 return;
1092 }
1093 thread = receiveThread;
1094 receiveThread = null;
1095 }
1096
1097 stopReceiveThread(thread);
1098 }
1099
1100 private void stopReceiveThread(final Thread thread) {
1101 if (context.getReceiveHelper().requestStopReceiveMessages()) {
1102 logger.debug("Receive stop requested, interrupting read from server.");
1103 thread.interrupt();
1104 }
1105 try {
1106 thread.join();
1107 } catch (InterruptedException ignored) {
1108 }
1109 }
1110
1111 @Override
1112 public boolean isReceiving() {
1113 if (isReceivingSynchronous) {
1114 return true;
1115 }
1116 synchronized (messageHandlers) {
1117 return !messageHandlers.isEmpty();
1118 }
1119 }
1120
1121 @Override
1122 public void receiveMessages(
1123 Optional<Duration> timeout, Optional<Integer> maxMessages, ReceiveMessageHandler handler
1124 ) throws IOException, AlreadyReceivingException {
1125 receiveMessages(timeout.orElse(Duration.ofMinutes(1)), timeout.isPresent(), maxMessages.orElse(null), handler);
1126 }
1127
1128 @Override
1129 public void stopReceiveMessages() {
1130 Thread thread = null;
1131 synchronized (messageHandlers) {
1132 if (isReceivingSynchronous) {
1133 thread = receiveThread;
1134 receiveThread = null;
1135 }
1136 }
1137 if (thread != null) {
1138 stopReceiveThread(thread);
1139 }
1140 }
1141
1142 private void receiveMessages(
1143 Duration timeout, boolean returnOnTimeout, Integer maxMessages, ReceiveMessageHandler handler
1144 ) throws IOException, AlreadyReceivingException {
1145 synchronized (messageHandlers) {
1146 if (isReceiving()) {
1147 throw new AlreadyReceivingException("Already receiving message.");
1148 }
1149 isReceivingSynchronous = true;
1150 receiveThread = Thread.currentThread();
1151 }
1152 try {
1153 context.getReceiveHelper().receiveMessages(timeout, returnOnTimeout, maxMessages, (envelope, e) -> {
1154 passReceivedMessageToHandlers(envelope, e);
1155 handler.handleMessage(envelope, e);
1156 });
1157 } finally {
1158 synchronized (messageHandlers) {
1159 receiveThread = null;
1160 isReceivingSynchronous = false;
1161 if (!messageHandlers.isEmpty()) {
1162 startReceiveThreadIfRequired();
1163 }
1164 }
1165 }
1166 }
1167
1168 @Override
1169 public void setReceiveConfig(final ReceiveConfig receiveConfig) {
1170 context.getReceiveHelper().setReceiveConfig(receiveConfig);
1171 }
1172
1173 @Override
1174 public boolean isContactBlocked(final RecipientIdentifier.Single recipient) {
1175 final RecipientId recipientId;
1176 try {
1177 recipientId = context.getRecipientHelper().resolveRecipient(recipient);
1178 } catch (UnregisteredRecipientException e) {
1179 return false;
1180 }
1181 return context.getContactHelper().isContactBlocked(recipientId);
1182 }
1183
1184 @Override
1185 public void sendContacts() throws IOException {
1186 context.getSyncHelper().sendContacts();
1187 }
1188
1189 @Override
1190 public List<Recipient> getRecipients(
1191 boolean onlyContacts,
1192 Optional<Boolean> blocked,
1193 Collection<RecipientIdentifier.Single> recipients,
1194 Optional<String> name
1195 ) {
1196 final var recipientIds = recipients.stream().map(a -> {
1197 try {
1198 return context.getRecipientHelper().resolveRecipient(a);
1199 } catch (UnregisteredRecipientException e) {
1200 return null;
1201 }
1202 }).filter(Objects::nonNull).collect(Collectors.toSet());
1203 if (!recipients.isEmpty() && recipientIds.isEmpty()) {
1204 return List.of();
1205 }
1206 // refresh profiles of explicitly given recipients
1207 context.getProfileHelper().refreshRecipientProfiles(recipientIds);
1208 return account.getRecipientStore()
1209 .getRecipients(onlyContacts, blocked, recipientIds, name)
1210 .stream()
1211 .map(s -> new Recipient(s.getRecipientId(),
1212 s.getAddress().toApiRecipientAddress(),
1213 s.getContact(),
1214 s.getProfileKey(),
1215 s.getExpiringProfileKeyCredential(),
1216 s.getProfile()))
1217 .toList();
1218 }
1219
1220 @Override
1221 public String getContactOrProfileName(RecipientIdentifier.Single recipient) {
1222 final RecipientId recipientId;
1223 try {
1224 recipientId = context.getRecipientHelper().resolveRecipient(recipient);
1225 } catch (UnregisteredRecipientException e) {
1226 return null;
1227 }
1228
1229 final var contact = account.getContactStore().getContact(recipientId);
1230 if (contact != null && !Util.isEmpty(contact.getName())) {
1231 return contact.getName();
1232 }
1233
1234 final var profile = context.getProfileHelper().getRecipientProfile(recipientId);
1235 if (profile != null) {
1236 return profile.getDisplayName();
1237 }
1238
1239 return null;
1240 }
1241
1242 @Override
1243 public Group getGroup(GroupId groupId) {
1244 return toGroup(context.getGroupHelper().getGroup(groupId));
1245 }
1246
1247 @Override
1248 public List<Identity> getIdentities() {
1249 return account.getIdentityKeyStore()
1250 .getIdentities()
1251 .stream()
1252 .map(this::toIdentity)
1253 .filter(Objects::nonNull)
1254 .toList();
1255 }
1256
1257 private Identity toIdentity(final IdentityInfo identityInfo) {
1258 if (identityInfo == null) {
1259 return null;
1260 }
1261
1262 final var address = account.getRecipientAddressResolver()
1263 .resolveRecipientAddress(account.getRecipientResolver().resolveRecipient(identityInfo.getServiceId()));
1264 if (address.serviceId().isPresent() && !Objects.equals(address.serviceId().get(),
1265 identityInfo.getServiceId())) {
1266 return null;
1267 }
1268 final var scannableFingerprint = context.getIdentityHelper()
1269 .computeSafetyNumberForScanning(identityInfo.getServiceId(), identityInfo.getIdentityKey());
1270 return new Identity(address.toApiRecipientAddress(),
1271 identityInfo.getIdentityKey(),
1272 context.getIdentityHelper()
1273 .computeSafetyNumber(identityInfo.getServiceId(), identityInfo.getIdentityKey()),
1274 scannableFingerprint == null ? null : scannableFingerprint.getSerialized(),
1275 identityInfo.getTrustLevel(),
1276 identityInfo.getDateAddedTimestamp());
1277 }
1278
1279 @Override
1280 public List<Identity> getIdentities(RecipientIdentifier.Single recipient) {
1281 ServiceId serviceId;
1282 try {
1283 final var address = account.getRecipientAddressResolver()
1284 .resolveRecipientAddress(context.getRecipientHelper().resolveRecipient(recipient));
1285 if (address.serviceId().isEmpty()) {
1286 return List.of();
1287 }
1288 serviceId = address.serviceId().get();
1289 } catch (UnregisteredRecipientException e) {
1290 return List.of();
1291 }
1292 final var identity = account.getIdentityKeyStore().getIdentityInfo(serviceId);
1293 return identity == null ? List.of() : List.of(toIdentity(identity));
1294 }
1295
1296 @Override
1297 public boolean trustIdentityVerified(
1298 RecipientIdentifier.Single recipient, IdentityVerificationCode verificationCode
1299 ) throws UnregisteredRecipientException {
1300 return switch (verificationCode) {
1301 case IdentityVerificationCode.Fingerprint fingerprint -> trustIdentity(recipient,
1302 r -> context.getIdentityHelper().trustIdentityVerified(r, fingerprint.fingerprint()));
1303 case IdentityVerificationCode.SafetyNumber safetyNumber -> trustIdentity(recipient,
1304 r -> context.getIdentityHelper().trustIdentityVerifiedSafetyNumber(r, safetyNumber.safetyNumber()));
1305 case IdentityVerificationCode.ScannableSafetyNumber safetyNumber -> trustIdentity(recipient,
1306 r -> context.getIdentityHelper().trustIdentityVerifiedSafetyNumber(r, safetyNumber.safetyNumber()));
1307 case null -> throw new AssertionError("Invalid verification code type");
1308 };
1309 }
1310
1311 @Override
1312 public boolean trustIdentityAllKeys(RecipientIdentifier.Single recipient) throws UnregisteredRecipientException {
1313 return trustIdentity(recipient, r -> context.getIdentityHelper().trustIdentityAllKeys(r));
1314 }
1315
1316 private boolean trustIdentity(
1317 RecipientIdentifier.Single recipient, Function<RecipientId, Boolean> trustMethod
1318 ) throws UnregisteredRecipientException {
1319 final var recipientId = context.getRecipientHelper().resolveRecipient(recipient);
1320 final var updated = trustMethod.apply(recipientId);
1321 if (updated && this.isReceiving()) {
1322 context.getReceiveHelper().setNeedsToRetryFailedMessages(true);
1323 }
1324 return updated;
1325 }
1326
1327 @Override
1328 public void addAddressChangedListener(final Runnable listener) {
1329 synchronized (addressChangedListeners) {
1330 addressChangedListeners.add(listener);
1331 }
1332 }
1333
1334 @Override
1335 public void addClosedListener(final Runnable listener) {
1336 synchronized (closedListeners) {
1337 closedListeners.add(listener);
1338 }
1339 }
1340
1341 @Override
1342 public InputStream retrieveAttachment(final String id) throws IOException {
1343 return context.getAttachmentHelper().retrieveAttachment(id).getStream();
1344 }
1345
1346 @Override
1347 public void close() {
1348 Thread thread;
1349 synchronized (messageHandlers) {
1350 weakHandlers.clear();
1351 messageHandlers.clear();
1352 thread = receiveThread;
1353 receiveThread = null;
1354 }
1355 if (thread != null) {
1356 stopReceiveThread(thread);
1357 }
1358 executor.close();
1359 context.close();
1360
1361 dependencies.getSignalWebSocket().disconnect();
1362 dependencies.getPushServiceSocket().close();
1363 disposable.dispose();
1364
1365 if (account != null) {
1366 account.close();
1367 }
1368
1369 synchronized (closedListeners) {
1370 closedListeners.forEach(Runnable::run);
1371 closedListeners.clear();
1372 }
1373
1374 account = null;
1375 }
1376 }