1 /*
   2  * Copyright (c) 2014, 2020, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.  Oracle designates this
   8  * particular file as subject to the "Classpath" exception as provided
   9  * by Oracle in the LICENSE file that accompanied this code.
  10  *
  11  * This code is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  14  * version 2 for more details (a copy is included in the LICENSE file that
  15  * accompanied this code).
  16  *
  17  * You should have received a copy of the GNU General Public License version
  18  * 2 along with this work; if not, write to the Free Software Foundation,
  19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  20  *
  21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  22  * or visit www.oracle.com if you need additional information or have any
  23  * questions.
  24  */
  25 
  26 package jdk.net;
  27 
  28 import java.io.FileDescriptor;
  29 import java.net.SocketException;
  30 import java.net.SocketOption;
  31 import java.security.AccessController;
  32 import java.security.PrivilegedAction;
  33 import java.util.Collections;
  34 import java.util.HashSet;
  35 import java.util.Set;
  36 import jdk.internal.access.JavaIOFileDescriptorAccess;
  37 import jdk.internal.access.SharedSecrets;
  38 
  39 /**
  40  * Defines extended socket options, beyond those defined in
  41  * {@link java.net.StandardSocketOptions}. These options may be platform
  42  * specific.
  43  *
  44  * @since 1.8
  45  */
  46 public final class ExtendedSocketOptions {
  47 
  48     private static class ExtSocketOption<T> implements SocketOption<T> {
  49         private final String name;
  50         private final Class<T> type;
  51         ExtSocketOption(String name, Class<T> type) {
  52             this.name = name;
  53             this.type = type;
  54         }
  55         @Override public String name() { return name; }
  56         @Override public Class<T> type() { return type; }
  57         @Override public String toString() { return name; }
  58     }
  59 
  60     private ExtendedSocketOptions() { }
  61 
  62     /**
  63      * Disable Delayed Acknowledgements.
  64      *
  65      * <p>
  66      * This socket option can be used to reduce or disable delayed
  67      * acknowledgments (ACKs). When {@code TCP_QUICKACK} is enabled, ACKs are
  68      * sent immediately, rather than delayed if needed in accordance to normal
  69      * TCP operation. This option is not permanent, it only enables a switch to
  70      * or from {@code TCP_QUICKACK} mode. Subsequent operations of the TCP
  71      * protocol will once again disable/enable {@code TCP_QUICKACK} mode
  72      * depending on internal protocol processing and factors such as delayed ACK
  73      * timeouts occurring and data transfer, therefore this option needs to be
  74      * set with {@code setOption} after each operation of TCP on a given socket.
  75      *
  76      * <p>
  77      * The value of this socket option is a {@code Boolean} that represents
  78      * whether the option is enabled or disabled. The socket option is specific
  79      * to stream-oriented sockets using the TCP/IP protocol. The exact semantics
  80      * of this socket option are socket type and system dependent.
  81      *
  82      * @since 10
  83      */
  84     public static final SocketOption<Boolean> TCP_QUICKACK =
  85             new ExtSocketOption<Boolean>("TCP_QUICKACK", Boolean.class);
  86 
  87     /**
  88      * Keep-Alive idle time.
  89      *
  90      * <p>
  91      * The value of this socket option is an {@code Integer} that is the number
  92      * of seconds of idle time before keep-alive initiates a probe. The socket
  93      * option is specific to stream-oriented sockets using the TCP/IP protocol.
  94      * The exact semantics of this socket option are system dependent.
  95      *
  96      * <p>
  97      * When the {@link java.net.StandardSocketOptions#SO_KEEPALIVE
  98      * SO_KEEPALIVE} option is enabled, TCP probes a connection that has been
  99      * idle for some amount of time. The default value for this idle period is
 100      * system dependent, but is typically 2 hours. The {@code TCP_KEEPIDLE}
 101      * option can be used to affect this value for a given socket.
 102      *
 103      * @since 11
 104      */
 105     public static final SocketOption<Integer> TCP_KEEPIDLE
 106             = new ExtSocketOption<Integer>("TCP_KEEPIDLE", Integer.class);
 107 
 108     /**
 109      * Keep-Alive retransmission interval time.
 110      *
 111      * <p>
 112      * The value of this socket option is an {@code Integer} that is the number
 113      * of seconds to wait before retransmitting a keep-alive probe. The socket
 114      * option is specific to stream-oriented sockets using the TCP/IP protocol.
 115      * The exact semantics of this socket option are system dependent.
 116      *
 117      * <p>
 118      * When the {@link java.net.StandardSocketOptions#SO_KEEPALIVE
 119      * SO_KEEPALIVE} option is enabled, TCP probes a connection that has been
 120      * idle for some amount of time. If the remote system does not respond to a
 121      * keep-alive probe, TCP retransmits the probe after some amount of time.
 122      * The default value for this retransmission interval is system dependent,
 123      * but is typically 75 seconds. The {@code TCP_KEEPINTERVAL} option can be
 124      * used to affect this value for a given socket.
 125      *
 126      * @since 11
 127      */
 128     public static final SocketOption<Integer> TCP_KEEPINTERVAL
 129             = new ExtSocketOption<Integer>("TCP_KEEPINTERVAL", Integer.class);
 130 
 131     /**
 132      * Keep-Alive retransmission maximum limit.
 133      *
 134      * <p>
 135      * The value of this socket option is an {@code Integer} that is the maximum
 136      * number of keep-alive probes to be sent. The socket option is specific to
 137      * stream-oriented sockets using the TCP/IP protocol. The exact semantics of
 138      * this socket option are system dependent.
 139      *
 140      * <p>
 141      * When the {@link java.net.StandardSocketOptions#SO_KEEPALIVE
 142      * SO_KEEPALIVE} option is enabled, TCP probes a connection that has been
 143      * idle for some amount of time. If the remote system does not respond to a
 144      * keep-alive probe, TCP retransmits the probe a certain number of times
 145      * before a connection is considered to be broken. The default value for
 146      * this keep-alive probe retransmit limit is system dependent, but is
 147      * typically 8. The {@code TCP_KEEPCOUNT} option can be used to affect this
 148      * value for a given socket.
 149      *
 150      * @since 11
 151      */
 152     public static final SocketOption<Integer> TCP_KEEPCOUNT
 153             = new ExtSocketOption<Integer>("TCP_KEEPCOUNT", Integer.class);
 154 
 155     /**
 156      * Identifies the receive queue that the last incoming packet for the socket
 157      * was received on.
 158      *
 159      * <p> The value of this socket option is a positive {@code Integer} that
 160      * identifies a receive queue that the application can use to split the
 161      * incoming flows among threads based on the queue identifier. The value is
 162      * {@code 0} when the socket is not bound, a packet has not been received,
 163      * or more generally, when there is no receive queue to identify.
 164      * The socket option is supported by both stream-oriented and datagram-oriented
 165      * sockets.
 166      *
 167      * <p> The socket option is read-only and an attempt to set the socket option
 168      * will throw {@code SocketException}.
 169      *
 170      * @apiNote
 171      * Network devices may have multiple queues or channels to transmit and receive
 172      * network packets. The {@code SO_INCOMING_NAPI_ID} socket option provides a hint
 173      * to the application to indicate the receive queue on which an incoming socket
 174      * connection or packets for that connection are directed to. An application may
 175      * take advantage of this by handling all socket connections assigned to a
 176      * specific queue on one thread.
 177      *
 178      * @since 15
 179      */
 180     public static final SocketOption<Integer> SO_INCOMING_NAPI_ID
 181             = new ExtSocketOption<Integer>("SO_INCOMING_NAPI_ID", Integer.class);
 182 
 183     /**
 184      * Unix Domain peer credentials.
 185      *
 186      * <p> The value of this socket option is a {@link UnixDomainPrincipal} that
 187      * represents the credentials of a peer connected to a Unix Domain socket.
 188      * The credentials are those that applied at the time the socket was first
 189      * connected or accepted.
 190      *
 191      * <p> The socket option is read-only and an attempt to set the socket option
 192      * will throw {@code SocketException}. {@code SocketException} is also thrown
 193      * when attempting to get the value of the socket option on an unconnected Unix
 194      * Domain socket.
 195      *
 196      * @since 16
 197      */
 198     public static final SocketOption<UnixDomainPrincipal> SO_PEERCRED
 199         = new ExtSocketOption<UnixDomainPrincipal>
 200             ("SO_PEERCRED", UnixDomainPrincipal.class);
 201 
 202     private static final PlatformSocketOptions platformSocketOptions =
 203             PlatformSocketOptions.get();
 204 
 205     private static final boolean quickAckSupported =
 206             platformSocketOptions.quickAckSupported();
 207     private static final boolean keepAliveOptSupported =
 208             platformSocketOptions.keepAliveOptionsSupported();
 209     private static final boolean peerCredentialsSupported =
 210             platformSocketOptions.peerCredentialsSupported();
 211     private static final boolean incomingNapiIdOptSupported  =
 212             platformSocketOptions.incomingNapiIdSupported();
 213     private static final Set<SocketOption<?>> extendedOptions = options();
 214 
 215     static Set<SocketOption<?>> options() {
 216         Set<SocketOption<?>> options = new HashSet<>();
 217         if (quickAckSupported) {
 218             options.add(TCP_QUICKACK);
 219         }
 220         if (incomingNapiIdOptSupported) {
 221             options.add(SO_INCOMING_NAPI_ID);
 222         }
 223         if (keepAliveOptSupported) {
 224             options.addAll(Set.of(TCP_KEEPCOUNT, TCP_KEEPIDLE, TCP_KEEPINTERVAL));
 225         }
 226         if (peerCredentialsSupported) {
 227             options.add(SO_PEERCRED);
 228         }
 229         return Collections.unmodifiableSet(options);
 230     }
 231 
 232     static {
 233         // Registers the extended socket options with the base module.
 234         sun.net.ext.ExtendedSocketOptions.register(
 235                 new sun.net.ext.ExtendedSocketOptions(extendedOptions) {
 236 
 237             @Override
 238             @SuppressWarnings("removal")
 239             public void setOption(FileDescriptor fd,
 240                                   SocketOption<?> option,
 241                                   Object value)
 242                 throws SocketException
 243             {
 244                 if (fd == null || !fd.valid())
 245                     throw new SocketException("socket closed");
 246 
 247                 if (option == TCP_QUICKACK) {
 248                     setQuickAckOption(fd, (boolean) value);
 249                 } else if (option == TCP_KEEPCOUNT) {
 250                     setTcpkeepAliveProbes(fd, (Integer) value);
 251                 } else if (option == TCP_KEEPIDLE) {
 252                     setTcpKeepAliveTime(fd, (Integer) value);
 253                 } else if (option == TCP_KEEPINTERVAL) {
 254                     setTcpKeepAliveIntvl(fd, (Integer) value);
 255                 } else if (option == SO_INCOMING_NAPI_ID) {
 256                     if (!incomingNapiIdOptSupported)
 257                         throw new UnsupportedOperationException("Attempt to set unsupported option " + option);
 258                     else
 259                         throw new SocketException("Attempt to set read only option " + option);
 260                 } else if (option == SO_PEERCRED) {
 261                     throw new SocketException("SO_PEERCRED cannot be set ");
 262                 } else {
 263                     throw new InternalError("Unexpected option " + option);
 264                 }
 265             }
 266 
 267             @Override
 268             @SuppressWarnings("removal")
 269             public Object getOption(FileDescriptor fd,
 270                                     SocketOption<?> option)
 271                 throws SocketException
 272             {
 273                 if (fd == null || !fd.valid())
 274                     throw new SocketException("socket closed");
 275 
 276                 if (option == TCP_QUICKACK) {
 277                     return getQuickAckOption(fd);
 278                 } else if (option == TCP_KEEPCOUNT) {
 279                     return getTcpkeepAliveProbes(fd);
 280                 } else if (option == TCP_KEEPIDLE) {
 281                     return getTcpKeepAliveTime(fd);
 282                 } else if (option == TCP_KEEPINTERVAL) {
 283                     return getTcpKeepAliveIntvl(fd);
 284                 } else if (option == SO_PEERCRED) {
 285                     return getSoPeerCred(fd);
 286                 } else if (option == SO_INCOMING_NAPI_ID) {
 287                     return getIncomingNapiId(fd);
 288                 } else {
 289                     throw new InternalError("Unexpected option " + option);
 290                 }
 291             }
 292         });
 293     }
 294 
 295     private static final JavaIOFileDescriptorAccess fdAccess =
 296             SharedSecrets.getJavaIOFileDescriptorAccess();
 297 
 298     private static void setQuickAckOption(FileDescriptor fd, boolean enable)
 299             throws SocketException {
 300         platformSocketOptions.setQuickAck(fdAccess.get(fd), enable);
 301     }
 302 
 303     private static Object getSoPeerCred(FileDescriptor fd)
 304             throws SocketException {
 305         return platformSocketOptions.getSoPeerCred(fdAccess.get(fd));
 306     }
 307 
 308     private static Object getQuickAckOption(FileDescriptor fd)
 309             throws SocketException {
 310         return platformSocketOptions.getQuickAck(fdAccess.get(fd));
 311     }
 312 
 313     private static void setTcpkeepAliveProbes(FileDescriptor fd, int value)
 314             throws SocketException {
 315         platformSocketOptions.setTcpkeepAliveProbes(fdAccess.get(fd), value);
 316     }
 317 
 318     private static void setTcpKeepAliveTime(FileDescriptor fd, int value)
 319             throws SocketException {
 320         platformSocketOptions.setTcpKeepAliveTime(fdAccess.get(fd), value);
 321     }
 322 
 323     private static void setTcpKeepAliveIntvl(FileDescriptor fd, int value)
 324             throws SocketException {
 325         platformSocketOptions.setTcpKeepAliveIntvl(fdAccess.get(fd), value);
 326     }
 327 
 328     private static int getTcpkeepAliveProbes(FileDescriptor fd) throws SocketException {
 329         return platformSocketOptions.getTcpkeepAliveProbes(fdAccess.get(fd));
 330     }
 331 
 332     private static int getTcpKeepAliveTime(FileDescriptor fd) throws SocketException {
 333         return platformSocketOptions.getTcpKeepAliveTime(fdAccess.get(fd));
 334     }
 335 
 336     private static int getTcpKeepAliveIntvl(FileDescriptor fd) throws SocketException {
 337         return platformSocketOptions.getTcpKeepAliveIntvl(fdAccess.get(fd));
 338     }
 339 
 340     private static int getIncomingNapiId(FileDescriptor fd) throws SocketException {
 341         return platformSocketOptions.getIncomingNapiId(fdAccess.get(fd));
 342     }
 343 
 344     static class PlatformSocketOptions {
 345 
 346         protected PlatformSocketOptions() {}
 347 
 348         @SuppressWarnings("unchecked")
 349         private static PlatformSocketOptions newInstance(String cn) {
 350             Class<PlatformSocketOptions> c;
 351             try {
 352                 c = (Class<PlatformSocketOptions>)Class.forName(cn);
 353                 return c.getConstructor(new Class<?>[] { }).newInstance();
 354             } catch (ReflectiveOperationException x) {
 355                 throw new AssertionError(x);
 356             }
 357         }
 358 
 359         private static PlatformSocketOptions create() {
 360             String osname = AccessController.doPrivileged(
 361                     new PrivilegedAction<String>() {
 362                         public String run() {
 363                             return System.getProperty("os.name");
 364                         }
 365                     });
 366             if ("Linux".equals(osname)) {
 367                 return newInstance("jdk.net.LinuxSocketOptions");
 368             } else if (osname.startsWith("Mac")) {
 369                 return newInstance("jdk.net.MacOSXSocketOptions");
 370             } else {
 371                 return new PlatformSocketOptions();
 372             }
 373         }
 374 
 375         private static final PlatformSocketOptions instance = create();
 376 
 377         static PlatformSocketOptions get() {
 378             return instance;
 379         }
 380 
 381         boolean peerCredentialsSupported() {
 382             return false;
 383         }
 384 
 385         void setQuickAck(int fd, boolean on) throws SocketException {
 386             throw new UnsupportedOperationException("unsupported TCP_QUICKACK option");
 387         }
 388 
 389         boolean getQuickAck(int fd) throws SocketException {
 390             throw new UnsupportedOperationException("unsupported TCP_QUICKACK option");
 391         }
 392 
 393         boolean quickAckSupported() {
 394             return false;
 395         }
 396 
 397         boolean keepAliveOptionsSupported() {
 398             return false;
 399         }
 400 
 401         void setTcpkeepAliveProbes(int fd, final int value) throws SocketException {
 402             throw new UnsupportedOperationException("unsupported TCP_KEEPCNT option");
 403         }
 404 
 405         void setTcpKeepAliveTime(int fd, final int value) throws SocketException {
 406             throw new UnsupportedOperationException("unsupported TCP_KEEPIDLE option");
 407         }
 408 
 409         UnixDomainPrincipal getSoPeerCred(int fd) throws SocketException {
 410             throw new UnsupportedOperationException("unsupported SO_PEERCRED option");
 411         }
 412 
 413         void setTcpKeepAliveIntvl(int fd, final int value) throws SocketException {
 414             throw new UnsupportedOperationException("unsupported TCP_KEEPINTVL option");
 415         }
 416 
 417         int getTcpkeepAliveProbes(int fd) throws SocketException {
 418             throw new UnsupportedOperationException("unsupported TCP_KEEPCNT option");
 419         }
 420 
 421         int getTcpKeepAliveTime(int fd) throws SocketException {
 422             throw new UnsupportedOperationException("unsupported TCP_KEEPIDLE option");
 423         }
 424 
 425         int getTcpKeepAliveIntvl(int fd) throws SocketException {
 426             throw new UnsupportedOperationException("unsupported TCP_KEEPINTVL option");
 427         }
 428 
 429         boolean incomingNapiIdSupported() {
 430             return false;
 431         }
 432 
 433         int getIncomingNapiId(int fd) throws SocketException {
 434             throw new UnsupportedOperationException("unsupported SO_INCOMING_NAPI_ID socket option");
 435         }
 436     }
 437 }