Commit | Line | Data |
---|---|---|
55fa3f63 | 1 | #!/usr/bin/python3 |
c06db49a | 2 | |
79662f8c | 3 | import sys, os, getopt, logging, collections |
c06db49a | 4 | import socket |
79662f8c | 5 | import ashd.scgi, ashd.serve |
c06db49a FT |
6 | |
7 | def usage(out): | |
78c8462c | 8 | out.write("usage: scgi-wsgi3 [-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-wsgi3(%(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"]) | |
55fa3f63 | 50 | except ImportError as exc: |
1f3d7aa3 | 51 | sys.stderr.write("scgi-wsgi3: handler %s not found: %s\n" % (args[0], exc.args[0])) |
c06db49a FT |
52 | sys.exit(1) |
53 | if not modwsgi_compat: | |
54 | if not hasattr(handlermod, "wmain"): | |
1f3d7aa3 | 55 | sys.stderr.write("scgi-wsgi3: handler %s has no `wmain' function\n" % args[0]) |
c06db49a | 56 | sys.exit(1) |
adb11d5f | 57 | handler = handlermod.wmain(*args[1:]) |
c06db49a FT |
58 | else: |
59 | if not hasattr(handlermod, "application"): | |
1f3d7aa3 | 60 | sys.stderr.write("scgi-wsgi3: handler %s has no `application' object\n" % args[0]) |
c06db49a FT |
61 | sys.exit(1) |
62 | handler = handlermod.application | |
63 | ||
79662f8c FT |
64 | def mkenv(head, sk): |
65 | try: | |
66 | env = ashd.scgi.decodehead(head, "utf-8") | |
67 | env["wsgi.uri_encoding"] = "utf-8" | |
68 | except UnicodeError: | |
69 | env = ashd.scgi.decodehead(head, "latin-1") | |
70 | env["wsgi.uri_encoding"] = "latin-1" | |
71 | env["wsgi.version"] = 1, 0 | |
72 | if "HTTP_X_ASH_PROTOCOL" in env: | |
73 | env["wsgi.url_scheme"] = env["HTTP_X_ASH_PROTOCOL"] | |
74 | elif "HTTPS" in env: | |
75 | env["wsgi.url_scheme"] = "https" | |
76 | else: | |
77 | env["wsgi.url_scheme"] = "http" | |
78 | env["wsgi.input"] = sk | |
79 | env["wsgi.errors"] = sys.stderr | |
80 | env["wsgi.multithread"] = True | |
81 | env["wsgi.multiprocess"] = False | |
82 | env["wsgi.run_once"] = False | |
83 | return env | |
84 | ||
85 | def recode(thing): | |
86 | if isinstance(thing, collections.ByteString): | |
87 | return thing | |
88 | else: | |
89 | return str(thing).encode("latin-1") | |
90 | ||
91 | class reqthread(ashd.serve.wsgithread): | |
92 | def __init__(self, sk): | |
93 | super().__init__() | |
94 | self.bsk = sk.dup() | |
95 | self.sk = self.bsk.makefile("rwb") | |
96 | ||
97 | def handlewsgi(self): | |
98 | return handler(self.env, self.startreq) | |
99 | ||
100 | def writehead(self, status, headers): | |
101 | buf = bytearray() | |
102 | buf += b"Status: " + recode(status) + b"\n" | |
103 | for nm, val in headers: | |
104 | buf += recode(nm) + b": " + recode(val) + b"\n" | |
105 | buf += b"\n" | |
106 | try: | |
107 | self.sk.write(buf) | |
108 | except IOError: | |
109 | raise ashd.serve.closed() | |
110 | ||
111 | def writedata(self, data): | |
112 | try: | |
113 | self.sk.write(data) | |
114 | self.sk.flush() | |
115 | except IOError: | |
116 | raise ashd.serve.closed() | |
117 | ||
118 | def handle(self): | |
119 | head = ashd.scgi.readhead(self.sk) | |
120 | self.env = mkenv(head, self.sk) | |
121 | super().handle() | |
122 | ||
123 | def run(self): | |
124 | try: | |
125 | super().run() | |
126 | finally: | |
127 | self.sk.close() | |
128 | self.bsk.close() | |
129 | ||
130 | while True: | |
131 | nsk, addr = sk.accept() | |
132 | try: | |
133 | reqthread(nsk).start() | |
134 | finally: | |
135 | nsk.close() |