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 | ||
c9aa6b28 | 98 | reqhandler = ashd.serve.freethread() |
79662f8c | 99 | |
c9aa6b28 FT |
100 | class request(ashd.serve.wsgirequest): |
101 | def __init__(self, *, sk, **kw): | |
102 | super().__init__(**kw) | |
103 | self.sk = sk.dup() | |
104 | ||
105 | def mkenv(self): | |
106 | fsk = self.sk.dup().makefile("rwb") | |
107 | try: | |
108 | return mkenv(ashd.scgi.readhead(fsk), fsk) | |
109 | finally: | |
110 | fsk.close() | |
111 | ||
112 | def handlewsgi(self, env, startreq): | |
113 | return handler(env, startreq) | |
114 | ||
115 | def fileno(self): | |
116 | return self.sk.fileno() | |
79662f8c FT |
117 | |
118 | def writehead(self, status, headers): | |
c9aa6b28 FT |
119 | w = self.buffer.extend |
120 | w(b"Status: " + recode(status) + b"\n") | |
79662f8c | 121 | for nm, val in headers: |
c9aa6b28 FT |
122 | w(recode(nm) + b": " + recode(val) + b"\n") |
123 | w(b"\n") | |
79662f8c | 124 | |
c9aa6b28 | 125 | def flush(self): |
79662f8c | 126 | try: |
c9aa6b28 FT |
127 | ret = self.sk.send(self.buffer, socket.MSG_DONTWAIT) |
128 | self.buffer[:ret] = b"" | |
79662f8c FT |
129 | except IOError: |
130 | raise ashd.serve.closed() | |
131 | ||
c9aa6b28 FT |
132 | def close(self): |
133 | self.sk.close() | |
79662f8c FT |
134 | |
135 | while True: | |
136 | nsk, addr = sk.accept() | |
137 | try: | |
c9aa6b28 | 138 | reqhandler.handle(request(sk=nsk, handler=reqhandler)) |
79662f8c FT |
139 | finally: |
140 | nsk.close() |