bp 插件临时代码记录

package burp;

import java.io.PrintWriter;
import java.util.List;
//import org.apache.http.client.methods.HttpPost;

public class BurpExtender implements IBurpExtender, IHttpListener,
        IProxyListener, IScannerListener, IExtensionStateListener {
    private IBurpExtenderCallbacks callbacks;
    private PrintWriter stdout;

    //
    // implement IBurpExtender
    //

    @Override
    public void registerExtenderCallbacks(IBurpExtenderCallbacks callbacks) {
        // keep a reference to our callbacks object
        this.callbacks = callbacks;

        // set our extension name
        callbacks.setExtensionName("Event listeners");

        // obtain our output stream
        stdout = new PrintWriter(callbacks.getStdout(), true);

        // register ourselves as an HTTP listener
        callbacks.registerHttpListener(this);

        // register ourselves as a Proxy listener
        callbacks.registerProxyListener(this);

        // register ourselves as a Scanner listener
        callbacks.registerScannerListener(this);

        // register ourselves as an extension state listener
        callbacks.registerExtensionStateListener(this);
    }

    //
    // implement IHttpListener
    //

    public String[] getHeaderNameValue(String header) {

        String arr[] = {"", ""};
        int index = header.indexOf(":");
        if (index ==-1){
            return arr;
        }
        arr[0] = header.substring(0, index);
        arr[1] = header.substring(index + 1);

        return arr;

    }


    @Override
    public void processHttpMessage(int toolFlag, boolean messageIsRequest, IHttpRequestResponse messageInfo) {

        if (!messageIsRequest) {

            IExtensionHelpers helpers = this.callbacks.getHelpers();
            IRequestInfo info = helpers.analyzeRequest(messageInfo.getRequest());
            List<String> headers = info.getHeaders();


            for (int i = 0; i < headers.size(); i++) {

//stdout.println(headers.get(i));
                String header[] = getHeaderNameValue(headers.get(i));
                stdout.println("header name : " + header[0] + " header value : " + header[1]);
            }



        }

    }

    //
    // implement IProxyListener
    //

    @Override
    public void processProxyMessage(boolean messageIsRequest, IInterceptedProxyMessage message) {
        stdout.println(
                (messageIsRequest ? "Proxy request to " : "Proxy response from ") +
                        message.getMessageInfo().getHttpService());
    }

    //
    // implement IScannerListener
    //

    @Override
    public void newScanIssue(IScanIssue issue) {
        stdout.println("New scan issue: " + issue.getIssueName());
    }

    //
    // implement IExtensionStateListener
    //

    @Override
    public void extensionUnloaded() {
        stdout.println("Extension was unloaded");
    }
}