| 1 | #!/usr/local/bin/python |
|---|
| 2 | |
|---|
| 3 | import sys |
|---|
| 4 | |
|---|
| 5 | from BaseHTTPServer import BaseHTTPRequestHandler |
|---|
| 6 | |
|---|
| 7 | from ZSI import Fault, ParseException, FaultFromNotUnderstood, \ |
|---|
| 8 | FaultFromZSIException, FaultFromException, ParsedSoap, SoapWriter |
|---|
| 9 | |
|---|
| 10 | from M2Crypto import SSL |
|---|
| 11 | from M2Crypto.SSL.SSLServer import SSLServer |
|---|
| 12 | import xmlrpclib |
|---|
| 13 | |
|---|
| 14 | from optparse import OptionParser |
|---|
| 15 | |
|---|
| 16 | from fedd_util import fedd_ssl_context, fedid |
|---|
| 17 | from fedd_proj import new_feddservice |
|---|
| 18 | |
|---|
| 19 | # The SSL server here is based on the implementation described at |
|---|
| 20 | # http://www.xml.com/pub/a/ws/2004/01/20/salz.html |
|---|
| 21 | |
|---|
| 22 | # Turn off the matching of hostname to certificate ID |
|---|
| 23 | SSL.Connection.clientPostConnectionCheck = None |
|---|
| 24 | |
|---|
| 25 | class fedd_server(SSLServer): |
|---|
| 26 | def __init__(self, ME, handler, ssl_ctx, impl): |
|---|
| 27 | SSLServer.__init__(self, ME, handler, ssl_ctx) |
|---|
| 28 | self.impl = impl |
|---|
| 29 | |
|---|
| 30 | class fedd_soap_handler(BaseHTTPRequestHandler): |
|---|
| 31 | server_version = "ZSI/2.0 fedd/0.1 " + BaseHTTPRequestHandler.server_version |
|---|
| 32 | |
|---|
| 33 | |
|---|
| 34 | def send_xml(self, text, code=200): |
|---|
| 35 | """Send an XML document as reply""" |
|---|
| 36 | self.send_response(code) |
|---|
| 37 | self.send_header('Content-type', 'text/xml; charset="utf-8"') |
|---|
| 38 | self.send_header('Content-Length', str(len(text))) |
|---|
| 39 | self.end_headers() |
|---|
| 40 | self.wfile.write(text) |
|---|
| 41 | self.wfile.flush() |
|---|
| 42 | |
|---|
| 43 | def send_fault(self, f, code=500): |
|---|
| 44 | """Send a SOAP encoded fault as reply""" |
|---|
| 45 | self.send_xml(f.AsSOAP(), code) |
|---|
| 46 | |
|---|
| 47 | def check_headers(self, ps): |
|---|
| 48 | """Send a fault for any required envelope headers""" |
|---|
| 49 | for (uri, localname) in ps.WhatMustIUnderstand(): |
|---|
| 50 | self.send_fault(FaultFromNotUnderstood(uri, lname, 'fedd')) |
|---|
| 51 | return False |
|---|
| 52 | return True |
|---|
| 53 | |
|---|
| 54 | def check_method(self, ps): |
|---|
| 55 | """Confirm that this class implements the namespace and SOAP method""" |
|---|
| 56 | root = ps.body_root |
|---|
| 57 | if root.namespaceURI != self.server.impl.soap_namespace: |
|---|
| 58 | self.send_fault(Fault(Fault.Client, |
|---|
| 59 | 'Unknown namespace "%s"' % root.namespaceURI)) |
|---|
| 60 | return False |
|---|
| 61 | |
|---|
| 62 | if getattr(root, 'localName', None) == None: |
|---|
| 63 | self.send_fault(Fault(Fault.Client, 'No method"')) |
|---|
| 64 | return False |
|---|
| 65 | return True |
|---|
| 66 | |
|---|
| 67 | def do_POST(self): |
|---|
| 68 | """Treat an HTTP POST request as a SOAP service call""" |
|---|
| 69 | try: |
|---|
| 70 | cl = int(self.headers['content-length']) |
|---|
| 71 | data = self.rfile.read(cl) |
|---|
| 72 | ps = ParsedSoap(data) |
|---|
| 73 | except ParseException, e: |
|---|
| 74 | self.send_fault(FaultFromZSIException(e)) |
|---|
| 75 | return |
|---|
| 76 | except Exception, e: |
|---|
| 77 | self.send_fault(FaultFromException(e, 0, sys.exc_info()[2])) |
|---|
| 78 | return |
|---|
| 79 | if not self.check_headers(ps): return |
|---|
| 80 | if not self.check_method(ps): return |
|---|
| 81 | try: |
|---|
| 82 | resp = self.server.impl.soap_dispatch(ps.body_root.localName, ps, |
|---|
| 83 | fedid(cert=self.request.get_peer_cert())) |
|---|
| 84 | except Fault, f: |
|---|
| 85 | self.send_fault(f) |
|---|
| 86 | resp = None |
|---|
| 87 | |
|---|
| 88 | if resp != None: |
|---|
| 89 | sw = SoapWriter() |
|---|
| 90 | sw.serialize(resp) |
|---|
| 91 | self.send_xml(str(sw)) |
|---|
| 92 | |
|---|
| 93 | class fedd_xmlrpc_handler(BaseHTTPRequestHandler): |
|---|
| 94 | server_version = "ZSI/2.0 fedd/0.1 " + BaseHTTPRequestHandler.server_version |
|---|
| 95 | |
|---|
| 96 | def send_xml(self, text, code=200): |
|---|
| 97 | """Send an XML document as reply""" |
|---|
| 98 | self.send_response(code) |
|---|
| 99 | self.send_header('Content-type', 'text/xml; charset="utf-8"') |
|---|
| 100 | self.send_header('Content-Length', str(len(text))) |
|---|
| 101 | self.end_headers() |
|---|
| 102 | self.wfile.write(text) |
|---|
| 103 | self.wfile.flush() |
|---|
| 104 | |
|---|
| 105 | def do_POST(self): |
|---|
| 106 | """Treat an HTTP POST request as an XMLRPC service call""" |
|---|
| 107 | |
|---|
| 108 | resp = None |
|---|
| 109 | data = None |
|---|
| 110 | cl = int(self.headers['content-length']) |
|---|
| 111 | data = self.rfile.read(cl) |
|---|
| 112 | params, method = xmlrpclib.loads(data) |
|---|
| 113 | |
|---|
| 114 | try: |
|---|
| 115 | resp = self.server.impl.xmlrpc_dispatch(method, params, |
|---|
| 116 | fedid(cert=self.request.get_peer_cert())) |
|---|
| 117 | data = xmlrpclib.dumps((resp,), methodresponse=True) |
|---|
| 118 | except Fault, f: |
|---|
| 119 | xf = xmlrpclib.Fault(f.code, f.string) |
|---|
| 120 | data = xmlrpclib.dumps(xf, methodresponse=True) |
|---|
| 121 | resp = None |
|---|
| 122 | except xmlrpclib.Fault, f: |
|---|
| 123 | data = xmlrpclib.dumps(f, methodresponse=True) |
|---|
| 124 | resp = None |
|---|
| 125 | |
|---|
| 126 | self.send_xml(data) |
|---|
| 127 | |
|---|
| 128 | |
|---|
| 129 | class fedd_opts(OptionParser): |
|---|
| 130 | """Encapsulate option processing in this class, rather than in main""" |
|---|
| 131 | def __init__(self): |
|---|
| 132 | OptionParser.__init__(self, usage="%prog [opts] (--help for details)", |
|---|
| 133 | version="0.1") |
|---|
| 134 | |
|---|
| 135 | self.set_defaults(host="localhost", port=23235, |
|---|
| 136 | transport="soap",debug=0) |
|---|
| 137 | |
|---|
| 138 | self.add_option("-d", "--debug", action="count", dest="debug", |
|---|
| 139 | help="Set debug. Repeat for more information") |
|---|
| 140 | self.add_option("-f", "--configfile", action="store", |
|---|
| 141 | dest="configfile", help="Configuration file (required)") |
|---|
| 142 | self.add_option("-H", "--host", action="store", type="string", |
|---|
| 143 | dest="host", help="Hostname to listen on (default %default)") |
|---|
| 144 | self.add_option("-p", "--port", action="store", type="int", |
|---|
| 145 | dest="port", help="Port to listen on (default %default)") |
|---|
| 146 | self.add_option("-x","--transport", action="store", type="choice", |
|---|
| 147 | choices=("xmlrpc", "soap"), |
|---|
| 148 | help="Transport for request (xmlrpc|soap) (Default: %default)") |
|---|
| 149 | self.add_option("--trace", action="store_const", dest="tracefile", |
|---|
| 150 | const=sys.stderr, help="Print SOAP exchange to stderr") |
|---|
| 151 | |
|---|
| 152 | opts, args = fedd_opts().parse_args() |
|---|
| 153 | |
|---|
| 154 | if opts.configfile != None: |
|---|
| 155 | try: |
|---|
| 156 | impl = new_feddservice(opts.configfile) |
|---|
| 157 | except RuntimeError, e: |
|---|
| 158 | sys.exit("Error configuring fedd: %s" % e.message) |
|---|
| 159 | else: |
|---|
| 160 | sys.exit("--configfile is required") |
|---|
| 161 | |
|---|
| 162 | SOAP_port = (opts.host, opts.port) |
|---|
| 163 | |
|---|
| 164 | if impl.cert_file == None: |
|---|
| 165 | sys.exit("Must supply certificate file (probably in config)") |
|---|
| 166 | |
|---|
| 167 | ctx = None |
|---|
| 168 | while ctx == None: |
|---|
| 169 | try: |
|---|
| 170 | ctx = fedd_ssl_context(impl.cert_file, impl.trusted_certs, |
|---|
| 171 | password=impl.cert_pwd) |
|---|
| 172 | except SSL.SSLError, e: |
|---|
| 173 | if str(e) != "bad decrypt" or impl.cert_pwd != None: |
|---|
| 174 | raise |
|---|
| 175 | |
|---|
| 176 | if opts.transport == "soap": |
|---|
| 177 | s = fedd_server(SOAP_port, fedd_soap_handler, ctx, impl) |
|---|
| 178 | elif opts.transport == "xmlrpc": |
|---|
| 179 | s = fedd_server(SOAP_port, fedd_xmlrpc_handler, ctx, impl) |
|---|
| 180 | else: |
|---|
| 181 | s = None |
|---|
| 182 | |
|---|
| 183 | s.serve_forever() |
|---|
