nmode's Git Repositories - signal-cli/blob - lib/src/main/java/org/asamk/signal/manager/ManagerImpl.java

   1 /*
   2   Copyright (C) 2015-2021 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.actions.HandleAction;
  20 import org.asamk.signal.manager.api.Configuration;
  21 import org.asamk.signal.manager.api.Device;
  22 import org.asamk.signal.manager.api.Group;
  23 import org.asamk.signal.manager.api.Identity;
  24 import org.asamk.signal.manager.api.InactiveGroupLinkException;
  25 import org.asamk.signal.manager.api.InvalidDeviceLinkException;
  26 import org.asamk.signal.manager.api.Message;
  27 import org.asamk.signal.manager.api.Pair;
  28 import org.asamk.signal.manager.api.RecipientIdentifier;
  29 import org.asamk.signal.manager.api.SendGroupMessageResults;
  30 import org.asamk.signal.manager.api.SendMessageResult;
  31 import org.asamk.signal.manager.api.SendMessageResults;
  32 import org.asamk.signal.manager.api.TypingAction;
  33 import org.asamk.signal.manager.api.UpdateGroup;
  34 import org.asamk.signal.manager.config.ServiceConfig;
  35 import org.asamk.signal.manager.config.ServiceEnvironmentConfig;
  36 import org.asamk.signal.manager.groups.GroupId;
  37 import org.asamk.signal.manager.groups.GroupInviteLinkUrl;
  38 import org.asamk.signal.manager.groups.GroupNotFoundException;
  39 import org.asamk.signal.manager.groups.GroupSendingNotAllowedException;
  40 import org.asamk.signal.manager.groups.LastGroupAdminException;
  41 import org.asamk.signal.manager.groups.NotAGroupMemberException;
  42 import org.asamk.signal.manager.helper.AttachmentHelper;
  43 import org.asamk.signal.manager.helper.ContactHelper;
  44 import org.asamk.signal.manager.helper.GroupHelper;
  45 import org.asamk.signal.manager.helper.GroupV2Helper;
  46 import org.asamk.signal.manager.helper.IdentityHelper;
  47 import org.asamk.signal.manager.helper.IncomingMessageHandler;
  48 import org.asamk.signal.manager.helper.PinHelper;
  49 import org.asamk.signal.manager.helper.PreKeyHelper;
  50 import org.asamk.signal.manager.helper.ProfileHelper;
  51 import org.asamk.signal.manager.helper.SendHelper;
  52 import org.asamk.signal.manager.helper.StorageHelper;
  53 import org.asamk.signal.manager.helper.SyncHelper;
  54 import org.asamk.signal.manager.helper.UnidentifiedAccessHelper;
  55 import org.asamk.signal.manager.jobs.Context;
  56 import org.asamk.signal.manager.storage.SignalAccount;
  57 import org.asamk.signal.manager.storage.groups.GroupInfo;
  58 import org.asamk.signal.manager.storage.identities.IdentityInfo;
  59 import org.asamk.signal.manager.storage.messageCache.CachedMessage;
  60 import org.asamk.signal.manager.storage.recipients.Contact;
  61 import org.asamk.signal.manager.storage.recipients.Profile;
  62 import org.asamk.signal.manager.storage.recipients.RecipientAddress;
  63 import org.asamk.signal.manager.storage.recipients.RecipientId;
  64 import org.asamk.signal.manager.storage.stickers.Sticker;
  65 import org.asamk.signal.manager.storage.stickers.StickerPackId;
  66 import org.asamk.signal.manager.util.KeyUtils;
  67 import org.asamk.signal.manager.util.StickerUtils;
  68 import org.slf4j.Logger;
  69 import org.slf4j.LoggerFactory;
  70 import org.whispersystems.libsignal.InvalidKeyException;
  71 import org.whispersystems.libsignal.ecc.ECPublicKey;
  72 import org.whispersystems.libsignal.util.guava.Optional;
  73 import org.whispersystems.signalservice.api.SignalSessionLock;
  74 import org.whispersystems.signalservice.api.messages.SignalServiceDataMessage;
  75 import org.whispersystems.signalservice.api.messages.SignalServiceEnvelope;
  76 import org.whispersystems.signalservice.api.messages.SignalServiceReceiptMessage;
  77 import org.whispersystems.signalservice.api.messages.SignalServiceTypingMessage;
  78 import org.whispersystems.signalservice.api.push.ACI;
  79 import org.whispersystems.signalservice.api.push.SignalServiceAddress;
  80 import org.whispersystems.signalservice.api.util.DeviceNameUtil;
  81 import org.whispersystems.signalservice.api.util.InvalidNumberException;
  82 import org.whispersystems.signalservice.api.util.PhoneNumberFormatter;
  83 import org.whispersystems.signalservice.api.websocket.WebSocketUnavailableException;
  84 import org.whispersystems.signalservice.internal.contacts.crypto.Quote;
  85 import org.whispersystems.signalservice.internal.contacts.crypto.UnauthenticatedQuoteException;
  86 import org.whispersystems.signalservice.internal.contacts.crypto.UnauthenticatedResponseException;
  87 import org.whispersystems.signalservice.internal.util.DynamicCredentialsProvider;
  88 import org.whispersystems.signalservice.internal.util.Hex;
  89 import org.whispersystems.signalservice.internal.util.Util;
  90
  91 import java.io.File;
  92 import java.io.IOException;
  93 import java.net.URI;
  94 import java.net.URISyntaxException;
  95 import java.net.URLEncoder;
  96 import java.nio.charset.StandardCharsets;
  97 import java.security.SignatureException;
  98 import java.util.Collection;
  99 import java.util.HashMap;
 100 import java.util.HashSet;
 101 import java.util.List;
 102 import java.util.Map;
 103 import java.util.Set;
 104 import java.util.UUID;
 105 import java.util.concurrent.ExecutorService;
 106 import java.util.concurrent.Executors;
 107 import java.util.concurrent.TimeUnit;
 108 import java.util.concurrent.TimeoutException;
 109 import java.util.concurrent.locks.ReentrantLock;
 110 import java.util.stream.Collectors;
 111 import java.util.stream.Stream;
 112
 113 import static org.asamk.signal.manager.config.ServiceConfig.capabilities;
 114
 115 public class ManagerImpl implements Manager {
 116
 117     private final static Logger logger = LoggerFactory.getLogger(ManagerImpl.class);
 118
 119     private final ServiceEnvironmentConfig serviceEnvironmentConfig;
 120     private final SignalDependencies dependencies;
 121
 122     private SignalAccount account;
 123
 124     private final ExecutorService executor = Executors.newCachedThreadPool();
 125
 126     private final ProfileHelper profileHelper;
 127     private final PinHelper pinHelper;
 128     private final StorageHelper storageHelper;
 129     private final SendHelper sendHelper;
 130     private final SyncHelper syncHelper;
 131     private final AttachmentHelper attachmentHelper;
 132     private final GroupHelper groupHelper;
 133     private final ContactHelper contactHelper;
 134     private final IncomingMessageHandler incomingMessageHandler;
 135     private final PreKeyHelper preKeyHelper;
 136     private final IdentityHelper identityHelper;
 137
 138     private final Context context;
 139     private boolean hasCaughtUpWithOldMessages = false;
 140     private boolean ignoreAttachments = false;
 141
 142     private Thread receiveThread;
 143     private final Set<ReceiveMessageHandler> weakHandlers = new HashSet<>();
 144     private final Set<ReceiveMessageHandler> messageHandlers = new HashSet<>();
 145     private boolean isReceivingSynchronous;
 146
 147     ManagerImpl(
 148             SignalAccount account,
 149             PathConfig pathConfig,
 150             ServiceEnvironmentConfig serviceEnvironmentConfig,
 151             String userAgent
 152     ) {
 153         this.account = account;
 154         this.serviceEnvironmentConfig = serviceEnvironmentConfig;
 155
 156         final var credentialsProvider = new DynamicCredentialsProvider(account.getAci(),
 157                 account.getUsername(),
 158                 account.getPassword(),
 159                 account.getDeviceId());
 160         final var sessionLock = new SignalSessionLock() {
 161             private final ReentrantLock LEGACY_LOCK = new ReentrantLock();
 162
 163             @Override
 164             public Lock acquire() {
 165                 LEGACY_LOCK.lock();
 166                 return LEGACY_LOCK::unlock;
 167             }
 168         };
 169         this.dependencies = new SignalDependencies(serviceEnvironmentConfig,
 170                 userAgent,
 171                 credentialsProvider,
 172                 account.getSignalProtocolStore(),
 173                 executor,
 174                 sessionLock);
 175         final var avatarStore = new AvatarStore(pathConfig.avatarsPath());
 176         final var attachmentStore = new AttachmentStore(pathConfig.attachmentsPath());
 177         final var stickerPackStore = new StickerPackStore(pathConfig.stickerPacksPath());
 178
 179         this.attachmentHelper = new AttachmentHelper(dependencies, attachmentStore);
 180         this.pinHelper = new PinHelper(dependencies.getKeyBackupService());
 181         final var unidentifiedAccessHelper = new UnidentifiedAccessHelper(account,
 182                 dependencies,
 183                 account::getProfileKey,
 184                 this::getRecipientProfile);
 185         this.profileHelper = new ProfileHelper(account,
 186                 dependencies,
 187                 avatarStore,
 188                 unidentifiedAccessHelper::getAccessFor,
 189                 this::resolveSignalServiceAddress);
 190         final GroupV2Helper groupV2Helper = new GroupV2Helper(profileHelper::getRecipientProfileKeyCredential,
 191                 this::getRecipientProfile,
 192                 account::getSelfRecipientId,
 193                 dependencies.getGroupsV2Operations(),
 194                 dependencies.getGroupsV2Api(),
 195                 this::resolveSignalServiceAddress);
 196         this.sendHelper = new SendHelper(account,
 197                 dependencies,
 198                 unidentifiedAccessHelper,
 199                 this::resolveSignalServiceAddress,
 200                 account.getRecipientStore(),
 201                 this::handleIdentityFailure,
 202                 this::getGroupInfo,
 203                 this::refreshRegisteredUser);
 204         this.groupHelper = new GroupHelper(account,
 205                 dependencies,
 206                 attachmentHelper,
 207                 sendHelper,
 208                 groupV2Helper,
 209                 avatarStore,
 210                 this::resolveSignalServiceAddress,
 211                 account.getRecipientStore());
 212         this.storageHelper = new StorageHelper(account, dependencies, groupHelper, profileHelper);
 213         this.contactHelper = new ContactHelper(account);
 214         this.syncHelper = new SyncHelper(account,
 215                 attachmentHelper,
 216                 sendHelper,
 217                 groupHelper,
 218                 avatarStore,
 219                 this::resolveSignalServiceAddress);
 220         preKeyHelper = new PreKeyHelper(account, dependencies);
 221
 222         this.context = new Context(account,
 223                 dependencies,
 224                 stickerPackStore,
 225                 sendHelper,
 226                 groupHelper,
 227                 syncHelper,
 228                 profileHelper,
 229                 storageHelper,
 230                 preKeyHelper);
 231         var jobExecutor = new JobExecutor(context);
 232
 233         this.incomingMessageHandler = new IncomingMessageHandler(account,
 234                 dependencies,
 235                 account.getRecipientStore(),
 236                 this::resolveSignalServiceAddress,
 237                 groupHelper,
 238                 contactHelper,
 239                 attachmentHelper,
 240                 syncHelper,
 241                 this::getRecipientProfile,
 242                 jobExecutor);
 243         this.identityHelper = new IdentityHelper(account,
 244                 dependencies,
 245                 this::resolveSignalServiceAddress,
 246                 syncHelper,
 247                 profileHelper);
 248     }
 249
 250     @Override
 251     public String getSelfNumber() {
 252         return account.getUsername();
 253     }
 254
 255     @Override
 256     public void checkAccountState() throws IOException {
 257         if (account.getLastReceiveTimestamp() == 0) {
 258             logger.info("The Signal protocol expects that incoming messages are regularly received.");
 259         } else {
 260             var diffInMilliseconds = System.currentTimeMillis() - account.getLastReceiveTimestamp();
 261             long days = TimeUnit.DAYS.convert(diffInMilliseconds, TimeUnit.MILLISECONDS);
 262             if (days > 7) {
 263                 logger.warn(
 264                         "Messages have been last received {} days ago. The Signal protocol expects that incoming messages are regularly received.",
 265                         days);
 266             }
 267         }
 268         preKeyHelper.refreshPreKeysIfNecessary();
 269         if (account.getAci() == null) {
 270             account.setAci(dependencies.getAccountManager().getOwnAci());
 271         }
 272         updateAccountAttributes(null);
 273     }
 274
 275     /**
 276      * This is used for checking a set of phone numbers for registration on Signal
 277      *
 278      * @param numbers The set of phone number in question
 279      * @return A map of numbers to canonicalized number and uuid. If a number is not registered the uuid is null.
 280      * @throws IOException if its unable to get the contacts to check if they're registered
 281      */
 282     @Override
 283     public Map<String, Pair<String, UUID>> areUsersRegistered(Set<String> numbers) throws IOException {
 284         Map<String, String> canonicalizedNumbers = numbers.stream().collect(Collectors.toMap(n -> n, n -> {
 285             try {
 286                 return PhoneNumberFormatter.formatNumber(n, account.getUsername());
 287             } catch (InvalidNumberException e) {
 288                 return "";
 289             }
 290         }));
 291
 292         // Note "registeredUsers" has no optionals. It only gives us info on users who are registered
 293         var registeredUsers = getRegisteredUsers(canonicalizedNumbers.values()
 294                 .stream()
 295                 .filter(s -> !s.isEmpty())
 296                 .collect(Collectors.toSet()));
 297
 298         return numbers.stream().collect(Collectors.toMap(n -> n, n -> {
 299             final var number = canonicalizedNumbers.get(n);
 300             final var aci = registeredUsers.get(number);
 301             return new Pair<>(number.isEmpty() ? null : number, aci == null ? null : aci.uuid());
 302         }));
 303     }
 304
 305     @Override
 306     public void updateAccountAttributes(String deviceName) throws IOException {
 307         final String encryptedDeviceName;
 308         if (deviceName == null) {
 309             encryptedDeviceName = account.getEncryptedDeviceName();
 310         } else {
 311             final var privateKey = account.getIdentityKeyPair().getPrivateKey();
 312             encryptedDeviceName = DeviceNameUtil.encryptDeviceName(deviceName, privateKey);
 313             account.setEncryptedDeviceName(encryptedDeviceName);
 314         }
 315         dependencies.getAccountManager()
 316                 .setAccountAttributes(encryptedDeviceName,
 317                         null,
 318                         account.getLocalRegistrationId(),
 319                         true,
 320                         null,
 321                         account.getPinMasterKey() == null ? null : account.getPinMasterKey().deriveRegistrationLock(),
 322                         account.getSelfUnidentifiedAccessKey(),
 323                         account.isUnrestrictedUnidentifiedAccess(),
 324                         capabilities,
 325                         account.isDiscoverableByPhoneNumber());
 326     }
 327
 328     @Override
 329     public Configuration getConfiguration() {
 330         final var configurationStore = account.getConfigurationStore();
 331         return new Configuration(java.util.Optional.ofNullable(configurationStore.getReadReceipts()),
 332                 java.util.Optional.ofNullable(configurationStore.getUnidentifiedDeliveryIndicators()),
 333                 java.util.Optional.ofNullable(configurationStore.getTypingIndicators()),
 334                 java.util.Optional.ofNullable(configurationStore.getLinkPreviews()));
 335     }
 336
 337     @Override
 338     public void updateConfiguration(
 339             Configuration configuration
 340     ) throws IOException, NotMasterDeviceException {
 341         if (!account.isMasterDevice()) {
 342             throw new NotMasterDeviceException();
 343         }
 344
 345         final var configurationStore = account.getConfigurationStore();
 346         if (configuration.readReceipts().isPresent()) {
 347             configurationStore.setReadReceipts(configuration.readReceipts().get());
 348         }
 349         if (configuration.unidentifiedDeliveryIndicators().isPresent()) {
 350             configurationStore.setUnidentifiedDeliveryIndicators(configuration.unidentifiedDeliveryIndicators().get());
 351         }
 352         if (configuration.typingIndicators().isPresent()) {
 353             configurationStore.setTypingIndicators(configuration.typingIndicators().get());
 354         }
 355         if (configuration.linkPreviews().isPresent()) {
 356             configurationStore.setLinkPreviews(configuration.linkPreviews().get());
 357         }
 358         syncHelper.sendConfigurationMessage();
 359     }
 360
 361     /**
 362      * @param givenName  if null, the previous givenName will be kept
 363      * @param familyName if null, the previous familyName will be kept
 364      * @param about      if null, the previous about text will be kept
 365      * @param aboutEmoji if null, the previous about emoji will be kept
 366      * @param avatar     if avatar is null the image from the local avatar store is used (if present),
 367      */
 368     @Override
 369     public void setProfile(
 370             String givenName, final String familyName, String about, String aboutEmoji, java.util.Optional<File> avatar
 371     ) throws IOException {
 372         profileHelper.setProfile(givenName,
 373                 familyName,
 374                 about,
 375                 aboutEmoji,
 376                 avatar == null ? null : Optional.fromNullable(avatar.orElse(null)));
 377         syncHelper.sendSyncFetchProfileMessage();
 378     }
 379
 380     @Override
 381     public void unregister() throws IOException {
 382         // When setting an empty GCM id, the Signal-Server also sets the fetchesMessages property to false.
 383         // If this is the master device, other users can't send messages to this number anymore.
 384         // If this is a linked device, other users can still send messages, but this device doesn't receive them anymore.
 385         dependencies.getAccountManager().setGcmId(Optional.absent());
 386
 387         account.setRegistered(false);
 388     }
 389
 390     @Override
 391     public void deleteAccount() throws IOException {
 392         try {
 393             pinHelper.removeRegistrationLockPin();
 394         } catch (IOException e) {
 395             logger.warn("Failed to remove registration lock pin");
 396         }
 397         account.setRegistrationLockPin(null, null);
 398
 399         dependencies.getAccountManager().deleteAccount();
 400
 401         account.setRegistered(false);
 402     }
 403
 404     @Override
 405     public void submitRateLimitRecaptchaChallenge(String challenge, String captcha) throws IOException {
 406         captcha = captcha == null ? null : captcha.replace("signalcaptcha://", "");
 407
 408         dependencies.getAccountManager().submitRateLimitRecaptchaChallenge(challenge, captcha);
 409     }
 410
 411     @Override
 412     public List<Device> getLinkedDevices() throws IOException {
 413         var devices = dependencies.getAccountManager().getDevices();
 414         account.setMultiDevice(devices.size() > 1);
 415         var identityKey = account.getIdentityKeyPair().getPrivateKey();
 416         return devices.stream().map(d -> {
 417             String deviceName = d.getName();
 418             if (deviceName != null) {
 419                 try {
 420                     deviceName = DeviceNameUtil.decryptDeviceName(deviceName, identityKey);
 421                 } catch (IOException e) {
 422                     logger.debug("Failed to decrypt device name, maybe plain text?", e);
 423                 }
 424             }
 425             return new Device(d.getId(),
 426                     deviceName,
 427                     d.getCreated(),
 428                     d.getLastSeen(),
 429                     d.getId() == account.getDeviceId());
 430         }).collect(Collectors.toList());
 431     }
 432
 433     @Override
 434     public void removeLinkedDevices(long deviceId) throws IOException {
 435         dependencies.getAccountManager().removeDevice(deviceId);
 436         var devices = dependencies.getAccountManager().getDevices();
 437         account.setMultiDevice(devices.size() > 1);
 438     }
 439
 440     @Override
 441     public void addDeviceLink(URI linkUri) throws IOException, InvalidDeviceLinkException {
 442         var info = DeviceLinkInfo.parseDeviceLinkUri(linkUri);
 443
 444         addDevice(info.deviceIdentifier(), info.deviceKey());
 445     }
 446
 447     private void addDevice(
 448             String deviceIdentifier, ECPublicKey deviceKey
 449     ) throws IOException, InvalidDeviceLinkException {
 450         var identityKeyPair = account.getIdentityKeyPair();
 451         var verificationCode = dependencies.getAccountManager().getNewDeviceVerificationCode();
 452
 453         try {
 454             dependencies.getAccountManager()
 455                     .addDevice(deviceIdentifier,
 456                             deviceKey,
 457                             identityKeyPair,
 458                             Optional.of(account.getProfileKey().serialize()),
 459                             verificationCode);
 460         } catch (InvalidKeyException e) {
 461             throw new InvalidDeviceLinkException("Invalid device link", e);
 462         }
 463         account.setMultiDevice(true);
 464     }
 465
 466     @Override
 467     public void setRegistrationLockPin(java.util.Optional<String> pin) throws IOException {
 468         if (!account.isMasterDevice()) {
 469             throw new RuntimeException("Only master device can set a PIN");
 470         }
 471         if (pin.isPresent()) {
 472             final var masterKey = account.getPinMasterKey() != null
 473                     ? account.getPinMasterKey()
 474                     : KeyUtils.createMasterKey();
 475
 476             pinHelper.setRegistrationLockPin(pin.get(), masterKey);
 477
 478             account.setRegistrationLockPin(pin.get(), masterKey);
 479         } else {
 480             // Remove KBS Pin
 481             pinHelper.removeRegistrationLockPin();
 482
 483             account.setRegistrationLockPin(null, null);
 484         }
 485     }
 486
 487     void refreshPreKeys() throws IOException {
 488         preKeyHelper.refreshPreKeys();
 489     }
 490
 491     @Override
 492     public Profile getRecipientProfile(RecipientIdentifier.Single recipient) throws IOException {
 493         return profileHelper.getRecipientProfile(resolveRecipient(recipient));
 494     }
 495
 496     private Profile getRecipientProfile(RecipientId recipientId) {
 497         return profileHelper.getRecipientProfile(recipientId);
 498     }
 499
 500     @Override
 501     public List<Group> getGroups() {
 502         return account.getGroupStore().getGroups().stream().map(this::toGroup).collect(Collectors.toList());
 503     }
 504
 505     private Group toGroup(final GroupInfo groupInfo) {
 506         if (groupInfo == null) {
 507             return null;
 508         }
 509
 510         return new Group(groupInfo.getGroupId(),
 511                 groupInfo.getTitle(),
 512                 groupInfo.getDescription(),
 513                 groupInfo.getGroupInviteLink(),
 514                 groupInfo.getMembers()
 515                         .stream()
 516                         .map(account.getRecipientStore()::resolveRecipientAddress)
 517                         .collect(Collectors.toSet()),
 518                 groupInfo.getPendingMembers()
 519                         .stream()
 520                         .map(account.getRecipientStore()::resolveRecipientAddress)
 521                         .collect(Collectors.toSet()),
 522                 groupInfo.getRequestingMembers()
 523                         .stream()
 524                         .map(account.getRecipientStore()::resolveRecipientAddress)
 525                         .collect(Collectors.toSet()),
 526                 groupInfo.getAdminMembers()
 527                         .stream()
 528                         .map(account.getRecipientStore()::resolveRecipientAddress)
 529                         .collect(Collectors.toSet()),
 530                 groupInfo.isBlocked(),
 531                 groupInfo.getMessageExpirationTimer(),
 532                 groupInfo.getPermissionAddMember(),
 533                 groupInfo.getPermissionEditDetails(),
 534                 groupInfo.getPermissionSendMessage(),
 535                 groupInfo.isMember(account.getSelfRecipientId()),
 536                 groupInfo.isAdmin(account.getSelfRecipientId()));
 537     }
 538
 539     @Override
 540     public SendGroupMessageResults quitGroup(
 541             GroupId groupId, Set<RecipientIdentifier.Single> groupAdmins
 542     ) throws GroupNotFoundException, IOException, NotAGroupMemberException, LastGroupAdminException {
 543         final var newAdmins = resolveRecipients(groupAdmins);
 544         return groupHelper.quitGroup(groupId, newAdmins);
 545     }
 546
 547     @Override
 548     public void deleteGroup(GroupId groupId) throws IOException {
 549         groupHelper.deleteGroup(groupId);
 550     }
 551
 552     @Override
 553     public Pair<GroupId, SendGroupMessageResults> createGroup(
 554             String name, Set<RecipientIdentifier.Single> members, File avatarFile
 555     ) throws IOException, AttachmentInvalidException {
 556         return groupHelper.createGroup(name, members == null ? null : resolveRecipients(members), avatarFile);
 557     }
 558
 559     @Override
 560     public SendGroupMessageResults updateGroup(
 561             final GroupId groupId, final UpdateGroup updateGroup
 562     ) throws IOException, GroupNotFoundException, AttachmentInvalidException, NotAGroupMemberException, GroupSendingNotAllowedException {
 563         return groupHelper.updateGroup(groupId,
 564                 updateGroup.getName(),
 565                 updateGroup.getDescription(),
 566                 updateGroup.getMembers() == null ? null : resolveRecipients(updateGroup.getMembers()),
 567                 updateGroup.getRemoveMembers() == null ? null : resolveRecipients(updateGroup.getRemoveMembers()),
 568                 updateGroup.getAdmins() == null ? null : resolveRecipients(updateGroup.getAdmins()),
 569                 updateGroup.getRemoveAdmins() == null ? null : resolveRecipients(updateGroup.getRemoveAdmins()),
 570                 updateGroup.isResetGroupLink(),
 571                 updateGroup.getGroupLinkState(),
 572                 updateGroup.getAddMemberPermission(),
 573                 updateGroup.getEditDetailsPermission(),
 574                 updateGroup.getAvatarFile(),
 575                 updateGroup.getExpirationTimer(),
 576                 updateGroup.getIsAnnouncementGroup());
 577     }
 578
 579     @Override
 580     public Pair<GroupId, SendGroupMessageResults> joinGroup(
 581             GroupInviteLinkUrl inviteLinkUrl
 582     ) throws IOException, InactiveGroupLinkException {
 583         return groupHelper.joinGroup(inviteLinkUrl);
 584     }
 585
 586     private SendMessageResults sendMessage(
 587             SignalServiceDataMessage.Builder messageBuilder, Set<RecipientIdentifier> recipients
 588     ) throws IOException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
 589         var results = new HashMap<RecipientIdentifier, List<SendMessageResult>>();
 590         long timestamp = System.currentTimeMillis();
 591         messageBuilder.withTimestamp(timestamp);
 592         for (final var recipient : recipients) {
 593             if (recipient instanceof RecipientIdentifier.Single single) {
 594                 final var recipientId = resolveRecipient(single);
 595                 final var result = sendHelper.sendMessage(messageBuilder, recipientId);
 596                 results.put(recipient,
 597                         List.of(SendMessageResult.from(result,
 598                                 account.getRecipientStore(),
 599                                 account.getRecipientStore()::resolveRecipientAddress)));
 600             } else if (recipient instanceof RecipientIdentifier.NoteToSelf) {
 601                 final var result = sendHelper.sendSelfMessage(messageBuilder);
 602                 results.put(recipient,
 603                         List.of(SendMessageResult.from(result,
 604                                 account.getRecipientStore(),
 605                                 account.getRecipientStore()::resolveRecipientAddress)));
 606             } else if (recipient instanceof RecipientIdentifier.Group group) {
 607                 final var result = sendHelper.sendAsGroupMessage(messageBuilder, group.groupId());
 608                 results.put(recipient,
 609                         result.stream()
 610                                 .map(sendMessageResult -> SendMessageResult.from(sendMessageResult,
 611                                         account.getRecipientStore(),
 612                                         account.getRecipientStore()::resolveRecipientAddress))
 613                                 .collect(Collectors.toList()));
 614             }
 615         }
 616         return new SendMessageResults(timestamp, results);
 617     }
 618
 619     private void sendTypingMessage(
 620             SignalServiceTypingMessage.Action action, Set<RecipientIdentifier> recipients
 621     ) throws IOException, UntrustedIdentityException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
 622         final var timestamp = System.currentTimeMillis();
 623         for (var recipient : recipients) {
 624             if (recipient instanceof RecipientIdentifier.Single) {
 625                 final var message = new SignalServiceTypingMessage(action, timestamp, Optional.absent());
 626                 final var recipientId = resolveRecipient((RecipientIdentifier.Single) recipient);
 627                 sendHelper.sendTypingMessage(message, recipientId);
 628             } else if (recipient instanceof RecipientIdentifier.Group) {
 629                 final var groupId = ((RecipientIdentifier.Group) recipient).groupId();
 630                 final var message = new SignalServiceTypingMessage(action, timestamp, Optional.of(groupId.serialize()));
 631                 sendHelper.sendGroupTypingMessage(message, groupId);
 632             }
 633         }
 634     }
 635
 636     @Override
 637     public void sendTypingMessage(
 638             TypingAction action, Set<RecipientIdentifier> recipients
 639     ) throws IOException, UntrustedIdentityException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
 640         sendTypingMessage(action.toSignalService(), recipients);
 641     }
 642
 643     @Override
 644     public void sendReadReceipt(
 645             RecipientIdentifier.Single sender, List<Long> messageIds
 646     ) throws IOException, UntrustedIdentityException {
 647         var receiptMessage = new SignalServiceReceiptMessage(SignalServiceReceiptMessage.Type.READ,
 648                 messageIds,
 649                 System.currentTimeMillis());
 650
 651         sendHelper.sendReceiptMessage(receiptMessage, resolveRecipient(sender));
 652     }
 653
 654     @Override
 655     public void sendViewedReceipt(
 656             RecipientIdentifier.Single sender, List<Long> messageIds
 657     ) throws IOException, UntrustedIdentityException {
 658         var receiptMessage = new SignalServiceReceiptMessage(SignalServiceReceiptMessage.Type.VIEWED,
 659                 messageIds,
 660                 System.currentTimeMillis());
 661
 662         sendHelper.sendReceiptMessage(receiptMessage, resolveRecipient(sender));
 663     }
 664
 665     @Override
 666     public SendMessageResults sendMessage(
 667             Message message, Set<RecipientIdentifier> recipients
 668     ) throws IOException, AttachmentInvalidException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
 669         final var messageBuilder = SignalServiceDataMessage.newBuilder();
 670         applyMessage(messageBuilder, message);
 671         return sendMessage(messageBuilder, recipients);
 672     }
 673
 674     private void applyMessage(
 675             final SignalServiceDataMessage.Builder messageBuilder, final Message message
 676     ) throws AttachmentInvalidException, IOException {
 677         messageBuilder.withBody(message.messageText());
 678         final var attachments = message.attachments();
 679         if (attachments != null) {
 680             messageBuilder.withAttachments(attachmentHelper.uploadAttachments(attachments));
 681         }
 682     }
 683
 684     @Override
 685     public SendMessageResults sendRemoteDeleteMessage(
 686             long targetSentTimestamp, Set<RecipientIdentifier> recipients
 687     ) throws IOException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
 688         var delete = new SignalServiceDataMessage.RemoteDelete(targetSentTimestamp);
 689         final var messageBuilder = SignalServiceDataMessage.newBuilder().withRemoteDelete(delete);
 690         return sendMessage(messageBuilder, recipients);
 691     }
 692
 693     @Override
 694     public SendMessageResults sendMessageReaction(
 695             String emoji,
 696             boolean remove,
 697             RecipientIdentifier.Single targetAuthor,
 698             long targetSentTimestamp,
 699             Set<RecipientIdentifier> recipients
 700     ) throws IOException, NotAGroupMemberException, GroupNotFoundException, GroupSendingNotAllowedException {
 701         var targetAuthorRecipientId = resolveRecipient(targetAuthor);
 702         var reaction = new SignalServiceDataMessage.Reaction(emoji,
 703                 remove,
 704                 resolveSignalServiceAddress(targetAuthorRecipientId),
 705                 targetSentTimestamp);
 706         final var messageBuilder = SignalServiceDataMessage.newBuilder().withReaction(reaction);
 707         return sendMessage(messageBuilder, recipients);
 708     }
 709
 710     @Override
 711     public SendMessageResults sendEndSessionMessage(Set<RecipientIdentifier.Single> recipients) throws IOException {
 712         var messageBuilder = SignalServiceDataMessage.newBuilder().asEndSessionMessage();
 713
 714         try {
 715             return sendMessage(messageBuilder,
 716                     recipients.stream().map(RecipientIdentifier.class::cast).collect(Collectors.toSet()));
 717         } catch (GroupNotFoundException | NotAGroupMemberException | GroupSendingNotAllowedException e) {
 718             throw new AssertionError(e);
 719         } finally {
 720             for (var recipient : recipients) {
 721                 final var recipientId = resolveRecipient(recipient);
 722                 account.getSessionStore().deleteAllSessions(recipientId);
 723             }
 724         }
 725     }
 726
 727     @Override
 728     public void setContactName(
 729             RecipientIdentifier.Single recipient, String name
 730     ) throws NotMasterDeviceException, IOException {
 731         if (!account.isMasterDevice()) {
 732             throw new NotMasterDeviceException();
 733         }
 734         contactHelper.setContactName(resolveRecipient(recipient), name);
 735     }
 736
 737     @Override
 738     public void setContactBlocked(
 739             RecipientIdentifier.Single recipient, boolean blocked
 740     ) throws NotMasterDeviceException, IOException {
 741         if (!account.isMasterDevice()) {
 742             throw new NotMasterDeviceException();
 743         }
 744         contactHelper.setContactBlocked(resolveRecipient(recipient), blocked);
 745         // TODO cycle our profile key
 746         syncHelper.sendBlockedList();
 747     }
 748
 749     @Override
 750     public void setGroupBlocked(
 751             final GroupId groupId, final boolean blocked
 752     ) throws GroupNotFoundException, IOException, NotMasterDeviceException {
 753         if (!account.isMasterDevice()) {
 754             throw new NotMasterDeviceException();
 755         }
 756         groupHelper.setGroupBlocked(groupId, blocked);
 757         // TODO cycle our profile key
 758         syncHelper.sendBlockedList();
 759     }
 760
 761     /**
 762      * Change the expiration timer for a contact
 763      */
 764     @Override
 765     public void setExpirationTimer(
 766             RecipientIdentifier.Single recipient, int messageExpirationTimer
 767     ) throws IOException {
 768         var recipientId = resolveRecipient(recipient);
 769         contactHelper.setExpirationTimer(recipientId, messageExpirationTimer);
 770         final var messageBuilder = SignalServiceDataMessage.newBuilder().asExpirationUpdate();
 771         try {
 772             sendMessage(messageBuilder, Set.of(recipient));
 773         } catch (NotAGroupMemberException | GroupNotFoundException | GroupSendingNotAllowedException e) {
 774             throw new AssertionError(e);
 775         }
 776     }
 777
 778     /**
 779      * Upload the sticker pack from path.
 780      *
 781      * @param path Path can be a path to a manifest.json file or to a zip file that contains a manifest.json file
 782      * @return if successful, returns the URL to install the sticker pack in the signal app
 783      */
 784     @Override
 785     public URI uploadStickerPack(File path) throws IOException, StickerPackInvalidException {
 786         var manifest = StickerUtils.getSignalServiceStickerManifestUpload(path);
 787
 788         var messageSender = dependencies.getMessageSender();
 789
 790         var packKey = KeyUtils.createStickerUploadKey();
 791         var packIdString = messageSender.uploadStickerManifest(manifest, packKey);
 792         var packId = StickerPackId.deserialize(Hex.fromStringCondensed(packIdString));
 793
 794         var sticker = new Sticker(packId, packKey);
 795         account.getStickerStore().updateSticker(sticker);
 796
 797         try {
 798             return new URI("https",
 799                     "signal.art",
 800                     "/addstickers/",
 801                     "pack_id="
 802                             + URLEncoder.encode(Hex.toStringCondensed(packId.serialize()), StandardCharsets.UTF_8)
 803                             + "&pack_key="
 804                             + URLEncoder.encode(Hex.toStringCondensed(packKey), StandardCharsets.UTF_8));
 805         } catch (URISyntaxException e) {
 806             throw new AssertionError(e);
 807         }
 808     }
 809
 810     @Override
 811     public void requestAllSyncData() throws IOException {
 812         syncHelper.requestAllSyncData();
 813         retrieveRemoteStorage();
 814     }
 815
 816     void retrieveRemoteStorage() throws IOException {
 817         if (account.getStorageKey() != null) {
 818             storageHelper.readDataFromStorage();
 819         }
 820     }
 821
 822     private RecipientId refreshRegisteredUser(RecipientId recipientId) throws IOException {
 823         final var address = resolveSignalServiceAddress(recipientId);
 824         if (!address.getNumber().isPresent()) {
 825             return recipientId;
 826         }
 827         final var number = address.getNumber().get();
 828         final var uuid = getRegisteredUser(number);
 829         return resolveRecipientTrusted(new SignalServiceAddress(uuid, number));
 830     }
 831
 832     private ACI getRegisteredUser(final String number) throws IOException {
 833         final Map<String, ACI> aciMap;
 834         try {
 835             aciMap = getRegisteredUsers(Set.of(number));
 836         } catch (NumberFormatException e) {
 837             throw new IOException(number, e);
 838         }
 839         final var uuid = aciMap.get(number);
 840         if (uuid == null) {
 841             throw new IOException(number, null);
 842         }
 843         return uuid;
 844     }
 845
 846     private Map<String, ACI> getRegisteredUsers(final Set<String> numbers) throws IOException {
 847         final Map<String, ACI> registeredUsers;
 848         try {
 849             registeredUsers = dependencies.getAccountManager()
 850                     .getRegisteredUsers(ServiceConfig.getIasKeyStore(),
 851                             numbers,
 852                             serviceEnvironmentConfig.getCdsMrenclave());
 853         } catch (Quote.InvalidQuoteFormatException | UnauthenticatedQuoteException | SignatureException | UnauthenticatedResponseException | InvalidKeyException e) {
 854             throw new IOException(e);
 855         }
 856
 857         // Store numbers as recipients, so we have the number/uuid association
 858         registeredUsers.forEach((number, aci) -> resolveRecipientTrusted(new SignalServiceAddress(aci, number)));
 859
 860         return registeredUsers;
 861     }
 862
 863     private void retryFailedReceivedMessages(ReceiveMessageHandler handler) {
 864         Set<HandleAction> queuedActions = new HashSet<>();
 865         for (var cachedMessage : account.getMessageCache().getCachedMessages()) {
 866             var actions = retryFailedReceivedMessage(handler, cachedMessage);
 867             if (actions != null) {
 868                 queuedActions.addAll(actions);
 869             }
 870         }
 871         handleQueuedActions(queuedActions);
 872     }
 873
 874     private List<HandleAction> retryFailedReceivedMessage(
 875             final ReceiveMessageHandler handler, final CachedMessage cachedMessage
 876     ) {
 877         var envelope = cachedMessage.loadEnvelope();
 878         if (envelope == null) {
 879             cachedMessage.delete();
 880             return null;
 881         }
 882
 883         final var result = incomingMessageHandler.handleRetryEnvelope(envelope, ignoreAttachments, handler);
 884         final var actions = result.first();
 885         final var exception = result.second();
 886
 887         if (exception instanceof UntrustedIdentityException) {
 888             if (System.currentTimeMillis() - envelope.getServerDeliveredTimestamp() > 1000L * 60 * 60 * 24 * 30) {
 889                 // Envelope is more than a month old, cleaning up.
 890                 cachedMessage.delete();
 891                 return null;
 892             }
 893             if (!envelope.hasSourceUuid()) {
 894                 final var identifier = ((UntrustedIdentityException) exception).getSender();
 895                 final var recipientId = account.getRecipientStore().resolveRecipient(identifier);
 896                 try {
 897                     account.getMessageCache().replaceSender(cachedMessage, recipientId);
 898                 } catch (IOException ioException) {
 899                     logger.warn("Failed to move cached message to recipient folder: {}", ioException.getMessage());
 900                 }
 901             }
 902             return null;
 903         }
 904
 905         // If successful and for all other errors that are not recoverable, delete the cached message
 906         cachedMessage.delete();
 907         return actions;
 908     }
 909
 910     @Override
 911     public void addReceiveHandler(final ReceiveMessageHandler handler, final boolean isWeakListener) {
 912         if (isReceivingSynchronous) {
 913             throw new IllegalStateException("Already receiving message synchronously.");
 914         }
 915         synchronized (messageHandlers) {
 916             if (isWeakListener) {
 917                 weakHandlers.add(handler);
 918             } else {
 919                 messageHandlers.add(handler);
 920                 startReceiveThreadIfRequired();
 921             }
 922         }
 923     }
 924
 925     private void startReceiveThreadIfRequired() {
 926         if (receiveThread != null) {
 927             return;
 928         }
 929         receiveThread = new Thread(() -> {
 930             logger.debug("Starting receiving messages");
 931             while (!Thread.interrupted()) {
 932                 try {
 933                     receiveMessagesInternal(1L, TimeUnit.HOURS, false, (envelope, e) -> {
 934                         synchronized (messageHandlers) {
 935                             Stream.concat(messageHandlers.stream(), weakHandlers.stream()).forEach(h -> {
 936                                 try {
 937                                     h.handleMessage(envelope, e);
 938                                 } catch (Exception ex) {
 939                                     logger.warn("Message handler failed, ignoring", ex);
 940                                 }
 941                             });
 942                         }
 943                     });
 944                     break;
 945                 } catch (IOException e) {
 946                     logger.warn("Receiving messages failed, retrying", e);
 947                 }
 948             }
 949             logger.debug("Finished receiving messages");
 950             hasCaughtUpWithOldMessages = false;
 951             synchronized (messageHandlers) {
 952                 receiveThread = null;
 953
 954                 // Check if in the meantime another handler has been registered
 955                 if (!messageHandlers.isEmpty()) {
 956                     logger.debug("Another handler has been registered, starting receive thread again");
 957                     startReceiveThreadIfRequired();
 958                 }
 959             }
 960         });
 961
 962         receiveThread.start();
 963     }
 964
 965     @Override
 966     public void removeReceiveHandler(final ReceiveMessageHandler handler) {
 967         final Thread thread;
 968         synchronized (messageHandlers) {
 969             weakHandlers.remove(handler);
 970             messageHandlers.remove(handler);
 971             if (!messageHandlers.isEmpty() || receiveThread == null || isReceivingSynchronous) {
 972                 return;
 973             }
 974             thread = receiveThread;
 975             receiveThread = null;
 976         }
 977
 978         stopReceiveThread(thread);
 979     }
 980
 981     private void stopReceiveThread(final Thread thread) {
 982         thread.interrupt();
 983         try {
 984             thread.join();
 985         } catch (InterruptedException ignored) {
 986         }
 987     }
 988
 989     @Override
 990     public boolean isReceiving() {
 991         if (isReceivingSynchronous) {
 992             return true;
 993         }
 994         synchronized (messageHandlers) {
 995             return messageHandlers.size() > 0;
 996         }
 997     }
 998
 999     @Override
1000     public void receiveMessages(long timeout, TimeUnit unit, ReceiveMessageHandler handler) throws IOException {
1001         receiveMessages(timeout, unit, true, handler);
1002     }
1003
1004     @Override
1005     public void receiveMessages(ReceiveMessageHandler handler) throws IOException {
1006         receiveMessages(1L, TimeUnit.HOURS, false, handler);
1007     }
1008
1009     private void receiveMessages(
1010             long timeout, TimeUnit unit, boolean returnOnTimeout, ReceiveMessageHandler handler
1011     ) throws IOException {
1012         if (isReceiving()) {
1013             throw new IllegalStateException("Already receiving message.");
1014         }
1015         isReceivingSynchronous = true;
1016         receiveThread = Thread.currentThread();
1017         try {
1018             receiveMessagesInternal(timeout, unit, returnOnTimeout, handler);
1019         } finally {
1020             receiveThread = null;
1021             hasCaughtUpWithOldMessages = false;
1022             isReceivingSynchronous = false;
1023         }
1024     }
1025
1026     private void receiveMessagesInternal(
1027             long timeout, TimeUnit unit, boolean returnOnTimeout, ReceiveMessageHandler handler
1028     ) throws IOException {
1029         retryFailedReceivedMessages(handler);
1030
1031         // Use a Map here because java Set doesn't have a get method ...
1032         Map<HandleAction, HandleAction> queuedActions = new HashMap<>();
1033
1034         final var signalWebSocket = dependencies.getSignalWebSocket();
1035         signalWebSocket.connect();
1036
1037         hasCaughtUpWithOldMessages = false;
1038         var backOffCounter = 0;
1039         final var MAX_BACKOFF_COUNTER = 9;
1040
1041         while (!Thread.interrupted()) {
1042             SignalServiceEnvelope envelope;
1043             final CachedMessage[] cachedMessage = {null};
1044             account.setLastReceiveTimestamp(System.currentTimeMillis());
1045             logger.debug("Checking for new message from server");
1046             try {
1047                 var result = signalWebSocket.readOrEmpty(unit.toMillis(timeout), envelope1 -> {
1048                     final var recipientId = envelope1.hasSourceUuid()
1049                             ? resolveRecipient(envelope1.getSourceAddress())
1050                             : null;
1051                     // store message on disk, before acknowledging receipt to the server
1052                     cachedMessage[0] = account.getMessageCache().cacheMessage(envelope1, recipientId);
1053                 });
1054                 backOffCounter = 0;
1055
1056                 if (result.isPresent()) {
1057                     envelope = result.get();
1058                     logger.debug("New message received from server");
1059                 } else {
1060                     logger.debug("Received indicator that server queue is empty");
1061                     handleQueuedActions(queuedActions.keySet());
1062                     queuedActions.clear();
1063
1064                     hasCaughtUpWithOldMessages = true;
1065                     synchronized (this) {
1066                         this.notifyAll();
1067                     }
1068
1069                     // Continue to wait another timeout for new messages
1070                     continue;
1071                 }
1072             } catch (AssertionError e) {
1073                 if (e.getCause() instanceof InterruptedException) {
1074                     Thread.currentThread().interrupt();
1075                     break;
1076                 } else {
1077                     throw e;
1078                 }
1079             } catch (IOException e) {
1080                 logger.debug("Pipe unexpectedly unavailable: {}", e.getMessage());
1081                 if (e instanceof WebSocketUnavailableException || "Connection closed!".equals(e.getMessage())) {
1082                     final var sleepMilliseconds = 100 * (long) Math.pow(2, backOffCounter);
1083                     backOffCounter = Math.min(backOffCounter + 1, MAX_BACKOFF_COUNTER);
1084                     logger.warn("Connection closed unexpectedly, reconnecting in {} ms", sleepMilliseconds);
1085                     try {
1086                         Thread.sleep(sleepMilliseconds);
1087                     } catch (InterruptedException interruptedException) {
1088                         return;
1089                     }
1090                     hasCaughtUpWithOldMessages = false;
1091                     signalWebSocket.connect();
1092                     continue;
1093                 }
1094                 throw e;
1095             } catch (TimeoutException e) {
1096                 backOffCounter = 0;
1097                 if (returnOnTimeout) return;
1098                 continue;
1099             }
1100
1101             final var result = incomingMessageHandler.handleEnvelope(envelope, ignoreAttachments, handler);
1102             for (final var h : result.first()) {
1103                 final var existingAction = queuedActions.get(h);
1104                 if (existingAction == null) {
1105                     queuedActions.put(h, h);
1106                 } else {
1107                     existingAction.mergeOther(h);
1108                 }
1109             }
1110             final var exception = result.second();
1111
1112             if (hasCaughtUpWithOldMessages) {
1113                 handleQueuedActions(queuedActions.keySet());
1114                 queuedActions.clear();
1115             }
1116             if (cachedMessage[0] != null) {
1117                 if (exception instanceof UntrustedIdentityException) {
1118                     logger.debug("Keeping message with untrusted identity in message cache");
1119                     final var address = ((UntrustedIdentityException) exception).getSender();
1120                     final var recipientId = resolveRecipient(address);
1121                     if (!envelope.hasSourceUuid()) {
1122                         try {
1123                             cachedMessage[0] = account.getMessageCache().replaceSender(cachedMessage[0], recipientId);
1124                         } catch (IOException ioException) {
1125                             logger.warn("Failed to move cached message to recipient folder: {}",
1126                                     ioException.getMessage());
1127                         }
1128                     }
1129                 } else {
1130                     cachedMessage[0].delete();
1131                 }
1132             }
1133         }
1134         handleQueuedActions(queuedActions.keySet());
1135         queuedActions.clear();
1136         dependencies.getSignalWebSocket().disconnect();
1137     }
1138
1139     @Override
1140     public void setIgnoreAttachments(final boolean ignoreAttachments) {
1141         this.ignoreAttachments = ignoreAttachments;
1142     }
1143
1144     @Override
1145     public boolean hasCaughtUpWithOldMessages() {
1146         return hasCaughtUpWithOldMessages;
1147     }
1148
1149     private void handleQueuedActions(final Collection<HandleAction> queuedActions) {
1150         logger.debug("Handling message actions");
1151         var interrupted = false;
1152         for (var action : queuedActions) {
1153             try {
1154                 action.execute(context);
1155             } catch (Throwable e) {
1156                 if ((e instanceof AssertionError || e instanceof RuntimeException)
1157                         && e.getCause() instanceof InterruptedException) {
1158                     interrupted = true;
1159                     continue;
1160                 }
1161                 logger.warn("Message action failed.", e);
1162             }
1163         }
1164         if (interrupted) {
1165             Thread.currentThread().interrupt();
1166         }
1167     }
1168
1169     @Override
1170     public boolean isContactBlocked(final RecipientIdentifier.Single recipient) {
1171         final RecipientId recipientId;
1172         try {
1173             recipientId = resolveRecipient(recipient);
1174         } catch (IOException e) {
1175             return false;
1176         }
1177         return contactHelper.isContactBlocked(recipientId);
1178     }
1179
1180     @Override
1181     public void sendContacts() throws IOException {
1182         syncHelper.sendContacts();
1183     }
1184
1185     @Override
1186     public List<Pair<RecipientAddress, Contact>> getContacts() {
1187         return account.getContactStore()
1188                 .getContacts()
1189                 .stream()
1190                 .map(p -> new Pair<>(account.getRecipientStore().resolveRecipientAddress(p.first()), p.second()))
1191                 .collect(Collectors.toList());
1192     }
1193
1194     @Override
1195     public String getContactOrProfileName(RecipientIdentifier.Single recipient) {
1196         final RecipientId recipientId;
1197         try {
1198             recipientId = resolveRecipient(recipient);
1199         } catch (IOException e) {
1200             return null;
1201         }
1202
1203         final var contact = account.getContactStore().getContact(recipientId);
1204         if (contact != null && !Util.isEmpty(contact.getName())) {
1205             return contact.getName();
1206         }
1207
1208         final var profile = getRecipientProfile(recipientId);
1209         if (profile != null) {
1210             return profile.getDisplayName();
1211         }
1212
1213         return null;
1214     }
1215
1216     @Override
1217     public Group getGroup(GroupId groupId) {
1218         return toGroup(groupHelper.getGroup(groupId));
1219     }
1220
1221     private GroupInfo getGroupInfo(GroupId groupId) {
1222         return groupHelper.getGroup(groupId);
1223     }
1224
1225     @Override
1226     public List<Identity> getIdentities() {
1227         return account.getIdentityKeyStore()
1228                 .getIdentities()
1229                 .stream()
1230                 .map(this::toIdentity)
1231                 .collect(Collectors.toList());
1232     }
1233
1234     private Identity toIdentity(final IdentityInfo identityInfo) {
1235         if (identityInfo == null) {
1236             return null;
1237         }
1238
1239         final var address = account.getRecipientStore().resolveRecipientAddress(identityInfo.getRecipientId());
1240         final var scannableFingerprint = identityHelper.computeSafetyNumberForScanning(identityInfo.getRecipientId(),
1241                 identityInfo.getIdentityKey());
1242         return new Identity(address,
1243                 identityInfo.getIdentityKey(),
1244                 identityHelper.computeSafetyNumber(identityInfo.getRecipientId(), identityInfo.getIdentityKey()),
1245                 scannableFingerprint == null ? null : scannableFingerprint.getSerialized(),
1246                 identityInfo.getTrustLevel(),
1247                 identityInfo.getDateAdded());
1248     }
1249
1250     @Override
1251     public List<Identity> getIdentities(RecipientIdentifier.Single recipient) {
1252         IdentityInfo identity;
1253         try {
1254             identity = account.getIdentityKeyStore().getIdentity(resolveRecipient(recipient));
1255         } catch (IOException e) {
1256             identity = null;
1257         }
1258         return identity == null ? List.of() : List.of(toIdentity(identity));
1259     }
1260
1261     /**
1262      * Trust this the identity with this fingerprint
1263      *
1264      * @param recipient   username of the identity
1265      * @param fingerprint Fingerprint
1266      */
1267     @Override
1268     public boolean trustIdentityVerified(RecipientIdentifier.Single recipient, byte[] fingerprint) {
1269         RecipientId recipientId;
1270         try {
1271             recipientId = resolveRecipient(recipient);
1272         } catch (IOException e) {
1273             return false;
1274         }
1275         return identityHelper.trustIdentityVerified(recipientId, fingerprint);
1276     }
1277
1278     /**
1279      * Trust this the identity with this safety number
1280      *
1281      * @param recipient    username of the identity
1282      * @param safetyNumber Safety number
1283      */
1284     @Override
1285     public boolean trustIdentityVerifiedSafetyNumber(RecipientIdentifier.Single recipient, String safetyNumber) {
1286         RecipientId recipientId;
1287         try {
1288             recipientId = resolveRecipient(recipient);
1289         } catch (IOException e) {
1290             return false;
1291         }
1292         return identityHelper.trustIdentityVerifiedSafetyNumber(recipientId, safetyNumber);
1293     }
1294
1295     /**
1296      * Trust this the identity with this scannable safety number
1297      *
1298      * @param recipient    username of the identity
1299      * @param safetyNumber Scannable safety number
1300      */
1301     @Override
1302     public boolean trustIdentityVerifiedSafetyNumber(RecipientIdentifier.Single recipient, byte[] safetyNumber) {
1303         RecipientId recipientId;
1304         try {
1305             recipientId = resolveRecipient(recipient);
1306         } catch (IOException e) {
1307             return false;
1308         }
1309         return identityHelper.trustIdentityVerifiedSafetyNumber(recipientId, safetyNumber);
1310     }
1311
1312     /**
1313      * Trust all keys of this identity without verification
1314      *
1315      * @param recipient username of the identity
1316      */
1317     @Override
1318     public boolean trustIdentityAllKeys(RecipientIdentifier.Single recipient) {
1319         RecipientId recipientId;
1320         try {
1321             recipientId = resolveRecipient(recipient);
1322         } catch (IOException e) {
1323             return false;
1324         }
1325         return identityHelper.trustIdentityAllKeys(recipientId);
1326     }
1327
1328     private void handleIdentityFailure(
1329             final RecipientId recipientId,
1330             final org.whispersystems.signalservice.api.messages.SendMessageResult.IdentityFailure identityFailure
1331     ) {
1332         this.identityHelper.handleIdentityFailure(recipientId, identityFailure);
1333     }
1334
1335     private SignalServiceAddress resolveSignalServiceAddress(RecipientId recipientId) {
1336         final var address = account.getRecipientStore().resolveRecipientAddress(recipientId);
1337         if (address.getUuid().isPresent()) {
1338             return address.toSignalServiceAddress();
1339         }
1340
1341         // Address in recipient store doesn't have a uuid, this shouldn't happen
1342         // Try to retrieve the uuid from the server
1343         final var number = address.getNumber().get();
1344         final ACI aci;
1345         try {
1346             aci = getRegisteredUser(number);
1347         } catch (IOException e) {
1348             logger.warn("Failed to get uuid for e164 number: {}", number, e);
1349             // Return SignalServiceAddress with unknown UUID
1350             return address.toSignalServiceAddress();
1351         }
1352         return resolveSignalServiceAddress(account.getRecipientStore().resolveRecipient(aci));
1353     }
1354
1355     private Set<RecipientId> resolveRecipients(Collection<RecipientIdentifier.Single> recipients) throws IOException {
1356         final var recipientIds = new HashSet<RecipientId>(recipients.size());
1357         for (var number : recipients) {
1358             final var recipientId = resolveRecipient(number);
1359             recipientIds.add(recipientId);
1360         }
1361         return recipientIds;
1362     }
1363
1364     private RecipientId resolveRecipient(final RecipientIdentifier.Single recipient) throws IOException {
1365         if (recipient instanceof RecipientIdentifier.Uuid uuidRecipient) {
1366             return account.getRecipientStore().resolveRecipient(ACI.from(uuidRecipient.uuid()));
1367         } else {
1368             final var number = ((RecipientIdentifier.Number) recipient).number();
1369             return account.getRecipientStore().resolveRecipient(number, () -> {
1370                 try {
1371                     return getRegisteredUser(number);
1372                 } catch (IOException e) {
1373                     return null;
1374                 }
1375             });
1376         }
1377     }
1378
1379     private RecipientId resolveRecipient(RecipientAddress address) {
1380         return account.getRecipientStore().resolveRecipient(address);
1381     }
1382
1383     private RecipientId resolveRecipient(SignalServiceAddress address) {
1384         return account.getRecipientStore().resolveRecipient(address);
1385     }
1386
1387     private RecipientId resolveRecipientTrusted(SignalServiceAddress address) {
1388         return account.getRecipientStore().resolveRecipientTrusted(address);
1389     }
1390
1391     @Override
1392     public void close() throws IOException {
1393         close(true);
1394     }
1395
1396     private void close(boolean closeAccount) throws IOException {
1397         Thread thread;
1398         synchronized (messageHandlers) {
1399             weakHandlers.clear();
1400             messageHandlers.clear();
1401             thread = receiveThread;
1402             receiveThread = null;
1403         }
1404         if (thread != null) {
1405             stopReceiveThread(thread);
1406         }
1407         executor.shutdown();
1408
1409         dependencies.getSignalWebSocket().disconnect();
1410
1411         if (closeAccount && account != null) {
1412             account.close();
1413         }
1414         account = null;
1415     }
1416 }