Actually write pre-keys as byte arrays

signalapp · May 26, 2023 · 4a8ad31 · 4a8ad31
1 parent a5f853c
commit 4a8ad31
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/service/src/main/java/org/whispersystems/textsecuregcm/storage/Keys.java b/service/src/main/java/org/whispersystems/textsecuregcm/storage/Keys.java
@@ -374,13 +374,13 @@ private Map<String, AttributeValue> getItemFromPreKey(final UUID accountUuid, fi
       return Map.of(
           KEY_ACCOUNT_UUID, getPartitionKey(accountUuid),
           KEY_DEVICE_ID_KEY_ID, getSortKey(deviceId, spk.getKeyId()),
-          KEY_PUBLIC_KEY, AttributeValues.fromString(spk.getPublicKey()),
-          KEY_SIGNATURE, AttributeValues.fromString(spk.getSignature()));
+          KEY_PUBLIC_KEY, AttributeValues.fromByteArray(Base64.getDecoder().decode(spk.getPublicKey())),
+          KEY_SIGNATURE, AttributeValues.fromByteArray(Base64.getDecoder().decode(spk.getSignature())));
     }
     return Map.of(
         KEY_ACCOUNT_UUID, getPartitionKey(accountUuid),
         KEY_DEVICE_ID_KEY_ID, getSortKey(deviceId, preKey.getKeyId()),
-        KEY_PUBLIC_KEY, AttributeValues.fromString(preKey.getPublicKey()));
+        KEY_PUBLIC_KEY, AttributeValues.fromByteArray(Base64.getDecoder().decode(preKey.getPublicKey())));
   }
 
   private PreKey getPreKeyFromItem(Map<String, AttributeValue> item) {