diff options
Diffstat (limited to 'sockettest.py')
| -rw-r--r-- | sockettest.py | 68 |
1 files changed, 37 insertions, 31 deletions
diff --git a/sockettest.py b/sockettest.py index 30a4deb81..882895b57 100644 --- a/sockettest.py +++ b/sockettest.py @@ -12,56 +12,62 @@ import random import string from Crypto.Cipher import AES from Crypto.Hash import SHA +from Crypto.Hash import MD5 +from module.remote.RequestObject import RequestObject -class handler: + +class Handler: def __init__(self): - key = SHA.new("pwhere") - self.aes = AES.new(key.hexdigest()[:32], AES.MODE_ECB) + key = SHA.new("pwhere") + key = MD5.new(key.hexdigest()) + self.aes = AES.new(key.hexdigest(), AES.MODE_ECB) def proceed(self, data): - return self.decrypt(self.encrypt(str(("lol","mehrlol","pff")))) + + return self.decrypt(self.encrypt(str(("lol","mehrlol","pff")))) def decrypt(self, dec_str): - try: - dec_str = base64.standard_b64decode(dec_str) - dec_str = self.aes.decrypt(dec_str) + try: + dec_str = base64.standard_b64decode(dec_str) + dec_str = self.aes.decrypt(dec_str) - dec_str = dec_str[:-(int(dec_str[-1],16)+1)] - obj = cPickle.loads(dec_str) - except: - obj = None - return obj + dec_str = dec_str[:-(int(dec_str[-1],16)+1)] + obj = cPickle.loads(dec_str) + except: + obj = None + return obj def encrypt(self, obj): - enc_str = cPickle.dumps(obj, 1) - padding = len(enc_str) % 16 - padding = 16 - padding - - p_str = "" - for i in range(padding - 1): - p_str += random.choice(string.letters+string.digits) - p_str += hex(len(p_str)).replace("0x","") - enc_str += p_str + enc_str = cPickle.dumps(obj, 1) + padding = len(enc_str) % 16 + padding = 16 - padding + + p_str = "" + for i in range(padding - 1): + p_str += random.choice(string.letters+string.digits) + p_str += hex(len(p_str)).replace("0x","") + enc_str += p_str - enc_str = self.aes.encrypt(enc_str) - enc_str = base64.standard_b64encode(enc_str) - return enc_str + enc_str = self.aes.encrypt(enc_str) + enc_str = base64.standard_b64encode(enc_str) + return enc_str import socket -handler = handler() +handler = Handler() sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) sock.connect(('localhost', 7272)) print "Connected to server" -data = """A few lines of data to test the operation of both server and client. -Und noch eine Zeile""" -for line in data.splitlines(): - sock.sendall(line+'\n') - print "Sent:", line + +obj = RequestObject() +obj.command = "exec" +obj.function = "get_downloads" + +sock.sendall(handler.encrypt(obj)+"\n") response = sock.recv(8192) -print "Received:", handler.decrypt(response) +print "Received:", handler.decrypt(response).response sock.close() |