/*
* Copyright (c) 2005, 2011, Oracle and/or its affiliates. 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. Oracle designates this
* particular file as subject to the "Classpath" exception as provided
* by Oracle 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.com if you need additional information or have any
* questions.
*/
#include <windows.h>
#include <stdlib.h>
#include <string.h>
#include <Psapi.h>
#include "jni.h"
#include "jni_util.h"
#include "sun_tools_attach_AttachProviderImpl.h"
/*
* Class: sun_tools_attach_AttachProviderImpl
* Method: tempPath
* Signature: ()Ljava/lang/String;
*/
JNIEXPORT jstring JNICALL
Java_sun_tools_attach_AttachProviderImpl_tempPath(JNIEnv *env, jclass cls)
{
char buf[256];
DWORD bufLen, actualLen;
jstring result = NULL;
bufLen = sizeof(buf) / sizeof(char);
actualLen = GetTempPath(bufLen, buf);
if (actualLen > 0) {
char* bufP = buf;
if (actualLen > bufLen) {
actualLen += sizeof(char);
bufP = (char*)malloc(actualLen * sizeof(char));
actualLen = GetTempPath(actualLen, bufP);
}
if (actualLen > 0) {
result = JNU_NewStringPlatform(env, bufP);
}
if (bufP != buf) {
free((void*)bufP);
}
}
return result;
}
/*
* Class: sun_tools_attach_AttachProviderImpl
* Method: volumeFlags
* Signature: ()J
*/
JNIEXPORT jlong JNICALL
Java_sun_tools_attach_AttachProviderImpl_volumeFlags(JNIEnv *env, jclass cls, jstring str)
{
jboolean isCopy;
const char* volume;
DWORD result = 0;
volume = JNU_GetStringPlatformChars(env, str, &isCopy);
if (volume != NULL) {
DWORD componentLen, flags;
BOOL res = GetVolumeInformation(volume,
NULL,
0,
NULL,
&componentLen,
&flags,
NULL,
0);
if (res != 0) {
result = flags;
}
if (isCopy) {
JNU_ReleaseStringPlatformChars(env, str, volume);
}
}
return result;
}
/*
* Class: sun_tools_attach_AttachProviderImpl
* Method: enumProcesses
* Signature: ([JI)I
*/
JNIEXPORT jint JNICALL
Java_sun_tools_attach_AttachProviderImpl_enumProcesses(JNIEnv *env, jclass cls,
jintArray arr, jint max)
{
DWORD size, bytesReturned;
DWORD* ptr;
jint result = 0;
size = max * sizeof(DWORD);
ptr = (DWORD*)malloc(size);
if (ptr != NULL) {
BOOL res = EnumProcesses(ptr, size, &bytesReturned);
if (res != 0) {
result = (jint)(bytesReturned / sizeof(DWORD));
(*env)->SetIntArrayRegion(env, arr, 0, (jsize)result, (jint*)ptr);
}
free((void*)ptr);
}
return result;
}
/*
* Class: sun_tools_attach_AttachProviderImpl
* Method: isLibraryLoadedByProcess
* Signature: (I[Ljava/lang/String;)Z
*/
JNIEXPORT jboolean JNICALL
Java_sun_tools_attach_AttachProviderImpl_isLibraryLoadedByProcess(JNIEnv *env, jclass cls,
jstring str, jint processId)
{
HANDLE hProcess;
jboolean isCopy;
const char* lib;
DWORD size, bytesReturned;
HMODULE* ptr;
jboolean result = JNI_FALSE;
hProcess = OpenProcess(PROCESS_QUERY_INFORMATION |
PROCESS_VM_READ,
FALSE, (DWORD)processId);
if (hProcess == NULL) {
return JNI_FALSE;
}
lib = JNU_GetStringPlatformChars(env, str, &isCopy);
if (lib == NULL) {
CloseHandle(hProcess);
return JNI_FALSE;
}
/*
* Enumerate the modules that the process has opened and see if we have a
* match.
*/
size = 1024 * sizeof(HMODULE);
ptr = (HMODULE*)malloc(size);
if (ptr != NULL) {
BOOL res = EnumProcessModules(hProcess, ptr, size, &bytesReturned);
if (res != 0) {
int count = bytesReturned / sizeof(HMODULE);
/**代码未完, 请加载全部代码(NowJava.com).**/