| OLD | NEW |
| (Empty) |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "chrome/browser/prefs/tracked/device_id.h" | |
| 6 | |
| 7 #include <windows.h> | |
| 8 #include <sddl.h> // For ConvertSidToStringSidA. | |
| 9 | |
| 10 #include "base/logging.h" | |
| 11 #include "base/macros.h" | |
| 12 #include "base/memory/scoped_ptr.h" | |
| 13 | |
| 14 MachineIdStatus GetDeterministicMachineSpecificId(std::string* machine_id) { | |
| 15 DCHECK(machine_id); | |
| 16 | |
| 17 wchar_t computer_name[MAX_COMPUTERNAME_LENGTH + 1] = {}; | |
| 18 DWORD computer_name_size = arraysize(computer_name); | |
| 19 | |
| 20 if (!::GetComputerNameW(computer_name, &computer_name_size)) | |
| 21 return MachineIdStatus::FAILURE; | |
| 22 | |
| 23 DWORD sid_size = SECURITY_MAX_SID_SIZE; | |
| 24 char sid_buffer[SECURITY_MAX_SID_SIZE]; | |
| 25 SID* sid = reinterpret_cast<SID*>(sid_buffer); | |
| 26 DWORD domain_size = 128; // Will expand below if needed. | |
| 27 scoped_ptr<wchar_t[]> domain_buffer(new wchar_t[domain_size]); | |
| 28 SID_NAME_USE sid_name_use; | |
| 29 | |
| 30 // Although the fifth argument to |LookupAccountNameW()|, | |
| 31 // |ReferencedDomainName|, is annotated as |_Out_opt_|, if a null | |
| 32 // value is passed in, zero is returned and |GetLastError()| will | |
| 33 // return |ERROR_INSUFFICIENT_BUFFER| (assuming that nothing else went | |
| 34 // wrong). In order to ensure that the call to |LookupAccountNameW()| | |
| 35 // has succeeded, it is necessary to include the following logic and | |
| 36 // obtain the domain name. | |
| 37 if (!::LookupAccountNameW(nullptr, computer_name, sid, &sid_size, | |
| 38 domain_buffer.get(), &domain_size, &sid_name_use)) { | |
| 39 // If the initial size of |domain_buffer| was too small, the | |
| 40 // required size is now found in |domain_size|. Resize and try | |
| 41 // again. | |
| 42 if (::GetLastError() != ERROR_INSUFFICIENT_BUFFER) | |
| 43 return MachineIdStatus::FAILURE; | |
| 44 | |
| 45 domain_buffer.reset(new wchar_t[domain_size]); | |
| 46 if (!::LookupAccountNameW(nullptr, computer_name, sid, &sid_size, | |
| 47 domain_buffer.get(), &domain_size, | |
| 48 &sid_name_use)) { | |
| 49 return MachineIdStatus::FAILURE; | |
| 50 } | |
| 51 } | |
| 52 | |
| 53 // Ensure that the correct type of SID was obtained. The | |
| 54 // |LookupAccountNameW()| function seems to always return | |
| 55 // |SidTypeDomain| instead of |SidTypeComputer| when the computer name | |
| 56 // is passed in as its second argument and therefore both enum values | |
| 57 // will be considered acceptable. If the computer name and user name | |
| 58 // coincide, |LookupAccountNameW()| seems to always return the machine | |
| 59 // SID and set the returned enum to |SidTypeDomain|. | |
| 60 DCHECK(sid_name_use == SID_NAME_USE::SidTypeComputer || | |
| 61 sid_name_use == SID_NAME_USE::SidTypeDomain); | |
| 62 | |
| 63 char* sid_string = nullptr; | |
| 64 if (!::ConvertSidToStringSidA(sid, &sid_string)) | |
| 65 return MachineIdStatus::FAILURE; | |
| 66 | |
| 67 *machine_id = sid_string; | |
| 68 ::LocalFree(sid_string); | |
| 69 | |
| 70 return MachineIdStatus::SUCCESS; | |
| 71 } | |
| OLD | NEW |