/*

 * Copyright (c) 2017, 2018, 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.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * 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.

 */

package jdk.internal.net.http.common;

import jdk.internal.net.http.common.SubscriberWrapper.SchedulingAction;

import javax.net.ssl.SSLEngine;

import javax.net.ssl.SSLEngineResult;

import javax.net.ssl.SSLEngineResult.HandshakeStatus;

import javax.net.ssl.SSLEngineResult.Status;

import javax.net.ssl.SSLException;

import java.io.IOException;

import java.lang.System.Logger.Level;

import java.nio.ByteBuffer;

import java.util.ArrayList;

import java.util.Collections;

import java.util.Iterator;

import java.util.LinkedList;

import java.util.List;

import java.util.concurrent.CompletableFuture;

import java.util.concurrent.ConcurrentLinkedQueue;

import java.util.concurrent.Executor;

import java.util.concurrent.Flow;

import java.util.concurrent.Flow.Subscriber;

import java.util.concurrent.atomic.AtomicInteger;

/**

 * Implements SSL using two SubscriberWrappers.

 *

 * <p> Constructor takes two Flow.Subscribers: one that receives the network

 * data (after it has been encrypted by SSLFlowDelegate) data, and one that

 * receives the application data (before it has been encrypted by SSLFlowDelegate).

 *

 * <p> Methods upstreamReader() and upstreamWriter() return the corresponding

 * Flow.Subscribers containing Flows for the encrypted/decrypted upstream data.

 * See diagram below.

 *

 * <p> How Flow.Subscribers are used in this class, and where they come from:

 * <pre>

 * {@code

 *

 *

 *

 * --------->  data flow direction

 *

 *

 *                         +------------------+

 *        upstreamWriter   |                  | downWriter

 *        ---------------> |                  | ------------>

 *  obtained from this     |                  | supplied to constructor

 *                         | SSLFlowDelegate  |

 *        downReader       |                  | upstreamReader

 *        <--------------- |                  | <--------------

 * supplied to constructor |                  | obtained from this

 *                         +------------------+

 *

 * Errors are reported to the downReader Flow.Subscriber

 *

 * }

 * </pre>

 */

public class SSLFlowDelegate {

    static final boolean DEBUG = Utils.DEBUG; // Revisit: temporary dev flag.

    final System.Logger debug =

            Utils.getDebugLogger(this::dbgString, DEBUG);

    final Executor exec;

    final Reader reader;

    final Writer writer;

    final SSLEngine engine;

    final String tubeName; // hack

    final CompletableFuture<String> alpnCF; // completes on initial handshake

    final static ByteBuffer SENTINEL = Utils.EMPTY_BYTEBUFFER;

    volatile boolean close_notify_received;

    static AtomicInteger scount = new AtomicInteger(1);

    final int id;

    /**

     * Creates an SSLFlowDelegate fed from two Flow.Subscribers. Each

     * Flow.Subscriber requires an associated {@link CompletableFuture}

     * for errors that need to be signaled from downstream to upstream.

     */

    public SSLFlowDelegate(SSLEngine engine,

                           Executor exec,

                           Subscriber<? super List<ByteBuffer>> downReader,

                           Subscriber<? super List<ByteBuffer>> downWriter)

    {

        this.id = scount.getAndIncrement();

        this.tubeName = String.valueOf(downWriter);

        this.reader = new Reader();

        this.writer = new Writer();

        this.engine = engine;

        this.exec = exec;

        this.handshakeState = new AtomicInteger(NOT_HANDSHAKING);

        CompletableFuture.allOf(reader.completion(), writer.completion())

            .thenRun(this::normalStop);

        this.alpnCF = new MinimalFuture<>();

        // connect the Reader to the downReader and the

        // Writer to the downWriter.

        connect(downReader, downWriter);

        //Monitor.add(this::monitor);

    }

    /**

     * Returns true if the SSLFlowDelegate has detected a TLS

     * close_notify from the server.

     * @return true, if a close_notify was detected.

     */

    public boolean closeNotifyReceived() {

        return close_notify_received;

    }

    /**

     * Connects the read sink (downReader) to the SSLFlowDelegate Reader,

     * and the write sink (downWriter) to the SSLFlowDelegate Writer.

     * Called from within the constructor. Overwritten by SSLTube.

     *

     * @param downReader  The left hand side read sink (typically, the

     *                    HttpConnection read subscriber).

     * @param downWriter  The right hand side write sink (typically

     *                    the SocketTube write subscriber).

     */

    void connect(Subscriber<? super List<ByteBuffer>> downReader,

                 Subscriber<? super List<ByteBuffer>> downWriter) {

        this.reader.subscribe(downReader);

        this.writer.subscribe(downWriter);

    }

   /**

    * Returns a CompletableFuture<String> which completes after

    * the initial handshake completes, and which contains the negotiated

    * alpn.

    */

    public CompletableFuture<String> alpn() {

        return alpnCF;

    }

    private void setALPN() {

        // Handshake is finished. So, can retrieve the ALPN now

        if (alpnCF.isDone())

            return;

        String alpn = engine.getApplicationProtocol();

        debug.log(Level.DEBUG, "setALPN = %s", alpn);

        alpnCF.complete(alpn);

    }

    public String monitor() {

        StringBuilder sb = new StringBuilder();

        sb.append("SSL: id ").append(id);

        sb.append(" HS state: " + states(handshakeState));

        sb.append(" Engine state: " + engine.getHandshakeStatus().toString());

        sb.append(" LL : ");

        for (String s: stateList) {

            sb.append(s).append(" ");

        }

        sb.append("\r\n");

        sb.append("Reader:: ").append(reader.toString());

        sb.append("\r\n");

        sb.append("Writer:: ").append(writer.toString());

        sb.append("\r\n===================================");

        return sb.toString();

    }

    protected SchedulingAction enterReadScheduling() {

        return SchedulingAction.CONTINUE;

    }

    /**

     * Processing function for incoming data. Pass it thru SSLEngine.unwrap().

     * Any decrypted buffers returned to be passed downstream.

     * Status codes:

     *     NEED_UNWRAP: do nothing. Following incoming data will contain

     *                  any required handshake data

     *     NEED_WRAP: call writer.addData() with empty buffer

     *     NEED_TASK: delegate task to executor

     *     BUFFER_OVERFLOW: allocate larger output buffer. Repeat unwrap

     *     BUFFER_UNDERFLOW: keep buffer and wait for more data

     *     OK: return generated buffers.

     *

     * Upstream subscription strategy is to try and keep no more than

     * TARGET_BUFSIZE bytes in readBuf

     */

    class Reader extends SubscriberWrapper {

        final SequentialScheduler scheduler;

        static final int TARGET_BUFSIZE = 16 * 1024;

        volatile ByteBuffer readBuf;

        volatile boolean completing = false;

        final Object readBufferLock = new Object();

        final System.Logger debugr =

            Utils.getDebugLogger(this::dbgString, DEBUG);

        class ReaderDownstreamPusher implements Runnable {

            @Override public void run() { processData(); }

        }

        Reader() {

            super();

            scheduler = SequentialScheduler.synchronizedScheduler(

                                                new ReaderDownstreamPusher());

            this.readBuf = ByteBuffer.allocate(1024);

            readBuf.limit(0); // keep in read mode

        }

        protected SchedulingAction enterScheduling() {

            return enterReadScheduling();

        }

        public final String dbgString() {

            return "SSL Reader(" + tubeName + ")";

        }

        /**

         * entry point for buffers delivered from upstream Subscriber

         */

        @Override

        public void incoming(List<ByteBuffer> buffers, boolean complete) {

            debugr.log(Level.DEBUG, () -> "Adding " + Utils.remaining(buffers)

                        + " bytes to read buffer");

            addToReadBuf(buffers, complete);

            scheduler.runOrSchedule();

        }

        @Override

        public String toString() {

            return "READER: " + super.toString() + " readBuf: " + readBuf.toString()

                    + " count: " + count.toString();

        }

        private void reallocReadBuf() {

            int sz = readBuf.capacity();

            ByteBuffer newb = ByteBuffer.allocate(sz*2);

            readBuf.flip();

            Utils.copy(readBuf, newb);

            readBuf = newb;

        }

        @Override

        protected long upstreamWindowUpdate(long currentWindow, long downstreamQsize) {

            if (readBuf.remaining() > TARGET_BUFSIZE) {

                return 0;

            } else {

                return super.upstreamWindowUpdate(currentWindow, downstreamQsize);

            }

        }

        // readBuf is kept ready for reading outside of this method

        private void addToReadBuf(List<ByteBuffer> buffers, boolean complete) {

            synchronized (readBufferLock) {

                for (ByteBuffer buf : buffers) {

                    readBuf.compact();

                    while (readBuf.remaining() < buf.remaining())

                        reallocReadBuf();

                    readBuf.put(buf);

                    readBuf.flip();

                }

                if (complete) {

                    this.completing = complete;

                }

            }

        }

        void schedule() {

            scheduler.runOrSchedule();

        }

        void stop() {

            debugr.log(Level.DEBUG, "stop");

            scheduler.stop();

        }

        AtomicInteger count = new AtomicInteger(0);

        // work function where it all happens

        void processData() {

            try {

                debugr.log(Level.DEBUG, () -> "processData: " + readBuf.remaining()

                           + " bytes to unwrap "

                           + states(handshakeState)

                           + ", " + engine.getHandshakeStatus());

                int len;

                boolean complete = false;

                while ((len = readBuf.remaining()) > 0) {

                    boolean handshaking = false;

                    try {

                        EngineResult result;

                        synchronized (readBufferLock) {

                            complete = this.completing;

                            result = unwrapBuffer(readBuf);

                            debugr.log(Level.DEBUG, "Unwrapped: %s", result.result);

                        }

                        if (result.bytesProduced() > 0) {

                            debugr.log(Level.DEBUG, "sending %d", result.bytesProduced());

                            count.addAndGet(result.bytesProduced());

                            outgoing(result.destBuffer, false);

                        }

                        if (result.status() == Status.BUFFER_UNDERFLOW) {

                            debugr.log(Level.DEBUG, "BUFFER_UNDERFLOW");

                            // not enough data in the read buffer...

                            requestMore();

                            synchronized (readBufferLock) {

                                // check if we have received some data

                                if (readBuf.remaining() > len) continue;

                                return;

                            }

                        }

                        if (complete && result.status() == Status.CLOSED) {

                            debugr.log(Level.DEBUG, "Closed: completing");

                            outgoing(Utils.EMPTY_BB_LIST, true);

                            return;

                        }

                        if (result.handshaking() && !complete) {

                            debugr.log(Level.DEBUG, "handshaking");

                            doHandshake(result, READER);

                            resumeActivity();

                            handshaking = true;

                        } else {

                            if ((handshakeState.getAndSet(NOT_HANDSHAKING) & ~DOING_TASKS) == HANDSHAKING) {

                                setALPN();

                                handshaking = false;

                                resumeActivity();

                            }

                        }

                    } catch (IOException ex) {

                        handleError(ex);

                    }

                    if (handshaking && !complete)

                        return;

                }

                if (!complete) {

                    synchronized (readBufferLock) {

                        complete = this.completing && !readBuf.hasRemaining();

                    }

                }

                if (complete) {

                    debugr.log(Level.DEBUG, "completing");

                    // Complete the alpnCF, if not already complete, regardless of

                    // whether or not the ALPN is available, there will be no more

                    // activity.

                    setALPN();

                    outgoing(Utils.EMPTY_BB_LIST, true);

                }

            } catch (Throwable ex) {

                handleError(ex);

            }

        }

    }

    public interface Monitorable {

        public String getInfo();

    }

    public static class Monitor extends Thread {

        final List<Monitorable> list;

        static Monitor themon;

        static {

            themon = new Monitor();

            themon.start(); // uncomment to enable Monitor

        }

        Monitor() {

            super("Monitor");

            setDaemon(true);

            list = Collections.synchronizedList(new LinkedList<>());

        }

        void addTarget(Monitorable o) {

            list.add(o);

        }

        public static void add(Monitorable o) {

            themon.addTarget(o);

        }

        @Override

        public void run() {

            System.out.println("Monitor starting");

            try {

                while (true) {

                    Thread.sleep(20 * 1000);

                    synchronized (list) {

                        for (Monitorable o : list) {

                            System.out.println(o.getInfo());

                            System.out.println("-------------------------");

                        }

                    }

                    System.out.println("--o-o-o-o-o-o-o-o-o-o-o-o-o-o-");

                }

            } catch (InterruptedException e) {

                System.out.println("Monitor exiting with " + e);

            }

        }

    }

    /**

     * Processing function for outgoing data. Pass it thru SSLEngine.wrap()

     * Any encrypted buffers generated are passed downstream to be written.

     * Status codes:

     *     NEED_UNWRAP: call reader.addData() with empty buffer

     *     NEED_WRAP: call addData() with empty buffer

     *     NEED_TASK: delegate task to executor

     *     BUFFER_OVERFLOW: allocate larger output buffer. Repeat wrap

     *     BUFFER_UNDERFLOW: shouldn't happen on writing side

     *     OK: return generated buffers

     */

    class Writer extends SubscriberWrapper {

        final SequentialScheduler scheduler;

        // queues of buffers received from upstream waiting

        // to be processed by the SSLEngine

        final List<ByteBuffer> writeList;

        final System.Logger debugw =

            Utils.getDebugLogger(this::dbgString, DEBUG);

        volatile boolean completing;

        boolean completed; // only accessed in processData

        class WriterDownstreamPusher extends SequentialScheduler.CompleteRestartableTask {

            @Override public void run() { processData(); }

        }

        Writer() {

            super();

            writeList = Collections.synchronizedList(new LinkedList<>());

            scheduler = new SequentialScheduler(new WriterDownstreamPusher());

        }

        @Override

        protected void incoming(List<ByteBuffer> buffers, boolean complete) {

            assert complete ? buffers ==  Utils.EMPTY_BB_LIST : true;

            assert buffers != Utils.EMPTY_BB_LIST ? complete == false : true;

            if (complete) {

                debugw.log(Level.DEBUG, "adding SENTINEL");

                completing = true;

                writeList.add(SENTINEL);

            } else {

                writeList.addAll(buffers);

            }

            debugw.log(Level.DEBUG, () -> "added " + buffers.size()

                        + " (" + Utils.remaining(buffers)

                        + " bytes) to the writeList");

            scheduler.runOrSchedule();

        }

        public final String dbgString() {

            return "SSL Writer(" + tubeName + ")";

        }

        protected void onSubscribe() {

            doHandshake(EngineResult.INIT, INIT);

            resumeActivity();

        }

        void schedule() {

            scheduler.runOrSchedule();

        }

        void stop() {

            debugw.log(Level.DEBUG, "stop");

            scheduler.stop();

        }

        @Override

        public boolean closing() {

            return closeNotifyReceived();

        }

        private boolean isCompleting() {

            return completing;

        }

        @Override

        protected long upstreamWindowUpdate(long currentWindow, long downstreamQsize) {

            if (writeList.size() > 10)

                return 0;

            else

                return super.upstreamWindowUpdate(currentWindow, downstreamQsize);

        }

        private boolean hsTriggered() {

            synchronized(writeList) {

                for (ByteBuffer b : writeList)

                    if (b == HS_TRIGGER)

                        return true;

                return false;

            }

        }

        private void processData() {