You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

76 lines
2.3 KiB

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import asyncio
import dataclasses
import socket
import traceback
import typing
@dataclasses.dataclass
class ScgiRequest:
headers: typing.Dict[bytes, bytes]
body: bytes
async def parse_scgi_request(reader: asyncio.StreamReader) -> ScgiRequest:
hlen = int((await reader.readuntil(b':'))[:-1])
header_raw = await reader.readexactly(hlen + 1)
assert len(header_raw) >= 16, "invalid request: too short (< 16)"
assert header_raw[-2:] == b'\0,', f"Invalid request: missing header/netstring terminator '\\x00,', got {header_raw[-2:]!r}"
header_list = header_raw[:-2].split(b'\0')
assert len(header_list) % 2 == 0, f"Invalid request: odd numbers of header entries (must be pairs), got {len(header_list)}"
assert header_list[0] == b'CONTENT_LENGTH', f"Invalid request: first header entry must be 'CONTENT_LENGTH', got {header_list[0]!r}"
clen = int(header_list[1])
headers = {}
i = 0
while i < len(header_list):
key = header_list[i]
value = header_list[i+1]
i += 2
assert not key in headers, f"Invalid request: duplicate header key {key!r}"
headers[key] = value
assert headers.get(b'SCGI') == b'1', "Invalid request: missing SCGI=1 header"
body = await reader.readexactly(clen)
return ScgiRequest(headers=headers, body=body)
async def handle_scgi(reader: asyncio.StreamReader, writer: asyncio.StreamWriter):
print(f"scgi-envcheck: Incoming connection", flush=True)
try:
req = await parse_scgi_request(reader)
envvar = req.headers[b'QUERY_STRING']
result = req.headers[envvar]
except KeyboardInterrupt:
raise
except Exception as e:
print(traceback.format_exc())
writer.write(b"Status: 500\r\nContent-Type: text/plain\r\n\r\n" + str(e).encode('utf-8'))
else:
writer.write(b"Status: 200\r\nContent-Type: text/plain\r\n\r\n" + result)
await writer.drain()
writer.close()
await writer.wait_closed()
async def main():
sock = socket.socket(fileno=0)
if sock.type == socket.AF_UNIX:
start_server = asyncio.start_unix_server
else:
start_server = asyncio.start_server
server = await start_server(handle_scgi, sock=sock, start_serving=False)
addr = server.sockets[0].getsockname()
print(f'Serving on {addr}', flush=True)
async with server:
await server.serve_forever()
try:
asyncio.run(main())
except KeyboardInterrupt:
pass