1/* 2 * Copyright (c) 2000, 2013, 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#include <windows.h> 27#include <winsock2.h> 28#include <ctype.h> 29#include "jni.h" 30#include "jni_util.h" 31#include "jvm.h" 32#include "jlong.h" 33#include "sun_nio_ch_SocketChannelImpl.h" 34 35#include "nio.h" 36#include "nio_util.h" 37#include "net_util.h" 38 39 40static jfieldID ia_addrID; /* java.net.InetAddress.address */ 41 42JNIEXPORT void JNICALL 43Java_sun_nio_ch_SocketChannelImpl_initIDs(JNIEnv *env, jclass cls) 44{ 45 CHECK_NULL(cls = (*env)->FindClass(env, "java/net/InetAddress")); 46 CHECK_NULL(ia_addrID = (*env)->GetFieldID(env, cls, "address", "I")); 47} 48 49jint 50handleSocketError(JNIEnv *env, int errorValue) 51{ 52 NET_ThrowNew(env, errorValue, NULL); 53 return IOS_THROWN; 54} 55 56JNIEXPORT jint JNICALL 57Java_sun_nio_ch_SocketChannelImpl_checkConnect(JNIEnv *env, jobject this, 58 jobject fdo, jboolean block, 59 jboolean ready) 60{ 61 int optError = 0; 62 int lastError = 0; 63 int result = 0; 64 int retry = 0; 65 int n = sizeof(int); 66 jint fd = fdval(env, fdo); 67 fd_set wr, ex; 68 struct timeval t = { 0, 0 }; 69 70 FD_ZERO(&wr); 71 FD_ZERO(&ex); 72 FD_SET((u_int)fd, &wr); 73 FD_SET((u_int)fd, &ex); 74 75 result = select(fd+1, 0, &wr, &ex, block ? NULL : &t); 76 77 /* save last winsock error */ 78 if (result == SOCKET_ERROR) { 79 lastError = WSAGetLastError(); 80 } 81 82 if (block) { /* must configure socket back to blocking state */ 83 u_long argp = 0; 84 int r = ioctlsocket(fd, FIONBIO, &argp); 85 if (r == SOCKET_ERROR) { 86 handleSocketError(env, WSAGetLastError()); 87 } 88 } 89 90 if (result == 0) { /* timeout */ 91 return block ? 0 : IOS_UNAVAILABLE; 92 } else { 93 if (result == SOCKET_ERROR) { /* select failed */ 94 handleSocketError(env, lastError); 95 return IOS_THROWN; 96 } 97 } 98 99 /* 100 * Socket is writable or error occurred. On some Windows editions 101 * the socket will appear writable when the connect fails so we 102 * check for error rather than writable. 103 */ 104 if (!FD_ISSET(fd, &ex)) { 105 return 1; /* connection established */ 106 } 107 108 /* 109 * A getsockopt( SO_ERROR ) may indicate success on NT4 even 110 * though the connection has failed. The workaround is to allow 111 * winsock to be scheduled and this is done via by yielding. 112 * As the yield approach is problematic in heavy load situations 113 * we attempt up to 3 times to get the failure reason. 114 */ 115 for (retry=0; retry<3; retry++) { 116 result = getsockopt((SOCKET)fd, 117 SOL_SOCKET, 118 SO_ERROR, 119 (char *)&optError, 120 &n); 121 if (result == SOCKET_ERROR) { 122 int lastError = WSAGetLastError(); 123 if (lastError == WSAEINPROGRESS) { 124 return IOS_UNAVAILABLE; 125 } 126 NET_ThrowNew(env, lastError, "getsockopt"); 127 return IOS_THROWN; 128 } 129 if (optError) { 130 break; 131 } 132 Sleep(0); 133 } 134 135 if (optError != NO_ERROR) { 136 handleSocketError(env, optError); 137 return IOS_THROWN; 138 } 139 140 return 0; 141} 142 143JNIEXPORT jint JNICALL 144Java_sun_nio_ch_SocketChannelImpl_sendOutOfBandData(JNIEnv* env, jclass this, 145 jobject fdo, jbyte b) 146{ 147 int n = send(fdval(env, fdo), (const char*)&b, 1, MSG_OOB); 148 if (n == SOCKET_ERROR) { 149 handleSocketError(env, WSAGetLastError()); 150 return IOS_THROWN; 151 } else { 152 return n; 153 } 154} 155