package org.owasp.webgoat.lessons;

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

import org.apache.ecs.Element;
import org.apache.ecs.ElementContainer;
import org.apache.ecs.StringElement;
import org.apache.ecs.html.BR;
import org.apache.ecs.html.H1;
import org.apache.ecs.html.IMG;
import org.apache.ecs.html.Input;
import org.apache.ecs.html.TD;
import org.apache.ecs.html.TR;
import org.apache.ecs.html.Table;
import org.owasp.webgoat.session.WebSession;

/*******************************************************************************
 * 
 * 
 * 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    October 28, 2006
 */

public class DOMInjection extends LessonAdapter
{

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

    private final static String KEY = "key";

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

    protected Element createContent(WebSession s)
    {

	String key = "K1JFWP8BSO8HI52LNPQS8F5L01N";
	ElementContainer ec = new ElementContainer();

	try
	{
	    String userKey = s.getParser().getRawParameter(KEY, "");
	    String fromAJAX = s.getParser().getRawParameter("from", "");
	    if (fromAJAX.equalsIgnoreCase("ajax") && userKey.length() != 0
		    && userKey.equals(key))
	    {
		s.getResponse().setContentType("text/html");
		s.getResponse().setHeader("Cache-Control", "no-cache");
		PrintWriter out = new PrintWriter(s.getResponse()
			.getOutputStream());
		out.print("document.forms[0].SUBMIT.disabled = false;");
		out.flush();
		out.close();
		return ec;
	    }
	    if (s.getRequest().getMethod().equalsIgnoreCase("POST"))
	    {
		makeSuccess(s);
	    }
	}
	catch (Exception e)
	{
	    s.setMessage("Error generating " + this.getClass().getName());
	    e.printStackTrace();
	}

	String lineSep = System.getProperty("line.separator");
	String script = "<script>" + lineSep + "function validate() {"
		+ lineSep + "var keyField = document.getElementById('key');"
		+ lineSep + "var url = '" + getLink() 
		+ "&from=ajax&key=' + encodeURIComponent(keyField.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 message = req.responseText;" + lineSep
		+ "			 eval(message);" + lineSep + "        }}}" + lineSep
		+ "</script>" + lineSep;

	ec.addElement(new StringElement(script));
	ec.addElement(new BR().addElement(new H1()
		.addElement("Welcome to WebGoat Registration Page:")));
	ec
		.addElement(new BR()
			.addElement("Please enter the license key that was emailed to you to start using the application."));
	ec.addElement(new BR());
	ec.addElement(new BR());
	Table t1 = new Table().setCellSpacing(0).setCellPadding(0).setBorder(0)
		.setWidth("70%").setAlign("center");

	TR tr = new TR();
	tr.addElement(new TD(new StringElement("License Key: ")));

	Input input1 = new Input(Input.TEXT, KEY, "");
	input1.addAttribute("onkeyup", "validate();");
	tr.addElement(new TD(input1));
	t1.addElement(tr);

	tr = new TR();
	tr.addElement(new TD("&nbsp;").setColSpan(2));

	t1.addElement(tr);

	tr = new TR();
	Input b = new Input();
	b.setType(Input.SUBMIT);
	b.setValue("Activate!");
	b.setName("SUBMIT");
	b.setDisabled(true);
	tr.addElement(new TD("&nbsp;"));
	tr.addElement(new TD(b));

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

	return ec;
    }


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


    protected Category getDefaultCategory()
    {

	return Category.AJAX_SECURITY;
    }


    protected Integer getDefaultRanking()
    {

	return DEFAULT_RANKING;
    }


    protected List<String> getHints(WebSession s)
    {

	List<String> hints = new ArrayList<String>();
	hints.add("This page is using XMLHTTP to comunicate with the server.");
	hints.add("Try to find a way to inject the DOM to enable the Activate button.");
	hints.add("Intercept the reply and replace the body with document.forms[0].SUBMIT.disabled = false;");
	return hints;
    }


    public String getTitle()
    {
	return "DOM Injection";
    }

}