package org.owasp.webgoat.lessons;

import org.owasp.webgoat.session.WebSession;

import org.apache.ecs.Element;
import org.apache.ecs.ElementContainer;
import org.apache.ecs.StringElement;
import org.apache.ecs.html.Div;
import org.apache.ecs.html.Form;
import org.apache.ecs.html.IMG;
import org.apache.ecs.html.Table;
import org.apache.ecs.html.TR;
import org.apache.ecs.html.TD;
import org.apache.ecs.html.Input;
import org.apache.ecs.html.BR;

import java.io.PrintWriter;
import java.util.List;
import java.util.ArrayList;

/*******************************************************************************
 * 
 * 
 * This file is part of WebGoat, an Open Web Application Security Project
 * utility. For details, please see http://www.owasp.org/
 * 
 * Copyright (c) 2002 - 2007 Bruce Mayhew
 * 
 * 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 2 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, write to the Free Software Foundation, Inc., 59 Temple
 * Place - Suite 330, Boston, MA 02111-1307, USA.
 * 
 * Getting Source ==============
 * 
 * Source for this application is maintained at code.google.com, a repository
 * for free software projects.
 * 
 * For details, please see http://code.google.com/p/webgoat/
 * 
 * @author     Sherif Koussa <a href="http://www.macadamian.com">Macadamian Technologies.</a>
 * @created    December 25, 2006
 */

public class JSONInjection extends LessonAdapter
{

    private final static Integer DEFAULT_RANKING = new Integer(30);

    private final static String TRAVEL_FROM = "travelFrom";

    private final static String TRAVEL_TO = "travelTo";

    private final static IMG MAC_LOGO = new IMG("images/logos/macadamian.gif").setAlt(
    "Macadamian Technologies").setBorder(0).setHspace(0).setVspace(0);

    public void handleRequest(WebSession s)
    {

	try
	{
	    if (s.getParser().getRawParameter("from", "").equals("ajax"))
	    {
		String lineSep = System.getProperty("line.separator");
		String jsonStr = "{"
			+ lineSep
			+ "\"From\": \"Boston\","
			+ lineSep
			+ "\"To\": \"Seattle\", "
			+ lineSep
			+ "\"flights\": ["
			+ lineSep
			+ "{\"stops\": \"0\", \"transit\" : \"N/A\", \"price\": \"$600\"},"
			+ lineSep
			+ "{\"stops\": \"2\", \"transit\" : \"Newark,Chicago\", \"price\": \"$300\"} "
			+ lineSep + "]" + lineSep + "}";
		s.getResponse().setContentType("text/html");
		s.getResponse().setHeader("Cache-Control", "no-cache");
		PrintWriter out = new PrintWriter(s.getResponse()
			.getOutputStream());
		out.print(jsonStr);
		out.flush();
		out.close();
		return;
	    }
	}
	catch (Exception ex)
	{
	    ex.printStackTrace();
	}

	Form form = new Form(getFormAction(), Form.POST).setName("form")
		.setEncType("");
	form.setOnSubmit("return check();");

	form.addElement(createContent(s));

	setContent(form);

    }


    /**
     * Description of the Method
     * 
     * @param s Current WebSession
     */

    protected Element createContent(WebSession s)
    {
	ElementContainer ec = new ElementContainer();
	String lineSep = System.getProperty("line.separator");
	String script = "<script>"
		+ lineSep
		+ "function getFlights() {"
		+ lineSep
		+ "var fromField = document.getElementById('"
		+ TRAVEL_FROM
		+ "');"
		+ lineSep
		+ "if (fromField.value.length < 3 || fromField.value!='BOS') { return; }"
		+ lineSep
		+ "var toField = document.getElementById('"
		+ TRAVEL_TO
		+ "');"
		+ lineSep
		+ "if (toField.value.length < 3 || toField.value!='SEA') { return; }"
		+ lineSep
		+ "var url = '" + getLink() 
		+ "&from=ajax&"
		+ TRAVEL_FROM
		+ "=' + encodeURIComponent(fromField.value) +"
		+ "'&"
		+ TRAVEL_TO
		+ "=' + encodeURIComponent(toField.value);"
		+ lineSep
		+ "if (typeof XMLHttpRequest != 'undefined') {"
		+ lineSep
		+ "req = new XMLHttpRequest();"
		+ lineSep
		+ "} else if (window.ActiveXObject) {"
		+ lineSep
		+ "req = new ActiveXObject('Microsoft.XMLHTTP');"
		+ lineSep
		+ "   }"
		+ lineSep
		+ "   req.open('GET', url, true);"
		+ lineSep
		+ "   req.onreadystatechange = callback;"
		+ lineSep
		+ "   req.send(null);"
		+ lineSep
		+ "}"
		+ lineSep
		+ "function callback() {"
		+ lineSep
		+ "    if (req.readyState == 4) { "
		+ lineSep
		+ "        if (req.status == 200) { "
		+ lineSep
		+ "                   var card = eval('(' + req.responseText + ')');"
		+ lineSep
		+ "			 var flightsDiv = document.getElementById('flightsDiv');"
		+ lineSep
		+ "				flightsDiv.innerHTML = '';"
		+ lineSep
		+ "				var strHTML='';"
		+ lineSep
		+ "				strHTML = '<tr><td>&nbsp;</td><td>No of Stops</td>';"
		+ lineSep
		+ "				strHTML = strHTML + '<td>Stops</td><td>Prices</td></tr>';"
		+ lineSep
		+ "			 for(var i=0; i<card.flights.length; i++){"
		+ lineSep
		+ "				var node = card.flights[i];"
		+ lineSep
		+ "				strHTML = strHTML + '<tr><td><input name=\"radio'+i+'\" type=\"radio\" id=\"radio'+i+'\"></td><td>';"
		+ lineSep
		+ "			    strHTML = strHTML + card.flights[i].stops + '</td><td>';"
		+ lineSep
		+ "			    strHTML = strHTML + card.flights[i].transit + '</td><td>';"
		+ lineSep
		+ "			    strHTML = strHTML + '<div name=\"priceID'+i+'\" id=\"priceID'+i+'\">' + card.flights[i].price + '</div></td></tr>';"
		+ lineSep
		+ "			 }"
		+ lineSep
		+ "				strHTML = '<table border=\"1\">' + strHTML + '</table>';"
		+ lineSep
		+ "               flightsDiv.innerHTML = strHTML;"
		+ lineSep
		+ "        }}}"
		+ lineSep
		+

		"function check(){"
		+ lineSep
		+ " if ( document.getElementById('radio0').checked  )"
		+ lineSep
		+ " { document.getElementById('price2Submit').value = document.getElementById('priceID0').innerHTML; return true;}"
		+ lineSep
		+ " else if ( document.getElementById('radio1').checked  )"
		+ lineSep
		+ " { document.getElementById('price2Submit').value = document.getElementById('priceID1').innerHTML; return true;}"
		+ lineSep + " else " + lineSep
		+ " { alert('Please choose one flight'); return false;}" + lineSep + "}"
		+ lineSep + "</script>" + lineSep;
	ec.addElement(new StringElement(script));
	Table t1 = new Table().setCellSpacing(0).setCellPadding(0).setBorder(0)
		.setWidth("90%").setAlign("center");

	TR tr = new TR();

	tr.addElement(new TD("From: "));
	Input in = new Input(Input.TEXT, TRAVEL_FROM, "");
	in.addAttribute("onkeyup", "getFlights();");
	in.addAttribute("id", TRAVEL_FROM);
	tr.addElement(new TD(in));

	t1.addElement(tr);

	tr = new TR();
	tr.addElement(new TD("To: "));
	in = new Input(Input.TEXT, TRAVEL_TO, "");
	in.addAttribute("onkeyup", "getFlights();");
	in.addAttribute("id", TRAVEL_TO);
	tr.addElement(new TD(in));

	t1.addElement(tr);
	ec.addElement(t1);

	ec.addElement(new BR());
	ec.addElement(new BR());
	Div div = new Div();
	div.addAttribute("name", "flightsDiv");
	div.addAttribute("id", "flightsDiv");
	ec.addElement(div);

	Input b = new Input();
	b.setType(Input.SUBMIT);
	b.setValue("Submit");
	b.setName("SUBMIT");
	ec.addElement(b);

	Input price2Submit = new Input();
	price2Submit.setType(Input.HIDDEN);
	price2Submit.setName("price2Submit");
	price2Submit.setValue("");
	price2Submit.addAttribute("id", "price2Submit");
	ec.addElement(price2Submit);
	if (s.getParser().getRawParameter("radio0", "").equals("on"))
	{
	    String price = s.getParser().getRawParameter("price2Submit", "");
	    price = price.replace("$", "");
	    if (Integer.parseInt(price) < 600)
	    {
		makeSuccess(s);
	    }
	    else
	    {
		s
			.setMessage("You are close, try to set the price for the non-stop flight to be less than $600");
	    }
	}
	return ec;
    }

    public Element getCredits()
    {
	return super.getCustomCredits("Created by Sherif Koussa", MAC_LOGO);
    }

    protected Category getDefaultCategory()
    {
	return Category.AJAX_SECURITY;
    }


    protected List<String> getHints(WebSession s)
    {
	List<String> hints = new ArrayList<String>();
	hints.add("JSON stands for JavaScript Object Notation.");
	hints.add("JSON is a way of representing data just like XML.");
	hints.add("The JSON payload is easily interceptable.");
	hints.add("Intercept the reply, change the $600 to $25.");
	return hints;

    }


    protected Integer getDefaultRanking()
    {
	return DEFAULT_RANKING;
    }


    /**
     *  Gets the title attribute of the HelloScreen object
     *
     * @return    The title value
     */
    public String getTitle()
    {
	return ("JSON Injection");
    }

}