dlls/psapi/psapi_main.c

   1 /*
   2  *      PSAPI library
   3  *
   4  * Copyright 1998 Patrik Stridvall
   5  * Copyright 2003 Eric Pouech
   6  *
   7  * This library is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * This library is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with this library; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  20  */
  21
  22 #include <stdarg.h>
  23
  24 #include "ntstatus.h"
  25 #define WIN32_NO_STATUS
  26 #include "windef.h"
  27 #include "winbase.h"
  28 #include "winerror.h"
  29 #include "wine/unicode.h"
  30 #include "wine/debug.h"
  31 #include "winnls.h"
  32 #include "winternl.h"
  33 #include "psapi.h"
  34
  35 WINE_DEFAULT_DEBUG_CHANNEL(psapi);
  36
  37 typedef struct
  38 {
  39     HANDLE hProcess;
  40     PLIST_ENTRY pHead, pCurrent;
  41     LDR_MODULE LdrModule;
  42 } MODULE_ITERATOR;
  43
  44 /***********************************************************************
  45  *           PSAPI_ModuleIteratorInit [internal]
  46  *
  47  * Prepares to iterate through the loaded modules of the given process.
  48  *
  49  * RETURNS
  50  *  Success: TRUE
  51  *  Failure: FALSE
  52  */
  53 static BOOL PSAPI_ModuleIteratorInit(MODULE_ITERATOR *iter, HANDLE hProcess)
  54 {
  55     PROCESS_BASIC_INFORMATION pbi;
  56     PPEB_LDR_DATA pLdrData;
  57     NTSTATUS status;
  58
  59     /* Get address of PEB */
  60     status = NtQueryInformationProcess(hProcess, ProcessBasicInformation,
  61                                        &pbi, sizeof(pbi), NULL);
  62     if (status != STATUS_SUCCESS)
  63     {
  64         SetLastError(RtlNtStatusToDosError(status));
  65         return FALSE;
  66     }
  67
  68     /* Read address of LdrData from PEB */
  69     if (!ReadProcessMemory(hProcess, &pbi.PebBaseAddress->LdrData,
  70                            &pLdrData, sizeof(pLdrData), NULL))
  71         return FALSE;
  72
  73     /* Read address of first module from LdrData */
  74     if (!ReadProcessMemory(hProcess,
  75                            &pLdrData->InLoadOrderModuleList.Flink,
  76                            &iter->pCurrent, sizeof(iter->pCurrent), NULL))
  77         return FALSE;
  78
  79     iter->pHead = &pLdrData->InLoadOrderModuleList;
  80     iter->hProcess = hProcess;
  81
  82     return TRUE;
  83 }
  84
  85 /***********************************************************************
  86  *           PSAPI_ModuleIteratorNext [internal]
  87  *
  88  * Iterates to the next module.
  89  *
  90  * RETURNS
  91  *   1 : Success
  92  *   0 : No more modules
  93  *  -1 : Failure
  94  *
  95  * NOTES
  96  *  Every function which uses this routine suffers from a race condition
  97  *  when a module is unloaded during the enumeration which can cause the
  98  *  function to fail. As there is no way to lock the loader of another
  99  *  process we can't avoid that.
 100  */
 101 static INT PSAPI_ModuleIteratorNext(MODULE_ITERATOR *iter)
 102 {
 103     if (iter->pCurrent == iter->pHead)
 104         return 0;
 105
 106     if (!ReadProcessMemory(iter->hProcess, CONTAINING_RECORD(iter->pCurrent,
 107                            LDR_MODULE, InLoadOrderModuleList),
 108                            &iter->LdrModule, sizeof(iter->LdrModule), NULL))
 109          return -1;
 110     else
 111          iter->pCurrent = iter->LdrModule.InLoadOrderModuleList.Flink;
 112
 113     return 1;
 114 }
 115
 116 /***********************************************************************
 117  *           PSAPI_GetLdrModule [internal]
 118  *
 119  * Reads the LDR_MODULE structure of the given module.
 120  *
 121  * RETURNS
 122  *  Success: TRUE
 123  *  Failure: FALSE
 124  */
 125
 126 static BOOL PSAPI_GetLdrModule(HANDLE hProcess, HMODULE hModule,
 127                                LDR_MODULE *pLdrModule)
 128 {
 129     MODULE_ITERATOR iter;
 130     INT ret;
 131
 132     if (!PSAPI_ModuleIteratorInit(&iter, hProcess))
 133         return FALSE;
 134
 135     while ((ret = PSAPI_ModuleIteratorNext(&iter)) > 0)
 136         /* When hModule is NULL we return the process image - which will be
 137          * the first module since our iterator uses InLoadOrderModuleList */
 138         if (!hModule || hModule == iter.LdrModule.BaseAddress)
 139         {
 140             *pLdrModule = iter.LdrModule;
 141             return TRUE;
 142         }
 143
 144     if (ret == 0)
 145         SetLastError(ERROR_INVALID_HANDLE);
 146
 147     return FALSE;
 148 }
 149
 150 /***********************************************************************
 151  *           EmptyWorkingSet (PSAPI.@)
 152  */
 153 BOOL WINAPI EmptyWorkingSet(HANDLE hProcess)
 154 {
 155     return SetProcessWorkingSetSize(hProcess, 0xFFFFFFFF, 0xFFFFFFFF);
 156 }
 157
 158 /***********************************************************************
 159  *           EnumDeviceDrivers (PSAPI.@)
 160  */
 161 BOOL WINAPI EnumDeviceDrivers(LPVOID *lpImageBase, DWORD cb, LPDWORD lpcbNeeded)
 162 {
 163     FIXME("(%p, %d, %p): stub\n", lpImageBase, cb, lpcbNeeded);
 164
 165     if (lpcbNeeded)
 166         *lpcbNeeded = 0;
 167
 168     return TRUE;
 169 }
 170
 171 /***********************************************************************
 172  *           EnumPageFilesA (PSAPI.@)
 173  */
 174 BOOL WINAPI EnumPageFilesA( PENUM_PAGE_FILE_CALLBACKA callback, LPVOID context )
 175 {
 176     FIXME("(%p, %p) stub\n", callback, context );
 177     return FALSE;
 178 }
 179
 180 /***********************************************************************
 181  *           EnumPageFilesW (PSAPI.@)
 182  */
 183 BOOL WINAPI EnumPageFilesW( PENUM_PAGE_FILE_CALLBACKW callback, LPVOID context )
 184 {
 185     FIXME("(%p, %p) stub\n", callback, context );
 186     return FALSE;
 187 }
 188
 189 /***********************************************************************
 190  *           EnumProcesses (PSAPI.@)
 191  */
 192 BOOL WINAPI EnumProcesses(DWORD *lpdwProcessIDs, DWORD cb, DWORD *lpcbUsed)
 193 {
 194     SYSTEM_PROCESS_INFORMATION *spi;
 195     NTSTATUS status;
 196     PVOID pBuf = NULL;
 197     ULONG nAlloc = 0x8000;
 198
 199     do {
 200         if (pBuf != NULL)
 201         {
 202             HeapFree(GetProcessHeap(), 0, pBuf);
 203             nAlloc *= 2;
 204         }
 205
 206         pBuf = HeapAlloc(GetProcessHeap(), 0, nAlloc);
 207         if (pBuf == NULL)
 208             return FALSE;
 209
 210         status = NtQuerySystemInformation(SystemProcessInformation, pBuf,
 211                                           nAlloc, NULL);
 212     } while (status == STATUS_INFO_LENGTH_MISMATCH);
 213
 214     if (status != STATUS_SUCCESS)
 215     {
 216         HeapFree(GetProcessHeap(), 0, pBuf);
 217         SetLastError(RtlNtStatusToDosError(status));
 218         return FALSE;
 219     }
 220
 221     spi = pBuf;
 222
 223     for (*lpcbUsed = 0; cb >= sizeof(DWORD); cb -= sizeof(DWORD))
 224     {
 225         *lpdwProcessIDs++ = HandleToUlong(spi->UniqueProcessId);
 226         *lpcbUsed += sizeof(DWORD);
 227
 228         if (spi->NextEntryOffset == 0)
 229             break;
 230
 231         spi = (SYSTEM_PROCESS_INFORMATION *)(((PCHAR)spi) + spi->NextEntryOffset);
 232     }
 233
 234     HeapFree(GetProcessHeap(), 0, pBuf);
 235     return TRUE;
 236 }
 237
 238 /***********************************************************************
 239  *           EnumProcessModules (PSAPI.@)
 240  *
 241  * NOTES
 242  *  Returned list is in load order.
 243  */
 244 BOOL WINAPI EnumProcessModules(HANDLE hProcess, HMODULE *lphModule,
 245                                DWORD cb, LPDWORD lpcbNeeded)
 246 {
 247     MODULE_ITERATOR iter;
 248     INT ret;
 249
 250     if (!PSAPI_ModuleIteratorInit(&iter, hProcess))
 251         return FALSE;
 252
 253     *lpcbNeeded = 0;
 254
 255     while ((ret = PSAPI_ModuleIteratorNext(&iter)) > 0)
 256     {
 257         if (cb >= sizeof(HMODULE))
 258         {
 259             *lphModule++ = iter.LdrModule.BaseAddress;
 260             cb -= sizeof(HMODULE);
 261         }
 262         *lpcbNeeded += sizeof(HMODULE);
 263     }
 264
 265     return (ret == 0);
 266 }
 267
 268 /***********************************************************************
 269  *          GetDeviceDriverBaseNameA (PSAPI.@)
 270  */
 271 DWORD WINAPI GetDeviceDriverBaseNameA(LPVOID ImageBase, LPSTR lpBaseName,
 272                                       DWORD nSize)
 273 {
 274     FIXME("(%p, %p, %d): stub\n", ImageBase, lpBaseName, nSize);
 275
 276     if (lpBaseName && nSize)
 277         lpBaseName[0] = '\0';
 278
 279     return 0;
 280 }
 281
 282 /***********************************************************************
 283  *           GetDeviceDriverBaseNameW (PSAPI.@)
 284  */
 285 DWORD WINAPI GetDeviceDriverBaseNameW(LPVOID ImageBase, LPWSTR lpBaseName,
 286                                       DWORD nSize)
 287 {
 288     FIXME("(%p, %p, %d): stub\n", ImageBase, lpBaseName, nSize);
 289
 290     if (lpBaseName && nSize)
 291         lpBaseName[0] = '\0';
 292
 293     return 0;
 294 }
 295
 296 /***********************************************************************
 297  *           GetDeviceDriverFileNameA (PSAPI.@)
 298  */
 299 DWORD WINAPI GetDeviceDriverFileNameA(LPVOID ImageBase, LPSTR lpFilename,
 300                                       DWORD nSize)
 301 {
 302     FIXME("(%p, %p, %d): stub\n", ImageBase, lpFilename, nSize);
 303
 304     if (lpFilename && nSize)
 305         lpFilename[0] = '\0';
 306
 307     return 0;
 308 }
 309
 310 /***********************************************************************
 311  *           GetDeviceDriverFileNameW (PSAPI.@)
 312  */
 313 DWORD WINAPI GetDeviceDriverFileNameW(LPVOID ImageBase, LPWSTR lpFilename,
 314                                       DWORD nSize)
 315 {
 316     FIXME("(%p, %p, %d): stub\n", ImageBase, lpFilename, nSize);
 317
 318     if (lpFilename && nSize)
 319         lpFilename[0] = '\0';
 320
 321     return 0;
 322 }
 323
 324 /***********************************************************************
 325  *           GetMappedFileNameA (PSAPI.@)
 326  */
 327 DWORD WINAPI GetMappedFileNameA(HANDLE hProcess, LPVOID lpv, LPSTR lpFilename,
 328                                 DWORD nSize)
 329 {
 330     FIXME("(%p, %p, %p, %d): stub\n", hProcess, lpv, lpFilename, nSize);
 331
 332     if (lpFilename && nSize)
 333         lpFilename[0] = '\0';
 334
 335     return 0;
 336 }
 337
 338 /***********************************************************************
 339  *           GetMappedFileNameW (PSAPI.@)
 340  */
 341 DWORD WINAPI GetMappedFileNameW(HANDLE hProcess, LPVOID lpv, LPWSTR lpFilename,
 342                                 DWORD nSize)
 343 {
 344     FIXME("(%p, %p, %p, %d): stub\n", hProcess, lpv, lpFilename, nSize);
 345
 346     if (lpFilename && nSize)
 347         lpFilename[0] = '\0';
 348
 349     return 0;
 350 }
 351
 352 /***********************************************************************
 353  *           GetModuleBaseNameA (PSAPI.@)
 354  */
 355 DWORD WINAPI GetModuleBaseNameA(HANDLE hProcess, HMODULE hModule,
 356                                 LPSTR lpBaseName, DWORD nSize)
 357 {
 358     WCHAR *lpBaseNameW;
 359     DWORD buflenW, ret = 0;
 360
 361     if(!lpBaseName || !nSize) {
 362         SetLastError(ERROR_INVALID_PARAMETER);
 363         return 0;
 364     }
 365     lpBaseNameW = HeapAlloc(GetProcessHeap(), 0, sizeof(WCHAR) * nSize);
 366     buflenW = GetModuleBaseNameW(hProcess, hModule, lpBaseNameW, nSize);
 367     TRACE("%d, %s\n", buflenW, debugstr_w(lpBaseNameW));
 368     if (buflenW)
 369     {
 370         ret = WideCharToMultiByte(CP_ACP, 0, lpBaseNameW, buflenW,
 371                                   lpBaseName, nSize, NULL, NULL);
 372         if (ret < nSize) lpBaseName[ret] = 0;
 373     }
 374     HeapFree(GetProcessHeap(), 0, lpBaseNameW);
 375     return ret;
 376 }
 377
 378 /***********************************************************************
 379  *           GetModuleBaseNameW (PSAPI.@)
 380  */
 381 DWORD WINAPI GetModuleBaseNameW(HANDLE hProcess, HMODULE hModule,
 382                                 LPWSTR lpBaseName, DWORD nSize)
 383 {
 384     LDR_MODULE LdrModule;
 385
 386     if (!PSAPI_GetLdrModule(hProcess, hModule, &LdrModule))
 387         return 0;
 388
 389     nSize = min(LdrModule.BaseDllName.Length / sizeof(WCHAR), nSize);
 390     if (!ReadProcessMemory(hProcess, LdrModule.BaseDllName.Buffer,
 391                            lpBaseName, nSize * sizeof(WCHAR), NULL))
 392         return 0;
 393
 394     lpBaseName[nSize] = 0;
 395     return nSize;
 396 }
 397
 398 /***********************************************************************
 399  *           GetModuleFileNameExA (PSAPI.@)
 400  */
 401 DWORD WINAPI GetModuleFileNameExA(HANDLE hProcess, HMODULE hModule,
 402                                   LPSTR lpFileName, DWORD nSize)
 403 {
 404     WCHAR *ptr;
 405
 406     TRACE("(hProcess=%p, hModule=%p, %p, %d)\n",
 407           hProcess, hModule, lpFileName, nSize);
 408
 409     if (!lpFileName || !nSize) return 0;
 410
 411     if ( hProcess == GetCurrentProcess() )
 412     {
 413         DWORD len = GetModuleFileNameA( hModule, lpFileName, nSize );
 414         if (nSize) lpFileName[nSize - 1] = '\0';
 415         return len;
 416     }
 417
 418     if (!(ptr = HeapAlloc(GetProcessHeap(), 0, nSize * sizeof(WCHAR)))) return 0;
 419
 420     if (!GetModuleFileNameExW(hProcess, hModule, ptr, nSize))
 421     {
 422         lpFileName[0] = '\0';
 423     }
 424     else
 425     {
 426         if (!WideCharToMultiByte( CP_ACP, 0, ptr, -1, lpFileName, nSize, NULL, NULL ))
 427             lpFileName[nSize - 1] = 0;
 428     }
 429
 430     HeapFree(GetProcessHeap(), 0, ptr);
 431     return strlen(lpFileName);
 432 }
 433
 434 /***********************************************************************
 435  *           GetModuleFileNameExW (PSAPI.@)
 436  */
 437 DWORD WINAPI GetModuleFileNameExW(HANDLE hProcess, HMODULE hModule,
 438                                   LPWSTR lpFileName, DWORD nSize)
 439 {
 440     LDR_MODULE LdrModule;
 441
 442     if(!PSAPI_GetLdrModule(hProcess, hModule, &LdrModule))
 443         return 0;
 444
 445     nSize = min(LdrModule.FullDllName.Length / sizeof(WCHAR), nSize);
 446     if (!ReadProcessMemory(hProcess, LdrModule.FullDllName.Buffer,
 447                            lpFileName, nSize * sizeof(WCHAR), NULL))
 448         return 0;
 449
 450     lpFileName[nSize] = 0;
 451     return nSize;
 452 }
 453
 454 /***********************************************************************
 455  *           GetModuleInformation (PSAPI.@)
 456  */
 457 BOOL WINAPI GetModuleInformation(HANDLE hProcess, HMODULE hModule,
 458                                  LPMODULEINFO lpmodinfo, DWORD cb)
 459 {
 460     LDR_MODULE LdrModule;
 461
 462     if (cb < sizeof(MODULEINFO))
 463     {
 464         SetLastError(ERROR_INSUFFICIENT_BUFFER);
 465         return FALSE;
 466     }
 467
 468     if (!PSAPI_GetLdrModule(hProcess, hModule, &LdrModule))
 469         return FALSE;
 470
 471     lpmodinfo->lpBaseOfDll = LdrModule.BaseAddress;
 472     lpmodinfo->SizeOfImage = LdrModule.SizeOfImage;
 473     lpmodinfo->EntryPoint  = LdrModule.EntryPoint;
 474     return TRUE;
 475 }
 476
 477 /***********************************************************************
 478  *           GetPerformanceInfo (PSAPI.@)
 479  */
 480 BOOL WINAPI GetPerformanceInfo( PPERFORMANCE_INFORMATION info, DWORD size )
 481 {
 482     NTSTATUS status;
 483
 484     TRACE( "(%p, %d)\n", info, size );
 485
 486     status = NtQuerySystemInformation( SystemPerformanceInformation, info, size, NULL );
 487
 488     if (status)
 489     {
 490         SetLastError( RtlNtStatusToDosError( status ) );
 491         return FALSE;
 492     }
 493     return TRUE;
 494 }
 495
 496 /***********************************************************************
 497  *           GetProcessImageFileNameA (PSAPI.@)
 498  */
 499 DWORD WINAPI GetProcessImageFileNameA( HANDLE process, LPSTR file, DWORD size )
 500 {
 501     FIXME("(%p, %p, %d) stub\n", process, file, size );
 502     return 0;
 503 }
 504
 505 /***********************************************************************
 506  *           GetProcessImageFileNameW (PSAPI.@)
 507  */
 508 DWORD WINAPI GetProcessImageFileNameW( HANDLE process, LPWSTR file, DWORD size )
 509 {
 510     BOOL success = QueryFullProcessImageNameW(process, PROCESS_NAME_NATIVE, file, &size);
 511     if (success)
 512         return size;
 513     else
 514         return 0;
 515 }
 516
 517 /***********************************************************************
 518  *           GetProcessMemoryInfo (PSAPI.@)
 519  *
 520  * Retrieve memory usage information for a given process
 521  *
 522  */
 523 BOOL WINAPI GetProcessMemoryInfo(HANDLE hProcess,
 524                                  PPROCESS_MEMORY_COUNTERS pmc, DWORD cb)
 525 {
 526     NTSTATUS status;
 527     VM_COUNTERS vmc;
 528
 529     if (cb < sizeof(PROCESS_MEMORY_COUNTERS))
 530     {
 531         SetLastError(ERROR_INSUFFICIENT_BUFFER);
 532         return FALSE;
 533     }
 534
 535     status = NtQueryInformationProcess(hProcess, ProcessVmCounters,
 536                                        &vmc, sizeof(vmc), NULL);
 537
 538     if (status)
 539     {
 540         SetLastError(RtlNtStatusToDosError(status));
 541         return FALSE;
 542     }
 543
 544     pmc->cb = sizeof(PROCESS_MEMORY_COUNTERS);
 545     pmc->PageFaultCount = vmc.PageFaultCount;
 546     pmc->PeakWorkingSetSize = vmc.PeakWorkingSetSize;
 547     pmc->WorkingSetSize = vmc.WorkingSetSize;
 548     pmc->QuotaPeakPagedPoolUsage = vmc.QuotaPeakPagedPoolUsage;
 549     pmc->QuotaPagedPoolUsage = vmc.QuotaPagedPoolUsage;
 550     pmc->QuotaPeakNonPagedPoolUsage = vmc.QuotaPeakNonPagedPoolUsage;
 551     pmc->QuotaNonPagedPoolUsage = vmc.QuotaNonPagedPoolUsage;
 552     pmc->PagefileUsage = vmc.PagefileUsage;
 553     pmc->PeakPagefileUsage = vmc.PeakPagefileUsage;
 554
 555     return TRUE;
 556 }
 557
 558 /***********************************************************************
 559  *           GetWsChanges (PSAPI.@)
 560  */
 561 BOOL WINAPI GetWsChanges( HANDLE process, PPSAPI_WS_WATCH_INFORMATION watchinfo, DWORD size )
 562 {
 563     NTSTATUS status;
 564
 565     TRACE( "(%p, %p, %d)\n", process, watchinfo, size );
 566
 567     status = NtQueryInformationProcess( process, ProcessWorkingSetWatch, watchinfo, size, NULL );
 568
 569     if (status)
 570     {
 571         SetLastError( RtlNtStatusToDosError( status ) );
 572         return FALSE;
 573     }
 574     return TRUE;
 575 }
 576
 577 /***********************************************************************
 578  *           InitializeProcessForWsWatch (PSAPI.@)
 579  */
 580 BOOL WINAPI InitializeProcessForWsWatch(HANDLE hProcess)
 581 {
 582     FIXME("(hProcess=%p): stub\n", hProcess);
 583
 584     return TRUE;
 585 }
 586
 587 /***********************************************************************
 588  *           QueryWorkingSet (PSAPI.@)
 589  */
 590 BOOL WINAPI QueryWorkingSet( HANDLE process, LPVOID buffer, DWORD size )
 591 {
 592     NTSTATUS status;
 593
 594     TRACE( "(%p, %p, %d)\n", process, buffer, size );
 595
 596     status = NtQueryVirtualMemory( process, NULL, MemoryWorkingSetList, buffer, size, NULL );
 597
 598     if (status)
 599     {
 600         SetLastError( RtlNtStatusToDosError( status ) );
 601         return FALSE;
 602     }
 603     return TRUE;
 604 }
 605
 606 /***********************************************************************
 607  *           QueryWorkingSetEx (PSAPI.@)
 608  */
 609 BOOL WINAPI QueryWorkingSetEx( HANDLE process, LPVOID buffer, DWORD size )
 610 {
 611     NTSTATUS status;
 612
 613     TRACE( "(%p, %p, %d)\n", process, buffer, size );
 614
 615     status = NtQueryVirtualMemory( process, NULL, MemoryWorkingSetList, buffer,  size, NULL );
 616
 617     if (status)
 618     {
 619         SetLastError( RtlNtStatusToDosError( status ) );
 620         return FALSE;
 621     }
 622     return TRUE;
 623 }