def main():
url = ''
t = 30
r = 10
c = ''
r = False
d = 5
try:
opt,argv = getopt.getopt(sys.argv[1:],'hu:t:',['help','url'])
if opt == []:
print '\t[!] Please input the parameter.'
usage()
exit()
except getopt.GetoptError as e:
print e
usage()
exit(0)
for o,v in opt:
if o in ('-h','--help'):
usage()
exit(0)
if o in ('-u','--url'):
if v == '':
print 'url can\'t be null'
exit(0)
url.strip()
if v[0:7] != 'http://':
v = 'http://' + v
if v[-1] != '/':
v += '/'
url = v
if o in ('-t','--thread_num'):
if v.isdigit():
thread_num = int(v)
else :
print 'Thread_num is NOT a digit!'
exit(0)
if o in ('-c','--cookie'):
c = serializeCookie(v)
if o in ('-r','--retry'):
if v.isdigit():
retry = int(v)
else :
print 'retry times is NOT a digit!'
exit(0)
# if o in ('-R','--recursive'):
# r = True
# if o in ('-d','--depth') and v.isdigit():
print '[*] Start'
sTime = time.time()
# print str(sTime)
a = Core(url,t,c,r)
a.createThread()
a.getRes()
print '[*] Used ' + str(time.time()-sTime) + ' s'
