1 // Copyright 2020 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "chrome/browser/media/webrtc/tab_capture_access_handler.h"
6
7 #include <memory>
8 #include <string>
9 #include <utility>
10
11 #include "base/bind.h"
12 #include "base/macros.h"
13 #include "base/run_loop.h"
14 #include "build/build_config.h"
15 #include "chrome/browser/extensions/api/tab_capture/tab_capture_registry.h"
16 #include "chrome/browser/media/webrtc/fake_desktop_media_picker_factory.h"
17 #include "chrome/test/base/chrome_render_view_host_test_harness.h"
18 #include "content/public/browser/browser_context.h"
19 #include "content/public/browser/desktop_media_id.h"
20 #include "content/public/browser/render_process_host.h"
21 #include "content/public/browser/web_contents.h"
22 #include "testing/gtest/include/gtest/gtest.h"
23 #include "third_party/blink/public/common/mediastream/media_stream_request.h"
24 #include "third_party/blink/public/mojom/mediastream/media_stream.mojom-shared.h"
25
26 #if defined(OS_CHROMEOS)
27 #include "chrome/browser/chromeos/policy/dlp/mock_dlp_content_manager.h"
28 #endif
29
30 class TabCaptureAccessHandlerTest : public ChromeRenderViewHostTestHarness {
31 public:
32 TabCaptureAccessHandlerTest() = default;
33 ~TabCaptureAccessHandlerTest() override = default;
34
SetUp()35 void SetUp() override {
36 ChromeRenderViewHostTestHarness::SetUp();
37 access_handler_ = std::make_unique<TabCaptureAccessHandler>();
38 }
39
ProcessRequest(const content::DesktopMediaID & fake_desktop_media_id_response,blink::mojom::MediaStreamRequestResult * request_result,blink::MediaStreamDevices * devices_result)40 void ProcessRequest(
41 const content::DesktopMediaID& fake_desktop_media_id_response,
42 blink::mojom::MediaStreamRequestResult* request_result,
43 blink::MediaStreamDevices* devices_result) {
44 content::MediaStreamRequest request(
45 web_contents()->GetMainFrame()->GetProcess()->GetID(),
46 web_contents()->GetMainFrame()->GetRoutingID(), /*page_request_id=*/0,
47 GURL("http://origin/"), /*user_gesture=*/false,
48 blink::MEDIA_GENERATE_STREAM,
49 /*requested_audio_device_id=*/std::string(),
50 /*requested_video_device_id=*/std::string(),
51 blink::mojom::MediaStreamType::NO_SERVICE,
52 blink::mojom::MediaStreamType::GUM_TAB_VIDEO_CAPTURE,
53 /*disable_local_echo=*/false,
54 /*request_pan_tilt_zoom_permission=*/false);
55
56 base::RunLoop wait_loop;
57 content::MediaResponseCallback callback = base::BindOnce(
58 [](base::RunLoop* wait_loop,
59 blink::mojom::MediaStreamRequestResult* request_result,
60 blink::MediaStreamDevices* devices_result,
61 const blink::MediaStreamDevices& devices,
62 blink::mojom::MediaStreamRequestResult result,
63 std::unique_ptr<content::MediaStreamUI> ui) {
64 *request_result = result;
65 *devices_result = devices;
66 wait_loop->Quit();
67 },
68 &wait_loop, request_result, devices_result);
69 access_handler_->HandleRequest(web_contents(), request, std::move(callback),
70 /*extension=*/nullptr);
71 wait_loop.Run();
72
73 access_handler_.reset();
74 }
75
76 protected:
77 std::unique_ptr<TabCaptureAccessHandler> access_handler_;
78 };
79
TEST_F(TabCaptureAccessHandlerTest,PermissionGiven)80 TEST_F(TabCaptureAccessHandlerTest, PermissionGiven) {
81 const content::DesktopMediaID source(
82 content::DesktopMediaID::TYPE_WEB_CONTENTS,
83 content::DesktopMediaID::kNullId,
84 content::WebContentsMediaCaptureId(
85 web_contents()->GetMainFrame()->GetProcess()->GetID(),
86 web_contents()->GetMainFrame()->GetRoutingID()));
87
88 extensions::TabCaptureRegistry::Get(profile())->AddRequest(
89 web_contents(), /*extension_id=*/"", /*is_anonymous=*/false,
90 GURL("http://origin/"), source, /*extension_name=*/"", web_contents());
91
92 blink::mojom::MediaStreamRequestResult result;
93 blink::MediaStreamDevices devices;
94 ProcessRequest(source, &result, &devices);
95
96 EXPECT_EQ(blink::mojom::MediaStreamRequestResult::OK, result);
97 EXPECT_EQ(1u, devices.size());
98 EXPECT_EQ(blink::mojom::MediaStreamType::GUM_TAB_VIDEO_CAPTURE,
99 devices[0].type);
100 }
101
102 #if defined(OS_CHROMEOS)
TEST_F(TabCaptureAccessHandlerTest,DlpRestricted)103 TEST_F(TabCaptureAccessHandlerTest, DlpRestricted) {
104 const content::DesktopMediaID source(
105 content::DesktopMediaID::TYPE_WEB_CONTENTS,
106 content::DesktopMediaID::kNullId,
107 content::WebContentsMediaCaptureId(
108 web_contents()->GetMainFrame()->GetProcess()->GetID(),
109 web_contents()->GetMainFrame()->GetRoutingID()));
110
111 // Setup Data Leak Prevention restriction.
112 policy::MockDlpContentManager mock_dlp_content_manager;
113 policy::DlpContentManager::SetDlpContentManagerForTesting(
114 &mock_dlp_content_manager);
115 EXPECT_CALL(mock_dlp_content_manager, IsScreenCaptureRestricted(source))
116 .Times(1)
117 .WillOnce(testing::Return(true));
118
119 extensions::TabCaptureRegistry::Get(profile())->AddRequest(
120 web_contents(), /*extension_id=*/"", /*is_anonymous=*/false,
121 GURL("http://origin/"), source, /*extension_name=*/"", web_contents());
122
123 blink::mojom::MediaStreamRequestResult result;
124 blink::MediaStreamDevices devices;
125 ProcessRequest(source, &result, &devices);
126
127 EXPECT_EQ(blink::mojom::MediaStreamRequestResult::PERMISSION_DENIED, result);
128 EXPECT_EQ(0u, devices.size());
129 }
130 #endif
131