2025-12-25 upload

venv/Lib/site-packages/mitmproxy/addons/keepserving.py

@@ -0,0 +1,54 @@
+from __future__ import annotations
+
+import asyncio
+
+from mitmproxy import ctx
+from mitmproxy.utils import asyncio_utils
+
+
+class KeepServing:
+    def load(self, loader):
+        loader.add_option(
+            "keepserving",
+            bool,
+            False,
+            """
+            Continue serving after client playback, server playback or file
+            read. This option is ignored by interactive tools, which always keep
+            serving.
+            """,
+        )
+
+    def keepgoing(self) -> bool:
+        # Checking for proxyserver.active_connections is important for server replay,
+        # the addon may report that replay is finished but not the entire response has been sent yet.
+        # (https://github.com/mitmproxy/mitmproxy/issues/7569)
+        checks = [
+            "readfile.reading",
+            "replay.client.count",
+            "replay.server.count",
+            "proxyserver.active_connections",
+        ]
+        return any([ctx.master.commands.call(c) for c in checks])
+
+    def shutdown(self):  # pragma: no cover
+        ctx.master.shutdown()
+
+    async def watch(self):
+        while True:
+            await asyncio.sleep(0.1)
+            if not self.keepgoing():
+                self.shutdown()
+
+    def running(self):
+        opts = [
+            ctx.options.client_replay,
+            ctx.options.server_replay,
+            ctx.options.rfile,
+        ]
+        if any(opts) and not ctx.options.keepserving:
+            asyncio_utils.create_task(
+                self.watch(),
+                name="keepserving",
+                keep_ref=True,
+            )