vcs.maemo.org Git - physicsfs/blob - lzma/CPP/Windows/MemoryLock.cpp

   1 // Common/MemoryLock.cpp
   2
   3 #include "StdAfx.h"
   4
   5 namespace NWindows {
   6 namespace NSecurity {
   7
   8 #ifndef _UNICODE
   9 typedef BOOL (WINAPI * OpenProcessTokenP)(HANDLE ProcessHandle, DWORD DesiredAccess, PHANDLE TokenHandle);
  10 typedef BOOL (WINAPI * LookupPrivilegeValueP)(LPCTSTR lpSystemName, LPCTSTR lpName, PLUID  lpLuid);
  11 typedef BOOL (WINAPI * AdjustTokenPrivilegesP)(HANDLE TokenHandle, BOOL DisableAllPrivileges,
  12     PTOKEN_PRIVILEGES NewState, DWORD BufferLength, PTOKEN_PRIVILEGES PreviousState,PDWORD ReturnLength);
  13 #endif
  14
  15 #ifdef _UNICODE
  16 bool EnableLockMemoryPrivilege(
  17 #else
  18 static bool EnableLockMemoryPrivilege2(HMODULE hModule,
  19 #endif
  20 bool enable)
  21 {
  22   #ifndef _UNICODE
  23   if (hModule == NULL)
  24     return false;
  25   OpenProcessTokenP openProcessToken = (OpenProcessTokenP)GetProcAddress(hModule, "OpenProcessToken");
  26   LookupPrivilegeValueP lookupPrivilegeValue = (LookupPrivilegeValueP)GetProcAddress(hModule, "LookupPrivilegeValueA" );
  27   AdjustTokenPrivilegesP adjustTokenPrivileges = (AdjustTokenPrivilegesP)GetProcAddress(hModule, "AdjustTokenPrivileges");
  28   if (openProcessToken == NULL || adjustTokenPrivileges == NULL || lookupPrivilegeValue == NULL)
  29     return false;
  30   #endif
  31
  32   HANDLE token;
  33   if (!
  34     #ifdef _UNICODE
  35     ::OpenProcessToken
  36     #else
  37     openProcessToken
  38     #endif
  39     (::GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES, &token))
  40     return false;
  41   TOKEN_PRIVILEGES tp;
  42   bool res = false;
  43   if (
  44     #ifdef _UNICODE
  45     ::LookupPrivilegeValue
  46     #else
  47     lookupPrivilegeValue
  48     #endif
  49     (NULL, SE_LOCK_MEMORY_NAME, &(tp.Privileges[0].Luid)))
  50   {
  51     tp.PrivilegeCount = 1;
  52     tp.Privileges[0].Attributes = enable ? SE_PRIVILEGE_ENABLED: 0;
  53     if (
  54       #ifdef _UNICODE
  55       ::AdjustTokenPrivileges
  56       #else
  57       adjustTokenPrivileges
  58       #endif
  59       (token, FALSE, &tp, 0, NULL, NULL))
  60       res = (GetLastError() == ERROR_SUCCESS);
  61   }
  62   ::CloseHandle(token);
  63   return res;
  64 }
  65
  66 #ifndef _UNICODE
  67 bool EnableLockMemoryPrivilege(bool enable)
  68 {
  69   HMODULE hModule = LoadLibrary(TEXT("Advapi32.dll"));
  70   if(hModule == NULL)
  71     return false;
  72   bool res = EnableLockMemoryPrivilege2(hModule, enable);
  73   ::FreeLibrary(hModule);
  74   return res;
  75 }
  76 #endif
  77
  78 }}