handle decode exceptions in myhttp

meprajjwal · Oct 30, 2018 · 7072ea3 · 7072ea3
1 parent fe79a26
commit 7072ea3
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/src/wfuzz/myhttp.py b/src/wfuzz/myhttp.py
@@ -192,10 +192,13 @@ def _read_multi_stack(self):
                 # Parse response
                 buff_body, buff_header, res, poolid = c.response_queue
 
-                res.history.from_http_object(c, buff_header.getvalue(), buff_body.getvalue())
-
-                # reset type to result otherwise backfeed items will enter an infinite loop
-                self.pool_map[poolid]["queue"].put(res.update())
+                try:
+                    res.history.from_http_object(c, buff_header.getvalue(), buff_body.getvalue())
+                except Exception as e:
+                    self.pool_map[poolid]["queue"].put(res.update(exception=e))
+                else:
+                    # reset type to result otherwise backfeed items will enter an infinite loop
+                    self.pool_map[poolid]["queue"].put(res.update())
 
                 self.m.remove_handle(c)
                 self.freelist.put(c)