python写目录爆破工具

先骂一下python，注释有中文也要报错，要先写encoding = utf-8
对url批量进行字典爆破

#!/usr/bin/python

def bruster(url)
	rep = request.get(url)
	rep_code = rep.status_code
	if(rep_code !=404):
		print "%d %s" % (rep_code,url)

if __name__ == "__main__":
	url = 'http://192.168.x.x/'
	filenames = open('dic.txt','r').readlines()#dic.txt为目录字典，生成filenames的列表形式。用readlines读的内容会具有换行
	for filename in filenames:
		filename = filename.replace('\n','')#去掉换行
		bruster(url+filename)

一个小御剑就写好了，但是这个御剑慢的一批，因为它是单线程的
本节重点：

多线程

线程也叫轻量级进程，是操作系统能够进行运算调度的最小单位，它被包涵在进程中
当一个进程启动后，会默认产生一个主线程，因为线程是程序执行流的最小单元，当设置多线程时，主线程会创建多个子线程

多线程库：threading

但是有一个问题，就是代码是同时在执行了，但是读取字典不能重复的读。这里就要用到python的队列了，先进先出使字典的空间变小

队列库：queue

q = queue.Queue()创建一个名为q的队列
q,put(argv)向队列添加一个元素
q.empty()判断队列是否为空
q.get()从队列中取出一个元素
q.qsize()返回队列大小

则上述代码改进如下：

import time
import queue
import threading

def bruster(url)
	while not q.empty():
		rep = request.get(url+q.get())#取出队列里的依次请求
	rep_code = rep.status_code
	if(rep_code !=404):
		print "%d %s" % (rep_code,url)

if __name__ == "__main__":
	target = 'http://192.168.x.x/'
	q = queue.Queue()
	start_time = time.time()#判断程序运行时间，看看多线程起用没
	filenames = open('./dic.txt','r').readlines()
	for filename in filenames:
		filename = filename.replace('\n','')
		q.put(filename)#将字典内的元素添加进q队列
	thread_list=[]#初始化线程
	for i in range(5):#初始化五个线程
		t= threading.Thread(target=bruster,args=(target,))#以target指定多线程运行的函数生成线程,arges指定传参（即bruster函数接收的参数）
		thread_list.append(t)#添加线程
	for t in thread_list:
		t.start()#启动线程运行
	
	print('main thread end!',threading.current_thread(),name)
	print('total time is',time.time()-start_time)

这个工具还不够智能，因为如果修改target或者字典路径还需要进入代码修改。我们将它改成接收外部传参的形式。就像sqlmap的–dump命令或者-u指定url一样

首先导入sys库：import sys
sys.argv[1] 接收外部第一个参数，比如python 1.py abc，即接收abc
同理sys.argv[2]``sys.argv[3]

最终版工具

接收外部参数，第一个为待扫url，第二个字典文本，第三个线程数

import time
import queue
import threading

def bruster(url)
	while not q.empty():
		rep = request.get(url+q.get())#取出队列里的依次请求
	rep_code = rep.status_code
	if(rep_code !=404):
		print "%d %s" % (rep_code,url)

if __name__ == "__main__":
	#target = 'http://192.168.x.x/'#不要了，换成读外部参数
	url = sys.argv[1]
	dict_txt = sys.argv[2]
	thread_number = sys.argv[3]

	q = queue.Queue()
	start_time = time.time()#判断程序运行时间，看看多线程起用没
	filenames = open('dict_txt','r').readlines()
	for filename in filenames:
		filename = filename.replace('\n','')
		q.put(filename)#将字典内的元素添加进q队列
	thread_list=[]#初始化线程
	for i in range(int(thread_number)):#初始化五个线程
		t= threading.Thread(target=bruster,args=(url,))#以target指定多线程运行的函数生成线程,arges指定传参（即bruster函数接收的参数）
		thread_list.append(t)#添加线程
	for t in thread_list:
		t.start()#启动线程运行
	
	print('main thread end!',threading.current_thread(),name)
	print('total time is',time.time()-start_time)