org.owasp.jbrofuzz.graph.canvas.StatusCodeChart Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jbrofuzz-encoder Show documentation
JBroFuzz is a stateless web application fuzzer for requests being made over HTTP and/or HTTPS. Its purpose is to provide a single, portable application that offers stable web protocol fuzzing capabilities. As a tool, it emerged from the needs of penetration testing.
There is a newer version: 2.5.1
Show newest version
/**
 * JbroFuzz 2.5
 *
 * JBroFuzz - A stateless network protocol fuzzer for web applications.
 * 
 * Copyright (C) 2007 - 2010 [email protected]
 *
 * 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 .
 * 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 java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.Map.Entry;

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.plot.Plot;
import org.jfree.data.general.DefaultPieDataset;
import org.jfree.ui.Align;
import org.owasp.jbrofuzz.version.ImageCreator;

public class StatusCodeChart {

	private static final int MAX_CHARS = 32;
	// The x-axis filenames
	private final String[] xData;
	// The y-axis data
	private final String[] yData;

	private DefaultPieDataset dataset;

	// Constants
	private static final String ERROR = "---";

	public StatusCodeChart() {

		this(0);

	}

	public StatusCodeChart(final int size) {

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

		dataset = new DefaultPieDataset();
		// Default + shape of the graph
		dataset.setValue("501", 25);
		dataset.setValue("302", 25);
		dataset.setValue("404", 25);
		dataset.setValue("200", 25);
	}

	private String calculateValue(final File inputFile) {

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

		String status = ERROR;

		BufferedReader inbuffReader = null;
		try {

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

			final StringBuffer one = new StringBuffer(MAX_CHARS);
			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("\n--\n") + 4);
			one.delete(one.indexOf("\n--"), one.length());

			// status = Integer.parseInt(one.toString());
			status = one.toString();

		} catch (final IOException e1) {

			// return -2;
			return ERROR;

		} catch (final StringIndexOutOfBoundsException e2) {

			// return -3;
			return ERROR;

		} catch (final NumberFormatException e3) {

			// return 0;
			return ERROR;

		} finally {

			IOUtils.closeQuietly(inbuffReader);

		}

		return status;
	}

	/**
	 * 
	 * Method for creating the final Chart.
	 * 
	 * 
	 * 
	 * @see #StatusCodeChart(int)
	 * @author [email protected]
	 * @version 1.5
	 * @since 1.2
	 */
	public void createFinalPlotCanvas() {

		final HashMap map = new HashMap();

		for (final String n : yData) {

			if (map.containsKey(n)) {
				map.put(n, map.get(n) + 1);
			}
			else {
				map.put(n, 1);
			}

			// if (n > 0) {
			//
			// final String value = "" + n;
			//
			// if (map.containsKey(value))
			// map.put(value, map.get(value) + 1);
			// else
			// map.put(value, 1);
			//
			// }
			// // Got response but no number identified
			// else if ((n == 0)) {
			// if (map.containsKey(ZERO))
			// map.put(ZERO, map.get(ZERO) + 1);
			// else
			// map.put(ZERO, 1);
			//
			// }
			// // Directory, IOException, String out of bounds
			// else {
			// if (map.containsKey(ERROR))
			// map.put(ERROR, map.get(ERROR) + 1);
			// else
			// map.put(ERROR, 1);
			//
			// }

		}

		dataset = new DefaultPieDataset();

		final Iterator> iter = map.entrySet().iterator();
		while ( iter.hasNext() ) {
			final Map.Entry entry = iter.next();

			dataset.setValue(entry.getKey(), (double) entry.getValue()
					/ (double) yData.length);

		}
	}

	public ChartPanel getPlotCanvas() {

		final JFreeChart chart = ChartFactory.createPieChart(
				"JBroFuzz Status Code Pie Chart", dataset, true, // legend?
				true, // tooltips?
				false // URLs?
		);

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

		return new ChartPanel(chart);

	}

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

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

	}

}