Commit c4379f63 authored by Nong Hoang Tu's avatar Nong Hoang Tu
Browse files

Add user-agent and proxy handle for session

parent c1fd13a4
......@@ -3,7 +3,7 @@ from vjyscan.cli.prints import *
class VJScan:
def __init__(self, verbose=False):
def __init__(self, ua, proxy, verbose):
self.print_vulnerable = print_vulnerable
self.print_found = print_found
if verbose:
......@@ -15,4 +15,6 @@ class VJScan:
self.print_not_vulnerable = dummy
self.print_not_found = dummy
self.http_client = requests.Session()
# TODO add proxy, cookie, agent handler
self.http_client.headers.update(ua)
if proxy:
self.http_client.proxies.update(proxy)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment