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