requestobject and working interface to the core

3 files changed, 59 insertions, 29 deletions

diff --git a/module/remote/RequestHandler.py b/module/remote/RequestHandler.py
index c72f6eaaa..09ad5b548 100644
--- a/module/remote/RequestHandler.py
+++ b/module/remote/RequestHandler.py
@@ -8,44 +8,57 @@ this module handels the incoming requests
 """
 
 import base64
-import cPickle
 import random
 import string
+
+import cPickle
 from Crypto.Cipher import AES
+from Crypto.Hash import MD5
 from Crypto.Hash import SHA
+from RequestObject import RequestObject
 
 
-class RequestHandler():
+class RequestHandler:
     def __init__(self, core):
-	self.core = core
-	key = SHA.new(core.config['remotepassword'])
-	self.aes = AES.new(key.hexdigest()[:32], AES.MODE_ECB)
+        self.core = core
+        key = SHA.new(core.config['remotepassword'])
+        key = MD5.new(key.hexdigest())
+        self.aes = AES.new(key.hexdigest(), AES.MODE_ECB)
 
     def proceed(self, data):
-	return self.encrypt({'befehl' : None , 'args':[1,2,3], 'test': 'lol'})
+        obj = self.decrypt(data)
 
-    def decrypt(self, dec_str):
-	dec_str = base64.standard_b64decode(dec_str)
-	dec_str = self.aes.decrypt(dec_str)
+        if obj.command == "exec":
+            func = getattr(self.core, obj.function)
+            obj.response = func()
+        else:
+            obj.response = "antwort"
+        
+        return self.encrypt(obj)
 
-	dec_str = dec_str[:-(int(dec_str[-1],16)+1)]
-	obj = cPickle.loads(dec_str)
-	return obj
+
+    def decrypt(self, dec_str):
+        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)
+        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 = self.aes.encrypt(enc_str)
-	enc_str = base64.standard_b64encode(enc_str)
-	return enc_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
 
 
 
diff --git a/module/remote/RequestObject.py b/module/remote/RequestObject.py
new file mode 100644
index 000000000..77322663a
--- /dev/null
+++ b/module/remote/RequestObject.py
@@ -0,0 +1,17 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+"""
+authored by:  RaNaN
+
+this module handels the incoming requests
+
+"""
+
+class RequestObject(object):
+    def __init__(self):
+        self.version = 0
+        self.sender = "ip"
+        self.command = None
+        self.function = ""
+        self.args = []
+        self.response = ""
\ No newline at end of file
diff --git a/module/remote/SocketServer.py b/module/remote/SocketServer.py
index 1252547ad..9000c1c71 100644
--- a/module/remote/SocketServer.py
+++ b/module/remote/SocketServer.py
@@ -48,13 +48,13 @@ class SecondaryServerSocket(asynchat.async_chat):
 	self.pycore = pycore
 	self.handler = RequestHandler(pycore)     
 	self.set_terminator('\n')
-        self.data = []
+        self.data = ""
     def collect_incoming_data(self, data):
-        self.data.append(data)
+        self.data += data
     def found_terminator(self):
-	rep = self.handler.proceed(self.data)
+        rep = self.handler.proceed(self.data)
         self.push(rep)
-        self.data = []
+        self.data = ""
         #having fun with the data
     def handle_close(self):
         print "Disconnected from", self.getpeername()