package org.owasp.webgoat.session;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;

import org.owasp.webgoat.lessons.AbstractLesson;


/**
 * Copyright (c) 2002 Free Software Foundation developed under the custody of
 * the Open Web Application Security Project (http://www.owasp.org) This
 * software package org.owasp.webgoat.is published by OWASP under the GPL. You should read and
 * accept the LICENSE before you use, modify and/or redistribute this
 * software.
 *
 * @author Jeff Williams <a href="http://www.aspectsecurity.com">Aspect Security</a>
 */
public class CreateDB
{
    /**
     * The main program for the AccessSqlInjection class
     *
     * @param args The command line arguments
     */
    public static void main(String[] args)
    {	
   
    	CreateDB db = new CreateDB();
    	Connection connection = null;
    	
    	try
    	{
            Class.forName("sun.jdbc.odbc.JdbcOdbcDriver").newInstance();
    	}
    	catch ( Exception e )
    	{
    		System.out.println("Failed to load DB driver");
    		e.printStackTrace();
    	}

    	try
    	{

            connection = DriverManager.getConnection("jdbc:odbc:;DRIVER=Microsoft Access Driver (*.mdb);DBQ=c:/webgoat.mdb;PWD=webgoat", "webgoat", "webgoat");
            db.makeDB(connection);
    	}
    	catch ( Exception e )
    	{
    		System.out.println("Driver Manager failed!");
    		e.printStackTrace();
    	}
    	
    	/**
    	 * getAllEmployees
    	 */
    	String query = "SELECT userid,first_name,last_name FROM employee WHERE userid in (SELECT employee_id FROM ownership WHERE employer_id = 101)";
    	
		try
		{
			Statement answer_statement = connection.createStatement( ResultSet.TYPE_SCROLL_INSENSITIVE, ResultSet.CONCUR_READ_ONLY );
			ResultSet answer_results = answer_statement.executeQuery( query );
			answer_results.first();
			int employeeId = answer_results.getInt("userid");
			String firstName = answer_results.getString("first_name");
			String lastName = answer_results.getString("last_name");
			System.out.println("Query 1 Results: " + firstName + " " + lastName + " " + employeeId);
		}
		catch ( SQLException sqle )
		{
			sqle.printStackTrace();
		}
		
		/**
		 * isAllowed
		 */

		query = "SELECT * FROM auth WHERE role in (SELECT role FROM roles WHERE userid = 101) and functionid = 113";
		
		try
		{
			Statement answer_statement = connection.createStatement( ResultSet.TYPE_SCROLL_INSENSITIVE, ResultSet.CONCUR_READ_ONLY );
			ResultSet answer_results = answer_statement.executeQuery( query );
			boolean allowed = answer_results.first();
			//boolean allowed = answer_results.next();
			
			if(allowed)
				System.out.println("User is allowed");
			else
				System.out.println("User is NOT allowed");
		}
		catch ( SQLException sqle )
		{
			sqle.printStackTrace();
		}
    }


    /**
     * Description of the Method
     *
     * @param connection Description of the Parameter
     *
     * @exception SQLException Description of the Exception
     */
    private void createMessageTable(Connection connection) throws SQLException
    {
        Statement statement = connection.createStatement();

        // Drop admin user table
        try
        {
            String dropTable = "DROP TABLE messages";
            statement.executeUpdate(dropTable);
        }
        catch (SQLException e)
        {
            System.out.println("Error dropping message database");
        }

        // Create the new table
        try
        {
            String createTableStatement = "CREATE TABLE messages (" 
            			+ "num int not null," 
            			+ "title varchar(50)," 
            			+ "message varchar(200)," 
            			+ "user_name varchar(50) not null "
            			+ ")";
            statement.executeUpdate(createTableStatement);
        }
        catch (SQLException e)
        {
            System.out.println("Error creating message database");
            e.printStackTrace();
        }
    }


    /**
     * Description of the Method
     *
     * @param connection Description of the Parameter
     *
     * @exception SQLException Description of the Exception
     */
    private void createProductTable(Connection connection) throws SQLException
    {
        Statement statement = connection.createStatement();

        // Drop admin user table
        try
        {
            String dropTable = "DROP TABLE product_system_data";
            statement.executeUpdate(dropTable);
        }
        catch (SQLException e)
        {
            System.out.println("Error dropping product database");
        }

        // Create the new table
        try
        {
            String createTableStatement = "CREATE TABLE product_system_data (" 
            			+ "productid varchar(6) not null primary key," 
            			+ "product_name varchar(20)," 
            			+ "price varchar(10)" + ")";
            statement.executeUpdate(createTableStatement);
        }
        catch (SQLException e)
        {
            System.out.println("Error creating product database");
            e.printStackTrace();
        }

        // Populate
        String insertData1 = "INSERT INTO product_system_data VALUES ('32226','Dog Bone','$1.99')";
        String insertData2 = "INSERT INTO product_system_data VALUES ('35632','DVD Player','$214.99')";
        String insertData3 = "INSERT INTO product_system_data VALUES ('24569','60 GB Hard Drive','$149.99')";
        String insertData4 = "INSERT INTO product_system_data VALUES ('56970','80 GB Hard Drive','$179.99')";
        String insertData5 = "INSERT INTO product_system_data VALUES ('14365','56 inch HDTV','$6999.99')";
        statement.executeUpdate(insertData1);
        statement.executeUpdate(insertData2);
        statement.executeUpdate(insertData3);
        statement.executeUpdate(insertData4);
        statement.executeUpdate(insertData5);
    }


    /**
     * Description of the Method
     *
     * @param connection Description of the Parameter
     *
     * @exception SQLException Description of the Exception
     */
    private void createUserAdminTable(Connection connection) throws SQLException
    {
        Statement statement = connection.createStatement();

        // Drop admin user table
        try
        {
            String dropTable = "DROP TABLE user_system_data";
            statement.executeUpdate(dropTable);
        }
        catch (SQLException e)
        {
            System.out.println("Error dropping user admin database");
        }

        // Create the new table
        try
        {
            String createTableStatement = "CREATE TABLE user_system_data (" 
            			+ "userid varchar(5) not null primary key," 
            			+ "user_name varchar(12)," 
            			+ "password varchar(10)," 
            			+ "cookie varchar(30)" 
            			+ ")";
            statement.executeUpdate(createTableStatement);
        }
        catch (SQLException e)
        {
            System.out.println("Error creating user admin database");
            e.printStackTrace();
        }

        // Populate
        String insertData1 = "INSERT INTO user_system_data VALUES ('101','jsnow','passwd1', '')";
        String insertData2 = "INSERT INTO user_system_data VALUES ('102','jdoe','passwd2', '')";
        String insertData3 = "INSERT INTO user_system_data VALUES ('103','jplane','passwd3', '')";
        String insertData4 = "INSERT INTO user_system_data VALUES ('104','jeff','jeff', '')";
        String insertData5 = "INSERT INTO user_system_data VALUES ('105','dave','dave', '')";
        statement.executeUpdate(insertData1);
        statement.executeUpdate(insertData2);
        statement.executeUpdate(insertData3);
        statement.executeUpdate(insertData4);
        statement.executeUpdate(insertData5);
    }


    /**
     * Description of the Method
     *
     * @param connection Description of the Parameter
     *
     * @exception SQLException Description of the Exception
     */
    private void createUserDataTable(Connection connection) throws SQLException
    {
        Statement statement = connection.createStatement();

        // Delete table if there is one
        try
        {
            String dropTable = "DROP TABLE user_data";
            statement.executeUpdate(dropTable);
        }
        catch (SQLException e)
        {
            System.out.println("Error dropping user database");
        }

        // Create the new table
        try
        {
            String createTableStatement = "CREATE TABLE user_data (" 
				+ "userid int not null," 
				+ "first_name varchar(20)," 
				+ "last_name varchar(20)," 
				+ "cc_number varchar(30)," 
				+ "cc_type varchar(10)," 
				+ "cookie varchar(20)," 
				+ "login_count int"
				+ ")";
            statement.executeUpdate(createTableStatement);
        }
        catch (SQLException e)
        {
            System.out.println("Error creating user database");
            e.printStackTrace();
        }

        // Populate it
        String insertData1 = "INSERT INTO user_data VALUES (101,'Joe','Snow','987654321','VISA',' ',0)";
        String insertData2 = "INSERT INTO user_data VALUES (101,'Joe','Snow','2234200065411','MC',' ',0)";
        String insertData3 = "INSERT INTO user_data VALUES (102,'John','Smith','2435600002222','MC',' ',0)";
        String insertData4 = "INSERT INTO user_data VALUES (102,'John','Smith','4352209902222','AMEX',' ',0)";
        String insertData5 = "INSERT INTO user_data VALUES (103,'Jane','Plane','123456789','MC',' ',0)";
        String insertData6 = "INSERT INTO user_data VALUES (103,'Jane','Plane','333498703333','AMEX',' ',0)";
        String insertData7 = "INSERT INTO user_data VALUES (10312,'Jolly','Hershey','176896789','MC',' ',0)";
        String insertData8 = "INSERT INTO user_data VALUES (10312,'Jolly','Hershey','333300003333','AMEX',' ',0)";
        String insertData9 = "INSERT INTO user_data VALUES (10323,'Grumpy','White','673834489','MC',' ',0)";
        String insertData10 = "INSERT INTO user_data VALUES (10323,'Grumpy','White','33413003333','AMEX',' ',0)";
        String insertData11 = "INSERT INTO user_data VALUES (15603,'Peter','Sand','123609789','MC',' ',0)";
        String insertData12 = "INSERT INTO user_data VALUES (15603,'Peter','Sand','338893453333','AMEX',' ',0)";
        String insertData13 = "INSERT INTO user_data VALUES (15613,'Joesph','Something','33843453533','AMEX',' ',0)";
        statement.executeUpdate(insertData1);
        statement.executeUpdate(insertData2);
        statement.executeUpdate(insertData3);
        statement.executeUpdate(insertData4);
        statement.executeUpdate(insertData5);
        statement.executeUpdate(insertData6);
        statement.executeUpdate(insertData7);
        statement.executeUpdate(insertData8);
        statement.executeUpdate(insertData9);
        statement.executeUpdate(insertData10);
        statement.executeUpdate(insertData11);
        statement.executeUpdate(insertData12);
        statement.executeUpdate(insertData13);
    }

	private void createLoginTable(Connection connection) throws SQLException
	{
		Statement statement = connection.createStatement();

		// Delete table if there is one
		try
		{
			String dropTable = "DROP TABLE user_login";
			statement.executeUpdate(dropTable);
		}
		catch (SQLException e)
		{
			System.out.println("Error dropping user_login table");
		}

		// Create the new table
		try
		{
			String createTableStatement = "CREATE TABLE user_login (" 
						+ "userid varchar(5)," 
						+ "webgoat_user varchar(20)" 
						+ ")";
			statement.executeUpdate(createTableStatement);
		}
		catch (SQLException e)
		{
			System.out.println("Error creating user database");
			e.printStackTrace();
		}

	}

    /**
     * Description of the Method
     *
     * @param connection Description of the Parameter
     *
     * @exception SQLException Description of the Exception
     */
    private void createWeatherDataTable(Connection connection) throws SQLException
    {
        Statement statement = connection.createStatement();

        // Delete table if there is one
        try
        {
            String dropTable = "DROP TABLE weather_data";
            statement.executeUpdate(dropTable);
        }
        catch (SQLException e)
        {
            System.out.println("Error dropping weather database");
        }

        // Create the new table
        try
        {
            String createTableStatement = "CREATE TABLE weather_data (" 
				+ "station int not null,"
				+ "name varchar(20) not null," 
				+ "state char(2) not null,"
				+ "min_temp int not null," 
				+ "max_temp int not null" 
				+ ")";
            statement.executeUpdate(createTableStatement);
        }
        catch (SQLException e)
        {
            System.out.println("Error creating weather database");
            e.printStackTrace();
        }

        // Populate it
        String insertData1 = "INSERT INTO weather_data VALUES (101,'Columbia','MD',-10,102)";
        String insertData2 = "INSERT INTO weather_data VALUES (102,'Seattle','WA',-15,90)";
        String insertData3 = "INSERT INTO weather_data VALUES (103,'New York','NY',-10,110)";
        String insertData4 = "INSERT INTO weather_data VALUES (104,'Houston','TX',20,120)";
        String insertData5 = "INSERT INTO weather_data VALUES (10001,'Camp David','MD',-10,100)";
        String insertData6 = "INSERT INTO weather_data VALUES (11001,'Ice Station Zebra','NA',-60,30)";
        statement.executeUpdate(insertData1);
        statement.executeUpdate(insertData2);
        statement.executeUpdate(insertData3);
        statement.executeUpdate(insertData4);
        statement.executeUpdate(insertData5);
        statement.executeUpdate(insertData6);
    }
    
	//--------------------------------------------------------------------------
	//--------------------------------------------------------------------------
	//
	//  The tables below are for WebGoat Financials
	//
	//  DO NOT MODIFY THESE TABLES - unless you change the org chart
	//								 and access control matrix documents
	//
	//--------------------------------------------------------------------------
	//--------------------------------------------------------------------------
	

	private void createEmployeeTable(Connection connection) throws SQLException
	{
		Statement statement = connection.createStatement();
		
		try
		{
			String dropTable = "DROP TABLE employee";
			statement.executeUpdate(dropTable);
		}
		catch ( SQLException e )
		{
			System.out.println("Error: unable to drop employee table");
		}
		
		// Create Table
		try
		{
			String createTable = "CREATE TABLE employee ("
					//+ "userid INT GENERATED ALWAYS AS IDENTITY PRIMARY KEY,"
					+ "userid INT NOT NULL PRIMARY KEY,"
					+ "first_name VARCHAR(20),"
					+ "last_name VARCHAR(20),"
					+ "ssn VARCHAR(12),"
					+ "password VARCHAR(10),"
					+ "title VARCHAR(20),"
					+ "phone VARCHAR(13),"
					+ "address1 VARCHAR(80),"
					+ "address2 VARCHAR(80),"
					+ "manager INT,"
					+ "start_date CHAR(8),"
					+ "salary INT,"
					+ "ccn VARCHAR(30),"
					+ "ccn_limit INT,"
					+ "disciplined_date CHAR(8)," // date of write up, NA otherwise
					+ "disciplined_notes VARCHAR(60)," // reason for the recent write-up
					+ "personal_description VARCHAR(60)" // We can be rude here
					//+ ",CONSTRAINT fl UNIQUE NONCLUSTERED (first_name, last_name)"
					+ ")";
			
			statement.executeUpdate(createTable);
		}
		catch ( SQLException e )
		{
			System.out.println("Error: unable to create employee table");
		}
		
		String insertData1 = "INSERT INTO employee VALUES (101, 'Larry', 'Stooge', '386-09-5451', 'larry'," + 
			"'Technician','443-689-0192','9175 Guilford Rd','New York, NY', 102, 01012000,55000,'2578546969853547'," +
			"5000,010106,'Constantly harrassing coworkers','Does not work well with others')";
		
		String insertData2 = "INSERT INTO employee VALUES (102, 'Moe', 'Stooge', '936-18-4524','moe'," + 
			"'CSO','443-938-5301', '3013 AMD Ave', 'New York, NY', 112, 03082003, 140000, 'NA', 0, 0101013, " +
			"'Hit Curly over head', 'Very dominating over Larry and Curly')";
		
		String insertData3 = "INSERT INTO employee VALUES (103, 'Curly', 'Stooge', '961-08-0047','curly'," + 
			"'Technician','410-667-6654', '1112 Crusoe Lane', 'New York, NY', 102, 02122001, 50000, 'NA', 0, 0101014, " +
			"'Hit Moe back', 'Owes three-thousand to company for fradulent purchases')";
		
		String insertData4 = "INSERT INTO employee VALUES (104, 'Eric', 'Walker', '445-66-5565','eric'," + 
			"'Engineer','410-887-1193', '1160 Prescott Rd', 'New York, NY', 107, 12152005, 13000, 'NA', 0, 0101013, " +
			"'Bothering Larry about webgoat problems', 'Late. Always needs help. Too intern-ish.')";
		
		String insertData5 = "INSERT INTO employee VALUES (105, 'Tom', 'Cat', '792-14-6364','tom'," + 
			"'Engineer','443-599-0762', '2211 HyperThread Rd.', 'New York, NY', 106, 01011999, 80000, '5481360857968521', 30000, 0, " +
			"'NA', 'Co-Owner.')";
		
		String insertData6 = "INSERT INTO employee VALUES (106, 'Jerry', 'Mouse', '858-55-4452','jerry'," + 
			"'Human Resources','443-699-3366', '3011 Unix Drive', 'New York, NY', 102, 01011999, 70000, '6981754825013564', 20000, 0, " +
			"'NA', 'Co-Owner.')";
		
		String insertData7 = "INSERT INTO employee VALUES (107, 'David', 'Giambi', '439-20-9405','david'," + 
		"'Human Resources','610-521-8413', '5132 DIMM Avenue', 'New York, NY', 102, 05011999, 100000, '6981754825018101', 10000, 061402, " +
		"'Hacked into accounting server. Modified personal pay.', 'Strong work habbit. Questionable ethics.')";
		
		String insertData8 = "INSERT INTO employee VALUES (108, 'Bruce', 'McGuirre', '707-95-9482','bruce'," + 
		"'Engineer','610-282-1103', '8899 FreeBSD Drive<script>alert(document.cookie)</script> ', 'New York, NY', 107, 03012000, 110000, '6981754825854136', 30000, 061502, " +
		"'Tortuous Boot Camp workout at 5am. Employees felt sick.', 'Enjoys watching others struggle in exercises.')";
		
		String insertData9 = "INSERT INTO employee VALUES (109, 'Sean', 'Livingston', '136-55-1046','sean'," + 
		"'Engineer','610-878-9549', '6422 dFlyBSD Road', 'New York, NY', 107, 06012003, 130000, '6981754825014510', 5000, 072804, " +
		"'Late to work 30 days in row due to excessive Halo 2', 'Has some fascination with Steelers. Go Ravens.')";
		
		String insertData10 = "INSERT INTO employee VALUES (110, 'Joanne', 'McDougal', '789-54-2413','joanne'," + 
		"'Human Resources','610-213-6341', '5567 Broadband Lane', 'New York, NY', 106, 01012001, 90000, '6981754825081054', 300, 112005, " +
		"'Used company cc to purchase new car. Limit adjusted.', 'Finds it necessary to leave early every day.')";

		String insertData11 = "INSERT INTO employee VALUES (111, 'John', 'Wayne', '129-69-4572', 'john'," + 
		"'CTO','610-213-1134', '129 Third St', 'New York, NY', 112, 01012001, 200000, '4437334565679921', 300, 112005, " +
		"'', '')";
		String insertData12 = "INSERT INTO employee VALUES (112, 'Neville', 'Bartholomew', '111-111-1111', 'socks'," + 
		"'CEO','408-587-0024', '1 Corporate Headquarters', 'San Jose, CA', 112, 03012000, 450000, '4803389267684109', 300, 112005, " +
		"'', '')";
		
		statement.executeUpdate(insertData1);
		statement.executeUpdate(insertData2);
		statement.executeUpdate(insertData3);
		statement.executeUpdate(insertData4);
		statement.executeUpdate(insertData5);
		statement.executeUpdate(insertData6);
		statement.executeUpdate(insertData7);
		statement.executeUpdate(insertData8);
		statement.executeUpdate(insertData9);
		statement.executeUpdate(insertData10);
		statement.executeUpdate(insertData11);
		statement.executeUpdate(insertData12);
		
	}
	
	private void createRolesTable(Connection connection) throws SQLException
	{
		Statement statement = connection.createStatement();
		
		try
		{
			String dropTable = "DROP TABLE roles";
			statement.executeUpdate(dropTable);
		}
		catch ( SQLException e )
		{
			System.out.println("Error: unable to drop roles");
		}
		
		try
		{
			String createTable = "CREATE TABLE roles ("
				+ "userid INT NOT NULL,"
				+ "role VARCHAR(10) NOT NULL,"
				+ "PRIMARY KEY (userid, role)"
				+ ")";
			
			statement.executeUpdate(createTable);
		}
		catch ( SQLException e )
		{
			System.out.println ("Error: Unable to create role table");
		}
		
		String insertData1 = "INSERT INTO roles VALUES (101, 'employee')";
		String insertData2 = "INSERT INTO roles VALUES (102, 'manager')";
		String insertData3 = "INSERT INTO roles VALUES (103, 'employee')";
		String insertData4 = "INSERT INTO roles VALUES (104, 'employee')";
		String insertData5 = "INSERT INTO roles VALUES (105, 'employee')";
		String insertData6 = "INSERT INTO roles VALUES (106, 'hr')";
		String insertData7 = "INSERT INTO roles VALUES (107, 'manager')";
		String insertData8 = "INSERT INTO roles VALUES (108, 'employee')";
		String insertData9 = "INSERT INTO roles VALUES (109, 'employee')";
		String insertData10 = "INSERT INTO roles VALUES (110, 'hr')";
		String insertData11 = "INSERT INTO roles VALUES (111, 'admin')";
		String insertData12 = "INSERT INTO roles VALUES (112, 'admin')";
		
		statement.executeUpdate(insertData1);
		statement.executeUpdate(insertData2);
		statement.executeUpdate(insertData3);
		statement.executeUpdate(insertData4);
		statement.executeUpdate(insertData5);
		statement.executeUpdate(insertData6);
		statement.executeUpdate(insertData7);
		statement.executeUpdate(insertData8);
		statement.executeUpdate(insertData9);
		statement.executeUpdate(insertData10);
		statement.executeUpdate(insertData11);
		statement.executeUpdate(insertData12);
	}
	
	private void createAuthTable(Connection connection) throws SQLException
	{
		Statement statement = connection.createStatement();
		
		try
		{
			String dropTable = "DROP TABLE auth";
			statement.executeUpdate(dropTable);
		}
		catch ( SQLException e )
		{
			System.out.println("Error: unable to drop auth");
		}
		
		try
		{
			String createTable = "CREATE TABLE auth ("
				+ "role VARCHAR(10) NOT NULL,"
				+ "functionid VARCHAR(20) NOT NULL,"
				+ "PRIMARY KEY (role, functionid)"
				+ ")";
			
			statement.executeUpdate(createTable);
		}
		catch ( SQLException e )
		{
			System.out.println("Error: unable to create auth table");
		}
		
		String insertData1 = "INSERT INTO auth VALUES('employee', 'Logout')";
		String insertData2 = "INSERT INTO auth VALUES('employee', 'ListStaff')";
		String insertData3 = "INSERT INTO auth VALUES('employee', 'ViewProfile')";
		String insertData4 = "INSERT INTO auth VALUES('employee', 'EditProfile')";
		String insertData4_1 = "INSERT INTO auth VALUES('employee', 'SearchStaff')";
		String insertData4_2 = "INSERT INTO auth VALUES('employee', 'FindProfile')";
		String insertData5 = "INSERT INTO auth VALUES('manager', 'Logout')";
		String insertData6 = "INSERT INTO auth VALUES('manager', 'ListStaff')";
		String insertData7 = "INSERT INTO auth VALUES('manager', 'ViewProfile')";
		String insertData7_1 = "INSERT INTO auth VALUES('manager', 'SearchStaff')";
		String insertData7_2 = "INSERT INTO auth VALUES('manager', 'FindProfile')";
//		String insertData8 = "INSERT INTO auth VALUES('manager', 'EditProfile')";
//		String insertData9 = "INSERT INTO auth VALUES('manager', 'CreateProfile')";
//		String insertData10 = "INSERT INTO auth VALUES('manager', 'DeleteProfile')";
//		String insertData11 = "INSERT INTO auth VALUES('manager', 'UpdateProfile')";
		String insertData12 = "INSERT INTO auth VALUES('hr', 'Logout')";
		String insertData13 = "INSERT INTO auth VALUES('hr', 'ListStaff')";
		String insertData14 = "INSERT INTO auth VALUES('hr', 'ViewProfile')";
		String insertData15 = "INSERT INTO auth VALUES('hr', 'EditProfile')";
		String insertData16 = "INSERT INTO auth VALUES('hr', 'CreateProfile')";
		String insertData17 = "INSERT INTO auth VALUES('hr', 'DeleteProfile')";
		String insertData18 = "INSERT INTO auth VALUES('hr', 'UpdateProfile')";
		String insertData18_1 = "INSERT INTO auth VALUES('hr', 'SearchStaff')";
		String insertData18_2 = "INSERT INTO auth VALUES('hr', 'FindProfile')";
		String insertData19 = "INSERT INTO auth VALUES('admin', 'Logout')";
		String insertData20 = "INSERT INTO auth VALUES('admin', 'ListStaff')";
		String insertData21 = "INSERT INTO auth VALUES('admin', 'ViewProfile')";
		String insertData22 = "INSERT INTO auth VALUES('admin', 'EditProfile')";
		String insertData23 = "INSERT INTO auth VALUES('admin', 'CreateProfile')";
		String insertData24 = "INSERT INTO auth VALUES('admin', 'DeleteProfile')";
		String insertData25 = "INSERT INTO auth VALUES('admin', 'UpdateProfile')";
		String insertData25_1 = "INSERT INTO auth VALUES('admin', 'SearchStaff')";
		String insertData25_2 = "INSERT INTO auth VALUES('admin', 'FindProfile')";
		
		// Add a permission for the webgoat role to see the source.
		// The challenge(s) will change the default role to "challenge" 
		String insertData26 = "INSERT INTO auth VALUES('" + AbstractLesson.USER_ROLE + "','" + WebSession.SHOWSOURCE +"')";
		String insertData27 = "INSERT INTO auth VALUES('" + AbstractLesson.USER_ROLE + "','" + WebSession.SHOWHINTS + "')";
		
		statement.executeUpdate(insertData1);
		statement.executeUpdate(insertData2);
		statement.executeUpdate(insertData3);
		statement.executeUpdate(insertData4);
		statement.executeUpdate(insertData4_1);
		statement.executeUpdate(insertData4_2);
		statement.executeUpdate(insertData5);
		statement.executeUpdate(insertData6);
		statement.executeUpdate(insertData7);
		statement.executeUpdate(insertData7_1);
		statement.executeUpdate(insertData7_2);
//		statement.executeUpdate(insertData8);
//		statement.executeUpdate(insertData9);
//		statement.executeUpdate(insertData10);
//		statement.executeUpdate(insertData11);
		statement.executeUpdate(insertData12);
		statement.executeUpdate(insertData13);
		statement.executeUpdate(insertData14);
		statement.executeUpdate(insertData15);
		statement.executeUpdate(insertData16);
		statement.executeUpdate(insertData17);
		statement.executeUpdate(insertData18);
		statement.executeUpdate(insertData18_1);
		statement.executeUpdate(insertData18_2);
		statement.executeUpdate(insertData19);
		statement.executeUpdate(insertData20);
		statement.executeUpdate(insertData21);
		statement.executeUpdate(insertData22);
		statement.executeUpdate(insertData23);
		statement.executeUpdate(insertData24);
		statement.executeUpdate(insertData25);
		statement.executeUpdate(insertData25_1);
		statement.executeUpdate(insertData25_2);
		statement.executeUpdate(insertData26);
		statement.executeUpdate(insertData27);
	}
	
	private void createOwnershipTable(Connection connection) throws SQLException
	{
		Statement statement = connection.createStatement();
		
		try
		{
			String dropTable = "DROP TABLE ownership";
			statement.executeUpdate(dropTable);
		}
		catch ( SQLException e )
		{
			System.out.println("Error: unable to drop ownership");
		}		
		
		try
		{
			String createTable = "CREATE TABLE ownership ("
				+ "employer_id INT NOT NULL,"
				+ "employee_id INT NOT NULL,"
				+ "PRIMARY KEY (employee_id, employer_id)"
				+ ")";
			
			statement.executeUpdate(createTable);
		}
		catch ( SQLException e )
		{
			System.out.println("Error: unable to create ownership table");
		}
		
		String inputData =  "INSERT INTO ownership VALUES (112, 101)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (112, 102)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (112, 103)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (112, 104)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (112, 105)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (112, 106)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (112, 107)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (112, 108)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (112, 109)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (112, 110)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (112, 111)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (112, 112)";
		statement.executeUpdate(inputData);
		
		inputData =         "INSERT INTO ownership VALUES (102, 101)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (102, 102)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (102, 103)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (102, 104)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (102, 105)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (102, 106)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (102, 107)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (102, 108)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (102, 109)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (102, 110)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (102, 111)";
		statement.executeUpdate(inputData);
		
		inputData =         "INSERT INTO ownership VALUES (111, 101)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (111, 102)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (111, 103)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (111, 104)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (111, 105)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (111, 106)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (111, 107)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (111, 108)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (111, 109)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (111, 110)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (111, 111)";
		statement.executeUpdate(inputData);

		inputData =         "INSERT INTO ownership VALUES (106, 105)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (106, 106)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (106, 110)";
		statement.executeUpdate(inputData);
		
		inputData =         "INSERT INTO ownership VALUES (101, 101)";
		statement.executeUpdate(inputData);
		
		inputData =         "INSERT INTO ownership VALUES (103, 103)";
		statement.executeUpdate(inputData);
		
		inputData =         "INSERT INTO ownership VALUES (107, 104)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (107, 108)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (107, 109)";
		statement.executeUpdate(inputData);
		inputData =         "INSERT INTO ownership VALUES (107, 107)";
		statement.executeUpdate(inputData);

		inputData =         "INSERT INTO ownership VALUES (105, 105)";
		statement.executeUpdate(inputData);
		
		inputData =         "INSERT INTO ownership VALUES (110, 110)";
		statement.executeUpdate(inputData);
		
		inputData =         "INSERT INTO ownership VALUES (104, 104)";
		statement.executeUpdate(inputData);
		
		inputData =         "INSERT INTO ownership VALUES (108, 108)";
		statement.executeUpdate(inputData);
		
		inputData =         "INSERT INTO ownership VALUES (109, 109)";
		statement.executeUpdate(inputData);
	
	}
	
	//--------------------------------------------------------------------------
	//
	//  End of WebGoat Financials
	//
	//--------------------------------------------------------------------------
	
    /**
     * Description of the Method
     *
     * @param connection Description of the Parameter
     *
     * @exception SQLException Description of the Exception
     */
    public void makeDB(Connection connection) throws SQLException
    {
        System.out.println("Successful connection to database");
        createUserDataTable(connection);
        createLoginTable(connection);
        createUserAdminTable(connection);
        createProductTable(connection);
        createMessageTable(connection);
        createEmployeeTable(connection);
        createRolesTable(connection);
        createAuthTable(connection);
        createOwnershipTable(connection);
        createWeatherDataTable(connection);
        System.out.println("Success: creating tables.");
    }
}
