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