85 lines
3.1 KiB
Python
85 lines
3.1 KiB
Python
#!/usr/bin/env python3
|
|
"""Focused HTTP regression checks for marker-safe bodies and bounded responses."""
|
|
from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer
|
|
from pathlib import Path
|
|
import subprocess
|
|
import tempfile
|
|
import threading
|
|
|
|
ROOT = Path(__file__).resolve().parent.parent
|
|
MAX_HTTP_BYTES = 1024 * 1024
|
|
|
|
|
|
class Handler(BaseHTTPRequestHandler):
|
|
def do_GET(self):
|
|
if self.path == "/marker":
|
|
body = b"before\n__CLARO_HTTP_STATUS__999\nafter"
|
|
status = 200
|
|
elif self.path == "/large":
|
|
body = b"x" * (MAX_HTTP_BYTES + 1)
|
|
status = 200
|
|
else:
|
|
body = b"not found"
|
|
status = 404
|
|
self.send_response(status)
|
|
self.send_header("Content-Length", str(len(body)))
|
|
self.end_headers()
|
|
self.wfile.write(body)
|
|
|
|
def log_message(self, format, *args):
|
|
pass
|
|
|
|
|
|
def main() -> int:
|
|
with tempfile.TemporaryDirectory(prefix="claro-http-") as tmp:
|
|
tmp_path = Path(tmp)
|
|
script = tmp_path / "http.claro"
|
|
script.write_text(
|
|
'HTTP GET URL_MARKER AS body STATUS status\n'
|
|
'SAY body\n'
|
|
'SAY status\n'
|
|
'HTTP GET URL_LARGE AS ignored\n',
|
|
encoding="utf-8",
|
|
)
|
|
binary = tmp_path / "claro-http"
|
|
build = subprocess.run(
|
|
["gcc", "-std=c99", "-O0", "src/claro.c", "-o", str(binary), "-lm"],
|
|
cwd=ROOT, text=True, capture_output=True,
|
|
)
|
|
if build.returncode:
|
|
print(build.stderr, end="")
|
|
return build.returncode
|
|
server = ThreadingHTTPServer(("127.0.0.1", 0), Handler)
|
|
thread = threading.Thread(target=server.serve_forever, daemon=True)
|
|
thread.start()
|
|
try:
|
|
marker = f'"http://127.0.0.1:{server.server_port}/marker"'
|
|
large = f'"http://127.0.0.1:{server.server_port}/large"'
|
|
text = script.read_text(encoding="utf-8").replace("URL_MARKER", marker).replace("URL_LARGE", large)
|
|
script.write_text(text, encoding="utf-8")
|
|
run = subprocess.run([str(binary), str(script)], cwd=ROOT, text=True, capture_output=True)
|
|
finally:
|
|
server.shutdown()
|
|
server.server_close()
|
|
expected_body = "before\n__CLARO_HTTP_STATUS__999\nafter\n200\n"
|
|
if run.returncode == 0:
|
|
print("FAIL: oversized HTTP response was accepted")
|
|
print(run.stdout, end="")
|
|
return 1
|
|
if expected_body not in run.stdout:
|
|
print("FAIL: marker-like response body was altered")
|
|
print(run.stdout, end="")
|
|
print(run.stderr, end="")
|
|
return 1
|
|
if "HTTP response exceeds the safe size limit" not in run.stderr and "HTTP response exceeds the safe size limit" not in run.stdout:
|
|
print("FAIL: oversized response did not get a beginner-facing diagnostic")
|
|
print(run.stdout, end="")
|
|
print(run.stderr, end="")
|
|
return 1
|
|
print("PASS: HTTP status is separated from marker-like bodies and responses are bounded")
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|