Signed-off-by: -LAN- <laipz8200@outlook.com>tags/0.13.0
| @@ -53,8 +53,6 @@ def make_request(method, url, max_retries=SSRF_DEFAULT_MAX_RETRIES, **kwargs): | |||
| response = client.request(method=method, url=url, **kwargs) | |||
| if response.status_code not in STATUS_FORCELIST: | |||
| if stream: | |||
| return response.iter_bytes() | |||
| return response | |||
| else: | |||
| logging.warning(f"Received status code {response.status_code} for URL {url} which is in the force list") | |||
| @@ -86,7 +86,7 @@ class WordExtractor(BaseExtractor): | |||
| image_count += 1 | |||
| if rel.is_external: | |||
| url = rel.reltype | |||
| response = ssrf_proxy.get(url, stream=True) | |||
| response = ssrf_proxy.get(url) | |||
| if response.status_code == 200: | |||
| image_ext = mimetypes.guess_extension(response.headers["Content-Type"]) | |||
| file_uuid = str(uuid.uuid4()) | |||