ExitImprovement.java

/*******************************************************************************
 * gMix open source project - https://svs.informatik.uni-hamburg.de/gmix/
 * Copyright (C) 2014  SVS
 *
 * This program is free software: you can redistribute it and/or modify 
 * it under the terms of the GNU General Public License as published by 
 * the Free Software Foundation, either version 3 of the License, or 
 * (at your option) any later version.
 *  
 * This program is distributed in the hope that it will be useful, 
 * but WITHOUT ANY WARRANTY; without even the implied warranty of 
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the 
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License 
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *******************************************************************************/
/**
 * 
 */
package userGeneratedContent.testbedPlugIns.layerPlugIns.layer5application.httpPush_v0_001.improvement;

import java.util.Hashtable;

import userGeneratedContent.testbedPlugIns.layerPlugIns.layer5application.httpPush_v0_001.dataObjects.Connection;
import userGeneratedContent.testbedPlugIns.layerPlugIns.layer5application.httpPush_v0_001.helper.LocalClassLoader;

/**
 * @author bash
 * 
 */
public class ExitImprovement {

    private ExitImprovementInterface improvementPlugin;

    /**
     * @throws IllegalAccessException
     * @throws InstantiationException
     * @throws ClassNotFoundException
     * 
     */
    public ExitImprovement(String improvement) throws ClassNotFoundException, InstantiationException,
            IllegalAccessException {
        improvementPlugin = LocalClassLoader.newInstance("plugIns.layer5application.httpPush_v0_001.plugin", improvement);
        // improvementPlugin = new noImprovementPlugin();
    }

    /**
     * Returns if a message should be written out on a connection
     * @return true if it should be written out, false otherwise
     */
    public boolean writeOut() {
        return improvementPlugin.isExitWriteOut();
    }

    /**
     * Method to improve a header
     * @param message
     * @param connection
     * @return the improve message
     */
    public byte[] improveHeader(byte[] message, Connection connection) {

        return improvementPlugin.fromWebHeaderImprovement(message, connection);
    }

    /**
     * Method to improve a body
     * @param message
     * @param connection
     * @return improved body
     */
    public byte[] improveBody(byte[] message, Connection connection) {

        return improvementPlugin.fromWebBodyImprovement(message, connection);
    }

    /**
     * Method to remove an improvement from a header
     * @param message
     * @param connection
     * @return unimproved header
     */
    public byte[] unImproveHeader(byte[] message, Connection connection) {

        return improvementPlugin.toWebHeaderImprovement(message, connection);
    }

    /**
     * Remove improvement from body
     * @param message
     * @param connection
     * @return unimproved body
     */
    public byte[] unImproveBody(byte[] message, Connection connection) {

        return improvementPlugin.toWebBodyImprovement(message, connection);
    }

    /**
     * Checks via header if a body sent to the mix is improveable
     * @param headerTable
     * @param connection
     * @return true if body is improveable, false otherwise
     */
    public boolean isBodyToMixImproveable(Hashtable<String, String> headerTable, Connection connection) {

        return improvementPlugin.isWebBodyToMixImproveable(headerTable, connection);

    }

    /**
     * Checks via header if a body is improveable from the mix
     * @param headerTable
     * @param connection
     * @return true if body is improveable, false otherwise
     */
    public boolean isBodyFromMixImproveable(Hashtable<String, String> headerTable, Connection connection) {

        return improvementPlugin.isWebBodyFromMixImproveable(headerTable, connection);

    }
}