实现和御剑类似功能,测试后并不理想,即使开了多线程,仅供参考
#-coding: utf-8
import sys
import requests
from threading import Thread
def geturl(url):
r = requests.get(url, timeout=1)
status_code = r.status_code
if status_code == 200:
print url + ' 200 ok'
savetxt(url)
url = raw_input('请输入要扫描目录的网站\n'.decode('utf-8').encode('gb2312'))
with open(sys.argv[1], 'r') as f: #字典文件
for fi in f.readlines():
fi = fi.strip('\n')
fullurl = url + '/' + fi
t = Thread(target=geturl, args=(fullurl,))
t.start()
t.join()