/*
 * Copyright (c) 2015, 2016, 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 com.sun.net.httpserver.HttpExchange;
import com.sun.net.httpserver.HttpHandler;
import com.sun.net.httpserver.HttpServer;
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.IOException;
import java.io.InputStream;
import java.net.Authenticator;
import java.net.InetSocketAddress;
import java.net.PasswordAuthentication;
import java.net.URL;
import java.net.URLConnection;
import java.util.List;
import sun.net.www.protocol.http.ntlm.NTLMAuthenticationCallback;

/*
 * @test
 * @bug 8137174
 * @modules java.base/sun.net.www.protocol.http.ntlm
 *          jdk.httpserver
 * @summary Checks if NTLM auth works fine if security manager set
 * @run main/othervm/java.security.policy=NTLMAuthWithSM.policy NTLMAuthWithSM
 */
public class NTLMAuthWithSM {

    public static void main(String[] args) throws Exception {
        // security manager is required
        if (System.getSecurityManager() == null) {
            throw new RuntimeException("Security manager not specified");
        }

        if (System.getProperty("os.name").startsWith("Windows")) {
            // disable transparent NTLM authentication on Windows
            NTLMAuthenticationCallback.setNTLMAuthenticationCallback(
                    new NTLMAuthenticationCallbackImpl());
        }

        try (LocalHttpServer server = LocalHttpServer.startServer()) {
            // set authenticator
            Authenticator.setDefault(new AuthenticatorImpl());

            String url = String.format("http://localhost:%d/test/",
                    server.getPort());

            // load a document which is protected with NTML authentication
            System.out.println("load() called: " + url);
            URLConnection conn = new URL(url).openConnection();
            try (BufferedReader reader = new BufferedReader(
                    new InputStreamReader(conn.getInputStream()))) {

                String line = reader.readLine();
                if (line == null) {
                    throw new IOException("Couldn't read a response");
                }
                do {
                    System.out.println(line);
                } while ((line = reader.readLine()) != null);
            }
        }

        System.out.println("Test passed");
    }

    private static class AuthenticatorImpl extends Authenticator {

        @Override
        public PasswordAuthentication getPasswordAuthentication() {
            System.out.println("getPasswordAuthentication() called, scheme: "
                    + getRequestingScheme());
            if (getRequestingScheme().equalsIgnoreCase("ntlm")) {
                return new PasswordAuthentication("test", "test".toCharArray());
            }
            return null;
        }
    }

    // local http server which pretends to support NTLM auth
    static class LocalHttpServer implements HttpHandler, AutoCloseable {

        private final HttpServer server;

        private LocalHttpServer(HttpServer server) {
            this.server = server;
        }

        static LocalHttpServer startServer() throws IOException {
            HttpServer httpServer = HttpServer.create(
                    new InetSocketAddress(0), 0);
            LocalHttpServer localHttpServer = new LocalHttpServer(httpServer);
            localHttpServer.start();

            return localHttpServer;
        }

        void start() {
            server.createContext("/test", this);
            server.start();
            System.out.println("HttpServer: started on port " + getPort());
        }

        void stop() {
            server.stop(0);
            System.out.println("HttpServer: stopped");
        }

        int getPort() {
            return server.getAddress().getPort();
        }

        @Override
        public void handle(HttpExchange t) throws IOException {
            System.out.println("HttpServer: handle connection");

            // read a request
            try (InputStream is = t.getRequestBody()) {
                while (is.read() > 0);
            }

            try {
                List<String> headers = t.getRequestHeaders()
                        .get("Authorization");
                if (headers != null && !headers.isEmpty()
                        && headers.get(0).trim().contains("NTLM")) {
                    byte[] output = "hello".getBytes();
                    t.sendResponseHeaders(200, output.length);
                    t.getResponseBody().write(output);
                    System.out.println("HttpServer: return 200");
                } else {
                    t.getResponseHeaders().set("WWW-Authenticate", "NTLM");
                    byte[] output = "forbidden".getBytes();
                    t.sendResponseHeaders(401, output.length);
                    t.getResponseBody().write(output);
                    System.out.println("HttpServer: return 401");
                }
            } catch (IOException e) {
                System.out.println("HttpServer: exception: " + e);
                System.out.println("HttpServer: return 500");
                t.sendResponseHeaders(500, 0);
            } finally {
                t.close();
            }
        }

        @Override
        public void close() {
            stop();
        }
    }

    private static class NTLMAuthenticationCallbackImpl
            extends NTLMAuthenticationCallback {

        // don't trust any site, so that no transparent NTLM auth happens
        @Override
        public boolean isTrustedSite(URL url) {
            System.out.println(
                    "NTLMAuthenticationCallbackImpl.isTrustedSite() called: "
                        + "return false");
            return false;
        }
    }
}