1 // Copyright 2015 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 <utility>
8
9 #include "chrome/browser/extensions/api/tab_capture/tab_capture_registry.h"
10 #include "chrome/browser/media/webrtc/media_capture_devices_dispatcher.h"
11 #include "chrome/browser/media/webrtc/media_stream_capture_indicator.h"
12 #include "chrome/browser/profiles/profile.h"
13 #include "chrome/common/pref_names.h"
14 #include "components/prefs/pref_service.h"
15 #include "content/public/browser/desktop_media_id.h"
16 #include "content/public/browser/web_contents.h"
17 #include "content/public/browser/web_contents_media_capture_id.h"
18 #include "extensions/common/permissions/permissions_data.h"
19 #include "third_party/blink/public/mojom/mediastream/media_stream.mojom-shared.h"
20
21 #if defined(OS_CHROMEOS)
22 #include "chrome/browser/chromeos/policy/dlp/dlp_content_manager.h"
23 #endif // defined(OS_CHROMEOS)
24
25 TabCaptureAccessHandler::TabCaptureAccessHandler() = default;
26
27 TabCaptureAccessHandler::~TabCaptureAccessHandler() = default;
28
SupportsStreamType(content::WebContents * web_contents,const blink::mojom::MediaStreamType type,const extensions::Extension * extension)29 bool TabCaptureAccessHandler::SupportsStreamType(
30 content::WebContents* web_contents,
31 const blink::mojom::MediaStreamType type,
32 const extensions::Extension* extension) {
33 return type == blink::mojom::MediaStreamType::GUM_TAB_VIDEO_CAPTURE ||
34 type == blink::mojom::MediaStreamType::GUM_TAB_AUDIO_CAPTURE;
35 }
36
CheckMediaAccessPermission(content::RenderFrameHost * render_frame_host,const GURL & security_origin,blink::mojom::MediaStreamType type,const extensions::Extension * extension)37 bool TabCaptureAccessHandler::CheckMediaAccessPermission(
38 content::RenderFrameHost* render_frame_host,
39 const GURL& security_origin,
40 blink::mojom::MediaStreamType type,
41 const extensions::Extension* extension) {
42 return false;
43 }
44
HandleRequest(content::WebContents * web_contents,const content::MediaStreamRequest & request,content::MediaResponseCallback callback,const extensions::Extension * extension)45 void TabCaptureAccessHandler::HandleRequest(
46 content::WebContents* web_contents,
47 const content::MediaStreamRequest& request,
48 content::MediaResponseCallback callback,
49 const extensions::Extension* extension) {
50 blink::MediaStreamDevices devices;
51 std::unique_ptr<content::MediaStreamUI> ui;
52
53 Profile* profile =
54 Profile::FromBrowserContext(web_contents->GetBrowserContext());
55 extensions::TabCaptureRegistry* tab_capture_registry =
56 extensions::TabCaptureRegistry::Get(profile);
57 if (!tab_capture_registry) {
58 NOTREACHED();
59 std::move(callback).Run(
60 devices, blink::mojom::MediaStreamRequestResult::INVALID_STATE,
61 std::move(ui));
62 return;
63 }
64 if (!profile->GetPrefs()->GetBoolean(prefs::kScreenCaptureAllowed)) {
65 std::move(callback).Run(
66 devices, blink::mojom::MediaStreamRequestResult::PERMISSION_DENIED,
67 std::move(ui));
68 return;
69 }
70
71 #if defined(OS_CHROMEOS)
72 if (request.video_type ==
73 blink::mojom::MediaStreamType::GUM_TAB_VIDEO_CAPTURE) {
74 content::DesktopMediaID media_id(
75 content::DesktopMediaID::TYPE_WEB_CONTENTS, /*id=*/0,
76 content::WebContentsMediaCaptureId(request.render_process_id,
77 request.render_frame_id));
78 if (policy::DlpContentManager::Get()->IsScreenCaptureRestricted(media_id)) {
79 std::move(callback).Run(
80 devices, blink::mojom::MediaStreamRequestResult::PERMISSION_DENIED,
81 std::move(ui));
82 return;
83 }
84 }
85 #endif
86
87 // |extension| may be null if the tabCapture starts with
88 // tabCapture.getMediaStreamId().
89 // TODO(crbug.com/831722): Deprecate tabCaptureRegistry soon.
90 const std::string extension_id = extension ? extension->id() : "";
91 const bool tab_capture_allowed = tab_capture_registry->VerifyRequest(
92 request.render_process_id, request.render_frame_id, extension_id);
93
94 if (request.audio_type ==
95 blink::mojom::MediaStreamType::GUM_TAB_AUDIO_CAPTURE &&
96 tab_capture_allowed) {
97 devices.push_back(blink::MediaStreamDevice(
98 blink::mojom::MediaStreamType::GUM_TAB_AUDIO_CAPTURE, std::string(),
99 std::string()));
100 }
101
102 if (request.video_type ==
103 blink::mojom::MediaStreamType::GUM_TAB_VIDEO_CAPTURE &&
104 tab_capture_allowed) {
105 devices.push_back(blink::MediaStreamDevice(
106 blink::mojom::MediaStreamType::GUM_TAB_VIDEO_CAPTURE, std::string(),
107 std::string()));
108 }
109
110 if (!devices.empty()) {
111 ui = MediaCaptureDevicesDispatcher::GetInstance()
112 ->GetMediaStreamCaptureIndicator()
113 ->RegisterMediaStream(web_contents, devices);
114 }
115 UpdateExtensionTrusted(request, extension);
116 std::move(callback).Run(
117 devices,
118 devices.empty() ? blink::mojom::MediaStreamRequestResult::INVALID_STATE
119 : blink::mojom::MediaStreamRequestResult::OK,
120 std::move(ui));
121 }
122