Use a sequence of bytes (representing AriesMessage) as input for Encr…

…yptionEnvelope::create (#1007) * Use a sequence of bytes (representing AriesMessage) as input for EncryptionEnvelope::create Signed-off-by: Naian <126972030+nain-F49FF806@users.noreply.github.com>
hyperledger · Oct 4, 2023 · 77c8d8f · 77c8d8f
1 parent 34c35de
commit 77c8d8f
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 10 deletions.
diff --git a/aries_vcx/src/protocols/connection/generic/mod.rs b/aries_vcx/src/protocols/connection/generic/mod.rs
@@ -180,7 +180,13 @@ impl GenericConnection {
             AriesVcxErrorKind::NotReady,
             "No DidDoc present",
         ))?;
-        EncryptionEnvelope::create(wallet, message, Some(sender_verkey), did_doc).await
+        EncryptionEnvelope::create(
+            wallet,
+            json!(message).to_string().as_bytes(),
+            Some(sender_verkey),
+            did_doc,
+        )
+        .await
     }
 
     pub async fn send_message<T>(

diff --git a/aries_vcx/src/protocols/connection/mod.rs b/aries_vcx/src/protocols/connection/mod.rs
@@ -100,7 +100,13 @@ where
         message: &AriesMessage,
     ) -> VcxResult<EncryptionEnvelope> {
         let sender_verkey = &self.pairwise_info().pw_vk;
-        EncryptionEnvelope::create(wallet, message, Some(sender_verkey), self.their_did_doc()).await
+        EncryptionEnvelope::create(
+            wallet,
+            json!(message).to_string().as_bytes(),
+            Some(sender_verkey),
+            self.their_did_doc(),
+        )
+        .await
     }
 
     pub fn remote_did(&self) -> &str {

diff --git a/aries_vcx/src/utils/encryption_envelope.rs b/aries_vcx/src/utils/encryption_envelope.rs
@@ -14,9 +14,11 @@ use crate::{errors::error::prelude::*, global::settings, utils::constants};
 pub struct EncryptionEnvelope(pub Vec<u8>);
 
 impl EncryptionEnvelope {
+    /// Create an Encryption Envelope from a plaintext AriesMessage encoded as sequence of bytes.
+    /// If did_doc includes routing_keys, then also wrap in appropriate layers of forward message.
     pub async fn create(
         wallet: &impl BaseWallet,
-        message: &AriesMessage,
+        message: &[u8],
         pw_verkey: Option<&str>,
         did_doc: &AriesDidDoc,
     ) -> VcxResult<EncryptionEnvelope> {
@@ -41,12 +43,10 @@ impl EncryptionEnvelope {
 
     async fn encrypt_for_pairwise(
         wallet: &impl BaseWallet,
-        message: &AriesMessage,
+        message: &[u8],
         pw_verkey: Option<&str>,
         did_doc: &AriesDidDoc,
     ) -> VcxResult<Vec<u8>> {
-        let message = json!(message).to_string();
-
         let receiver_keys = json!(did_doc.recipient_keys()?).to_string();
 
         debug!(
@@ -55,7 +55,7 @@ impl EncryptionEnvelope {
         );
 
         wallet
-            .pack_message(pw_verkey, &receiver_keys, message.as_bytes())
+            .pack_message(pw_verkey, &receiver_keys, message)
             .await
             .map_err(|err| err.into())
     }

diff --git a/aries_vcx/src/utils/mod.rs b/aries_vcx/src/utils/mod.rs
@@ -56,8 +56,14 @@ pub async fn send_message(
         message,
         &did_doc
     );
-    let EncryptionEnvelope(envelope) =
-        EncryptionEnvelope::create(wallet, &message, Some(&sender_verkey), &did_doc).await?;
+
+    let EncryptionEnvelope(envelope) = EncryptionEnvelope::create(
+        wallet,
+        json!(message).to_string().as_bytes(),
+        Some(&sender_verkey),
+        &did_doc,
+    )
+    .await?;
 
     // TODO: Extract from agency client
     agency_client::httpclient::post_message(
@@ -81,7 +87,8 @@ pub async fn send_message_anonymously(
         &did_doc
     );
     let EncryptionEnvelope(envelope) =
-        EncryptionEnvelope::create(wallet, message, None, did_doc).await?;
+        EncryptionEnvelope::create(wallet, json!(message).to_string().as_bytes(), None, did_doc)
+            .await?;
 
     agency_client::httpclient::post_message(
         envelope,