CIRCL
/
lookyloo
mirror of https://github.com/CIRCL/lookyloo
2
0
Fork 0
Code Issues Releases Wiki Activity

fix: error in case of invalid proxy + username/password support

pull/223/head
felalex 2021-06-29 18:00:44 +02:00
parent d3e0b7f394
commit 6a7787d8af
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
website/web/__init__.py
View File

@ -12,7 +12,7 @@ import calendar
from typing import Optional, Dict, Any, Union, List
import logging
import hashlib
from urllib.parse import quote_plus, unquote_plus
from urllib.parse import quote_plus, unquote_plus, urlparse
import time
from flask import Flask, render_template, request, send_file, redirect, url_for, Response, flash, jsonify
@ -758,7 +758,17 @@ def capture_web():
capture_query['referer'] = request.form['referer']
if request.form.get('proxy'):
capture_query['proxy'] = request.form['proxy']
parsed_proxy = urlparse(request.form['proxy'])
if parsed_proxy.scheme and parsed_proxy.hostname and parsed_proxy.port:
if parsed_proxy.scheme in ['http', 'https', 'socks5']:
if (parsed_proxy.username and parsed_proxy.password) != (not parsed_proxy.username and not parsed_proxy.password):
capture_query['proxy'] = request.form['proxy']
else:
flash('You need to enter a username AND a password for your proxy.', 'error')
else:
flash('Proxy scheme not supported: must be http(s) or socks5.', 'error')
else:
flash('Invalid proxy: Check that you entered a scheme, a hostname and a port.', 'error')
perma_uuid = lookyloo.enqueue_capture(capture_query, source='web', user=user, authenticated=flask_login.current_user.is_authenticated)
time.sleep(30)