1 /*

     2  * Copyright (c) 2015, 2016, Oracle and/or its affiliates. All rights reserved.

     3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

     4  *

     5  * This code is free software; you can redistribute it and/or modify it

     6  * under the terms of the GNU General Public License version 2 only, as

     7  * published by the Free Software Foundation.  Oracle designates this

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

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

    10  *

    11  * This code is distributed in the hope that it will be useful, but WITHOUT

    12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

    13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

    14  * version 2 for more details (a copy is included in the LICENSE file that

    15  * accompanied this code).

    16  *

    17  * You should have received a copy of the GNU General Public License version

    18  * 2 along with this work; if not, write to the Free Software Foundation,

    19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

    20  *

    21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

    22  * or visit www.oracle.com if you need additional information or have any

    23  * questions.

    24  */

    25 

    26 package jdk.incubator.http;

    27 

    28 import jdk.incubator.http.internal.common.Utils;

    29 import jdk.incubator.http.internal.websocket.RawChannel;

    30 

    31 import java.io.IOException;

    32 import java.nio.ByteBuffer;

    33 import java.nio.channels.SelectableChannel;

    34 import java.nio.channels.SocketChannel;

    35 

    36 /*

    37  * Each RawChannel corresponds to a TCP connection (SocketChannel) but is

    38  * connected to a Selector and an ExecutorService for invoking the send and

    39  * receive callbacks. Also includes SSL processing.

    40  */

    41 final class RawChannelImpl implements RawChannel {

    42 

    43     private final HttpClientImpl client;

    44     private final HttpConnection connection;

    45     private final Object         initialLock = new Object();

    46     private ByteBuffer           initial;

    47 

    48     RawChannelImpl(HttpClientImpl client,

    49                    HttpConnection connection,

    50                    ByteBuffer initial)

    51             throws IOException

    52     {

    53         this.client = client;

    54         this.connection = connection;

    55         SocketChannel chan = connection.channel();

    56         client.cancelRegistration(chan);

    57         // Constructing a RawChannel is supposed to have a "hand over"

    58         // semantics, in other words if construction fails, the channel won't be

    59         // needed by anyone, in which case someone still needs to close it

    60         try {

    61             chan.configureBlocking(false);

    62         } catch (IOException e) {

    63             try {

    64                 chan.close();

    65             } catch (IOException e1) {

    66                 e.addSuppressed(e1);

    67             }

    68             throw e;

    69         }

    70         // empty the initial buffer into our own copy.

    71         synchronized (initialLock) {

    72             this.initial = initial.hasRemaining()

    73                     ? Utils.copy(initial)

    74                     : Utils.EMPTY_BYTEBUFFER;

    75         }

    76     }

    77 

    78     private class NonBlockingRawAsyncEvent extends AsyncEvent {

    79 

    80         private final RawEvent re;

    81 

    82         NonBlockingRawAsyncEvent(RawEvent re) {

    83             super(0); // !BLOCKING & !REPEATING

    84             this.re = re;

    85         }

    86 

    87         @Override

    88         public SelectableChannel channel() {

    89             return connection.channel();

    90         }

    91 

    92         @Override

    93         public int interestOps() {

    94             return re.interestOps();

    95         }

    96 

    97         @Override

    98         public void handle() {

    99             re.handle();

   100         }

   101 

   102         @Override

   103         public void abort() { }

   104     }

   105 

   106     @Override

   107     public void registerEvent(RawEvent event) throws IOException {

   108         client.registerEvent(new NonBlockingRawAsyncEvent(event));

   109     }

   110 

   111     @Override

   112     public ByteBuffer read() throws IOException {

   113         assert !connection.channel().isBlocking();

   114         return connection.read();

   115     }

   116 

   117     @Override

   118     public ByteBuffer initialByteBuffer() {

   119         synchronized (initialLock) {

   120             if (initial == null) {

   121                 throw new IllegalStateException();

   122             }

   123             ByteBuffer ref = initial;

   124             initial = null;

   125             return ref;

   126         }

   127     }

   128 

   129     @Override

   130     public long write(ByteBuffer[] src, int offset, int len) throws IOException {

   131         return connection.write(src, offset, len);

   132     }

   133 

   134     @Override

   135     public void shutdownInput() throws IOException {

   136         connection.shutdownInput();

   137     }

   138 

   139     @Override

   140     public void shutdownOutput() throws IOException {

   141         connection.shutdownOutput();

   142     }

   143 

   144     @Override

   145     public void close() throws IOException {

   146         connection.close();

   147     }

   148 }