| 1 | import sys |
| 2 | import threading |
| 3 | |
| 4 | class protoerr(Exception): |
| 5 | pass |
| 6 | |
| 7 | def readns(sk): |
| 8 | hln = 0 |
| 9 | while True: |
| 10 | c = sk.read(1) |
| 11 | if c == ':': |
| 12 | break |
| 13 | elif c >= '0' or c <= '9': |
| 14 | hln = (hln * 10) + (ord(c) - ord('0')) |
| 15 | else: |
| 16 | raise protoerr, "Invalid netstring length byte: " + c |
| 17 | ret = sk.read(hln) |
| 18 | if sk.read(1) != ',': |
| 19 | raise protoerr, "Non-terminated netstring" |
| 20 | return ret |
| 21 | |
| 22 | def readhead(sk): |
| 23 | parts = readns(sk).split('\0')[:-1] |
| 24 | if len(parts) % 2 != 0: |
| 25 | raise protoerr, "Malformed headers" |
| 26 | ret = {} |
| 27 | i = 0 |
| 28 | while i < len(parts): |
| 29 | ret[parts[i]] = parts[i + 1] |
| 30 | i += 2 |
| 31 | return ret |
| 32 | |
| 33 | class reqthread(threading.Thread): |
| 34 | def __init__(self, sk, handler): |
| 35 | super(reqthread, self).__init__(name = "SCGI request handler") |
| 36 | self.sk = sk.dup().makefile("r+") |
| 37 | self.handler = handler |
| 38 | |
| 39 | def run(self): |
| 40 | try: |
| 41 | head = readhead(self.sk) |
| 42 | self.handler(head, self.sk) |
| 43 | finally: |
| 44 | self.sk.close() |
| 45 | |
| 46 | def handlescgi(sk, handler): |
| 47 | t = reqthread(sk, handler) |
| 48 | t.start() |
| 49 | |
| 50 | def servescgi(socket, handler): |
| 51 | while True: |
| 52 | nsk, addr = socket.accept() |
| 53 | try: |
| 54 | handlescgi(nsk, handler) |
| 55 | finally: |
| 56 | nsk.close() |
| 57 | |
| 58 | def wrapwsgi(handler): |
| 59 | def handle(head, sk): |
| 60 | env = dict(head) |
| 61 | env["wsgi.version"] = 1, 0 |
| 62 | if "HTTP_X_ASH_PROTOCOL" in env: |
| 63 | env["wsgi.url_scheme"] = env["HTTP_X_ASH_PROTOCOL"] |
| 64 | elif "HTTPS" in env: |
| 65 | env["wsgi.url_scheme"] = "https" |
| 66 | else: |
| 67 | env["wsgi.url_scheme"] = "http" |
| 68 | env["wsgi.input"] = sk |
| 69 | env["wsgi.errors"] = sys.stderr |
| 70 | env["wsgi.multithread"] = True |
| 71 | env["wsgi.multiprocess"] = False |
| 72 | env["wsgi.run_once"] = False |
| 73 | |
| 74 | resp = [] |
| 75 | respsent = [] |
| 76 | |
| 77 | def flushreq(): |
| 78 | if not respsent: |
| 79 | if not resp: |
| 80 | raise Exception, "Trying to write data before starting response." |
| 81 | status, headers = resp |
| 82 | respsent[:] = [True] |
| 83 | sk.write("Status: %s\n" % status) |
| 84 | for nm, val in headers: |
| 85 | sk.write("%s: %s\n" % (nm, val)) |
| 86 | sk.write("\n") |
| 87 | |
| 88 | def write(data): |
| 89 | if not data: |
| 90 | return |
| 91 | flushreq() |
| 92 | sk.write(data) |
| 93 | sk.flush() |
| 94 | |
| 95 | def startreq(status, headers, exc_info = None): |
| 96 | if resp: |
| 97 | if exc_info: # Interesting, this... |
| 98 | try: |
| 99 | if respsent: |
| 100 | raise exc_info[0], exc_info[1], exc_info[2] |
| 101 | finally: |
| 102 | exc_info = None # CPython GC bug? |
| 103 | else: |
| 104 | raise Exception, "Can only start responding once." |
| 105 | resp[:] = status, headers |
| 106 | return write |
| 107 | |
| 108 | respiter = handler(env, startreq) |
| 109 | try: |
| 110 | for data in respiter: |
| 111 | write(data) |
| 112 | if resp: |
| 113 | flushresp() |
| 114 | finally: |
| 115 | if hasattr(respiter, "close"): |
| 116 | respiter.close() |
| 117 | return handle |