Write first version of command daemon

command_daemon/command_daemon.py

@@ -0,0 +1,109 @@
+#!/usr/bin/env python
+
+import sys, os
+import subprocess
+import json, urlparse
+import SocketServer, SimpleHTTPServer
+from optparse import OptionParser
+
+parser = OptionParser()
+(options, cmdargs) = parser.parse_args()
+
+try:
+	f = open("session_key", "r")
+	session_key = f.read().strip()
+	f.close()
+except IOError, e:
+	sys.stderr.write("You must specify a session key.\n")
+	exit(1)
+	
+os.remove("session_key")
+
+class CommandHandler(SimpleHTTPServer.SimpleHTTPRequestHandler):
+	def do_GET(self):
+		global session_key
+		
+		req = urlparse.urlparse(self.path)
+		get_params = urlparse.parse_qs(req.query)
+		path = req.path
+		
+		if path=='/':
+			try:
+				command = json.loads(get_params['command'][0])
+			except KeyError, e:
+				self.send_404()
+				return
+			except IndexError, e:
+				self.send_404()
+				return
+			except ValueError, e:
+				self.send_404()
+				return
+				
+			try:
+				key = get_params['key'][0]
+			except KeyError, e:
+				self.send_403()
+				return
+			except IndexError, e:
+				self.send_403()
+				return
+				
+			if key != session_key:
+				self.send_403()
+				return
+			
+			try:
+				result = json.dumps(self.run_command(command))
+			except Exception, e:
+				print e
+				self.send_404()
+				return
+			
+			self.send_response(200)
+			self.send_header('Content-type','text/json')
+			self.end_headers()
+			self.wfile.write(result)
+		else:
+			self.send_404()
+			return
+			
+	def send_404(self):
+		self.send_response(404)
+		self.send_header('Content-type','text/plain')
+		self.end_headers()
+		self.wfile.write("404 Not Found")
+		
+	def send_403(self):
+		self.send_response(403)
+		self.send_header('Content-type','text/plain')
+		self.end_headers()
+		self.wfile.write("403 Forbidden")
+		
+	def run_command(self, args):
+		pr = subprocess.Popen(args, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+		data = pr.communicate()
+		pr.wait()
+
+		return {
+			'stdout': data[0],
+			'stderr': data[1],
+			'returncode': pr.returncode
+		}
+
+if os.fork(): exit(0)
+os.umask(0) 
+os.setsid() 
+if os.fork(): exit(0)
+
+sys.stdout.flush()
+sys.stderr.flush()
+si = file('/dev/null', 'r')
+so = file('/dev/null', 'a+')
+se = file('/dev/null', 'a+', 0)
+os.dup2(si.fileno(), sys.stdin.fileno())
+os.dup2(so.fileno(), sys.stdout.fileno())
+os.dup2(se.fileno(), sys.stderr.fileno())
+
+httpd = SocketServer.ThreadingTCPServer(("localhost", 3434), CommandHandler)
+httpd.serve_forever()