Consolidate.py

from burp import IBurpExtender, IScannerCheck, IScanIssue


class BurpExtender(IBurpExtender):
    def registerExtenderCallbacks(self, callbacks):
        callbacks.registerScannerCheck(ScanCheck(callbacks))


class ScanCheck(IScannerCheck):
    def __init__(self, callbacks):
        self.callbacks = callbacks

    def doPassiveScan(self, base):
        self.callbacks.printOutput("PASSIVE!")
        reqInfo = self.callbacks.getHelpers().analyzeRequest(base.getHttpService(), base.getRequest())
        return [CustomIssue(reqInfo.getUrl(), base.getHttpService())]

    def doActiveScan(self, baseRequestResponse, insertionPoint):
        return []

    def consolidateDuplicateIssues(self, existing, new):
        self.callbacks.printOutput("Consolidate!")
        return 1


class CustomIssue(IScanIssue):
    def __init__(self, url, httpService):
        self.xurl = url
        self.xhttpService = httpService

    def getUrl(self):
        return self.xurl

    def getIssueName(self):
        return "Custom issue"

    def getIssueType(self):
        return 0x0000

    def getSeverity(self):
        return "Low"

    def getConfidence(self):
        return "Firm"

    def getIssueBackground(self):
        return "Custom issue"

    def getRemediationBackground(self):
        return "Panic"

    def getIssueDetail(self):
        return "Custom issue"

    def getRemediationDetail(self):
        return "Run in circles, scream and shout"

    def getHttpMessages(self):
        return []

    def getHttpService(self):
        return self.xhttpService