| 1 | #!/usr/bin/python |
| 2 | |
| 3 | import sys, os, getopt, logging |
| 4 | import socket |
| 5 | import ashd.scgi, ashd.serve |
| 6 | |
| 7 | def usage(out): |
| 8 | out.write("usage: scgi-wsgi [-hAL] [-p MODPATH] [-T [HOST:]PORT] HANDLER-MODULE [ARGS...]\n") |
| 9 | |
| 10 | sk = None |
| 11 | modwsgi_compat = False |
| 12 | setlog = True |
| 13 | opts, args = getopt.getopt(sys.argv[1:], "+hALp:T:") |
| 14 | for o, a in opts: |
| 15 | if o == "-h": |
| 16 | usage(sys.stdout) |
| 17 | sys.exit(0) |
| 18 | elif o == "-p": |
| 19 | sys.path.insert(0, a) |
| 20 | elif o == "-L": |
| 21 | setlog = False |
| 22 | elif o == "-T": |
| 23 | sk = socket.socket(socket.AF_INET, socket.SOCK_STREAM) |
| 24 | sk.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) |
| 25 | p = a.rfind(":") |
| 26 | if p < 0: |
| 27 | bindhost = "localhost" |
| 28 | bindport = int(a) |
| 29 | else: |
| 30 | bindhost = a[:p] |
| 31 | bindport = int(a[p + 1:]) |
| 32 | sk.bind((bindhost, bindport)) |
| 33 | sk.listen(32) |
| 34 | elif o == "-A": |
| 35 | modwsgi_compat = True |
| 36 | if len(args) < 1: |
| 37 | usage(sys.stderr) |
| 38 | sys.exit(1) |
| 39 | if setlog: |
| 40 | logging.basicConfig(format="scgi-wsgi(%(name)s): %(levelname)s: %(message)s") |
| 41 | |
| 42 | if sk is None: |
| 43 | # This is suboptimal, since the socket on stdin is not necessarily |
| 44 | # AF_UNIX, but Python does not seem to offer any way around it, |
| 45 | # that I can find. |
| 46 | sk = socket.fromfd(0, socket.AF_UNIX, socket.SOCK_STREAM) |
| 47 | |
| 48 | try: |
| 49 | handlermod = __import__(args[0], fromlist = ["dummy"]) |
| 50 | except ImportError, exc: |
| 51 | sys.stderr.write("scgi-wsgi: handler %s not found: %s\n" % (args[0], exc.message)) |
| 52 | sys.exit(1) |
| 53 | if not modwsgi_compat: |
| 54 | if not hasattr(handlermod, "wmain"): |
| 55 | sys.stderr.write("scgi-wsgi: handler %s has no `wmain' function\n" % args[0]) |
| 56 | sys.exit(1) |
| 57 | handler = handlermod.wmain(*args[1:]) |
| 58 | else: |
| 59 | if not hasattr(handlermod, "application"): |
| 60 | sys.stderr.write("scgi-wsgi: handler %s has no `application' object\n" % args[0]) |
| 61 | sys.exit(1) |
| 62 | handler = handlermod.application |
| 63 | |
| 64 | def mkenv(head, sk): |
| 65 | env = dict(head) |
| 66 | env["wsgi.version"] = 1, 0 |
| 67 | if "HTTP_X_ASH_PROTOCOL" in env: |
| 68 | env["wsgi.url_scheme"] = env["HTTP_X_ASH_PROTOCOL"] |
| 69 | elif "HTTPS" in env: |
| 70 | env["wsgi.url_scheme"] = "https" |
| 71 | else: |
| 72 | env["wsgi.url_scheme"] = "http" |
| 73 | env["wsgi.input"] = sk |
| 74 | env["wsgi.errors"] = sys.stderr |
| 75 | env["wsgi.multithread"] = True |
| 76 | env["wsgi.multiprocess"] = False |
| 77 | env["wsgi.run_once"] = False |
| 78 | return env |
| 79 | |
| 80 | class reqthread(ashd.serve.wsgithread): |
| 81 | def __init__(self, sk): |
| 82 | super(reqthread, self).__init__() |
| 83 | self.bsk = sk.dup() |
| 84 | self.sk = self.bsk.makefile("r+") |
| 85 | |
| 86 | def handlewsgi(self): |
| 87 | return handler(self.env, self.startreq) |
| 88 | |
| 89 | def writehead(self, status, headers): |
| 90 | try: |
| 91 | self.sk.write("Status: %s\n" % status) |
| 92 | for nm, val in headers: |
| 93 | self.sk.write("%s: %s\n" % (nm, val)) |
| 94 | self.sk.write("\n") |
| 95 | except IOError: |
| 96 | raise ashd.serve.closed() |
| 97 | |
| 98 | def writedata(self, data): |
| 99 | try: |
| 100 | self.sk.write(data) |
| 101 | self.sk.flush() |
| 102 | except IOError: |
| 103 | raise ashd.serve.closed() |
| 104 | |
| 105 | def handle(self): |
| 106 | head = ashd.scgi.readhead(self.sk) |
| 107 | self.env = mkenv(head, self.sk) |
| 108 | super(reqthread, self).handle() |
| 109 | |
| 110 | def run(self): |
| 111 | try: |
| 112 | super(reqthread, self).run() |
| 113 | finally: |
| 114 | self.sk.close() |
| 115 | self.bsk.close() |
| 116 | |
| 117 | while True: |
| 118 | nsk, addr = sk.accept() |
| 119 | try: |
| 120 | reqthread(nsk).start() |
| 121 | finally: |
| 122 | nsk.close() |