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