Decrypt device names in linked devices views.

SignalServiceKit/src/Devices/OWSDevice.m

@@ -13,6 +13,7 @@
 #import "YapDatabaseTransaction.h"
 #import <Mantle/MTLValueTransformer.h>
 #import <SignalCoreKit/NSDate+OWS.h>
+#import <SignalServiceKit/OWSIdentityManager.h>
 #import <SignalServiceKit/SignalServiceKit-Swift.h>
 
 NS_ASSUME_NONNULL_BEGIN
@@ -130,6 +131,13 @@ NSString *const kOWSPrimaryStorage_MayHaveLinkedDevices = @"kTSStorageManager_Ma
     return TSAccountManager.sharedInstance;
 }
 
+- (OWSIdentityManager *)identityManager
+{
+    OWSAssertDebug(SSKEnvironment.shared.identityManager);
+
+    return SSKEnvironment.shared.identityManager;
+}
+
 #pragma mark -
 
 - (void)saveWithTransaction:(YapDatabaseReadWriteTransaction *)transaction
@@ -275,6 +283,20 @@ NSString *const kOWSPrimaryStorage_MayHaveLinkedDevices = @"kTSStorageManager_Ma
 - (NSString *)displayName
 {
     if (self.name) {
+        ECKeyPair *_Nullable identityKeyPair = self.identityManager.identityKeyPair;
+        OWSAssertDebug(identityKeyPair);
+        if (identityKeyPair) {
+            NSError *error;
+            NSString *_Nullable decryptedName =
+                [DeviceNames decryptDeviceNameWithInputString:self.name identityKeyPair:identityKeyPair error:&error];
+            if (error) {
+                // Not necessarily an error; might be a legacy device name.
+                OWSLogError(@"Could not decrypt device name: %@", error);
+            } else if (decryptedName) {
+                return decryptedName;
+            }
+        }
+
         return self.name;
     }
 

SignalServiceKit/src/Util/DeviceNames.swift

@@ -104,10 +104,22 @@ public class DeviceNames: NSObject {
     }
 
     @objc
-    public class func decryptDeviceName(input: Data,
+    public class func decryptDeviceName(inputString: String,
+                                        identityKeyPair: ECKeyPair) throws -> String {
+        guard let inputData = Data(base64Encoded: inputString) else {
+            // Not necessarily an error; might be a legacy device name.
+            throw DeviceNameError.invalidInput
+        }
+
+        return try decryptDeviceName(inputData: inputData,
+                                     identityKeyPair: identityKeyPair)
+    }
+
+    @objc
+    public class func decryptDeviceName(inputData: Data,
                                         identityKeyPair: ECKeyPair) throws -> String {
 
-        guard let protoData = Data(base64Encoded: input) else {
+        guard let protoData = Data(base64Encoded: inputData) else {
             // Not necessarily an error; might be a legacy device name.
             throw DeviceNameError.invalidInput
         }

SignalServiceKit/tests/Util/DeviceNamesTest.swift

@@ -31,7 +31,7 @@ class DeviceNamesTest: SSKBaseTestSwift {
         }
 
         do {
-            _ = try DeviceNames.decryptDeviceName(input: plaintextData,
+            _ = try DeviceNames.decryptDeviceName(inputData: plaintextData,
                                                   identityKeyPair: identityKeyPair)
             XCTFail("Unexpectedly did not throw error.")
         } catch {
@@ -55,7 +55,7 @@ class DeviceNamesTest: SSKBaseTestSwift {
 
         let decrypted: String
         do {
-            decrypted = try DeviceNames.decryptDeviceName(input: encrypted,
+            decrypted = try DeviceNames.decryptDeviceName(inputData: encrypted,
                                                           identityKeyPair: identityKeyPair)
         } catch {
             XCTFail("Failed with error: \(error)")