/*
 * Copyright (c) 2005, Oracle and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.
 *
 * This code 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
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */

/*
 * @test
 * @bug 6223624
 * @summary SSLSocket.setUseClientMode() fails to throw expected
 *        IllegalArgumentException
 */

/*
 * Attempts to replicate a TCK test failure which creates SSLServerSockets
 * and then runs client threads which connect and start handshaking. Once
 * handshaking is begun the server side attempts to invoke
 * SSLSocket.setUseClientMode() on one or the other of the ends of the
 * connection, expecting an IllegalArgumentException.
 *
 * If the server side of the connection tries setUseClientMode() we
 * see the expected exception. If the setting is tried on the
 * client side SSLSocket, we do *not* see the exception, except
 * occasionally on the very first iteration.
 */

import java.io.*;
import java.lang.*;
import java.net.*;
import javax.net.ssl.*;
import java.security.*;
import java.security.cert.*;

public class SetClientMode {
    private static String[] algorithms = {"TLS", "SSL", "SSLv3", "TLS"};
    volatile int serverPort = 0;

    /*
     * Where do we find the keystores?
     */
    static String pathToStores = "../../../../../../../etc";
    static String keyStoreFile = "keystore";
    static String trustStoreFile = "truststore";
    static String passwd = "passphrase";


    public SetClientMode() {
        // trivial constructor
    }

    public static void main(String[] args) throws Exception {
        String keyFilename =
            System.getProperty("test.src", "./") + "/" + pathToStores +
                "/" + keyStoreFile;
        String trustFilename =
            System.getProperty("test.src", "./") + "/" + pathToStores +
                "/" + trustStoreFile;

        System.setProperty("javax.net.ssl.keyStore", keyFilename);
        System.setProperty("javax.net.ssl.keyStorePassword", passwd);
        System.setProperty("javax.net.ssl.trustStore", trustFilename);
        System.setProperty("javax.net.ssl.trustStorePassword", passwd);

        new SetClientMode().run();
    }

    public void run() throws Exception {
        for (int i = 0; i < algorithms.length; i++) {
            testCombo( algorithms[i] );
        }
    }

    public void testCombo(String algorithm) throws Exception {
        Exception modeException = null ;

        // Create a server socket
        SSLServerSocketFactory ssf =
            (SSLServerSocketFactory)SSLServerSocketFactory.getDefault();
        SSLServerSocket serverSocket =
            (SSLServerSocket)ssf.createServerSocket(serverPort);
        serverPort = serverSocket.getLocalPort();

        // Create a client socket
        SSLSocketFactory sf = (SSLSocketFactory)SSLSocketFactory.getDefault();
        SSLSocket clientSocket = (SSLSocket)sf.createSocket(
                                InetAddress.getLocalHost(),
                                serverPort );

        // Create a client which will use the SSLSocket to talk to the server
        SocketClient client = new SocketClient(clientSocket);

        // Start the client and then accept any connection
        client.start();

        SSLSocket connectedSocket = (SSLSocket)serverSocket.accept();

        // force handshaking to complete
        connectedSocket.getSession();

        try {
            // Now try invoking setClientMode() on one
            // or the other of our two sockets. We expect
            // to see an IllegalArgumentException because
            // handshaking has begun.
            clientSocket.setUseClientMode(false);

            modeException = new Exception("no IllegalArgumentException");
        } catch (IllegalArgumentException iae) {
            System.out.println("succeeded, we can't set the client mode");
        } catch (Exception e) {
            modeException = e;
        } finally {
            // Shut down.
            connectedSocket.close();
            serverSocket.close();

            if (modeException != null) {
                throw modeException;
            }
        }

        return;
    }

    // A thread-based client which does nothing except
    // start handshaking on the socket it's given.
    class SocketClient extends Thread {
        SSLSocket clientsideSocket;
        Exception clientException = null;
        boolean done = false;

        public SocketClient( SSLSocket s ) {
            clientsideSocket = s;
        }

        public void run() {
            try {
                clientsideSocket.startHandshake();

                // If we were to invoke setUseClientMode()
                // here, the expected exception will happen.
                //clientsideSocket.getSession();
                //clientsideSocket.setUseClientMode( false );
            } catch ( Exception e ) {
                e.printStackTrace();
                clientException = e;
            } finally {
                done = true;
                try {
                    clientsideSocket.close();
                } catch ( IOException e ) {
                    // eat it
                }
            }
            return;
        }

        boolean isDone() {
            return done;
        }

        Exception getException() {
            return clientException;
        }
    }
}