3 import sys, os, getopt, threading, logging, time, locale, collections
4 import ashd.proto, ashd.util, ashd.perf
11 out.write("usage: ashd-wsgi3 [-hAL] [-m PDM-SPEC] [-p MODPATH] [-l REQLIMIT] HANDLER-MODULE [ARGS...]\n")
14 modwsgi_compat = False
16 opts, args = getopt.getopt(sys.argv[1:], "+hALp:l:m:")
36 logging.basicConfig(format="ashd-wsgi3(%(name)s): %(levelname)s: %(message)s")
39 handlermod = __import__(args[0], fromlist = ["dummy"])
40 except ImportError as exc:
41 sys.stderr.write("ashd-wsgi3: handler %s not found: %s\n" % (args[0], exc.args[0]))
43 if not modwsgi_compat:
44 if not hasattr(handlermod, "wmain"):
45 sys.stderr.write("ashd-wsgi3: handler %s has no `wmain' function\n" % args[0])
47 handler = handlermod.wmain(*args[1:])
49 if not hasattr(handlermod, "application"):
50 sys.stderr.write("ashd-wsgi3: handler %s has no `application' object\n" % args[0])
52 handler = handlermod.application
54 class closed(IOError):
56 super().__init__("The client has closed the connection.")
61 return os.path.join(cwd, path)
73 if ord(b'0') <= url[i] <= ord(b'9'):
74 c |= (url[i] - ord(b'0')) << 4
75 elif ord(b'a') <= url[i] <= ord(b'f'):
76 c |= (url[i] - ord(b'a') + 10) << 4
77 elif ord(b'A') <= url[i] <= ord(b'F'):
78 c |= (url[i] - ord(b'A') + 10) << 4
80 raise ValueError("Illegal URL escape character")
81 if ord(b'0') <= url[i + 1] <= ord(b'9'):
82 c |= url[i + 1] - ord('0')
83 elif ord(b'a') <= url[i + 1] <= ord(b'f'):
84 c |= url[i + 1] - ord(b'a') + 10
85 elif ord(b'A') <= url[i + 1] <= ord(b'F'):
86 c |= url[i + 1] - ord(b'A') + 10
88 raise ValueError("Illegal URL escape character")
92 raise ValueError("Incomplete URL escape character")
99 env["wsgi.version"] = 1, 0
100 for key, val in req.headers:
101 env["HTTP_" + key.upper().replace(b"-", b"_").decode("latin-1")] = val.decode("latin-1")
102 env["SERVER_SOFTWARE"] = "ashd-wsgi/1"
103 env["GATEWAY_INTERFACE"] = "CGI/1.1"
104 env["SERVER_PROTOCOL"] = req.ver.decode("latin-1")
105 env["REQUEST_METHOD"] = req.method.decode("latin-1")
107 rawpi = unquoteurl(req.rest)
111 name, rest, pi = (v.decode("utf-8") for v in (req.url, req.rest, rawpi))
112 env["wsgi.uri_encoding"] = "utf-8"
113 except UnicodeError as exc:
114 name, rest, pi = (v.decode("latin-1") for v in (req.url, req.rest, rawpi))
115 env["wsgi.uri_encoding"] = "latin-1"
116 env["REQUEST_URI"] = name
119 env["QUERY_STRING"] = name[p + 1:]
122 env["QUERY_STRING"] = ""
123 if name[-len(rest):] == rest:
124 # This is the same hack used in call*cgi.
125 name = name[:-len(rest)]
127 # This seems to be normal CGI behavior, but see callcgi.c for
131 env["SCRIPT_NAME"] = name
132 env["PATH_INFO"] = pi
133 for src, tgt in [("HTTP_HOST", "SERVER_NAME"), ("HTTP_X_ASH_SERVER_PORT", "SERVER_PORT"),
134 ("HTTP_X_ASH_ADDRESS", "REMOTE_ADDR"), ("HTTP_CONTENT_TYPE", "CONTENT_TYPE"),
135 ("HTTP_CONTENT_LENGTH", "CONTENT_LENGTH"), ("HTTP_X_ASH_PROTOCOL", "wsgi.url_scheme")]:
136 if src in env: env[tgt] = env[src]
137 if "X-Ash-Protocol" in req and req["X-Ash-Protocol"] == b"https": env["HTTPS"] = "on"
138 if "X-Ash-File" in req: env["SCRIPT_FILENAME"] = absolutify(req["X-Ash-File"].decode(locale.getpreferredencoding()))
139 env["wsgi.input"] = req.sk
140 env["wsgi.errors"] = sys.stderr
141 env["wsgi.multithread"] = True
142 env["wsgi.multiprocess"] = False
143 env["wsgi.run_once"] = False
149 if isinstance(thing, collections.ByteString):
152 return str(thing).encode("latin-1")
157 raise Exception("Trying to write data before starting response.")
158 status, headers = resp
161 buf += b"HTTP/1.1 " + recode(status) + b"\n"
162 for nm, val in headers:
163 buf += recode(nm) + b": " + recode(val) + b"\n"
180 def startreq(status, headers, exc_info = None):
182 if exc_info: # Interesting, this...
187 exc_info = None # CPython GC bug?
189 raise Exception("Can only start responding once.")
190 resp[:] = status, headers
193 with ashd.perf.request(env) as reqevent:
194 respiter = handler(env, startreq)
197 for data in respiter:
204 if hasattr(respiter, "close"):
207 reqevent.response(resp)
209 flightlock = threading.Condition()
212 class reqthread(threading.Thread):
213 def __init__(self, req):
214 super().__init__(name = "Request handler")
223 while inflight >= reqlimit:
225 if time.time() - start > 10:
239 reqthread(req).start()
241 ashd.util.serveloop(handle)