文章目录

  • python写目录爆破工具
  • 多线程
  • 最终版工具


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)

如果url是文本,还要重新读文件写进列表,循环依次检测,我不写了,懒,建议一次只查一个url。额,这种写不能断掉进程,因为Linux ctrl C默认断主进程,但是主线程执行完自己任务后就退出了,子线程会继续执行自己的任务

环境要求:python环境、pip、os、queue、requests、time、threading库

python写爆破ssh python 爆破_开发语言

可以看出很简单,大家可以很清楚的看出,目录爆破就是在比谁的字典强大,所以平时要自己收集一些字典啊