OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 "base/bind.h" | 5 #include "base/bind.h" |
6 #include "base/run_loop.h" | 6 #include "base/run_loop.h" |
7 #include "base/strings/utf_string_conversions.h" | 7 #include "base/strings/utf_string_conversions.h" |
8 #include "content/public/browser/browser_thread.h" | 8 #include "content/public/browser/browser_thread.h" |
9 #include "content/public/browser/utility_process_host.h" | 9 #include "content/public/browser/utility_process_host.h" |
10 #include "content/public/browser/utility_process_host_client.h" | 10 #include "content/public/browser/utility_process_host_client.h" |
11 #include "content/public/common/service_registry.h" | 11 #include "content/public/common/service_registry.h" |
12 #include "content/public/test/content_browser_test.h" | 12 #include "content/public/test/content_browser_test.h" |
13 #include "content/public/test/content_browser_test_utils.h" | 13 #include "content/public/test/content_browser_test_utils.h" |
14 #include "content/public/test/test_mojo_service.mojom.h" | 14 #include "content/public/test/test_mojo_service.mojom.h" |
15 | 15 |
16 namespace content { | 16 namespace content { |
17 | 17 |
18 class UtilityProcessHostImplBrowserTest : public ContentBrowserTest { | 18 class UtilityProcessHostImplBrowserTest : public ContentBrowserTest { |
19 public: | 19 public: |
20 void RunUtilityProcess() { | 20 void RunUtilityProcess(bool elevated) { |
21 base::RunLoop run_loop; | 21 base::RunLoop run_loop; |
22 done_closure_ = run_loop.QuitClosure(); | 22 done_closure_ = run_loop.QuitClosure(); |
23 BrowserThread::PostTask( | 23 BrowserThread::PostTask( |
24 BrowserThread::IO, FROM_HERE, | 24 BrowserThread::IO, FROM_HERE, |
25 base::Bind( | 25 base::Bind( |
26 &UtilityProcessHostImplBrowserTest::RunUtilityProcessOnIOThread, | 26 &UtilityProcessHostImplBrowserTest::RunUtilityProcessOnIOThread, |
27 base::Unretained(this))); | 27 base::Unretained(this), elevated)); |
28 run_loop.Run(); | 28 run_loop.Run(); |
29 } | 29 } |
30 | 30 |
31 protected: | 31 protected: |
32 void RunUtilityProcessOnIOThread() { | 32 void RunUtilityProcessOnIOThread(bool elevated) { |
33 UtilityProcessHost* host = UtilityProcessHost::Create(nullptr, nullptr); | 33 UtilityProcessHost* host = UtilityProcessHost::Create(nullptr, nullptr); |
34 host->SetName(base::ASCIIToUTF16("TestProcess")); | 34 host->SetName(base::ASCIIToUTF16("TestProcess")); |
| 35 #if defined(OS_WIN) |
| 36 if (elevated) |
| 37 host->ElevatePrivileges(); |
| 38 #endif |
35 EXPECT_TRUE(host->Start()); | 39 EXPECT_TRUE(host->Start()); |
36 | 40 |
37 ServiceRegistry* service_registry = host->GetServiceRegistry(); | 41 ServiceRegistry* service_registry = host->GetServiceRegistry(); |
38 service_registry->ConnectToRemoteService(mojo::GetProxy(&service_)); | 42 service_registry->ConnectToRemoteService(mojo::GetProxy(&service_)); |
39 service_->DoSomething(base::Bind( | 43 service_->DoSomething(base::Bind( |
40 &UtilityProcessHostImplBrowserTest::OnSomething, | 44 &UtilityProcessHostImplBrowserTest::OnSomething, |
41 base::Unretained(this))); | 45 base::Unretained(this))); |
42 } | 46 } |
43 | 47 |
44 void OnSomething() { | 48 void OnSomething() { |
45 service_.reset(); | 49 service_.reset(); |
46 BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, done_closure_); | 50 BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, done_closure_); |
47 } | 51 } |
48 | 52 |
49 mojom::TestMojoServicePtr service_; | 53 mojom::TestMojoServicePtr service_; |
50 base::Closure done_closure_; | 54 base::Closure done_closure_; |
51 }; | 55 }; |
52 | 56 |
53 IN_PROC_BROWSER_TEST_F(UtilityProcessHostImplBrowserTest, LaunchProcess) { | 57 IN_PROC_BROWSER_TEST_F(UtilityProcessHostImplBrowserTest, LaunchProcess) { |
54 RunUtilityProcess(); | 58 RunUtilityProcess(false); |
55 } | 59 } |
56 | 60 |
| 61 #if defined(OS_WIN) |
| 62 IN_PROC_BROWSER_TEST_F(UtilityProcessHostImplBrowserTest, |
| 63 LaunchElevatedProcess) { |
| 64 RunUtilityProcess(true); |
| 65 } |
| 66 #endif |
| 67 |
57 } // namespace content | 68 } // namespace content |
OLD | NEW |