-
Notifications
You must be signed in to change notification settings - Fork 285
/
PubnubCrypto.py
94 lines (82 loc) · 2.23 KB
/
PubnubCrypto.py
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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
## www.pubnub.com - PubNub Real-time push service in the cloud.
# coding=utf8
## PubNub Real-time Push APIs and Notifications Framework
## Copyright (c) 2010 Stephen Blum
## http://www.pubnub.com/
## -----------------------------------
## PubNub 3.1 Real-time Push Cloud API
## -----------------------------------
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.
#**
"""
return msg + ((block_size - len(msg) % block_size) * chr(block_size - len(msg) % block_size))
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
#**
"""
m = hashlib.sha256()
m.update(key)
return m.hexdigest()[:32]
def encrypt( self, key, msg ):
"""
#**
#* encrypt
#*
#* encrypts the message
#* @return message in encrypted format
#**
"""
secret = self.getSecret(key)
Initial16bytes='0123456789012345'
cipher = AES.new(secret,AES.MODE_CBC,Initial16bytes)
return encodestring(cipher.encrypt(self.pad(msg)))
def decrypt( self, key, msg ):
"""
#**
#* decrypt
#*
#* decrypts the message
#* @return message in decryped format
#**
"""
secret = self.getSecret(key)
Initial16bytes='0123456789012345'
cipher = AES.new(secret,AES.MODE_CBC,Initial16bytes)
return self.depad((cipher.decrypt(decodestring(msg))))