OLD | NEW |
| (Empty) |
1 // Copyright 2007-2009 Google Inc. | |
2 // | |
3 // Licensed under the Apache License, Version 2.0 (the "License"); | |
4 // you may not use this file except in compliance with the License. | |
5 // You may obtain a copy of the License at | |
6 // | |
7 // http://www.apache.org/licenses/LICENSE-2.0 | |
8 // | |
9 // Unless required by applicable law or agreed to in writing, software | |
10 // distributed under the License is distributed on an "AS IS" BASIS, | |
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
12 // See the License for the specific language governing permissions and | |
13 // limitations under the License. | |
14 // ======================================================================== | |
15 | |
16 #include "omaha/base/app_util.h" | |
17 #include "omaha/base/path.h" | |
18 #include "omaha/base/utils.h" | |
19 #include "omaha/base/vista_utils.h" | |
20 #include "omaha/base/vistautil.h" | |
21 #include "omaha/testing/unit_test.h" | |
22 | |
23 namespace omaha { | |
24 | |
25 namespace vista { | |
26 | |
27 // Exercises RunAsUser() with explorer token. For Vista, the call to | |
28 // StartProcessWithTokenOfProcess() will succeed only if the caller is SYSTEM. | |
29 TEST(VistaUtilsTest, StartProcessWithExplorerTokenTest) { | |
30 CString path = ConcatenatePath(app_util::GetSystemDir(), _T("cmd.exe")); | |
31 EnclosePath(&path); | |
32 path += _T(" /c exit 702"); | |
33 uint32 pid(0); | |
34 EXPECT_SUCCEEDED(GetExplorerPidForCurrentUserOrSession(&pid)); | |
35 | |
36 HRESULT hr = StartProcessWithTokenOfProcess(pid, path); | |
37 if (!vista_util::IsVistaOrLater()) { | |
38 EXPECT_SUCCEEDED(hr); | |
39 return; | |
40 } | |
41 | |
42 bool is_system = false; | |
43 EXPECT_SUCCEEDED(IsSystemProcess(&is_system)); | |
44 if (is_system) { | |
45 EXPECT_SUCCEEDED(hr); | |
46 return; | |
47 } | |
48 | |
49 EXPECT_EQ(HRESULT_FROM_WIN32(ERROR_PRIVILEGE_NOT_HELD), hr); | |
50 } | |
51 | |
52 // Exercises RunAsUser() with current user token. | |
53 TEST(VistaUtilsTest, RunAsCurrentUserTest) { | |
54 CString path = ConcatenatePath(app_util::GetSystemDir(), _T("cmd.exe")); | |
55 EnclosePath(&path); | |
56 path += _T(" /c exit 702"); | |
57 EXPECT_SUCCEEDED(vista::RunAsCurrentUser(path)); | |
58 } | |
59 | |
60 } // namespace vista | |
61 | |
62 } // namespace omaha | |
63 | |
OLD | NEW |