Commit | Line | Data |
---|---|---|
173e0e9e | 1 | #!/usr/bin/python |
c06db49a | 2 | |
78c8462c | 3 | import sys, os, getopt, logging |
c06db49a | 4 | import socket |
79662f8c | 5 | import ashd.scgi, ashd.serve |
c06db49a FT |
6 | |
7 | def usage(out): | |
78c8462c | 8 | out.write("usage: scgi-wsgi [-hAL] [-p MODPATH] [-T [HOST:]PORT] HANDLER-MODULE [ARGS...]\n") |
c06db49a FT |
9 | |
10 | sk = None | |
11 | modwsgi_compat = False | |
78c8462c FT |
12 | setlog = True |
13 | opts, args = getopt.getopt(sys.argv[1:], "+hALp:T:") | |
c06db49a FT |
14 | for o, a in opts: |
15 | if o == "-h": | |
16 | usage(sys.stdout) | |
17 | sys.exit(0) | |
18 | elif o == "-p": | |
e4769c65 | 19 | sys.path.insert(0, a) |
78c8462c FT |
20 | elif o == "-L": |
21 | setlog = False | |
c06db49a FT |
22 | elif o == "-T": |
23 | sk = socket.socket(socket.AF_INET, socket.SOCK_STREAM) | |
78c8462c | 24 | sk.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) |
c06db49a FT |
25 | p = a.rfind(":") |
26 | if p < 0: | |
075a379e | 27 | bindhost = "localhost" |
c06db49a FT |
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) | |
78c8462c | 39 | if setlog: |
b327e4c1 | 40 | logging.basicConfig(format="scgi-wsgi(%(name)s): %(levelname)s: %(message)s") |
c06db49a FT |
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"]) | |
173e0e9e FT |
50 | except ImportError, exc: |
51 | sys.stderr.write("scgi-wsgi: handler %s not found: %s\n" % (args[0], exc.message)) | |
c06db49a FT |
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) | |
adb11d5f | 57 | handler = handlermod.wmain(*args[1:]) |
c06db49a FT |
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 | ||
79662f8c FT |
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() |