diff --git a/stapler/handlers.py b/stapler/handlers.py index 438260a..40456c4 100644 --- a/stapler/handlers.py +++ b/stapler/handlers.py @@ -126,6 +126,7 @@ class BaseHandler(abc.ABC, http.server.BaseHTTPRequestHandler): self.send_response(code, message) self.send_header("Content-Type", f"{content_type}; charset=UTF-8") self.send_header("Content-Length", str(len(encoded))) + self.send_header("Connection", "close") self.end_headers() self.wfile.write(encoded) self.close_connection = True @@ -140,6 +141,7 @@ class BaseHandler(abc.ABC, http.server.BaseHTTPRequestHandler): headers = {} self.send_response(code, message) self.send_header("Content-Length", "0") + self.send_header("Connection", "close") for header, value in headers.items(): self.send_header(header, value) self.end_headers() @@ -188,9 +190,11 @@ class BaseHandler(abc.ABC, http.server.BaseHTTPRequestHandler): "transfer-encoding", "server", "date", + "connection", ]: self.send_header(header, value.replace(target_host, self.host)) self.send_header("Content-Length", str(out_size := len(response.content))) + self.send_header("Connection", "close") self.end_headers() if out_size > 0: self.wfile.write(response.content)