burp.MSLTab Maven / Gradle / Ivy

Go to download
/**
 * Copyright (c) 2014 Netflix, Inc.  All rights reserved.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package burp;

import burp.msl.WiretapException;

import java.awt.*;

/**
 * User: skommidi
 * Date: 9/25/14
 */
public class MSLTab implements IMessageEditorTab {

    public MSLTab(IMessageEditorController controller, boolean editable, IBurpExtenderCallbacks callbacks, IExtensionHelpers helpers) {
        this.editable = editable;
        this.callbacks = callbacks;
        this.helpers = helpers;

        // create an instance of Burp's text editor
        txtInput = callbacks.createTextEditor();
        txtInput.setEditable(editable);

        this.mslHttpListener = new MSLHttpListener(this.callbacks, this.helpers);
    }

    @Override
    public String getTabCaption() {
        return "MSL Parse";
    }

    @Override
    public Component getUiComponent() {
        return txtInput.getComponent();
    }

    @Override
    public boolean isEnabled(byte[] content, boolean isRequest) {
        return true;
    }

    @Override
    public void setMessage(byte[] content, boolean isRequest) {
        if(content == null) {
            // clear our display
            txtInput.setText(null);
            txtInput.setEditable(false);
        } else {
            // TODO: fix setText
            String data = null;
            String body = mslHttpListener.getBody(content);
            try {
                data = mslHttpListener.processMslMessage(body);
            } catch (WiretapException e) {
                throw new RuntimeException(e.getMessage(), e);
            }
            txtInput.setText(data.getBytes());
            txtInput.setEditable(editable);
        }

        // remember the displayed content
        currentMessage = content;
    }

    @Override
    public byte[] getMessage() {
        // determine whether the user modified the deserialized data
        if (txtInput.isTextModified()) {
            // reserialize the data
            // TODO: fix the reserialization part
            return currentMessage;
        } else
            return currentMessage;
    }

    @Override
    public boolean isModified() {
        return txtInput.isTextModified();
    }

    @Override
    public byte[] getSelectedData() {
        return txtInput.getSelectedText();
    }

    private final IBurpExtenderCallbacks callbacks;
    private final IExtensionHelpers helpers;
    private final boolean editable;
    private final MSLHttpListener mslHttpListener;

    private ITextEditor txtInput;
    private byte[] currentMessage;
}