Use proper TCP context manager on aiohttp.request

y0k4i-1337 · May 23, 2024 · 77bf7b8 · 77bf7b8
1 parent b6fec67
commit 77bf7b8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/copy-as-python-aiohttp/src/main/java/burp/ReqParser.java b/copy-as-python-aiohttp/src/main/java/burp/ReqParser.java
@@ -177,9 +177,12 @@ private String requestToFunction(HttpRequest request, int baseIndent, boolean se
             sb.append(Utility.indent(baseIndent + 1) + "async with client.request(\n");
             sb.append(Utility.indent(baseIndent + 2) + " ssl=ssl,\n");
         } else {
+            sb.append(Utility.indent(baseIndent + 1)
+                    + "async with aiohttp.TCPConnector(ssl=ssl) as conn:\n");
+            baseIndent++;
             sb.append(Utility.indent(baseIndent + 1) + "async with aiohttp.request(\n");
             sb.append(
-                    Utility.indent(baseIndent + 2) + "connector=aiohttp.TCPConnector(ssl=ssl),\n");
+                    Utility.indent(baseIndent + 2) + "connector=conn,\n");
         }
         sb.append(Utility.indent(baseIndent + 2) + "method=method,\n");
         sb.append(Utility.indent(baseIndent + 2) + "url=url,\n");