2 Commits

Author SHA1 Message Date
klemek 04360b42d8 chore: release 1.3.1
Python Lint CI / ruff (push) Successful in 2m18s
Docker CI / docker-build (push) Successful in 3m2s
Python Lint CI / ruff-format-check (push) Successful in 1m31s
Python Lint CI / ty (push) Successful in 2m17s
Python Test CI / coverage (push) Successful in 2m5s
2026-05-06 16:01:26 +02:00
klemek 4edcc6acc7 fix: http/1.1 and force close connection 2026-05-06 16:01:15 +02:00
6 changed files with 23 additions and 8 deletions
+1 -1
View File
@@ -1,6 +1,6 @@
[project] [project]
name = "stapler" name = "stapler"
version = "1.3.0" version = "1.3.1"
description = "Static pages as simple as a gzip file" description = "Static pages as simple as a gzip file"
requires-python = ">=3.14" requires-python = ">=3.14"
dependencies = [ dependencies = [
+1
View File
@@ -200,6 +200,7 @@ class CertManager:
cert_file, cert_file,
key_file, key_file,
) )
new_context.set_alpn_protocols(["http/1.1"])
socket.context = new_context socket.context = new_context
except Exception: except Exception:
self.logger.exception("Could not create HTTPS context for %s", host) self.logger.exception("Could not create HTTPS context for %s", host)
+12 -2
View File
@@ -27,6 +27,7 @@ if typing.TYPE_CHECKING:
class BaseHandler(abc.ABC, http.server.BaseHTTPRequestHandler): class BaseHandler(abc.ABC, http.server.BaseHTTPRequestHandler):
timeout = 10 timeout = 10
protocol_version = "HTTP/1.1"
REQUEST_COUNT = 0 REQUEST_COUNT = 0
@typing.override @typing.override
@@ -111,6 +112,8 @@ class BaseHandler(abc.ABC, http.server.BaseHTTPRequestHandler):
@typing.override @typing.override
def log_request(self, code: str = "?", size: str = "-") -> None: # ty:ignore[invalid-method-override] # pragma: no cover def log_request(self, code: str = "?", size: str = "-") -> None: # ty:ignore[invalid-method-override] # pragma: no cover
if isinstance(code, http.HTTPStatus): if isinstance(code, http.HTTPStatus):
code = code.value
if isinstance(code, int):
color = logs.TermColor.RED color = logs.TermColor.RED
if 100 <= code < 200: if 100 <= code < 200:
color = logs.TermColor.CYAN color = logs.TermColor.CYAN
@@ -120,7 +123,7 @@ class BaseHandler(abc.ABC, http.server.BaseHTTPRequestHandler):
color = logs.TermColor.BLUE color = logs.TermColor.BLUE
elif 400 <= code < 500: elif 400 <= code < 500:
color = logs.TermColor.YELLOW color = logs.TermColor.YELLOW
code = color + str(code.value) + logs.TermColor.RESET code = color + str(code) + logs.TermColor.RESET
if size == "" and self.out_size > 0: if size == "" and self.out_size > 0:
size = str(self.out_size) size = str(self.out_size)
args = ( args = (
@@ -193,6 +196,7 @@ class BaseHandler(abc.ABC, http.server.BaseHTTPRequestHandler):
headers=headers, headers=headers,
allow_redirects=False, allow_redirects=False,
timeout=480, timeout=480,
stream=False,
) )
except Exception as e: except Exception as e:
self.send_error( self.send_error(
@@ -385,6 +389,7 @@ class RequestHandler(http.server.SimpleHTTPRequestHandler, BaseHandler):
self._pre_log_request() self._pre_log_request()
if not self._proxy_or_redirect(): if not self._proxy_or_redirect():
super().do_HEAD() super().do_HEAD()
self.close_connection = True
@typing.override @typing.override
def do_GET(self) -> None: def do_GET(self) -> None:
@@ -394,7 +399,9 @@ class RequestHandler(http.server.SimpleHTTPRequestHandler, BaseHandler):
return None return None
if self.path == "/" and self.host == self.default_host: if self.path == "/" and self.host == self.default_host:
return self.send_basic_body(self.server_signature()) return self.send_basic_body(self.server_signature())
return super().do_GET() super().do_GET()
self.close_connection = True
return None
def do_PUT(self) -> None: def do_PUT(self) -> None:
with self.handle_errors(): with self.handle_errors():
@@ -620,16 +627,19 @@ class RequestHandler(http.server.SimpleHTTPRequestHandler, BaseHandler):
class UpgradeHandler(RequestHandler): class UpgradeHandler(RequestHandler):
protocol_version = "HTTP/1.0"
server_version = "StaplerUpgradeServer/" + PKG_VERSION server_version = "StaplerUpgradeServer/" + PKG_VERSION
def do_HEAD(self) -> None: def do_HEAD(self) -> None:
with self.handle_errors(): with self.handle_errors():
self._pre_log_request() self._pre_log_request()
self.send_redirect(f"https://{self.host}{self.path}") self.send_redirect(f"https://{self.host}{self.path}")
self.close_connection = True
def do_GET(self) -> None: def do_GET(self) -> None:
with self.handle_errors(): with self.handle_errors():
if self.path.startswith(self.CERTBOT_CHALLENGE_PATH): if self.path.startswith(self.CERTBOT_CHALLENGE_PATH):
super().do_GET() super().do_GET()
self.close_connection = True
else: else:
self.do_HEAD() self.do_HEAD()
+2 -4
View File
@@ -29,7 +29,6 @@ class StaplerServer:
"logger", "logger",
"params", "params",
"registry", "registry",
"server",
"token_manager", "token_manager",
] ]
@@ -41,7 +40,6 @@ class StaplerServer:
self.token_manager: TokenManager = TokenManager(params, self.registry) self.token_manager: TokenManager = TokenManager(params, self.registry)
self.data_dir: DataDir = DataDir(params.data_dir) self.data_dir: DataDir = DataDir(params.data_dir)
self.default_host: str = params.host.split(":", maxsplit=2)[0] self.default_host: str = params.host.split(":", maxsplit=2)[0]
self.server: http.server.ThreadingHTTPServer | None = None
def __get_all_hosts(self) -> list[str]: def __get_all_hosts(self) -> list[str]:
return [self.default_host, *self.registry.get_hosts()] return [self.default_host, *self.registry.get_hosts()]
@@ -131,7 +129,7 @@ class StaplerServer:
for line in STAPLER_ASCII.split("\n"): for line in STAPLER_ASCII.split("\n"):
self.logger.debug(line.ljust(36)) self.logger.debug(line.ljust(36))
self.__startup() self.__startup()
self.server = self.__create_base_server() base_server = self.__create_base_server()
upgrade_server = self.__start_upgrade_server() if self.params.https else None upgrade_server = self.__start_upgrade_server() if self.params.https else None
self.logger.info( self.logger.info(
"Server up and ready on %s://%s", "Server up and ready on %s://%s",
@@ -140,7 +138,7 @@ class StaplerServer:
) )
self.__start_background_tasks() self.__start_background_tasks()
with contextlib.suppress(KeyboardInterrupt): with contextlib.suppress(KeyboardInterrupt):
self.server.serve_forever() base_server.serve_forever()
self.logger.info("Shutting down...") self.logger.info("Shutting down...")
if upgrade_server is not None: if upgrade_server is not None:
upgrade_server.shutdown() upgrade_server.shutdown()
+6
View File
@@ -859,6 +859,7 @@ class TestRequestHandler(BaseHandlerTestCase):
}, },
"allow_redirects": False, "allow_redirects": False,
"timeout": 480, "timeout": 480,
"stream": False,
}, },
), ),
self.expects_status_only(handler, 200, "OK"), self.expects_status_only(handler, 200, "OK"),
@@ -903,6 +904,7 @@ class TestRequestHandler(BaseHandlerTestCase):
}, },
"allow_redirects": False, "allow_redirects": False,
"timeout": 480, "timeout": 480,
"stream": False,
}, },
), ),
self.expects_status_only(handler, 200, "OK"), self.expects_status_only(handler, 200, "OK"),
@@ -945,6 +947,7 @@ class TestRequestHandler(BaseHandlerTestCase):
}, },
"allow_redirects": False, "allow_redirects": False,
"timeout": 480, "timeout": 480,
"stream": False,
}, },
), ),
self.expects_basic_body(handler, "hello", message="OK"), self.expects_basic_body(handler, "hello", message="OK"),
@@ -979,6 +982,7 @@ class TestRequestHandler(BaseHandlerTestCase):
}, },
"allow_redirects": False, "allow_redirects": False,
"timeout": 480, "timeout": 480,
"stream": False,
}, },
) as request_mock, ) as request_mock,
self.expects_status_only( self.expects_status_only(
@@ -1022,6 +1026,7 @@ class TestRequestHandler(BaseHandlerTestCase):
}, },
"allow_redirects": False, "allow_redirects": False,
"timeout": 480, "timeout": 480,
"stream": False,
}, },
), ),
self.expects_status_only(handler, 200, "OK"), self.expects_status_only(handler, 200, "OK"),
@@ -1062,6 +1067,7 @@ class TestRequestHandler(BaseHandlerTestCase):
}, },
"allow_redirects": False, "allow_redirects": False,
"timeout": 480, "timeout": 480,
"stream": False,
}, },
), ),
self.expects_status_only(handler, 200, "OK"), self.expects_status_only(handler, 200, "OK"),
Generated
+1 -1
View File
@@ -212,7 +212,7 @@ wheels = [
[[package]] [[package]]
name = "stapler" name = "stapler"
version = "1.3.0" version = "1.3.1"
source = { editable = "." } source = { editable = "." }
dependencies = [ dependencies = [
{ name = "requests" }, { name = "requests" },