From 1ce0b5f83853f93ea42d478f23947ab4054bbaaa Mon Sep 17 00:00:00 2001
From: "Mr. eXoDia" <mr.exodia.tpodt@gmail.com>
Date: Wed, 16 Apr 2014 16:39:09 +0200
Subject: [PATCH] EngineSetDebugPrivilege function

---
 TitanEngine/Global.Engine.cpp | 26 ++++++++++++++++++++++++++
 TitanEngine/Global.Engine.h   |  1 +
 2 files changed, 27 insertions(+)

diff --git a/TitanEngine/Global.Engine.cpp b/TitanEngine/Global.Engine.cpp
index 8a7055e..6e9400b 100644
--- a/TitanEngine/Global.Engine.cpp
+++ b/TitanEngine/Global.Engine.cpp
@@ -2021,3 +2021,29 @@ ULONG_PTR EngineGlobalAPIHandler(HANDLE handleProcess, ULONG_PTR EnumedModulesBa
     }
     return(NULL);
 }
+
+DWORD EngineSetDebugPrivilege(HANDLE hProcess, bool bEnablePrivilege)
+{
+    DWORD dwLastError;
+    HANDLE hToken = 0;
+    if(!OpenProcessToken(hProcess, TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken))
+    {
+        dwLastError = GetLastError();
+        if(hToken)
+            CloseHandle(hToken);
+        return dwLastError;
+    }
+    TOKEN_PRIVILEGES tokenPrivileges;
+    memset(&tokenPrivileges, 0, sizeof(TOKEN_PRIVILEGES));
+    LUID luid;
+    if(!LookupPrivilegeValue(NULL, SE_DEBUG_NAME, &luid))
+        return false;
+    tokenPrivileges.PrivilegeCount = 1; 
+    tokenPrivileges.Privileges[0].Luid = luid; 
+    if(bEnablePrivilege)
+        tokenPrivileges.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED; 
+    else 
+        tokenPrivileges.Privileges[0].Attributes = 0;
+    AdjustTokenPrivileges(hToken, FALSE, &tokenPrivileges, sizeof(TOKEN_PRIVILEGES), NULL, NULL); 
+    return GetLastError();
+}
diff --git a/TitanEngine/Global.Engine.h b/TitanEngine/Global.Engine.h
index 96db561..1c0726d 100644
--- a/TitanEngine/Global.Engine.h
+++ b/TitanEngine/Global.Engine.h
@@ -51,5 +51,6 @@ ULONG_PTR EngineSimulateDllLoaderW(HANDLE hProcess, wchar_t* szFileName);
 ULONG_PTR EngineGetProcAddress(ULONG_PTR ModuleBase, char* szAPIName);
 bool EngineGetLibraryOrdinalData(ULONG_PTR ModuleBase, LPDWORD ptrOrdinalBase, LPDWORD ptrOrdinalCount);
 ULONG_PTR EngineGlobalAPIHandler(HANDLE handleProcess, ULONG_PTR EnumedModulesBases, ULONG_PTR APIAddress, const char* szAPIName, DWORD ReturnType);
+DWORD EngineSetDebugPrivilege(HANDLE hProcess, bool bEnablePrivilege);
 
 #endif //_GLOBAL_ENGINE_H
\ No newline at end of file