/*
 * Copyright (c) 2018, 2019, 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.
 */

import javax.naming.Context;
import javax.naming.NamingException;
import javax.naming.directory.DirContext;
import javax.naming.directory.InitialDirContext;
import java.io.IOException;
import java.net.InetAddress;
import java.net.ServerSocket;
import java.util.Hashtable;

/*
 * @test
 * @bug 8205330
 * @summary Test that If a connection has already been established and then
 *          the LDAP directory server sends an (unsolicited)
 *          "Notice of Disconnection", make sure client handle it correctly,
 *          no NPE been thrown.
 * @library lib/
 * @run main/othervm DisconnectNPETest
 */

public class DisconnectNPETest {
    // Normally the NPE bug should be hit less than 100 times run, but just in
    // case, we set repeat count to 1000 here.
    private static final int REPEAT_COUNT = 1000;

    // "Notice of Disconnection" message
    private static final byte[] DISCONNECT_MSG = { 0x30, 0x4C, 0x02, 0x01, 0x00,
            0x78, 0x47, 0x0A, 0x01, 0x34, 0x04, 0x00, 0x04, 0x28, 0x55, 0x4E,
            0x41, 0x56, 0x41, 0x49, 0x4C, 0x41, 0x42, 0x4C, 0x45, 0x3A, 0x20,
            0x54, 0x68, 0x65, 0x20, 0x73, 0x65, 0x72, 0x76, 0x65, 0x72, 0x20,
            0x77, 0x69, 0x6C, 0x6C, 0x20, 0x64, 0x69, 0x73, 0x63, 0x6F, 0x6E,
            0x6E, 0x65, 0x63, 0x74, 0x21, (byte) 0x8A, 0x16, 0x31, 0x2E, 0x33,
            0x2E, 0x36, 0x2E, 0x31, 0x2E, 0x34, 0x2E, 0x31, 0x2E, 0x31, 0x34,
            0x36, 0x36, 0x2E, 0x32, 0x30, 0x30, 0x33, 0x36 };
    private static final byte[] BIND_RESPONSE = { 0x30, 0x0C, 0x02, 0x01, 0x01,
            0x61, 0x07, 0x0A, 0x01, 0x00, 0x04, 0x00, 0x04, 0x00 };

    public static void main(String[] args) throws IOException {
        new DisconnectNPETest().run();
    }

    private ServerSocket serverSocket;
    private Hashtable<Object, Object> env;

    private void initRes() throws IOException {
        serverSocket = new ServerSocket(0, 0, InetAddress.getLoopbackAddress());
    }

    private void initTest() {
        env = new Hashtable<>();
        env.put(Context.INITIAL_CONTEXT_FACTORY,
                "com.sun.jndi.ldap.LdapCtxFactory");
        env.put(Context.PROVIDER_URL, String.format("ldap://%s:%d/",
                InetAddress.getLoopbackAddress().getHostName(),
                serverSocket.getLocalPort()));
        env.put(Context.SECURITY_AUTHENTICATION, "simple");
        env.put(Context.SECURITY_PRINCIPAL,
                "cn=8205330,ou=Client6,ou=Vendor1,o=IMC,c=US");
        env.put(Context.SECURITY_CREDENTIALS, "secret123");
    }

    private void run() throws IOException {
        initRes();
        initTest();
        int count = 0;
        try (BaseLdapServer ignored = new BaseLdapServer(serverSocket)
                .setCommonRequestHandler((msg, out) -> {
                    if (msg.getOperation()
                            == LdapMessage.Operation.BIND_REQUEST) {
                        out.write(BIND_RESPONSE);
                        out.write(DISCONNECT_MSG);
                    }
                }).startServer()) {
            while (count < REPEAT_COUNT) {
                count++;
                InitialDirContext context = null;
                try {
                    context = new InitialDirContext(env);
                } catch (NamingException ne) {
                    System.out.println("(" + count + "/" + REPEAT_COUNT
                            + ") It's ok to get NamingException: " + ne);
                    // for debug
                    ne.printStackTrace(System.out);
                } finally {
                    cleanupContext(context);
                }
            }
        } finally {
            System.out.println("Test count: " + count + "/" + REPEAT_COUNT);
        }
    }

    private void cleanupContext(DirContext context) {
        if (context != null) {
            try {
                context.close();
            } catch (NamingException e) {
                // ignore
            }
        }
    }
}