From 02cd9d60641cd40cfc6c7b8216d5861f40f92494 Mon Sep 17 00:00:00 2001
From: Victor Shyba <victor1984@riseup.net>
Date: Wed, 15 Mar 2017 08:26:56 -0300
Subject: [feature] ascii armor disable option

---
 client/src/leap/soledad/client/_crypto.py | 30 +++++++++++++-----------------
 1 file changed, 13 insertions(+), 17 deletions(-)

(limited to 'client')

diff --git a/client/src/leap/soledad/client/_crypto.py b/client/src/leap/soledad/client/_crypto.py
index 8dbdc094..923891fc 100644
--- a/client/src/leap/soledad/client/_crypto.py
+++ b/client/src/leap/soledad/client/_crypto.py
@@ -300,8 +300,7 @@ class BlobEncryptor(object):
             result.write(
                 base64.urlsafe_b64encode(encrypted + self.tag))
         else:
-            result.write(encrypted)
-            result.write(self.tag)
+            result.write(encrypted + self.tag)
 
         result.seek(0)
         return defer.succeed(result)
@@ -319,28 +318,25 @@ class BlobDecryptor(object):
     # TODO enable the ascii armor = False
 
     def __init__(self, doc_info, ciphertext_fd, result=None,
-                 secret=None, armor=True, start_stream=True):
+                 secret=None, armor=True, start_stream=True, tag=None):
         if not secret:
             raise EncryptionDecryptionError('no secret given')
-        if armor is False:
-            raise NotImplementedError
 
         self.doc_id = doc_info.doc_id
         self.rev = doc_info.rev
         self.fd = ciphertext_fd
         self.armor = armor
         self._producer = None
+        self.result = result or BytesIO()
+        self.sym_key = _get_sym_key_for_doc(doc_info.doc_id, secret)
         self.size = None
-
+        self.tag = tag
         preamble, iv = self._consume_preamble()
         assert preamble
         assert iv
-
-        self.result = result or BytesIO()
-        sym_key = _get_sym_key_for_doc(doc_info.doc_id, secret)
-
-        self._aes = AESWriter(sym_key, iv, self.result, tag=self.tag)
+        self._aes = AESWriter(self.sym_key, iv, self.result, tag=self.tag)
         self._aes.authenticate(preamble)
+
         if start_stream:
             self._start_stream()
 
@@ -348,14 +344,14 @@ class BlobDecryptor(object):
         self._producer = FileBodyProducer(self.fd, readSize=2**16)
 
     def _consume_preamble(self):
-
         self.fd.seek(0)
         try:
-            parts = self.fd.getvalue().split()
-            encoded_preamble = parts[0]
-            preamble = base64.urlsafe_b64decode(encoded_preamble)
-            ciphertext = base64.urlsafe_b64decode(parts[1])
-            self.tag, ciphertext = ciphertext[-16:], ciphertext[:-16]
+            preamble, ciphertext = self.fd.getvalue().split(SEPARATOR, 1)
+            preamble = base64.urlsafe_b64decode(preamble)
+            if self.armor:
+                ciphertext = base64.urlsafe_b64decode(ciphertext)
+            tag, ciphertext = ciphertext[-16:], ciphertext[:-16]
+            self.tag = self.tag or tag
 
         except (TypeError, ValueError):
             raise InvalidBlob
-- 
cgit v1.2.3