package org.owasp.webgoat.lessons;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
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.HR;
import org.apache.ecs.html.TD;
import org.apache.ecs.html.TR;
import org.apache.ecs.html.Table;
import org.owasp.webgoat.session.ECSFactory;
import org.owasp.webgoat.session.WebSession;
import org.owasp.webgoat.util.WebGoatI18N;


/***************************************************************************************************
 * 
 * 
 * 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 Bruce Mayhew <a href="http://code.google.com/p/webgoat">WebGoat</a>
 * @created October 28, 2003
 */
public class PathBasedAccessControl extends LessonAdapter
{

	private final static String FILE = "File";

	/**
	 * Description of the Method
	 * 
	 * @param s
	 *            Description of the Parameter
	 * @return Description of the Return Value
	 */
	protected Element createContent(WebSession s)
	{
		ElementContainer ec = new ElementContainer();

		try
		{
			String dir = s.getContext().getRealPath("/lesson_plans/English");
			File d = new File(dir);

			Table t = new Table().setCellSpacing(0).setCellPadding(2).setWidth("90%").setAlign("center");

			if (s.isColor())
			{
				t.setBorder(1);
			}

			String[] list = d.list();
			String listing = " <p><B>"+WebGoatI18N.get("CurrentDirectory")+"</B> " + Encoding.urlDecode(dir)
					+ "<br><br>"+WebGoatI18N.get("ChooseFileToView")+"</p>";

			TR tr = new TR();
			tr.addElement(new TD().setColSpan(2).addElement(new StringElement(listing)));
			t.addElement(tr);

			tr = new TR();
			tr.addElement(new TD().setWidth("35%").addElement(ECSFactory.makePulldown(FILE, list, "", 15)));
			tr.addElement(new TD().addElement(ECSFactory.makeButton(WebGoatI18N.get("ViewFile"))));
			t.addElement(tr);

			ec.addElement(t);

			// FIXME: would be cool to allow encodings here -- hex, percent,
			// url, etc...
			String file = s.getParser().getRawParameter(FILE, "");

			// defuse file searching
			boolean illegalCommand = getWebgoatContext().isDefuseOSCommands();
			if (getWebgoatContext().isDefuseOSCommands())
			{
				// allow them to look at any file in the webgoat hierachy. Don't
				// allow them
				// to look about the webgoat root, except to see the LICENSE
				// file
				if (upDirCount(file) == 3 && !file.endsWith("LICENSE"))
				{
					s.setMessage(WebGoatI18N.get("AccessDenied"));
					s.setMessage(WebGoatI18N.get("ItAppears1"));
				}
				else if (upDirCount(file) > 3)
				{
					s.setMessage(WebGoatI18N.get("AccessDenied"));
					s.setMessage(WebGoatI18N.get("ItAppears2"));
				}
				else
				{
					illegalCommand = false;
				}
			}

			// Using the URI supports encoding of the data.
			// We could force the user to use encoded '/'s == %2f to make the lesson more difficult.
			// We url Encode our dir name to avoid problems with special characters in our own path.
			// File f = new File( new URI("file:///" +
			// Encoding.urlEncode(dir).replaceAll("\\\\","/") + "/" +
			// file.replaceAll("\\\\","/")) );
			File f = new File((dir + "\\" + file).replaceAll("\\\\", "/"));

			if (s.isDebug())
			{

				s.setMessage(WebGoatI18N.get("File") + file);
				s.setMessage(WebGoatI18N.get("Dir")+ dir);
				// s.setMessage("File URI: " + "file:///" +
				// (Encoding.urlEncode(dir) + "\\" +
				// Encoding.urlEncode(file)).replaceAll("\\\\","/"));
				s.setMessage(WebGoatI18N.get("IsFile")+ f.isFile());
				s.setMessage(WebGoatI18N.get("Exists") + f.exists());
			}
			if (!illegalCommand)
			{
				if (f.isFile() && f.exists())
				{
					// Don't set completion if they are listing files in the
					// directory listing we gave them.
					if (upDirCount(file) >= 1)
					{
						s.setMessage(WebGoatI18N.get("CongratsAccessToFileAllowed"));
						s.setMessage(" ==> " + Encoding.urlDecode(f.getCanonicalPath()));
						makeSuccess(s);
					}
					else
					{
						s.setMessage(WebGoatI18N.get("FileInAllowedDirectory"));
						s.setMessage(" ==> " + Encoding.urlDecode(f.getCanonicalPath()));
					}
				}
				else if (file != null && file.length() != 0)
				{
					s
							.setMessage(WebGoatI18N.get("AccessToFileDenied1") + Encoding.urlDecode(f.getCanonicalPath())
									+  WebGoatI18N.get("AccessToFileDenied2"));
				}
				else
				{
					// do nothing, probably entry screen
				}

				try
				{
					// Show them the file
					// Strip out some of the extra html from the "help" file
					ec.addElement(new BR());
					ec.addElement(new BR());
					ec.addElement(new HR().setWidth("100%"));
					ec.addElement(WebGoatI18N.get("ViewingFile")+ f.getCanonicalPath());
					ec.addElement(new HR().setWidth("100%"));
					if (f.length() > 80000) { throw new Exception(WebGoatI18N.get("FileTooLarge")); }
					String fileData = getFileText(new BufferedReader(new FileReader(f)), false);
					if (fileData.indexOf(0x00) != -1) { throw new Exception(WebGoatI18N.get("FileBinary")); }
					ec.addElement(new StringElement(fileData.replaceAll(System.getProperty("line.separator"), "<br>")
							.replaceAll("(?s)<!DOCTYPE.*/head>", "").replaceAll("<br><br>", "<br>")
							.replaceAll("<br>\\s<br>", "<br>").replaceAll("<\\?", "&lt;").replaceAll("<(r|u|t)",
																										"&lt;$1")));
				} catch (Exception e)
				{
					ec.addElement(new BR());
					ec.addElement(WebGoatI18N.get("TheFollowingError"));
					ec.addElement(e.getMessage());
				}
			}
		} catch (Exception e)
		{
			s.setMessage(WebGoatI18N.get("ErrorGenerating")+ this.getClass().getName());
			e.printStackTrace();
		}

		return (ec);
	}

	private int upDirCount(String fileName)
	{
		int count = 0;
		int startIndex = fileName.indexOf("..");
		while (startIndex != -1)
		{
			count++;
			startIndex = fileName.indexOf("..", startIndex + 1);
		}
		return count;
	}

	/**
	 * DOCUMENT ME!
	 * 
	 * @return DOCUMENT ME!
	 */
	protected Category getDefaultCategory()
	{
		return Category.ACCESS_CONTROL;
	}

	/**
	 * Gets the hints attribute of the AccessControlScreen object
	 * 
	 * @return The hints value
	 */
	protected List<String> getHints(WebSession s)
	{
		List<String> hints = new ArrayList<String>();
		hints.add(WebGoatI18N.get("PathBasedAccessControlHint1"));
		hints.add(WebGoatI18N.get("PathBasedAccessControlHint2"));
		hints.add(WebGoatI18N.get("PathBasedAccessControlHint3"));
		hints.add(WebGoatI18N.get("PathBasedAccessControlHint4"));
		
		return hints;
	}

	/**
	 * Gets the instructions attribute of the WeakAccessControl object
	 * 
	 * @return The instructions value
	 */
	public String getInstructions(WebSession s)
	{
		String instructions = WebGoatI18N.get("PathBasedAccessControlInstr1")+ s.getUserName() + WebGoatI18N.get("PathBasedAccessControlInstr2");

		return (instructions);
	}

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

	protected Integer getDefaultRanking()
	{
		return DEFAULT_RANKING;
	}

	/**
	 * Gets the title attribute of the AccessControlScreen object
	 * 
	 * @return The title value
	 */
	public String getTitle()
	{
		return ("Bypass a Path Based Access Control Scheme");
	}
}