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