1 /*

     2  * Copyright 2008-2009 Sun Microsystems, Inc.  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.  Sun designates this

     8  * particular file as subject to the "Classpath" exception as provided

     9  * by Sun 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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,

    22  * CA 95054 USA or visit www.sun.com if you need additional information or

    23  * have any questions.

    24  */

    25 

    26 #include <windows.h>

    27 

    28 #include "jni.h"

    29 #include "jni_util.h"

    30 #include "jlong.h"

    31 #include "nio.h"

    32 #include "nio_util.h"

    33 

    34 #include "sun_nio_ch_Iocp.h"

    35 

    36 

    37 static jfieldID completionStatus_error;

    38 static jfieldID completionStatus_bytesTransferred;

    39 static jfieldID completionStatus_completionKey;

    40 static jfieldID completionStatus_overlapped;

    41 

    42 

    43 JNIEXPORT void JNICALL

    44 Java_sun_nio_ch_Iocp_initIDs(JNIEnv* env, jclass this)

    45 {

    46     jclass clazz;

    47 

    48     clazz = (*env)->FindClass(env, "sun/nio/ch/Iocp$CompletionStatus");

    49     if (clazz == NULL) {

    50         return;

    51     }

    52     completionStatus_error = (*env)->GetFieldID(env, clazz, "error", "I");

    53     if (completionStatus_error == NULL) return;

    54     completionStatus_bytesTransferred = (*env)->GetFieldID(env, clazz, "bytesTransferred", "I");

    55     if (completionStatus_bytesTransferred == NULL) return;

    56     completionStatus_completionKey = (*env)->GetFieldID(env, clazz, "completionKey", "I");

    57     if (completionStatus_completionKey == NULL) return;

    58     completionStatus_overlapped = (*env)->GetFieldID(env, clazz, "overlapped", "J");

    59 }

    60 

    61 JNIEXPORT jlong JNICALL

    62 Java_sun_nio_ch_Iocp_createIoCompletionPort(JNIEnv* env, jclass this,

    63     jlong handle, jlong existingPort, jint completionKey, jint concurrency)

    64 {

    65     HANDLE port = CreateIoCompletionPort((HANDLE)jlong_to_ptr(handle),

    66                                          (HANDLE)jlong_to_ptr(existingPort),

    67                                          (DWORD)completionKey,

    68                                          (DWORD)concurrency);

    69     if (port == NULL) {

    70         JNU_ThrowIOExceptionWithLastError(env, "CreateIoCompletionPort failed");

    71     }

    72     return ptr_to_jlong(port);

    73 }

    74 

    75 JNIEXPORT void JNICALL

    76 Java_sun_nio_ch_Iocp_close0(JNIEnv* env, jclass this,

    77     jlong handle)

    78 {

    79     HANDLE h = (HANDLE)jlong_to_ptr(handle);

    80     CloseHandle(h);

    81 }

    82 

    83 

    84 JNIEXPORT void JNICALL

    85 Java_sun_nio_ch_Iocp_getQueuedCompletionStatus(JNIEnv* env, jclass this,

    86     jlong completionPort, jobject obj)

    87 {

    88     DWORD bytesTransferred;

    89     DWORD completionKey;

    90     OVERLAPPED *lpOverlapped;

    91     BOOL res;

    92 

    93     res = GetQueuedCompletionStatus((HANDLE)jlong_to_ptr(completionPort),

    94                                   &bytesTransferred,

    95                                   &completionKey,

    96                                   &lpOverlapped,

    97                                   INFINITE);

    98     if (res == 0 && lpOverlapped == NULL) {

    99         JNU_ThrowIOExceptionWithLastError(env, "GetQueuedCompletionStatus failed");

   100     } else {

   101         DWORD ioResult = (res == 0) ? GetLastError() : 0;

   102         (*env)->SetIntField(env, obj, completionStatus_error, ioResult);

   103         (*env)->SetIntField(env, obj, completionStatus_bytesTransferred,

   104             (jint)bytesTransferred);

   105         (*env)->SetIntField(env, obj, completionStatus_completionKey,

   106             (jint)completionKey);

   107         (*env)->SetLongField(env, obj, completionStatus_overlapped,

   108             ptr_to_jlong(lpOverlapped));

   109 

   110     }

   111 }

   112 

   113 JNIEXPORT void JNICALL

   114 Java_sun_nio_ch_Iocp_postQueuedCompletionStatus(JNIEnv* env, jclass this,

   115     jlong completionPort, jint completionKey)

   116 {

   117     BOOL res;

   118 

   119     res = PostQueuedCompletionStatus((HANDLE)jlong_to_ptr(completionPort),

   120                                      (DWORD)0,

   121                                      (DWORD)completionKey,

   122                                      NULL);

   123     if (res == 0) {

   124         JNU_ThrowIOExceptionWithLastError(env, "PostQueuedCompletionStatus");

   125     }

   126 }

   127 

   128 JNIEXPORT jstring JNICALL

   129 Java_sun_nio_ch_Iocp_getErrorMessage(JNIEnv* env, jclass this, jint errorCode)

   130 {

   131     WCHAR message[255];

   132 

   133     DWORD len = FormatMessageW(FORMAT_MESSAGE_FROM_SYSTEM,

   134                                NULL,

   135                                (DWORD)errorCode,

   136                                0,

   137                                &message[0],

   138                                255,

   139                                NULL);

   140 

   141 

   142     if (len == 0) {

   143         return NULL;

   144     } else {

   145         return (*env)->NewString(env, (const jchar *)message, (jsize)wcslen(message));

   146     }

   147 }