From 2b3f7138f02cd89dfb4f79a75b816e3a52f03aff Mon Sep 17 00:00:00 2001 From: Egor Tensin Date: Fri, 16 Sep 2016 05:06:40 +0300 Subject: refactoring --- src/main.cpp | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) (limited to 'src/main.cpp') diff --git a/src/main.cpp b/src/main.cpp index 8731047..4e31e9f 100644 --- a/src/main.cpp +++ b/src/main.cpp @@ -24,21 +24,22 @@ HWND get_dialog_item(HWND wnd, int id) bool is_administrator() { - const auto token = token::impersonate( - token::get_linked( - token::open_for_current_process(token::permissions::query() | token::permissions::duplicate()))); + auto token = token::open_for_current_process( + token::permissions::query() | token::permissions::duplicate()); + if (!token::get_linked(token)) + token = token::get_for_identification(token); - return token::belongs(token, sid::builtin_administrators()); + return token::check_belongs(token, sid::builtin_administrators()); } bool is_run_as_administrator() { - return token::belongs(token::dumb(), sid::builtin_administrators()); + return token::check_belongs(token::dumb(), sid::builtin_administrators()); } bool is_elevated() { - return token::is_elevated(token::open_for_current_process()); + return token::query_elevation(token::open_for_current_process()); } void decorate_elevate_button(HWND wnd, bool decorate) @@ -49,8 +50,7 @@ void decorate_elevate_button(HWND wnd, bool decorate) DWORD get_integrity_level() { - return token::query_integrity_level( - token::open_for_current_process()); + return token::query_integrity_level(token::open_for_current_process()); } std::wstring get_integrity_level_as_string() -- cgit v1.2.3