org.owasp.jbrofuzz.graph.canvas.ResponseHeaderSizeChart.java Source code

Introduction

Here is the source code for org.owasp.jbrofuzz.graph.canvas.ResponseHeaderSizeChart.java

Source

/**
 * JbroFuzz 2.5
 *
 * JBroFuzz - A stateless network protocol fuzzer for web applications.
 * 
 * Copyright (C) 2007 - 2010 subere@uncon.org
 *
 * This file is part of JBroFuzz.
 * 
 * JBroFuzz 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.
 * 
 * JBroFuzz 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 JBroFuzz.  If not, see <http://www.gnu.org/licenses/>.
 * Alternatively, write to the Free Software Foundation, Inc., 51 
 * Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 * 
 * Verbatim copying and distribution of this entire program file is 
 * permitted in any medium without royalty provided this notice 
 * is preserved. 
 * 
 */
package org.owasp.jbrofuzz.graph.canvas;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;

import org.apache.commons.io.IOUtils;
import org.jfree.chart.ChartFactory;
import org.jfree.chart.ChartPanel;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.labels.StandardCategoryToolTipGenerator;
import org.jfree.chart.plot.Plot;
import org.jfree.chart.plot.PlotOrientation;
import org.jfree.chart.renderer.category.CategoryItemRenderer;
import org.jfree.data.category.DefaultCategoryDataset;
import org.jfree.ui.Align;
import org.owasp.jbrofuzz.version.ImageCreator;

public class ResponseHeaderSizeChart {

    /**
     * <p>
     * The maximum number of characters that will be read from the stream.
     * </p>
     * <p>
     * In this case, 1 Mb.
     * </p>
     */
    private static final int MAX_CHARS = 1048576;
    // The x-axis filenames
    private final String[] xData;
    // The y-axis data
    private final int[] yData;

    private final DefaultCategoryDataset dataset;

    private static final String END_SIGNATURE = "--jbrofuzz-->\n";

    public ResponseHeaderSizeChart() {

        this(0);

    }

    public ResponseHeaderSizeChart(final int size) {

        xData = new String[size];
        yData = new int[size];

        dataset = new DefaultCategoryDataset();

    }

    private int calculateValue(final File inputFile) {

        if (inputFile.isDirectory()) {
            return -1;
        }

        int headerLength = 0;

        BufferedReader inBuffReader = null;
        try {

            inBuffReader = new BufferedReader(new FileReader(inputFile));

            final StringBuffer one = new StringBuffer();
            int counter = 0;
            int got;
            while (((got = inBuffReader.read()) > 0) && (counter < MAX_CHARS)) {

                one.append((char) got);
                counter++;

            }
            inBuffReader.close();

            one.delete(0, one.indexOf(END_SIGNATURE) + END_SIGNATURE.length());

            headerLength = one.indexOf("\r\n\r\n");

            if (headerLength < 0) {

                headerLength = one.indexOf("\n\n");

            }

        } catch (final IOException e1) {

            return -2;

        } catch (final StringIndexOutOfBoundsException e2) {

            return -3;

        } catch (final NumberFormatException e3) {

            return -4;

        } finally {

            IOUtils.closeQuietly(inBuffReader);

        }

        return headerLength;
    }

    public void createFinalPlotCanvas() {

    }

    public ChartPanel getPlotCanvas() {

        final JFreeChart chart = ChartFactory.createBarChart("JBroFuzz Response Header Size Bar Chart", // chart title
                "File Name", // domain axis label
                "Response Header Size (bytes)", // range axis label
                dataset, // data
                PlotOrientation.VERTICAL, // orientation
                false, // include legend
                true, // tooltips?
                true // URLs?
        );

        final Plot plot = chart.getPlot();
        plot.setBackgroundImage(ImageCreator.IMG_OWASP_MED.getImage());
        plot.setBackgroundImageAlignment(Align.TOP_RIGHT);

        final CategoryItemRenderer renderer = chart.getCategoryPlot().getRenderer();
        renderer.setBaseToolTipGenerator(new StandardCategoryToolTipGenerator());

        return new ChartPanel(chart);

    }

    public void setValueAt(final int index, final File inputFile) {

        xData[index] = inputFile.getName();
        yData[index] = calculateValue(inputFile);

        dataset.addValue(yData[index], "Row 1", xData[index]);

    }

}