利用Python开发Burp Suite插件(二) - SQL注入检测插件开发指南

前言

本教程是Burp Suite插件开发系列的第二部分，将详细介绍如何使用Python开发一个简单的SQL注入检测插件。该插件会在每个参数后面添加单引号，用于检测SQL注入漏洞。我们将逐步讲解每个开发步骤和关键API的使用方法。

开发环境准备

在开始之前，请确保已完成以下准备工作：

1. 已安装Java运行环境(JRE)
2. 已安装Burp Suite(免费版或专业版)
3. 已配置Jython环境(参考系列教程第一部分)

核心接口类介绍

1. IBurpExtender接口

所有Burp Suite插件都必须实现这个接口：

• 类名必须为"BurpExtender"
• 必须提供默认构造器
• 需要实现registerExtenderCallbacks方法

from burp import IBurpExtender

class BurpExtender(IBurpExtender):
    def registerExtenderCallbacks(self, callbacks):
        self._callbacks = callbacks
        self._helpers = callbacks.getHelpers()
        callbacks.setExtensionName("SQL Injection Detector")

2. IIntruderPayloadGeneratorFactory接口

用于注册payload生成器工厂：

• 需要实现createNewInstance和getGeneratorName方法

from burp import IIntruderPayloadGeneratorFactory

class BurpExtender(IBurpExtender, IIntruderPayloadGeneratorFactory):
    def registerExtenderCallbacks(self, callbacks):
        # ...之前的代码...
        callbacks.registerIntruderPayloadGeneratorFactory(self)
    
    def createNewInstance(self, attack):
        return SQLiPayloadGenerator(self._helpers)
    
    def getGeneratorName(self):
        return "SQL Injection Payload Generator"

3. IIntruderPayloadGenerator接口

定义payload生成器的具体行为：

• 需要实现getNextPayload, hasMorePayloads和reset方法

from burp import IIntruderPayloadGenerator

class SQLiPayloadGenerator(IIntruderPayloadGenerator):
    def __init__(self, helpers):
        self._helpers = helpers
        self.payload_index = 0
        self.payloads = ["'", "\"", "'--", "\"--", "'#"]
    
    def getNextPayload(self, baseValue):
        payload = baseValue + self.payloads[self.payload_index]
        self.payload_index += 1
        return self._helpers.stringToBytes(payload)
    
    def hasMorePayloads(self):
        return self.payload_index < len(self.payloads)
    
    def reset(self):
        self.payload_index = 0

完整插件代码

from burp import IBurpExtender, IIntruderPayloadGeneratorFactory, IIntruderPayloadGenerator

class BurpExtender(IBurpExtender, IIntruderPayloadGeneratorFactory):
    def registerExtenderCallbacks(self, callbacks):
        self._callbacks = callbacks
        self._helpers = callbacks.getHelpers()
        callbacks.setExtensionName("SQL Injection Detector")
        callbacks.registerIntruderPayloadGeneratorFactory(self)
        
    def createNewInstance(self, attack):
        return SQLiPayloadGenerator(self._helpers)
    
    def getGeneratorName(self):
        return "SQL Injection Payload Generator"

class SQLiPayloadGenerator(IIntruderPayloadGenerator):
    def __init__(self, helpers):
        self._helpers = helpers
        self.payload_index = 0
        self.payloads = ["'", "\"", "'--", "\"--", "'#"]
    
    def getNextPayload(self, baseValue):
        payload = baseValue + self.payloads[self.payload_index]
        print("Current payload: " + payload)
        self.payload_index += 1
        return self._helpers.stringToBytes(payload)
    
    def hasMorePayloads(self):
        return self.payload_index < len(self.payloads)
    
    def reset(self):
        self.payload_index = 0

插件工作原理

1. 注册流程:

   • Burp Suite加载插件时调用registerExtenderCallbacks
   • 插件注册自己为Payload生成器工厂
   • 设置插件显示名称

2. Payload生成流程:

   • 当用户发起Intruder攻击时，Burp调用createNewInstance
   • 创建新的Payload生成器实例
   • 依次调用hasMorePayloads和getNextPayload获取payload
   • 所有payload用完后或用户停止攻击时结束

3. Payload设计:

   • 基础payload包括单引号、双引号及其组合
   • 可以扩展更多SQL注入测试字符

测试与使用

1. 将插件保存为.py文件并加载到Burp Suite
2. 在DVWA或其他测试环境中拦截SQL注入请求
3. 发送请求到Intruder模块
4. 在Payloads标签中选择"Extension-generated"作为Payload类型
5. 选择我们开发的"SQL Injection Payload Generator"
6. 开始攻击并观察响应中的SQL错误信息

扩展与改进建议

1. 增加更多Payload:

   self.payloads = [
       "'", "\"", 
       "'--", "\"--", 
       "'#", "' OR 1=1--", 
       "\" OR \"\"=\"", 
       "' OR 'a'='a"
   ]
   

2. 添加错误检测逻辑:

   • 可以扩展插件自动检测响应中的SQL错误信息

3. 支持更多攻击类型:

   • 除了Intruder，还可以实现Scanner检查等功能

4. 添加配置界面:

   • 使用ITab接口添加配置界面，让用户自定义Payload

总结

本教程详细介绍了如何使用Python开发Burp Suite的SQL注入检测插件，重点讲解了三个核心接口的使用方法。通过这个简单的示例，你可以进一步扩展开发更复杂的安全测试插件。关键点包括：

1. 必须实现IBurpExtender接口
2. Payload生成器工厂和生成器的关系
3. Payload的设计和生成流程
4. 与Burp Suite其他组件的交互方式

希望这篇教程能帮助你理解Burp Suite插件开发的基本原理，并为开发更强大的安全测试工具打下基础。