OLD | NEW |
1 // Copyright (c) 2006-2010 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/tests/common/controller.h" | 5 #include "sandbox/tests/common/controller.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 | 8 |
| 9 #include "base/win/windows_version.h" |
9 #include "sandbox/src/sandbox_factory.h" | 10 #include "sandbox/src/sandbox_factory.h" |
10 #include "sandbox/src/sandbox_utils.h" | 11 #include "sandbox/src/sandbox_utils.h" |
11 #include "sandbox/src/wow64.h" | |
12 | 12 |
13 namespace { | 13 namespace { |
14 | 14 |
15 static const int kDefaultTimeout = 3000; | 15 static const int kDefaultTimeout = 3000; |
16 | 16 |
17 // Constructs a full path to a file inside the system32 folder. | 17 // Constructs a full path to a file inside the system32 folder. |
18 std::wstring MakePathToSys32(const wchar_t* name, bool is_obj_man_path) { | 18 std::wstring MakePathToSys32(const wchar_t* name, bool is_obj_man_path) { |
19 wchar_t windows_path[MAX_PATH] = {0}; | 19 wchar_t windows_path[MAX_PATH] = {0}; |
20 if (0 == ::GetSystemWindowsDirectoryW(windows_path, MAX_PATH)) | 20 if (0 == ::GetSystemWindowsDirectoryW(windows_path, MAX_PATH)) |
21 return std::wstring(); | 21 return std::wstring(); |
(...skipping 26 matching lines...) Expand all Loading... |
48 full_path += L"\\SysWOW64\\"; | 48 full_path += L"\\SysWOW64\\"; |
49 full_path += name; | 49 full_path += name; |
50 return full_path; | 50 return full_path; |
51 } | 51 } |
52 | 52 |
53 } // namespace | 53 } // namespace |
54 | 54 |
55 namespace sandbox { | 55 namespace sandbox { |
56 | 56 |
57 std::wstring MakePathToSys(const wchar_t* name, bool is_obj_man_path) { | 57 std::wstring MakePathToSys(const wchar_t* name, bool is_obj_man_path) { |
58 Wow64 current_proc(NULL, NULL); | 58 if (base::win::GetWOW64Status() == base::win::WOW64_ENABLED) |
59 if (current_proc.IsWow64()) | |
60 return MakePathToSysWow64(name, is_obj_man_path); | 59 return MakePathToSysWow64(name, is_obj_man_path); |
61 else | 60 return MakePathToSys32(name, is_obj_man_path); |
62 return MakePathToSys32(name, is_obj_man_path); | |
63 } | 61 } |
64 | 62 |
65 BrokerServices* GetBroker() { | 63 BrokerServices* GetBroker() { |
66 static BrokerServices* broker = SandboxFactory::GetBrokerServices(); | 64 static BrokerServices* broker = SandboxFactory::GetBrokerServices(); |
67 static bool is_initialized = false; | 65 static bool is_initialized = false; |
68 | 66 |
69 if (!broker) { | 67 if (!broker) { |
70 return NULL; | 68 return NULL; |
71 } | 69 } |
72 | 70 |
(...skipping 60 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
133 if (!is_init_) | 131 if (!is_init_) |
134 return false; | 132 return false; |
135 | 133 |
136 std::wstring win32_path = MakePathToSys32(pattern, false); | 134 std::wstring win32_path = MakePathToSys32(pattern, false); |
137 if (win32_path.empty()) | 135 if (win32_path.empty()) |
138 return false; | 136 return false; |
139 | 137 |
140 if (!AddRule(TargetPolicy::SUBSYS_FILES, semantics, win32_path.c_str())) | 138 if (!AddRule(TargetPolicy::SUBSYS_FILES, semantics, win32_path.c_str())) |
141 return false; | 139 return false; |
142 | 140 |
143 Wow64 current_proc(NULL, NULL); | 141 if (base::win::GetWOW64Status() != base::win::WOW64_ENABLED) |
144 if (!current_proc.IsWow64()) | |
145 return true; | 142 return true; |
146 | 143 |
147 win32_path = MakePathToSysWow64(pattern, false); | 144 win32_path = MakePathToSysWow64(pattern, false); |
148 if (win32_path.empty()) | 145 if (win32_path.empty()) |
149 return false; | 146 return false; |
150 | 147 |
151 return AddRule(TargetPolicy::SUBSYS_FILES, semantics, win32_path.c_str()); | 148 return AddRule(TargetPolicy::SUBSYS_FILES, semantics, win32_path.c_str()); |
152 } | 149 } |
153 | 150 |
154 bool TestRunner::AddFsRule(TargetPolicy::Semantics semantics, | 151 bool TestRunner::AddFsRule(TargetPolicy::Semantics semantics, |
(...skipping 128 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
283 if (BEFORE_REVERT == state) | 280 if (BEFORE_REVERT == state) |
284 return command(argc - 4, argv + 4); | 281 return command(argc - 4, argv + 4); |
285 else if (EVERY_STATE == state) | 282 else if (EVERY_STATE == state) |
286 command(argc - 4, argv + 4); | 283 command(argc - 4, argv + 4); |
287 | 284 |
288 target->LowerToken(); | 285 target->LowerToken(); |
289 return command(argc - 4, argv + 4); | 286 return command(argc - 4, argv + 4); |
290 } | 287 } |
291 | 288 |
292 } // namespace sandbox | 289 } // namespace sandbox |
OLD | NEW |