simplify requests session argument

knownsec · Jan 6, 2024 · 3249bc3 · 3249bc3
1 parent 66c9ab3
commit 3249bc3
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/CONTRIBUTORS.md b/CONTRIBUTORS.md
@@ -61,4 +61,5 @@ HomerQing <https://github.com/HomerQing>
 * contributing to fix ipv6 compatibility issue in build_url
 
 Xxcdd <https://github.com/xxcdd>
-* Add support for requests session reuse
+* Add support for requests session reuse
+* Add support for web hook
diff --git a/pocsuite3/lib/core/settings.py b/pocsuite3/lib/core/settings.py
@@ -191,8 +191,8 @@
     "connect_back_host",
     "connect_back_port",
 
-    "requests-session-reuse",
-    "requests-session-reuse-num",
+    "session-reuse",
+    "session-reuse-num",
 
     # web hook
     "dingtalk-token",

diff --git a/pocsuite3/lib/parse/cmd.py b/pocsuite3/lib/parse/cmd.py
@@ -73,9 +73,9 @@ def cmd_line_parser(argv=None):
         request.add_argument("--delay", dest="delay", help="Delay between two request of one thread")
         request.add_argument("--headers", dest="headers", help="Extra headers (e.g. \"key1: value1\\nkey2: value2\")")
         request.add_argument("--http-debug", dest="http_debug", type=int, default=0, help="HTTP debug level (default 0)")
-        request.add_argument("--requests-session-reuse", dest="requests_session_reuse", action="store_true",
+        request.add_argument("--session-reuse", dest="requests_session_reuse", action="store_true",
                              help="Enable requests session reuse")
-        request.add_argument("--requests-session-reuse-num", type=int, dest="requests_session_reuse_num", default=10,
+        request.add_argument("--session-reuse-num", type=int, dest="requests_session_reuse_num", default=10,
                              help="Requests session reuse number")
 
         # Account options