Webアプリケーション診断ツール「Burp Suite」のExtention紹介（二重登録エラー回避機能）

更新日：2024.06.06

こんにちは。牧田です。

Webアプリケーション診断において、Burp SuiteのMacro機能で対応できない案件がたまにあります。
例えば、会員登録の確定処理において、指定したメールアドレスが既に登録済みだとエラーになってしまうような場合です。
本来は正常に登録できるリクエストに対して診断を行いたいところですが、こういう場合はIntruderやScannerをかけても二重登録エラーになってしまうため、きちんと診断することができません。（よって、手動でひたすら頑張ることになります...）

今回はそんな時に使えるExtentionの紹介です。
このExtentionを使うことによって、指定したパラメータの値を任意のものに変更できるので、例えば、会員登録の確定処理で毎回メールアドレスを有効なものに変更しつつ、他のパラメータに対するスキャンができるようになります。

なお、Pythonで書くのは初めてなので、サンプルコードの紹介程度ということでよろしくお願いいたします。

ソースコード


# coding: UTF-8

from burp import IBurpExtender
from burp import IBurpExtenderCallbacks
from burp import ITab
from burp import IHttpListener
from burp import IParameter
from javax import swing

count = 0

class BurpExtender(IBurpExtender,ITab,IHttpListener,IParameter):

    def processHttpMessage(self, toolFlag, messageIsRequest, currentRequest):
        #Requestではない場合はスルー
        if not messageIsRequest:
            return

        #対象スコープではない場合はスルー
        if not self._callbacks.isInScope( currentRequest.getUrl() ):
            return

        #対象パラメータ名、置換後のリスト取得
        target = self._jTextIn.getText()
        words = self._jTextOut.getText().split("\n")
        while "" in words: words.remove("")

        #対象パラメータの値を置換
        if target != '':
            parameters = self._helpers.analyzeRequest(currentRequest.getRequest()).getParameters()
            for parameter in parameters:
                if target == parameter.getName():
                    current_value = parameter.getValue()

                    #リストが空の場合は数値を元の値の末尾に追加する
                    if not words:
                        global count
                        word = current_value + str(count)
                        count += 1

                    #リストの上から値を取得
                    else:
                        words = self._jTextOut.getText().split("\n")
                        word = words[0]
                        words.pop(0)
                        newwords = '\n'.join(words)
                        self._jTextOut.setText(newwords)

                    print "changed:" + word

                    #値を置換して送信する
                    newMessage = self._helpers.updateParameter(currentRequest.getRequest(), self._helpers.buildParameter(target,word,parameter.getType()))
                    currentRequest.setRequest(newMessage)

        return

    def registerExtenderCallbacks(self, callbacks):
        print "Parameter changer"

        self._callbacks = callbacks
        self._helpers = callbacks.getHelpers()

        callbacks.setExtensionName("Parameter changer")
        callbacks.registerHttpListener(self)

        #GUIの構築...最低限のところのみ
        self._jPanel = swing.JPanel()
        self._jPanel.setLayout(swing.BoxLayout(self._jPanel, swing.BoxLayout.Y_AXIS))
        self._jTextIn = swing.JTextField()
        self._jTextIn.setText("Write your parameter name")
        self._jScrollPaneIn = swing.JScrollPane(self._jTextIn)
        self._jTextOut = swing.JTextArea("Write your word list")
        self._jScrollPaneOut = swing.JScrollPane(self._jTextOut)
        self._jPanel.add(self._jScrollPaneIn)
        self._jPanel.add(self._jScrollPaneOut)

        callbacks.customizeUiComponent(self._jPanel)
        callbacks.addSuiteTab(self)
        return

    def getTabCaption(self):
        return "Parameter changer"

    def getUiComponent(self):
        return self._jPanel