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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
from Crypto.Cipher import AES
from Crypto.Hash import MD5
from base64 import encodestring, decodestring
import hashlib
import hmac
class PubnubCrypto() :
"""
#**
#* PubnubCrypto
#*
#**
## Initiate Class
pc = PubnubCrypto
"""
def pad( self, msg, block_size=16 ):
"""
#**
#* pad
#*
#* pad the text to be encrypted
#* appends a padding character to the end of the String
#* until the string has block_size length
#* @return msg with padding.
#**
"""
padding = block_size - (len(msg) % block_size)
return msg + (chr(padding)*padding).encode('utf-8')
def depad( self, msg ):
"""
#**
#* depad
#*
#* depad the decryptet message"
#* @return msg without padding.
#**
"""
return msg[0:-ord(msg[-1])]
def getSecret( self, key ):
"""
#**
#* getSecret
#*
#* hases the key to MD5
#* @return key in MD5 format
#**
"""
return hashlib.sha256(key.encode("utf-8")).hexdigest()
def encrypt( self, key, msg ):
"""
#**
#* encrypt
#*
#* encrypts the message
#* @return message in encrypted format
#**
"""
secret = self.getSecret(key)
Initial16bytes='0123456789012345'
cipher = AES.new(secret[0:32],AES.MODE_CBC,Initial16bytes)
return encodestring(cipher.encrypt(self.pad(msg.encode('utf-8')))).decode('utf-8')
def decrypt( self, key, msg ):
"""
#**
#* decrypt
#*
#* decrypts the message
#* @return message in decryped format
#**
"""
secret = self.getSecret(key)
Initial16bytes='0123456789012345'
cipher = AES.new(secret[0:32],AES.MODE_CBC,Initial16bytes)
return (cipher.decrypt(decodestring(msg.encode('utf-8')))).decode('utf-8')
|
