Implemented GetModuleBaseName(AW), GetModuleFileNameEx(AW) and

GetModuleInformation.
This commit is contained in:
Eric Pouech 2003-01-09 00:02:26 +00:00 committed by Alexandre Julliard
parent 2359b57574
commit d93cb70558
1 changed files with 295 additions and 222 deletions

View File

@ -2,6 +2,7 @@
* PSAPI library * PSAPI library
* *
* Copyright 1998 Patrik Stridvall * Copyright 1998 Patrik Stridvall
* Copyright 2003 Eric Pouech
* *
* This library is free software; you can redistribute it and/or * This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public * modify it under the terms of the GNU Lesser General Public
@ -23,32 +24,11 @@
#include "winerror.h" #include "winerror.h"
#include "wine/server.h" #include "wine/server.h"
#include "wine/debug.h" #include "wine/debug.h"
#include "tlhelp32.h" #include "winnls.h"
#include "psapi.h" #include "psapi.h"
WINE_DEFAULT_DEBUG_CHANNEL(psapi); WINE_DEFAULT_DEBUG_CHANNEL(psapi);
#include <string.h>
/***********************************************************************
* get pid from hProcess (internal)
*/
static DWORD get_pid_from_process_handle(HANDLE hProcess)
{
DWORD ret = 0;
SERVER_START_REQ( get_process_info )
{
req->handle = hProcess;
if ( !wine_server_call_err( req ) )
ret = (DWORD)reply->pid;
}
SERVER_END_REQ;
return ret;
}
/*********************************************************************** /***********************************************************************
* EmptyWorkingSet (PSAPI.@) * EmptyWorkingSet (PSAPI.@)
*/ */
@ -60,12 +40,11 @@ BOOL WINAPI EmptyWorkingSet(HANDLE hProcess)
/*********************************************************************** /***********************************************************************
* EnumDeviceDrivers (PSAPI.@) * EnumDeviceDrivers (PSAPI.@)
*/ */
BOOL WINAPI EnumDeviceDrivers( BOOL WINAPI EnumDeviceDrivers(LPVOID *lpImageBase, DWORD cb, LPDWORD lpcbNeeded)
LPVOID *lpImageBase, DWORD cb, LPDWORD lpcbNeeded)
{ {
FIXME("(%p, %ld, %p): stub\n", lpImageBase, cb, lpcbNeeded); FIXME("(%p, %ld, %p): stub\n", lpImageBase, cb, lpcbNeeded);
if(lpcbNeeded) if (lpcbNeeded)
*lpcbNeeded = 0; *lpcbNeeded = 0;
return TRUE; return TRUE;
@ -77,45 +56,59 @@ BOOL WINAPI EnumDeviceDrivers(
*/ */
BOOL WINAPI EnumProcesses(DWORD *lpidProcess, DWORD cb, DWORD *lpcbNeeded) BOOL WINAPI EnumProcesses(DWORD *lpidProcess, DWORD cb, DWORD *lpcbNeeded)
{ {
PROCESSENTRY32 pe;
HANDLE hSnapshot; HANDLE hSnapshot;
BOOL res;
DWORD count; DWORD count;
DWORD countMax; DWORD countMax;
DWORD pid;
int ret;
FIXME("(%p, %ld, %p)\n", lpidProcess,cb, lpcbNeeded); TRACE("(%p, %ld, %p)\n", lpidProcess,cb, lpcbNeeded);
if ( lpidProcess == NULL ) if ( lpidProcess == NULL )
cb = 0; cb = 0;
if ( lpcbNeeded != NULL ) if ( lpcbNeeded != NULL )
*lpcbNeeded = 0; *lpcbNeeded = 0;
hSnapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS,0); SERVER_START_REQ( create_snapshot )
if ( hSnapshot == INVALID_HANDLE_VALUE ) {
req->flags = SNAP_PROCESS;
req->inherit = FALSE;
req->pid = 0;
wine_server_call_err( req );
hSnapshot = reply->handle;
}
SERVER_END_REQ;
if ( hSnapshot == 0 )
{ {
FIXME("cannot create snapshot\n"); FIXME("cannot create snapshot\n");
return FALSE; return FALSE;
} }
count = 0; count = 0;
countMax = cb / sizeof(DWORD); countMax = cb / sizeof(DWORD);
while (1) for (;;)
{ {
ZeroMemory( &pe, sizeof(PROCESSENTRY32) ); SERVER_START_REQ( next_process )
pe.dwSize = sizeof(PROCESSENTRY32); {
res = (count == 0) ? Process32First( hSnapshot, &pe ) : Process32Next( hSnapshot, &pe ); req->handle = hSnapshot;
if ( !res ) req->reset = (count == 0);
break; wine_server_set_reply( req, NULL, 0);
TRACE("process 0x%08lx\n",(long)pe.th32ProcessID); if ((ret = !wine_server_call_err( req )))
pid = reply->pid;
}
SERVER_END_REQ;
if (!ret) break;
TRACE("process 0x%08lx\n", pid);
if ( count < countMax ) if ( count < countMax )
lpidProcess[count] = pe.th32ProcessID; lpidProcess[count] = pid;
count ++; count++;
} }
CloseHandle( hSnapshot ); CloseHandle( hSnapshot );
if ( lpcbNeeded != NULL ) if ( lpcbNeeded != NULL )
*lpcbNeeded = sizeof(DWORD) * count; *lpcbNeeded = sizeof(DWORD) * count;
TRACE("return %lu processes\n",count); TRACE("return %lu processes\n", count);
return TRUE; return TRUE;
} }
@ -123,17 +116,17 @@ BOOL WINAPI EnumProcesses(DWORD *lpidProcess, DWORD cb, DWORD *lpcbNeeded)
/*********************************************************************** /***********************************************************************
* EnumProcessModules (PSAPI.@) * EnumProcessModules (PSAPI.@)
*/ */
BOOL WINAPI EnumProcessModules( BOOL WINAPI EnumProcessModules(HANDLE hProcess, HMODULE *lphModule,
HANDLE hProcess, HMODULE *lphModule, DWORD cb, LPDWORD lpcbNeeded) DWORD cb, LPDWORD lpcbNeeded)
{ {
MODULEENTRY32 me;
HANDLE hSnapshot; HANDLE hSnapshot;
BOOL res;
DWORD pid; DWORD pid;
DWORD count; DWORD count;
DWORD countMax; DWORD countMax;
int ret;
HMODULE hModule;
FIXME("(hProcess=%p, %p, %ld, %p)\n", TRACE("(hProcess=%p, %p, %ld, %p)\n",
hProcess, lphModule, cb, lpcbNeeded ); hProcess, lphModule, cb, lpcbNeeded );
if ( lphModule == NULL ) if ( lphModule == NULL )
@ -141,39 +134,63 @@ BOOL WINAPI EnumProcessModules(
if ( lpcbNeeded != NULL ) if ( lpcbNeeded != NULL )
*lpcbNeeded = 0; *lpcbNeeded = 0;
pid = get_pid_from_process_handle(hProcess); SERVER_START_REQ( get_process_info )
{
req->handle = hProcess;
if ( !wine_server_call_err( req ) )
pid = (DWORD)reply->pid;
else
pid = 0;
}
SERVER_END_REQ;
if ( pid == 0 ) if ( pid == 0 )
{ {
FIXME("no pid for hProcess %p\n",hProcess); FIXME("no pid for hProcess %p\n" ,hProcess);
return FALSE; return FALSE;
} }
hSnapshot = CreateToolhelp32Snapshot(TH32CS_SNAPMODULE,pid); SERVER_START_REQ( create_snapshot )
if ( hSnapshot == INVALID_HANDLE_VALUE ) {
req->flags = SNAP_MODULE;
req->inherit = FALSE;
req->pid = pid;
wine_server_call_err( req );
hSnapshot = reply->handle;
}
SERVER_END_REQ;
if ( hSnapshot == 0 )
{ {
FIXME("cannot create snapshot\n"); FIXME("cannot create snapshot\n");
return FALSE; return FALSE;
} }
count = 0; count = 0;
countMax = cb / sizeof(HMODULE); countMax = cb / sizeof(HMODULE);
while (1) for (;;)
{ {
ZeroMemory( &me, sizeof(MODULEENTRY32) ); SERVER_START_REQ( next_module )
me.dwSize = sizeof(MODULEENTRY32); {
res = (count == 0) ? Module32First( hSnapshot, &me ) : Module32Next( hSnapshot, &me ); req->handle = hSnapshot;
if ( !res ) req->reset = (count == 0);
break; wine_server_set_reply( req, NULL, 0 );
TRACE("module 0x%08lx\n",(long)me.hModule); if ((ret = !wine_server_call_err( req )))
{
hModule = (HMODULE)reply->base;
}
}
SERVER_END_REQ;
if ( !ret ) break;
TRACE("module 0x%p\n", hModule);
if ( count < countMax ) if ( count < countMax )
lphModule[count] = me.hModule; lphModule[count] = hModule;
count ++; count++;
} }
CloseHandle( hSnapshot ); CloseHandle( hSnapshot );
if ( lpcbNeeded != NULL ) if ( lpcbNeeded != NULL )
*lpcbNeeded = sizeof(HMODULE) * count; *lpcbNeeded = sizeof(HMODULE) * count;
TRACE("return %lu modules\n",count); TRACE("return %lu modules\n", count);
return TRUE; return TRUE;
} }
@ -181,14 +198,13 @@ BOOL WINAPI EnumProcessModules(
/*********************************************************************** /***********************************************************************
* GetDeviceDriverBaseNameA (PSAPI.@) * GetDeviceDriverBaseNameA (PSAPI.@)
*/ */
DWORD WINAPI GetDeviceDriverBaseNameA( DWORD WINAPI GetDeviceDriverBaseNameA(LPVOID ImageBase, LPSTR lpBaseName,
LPVOID ImageBase, LPSTR lpBaseName, DWORD nSize) DWORD nSize)
{ {
FIXME("(%p, %s, %ld): stub\n", FIXME("(%p, %s, %ld): stub\n",
ImageBase, debugstr_a(lpBaseName), nSize ImageBase, debugstr_a(lpBaseName), nSize);
);
if(lpBaseName && nSize) if (lpBaseName && nSize)
lpBaseName[0] = '\0'; lpBaseName[0] = '\0';
return 0; return 0;
@ -197,14 +213,13 @@ DWORD WINAPI GetDeviceDriverBaseNameA(
/*********************************************************************** /***********************************************************************
* GetDeviceDriverBaseNameW (PSAPI.@) * GetDeviceDriverBaseNameW (PSAPI.@)
*/ */
DWORD WINAPI GetDeviceDriverBaseNameW( DWORD WINAPI GetDeviceDriverBaseNameW(LPVOID ImageBase, LPWSTR lpBaseName,
LPVOID ImageBase, LPWSTR lpBaseName, DWORD nSize) DWORD nSize)
{ {
FIXME("(%p, %s, %ld): stub\n", FIXME("(%p, %s, %ld): stub\n",
ImageBase, debugstr_w(lpBaseName), nSize ImageBase, debugstr_w(lpBaseName), nSize);
);
if(lpBaseName && nSize) if (lpBaseName && nSize)
lpBaseName[0] = '\0'; lpBaseName[0] = '\0';
return 0; return 0;
@ -213,14 +228,13 @@ DWORD WINAPI GetDeviceDriverBaseNameW(
/*********************************************************************** /***********************************************************************
* GetDeviceDriverFileNameA (PSAPI.@) * GetDeviceDriverFileNameA (PSAPI.@)
*/ */
DWORD WINAPI GetDeviceDriverFileNameA( DWORD WINAPI GetDeviceDriverFileNameA(LPVOID ImageBase, LPSTR lpFilename,
LPVOID ImageBase, LPSTR lpFilename, DWORD nSize) DWORD nSize)
{ {
FIXME("(%p, %s, %ld): stub\n", FIXME("(%p, %s, %ld): stub\n",
ImageBase, debugstr_a(lpFilename), nSize ImageBase, debugstr_a(lpFilename), nSize);
);
if(lpFilename && nSize) if (lpFilename && nSize)
lpFilename[0] = '\0'; lpFilename[0] = '\0';
return 0; return 0;
@ -229,14 +243,13 @@ DWORD WINAPI GetDeviceDriverFileNameA(
/*********************************************************************** /***********************************************************************
* GetDeviceDriverFileNameW (PSAPI.@) * GetDeviceDriverFileNameW (PSAPI.@)
*/ */
DWORD WINAPI GetDeviceDriverFileNameW( DWORD WINAPI GetDeviceDriverFileNameW(LPVOID ImageBase, LPWSTR lpFilename,
LPVOID ImageBase, LPWSTR lpFilename, DWORD nSize) DWORD nSize)
{ {
FIXME("(%p, %s, %ld): stub\n", FIXME("(%p, %s, %ld): stub\n",
ImageBase, debugstr_w(lpFilename), nSize ImageBase, debugstr_w(lpFilename), nSize);
);
if(lpFilename && nSize) if (lpFilename && nSize)
lpFilename[0] = '\0'; lpFilename[0] = '\0';
return 0; return 0;
@ -245,14 +258,13 @@ DWORD WINAPI GetDeviceDriverFileNameW(
/*********************************************************************** /***********************************************************************
* GetMappedFileNameA (PSAPI.@) * GetMappedFileNameA (PSAPI.@)
*/ */
DWORD WINAPI GetMappedFileNameA( DWORD WINAPI GetMappedFileNameA(HANDLE hProcess, LPVOID lpv, LPSTR lpFilename,
HANDLE hProcess, LPVOID lpv, LPSTR lpFilename, DWORD nSize) DWORD nSize)
{ {
FIXME("(hProcess=%p, %p, %s, %ld): stub\n", FIXME("(hProcess=%p, %p, %s, %ld): stub\n",
hProcess, lpv, debugstr_a(lpFilename), nSize hProcess, lpv, debugstr_a(lpFilename), nSize);
);
if(lpFilename && nSize) if (lpFilename && nSize)
lpFilename[0] = '\0'; lpFilename[0] = '\0';
return 0; return 0;
@ -261,14 +273,13 @@ DWORD WINAPI GetMappedFileNameA(
/*********************************************************************** /***********************************************************************
* GetMappedFileNameW (PSAPI.@) * GetMappedFileNameW (PSAPI.@)
*/ */
DWORD WINAPI GetMappedFileNameW( DWORD WINAPI GetMappedFileNameW(HANDLE hProcess, LPVOID lpv, LPWSTR lpFilename,
HANDLE hProcess, LPVOID lpv, LPWSTR lpFilename, DWORD nSize) DWORD nSize)
{ {
FIXME("(hProcess=%p, %p, %s, %ld): stub\n", FIXME("(hProcess=%p, %p, %s, %ld): stub\n",
hProcess, lpv, debugstr_w(lpFilename), nSize hProcess, lpv, debugstr_w(lpFilename), nSize);
);
if(lpFilename && nSize) if (lpFilename && nSize)
lpFilename[0] = '\0'; lpFilename[0] = '\0';
return 0; return 0;
@ -277,83 +288,147 @@ DWORD WINAPI GetMappedFileNameW(
/*********************************************************************** /***********************************************************************
* GetModuleBaseNameA (PSAPI.@) * GetModuleBaseNameA (PSAPI.@)
*/ */
DWORD WINAPI GetModuleBaseNameA( DWORD WINAPI GetModuleBaseNameA(HANDLE hProcess, HMODULE hModule,
HANDLE hProcess, HMODULE hModule, LPSTR lpBaseName, DWORD nSize) LPSTR lpBaseName, DWORD nSize)
{ {
FIXME("(hProcess=%p, hModule=%p, %s, %ld): stub\n", char tmp[MAX_PATH];
hProcess, hModule, debugstr_a(lpBaseName), nSize char* ptr;
);
if(lpBaseName && nSize)
lpBaseName[0] = '\0';
if (!GetModuleFileNameExA(hProcess, hModule, tmp, sizeof(tmp)))
return 0; return 0;
if ((ptr = strrchr(tmp, '\\')) != NULL) ptr++; else ptr = tmp;
strncpy(lpBaseName, ptr, nSize);
lpBaseName[nSize - 1] = '\0';
return strlen(lpBaseName);
} }
/*********************************************************************** /***********************************************************************
* GetModuleBaseNameW (PSAPI.@) * GetModuleBaseNameW (PSAPI.@)
*/ */
DWORD WINAPI GetModuleBaseNameW( DWORD WINAPI GetModuleBaseNameW(HANDLE hProcess, HMODULE hModule,
HANDLE hProcess, HMODULE hModule, LPWSTR lpBaseName, DWORD nSize) LPWSTR lpBaseName, DWORD nSize)
{ {
FIXME("(hProcess=%p, hModule=%p, %s, %ld): stub\n", char* ptr;
hProcess, hModule, debugstr_w(lpBaseName), nSize); DWORD len;
if(lpBaseName && nSize) TRACE("(hProcess=%p, hModule=%p, %p, %ld)\n",
hProcess, hModule, lpBaseName, nSize);
if (!lpBaseName || !nSize) return 0;
ptr = HeapAlloc(GetProcessHeap(), 0, nSize / 2);
if (!ptr) return 0;
len = GetModuleBaseNameA(hProcess, hModule, ptr, nSize / 2);
if (len == 0)
{
lpBaseName[0] = '\0'; lpBaseName[0] = '\0';
}
else
{
if (!MultiByteToWideChar( CP_ACP, 0, ptr, -1, lpBaseName, nSize / 2 ))
lpBaseName[nSize / 2 - 1] = 0;
}
return 0; return len;
} }
/*********************************************************************** /***********************************************************************
* GetModuleFileNameExA (PSAPI.@) * GetModuleFileNameExA (PSAPI.@)
*/ */
DWORD WINAPI GetModuleFileNameExA( DWORD WINAPI GetModuleFileNameExA(HANDLE hProcess, HMODULE hModule,
HANDLE hProcess, HMODULE hModule, LPSTR lpFilename, DWORD nSize) LPSTR lpFileName, DWORD nSize)
{ {
FIXME("(hProcess=%p,hModule=%p, %s, %ld): stub\n", DWORD len = 0;
hProcess, hModule, debugstr_a(lpFilename), nSize
);
if ( get_pid_from_process_handle(hProcess) == GetCurrentProcessId() ) TRACE("(hProcess=%p, hModule=%p, %p, %ld)\n",
return GetModuleFileNameA( hModule, lpFilename, nSize ); hProcess, hModule, lpFileName, nSize);
if(lpFilename&&nSize) if (!lpFileName || !nSize) return 0;
lpFilename[0]='\0';
return 0; if ( hProcess == GetCurrentProcess() )
return GetModuleFileNameA( hModule, lpFileName, nSize );
lpFileName[0] = 0;
SERVER_START_REQ( get_dll_info )
{
req->handle = hProcess;
req->base_address = (void*)hModule;
wine_server_set_reply( req, lpFileName, nSize - 1);
if (!wine_server_call_err( req ))
{
len = wine_server_reply_size(reply);
lpFileName[len] = 0;
}
}
SERVER_END_REQ;
TRACE("return %s (%lu)\n", lpFileName, len);
return len;
} }
/*********************************************************************** /***********************************************************************
* GetModuleFileNameExW (PSAPI.@) * GetModuleFileNameExW (PSAPI.@)
*/ */
DWORD WINAPI GetModuleFileNameExW( DWORD WINAPI GetModuleFileNameExW(HANDLE hProcess, HMODULE hModule,
HANDLE hProcess, HMODULE hModule, LPWSTR lpFilename, DWORD nSize) LPWSTR lpFileName, DWORD nSize)
{ {
FIXME("(hProcess=%p,hModule=%p, %s, %ld): stub\n", char* ptr;
hProcess, hModule, debugstr_w(lpFilename), nSize DWORD len;
);
if ( get_pid_from_process_handle(hProcess) == GetCurrentProcessId() ) TRACE("(hProcess=%p,hModule=%p, %p, %ld)\n",
return GetModuleFileNameW( hModule, lpFilename, nSize ); hProcess, hModule, lpFileName, nSize);
if(lpFilename && nSize) if (!lpFileName || !nSize) return 0;
lpFilename[0] = '\0';
return 0; if ( hProcess == GetCurrentProcess() )
return GetModuleFileNameW( hModule, lpFileName, nSize );
ptr = HeapAlloc(GetProcessHeap(), 0, nSize / 2);
if (!ptr) return 0;
len = GetModuleFileNameExA(hProcess, hModule, ptr, nSize / 2);
if (len == 0)
{
lpFileName[0] = '\0';
}
else
{
if (!MultiByteToWideChar( CP_ACP, 0, ptr, -1, lpFileName, nSize / 2 ))
lpFileName[nSize / 2 - 1] = 0;
}
return len;
} }
/*********************************************************************** /***********************************************************************
* GetModuleInformation (PSAPI.@) * GetModuleInformation (PSAPI.@)
*/ */
BOOL WINAPI GetModuleInformation( BOOL WINAPI GetModuleInformation(HANDLE hProcess, HMODULE hModule,
HANDLE hProcess, HMODULE hModule, LPMODULEINFO lpmodinfo, DWORD cb) LPMODULEINFO lpmodinfo, DWORD cb)
{ {
FIXME("(hProcess=%p, hModule=%p, %p, %ld): stub\n", BOOL ret = FALSE;
hProcess, hModule, lpmodinfo, cb
);
memset(lpmodinfo, 0, cb); TRACE("(hProcess=%p, hModule=%p, %p, %ld)\n",
hProcess, hModule, lpmodinfo, cb);
if (cb < sizeof(MODULEINFO)) return FALSE;
SERVER_START_REQ( get_dll_info )
{
req->handle = hProcess;
req->base_address = (void*)hModule;
if (!wine_server_call_err( req ))
{
ret = TRUE;
lpmodinfo->lpBaseOfDll = (void*)hModule;
lpmodinfo->SizeOfImage = reply->size;
lpmodinfo->EntryPoint = reply->entry_point;
}
}
SERVER_END_REQ;
return TRUE; return TRUE;
} }
@ -361,12 +436,11 @@ BOOL WINAPI GetModuleInformation(
/*********************************************************************** /***********************************************************************
* GetProcessMemoryInfo (PSAPI.@) * GetProcessMemoryInfo (PSAPI.@)
*/ */
BOOL WINAPI GetProcessMemoryInfo( BOOL WINAPI GetProcessMemoryInfo(HANDLE Process,
HANDLE Process, PPROCESS_MEMORY_COUNTERS ppsmemCounters, DWORD cb) PPROCESS_MEMORY_COUNTERS ppsmemCounters, DWORD cb)
{ {
FIXME("(hProcess=%p, %p, %ld): stub\n", FIXME("(hProcess=%p, %p, %ld): stub\n",
Process, ppsmemCounters, cb Process, ppsmemCounters, cb);
);
memset(ppsmemCounters, 0, cb); memset(ppsmemCounters, 0, cb);
@ -376,12 +450,11 @@ BOOL WINAPI GetProcessMemoryInfo(
/*********************************************************************** /***********************************************************************
* GetWsChanges (PSAPI.@) * GetWsChanges (PSAPI.@)
*/ */
BOOL WINAPI GetWsChanges( BOOL WINAPI GetWsChanges(HANDLE hProcess,
HANDLE hProcess, PPSAPI_WS_WATCH_INFORMATION lpWatchInfo, DWORD cb) PPSAPI_WS_WATCH_INFORMATION lpWatchInfo, DWORD cb)
{ {
FIXME("(hProcess=%p, %p, %ld): stub\n", FIXME("(hProcess=%p, %p, %ld): stub\n",
hProcess, lpWatchInfo, cb hProcess, lpWatchInfo, cb);
);
memset(lpWatchInfo, 0, cb); memset(lpWatchInfo, 0, cb);
@ -408,7 +481,7 @@ BOOL WINAPI QueryWorkingSet(HANDLE hProcess, LPVOID pv, DWORD cb)
{ {
FIXME("(hProcess=%p, %p, %ld)\n", hProcess, pv, cb); FIXME("(hProcess=%p, %p, %ld)\n", hProcess, pv, cb);
if(pv && cb) if (pv && cb)
((DWORD *) pv)[0] = 0; /* Empty WorkingSet */ ((DWORD *) pv)[0] = 0; /* Empty WorkingSet */
return TRUE; return TRUE;