close httpool in fuzz

meprajjwal · Apr 26, 2019 · e25f418 · e25f418
1 parent b84f051
commit e25f418
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/docs/library/guide.rst b/docs/library/guide.rst
@@ -75,7 +75,6 @@ The FuzzSession object allows you to persist certain parameters across fuzzing s
     00060:  C=301      7 L        12 W          184 Ch        "admin"
     00183:  C=403     10 L        29 W          263 Ch        "cgi-bin"
     ...
-    >>> s.close()
 
 FuzzSession can also be used as context manager::
 
@@ -104,7 +103,6 @@ Generating a new payload and start fuzzing is really simple::
     00014:  C=404      7 L        12 W          168 Ch        "2"
     00015:  C=404      7 L        12 W          168 Ch        "3"
     00016:  C=404      7 L        12 W          168 Ch        "4"
-    >>> s.close()
 
 The get_payloads method can be used when various payloads are needed::
 
@@ -123,7 +121,6 @@ The get_payloads method can be used when various payloads are needed::
     00020:  C=404      7 L        12 W          168 Ch        "0 - b"
     00023:  C=404      7 L        12 W          168 Ch        "2 - a"
     00019:  C=404      7 L        12 W          168 Ch        "0 - a"
-    >>> s.close()
 
 Get session
 ===========
@@ -147,7 +144,6 @@ The get_session function generates a Wfuzz session object from the specified com
     00007:  C=404      7 L        12 W          168 Ch        "6"
     00009:  C=404      7 L        12 W          168 Ch        "8"
     00010:  C=404      7 L        12 W          168 Ch        "9"
-    >>> s.close()
 
 Interacting with the results
 ============================

diff --git a/src/wfuzz/options.py b/src/wfuzz/options.py
@@ -221,6 +221,10 @@ def fuzz(self, **kwargs):
                 fz.cancel_job()
                 self.stats.update(fz.genReq.stats)
 
+            if self.http_pool:
+                self.http_pool.deregister()
+                self.http_pool = None
+
     def get_payloads(self, iterator):
         self.data["dictio"] = iterator
 
@@ -303,3 +307,4 @@ def compile(self):
     def close(self):
         if self.http_pool:
             self.http_pool.deregister()
+            self.http_pool = None