From 7e974f306adb82dd38cc4adc4adbe624619c6f1a Mon Sep 17 00:00:00 2001
From: root <root@raspberrypi.(none)>
Date: Fri, 3 Apr 2015 16:41:45 +0200
Subject: add lib

---
 lib/beaker/crypto/pycrypto.py | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)
 create mode 100644 lib/beaker/crypto/pycrypto.py

(limited to 'lib/beaker/crypto/pycrypto.py')

diff --git a/lib/beaker/crypto/pycrypto.py b/lib/beaker/crypto/pycrypto.py
new file mode 100644
index 000000000..a3eb4d9db
--- /dev/null
+++ b/lib/beaker/crypto/pycrypto.py
@@ -0,0 +1,31 @@
+"""Encryption module that uses pycryptopp or pycrypto"""
+try:
+    # Pycryptopp is preferred over Crypto because Crypto has had
+    # various periods of not being maintained, and pycryptopp uses
+    # the Crypto++ library which is generally considered the 'gold standard'
+    # of crypto implementations
+    from pycryptopp.cipher import aes
+
+    def aesEncrypt(data, key):
+        cipher = aes.AES(key)
+        return cipher.process(data)
+    
+    # magic.
+    aesDecrypt = aesEncrypt
+    
+except ImportError:
+    from Crypto.Cipher import AES
+
+    def aesEncrypt(data, key):
+        cipher = AES.new(key)
+        
+        data = data + (" " * (16 - (len(data) % 16)))
+        return cipher.encrypt(data)
+
+    def aesDecrypt(data, key):
+        cipher = AES.new(key)
+
+        return cipher.decrypt(data).rstrip()
+
+def getKeyLength():
+    return 32
-- 
cgit v1.2.3