/*

 * 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;

}