summaryrefslogtreecommitdiffstats
path: root/src/lib/simpleserver.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/lib/simpleserver.py')
-rw-r--r--src/lib/simpleserver.py95
1 files changed, 95 insertions, 0 deletions
diff --git a/src/lib/simpleserver.py b/src/lib/simpleserver.py
new file mode 100644
index 000000000..63f07d71d
--- /dev/null
+++ b/src/lib/simpleserver.py
@@ -0,0 +1,95 @@
+from SocketServer import *
+from SimpleXMLRPCServer import *
+from tlslite.api import *
+#from Bcfg2.tlslite.api import *
+
+s = open("./serverX509Cert.pem").read()
+x509 = X509()
+x509.parse(s)
+certChain = X509CertChain([x509])
+s = open("./serverX509Key.pem").read()
+privateKey = parsePEMKey(s, private=True)
+
+sessionCache = SessionCache()
+
+class MyHTTPServer(ThreadingMixIn, TLSSocketServerMixIn, SimpleXMLRPCServer):
+ db = VerifierDB("./verifierDB")
+ db.open()
+
+ def handshake(self, tlsConnection):
+ try:
+ tlsConnection.handshakeServer(certChain=certChain,
+ privateKey=privateKey,
+ verifierDB=self.db,
+ sessionCache=sessionCache)
+ tlsConnection.ignoreAbruptClose = True
+ return True
+ except TLSError, error:
+ print "Handshake failure:", str(error)
+ return False
+
+class TLSXMLRPCRequestHandler(SimpleXMLRPCRequestHandler):
+ '''TLSXMLRPCRequestHandler overrides SimpleXMLRPCRequestHandler to close
+ connections without causing problems. (just the do_POST() is broken)'''
+ def do_POST(self):
+ """Handles the HTTP POST request.
+
+ Attempts to interpret all HTTP POST requests as XML-RPC calls,
+ which are forwarded to the server's _dispatch method for handling.
+ """
+
+ # Check that the path is legal
+ if not self.is_rpc_path_valid():
+ self.report_404()
+ return
+
+ try:
+ # Get arguments by reading body of request.
+ # We read this in chunks to avoid straining
+ # socket.read(); around the 10 or 15Mb mark, some platforms
+ # begin to have problems (bug #792570).
+ max_chunk_size = 10*1024*1024
+ size_remaining = int(self.headers["content-length"])
+ L = []
+ while size_remaining:
+ chunk_size = min(size_remaining, max_chunk_size)
+ L.append(self.rfile.read(chunk_size))
+ size_remaining -= len(L[-1])
+ data = ''.join(L)
+
+ # In previous versions of SimpleXMLRPCServer, _dispatch
+ # could be overridden in this class, instead of in
+ # SimpleXMLRPCDispatcher. To maintain backwards compatibility,
+ # check to see if a subclass implements _dispatch and dispatch
+ # using that method if present.
+ response = self.server._marshaled_dispatch(
+ data, getattr(self, '_dispatch', None)
+ )
+ except: # This should only happen if the module is buggy
+ # internal error, report as HTTP server error
+ self.send_response(500)
+ self.end_headers()
+ else:
+ # got a valid XML RPC response
+ self.send_response(200)
+ self.send_header("Content-type", "text/xml")
+ self.send_header("Content-length", str(len(response)))
+ self.end_headers()
+ self.wfile.write(response)
+
+ # shut down the connection
+ self.wfile.flush()
+ #THIS IS THE ONLY MODIFICATION FROM SimpleXMLRPCRequestHandler's IMPLEMENTATION:
+ #self.connection.shutdown(1)
+ self.connection.close()
+
+
+def silly(arg="string"):
+ print arg
+ return arg[::-1]
+
+
+httpd = MyHTTPServer(('localhost', 8505), TLSXMLRPCRequestHandler)
+httpd.register_introspection_functions()
+httpd.register_function(silly)
+httpd.serve_forever()