Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 #include "sandbox/win/src/handle_closer.h" | 5 #include "sandbox/win/src/handle_closer.h" |
| 6 | 6 |
| 7 #include "base/logging.h" | 7 #include "base/logging.h" |
| 8 #include "base/memory/scoped_ptr.h" | 8 #include "base/memory/scoped_ptr.h" |
| 9 #include "base/win/windows_version.h" | 9 #include "base/win/windows_version.h" |
| 10 #include "sandbox/win/src/interceptors.h" | 10 #include "sandbox/win/src/interceptors.h" |
| (...skipping 21 matching lines...) Expand all Loading... | |
| 32 // Memory buffer mapped from the parent, with the list of handles. | 32 // Memory buffer mapped from the parent, with the list of handles. |
| 33 SANDBOX_INTERCEPT HandleCloserInfo* g_handles_to_close; | 33 SANDBOX_INTERCEPT HandleCloserInfo* g_handles_to_close; |
| 34 | 34 |
| 35 HandleCloser::HandleCloser() {} | 35 HandleCloser::HandleCloser() {} |
| 36 | 36 |
| 37 ResultCode HandleCloser::AddHandle(const base::char16* handle_type, | 37 ResultCode HandleCloser::AddHandle(const base::char16* handle_type, |
| 38 const base::char16* handle_name) { | 38 const base::char16* handle_name) { |
| 39 if (!handle_type) | 39 if (!handle_type) |
| 40 return SBOX_ERROR_BAD_PARAMS; | 40 return SBOX_ERROR_BAD_PARAMS; |
| 41 | 41 |
| 42 base::string16 resolved_name; | |
| 43 if (handle_name) { | |
| 44 resolved_name = handle_name; | |
| 45 if (handle_type == base::string16(L"Key")) | |
|
cpu_(ooo_6.6-7.5)
2014/08/13 23:18:35
can you use == L"key"; so avoids constructing a st
| |
| 46 if (!ResolveRegistryName(resolved_name, &resolved_name)) | |
| 47 return SBOX_ERROR_BAD_PARAMS; | |
| 48 } | |
| 49 | |
| 42 HandleMap::iterator names = handles_to_close_.find(handle_type); | 50 HandleMap::iterator names = handles_to_close_.find(handle_type); |
| 43 if (names == handles_to_close_.end()) { // We have no entries for this type. | 51 if (names == handles_to_close_.end()) { // We have no entries for this type. |
| 44 std::pair<HandleMap::iterator, bool> result = handles_to_close_.insert( | 52 std::pair<HandleMap::iterator, bool> result = handles_to_close_.insert( |
| 45 HandleMap::value_type(handle_type, HandleMap::mapped_type())); | 53 HandleMap::value_type(handle_type, HandleMap::mapped_type())); |
| 46 names = result.first; | 54 names = result.first; |
| 47 if (handle_name) | 55 if (handle_name) |
| 48 names->second.insert(handle_name); | 56 names->second.insert(resolved_name); |
| 49 } else if (!handle_name) { // Now we need to close all handles of this type. | 57 } else if (!handle_name) { // Now we need to close all handles of this type. |
| 50 names->second.clear(); | 58 names->second.clear(); |
| 51 } else if (!names->second.empty()) { // Add another name for this type. | 59 } else if (!names->second.empty()) { // Add another name for this type. |
| 52 names->second.insert(handle_name); | 60 names->second.insert(resolved_name); |
| 53 } // If we're already closing all handles of type then we're done. | 61 } // If we're already closing all handles of type then we're done. |
| 54 | 62 |
| 55 return SBOX_ALL_OK; | 63 return SBOX_ALL_OK; |
| 56 } | 64 } |
| 57 | 65 |
| 58 size_t HandleCloser::GetBufferSize() { | 66 size_t HandleCloser::GetBufferSize() { |
| 59 size_t bytes_total = offsetof(HandleCloserInfo, handle_entries); | 67 size_t bytes_total = offsetof(HandleCloserInfo, handle_entries); |
| 60 | 68 |
| 61 for (HandleMap::iterator i = handles_to_close_.begin(); | 69 for (HandleMap::iterator i = handles_to_close_.begin(); |
| 62 i != handles_to_close_.end(); ++i) { | 70 i != handles_to_close_.end(); ++i) { |
| (...skipping 110 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 173 | 181 |
| 174 if (NT_SUCCESS(result) && name->Buffer && name->Length) | 182 if (NT_SUCCESS(result) && name->Buffer && name->Length) |
| 175 handle_name->assign(name->Buffer, name->Length / sizeof(wchar_t)); | 183 handle_name->assign(name->Buffer, name->Length / sizeof(wchar_t)); |
| 176 else | 184 else |
| 177 handle_name->clear(); | 185 handle_name->clear(); |
| 178 | 186 |
| 179 return NT_SUCCESS(result); | 187 return NT_SUCCESS(result); |
| 180 } | 188 } |
| 181 | 189 |
| 182 } // namespace sandbox | 190 } // namespace sandbox |
| OLD | NEW |