| /* |
| * Copyright (c) 2008, 2009, Oracle and/or its affiliates. All rights reserved. |
| * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
| * |
| * This code is free software; you can redistribute it and/or modify it |
| * under the terms of the GNU General Public License version 2 only, as |
| * published by the Free Software Foundation. Oracle designates this |
| * particular file as subject to the "Classpath" exception as provided |
| * by Oracle in the LICENSE file that accompanied this code. |
| * |
| * This code is distributed in the hope that it will be useful, but WITHOUT |
| * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
| * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
| * version 2 for more details (a copy is included in the LICENSE file that |
| * accompanied this code). |
| * |
| * You should have received a copy of the GNU General Public License version |
| * 2 along with this work; if not, write to the Free Software Foundation, |
| * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
| * |
| * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
| * or visit www.oracle.com if you need additional information or have any |
| * questions. |
| */ |
| |
| #include <windows.h> |
| #include <winsock2.h> |
| #include <stddef.h> |
| |
| #include "jni.h" |
| #include "jni_util.h" |
| #include "jlong.h" |
| #include "nio.h" |
| #include "nio_util.h" |
| #include "net_util.h" |
| |
| #include "sun_nio_ch_WindowsAsynchronousSocketChannelImpl.h" |
| |
| #ifndef WSAID_CONNECTEX |
| #define WSAID_CONNECTEX {0x25a207b9,0xddf3,0x4660,{0x8e,0xe9,0x76,0xe5,0x8c,0x74,0x06,0x3e}} |
| #endif |
| |
| #ifndef SO_UPDATE_CONNECT_CONTEXT |
| #define SO_UPDATE_CONNECT_CONTEXT 0x7010 |
| #endif |
| |
| typedef BOOL (*ConnectEx_t) |
| ( |
| SOCKET s, |
| const struct sockaddr* name, |
| int namelen, |
| PVOID lpSendBuffer, |
| DWORD dwSendDataLength, |
| LPDWORD lpdwBytesSent, |
| LPOVERLAPPED lpOverlapped |
| ); |
| |
| static ConnectEx_t ConnectEx_func; |
| |
| |
| JNIEXPORT void JNICALL |
| Java_sun_nio_ch_WindowsAsynchronousSocketChannelImpl_initIDs(JNIEnv* env, jclass this) { |
| GUID GuidConnectEx = WSAID_CONNECTEX; |
| SOCKET s; |
| int rv; |
| DWORD dwBytes; |
| |
| s = socket(AF_INET, SOCK_STREAM, 0); |
| if (s == INVALID_SOCKET) { |
| JNU_ThrowIOExceptionWithLastError(env, "socket failed"); |
| return; |
| } |
| rv = WSAIoctl(s, |
| SIO_GET_EXTENSION_FUNCTION_POINTER, |
| (LPVOID)&GuidConnectEx, |
| sizeof(GuidConnectEx), |
| &ConnectEx_func, |
| sizeof(ConnectEx_func), |
| &dwBytes, |
| NULL, |
| NULL); |
| if (rv != 0) |
| JNU_ThrowIOExceptionWithLastError(env, "WSAIoctl failed"); |
| closesocket(s); |
| } |
| |
| JNIEXPORT jint JNICALL |
| Java_sun_nio_ch_WindowsAsynchronousSocketChannelImpl_connect0(JNIEnv* env, jclass this, |
| jlong socket, jboolean preferIPv6, jobject iao, jint port, jlong ov) |
| { |
| SOCKET s = (SOCKET) jlong_to_ptr(socket); |
| OVERLAPPED* lpOverlapped = (OVERLAPPED*) jlong_to_ptr(ov); |
| |
| SOCKETADDRESS sa; |
| int sa_len; |
| BOOL res; |
| |
| if (NET_InetAddressToSockaddr(env, iao, port, (struct sockaddr *)&sa, &sa_len, preferIPv6) != 0) { |
| return IOS_THROWN; |
| } |
| |
| ZeroMemory((PVOID)lpOverlapped, sizeof(OVERLAPPED)); |
| |
| res = (*ConnectEx_func)(s, |
| (struct sockaddr *)&sa, |
| sa_len, |
| NULL, |
| 0, |
| NULL, |
| lpOverlapped); |
| if (res == 0) { |
| int error = GetLastError(); |
| if (error == ERROR_IO_PENDING) { |
| return IOS_UNAVAILABLE; |
| } |
| JNU_ThrowIOExceptionWithLastError(env, "ConnectEx failed"); |
| return IOS_THROWN; |
| } |
| return 0; |
| } |
| |
| JNIEXPORT void JNICALL |
| Java_sun_nio_ch_WindowsAsynchronousSocketChannelImpl_updateConnectContext(JNIEnv* env, jclass this, |
| jlong socket) |
| { |
| SOCKET s = (SOCKET)jlong_to_ptr(socket); |
| setsockopt(s, SOL_SOCKET, SO_UPDATE_CONNECT_CONTEXT, NULL, 0); |
| } |
| |
| |
| JNIEXPORT void JNICALL |
| Java_sun_nio_ch_WindowsAsynchronousSocketChannelImpl_shutdown0(JNIEnv *env, jclass cl, |
| jlong socket, jint how) |
| { |
| SOCKET s =(SOCKET) jlong_to_ptr(socket); |
| if (shutdown(s, how) == SOCKET_ERROR) { |
| JNU_ThrowIOExceptionWithLastError(env, "shutdown failed"); |
| } |
| } |
| |
| |
| JNIEXPORT void JNICALL |
| Java_sun_nio_ch_WindowsAsynchronousSocketChannelImpl_closesocket0(JNIEnv* env, jclass this, |
| jlong socket) |
| { |
| SOCKET s = (SOCKET)jlong_to_ptr(socket); |
| if (closesocket(s) == SOCKET_ERROR) |
| JNU_ThrowIOExceptionWithLastError(env, "closesocket failed"); |
| } |
| |
| |
| JNIEXPORT jint JNICALL |
| Java_sun_nio_ch_WindowsAsynchronousSocketChannelImpl_read0(JNIEnv* env, jclass this, |
| jlong socket, jint count, jlong address, jlong ov) |
| { |
| SOCKET s = (SOCKET) jlong_to_ptr(socket); |
| WSABUF* lpWsaBuf = (WSABUF*) jlong_to_ptr(address); |
| OVERLAPPED* lpOverlapped = (OVERLAPPED*) jlong_to_ptr(ov); |
| BOOL res; |
| DWORD flags = 0; |
| |
| ZeroMemory((PVOID)lpOverlapped, sizeof(OVERLAPPED)); |
| res = WSARecv(s, |
| lpWsaBuf, |
| (DWORD)count, |
| NULL, |
| &flags, |
| lpOverlapped, |
| NULL); |
| |
| if (res == SOCKET_ERROR) { |
| int error = WSAGetLastError(); |
| if (error == WSA_IO_PENDING) { |
| return IOS_UNAVAILABLE; |
| } |
| if (error == WSAESHUTDOWN) { |
| return IOS_EOF; // input shutdown |
| } |
| JNU_ThrowIOExceptionWithLastError(env, "WSARecv failed"); |
| return IOS_THROWN; |
| } |
| return IOS_UNAVAILABLE; |
| } |
| |
| JNIEXPORT jint JNICALL |
| Java_sun_nio_ch_WindowsAsynchronousSocketChannelImpl_write0(JNIEnv* env, jclass this, |
| jlong socket, jint count, jlong address, jlong ov) |
| { |
| SOCKET s = (SOCKET) jlong_to_ptr(socket); |
| WSABUF* lpWsaBuf = (WSABUF*) jlong_to_ptr(address); |
| OVERLAPPED* lpOverlapped = (OVERLAPPED*) jlong_to_ptr(ov); |
| BOOL res; |
| |
| ZeroMemory((PVOID)lpOverlapped, sizeof(OVERLAPPED)); |
| res = WSASend(s, |
| lpWsaBuf, |
| (DWORD)count, |
| NULL, |
| 0, |
| lpOverlapped, |
| NULL); |
| |
| if (res == SOCKET_ERROR) { |
| int error = WSAGetLastError(); |
| if (error == WSA_IO_PENDING) { |
| return IOS_UNAVAILABLE; |
| } |
| if (error == WSAESHUTDOWN) { |
| return IOS_EOF; // output shutdown |
| } |
| JNU_ThrowIOExceptionWithLastError(env, "WSASend failed"); |
| return IOS_THROWN; |
| } |
| return IOS_UNAVAILABLE; |
| } |