src/java.base/share/classes/sun/nio/ch/NioSocketImpl.java
author alanb
Sun, 17 Feb 2019 10:00:26 +0000
branchniosocketimpl-branch
changeset 57188 1f2101ee432d
parent 57187 056911ad3ee7
child 57189 c56554b46dec
permissions -rw-r--r--
Change Socket.getInputStream/getOutputStream to return the same stream each time
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
     1
/*
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
     2
 * Copyright (c) 2018, Oracle and/or its affiliates. All rights reserved.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
     3
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
     4
 *
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
     5
 * This code is free software; you can redistribute it and/or modify it
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
     6
 * under the terms of the GNU General Public License version 2 only, as
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
     7
 * published by the Free Software Foundation.  Oracle designates this
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
     8
 * particular file as subject to the "Classpath" exception as provided
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
     9
 * by Oracle in the LICENSE file that accompanied this code.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    10
 *
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    11
 * This code is distributed in the hope that it will be useful, but WITHOUT
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    12
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    13
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    14
 * version 2 for more details (a copy is included in the LICENSE file that
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    15
 * accompanied this code).
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    16
 *
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    17
 * You should have received a copy of the GNU General Public License version
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    18
 * 2 along with this work; if not, write to the Free Software Foundation,
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    19
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    20
 *
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    21
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    22
 * or visit www.oracle.com if you need additional information or have any
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    23
 * questions.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    24
 */
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    25
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    26
package sun.nio.ch;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    27
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    28
import java.io.FileDescriptor;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    29
import java.io.IOException;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    30
import java.io.InputStream;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    31
import java.io.OutputStream;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    32
import java.lang.invoke.MethodHandles;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    33
import java.lang.invoke.VarHandle;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    34
import java.lang.reflect.Field;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    35
import java.net.InetAddress;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    36
import java.net.InetSocketAddress;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    37
import java.net.ProtocolFamily;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    38
import java.net.SocketAddress;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    39
import java.net.SocketException;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    40
import java.net.SocketImpl;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    41
import java.net.SocketOption;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    42
import java.net.SocketTimeoutException;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    43
import java.net.StandardProtocolFamily;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    44
import java.net.StandardSocketOptions;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    45
import java.net.UnknownHostException;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    46
import java.nio.ByteBuffer;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    47
import java.security.AccessController;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    48
import java.security.PrivilegedActionException;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    49
import java.security.PrivilegedExceptionAction;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    50
import java.util.Collections;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    51
import java.util.HashSet;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    52
import java.util.Objects;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    53
import java.util.Set;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    54
import java.util.concurrent.TimeUnit;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    55
import java.util.concurrent.locks.ReentrantLock;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    56
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    57
import jdk.internal.access.SharedSecrets;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    58
import jdk.internal.ref.CleanerFactory;
57186
997178749c87 Throw SocketException consistently after connection reset detected
alanb
parents: 57179
diff changeset
    59
import sun.net.ConnectionResetException;
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    60
import sun.net.NetHooks;
57176
726630bc6a4c Cleanup
michaelm
parents: 57174
diff changeset
    61
import sun.net.PlatformSocketImpl;
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    62
import sun.net.ResourceManager;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    63
import sun.net.ext.ExtendedSocketOptions;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    64
import sun.net.util.SocketExceptions;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    65
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    66
import static java.util.concurrent.TimeUnit.MILLISECONDS;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    67
import static java.util.concurrent.TimeUnit.NANOSECONDS;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    68
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    69
/**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    70
 * NIO based SocketImpl.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    71
 *
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    72
 * This implementation attempts to be compatible with legacy PlainSocketImpl,
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    73
 * including behavior and exceptions that are not specified by SocketImpl.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    74
 *
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    75
 * The underlying socket used by this SocketImpl is initially configured
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    76
 * blocking. If a connect, accept or read is attempted with a timeout then the
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    77
 * socket is changed to non-blocking mode. When in non-blocking mode, operations
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    78
 * that don't complete immediately will poll the socket.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    79
 */
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    80
57176
726630bc6a4c Cleanup
michaelm
parents: 57174
diff changeset
    81
public final class NioSocketImpl extends SocketImpl implements PlatformSocketImpl {
57186
997178749c87 Throw SocketException consistently after connection reset detected
alanb
parents: 57179
diff changeset
    82
    private static final NativeDispatcher nd = new SocketDispatcher(true);
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    83
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    84
    // The maximum number of bytes to read/write per syscall to avoid needing
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    85
    // a huge buffer from the temporary buffer cache
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    86
    private static final int MAX_BUFFER_SIZE = 128 * 1024;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    87
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    88
    // true if this is a SocketImpl for a ServerSocket
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    89
    private final boolean server;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    90
57111
a57c4dc7e2fe implConnect only requires one lock
alanb
parents: 57110
diff changeset
    91
    // Lock held when reading (also used when accepting or connecting)
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    92
    private final ReentrantLock readLock = new ReentrantLock();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    93
57111
a57c4dc7e2fe implConnect only requires one lock
alanb
parents: 57110
diff changeset
    94
    // Lock held when writing
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    95
    private final ReentrantLock writeLock = new ReentrantLock();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    96
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    97
    // The stateLock for read/changing state
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    98
    private final Object stateLock = new Object();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
    99
    private static final int ST_NEW = 0;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   100
    private static final int ST_UNCONNECTED = 1;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   101
    private static final int ST_CONNECTING = 2;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   102
    private static final int ST_CONNECTED = 3;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   103
    private static final int ST_CLOSING = 4;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   104
    private static final int ST_CLOSED = 5;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   105
    private volatile int state;  // need stateLock to change
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   106
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   107
    // set by SocketImpl.create, protected by stateLock
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   108
    private boolean stream;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   109
    private FileDescriptorCloser closer;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   111
    // lazily set to true when the socket is configured non-blocking
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   112
    private volatile boolean nonBlocking;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   113
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   114
    // used by connect/read/write/accept, protected by stateLock
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   115
    private long readerThread;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   116
    private long writerThread;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   117
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   118
    // used when SO_REUSEADDR is emulated
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   119
    private boolean isReuseAddress;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   120
57187
056911ad3ee7 Allow setTrafficClass on streams socket
alanb
parents: 57186
diff changeset
   121
    // cached value of IPV6_TCLASS or IP_TOS socket option
056911ad3ee7 Allow setTrafficClass on streams socket
alanb
parents: 57186
diff changeset
   122
    private int trafficClass;
056911ad3ee7 Allow setTrafficClass on streams socket
alanb
parents: 57186
diff changeset
   123
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   124
    // read or accept timeout in millis
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   125
    private volatile int timeout;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   126
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   127
    // flags to indicate if the connection is shutdown for input and output
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   128
    private volatile boolean isInputClosed;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   129
    private volatile boolean isOutputClosed;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   130
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   131
    /**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   132
     * Creates a instance of this SocketImpl.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   133
     * @param server true if this is a SocketImpl for a ServerSocket
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   134
     */
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   135
    public NioSocketImpl(boolean server) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   136
        this.server = server;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   137
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   138
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   139
    /**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   140
     * Returns true if the socket is open.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   141
     */
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   142
    private boolean isOpen() {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   143
        return state < ST_CLOSING;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   144
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   145
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   146
    /**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   147
     * Throws SocketException if the socket is not open.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   148
     */
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   149
    private void ensureOpen() throws SocketException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   150
        if (state >= ST_CLOSING)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   151
            throw new SocketException("Socket closed");
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   152
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   153
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   154
    /**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   155
     * Throws SocketException if the socket is not open and connected.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   156
     */
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   157
    private void ensureOpenAndConnected() throws SocketException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   158
        int state = this.state;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   159
        if (state < ST_CONNECTED)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   160
            throw new SocketException("Not connected");
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   161
        if (state > ST_CONNECTED)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   162
            throw new SocketException("Socket closed");
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   163
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   164
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   165
    /**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   166
     * Disables the current thread for scheduling purposes until the socket is
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   167
     * ready for I/O, or is asynchronously closed, for up to the specified
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   168
     * waiting time.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   169
     * @throws IOException if an I/O error occurs
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   170
     */
57128
3d6cee596b33 Minor cleanup to NioSocketImpl to avoid accessing fd field without stateLock
alanb
parents: 57124
diff changeset
   171
    private void park(FileDescriptor fd, int event, long nanos) throws IOException {
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   172
        long millis;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   173
        if (nanos == 0) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   174
            millis = -1;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   175
        } else {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   176
            millis = MILLISECONDS.convert(nanos, NANOSECONDS);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   177
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   178
        Net.poll(fd, event, millis);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   179
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   180
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   181
    /**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   182
     * Disables the current thread for scheduling purposes until the socket is
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   183
     * ready for I/O or is asynchronously closed.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   184
     * @throws IOException if an I/O error occurs
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   185
     */
57128
3d6cee596b33 Minor cleanup to NioSocketImpl to avoid accessing fd field without stateLock
alanb
parents: 57124
diff changeset
   186
    private void park(FileDescriptor fd, int event) throws IOException {
3d6cee596b33 Minor cleanup to NioSocketImpl to avoid accessing fd field without stateLock
alanb
parents: 57124
diff changeset
   187
        park(fd, event, 0);
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   188
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   189
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   190
    /**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   191
     * Ensures that the socket is configured non-blocking when a timeout is specified.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   192
     * @throws IOException if there is an I/O error changing the blocking mode
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   193
     */
57128
3d6cee596b33 Minor cleanup to NioSocketImpl to avoid accessing fd field without stateLock
alanb
parents: 57124
diff changeset
   194
    private void configureNonBlockingIfNeeded(FileDescriptor fd, int timeout)
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   195
        throws IOException
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   196
    {
57124
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   197
        if (timeout > 0 && !nonBlocking) {
57128
3d6cee596b33 Minor cleanup to NioSocketImpl to avoid accessing fd field without stateLock
alanb
parents: 57124
diff changeset
   198
            assert readLock.isHeldByCurrentThread() || writeLock.isHeldByCurrentThread();
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   199
            IOUtil.configureBlocking(fd, false);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   200
            nonBlocking = true;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   201
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   202
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   203
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   204
    /**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   205
     * Marks the beginning of a read operation that might block.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   206
     * @throws SocketException if the socket is closed or not connected
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   207
     */
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   208
    private FileDescriptor beginRead() throws SocketException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   209
        synchronized (stateLock) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   210
            ensureOpenAndConnected();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   211
            readerThread = NativeThread.current();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   212
            return fd;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   213
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   214
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   215
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   216
    /**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   217
     * Marks the end of a read operation that may have blocked.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   218
     * @throws SocketException is the socket is closed
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   219
     */
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   220
    private void endRead(boolean completed) throws SocketException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   221
        synchronized (stateLock) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   222
            readerThread = 0;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   223
            int state = this.state;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   224
            if (state == ST_CLOSING)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   225
                stateLock.notifyAll();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   226
            if (!completed && state >= ST_CLOSING)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   227
                throw new SocketException("Socket closed");
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   228
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   229
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   230
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   231
    /**
57124
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   232
     * Try to read bytes from the socket into the given byte array.
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   233
     */
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   234
    private int tryRead(FileDescriptor fd, byte[] b, int off, int len)
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   235
        throws IOException
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   236
    {
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   237
        ByteBuffer dst = Util.getTemporaryDirectBuffer(len);
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   238
        assert dst.position() == 0;
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   239
        try {
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   240
            int n = nd.read(fd, ((DirectBuffer)dst).address(), len);
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   241
            if (n > 0) {
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   242
                dst.get(b, off, n);
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   243
            }
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   244
            return n;
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   245
        } finally{
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   246
            Util.offerFirstTemporaryDirectBuffer(dst);
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   247
        }
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   248
    }
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   249
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   250
    /**
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   251
     * Reads bytes from the socket into the given byte array.
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   252
     * @return the number of bytes read
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   253
     * @throws IOException if the socket is closed or an I/O occurs
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   254
     * @throws SocketTimeoutException if the read timeout elapses
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   255
     */
57124
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   256
    private int read(byte[] b, int off, int len) throws IOException {
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   257
        readLock.lock();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   258
        try {
57128
3d6cee596b33 Minor cleanup to NioSocketImpl to avoid accessing fd field without stateLock
alanb
parents: 57124
diff changeset
   259
            int timeout = this.timeout;
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   260
            int n = 0;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   261
            FileDescriptor fd = beginRead();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   262
            try {
57128
3d6cee596b33 Minor cleanup to NioSocketImpl to avoid accessing fd field without stateLock
alanb
parents: 57124
diff changeset
   263
                if (isInputClosed)
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   264
                    return IOStatus.EOF;
57128
3d6cee596b33 Minor cleanup to NioSocketImpl to avoid accessing fd field without stateLock
alanb
parents: 57124
diff changeset
   265
                configureNonBlockingIfNeeded(fd, timeout);
57124
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   266
                n = tryRead(fd, b, off, len);
57114
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   267
                if (IOStatus.okayToRetry(n) && isOpen()) {
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   268
                    if (timeout > 0) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   269
                        // read with timeout
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   270
                        long nanos = NANOSECONDS.convert(timeout, TimeUnit.MILLISECONDS);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   271
                        do {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   272
                            long startTime = System.nanoTime();
57128
3d6cee596b33 Minor cleanup to NioSocketImpl to avoid accessing fd field without stateLock
alanb
parents: 57124
diff changeset
   273
                            park(fd, Net.POLLIN, nanos);
57124
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   274
                            n = tryRead(fd, b, off, len);
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   275
                            if (n == IOStatus.UNAVAILABLE) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   276
                                nanos -= System.nanoTime() - startTime;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   277
                                if (nanos <= 0)
57178
738431fa8fb2 Improve clarity of implAccept, align timeout messages with PSI
alanb
parents: 57176
diff changeset
   278
                                    throw new SocketTimeoutException("Read timed out");
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   279
                            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   280
                        } while (n == IOStatus.UNAVAILABLE && isOpen());
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   281
                    } else {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   282
                        // read, no timeout
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   283
                        do {
57128
3d6cee596b33 Minor cleanup to NioSocketImpl to avoid accessing fd field without stateLock
alanb
parents: 57124
diff changeset
   284
                            park(fd, Net.POLLIN);
57124
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   285
                            n = tryRead(fd, b, off, len);
57114
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   286
                        } while (IOStatus.okayToRetry(n) && isOpen());
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   287
                    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   288
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   289
                return n;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   290
            } finally {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   291
                endRead(n > 0);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   292
            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   293
        } finally {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   294
            readLock.unlock();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   295
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   296
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   297
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   298
    /**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   299
     * Marks the beginning of a write operation that might block.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   300
     * @throws SocketException if the socket is closed or not connected
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   301
     */
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   302
    private FileDescriptor beginWrite() throws SocketException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   303
        synchronized (stateLock) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   304
            ensureOpenAndConnected();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   305
            writerThread = NativeThread.current();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   306
            return fd;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   307
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   308
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   309
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   310
    /**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   311
     * Marks the end of a write operation that may have blocked.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   312
     * @throws SocketException is the socket is closed
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   313
     */
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   314
    private void endWrite(boolean completed) throws SocketException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   315
        synchronized (stateLock) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   316
            writerThread = 0;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   317
            int state = this.state;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   318
            if (state == ST_CLOSING)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   319
                stateLock.notifyAll();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   320
            if (!completed && state >= ST_CLOSING)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   321
                throw new SocketException("Socket closed");
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   322
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   323
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   324
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   325
    /**
57124
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   326
     * Try to write a sequence of bytes to this socket from the given byte array
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   327
     */
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   328
    private int tryWrite(FileDescriptor fd, byte[] b, int off, int len)
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   329
        throws IOException
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   330
    {
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   331
        ByteBuffer src = Util.getTemporaryDirectBuffer(len);
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   332
        assert src.position() == 0;
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   333
        try {
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   334
            src.put(b, off, len);
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   335
            return nd.write(fd, ((DirectBuffer)src).address(), len);
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   336
        } finally {
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   337
            Util.offerFirstTemporaryDirectBuffer(src);
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   338
        }
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   339
    }
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   340
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   341
    /**
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   342
     * Writes a sequence of bytes to this socket from the given byte array.
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   343
     * @return the number of bytes written
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   344
     * @throws IOException if the socket is closed or an I/O occurs
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   345
     */
57124
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   346
    private int write(byte[] b, int off, int len) throws IOException {
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   347
        writeLock.lock();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   348
        try {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   349
            int n = 0;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   350
            FileDescriptor fd = beginWrite();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   351
            try {
57124
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   352
                n = tryWrite(fd, b, off, len);
57114
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   353
                while (IOStatus.okayToRetry(n) && isOpen()) {
57128
3d6cee596b33 Minor cleanup to NioSocketImpl to avoid accessing fd field without stateLock
alanb
parents: 57124
diff changeset
   354
                    park(fd, Net.POLLOUT);
57124
8bb7df86576a Improve read/write performance
alanb
parents: 57114
diff changeset
   355
                    n = tryWrite(fd, b, off, len);
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   356
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   357
                return n;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   358
            } finally {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   359
                endWrite(n > 0);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   360
            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   361
        } finally {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   362
            writeLock.unlock();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   363
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   364
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   365
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   366
    /**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   367
     * Creates the socket.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   368
     * @param stream {@code true} for a streams socket
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   369
     */
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   370
    @Override
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   371
    protected void create(boolean stream) throws IOException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   372
        synchronized (stateLock) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   373
            assert state == ST_NEW;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   374
            if (!stream)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   375
                ResourceManager.beforeUdpCreate();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   376
            FileDescriptor fd;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   377
            try {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   378
                if (server) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   379
                    assert stream;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   380
                    fd = Net.serverSocket(true);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   381
                } else {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   382
                    fd = Net.socket(stream);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   383
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   384
            } catch (IOException ioe) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   385
                if (!stream)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   386
                    ResourceManager.afterUdpClose();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   387
                throw ioe;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   388
            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   389
            this.fd = fd;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   390
            this.stream = stream;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   391
            this.closer = FileDescriptorCloser.create(this);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   392
            this.state = ST_UNCONNECTED;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   393
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   394
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   395
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   396
    /**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   397
     * For use by ServerSocket to set the state and other fields after a
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   398
     * connection is accepted by a ServerSocket using a custom SocketImpl.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   399
     * The protected fields defined by SocketImpl should be set.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   400
     */
57174
899641440751 More cleanup
alanb
parents: 57172
diff changeset
   401
    @Override
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   402
    public void postCustomAccept() throws IOException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   403
        synchronized (stateLock) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   404
            assert state == ST_NEW;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   405
            assert fd.valid() && localport != 0 && address != null && port != 0;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   406
            IOUtil.configureBlocking(fd, true);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   407
            stream = true;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   408
            closer = FileDescriptorCloser.create(this);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   409
            state = ST_CONNECTED;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   410
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   411
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   412
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   413
    /**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   414
     * For use by ServerSocket to copy the state from this connected SocketImpl
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   415
     * to a target SocketImpl. If the target SocketImpl is not a newly created
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   416
     * SocketImpl then it is first closed to release any resources. The target
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   417
     * SocketImpl becomes the owner of the file descriptor, this SocketImpl
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   418
     * is marked as closed and should be discarded.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   419
     */
57174
899641440751 More cleanup
alanb
parents: 57172
diff changeset
   420
    @Override
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   421
    public void copyTo(SocketImpl si) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   422
        if (si instanceof NioSocketImpl) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   423
            NioSocketImpl nsi = (NioSocketImpl) si;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   424
            if (nsi.state != ST_NEW) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   425
                try {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   426
                    nsi.close();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   427
                } catch (IOException ignore) { }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   428
            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   429
            synchronized (nsi.stateLock) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   430
                assert nsi.state == ST_NEW || nsi.state == ST_CLOSED;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   431
                synchronized (this.stateLock) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   432
                    // this SocketImpl should be connected
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   433
                    assert state == ST_CONNECTED && fd.valid()
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   434
                        && localport != 0 && address != null && port != 0;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   435
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   436
                    // copy fields
57179
1774a7b73c72 Extend field resetting in copyTo(SocketImpl) implementations
alanb
parents: 57178
diff changeset
   437
                    nsi.stream = this.stream;
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   438
                    nsi.fd = this.fd;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   439
                    nsi.localport = this.localport;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   440
                    nsi.address = this.address;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   441
                    nsi.port = this.port;
57179
1774a7b73c72 Extend field resetting in copyTo(SocketImpl) implementations
alanb
parents: 57178
diff changeset
   442
1774a7b73c72 Extend field resetting in copyTo(SocketImpl) implementations
alanb
parents: 57178
diff changeset
   443
                    // reset fields; do not reset timeout
1774a7b73c72 Extend field resetting in copyTo(SocketImpl) implementations
alanb
parents: 57178
diff changeset
   444
                    nsi.nonBlocking = false;
1774a7b73c72 Extend field resetting in copyTo(SocketImpl) implementations
alanb
parents: 57178
diff changeset
   445
                    nsi.isInputClosed = false;
1774a7b73c72 Extend field resetting in copyTo(SocketImpl) implementations
alanb
parents: 57178
diff changeset
   446
                    nsi.isOutputClosed = false;
1774a7b73c72 Extend field resetting in copyTo(SocketImpl) implementations
alanb
parents: 57178
diff changeset
   447
                    nsi.isReuseAddress = false;
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   448
                    nsi.state = ST_CONNECTED;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   449
57179
1774a7b73c72 Extend field resetting in copyTo(SocketImpl) implementations
alanb
parents: 57178
diff changeset
   450
                    // GC'ing of this impl should not close the file descriptor
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   451
                    this.closer.disable();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   452
                    this.state = ST_CLOSED;
57179
1774a7b73c72 Extend field resetting in copyTo(SocketImpl) implementations
alanb
parents: 57178
diff changeset
   453
1774a7b73c72 Extend field resetting in copyTo(SocketImpl) implementations
alanb
parents: 57178
diff changeset
   454
                    // create new closer to execute when nsi is GC'ed
1774a7b73c72 Extend field resetting in copyTo(SocketImpl) implementations
alanb
parents: 57178
diff changeset
   455
                    nsi.closer = FileDescriptorCloser.create(nsi);
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   456
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   457
            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   458
        } else {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   459
            synchronized (this.stateLock) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   460
                // this SocketImpl should be connected
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   461
                assert state == ST_CONNECTED && fd.valid()
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   462
                        && localport != 0 && address != null && port != 0;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   463
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   464
                // set fields in foreign impl
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   465
                setSocketImplFields(si, fd, localport, address, port);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   466
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   467
                // disable closer to prevent GC'ing of this impl from
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   468
                // closing the file descriptor
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   469
                this.closer.disable();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   470
                this.state = ST_CLOSED;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   471
            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   472
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   473
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   474
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   475
    /**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   476
     * Marks the beginning of a connect operation that might block.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   477
     * @throws SocketException if the socket is closed or already connected
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   478
     */
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   479
    private FileDescriptor beginConnect(InetAddress address, int port)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   480
        throws IOException
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   481
    {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   482
        synchronized (stateLock) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   483
            int state = this.state;
57114
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   484
            if (state != ST_UNCONNECTED) {
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   485
                if (state == ST_CONNECTING)
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   486
                    throw new SocketException("Connection in progress");
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   487
                if (state == ST_CONNECTED)
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   488
                    throw new SocketException("Already connected");
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   489
                if (state >= ST_CLOSING)
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   490
                    throw new SocketException("Socket closed");
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   491
                assert false;
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   492
            }
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   493
            this.state = ST_CONNECTING;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   494
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   495
            // invoke beforeTcpConnect hook if not already bound
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   496
            if (localport == 0) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   497
                NetHooks.beforeTcpConnect(fd, address, port);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   498
            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   499
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   500
            // save the remote address/port
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   501
            this.address = address;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   502
            this.port = port;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   503
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   504
            readerThread = NativeThread.current();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   505
            return fd;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   506
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   507
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   508
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   509
    /**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   510
     * Marks the end of a connect operation that may have blocked.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   511
     * @throws SocketException is the socket is closed
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   512
     */
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   513
    private void endConnect(boolean completed) throws IOException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   514
        synchronized (stateLock) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   515
            readerThread = 0;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   516
            int state = this.state;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   517
            if (state == ST_CLOSING)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   518
                stateLock.notifyAll();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   519
            if (completed && state == ST_CONNECTING) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   520
                this.state = ST_CONNECTED;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   521
                localport = Net.localAddress(fd).getPort();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   522
            } else if (!completed && state >= ST_CLOSING) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   523
                throw new SocketException("Socket closed");
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   524
            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   525
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   526
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   527
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   528
    /**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   529
     * Connect the socket. Closes the socket if connection cannot be established.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   530
     * @throws IllegalArgumentException if the address is not an InetSocketAddress
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   531
     * @throws UnknownHostException if the InetSocketAddress is not resolved
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   532
     * @throws IOException if the connection cannot be established
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   533
     */
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   534
    private void implConnect(SocketAddress remote, int millis) throws IOException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   535
        if (!(remote instanceof InetSocketAddress))
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   536
            throw new IllegalArgumentException("Unsupported address type");
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   537
        InetSocketAddress isa = (InetSocketAddress) remote;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   538
        if (isa.isUnresolved()) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   539
            throw new UnknownHostException(isa.getHostName());
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   540
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   541
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   542
        InetAddress address = isa.getAddress();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   543
        if (address.isAnyLocalAddress())
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   544
            address = InetAddress.getLocalHost();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   545
        int port = isa.getPort();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   546
57114
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   547
        ReentrantLock connectLock = readLock;
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   548
        try {
57114
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   549
            connectLock.lock();
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   550
            try {
57111
a57c4dc7e2fe implConnect only requires one lock
alanb
parents: 57110
diff changeset
   551
                boolean connected = false;
a57c4dc7e2fe implConnect only requires one lock
alanb
parents: 57110
diff changeset
   552
                FileDescriptor fd = beginConnect(address, port);
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   553
                try {
57128
3d6cee596b33 Minor cleanup to NioSocketImpl to avoid accessing fd field without stateLock
alanb
parents: 57124
diff changeset
   554
                    configureNonBlockingIfNeeded(fd, millis);
57111
a57c4dc7e2fe implConnect only requires one lock
alanb
parents: 57110
diff changeset
   555
                    int n = Net.connect(fd, address, port);
57114
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   556
                    if (IOStatus.okayToRetry(n) && isOpen()) {
57111
a57c4dc7e2fe implConnect only requires one lock
alanb
parents: 57110
diff changeset
   557
                        if (millis > 0) {
a57c4dc7e2fe implConnect only requires one lock
alanb
parents: 57110
diff changeset
   558
                            // connect with timeout
a57c4dc7e2fe implConnect only requires one lock
alanb
parents: 57110
diff changeset
   559
                            assert nonBlocking;
a57c4dc7e2fe implConnect only requires one lock
alanb
parents: 57110
diff changeset
   560
                            long nanos = NANOSECONDS.convert(millis, MILLISECONDS);
a57c4dc7e2fe implConnect only requires one lock
alanb
parents: 57110
diff changeset
   561
                            do {
a57c4dc7e2fe implConnect only requires one lock
alanb
parents: 57110
diff changeset
   562
                                long startTime = System.nanoTime();
57128
3d6cee596b33 Minor cleanup to NioSocketImpl to avoid accessing fd field without stateLock
alanb
parents: 57124
diff changeset
   563
                                park(fd, Net.POLLOUT, nanos);
57114
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   564
                                n = Net.pollConnectNow(fd);
57111
a57c4dc7e2fe implConnect only requires one lock
alanb
parents: 57110
diff changeset
   565
                                if (n == 0) {
a57c4dc7e2fe implConnect only requires one lock
alanb
parents: 57110
diff changeset
   566
                                    nanos -= System.nanoTime() - startTime;
a57c4dc7e2fe implConnect only requires one lock
alanb
parents: 57110
diff changeset
   567
                                    if (nanos <= 0)
57178
738431fa8fb2 Improve clarity of implAccept, align timeout messages with PSI
alanb
parents: 57176
diff changeset
   568
                                        throw new SocketTimeoutException("Connect timed out");
57111
a57c4dc7e2fe implConnect only requires one lock
alanb
parents: 57110
diff changeset
   569
                                }
a57c4dc7e2fe implConnect only requires one lock
alanb
parents: 57110
diff changeset
   570
                            } while (n == 0 && isOpen());
a57c4dc7e2fe implConnect only requires one lock
alanb
parents: 57110
diff changeset
   571
                        } else {
a57c4dc7e2fe implConnect only requires one lock
alanb
parents: 57110
diff changeset
   572
                            // connect, no timeout
a57c4dc7e2fe implConnect only requires one lock
alanb
parents: 57110
diff changeset
   573
                            do {
57128
3d6cee596b33 Minor cleanup to NioSocketImpl to avoid accessing fd field without stateLock
alanb
parents: 57124
diff changeset
   574
                                park(fd, Net.POLLOUT);
57114
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   575
                                n = Net.pollConnectNow(fd);
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   576
                            } while (n == 0 && isOpen());
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   577
                        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   578
                    }
57111
a57c4dc7e2fe implConnect only requires one lock
alanb
parents: 57110
diff changeset
   579
                    connected = (n > 0) && isOpen();
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   580
                } finally {
57111
a57c4dc7e2fe implConnect only requires one lock
alanb
parents: 57110
diff changeset
   581
                    endConnect(connected);
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   582
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   583
            } finally {
57114
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   584
                connectLock.unlock();
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   585
            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   586
        } catch (IOException ioe) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   587
            close();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   588
            throw SocketExceptions.of(ioe, isa);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   589
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   590
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   591
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   592
    @Override
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   593
    protected void connect(String host, int port) throws IOException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   594
        implConnect(new InetSocketAddress(host, port), timeout);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   595
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   596
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   597
    @Override
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   598
    protected void connect(InetAddress address, int port) throws IOException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   599
        implConnect(new InetSocketAddress(address, port), timeout);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   600
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   601
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   602
    @Override
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   603
    protected void connect(SocketAddress address, int timeout) throws IOException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   604
        implConnect(address, timeout);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   605
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   606
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   607
    @Override
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   608
    protected void bind(InetAddress host, int port) throws IOException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   609
        synchronized (stateLock) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   610
            ensureOpen();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   611
            if (localport != 0)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   612
                throw new SocketException("Already bound");
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   613
            NetHooks.beforeTcpBind(fd, host, port);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   614
            Net.bind(fd, host, port);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   615
            // set the address field to the address specified to the method to
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   616
            // keep compatibility with PlainSocketImpl. When binding to 0.0.0.0
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   617
            // then the actual local address will be ::0 when IPv6 is enabled.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   618
            address = host;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   619
            localport = Net.localAddress(fd).getPort();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   620
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   621
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   622
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   623
    @Override
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   624
    protected void listen(int backlog) throws IOException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   625
        synchronized (stateLock) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   626
            ensureOpen();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   627
            if (localport == 0)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   628
                throw new SocketException("Not bound");
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   629
            Net.listen(fd, backlog < 1 ? 50 : backlog);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   630
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   631
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   632
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   633
    /**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   634
     * Marks the beginning of an accept operation that might block.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   635
     * @throws SocketException if the socket is closed
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   636
     */
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   637
    private FileDescriptor beginAccept() throws SocketException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   638
        synchronized (stateLock) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   639
            ensureOpen();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   640
            if (!stream)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   641
                throw new SocketException("Not a stream socket");
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   642
            if (localport == 0)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   643
                throw new SocketException("Not bound");
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   644
            readerThread = NativeThread.current();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   645
            return fd;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   646
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   647
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   648
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   649
    /**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   650
     * Marks the end of an accept operation that may have blocked.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   651
     * @throws SocketException is the socket is closed
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   652
     */
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   653
    private void endAccept(boolean completed) throws SocketException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   654
        synchronized (stateLock) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   655
            int state = this.state;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   656
            readerThread = 0;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   657
            if (state == ST_CLOSING)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   658
                stateLock.notifyAll();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   659
            if (!completed && state >= ST_CLOSING)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   660
                throw new SocketException("Socket closed");
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   661
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   662
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   663
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   664
    @Override
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   665
    protected void accept(SocketImpl si) throws IOException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   666
        // accept a connection
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   667
        FileDescriptor newfd = new FileDescriptor();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   668
        InetSocketAddress[] isaa = new InetSocketAddress[1];
57114
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   669
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   670
        ReentrantLock acceptLock = readLock;
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   671
        acceptLock.lock();
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   672
        try {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   673
            int n = 0;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   674
            FileDescriptor fd = beginAccept();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   675
            try {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   676
                int timeout = this.timeout;
57128
3d6cee596b33 Minor cleanup to NioSocketImpl to avoid accessing fd field without stateLock
alanb
parents: 57124
diff changeset
   677
                configureNonBlockingIfNeeded(fd, timeout);
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   678
                n = ServerSocketChannelImpl.accept0(fd, newfd, isaa);
57114
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   679
                if (IOStatus.okayToRetry(n) && isOpen()) {
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   680
                    if (timeout > 0) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   681
                        // accept with timeout
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   682
                        assert nonBlocking;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   683
                        long nanos = NANOSECONDS.convert(timeout, TimeUnit.MILLISECONDS);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   684
                        do {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   685
                            long startTime = System.nanoTime();
57128
3d6cee596b33 Minor cleanup to NioSocketImpl to avoid accessing fd field without stateLock
alanb
parents: 57124
diff changeset
   686
                            park(fd, Net.POLLIN, nanos);
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   687
                            n = ServerSocketChannelImpl.accept0(fd, newfd, isaa);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   688
                            if (n == IOStatus.UNAVAILABLE) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   689
                                nanos -= System.nanoTime() - startTime;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   690
                                if (nanos <= 0)
57178
738431fa8fb2 Improve clarity of implAccept, align timeout messages with PSI
alanb
parents: 57176
diff changeset
   691
                                    throw new SocketTimeoutException("Accept timed out");
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   692
                            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   693
                        } while (n == IOStatus.UNAVAILABLE && isOpen());
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   694
                    } else {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   695
                        // accept, no timeout
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   696
                        do {
57128
3d6cee596b33 Minor cleanup to NioSocketImpl to avoid accessing fd field without stateLock
alanb
parents: 57124
diff changeset
   697
                            park(fd, Net.POLLIN);
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   698
                            n = ServerSocketChannelImpl.accept0(fd, newfd, isaa);
57114
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   699
                        } while (IOStatus.okayToRetry(n) && isOpen());
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   700
                    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   701
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   702
            } finally {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   703
                endAccept(n > 0);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   704
                assert IOStatus.check(n);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   705
            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   706
        } finally {
57114
e613cc3bc9d4 NioSocketImpl cleanup
alanb
parents: 57111
diff changeset
   707
            acceptLock.unlock();
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   708
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   709
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   710
        // get local address and configure accepted socket to blocking mode
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   711
        InetSocketAddress localAddress;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   712
        try {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   713
            localAddress = Net.localAddress(newfd);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   714
            IOUtil.configureBlocking(newfd, true);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   715
        } catch (IOException ioe) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   716
            nd.close(newfd);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   717
            throw ioe;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   718
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   719
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   720
        // set the fields
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   721
        InetSocketAddress remoteAddress = isaa[0];
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   722
        if (si instanceof NioSocketImpl) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   723
            NioSocketImpl nsi = (NioSocketImpl) si;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   724
            synchronized (nsi.stateLock) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   725
                nsi.fd = newfd;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   726
                nsi.stream = true;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   727
                nsi.closer = FileDescriptorCloser.create(nsi);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   728
                nsi.localport = localAddress.getPort();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   729
                nsi.address = remoteAddress.getAddress();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   730
                nsi.port = remoteAddress.getPort();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   731
                nsi.state = ST_CONNECTED;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   732
            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   733
        } else {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   734
            // set fields in foreign impl
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   735
            setSocketImplFields(si, newfd,
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   736
                                localAddress.getPort(),
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   737
                                remoteAddress.getAddress(),
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   738
                                remoteAddress.getPort());
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   739
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   740
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   741
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   742
    @Override
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   743
    protected InputStream getInputStream() {
57188
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   744
        return new InputStream() {
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   745
            // EOF or connection reset detected, not thread safe
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   746
            private boolean eof, reset;
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   747
            @Override
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   748
            public int read() throws IOException {
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   749
                byte[] a = new byte[1];
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   750
                int n = read(a, 0, 1);
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   751
                return (n > 0) ? (a[0] & 0xff) : -1;
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   752
            }
57188
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   753
            @Override
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   754
            public int read(byte[] b, int off, int len) throws IOException {
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   755
                Objects.checkFromIndexSize(off, len, b.length);
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   756
                if (eof) {
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   757
                    return -1;
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   758
                } else if (reset) {
57186
997178749c87 Throw SocketException consistently after connection reset detected
alanb
parents: 57179
diff changeset
   759
                    throw new SocketException("Connection reset");
57188
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   760
                } else if (len == 0) {
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   761
                    return 0;
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   762
                } else {
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   763
                    try {
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   764
                        // read up to MAX_BUFFER_SIZE bytes
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   765
                        int size = Math.min(len, MAX_BUFFER_SIZE);
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   766
                        int n = NioSocketImpl.this.read(b, off, size);
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   767
                        if (n == -1)
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   768
                            eof = true;
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   769
                        return n;
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   770
                    } catch (ConnectionResetException e) {
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   771
                        reset = true;
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   772
                        throw new SocketException("Connection reset");
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   773
                    } catch (SocketTimeoutException e) {
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   774
                        throw e;
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   775
                    } catch (IOException ioe) {
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   776
                        throw new SocketException(ioe.getMessage());
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   777
                    }
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   778
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   779
            }
57188
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   780
            @Override
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   781
            public int available() throws IOException {
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   782
                return NioSocketImpl.this.available();
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   783
            }
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   784
            @Override
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   785
            public void close() throws IOException {
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   786
                NioSocketImpl.this.close();
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   787
            }
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   788
        };
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   789
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   790
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   791
    @Override
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   792
    protected OutputStream getOutputStream() {
57188
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   793
        return new OutputStream() {
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   794
            @Override
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   795
            public void write(int b) throws IOException {
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   796
                byte[] a = new byte[]{(byte) b};
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   797
                write(a, 0, 1);
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   798
            }
57188
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   799
            @Override
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   800
            public void write(byte[] b, int off, int len) throws IOException {
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   801
                Objects.checkFromIndexSize(off, len, b.length);
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   802
                if (len > 0) {
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   803
                    try {
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   804
                        int pos = off;
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   805
                        int end = off + len;
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   806
                        while (pos < end) {
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   807
                            // write up to MAX_BUFFER_SIZE bytes
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   808
                            int size = Math.min((end - pos), MAX_BUFFER_SIZE);
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   809
                            int n = NioSocketImpl.this.write(b, pos, size);
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   810
                            pos += n;
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   811
                        }
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   812
                    } catch (IOException ioe) {
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   813
                        throw new SocketException(ioe.getMessage());
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   814
                    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   815
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   816
            }
57188
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   817
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   818
            @Override
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   819
            public void close() throws IOException {
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   820
                NioSocketImpl.this.close();
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   821
            }
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   822
        };
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   823
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   824
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   825
    @Override
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   826
    protected int available() throws IOException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   827
        readLock.lock();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   828
        try {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   829
            ensureOpenAndConnected();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   830
            if (isInputClosed) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   831
                return 0;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   832
            } else {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   833
                return Net.available(fd);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   834
            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   835
        } finally {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   836
            readLock.unlock();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   837
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   838
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   839
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   840
    /**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   841
     * Closes the socket, signalling and waiting for blocking I/O operations
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   842
     * to complete.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   843
     */
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   844
    @Override
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   845
    protected void close() throws IOException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   846
        boolean interrupted = false;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   847
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   848
        synchronized (stateLock) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   849
            int state = this.state;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   850
            if (state >= ST_CLOSING)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   851
                return;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   852
            if (state == ST_NEW) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   853
                // stillborn
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   854
                this.state = ST_CLOSED;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   855
                return;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   856
            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   857
            this.state = ST_CLOSING;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   858
            assert fd != null && closer != null;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   859
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   860
            // shutdown output when linger interval not set
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   861
            try {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   862
                var SO_LINGER = StandardSocketOptions.SO_LINGER;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   863
                if ((int) Net.getSocketOption(fd, SO_LINGER) != 0) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   864
                    Net.shutdown(fd, Net.SHUT_WR);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   865
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   866
            } catch (IOException ignore) { }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   867
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   868
            // interrupt and wait for kernel threads to complete I/O operations
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   869
            long reader = readerThread;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   870
            long writer = writerThread;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   871
            if (reader != 0 || writer != 0) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   872
                nd.preClose(fd);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   873
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   874
                if (reader != 0)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   875
                    NativeThread.signal(reader);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   876
                if (writer != 0)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   877
                    NativeThread.signal(writer);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   878
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   879
                // wait for blocking I/O operations to end
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   880
                while (readerThread != 0 || writerThread != 0) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   881
                    try {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   882
                        stateLock.wait();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   883
                    } catch (InterruptedException e) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   884
                        interrupted = true;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   885
                    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   886
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   887
            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   888
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   889
            // close file descriptor
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   890
            try {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   891
                closer.run();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   892
            } finally {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   893
                this.state = ST_CLOSED;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   894
            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   895
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   896
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   897
        // restore interrupt status
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   898
        if (interrupted)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   899
            Thread.currentThread().interrupt();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   900
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   901
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   902
    @Override
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   903
    protected Set<SocketOption<?>> supportedOptions() {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   904
        Set<SocketOption<?>> options = new HashSet<>();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   905
        options.addAll(super.supportedOptions());
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   906
        if (server) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   907
            options.addAll(ExtendedSocketOptions.serverSocketOptions());
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   908
        } else {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   909
            options.addAll(ExtendedSocketOptions.clientSocketOptions());
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   910
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   911
        if (Net.isReusePortAvailable())
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   912
            options.add(StandardSocketOptions.SO_REUSEPORT);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   913
        return Collections.unmodifiableSet(options);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   914
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   915
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   916
    private boolean booleanValue(Object value, String desc) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   917
        if (!(value instanceof Boolean))
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   918
            throw new IllegalArgumentException("Bad value for " + desc);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   919
        return (boolean) value;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   920
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   921
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   922
    private int intValue(Object value, String desc) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   923
        if (!(value instanceof Integer))
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   924
            throw new IllegalArgumentException("Bad value for " + desc);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   925
        return (int) value;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   926
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   927
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   928
    @Override
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   929
    public void setOption(int opt, Object value) throws SocketException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   930
        synchronized (stateLock) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   931
            ensureOpen();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   932
            try {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   933
                switch (opt) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   934
                case SO_LINGER: {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   935
                    // the value is "false" to disable, or linger interval to enable
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   936
                    int i;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   937
                    if (value instanceof Boolean && ((boolean) value) == false) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   938
                        i = -1;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   939
                    } else {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   940
                        i = intValue(value, "SO_LINGER");
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   941
                    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   942
                    Net.setSocketOption(fd, StandardSocketOptions.SO_LINGER, i);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   943
                    break;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   944
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   945
                case SO_TIMEOUT: {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   946
                    int i = intValue(value, "SO_TIMEOUT");
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   947
                    if (i < 0)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   948
                        throw new IllegalArgumentException("timeout < 0");
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   949
                    timeout = i;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   950
                    break;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   951
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   952
                case IP_TOS: {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   953
                    int i = intValue(value, "IP_TOS");
57188
1f2101ee432d Change Socket.getInputStream/getOutputStream to return the same stream each time
alanb
parents: 57187
diff changeset
   954
                    Net.setSocketOption(fd, family(), StandardSocketOptions.IP_TOS, i);
57187
056911ad3ee7 Allow setTrafficClass on streams socket
alanb
parents: 57186
diff changeset
   955
                    trafficClass = i;
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   956
                    break;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   957
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   958
                case TCP_NODELAY: {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   959
                    boolean b = booleanValue(value, "TCP_NODELAY");
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   960
                    Net.setSocketOption(fd, StandardSocketOptions.TCP_NODELAY, b);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   961
                    break;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   962
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   963
                case SO_SNDBUF: {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   964
                    int i = intValue(value, "SO_SNDBUF");
57187
056911ad3ee7 Allow setTrafficClass on streams socket
alanb
parents: 57186
diff changeset
   965
                    if (i <= 0)
056911ad3ee7 Allow setTrafficClass on streams socket
alanb
parents: 57186
diff changeset
   966
                        throw new IllegalArgumentException("SO_SNDBUF < 0");
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   967
                    Net.setSocketOption(fd, StandardSocketOptions.SO_SNDBUF, i);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   968
                    break;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   969
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   970
                case SO_RCVBUF: {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   971
                    int i = intValue(value, "SO_RCVBUF");
57187
056911ad3ee7 Allow setTrafficClass on streams socket
alanb
parents: 57186
diff changeset
   972
                    if (i <= 0)
056911ad3ee7 Allow setTrafficClass on streams socket
alanb
parents: 57186
diff changeset
   973
                        throw new IllegalArgumentException("SO_RCVBUF < 0");
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   974
                    Net.setSocketOption(fd, StandardSocketOptions.SO_RCVBUF, i);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   975
                    break;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   976
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   977
                case SO_KEEPALIVE: {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   978
                    boolean b = booleanValue(value, "SO_KEEPALIVE");
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   979
                    Net.setSocketOption(fd, StandardSocketOptions.SO_KEEPALIVE, b);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   980
                    break;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   981
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   982
                case SO_OOBINLINE: {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   983
                    boolean b = booleanValue(value, "SO_OOBINLINE");
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   984
                    Net.setSocketOption(fd, ExtendedSocketOption.SO_OOBINLINE, b);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   985
                    break;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   986
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   987
                case SO_REUSEADDR: {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   988
                    boolean b = booleanValue(value, "SO_REUSEADDR");
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   989
                    if (Net.useExclusiveBind()) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   990
                        isReuseAddress = b;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   991
                    } else {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   992
                        Net.setSocketOption(fd, StandardSocketOptions.SO_REUSEADDR, b);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   993
                    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   994
                    break;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   995
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   996
                case SO_REUSEPORT: {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   997
                    if (!Net.isReusePortAvailable())
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   998
                        throw new UnsupportedOperationException("SO_REUSEPORT not supported");
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
   999
                    boolean b = booleanValue(value, "SO_REUSEPORT");
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1000
                    Net.setSocketOption(fd, StandardSocketOptions.SO_REUSEPORT, b);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1001
                    break;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1002
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1003
                default:
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1004
                    throw new SocketException("Unknown option " + opt);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1005
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1006
            } catch (IOException ioe) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1007
                throw new SocketException(ioe.getMessage());
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1008
            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1009
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1010
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1011
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1012
    @Override
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1013
    public Object getOption(int opt) throws SocketException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1014
        synchronized (stateLock) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1015
            ensureOpen();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1016
            try {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1017
                switch (opt) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1018
                case SO_TIMEOUT:
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1019
                    return timeout;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1020
                case TCP_NODELAY:
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1021
                    return Net.getSocketOption(fd, StandardSocketOptions.TCP_NODELAY);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1022
                case SO_OOBINLINE:
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1023
                    return Net.getSocketOption(fd, ExtendedSocketOption.SO_OOBINLINE);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1024
                case SO_LINGER: {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1025
                    // return "false" when disabled, linger interval when enabled
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1026
                    int i = (int) Net.getSocketOption(fd, StandardSocketOptions.SO_LINGER);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1027
                    if (i == -1) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1028
                        return Boolean.FALSE;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1029
                    } else {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1030
                        return i;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1031
                    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1032
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1033
                case SO_REUSEADDR:
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1034
                    if (Net.useExclusiveBind()) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1035
                        return isReuseAddress;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1036
                    } else {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1037
                        return Net.getSocketOption(fd, StandardSocketOptions.SO_REUSEADDR);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1038
                    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1039
                case SO_BINDADDR:
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1040
                    return Net.localAddress(fd).getAddress();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1041
                case SO_SNDBUF:
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1042
                    return Net.getSocketOption(fd, StandardSocketOptions.SO_SNDBUF);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1043
                case SO_RCVBUF:
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1044
                    return Net.getSocketOption(fd, StandardSocketOptions.SO_RCVBUF);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1045
                case IP_TOS:
57187
056911ad3ee7 Allow setTrafficClass on streams socket
alanb
parents: 57186
diff changeset
  1046
                    return trafficClass;
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1047
                case SO_KEEPALIVE:
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1048
                    return Net.getSocketOption(fd, StandardSocketOptions.SO_KEEPALIVE);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1049
                case SO_REUSEPORT:
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1050
                    if (!Net.isReusePortAvailable())
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1051
                        throw new UnsupportedOperationException("SO_REUSEPORT not supported");
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1052
                    return Net.getSocketOption(fd, StandardSocketOptions.SO_REUSEPORT);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1053
                default:
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1054
                    throw new SocketException("Unknown option " + opt);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1055
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1056
            } catch (IOException ioe) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1057
                throw new SocketException(ioe.getMessage());
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1058
            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1059
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1060
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1061
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1062
    @Override
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1063
    protected <T> void setOption(SocketOption<T> opt, T value) throws IOException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1064
        synchronized (stateLock) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1065
            ensureOpen();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1066
            if (supportedOptions().contains(opt)) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1067
                ExtendedSocketOptions extended = ExtendedSocketOptions.getInstance();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1068
                if (extended.isOptionSupported(opt)) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1069
                    extended.setOption(fd, opt, value);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1070
                } else {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1071
                    super.setOption(opt, value);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1072
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1073
            } else {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1074
                throw new UnsupportedOperationException(opt.name());
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1075
            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1076
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1077
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1078
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1079
    @SuppressWarnings("unchecked")
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1080
    protected <T> T getOption(SocketOption<T> opt) throws IOException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1081
        synchronized (stateLock) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1082
            ensureOpen();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1083
            if (supportedOptions().contains(opt)) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1084
                ExtendedSocketOptions extended = ExtendedSocketOptions.getInstance();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1085
                if (extended.isOptionSupported(opt)) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1086
                    return (T) extended.getOption(fd, opt);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1087
                } else {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1088
                    return super.getOption(opt);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1089
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1090
            } else {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1091
                throw new UnsupportedOperationException(opt.name());
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1092
            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1093
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1094
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1095
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1096
    @Override
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1097
    protected void shutdownInput() throws IOException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1098
        synchronized (stateLock) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1099
            ensureOpenAndConnected();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1100
            if (!isInputClosed) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1101
                Net.shutdown(fd, Net.SHUT_RD);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1102
                long reader = readerThread;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1103
                if (reader != 0)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1104
                    NativeThread.signal(reader);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1105
                isInputClosed = true;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1106
            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1107
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1108
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1109
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1110
    @Override
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1111
    protected void shutdownOutput() throws IOException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1112
        synchronized (stateLock) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1113
            ensureOpenAndConnected();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1114
            if (!isOutputClosed) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1115
                Net.shutdown(fd, Net.SHUT_WR);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1116
                long writer = writerThread;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1117
                if (writer != 0)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1118
                    NativeThread.signal(writer);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1119
                isOutputClosed = true;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1120
            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1121
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1122
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1123
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1124
    @Override
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1125
    protected boolean supportsUrgentData() {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1126
        return true;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1127
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1128
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1129
    @Override
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1130
    protected void sendUrgentData(int data) throws IOException {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1131
        writeLock.lock();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1132
        try {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1133
            int n = 0;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1134
            FileDescriptor fd = beginWrite();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1135
            try {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1136
                do {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1137
                    n = Net.sendOOB(fd, (byte) data);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1138
                } while (n == IOStatus.INTERRUPTED && isOpen());
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1139
                if (n == IOStatus.UNAVAILABLE) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1140
                    throw new RuntimeException("not implemented yet");
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1141
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1142
            } finally {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1143
                endWrite(n > 0);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1144
            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1145
        } finally {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1146
            writeLock.unlock();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1147
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1148
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1149
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1150
    /**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1151
     * A task that closes a SocketImpl's file descriptor. The task runs when the
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1152
     * SocketImpl is explicitly closed and when the SocketImpl becomes phantom
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1153
     * reachable.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1154
     */
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1155
    private static class FileDescriptorCloser implements Runnable {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1156
        private static final VarHandle CLOSED;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1157
        static {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1158
            try {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1159
                MethodHandles.Lookup l = MethodHandles.lookup();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1160
                CLOSED = l.findVarHandle(FileDescriptorCloser.class,
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1161
                                         "closed",
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1162
                                         boolean.class);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1163
            } catch (Exception e) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1164
                throw new InternalError(e);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1165
            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1166
        }
57176
726630bc6a4c Cleanup
michaelm
parents: 57174
diff changeset
  1167
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1168
        private final FileDescriptor fd;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1169
        private final boolean stream;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1170
        private volatile boolean closed;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1171
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1172
        FileDescriptorCloser(FileDescriptor fd, boolean stream) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1173
            this.fd = fd;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1174
            this.stream = stream;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1175
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1176
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1177
        static FileDescriptorCloser create(NioSocketImpl impl) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1178
            assert Thread.holdsLock(impl.stateLock);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1179
            var closer = new FileDescriptorCloser(impl.fd, impl.stream);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1180
            CleanerFactory.cleaner().register(impl, closer);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1181
            return closer;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1182
        }
57176
726630bc6a4c Cleanup
michaelm
parents: 57174
diff changeset
  1183
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1184
        @Override
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1185
        public void run() {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1186
            if (CLOSED.compareAndSet(this, false, true)) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1187
                try {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1188
                    nd.close(fd);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1189
                } catch (IOException ioe) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1190
                    throw new RuntimeException(ioe);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1191
                } finally {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1192
                    if (!stream) {
57176
726630bc6a4c Cleanup
michaelm
parents: 57174
diff changeset
  1193
                        // decrement
57110
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1194
                        ResourceManager.afterUdpClose();
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1195
                    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1196
                }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1197
            }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1198
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1199
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1200
        boolean disable() {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1201
            return CLOSED.compareAndSet(this, false, true);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1202
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1203
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1204
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1205
    /**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1206
     * Returns the socket protocol family
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1207
     */
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1208
    private static ProtocolFamily family() {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1209
        if (Net.isIPv6Available()) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1210
            return StandardProtocolFamily.INET6;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1211
        } else {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1212
            return StandardProtocolFamily.INET;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1213
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1214
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1215
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1216
    /**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1217
     * Returns the native file descriptor
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1218
     */
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1219
    private static int fdVal(FileDescriptor fd) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1220
        int fdVal = SharedSecrets.getJavaIOFileDescriptorAccess().get(fd);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1221
        assert fdVal == IOUtil.fdVal(fd);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1222
        return fdVal;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1223
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1224
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1225
    /**
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1226
     * Sets the SocketImpl fields to the given values.
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1227
     */
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1228
    private static void setSocketImplFields(SocketImpl si,
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1229
                                            FileDescriptor fd,
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1230
                                            int localport,
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1231
                                            InetAddress address,
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1232
                                            int port)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1233
    {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1234
        PrivilegedExceptionAction<Void> pa = () -> {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1235
            setSocketImplField(si, "fd", fd);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1236
            setSocketImplField(si, "localport", localport);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1237
            setSocketImplField(si, "address", address);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1238
            setSocketImplField(si, "port", port);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1239
            return null;
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1240
        };
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1241
        try {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1242
            AccessController.doPrivileged(pa);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1243
        } catch (PrivilegedActionException pae) {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1244
            throw new InternalError(pae);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1245
        }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1246
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1247
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1248
    private static void setSocketImplField(SocketImpl si, String name, Object value)
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1249
        throws Exception
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1250
    {
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1251
        Field field = SocketImpl.class.getDeclaredField(name);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1252
        field.setAccessible(true);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1253
        field.set(si, value);
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1254
    }
b848ca1ef778 Prototype of NIO based SocketImpl
alanb
parents:
diff changeset
  1255
}