Rewrite.py

from burp import IBurpExtender, IHttpListener
import json


class BurpExtender(IBurpExtender, IHttpListener):
    def registerExtenderCallbacks(self, callbacks):
        self.helpers = callbacks.getHelpers()
        callbacks.registerHttpListener(self)

    def processHttpMessage(self, toolFlag, messageIsRequest, message):
        if messageIsRequest:
            req_info = self.helpers.analyzeRequest(message)
            headers = req_info.getHeaders()
            if 'GET /rest/requests/detail' not in headers[0]:
                 return
            headers[0] = headers[0].replace('?id=', '/')
            message.setRequest(self.helpers.buildHttpMessage(headers, None))

        else:
            req_info = self.helpers.analyzeRequest(message)
            headers = req_info.getHeaders()
            if 'POST /rest/requests' not in headers[0]:
                 return
            resp_info = self.helpers.analyzeResponse(message.getResponse())
            headers = resp_info.getHeaders()
            json_str = self.helpers.bytesToString(message.getResponse()[resp_info.getBodyOffset():])
            json_data = json.loads(json_str)
            headers[-1] = 'Content-type: text/html'
            resp = '<form action="/rest/requests/detail" method="GET"><input type="hidden" name="id" value="%s"></form>\n\n' % json_data['ID']
            message.setResponse(self.helpers.buildHttpMessage(headers, self.helpers.stringToBytes(resp)))