1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
#!/usr/bin/env python
import os
__author__ = 'rolandh'
import M2Crypto
from base64 import b64encode, b64decode
class AES(object):
def __init__(self, iv=None):
if iv is None:
self.iv = '\0' * 16
else:
self.iv = iv
def build_cipher(self, key, iv, op=1, alg="aes_128_cbc"):
"""
:param key: encryption key
:param iv: init vector
:param op: key usage - 1 (encryption) or 0 (decryption)
:param alg: cipher algorithm
:return: A Cipher instance
"""
return M2Crypto.EVP.Cipher(alg=alg, key=key, iv=iv, op=op)
def encrypt(self, key, msg, iv=None):
"""
:param key: The encryption key
:param iv: init vector
:param msg: Message to be encrypted
:return: The encrypted message base64 encoded
"""
if iv is None:
iv = self.iv
cipher = self.build_cipher(key, iv, 1)
v = cipher.update(msg)
v = v + cipher.final()
v = b64encode(v)
return v
def decrypt(self, key, msg, iv=None):
"""
:param key: The encryption key
:param iv: init vector
:param msg: Base64 encoded message to be decrypted
:return: The decrypted message
"""
if iv is None:
iv = self.iv
data = b64decode(msg)
cipher = self.build_cipher(key, iv, 0)
v = cipher.update(data)
v = v + cipher.final()
return v
if __name__ == "__main__":
key = "123452345"
msg = "ToBeOrNotTobe W.S."
iv = os.urandom(16)
aes = AES()
encrypted_msg = aes.encrypt(key, msg, iv)
print aes.decrypt(key, encrypted_msg, iv)
|
