| 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 |