6.5

1 files changed, 0 insertions, 138 deletions

diff --git a/zen/privilege.cpp b/zen/privilege.cpp
deleted file mode 100644
index 90612756..00000000
--- a/zen/privilege.cpp
+++ /dev/null
@@ -1,138 +0,0 @@
-#include "privilege.h"
-#include <map>
-#include <mutex>
-#include "zstring.h"
-#include "scope_guard.h"
-#include "win_ver.h"
-
-using namespace zen;
-
-
-namespace
-{
-bool privilegeIsActive(LPCTSTR privilege) //throw FileError
-{
-    HANDLE hToken = nullptr;
-    if (!::OpenProcessToken(::GetCurrentProcess(), //__in   HANDLE ProcessHandle,
-                            TOKEN_QUERY,           //__in   DWORD DesiredAccess,
-                            &hToken))              //__out  PHANDLE TokenHandle
-        throw FileError(replaceCpy(_("Cannot set privilege %x."), L"%x", std::wstring(L"\"") + privilege +  L"\""), formatSystemError(L"OpenProcessToken", getLastError()));
-    ZEN_ON_SCOPE_EXIT(::CloseHandle(hToken));
-
-    LUID luid = {};
-    if (!::LookupPrivilegeValue(nullptr,   //__in_opt  LPCTSTR lpSystemName,
-                                privilege, //__in      LPCTSTR lpName,
-                                &luid ))   //__out     PLUID lpLuid
-        throw FileError(replaceCpy(_("Cannot set privilege %x."), L"%x", std::wstring(L"\"") + privilege +  L"\""), formatSystemError(L"LookupPrivilegeValue", getLastError()));
-
-    PRIVILEGE_SET priv  = {};
-    priv.PrivilegeCount = 1;
-    priv.Control        = PRIVILEGE_SET_ALL_NECESSARY;
-    priv.Privilege[0].Luid = luid;
-    priv.Privilege[0].Attributes = SE_PRIVILEGE_ENABLED;
-
-    BOOL alreadyGranted = FALSE;
-    if (!::PrivilegeCheck(hToken,           //__in     HANDLE ClientToken,
-                          &priv,            //__inout  PPRIVILEGE_SET RequiredPrivileges,
-                          &alreadyGranted)) //__out    LPBOOL pfResult
-        throw FileError(replaceCpy(_("Cannot set privilege %x."), L"%x", std::wstring(L"\"") + privilege +  L"\""), formatSystemError(L"PrivilegeCheck", getLastError()));
-
-    return alreadyGranted != FALSE;
-}
-
-
-void setPrivilege(LPCTSTR privilege, bool enable) //throw FileError
-{
-    HANDLE hToken = nullptr;
-    if (!::OpenProcessToken(::GetCurrentProcess(),   //__in   HANDLE ProcessHandle,
-                            TOKEN_ADJUST_PRIVILEGES, //__in   DWORD DesiredAccess,
-                            &hToken))                //__out  PHANDLE TokenHandle
-        throw FileError(replaceCpy(_("Cannot set privilege %x."), L"%x", std::wstring(L"\"") + privilege +  L"\""), formatSystemError(L"OpenProcessToken", getLastError()));
-    ZEN_ON_SCOPE_EXIT(::CloseHandle(hToken));
-
-    LUID luid = {};
-    if (!::LookupPrivilegeValue(nullptr,   //__in_opt  LPCTSTR lpSystemName,
-                                privilege, //__in      LPCTSTR lpName,
-                                &luid ))   //__out     PLUID lpLuid
-        throw FileError(replaceCpy(_("Cannot set privilege %x."), L"%x", std::wstring(L"\"") + privilege +  L"\""), formatSystemError(L"LookupPrivilegeValue", getLastError()));
-
-    TOKEN_PRIVILEGES tp = {};
-    tp.PrivilegeCount   = 1;
-    tp.Privileges[0].Luid = luid;
-    tp.Privileges[0].Attributes = enable ? SE_PRIVILEGE_ENABLED : 0;
-
-    if (!::AdjustTokenPrivileges(hToken,   //__in       HANDLE TokenHandle,
-                                 false,    //__in       BOOL DisableAllPrivileges,
-                                 &tp,      //__in_opt   PTOKEN_PRIVILEGES NewState,
-                                 0,        //__in       DWORD BufferLength,
-                                 nullptr,  //__out_opt  PTOKEN_PRIVILEGES PreviousState,
-                                 nullptr)) //__out_opt  PDWORD ReturnLength
-        throw FileError(replaceCpy(_("Cannot set privilege %x."), L"%x", std::wstring(L"\"") + privilege +  L"\""), formatSystemError(L"AdjustTokenPrivileges", getLastError()));
-
-    ErrorCode lastError = getLastError();
-    if (lastError == ERROR_NOT_ALL_ASSIGNED) //check although previous function returned with success!
-    {
-#ifdef __MINGW32__ //Shobjidl.h
-#define ERROR_ELEVATION_REQUIRED         740L
-#endif
-        if (vistaOrLater()) //replace this useless error code with what it *really* means!
-            lastError = ERROR_ELEVATION_REQUIRED;
-
-        throw FileError(replaceCpy(_("Cannot set privilege %x."), L"%x", std::wstring(L"\"") + privilege +  L"\""), formatSystemError(L"AdjustTokenPrivileges", lastError));
-    }
-}
-
-
-class Privileges
-{
-public:
-    static Privileges& getInstance()
-    {
-        static Privileges inst;
-        return inst;
-    }
-
-    void ensureActive(LPCTSTR privilege) //throw FileError
-    {
-        if (activePrivileges.find(privilege) != activePrivileges.end())
-            return; //privilege already active
-
-        if (privilegeIsActive(privilege)) //privilege was already active before starting this tool
-            activePrivileges.insert(std::make_pair(privilege, false));
-        else
-        {
-            setPrivilege(privilege, true);
-            activePrivileges.insert(std::make_pair(privilege, true));
-        }
-    }
-
-private:
-    Privileges() {}
-    Privileges(Privileges&);
-    Privileges& operator=(const Privileges&);
-
-    ~Privileges() //clean up: deactivate all privileges that have been activated by this application
-    {
-        for (const auto& priv : activePrivileges)
-            if (priv.second)
-            {
-                try
-                {
-                    setPrivilege(priv.first.c_str(), false); //throw FileError
-                }
-                catch (FileError&) {}
-            }
-    }
-
-    std::map<Zstring, bool> activePrivileges; //bool: enabled by this application
-};
-
-std::mutex lockPrivileges;
-}
-
-
-void zen::activatePrivilege(LPCTSTR privilege) //throw FileError
-{
-    std::lock_guard<std::mutex> dummy(lockPrivileges);
-    Privileges::getInstance().ensureActive(privilege);
-}