/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.apache.tomcat.jni;

import junit.framework.Test;
import junit.framework.TestSuite;
import junit.textui.TestRunner;
import junit.framework.TestCase;

import java.io.OutputStream;

/**
 * A basic test suite that tests Socket Server feature.
 * 
 * @see org.apache.tomcat.jni
 */
public class SocketServerTestSuite extends TestCase {

    private long serverSock = 0;
    private int port=6666;
    private String host="localhost";

    public static long serverPool = 0;

    public void testSocketServerTestSuite() throws Exception {

        /* Load APR library */
        Library.initialize(null);

        /* Create the server socket and listen on it */
        serverPool = Pool.create(0);
        long inetAddress = Address.info(host, Socket.APR_INET,
                                        port, 0, serverPool);
        serverSock = Socket.create(Socket.APR_INET, Socket.SOCK_STREAM,
                                   Socket.APR_PROTO_TCP, serverPool);
        int rc = Socket.bind(serverSock, inetAddress);
        if (rc != 0) {
            Socket.close(serverSock);
            throw(new Exception("Can't bind: " + Error.strerror(rc)));
        }
        Socket.listen(serverSock, 5);

        /* Start the client that connects to the server */
        Client client = new Client();
        client.start(); 
        java.lang.Thread.sleep(100);
    
        /* Accept it */
        long clientSock = Socket.accept(serverSock);

        /* Try 2 milliseconds timeout */
        Socket.timeoutSet(clientSock, 2000);
        long timeout = Socket.timeoutGet(clientSock);
        if (timeout != 2000) {
            Socket.close(clientSock);
            Socket.close(serverSock);
            throw new Exception("Socket.timeoutGet clientSock failed");
        }

        long start = System.currentTimeMillis();
        byte [] buf = new byte[1];
        while (Socket.recv(clientSock, buf, 0, 1) == 1) {
        }
        long wait = System.currentTimeMillis() - start;
        /* allow some margin between set timeout and observed return time */
        if (wait < 1 || wait > 12) {
            Socket.close(clientSock);
            Socket.close(serverSock);
            throw new Exception("2 milliseconds client Socket.timeoutSet failed");
        }

        /* Try 2 milliseconds timeout on accept socket */
        Socket.timeoutSet(serverSock, 2000);
        timeout = Socket.timeoutGet(serverSock);
        if (timeout != 2000) {
            Socket.close(serverSock);
            throw new Exception("Socket.timeoutGet serverSock failed");
        }

        start = System.currentTimeMillis();
        boolean ok = false;
        try {
            clientSock = Socket.accept(serverSock);
        } catch (Exception ex) {
            ok = true;
        }
        wait = System.currentTimeMillis() - start;
        /* allow some margin between set timeout and observed return time */
        if (wait < 1 || wait > 12 && ! ok) {
            Socket.close(clientSock);
            Socket.close(serverSock);
            throw new Exception("2 milliseconds accept Socket.timeoutSet failed");
        }

        /* Try APR_SO_NONBLOCK */
        Socket.optSet(serverSock, Socket.APR_SO_NONBLOCK, 1);
        int val = Socket.optGet(serverSock, Socket.APR_SO_NONBLOCK);
        if (val != 1) {
            Socket.close(serverSock);
            throw new Exception("Socket.optGet serverSock failed");
        }

        start = System.currentTimeMillis();
        ok = false;
        try {
            clientSock = Socket.accept(serverSock);
        } catch (Exception ex) {
            ok = true;
        }
        wait = System.currentTimeMillis() - start;
        if (wait > 1 && ! ok) {
            Socket.close(clientSock);
            Socket.close(serverSock);
            throw new Exception("non_blocking accept Socket.APR_SO_NONBLOCK failed");
        }
        client.join();
        Socket.close(clientSock);

        /* Try the same on client socket */
        client = new Client();
        client.start(); 
        java.lang.Thread.sleep(100);
        clientSock = Socket.accept(serverSock);
        Socket.optSet(clientSock, Socket.APR_SO_NONBLOCK, 1);
        val = Socket.optGet(clientSock, Socket.APR_SO_NONBLOCK);
        if (val != 1) {
            Socket.close(clientSock);
            Socket.close(serverSock);
            throw new Exception("Socket.optGet clientSock failed");
        }
        start = System.currentTimeMillis();
        while (Socket.recv(clientSock, buf, 0, 1) == 1) {
        }
        wait = System.currentTimeMillis() - start;
        if (wait > 1) {
            Socket.close(clientSock);
            Socket.close(serverSock);
            throw new Exception("non_blocking client Socket.APR_SO_NONBLOCK failed");
        }

        /* Now blocking */
        Socket.optSet(clientSock, Socket.APR_SO_NONBLOCK, 0);
        start = System.currentTimeMillis();
        while (Socket.recv(clientSock, buf, 0, 1) == 1) {
        }
        wait = System.currentTimeMillis() - start;
        if (wait < 1) {
            Socket.close(clientSock);
            Socket.close(serverSock);
            throw new Exception("non_blocking client Socket.APR_SO_NONBLOCK false failed");
        }

        client.join();
        Socket.close(clientSock);
        Socket.close(serverSock);
        Library.terminate();
    }

    /* small client that connects and sends one byte */
    private class Client extends java.lang.Thread {
        java.net.Socket sock;
        public Client() throws Exception {
            sock = new java.net.Socket(host, port);
        }
        public void run() {
            try {
                OutputStream ou = sock.getOutputStream();
                ou.write('A');
                ou.flush();
                java.lang.Thread.sleep(10000);
                sock.close();
            } catch(Exception ex ) {
                ex.printStackTrace();
            }
        }
    }
    
    public static void main(String[] args) {
        TestRunner.run(suite());
    }
    
    public static Test suite()
    {
        TestSuite suite = new TestSuite( "Tomcat Native Server Socket" );
        suite.addTest(new TestSuite(SocketServerTestSuite.class));
        suite.addTest(new TestSuite(SocketServerTestBind.class));
        return suite;
    }
}