1 /*
2  * Copyright (c) 2005, 2011, Oracle and/or its affiliates. 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.  Oracle designates this
8  * particular file as subject to the "Classpath" exception as provided
9  * by Oracle 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
22  * or visit www.oracle.com if you need additional information or have any
23  * questions.
24  */
25 
26 #include <stdio.h>
27 #include <stdlib.h>
28 #include <string.h>
29 
30 #include <jni_util.h>
31 
32 #include "j2secmod.h"
33 
34 extern void throwNullPointerException(JNIEnv *env, const char *message);
35 extern void throwIOException(JNIEnv *env, const char *message);
36 
findFunction(JNIEnv * env,jlong jHandle,const char * functionName)37 void *findFunction(JNIEnv *env, jlong jHandle, const char *functionName) {
38     HINSTANCE hModule = (HINSTANCE)jHandle;
39     void *fAddress = GetProcAddress(hModule, functionName);
40     if (fAddress == NULL) {
41         char errorMessage[256];
42         _snprintf(errorMessage, sizeof(errorMessage), "Symbol not found: %s", functionName);
43         throwNullPointerException(env, errorMessage);
44         return NULL;
45     }
46     return fAddress;
47 }
48 
Java_sun_security_pkcs11_Secmod_nssGetLibraryHandle(JNIEnv * env,jclass thisClass,jstring jLibName)49 JNIEXPORT jlong JNICALL Java_sun_security_pkcs11_Secmod_nssGetLibraryHandle
50   (JNIEnv *env, jclass thisClass, jstring jLibName)
51 {
52     const char *libName = (*env)->GetStringUTFChars(env, jLibName, NULL);
53     HMODULE hModule = GetModuleHandle(libName);
54     dprintf2("-handle for %s: %d\n", libName, hModule);
55     (*env)->ReleaseStringUTFChars(env, jLibName, libName);
56     return (jlong)hModule;
57 }
58 
Java_sun_security_pkcs11_Secmod_nssLoadLibrary(JNIEnv * env,jclass thisClass,jstring jName)59 JNIEXPORT jlong JNICALL Java_sun_security_pkcs11_Secmod_nssLoadLibrary
60   (JNIEnv *env, jclass thisClass, jstring jName)
61 {
62     HINSTANCE hModule;
63     LPVOID lpMsgBuf;
64 
65     const char *libName = (*env)->GetStringUTFChars(env, jName, NULL);
66     dprintf1("-lib %s\n", libName);
67 
68     hModule = LoadLibrary(libName);
69     (*env)->ReleaseStringUTFChars(env, jName, libName);
70 
71     if (hModule == NULL) {
72         FormatMessage(
73             FORMAT_MESSAGE_ALLOCATE_BUFFER |
74             FORMAT_MESSAGE_FROM_SYSTEM |
75             FORMAT_MESSAGE_IGNORE_INSERTS,
76             NULL,
77             GetLastError(),
78             0, /* Default language */
79             (LPTSTR) &lpMsgBuf,
80             0,
81             NULL
82         );
83         dprintf1("-error: %s\n", lpMsgBuf);
84         throwIOException(env, (char*)lpMsgBuf);
85         LocalFree(lpMsgBuf);
86         return 0;
87     }
88     dprintf2("-handle: %d (0X%X)\n", hModule, hModule);
89     return (jlong)hModule;
90 }
91