Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 // | 4 // |
| 5 // Information about the current process. | 5 // Information about the current process. |
| 6 | 6 |
| 7 #include "rlz/win/lib/process_info.h" | 7 #include "rlz/win/lib/process_info.h" |
| 8 | 8 |
| 9 #include <windows.h> | 9 #include <windows.h> |
| 10 #include <Sddl.h> // For ConvertSidToStringSid. | 10 #include <Sddl.h> // For ConvertSidToStringSid. |
| (...skipping 15 matching lines...) Expand all Loading... | |
| 26 | 26 |
| 27 // Get the current username & domain the hard way. (GetUserNameEx would be | 27 // Get the current username & domain the hard way. (GetUserNameEx would be |
| 28 // nice, but unfortunately requires connectivity to a domain controller. | 28 // nice, but unfortunately requires connectivity to a domain controller. |
| 29 // Useless.) | 29 // Useless.) |
| 30 | 30 |
| 31 // (Following call doesn't work if running as a Service - because a Service | 31 // (Following call doesn't work if running as a Service - because a Service |
| 32 // runs under special accounts like LOCAL_SYSTEM, not as the logged in user. | 32 // runs under special accounts like LOCAL_SYSTEM, not as the logged in user. |
| 33 // In which case, search for and use the process handle of a running | 33 // In which case, search for and use the process handle of a running |
| 34 // Explorer.exe.) | 34 // Explorer.exe.) |
| 35 HANDLE token; | 35 HANDLE token; |
| 36 if (!::OpenProcessToken(::GetCurrentProcess(), TOKEN_QUERY, &token)) | 36 |
| 37 return E_FAIL; | 37 CHECK(::OpenProcessToken(::GetCurrentProcess(), TOKEN_QUERY, &token)); |
| 38 | 38 |
| 39 base::win::ScopedHandle scoped_process_token(token); | 39 base::win::ScopedHandle scoped_process_token(token); |
| 40 | 40 |
| 41 // (Following call will fail with ERROR_INSUFFICIENT_BUFFER and give us the | 41 // (Following call will fail with ERROR_INSUFFICIENT_BUFFER and give us the |
| 42 // required size.) | 42 // required size.) |
| 43 scoped_ptr<char[]> token_user_bytes; | 43 scoped_ptr<char[]> token_user_bytes; |
| 44 DWORD token_user_size; | 44 DWORD token_user_size; |
| 45 DWORD token_user_size2; | 45 DWORD token_user_size2; |
| 46 BOOL result = ::GetTokenInformation(token, TokenUser, NULL, 0, | 46 BOOL result = ::GetTokenInformation(token, TokenUser, NULL, 0, |
| 47 &token_user_size); | 47 &token_user_size); |
| 48 err = ::GetLastError(); | 48 err = ::GetLastError(); |
| 49 CHECK(!result && err == ERROR_INSUFFICIENT_BUFFER); | 49 CHECK(!result && err == ERROR_INSUFFICIENT_BUFFER); |
| 50 | 50 |
| 51 token_user_bytes.reset(new char[token_user_size]); | 51 token_user_bytes.reset(new char[token_user_size]); |
| 52 if (!token_user_bytes.get()) | 52 CHECK(token_user_bytes.get()); |
| 53 return E_OUTOFMEMORY; | |
| 54 | 53 |
| 55 if (!::GetTokenInformation(token, TokenUser, token_user_bytes.get(), | 54 CHECK(::GetTokenInformation(token, TokenUser, token_user_bytes.get(), |
| 56 token_user_size, &token_user_size2)) { | 55 token_user_size, &token_user_size2)) |
|
Roger Tawa OOO till Jul 10th
2014/03/01 14:54:58
Remove extra space, add missing ; at end.
yiyaoliu
2014/03/05 21:32:44
Done.
| |
| 57 return E_FAIL; | |
| 58 } | |
| 59 | 56 |
| 60 WCHAR user_name[UNLEN + 1]; // max username length | 57 WCHAR user_name[UNLEN + 1]; // max username length |
| 61 WCHAR domain_name[UNLEN + 1]; | 58 WCHAR domain_name[UNLEN + 1]; |
| 62 DWORD user_name_size = UNLEN + 1; | 59 DWORD user_name_size = UNLEN + 1; |
| 63 DWORD domain_name_size = UNLEN + 1; | 60 DWORD domain_name_size = UNLEN + 1; |
| 64 SID_NAME_USE sid_type; | 61 SID_NAME_USE sid_type; |
| 65 TOKEN_USER* token_user = | 62 TOKEN_USER* token_user = |
| 66 reinterpret_cast<TOKEN_USER*>(token_user_bytes.get()); | 63 reinterpret_cast<TOKEN_USER*>(token_user_bytes.get()); |
| 67 if (!token_user) | 64 if (!token_user) |
| 68 return E_FAIL; | 65 CHECK(0); |
|
Roger Tawa OOO till Jul 10th
2014/03/01 14:54:58
You can replace lines 64-65 with:
CHECK(token_
yiyaoliu
2014/03/05 21:32:44
Done.
| |
| 66 | |
| 69 PSID user_sid = token_user->User.Sid; | 67 PSID user_sid = token_user->User.Sid; |
| 70 if (!::LookupAccountSidW(NULL, user_sid, user_name, &user_name_size, | 68 CHECK(::LookupAccountSidW(NULL, user_sid, user_name, &user_name_size, |
| 71 domain_name, &domain_name_size, &sid_type)) { | 69 domain_name, &domain_name_size, &sid_type)) |
| 72 return E_FAIL; | |
| 73 } | |
| 74 | 70 |
| 75 if (name != NULL) { | 71 if (name != NULL) { |
| 76 *name = user_name; | 72 *name = user_name; |
| 77 } | 73 } |
| 78 if (domain != NULL) { | 74 if (domain != NULL) { |
| 79 *domain = domain_name; | 75 *domain = domain_name; |
| 80 } | 76 } |
| 81 if (sid != NULL) { | 77 if (sid != NULL) { |
| 82 LPWSTR string_sid; | 78 LPWSTR string_sid; |
| 83 ConvertSidToStringSidW(user_sid, &string_sid); | 79 ConvertSidToStringSidW(user_sid, &string_sid); |
| (...skipping 102 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 186 } | 182 } |
| 187 | 183 |
| 188 evaluated = true; | 184 evaluated = true; |
| 189 if (!has_rights) | 185 if (!has_rights) |
| 190 ASSERT_STRING("ProcessInfo::HasAdminRights: Does not have admin rights."); | 186 ASSERT_STRING("ProcessInfo::HasAdminRights: Does not have admin rights."); |
| 191 | 187 |
| 192 return has_rights; | 188 return has_rights; |
| 193 } | 189 } |
| 194 | 190 |
| 195 }; // namespace | 191 }; // namespace |
| OLD | NEW |