summaryrefslogtreecommitdiff
path: root/testing/tests/client/test_crypto2.py
diff options
context:
space:
mode:
authorKali Kaneko <kali@leap.se>2016-09-14 01:52:36 -0400
committerdrebs <drebs@leap.se>2016-12-12 09:11:59 -0200
commitfcf3b3046dd2005992638ebf993d53897af8ed3a (patch)
tree5953ab7be83d79077d4072f5ec955dbf5d74db61 /testing/tests/client/test_crypto2.py
parent77b952eeec20623e3b2e6f47597c59124c83f3d4 (diff)
[refactor] remove encryption pool
Diffstat (limited to 'testing/tests/client/test_crypto2.py')
-rw-r--r--testing/tests/client/test_crypto2.py63
1 files changed, 63 insertions, 0 deletions
diff --git a/testing/tests/client/test_crypto2.py b/testing/tests/client/test_crypto2.py
new file mode 100644
index 00000000..ae280020
--- /dev/null
+++ b/testing/tests/client/test_crypto2.py
@@ -0,0 +1,63 @@
+# -*- coding: utf-8 -*-
+# test_crypto2.py
+# Copyright (C) 2016 LEAP Encryption Access Project
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+"""
+Tests for the _crypto module
+"""
+
+import StringIO
+
+
+import leap.soledad.client
+from leap.soledad.client import _crypto
+
+
+from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
+from cryptography.hazmat.backends import default_backend
+
+
+def _aes_encrypt(key, iv, data):
+ backend = default_backend()
+ cipher = Cipher(algorithms.AES(key), modes.CTR(iv), backend=backend)
+ encryptor = cipher.encryptor()
+ return encryptor.update(data) + encryptor.finalize()
+
+
+def test_chunked_encryption():
+ key = 'A' * 32
+ iv = 'A' * 16
+ data = (
+ "You can't come up against "
+ "the world's most powerful intelligence "
+ "agencies and not accept the risk. "
+ "If they want to get you, over time "
+ "they will.")
+
+ fd = StringIO.StringIO()
+ aes = _crypto.AESWriter(key, fd, iv)
+
+ block = 16
+
+ for i in range(len(data)/block):
+ chunk = data[i * block:(i+1)*block]
+ aes.write(chunk)
+ aes.end()
+
+ ciphertext_chunked = fd.getvalue()
+ ciphertext = _aes_encrypt(key, iv, data)
+
+ assert ciphertext_chunked == ciphertext