| 1 | /* |
| 2 | * Copyright (c) 2017, 2018, 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 | #include <sys/socket.h> |
| 26 | #include <string.h> |
| 27 | #include <errno.h> |
| 28 | #include <unistd.h> |
| 29 | |
| 30 | #include <jni.h> |
| 31 | #include <netinet/tcp.h> |
| 32 | #include <netinet/in.h> |
| 33 | #include "jni_util.h" |
| 34 | #include "jdk_net_LinuxSocketOptions.h" |
| 35 | |
| 36 | /* |
| 37 | * Class: jdk_net_LinuxSocketOptions |
| 38 | * Method: setQuickAck |
| 39 | * Signature: (II)V |
| 40 | */ |
| 41 | JNIEXPORT void JNICALL Java_jdk_net_LinuxSocketOptions_setQuickAck0 |
| 42 | (JNIEnv *env, jobject unused, jint fd, jboolean on) { |
| 43 | int optval; |
| 44 | int rv; |
| 45 | optval = (on ? 1 : 0); |
| 46 | rv = setsockopt(fd, SOL_SOCKET, TCP_QUICKACK, &optval, sizeof (optval)); |
| 47 | if (rv < 0) { |
| 48 | if (errno == ENOPROTOOPT) { |
| 49 | JNU_ThrowByName(env, "java/lang/UnsupportedOperationException" , |
| 50 | "unsupported socket option" ); |
| 51 | } else { |
| 52 | JNU_ThrowByNameWithLastError(env, "java/net/SocketException" , |
| 53 | "set option TCP_QUICKACK failed" ); |
| 54 | } |
| 55 | } |
| 56 | } |
| 57 | |
| 58 | /* |
| 59 | * Class: jdk_net_LinuxSocketOptions |
| 60 | * Method: getQuickAck |
| 61 | * Signature: (I)Z; |
| 62 | */ |
| 63 | JNIEXPORT jboolean JNICALL Java_jdk_net_LinuxSocketOptions_getQuickAck0 |
| 64 | (JNIEnv *env, jobject unused, jint fd) { |
| 65 | int on; |
| 66 | socklen_t sz = sizeof (on); |
| 67 | int rv = getsockopt(fd, SOL_SOCKET, TCP_QUICKACK, &on, &sz); |
| 68 | if (rv < 0) { |
| 69 | if (errno == ENOPROTOOPT) { |
| 70 | JNU_ThrowByName(env, "java/lang/UnsupportedOperationException" , |
| 71 | "unsupported socket option" ); |
| 72 | } else { |
| 73 | JNU_ThrowByNameWithLastError(env, "java/net/SocketException" , |
| 74 | "get option TCP_QUICKACK failed" ); |
| 75 | } |
| 76 | } |
| 77 | return on != 0; |
| 78 | } |
| 79 | |
| 80 | /* |
| 81 | * Class: jdk_net_LinuxSocketOptions |
| 82 | * Method: quickAckSupported |
| 83 | * Signature: ()Z |
| 84 | */ |
| 85 | JNIEXPORT jboolean JNICALL Java_jdk_net_LinuxSocketOptions_quickAckSupported0 |
| 86 | (JNIEnv *env, jobject unused) { |
| 87 | int one = 1; |
| 88 | int rv, s; |
| 89 | s = socket(PF_INET, SOCK_STREAM, 0); |
| 90 | if (s < 0) { |
| 91 | return JNI_FALSE; |
| 92 | } |
| 93 | rv = setsockopt(s, SOL_SOCKET, TCP_QUICKACK, (void *) &one, sizeof (one)); |
| 94 | if (rv != 0 && errno == ENOPROTOOPT) { |
| 95 | rv = JNI_FALSE; |
| 96 | } else { |
| 97 | rv = JNI_TRUE; |
| 98 | } |
| 99 | close(s); |
| 100 | return rv; |
| 101 | } |
| 102 | |
| 103 | static jint socketOptionSupported(jint sockopt) { |
| 104 | jint one = 1; |
| 105 | jint rv, s; |
| 106 | s = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP); |
| 107 | if (s < 0) { |
| 108 | return 0; |
| 109 | } |
| 110 | rv = setsockopt(s, SOL_TCP, sockopt, (void *) &one, sizeof (one)); |
| 111 | if (rv != 0 && errno == ENOPROTOOPT) { |
| 112 | rv = 0; |
| 113 | } else { |
| 114 | rv = 1; |
| 115 | } |
| 116 | close(s); |
| 117 | return rv; |
| 118 | } |
| 119 | |
| 120 | static void handleError(JNIEnv *env, jint rv, const char *errmsg) { |
| 121 | if (rv < 0) { |
| 122 | if (errno == ENOPROTOOPT) { |
| 123 | JNU_ThrowByName(env, "java/lang/UnsupportedOperationException" , |
| 124 | "unsupported socket option" ); |
| 125 | } else { |
| 126 | JNU_ThrowByNameWithLastError(env, "java/net/SocketException" , errmsg); |
| 127 | } |
| 128 | } |
| 129 | } |
| 130 | |
| 131 | /* |
| 132 | * Class: jdk_net_LinuxSocketOptions |
| 133 | * Method: keepAliveOptionsSupported0 |
| 134 | * Signature: ()Z |
| 135 | */ |
| 136 | JNIEXPORT jboolean JNICALL Java_jdk_net_LinuxSocketOptions_keepAliveOptionsSupported0 |
| 137 | (JNIEnv *env, jobject unused) { |
| 138 | return socketOptionSupported(TCP_KEEPIDLE) && socketOptionSupported(TCP_KEEPCNT) |
| 139 | && socketOptionSupported(TCP_KEEPINTVL); |
| 140 | } |
| 141 | |
| 142 | /* |
| 143 | * Class: jdk_net_LinuxSocketOptions |
| 144 | * Method: setTcpkeepAliveProbes0 |
| 145 | * Signature: (II)V |
| 146 | */ |
| 147 | JNIEXPORT void JNICALL Java_jdk_net_LinuxSocketOptions_setTcpkeepAliveProbes0 |
| 148 | (JNIEnv *env, jobject unused, jint fd, jint optval) { |
| 149 | jint rv = setsockopt(fd, SOL_TCP, TCP_KEEPCNT, &optval, sizeof (optval)); |
| 150 | handleError(env, rv, "set option TCP_KEEPCNT failed" ); |
| 151 | } |
| 152 | |
| 153 | /* |
| 154 | * Class: jdk_net_LinuxSocketOptions |
| 155 | * Method: setTcpKeepAliveTime0 |
| 156 | * Signature: (II)V |
| 157 | */ |
| 158 | JNIEXPORT void JNICALL Java_jdk_net_LinuxSocketOptions_setTcpKeepAliveTime0 |
| 159 | (JNIEnv *env, jobject unused, jint fd, jint optval) { |
| 160 | jint rv = setsockopt(fd, SOL_TCP, TCP_KEEPIDLE, &optval, sizeof (optval)); |
| 161 | handleError(env, rv, "set option TCP_KEEPIDLE failed" ); |
| 162 | } |
| 163 | |
| 164 | /* |
| 165 | * Class: jdk_net_LinuxSocketOptions |
| 166 | * Method: setTcpKeepAliveIntvl0 |
| 167 | * Signature: (II)V |
| 168 | */ |
| 169 | JNIEXPORT void JNICALL Java_jdk_net_LinuxSocketOptions_setTcpKeepAliveIntvl0 |
| 170 | (JNIEnv *env, jobject unused, jint fd, jint optval) { |
| 171 | jint rv = setsockopt(fd, SOL_TCP, TCP_KEEPINTVL, &optval, sizeof (optval)); |
| 172 | handleError(env, rv, "set option TCP_KEEPINTVL failed" ); |
| 173 | } |
| 174 | |
| 175 | /* |
| 176 | * Class: jdk_net_LinuxSocketOptions |
| 177 | * Method: getTcpkeepAliveProbes0 |
| 178 | * Signature: (I)I; |
| 179 | */ |
| 180 | JNIEXPORT jint JNICALL Java_jdk_net_LinuxSocketOptions_getTcpkeepAliveProbes0 |
| 181 | (JNIEnv *env, jobject unused, jint fd) { |
| 182 | jint optval, rv; |
| 183 | socklen_t sz = sizeof (optval); |
| 184 | rv = getsockopt(fd, SOL_TCP, TCP_KEEPCNT, &optval, &sz); |
| 185 | handleError(env, rv, "get option TCP_KEEPCNT failed" ); |
| 186 | return optval; |
| 187 | } |
| 188 | |
| 189 | /* |
| 190 | * Class: jdk_net_LinuxSocketOptions |
| 191 | * Method: getTcpKeepAliveTime0 |
| 192 | * Signature: (I)I; |
| 193 | */ |
| 194 | JNIEXPORT jint JNICALL Java_jdk_net_LinuxSocketOptions_getTcpKeepAliveTime0 |
| 195 | (JNIEnv *env, jobject unused, jint fd) { |
| 196 | jint optval, rv; |
| 197 | socklen_t sz = sizeof (optval); |
| 198 | rv = getsockopt(fd, SOL_TCP, TCP_KEEPIDLE, &optval, &sz); |
| 199 | handleError(env, rv, "get option TCP_KEEPIDLE failed" ); |
| 200 | return optval; |
| 201 | } |
| 202 | |
| 203 | /* |
| 204 | * Class: jdk_net_LinuxSocketOptions |
| 205 | * Method: getTcpKeepAliveIntvl0 |
| 206 | * Signature: (I)I; |
| 207 | */ |
| 208 | JNIEXPORT jint JNICALL Java_jdk_net_LinuxSocketOptions_getTcpKeepAliveIntvl0 |
| 209 | (JNIEnv *env, jobject unused, jint fd) { |
| 210 | jint optval, rv; |
| 211 | socklen_t sz = sizeof (optval); |
| 212 | rv = getsockopt(fd, SOL_TCP, TCP_KEEPINTVL, &optval, &sz); |
| 213 | handleError(env, rv, "get option TCP_KEEPINTVL failed" ); |
| 214 | return optval; |
| 215 | } |
| 216 | |