def gen_traffic(self, url):
domain = get_domain_from_url(url)
# add cookie to DEFAULT_HEADER
cookie = get_cookie(domain)
self.DEFAULT_HEADER['Cookie'] = cookie
# add referer
self.DEFAULT_HEADER['Referer'] = 'https"//' + domain + '/'
request = HttpRequest(method='GET', url=url, headers=self.DEFAULT_HEADER, body='')
req = urllib2.Request(url=url, headers=self.DEFAULT_HEADER)
with gevent.Timeout(10, False)as t:
try:
resp = urllib2.urlopen(req)
except urllib2.URLError, e:
REQUEST_ERROR.append(('gen_traffic()', url, e.reason))
except CertificateError:
REQUEST_ERROR.append(('gen_traffic()', url, 'ssl.CertificateError'))
# default 2 if use browser
num = 2
if args.url:
num = 1
if args.process:
num = args.process
if args.coroutine:
coroutine = args.coroutine
if args.cookie:
from cookie import save_cookie_ip, is_ip
if file:
with open(file) as f:
scope_url = f.readline().strip()
elif url:
scope_url = url
domain = get_domain_from_url(scope_url)
if is_ip(scope_url):
save_cookie_ip(args.cookie, domain)
else:
from cookie import save_cookie
save_cookie(args.cookie, domain)
if url or file or burp or args.id or args.filter:
if args.id:
id = args.id
if not Engine.is_scanned(id):
print 'Task %s not found,exit.'
exit(0)
else:
id = gen_id()
engine = Engine(id=id,
url=url,