--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/jdk/src/windows/native/sun/nio/ch/WindowsAsynchronousSocketChannelImpl.c Sun Feb 15 12:25:54 2009 +0000
@@ -0,0 +1,222 @@
+/*
+ * Copyright 2008-2009 Sun Microsystems, Inc. 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. Sun designates this
+ * particular file as subject to the "Classpath" exception as provided
+ * by Sun 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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
+ * CA 95054 USA or visit www.sun.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 nread = 0;
+ DWORD flags = 0;
+
+ ZeroMemory((PVOID)lpOverlapped, sizeof(OVERLAPPED));
+ res = WSARecv(s,
+ lpWsaBuf,
+ (DWORD)count,
+ &nread,
+ &flags,
+ lpOverlapped,
+ NULL);
+
+ if (res == SOCKET_ERROR) {
+ int error = WSAGetLastError();
+ if (error == WSA_IO_PENDING) {
+ return IOS_UNAVAILABLE;
+ }
+ if (error == WSAESHUTDOWN) {
+ return 0; // input shutdown
+ }
+ JNU_ThrowIOExceptionWithLastError(env, "WSARecv failed");
+ return IOS_THROWN;
+ }
+ if (nread == 0) {
+ // Handle graceful close or bytes not yet available cases
+ // via completion port notification.
+ return IOS_UNAVAILABLE;
+ }
+ return (jint)nread;
+}
+
+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;
+ DWORD nwritten;
+
+ ZeroMemory((PVOID)lpOverlapped, sizeof(OVERLAPPED));
+ res = WSASend(s,
+ lpWsaBuf,
+ (DWORD)count,
+ &nwritten,
+ 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 (jint)nwritten;
+}