15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Implements the Chrome Extensions Tab Capture API. 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/extensions/api/tab_capture/tab_capture_api.h" 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)#include <set> 10d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)#include <string> 11d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)#include <vector> 12d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) 13c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "base/command_line.h" 14868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/stringprintf.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/values.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/extensions/api/tab_capture/tab_capture_registry.h" 172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "chrome/browser/extensions/extension_renderer_state.h" 18c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "chrome/browser/profiles/profile.h" 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/sessions/session_tab_helper.h" 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/ui/browser.h" 212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "chrome/browser/ui/browser_finder.h" 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/ui/tabs/tab_strip_model.h" 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/browser/render_process_host.h" 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/browser/render_view_host.h" 253551c9c881056c480085172ff9840cab31610854Torne (Richard Coles)#include "extensions/common/features/feature.h" 26ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch#include "extensions/common/features/feature_provider.h" 27c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch#include "extensions/common/features/simple_feature.h" 28f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)#include "extensions/common/permissions/permissions_data.h" 29e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch#include "extensions/common/switches.h" 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)using extensions::api::tab_capture::MediaStreamConstraint; 322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace TabCapture = extensions::api::tab_capture; 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace GetCapturedTabs = TabCapture::GetCapturedTabs; 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace extensions { 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kCapturingSameTab[] = "Cannot capture a tab with an active stream."; 412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const char kFindingTabError[] = "Error finding tab to capture."; 422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const char kNoAudioOrVideo[] = "Capture failed. No audio or video requested."; 43c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)const char kGrantError[] = 44c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) "Extension has not been invoked for the current page (see activeTab " 45c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) "permission). Chrome pages cannot be captured."; 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Keys/values for media stream constraints. 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kMediaStreamSource[] = "chromeMediaSource"; 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kMediaStreamSourceId[] = "chromeMediaSourceId"; 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kMediaStreamSourceTab[] = "tab"; 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 52d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)// Whitelisted extensions that do not check for a browser action grant because 53d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)// they provide API's. 54d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)const char* whitelisted_extensions[] = { 55d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) "enhhojjnijigcajfphajepfemndkmdlo", // Dev 56d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) "pkedcjkdefgpdelpbcmbmeomcjbeemfm", // Trusted Tester 57d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) "fmfcbgogabcbclcofgocippekhfcmgfj", // Staging 58d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) "hfaagokkkhdbgiakmmlclaapfelnkoah", // Canary 59d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) "F155646B5D1CA545F7E1E4E20D573DFDD44C2540", // Trusted Tester (public) 60d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) "16CA7A47AAE4BE49B1E75A6B960C3875E945B264" // Release 61d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)}; 62d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 655c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liubool TabCaptureCaptureFunction::RunSync() { 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<api::tab_capture::Capture::Params> params = 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TabCapture::Capture::Params::Create(*args_); 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXTENSION_FUNCTION_VALIDATE(params.get()); 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Figure out the active WebContents and retrieve the needed ids. 711e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) Browser* target_browser = chrome::FindAnyBrowser( 721e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) GetProfile(), include_incognito(), chrome::GetActiveDesktop()); 732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!target_browser) { 742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) error_ = kFindingTabError; 752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) content::WebContents* target_contents = 792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) target_browser->tab_strip_model()->GetActiveWebContents(); 802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!target_contents) { 812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) error_ = kFindingTabError; 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 85c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const Extension* extension = GetExtension(); 86c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const std::string& extension_id = extension->id(); 87c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 88d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) const int tab_id = SessionID::IdForTab(target_contents); 89d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) 90c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Make sure either we have been granted permission to capture through an 91c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // extension icon click or our extension is whitelisted. 9246d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) if (!extension->permissions_data()->HasAPIPermissionForTab( 9346d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) tab_id, APIPermission::kTabCaptureForTab) && 94c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) CommandLine::ForCurrentProcess()->GetSwitchValueASCII( 95c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) switches::kWhitelistedExtensionID) != extension_id && 96010d83a9304c5a91596085d917d248abff47903aTorne (Richard Coles) !SimpleFeature::IsIdInList( 97d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) extension_id, 98d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) std::set<std::string>( 99d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) whitelisted_extensions, 100d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) whitelisted_extensions + arraysize(whitelisted_extensions)))) { 101c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) error_ = kGrantError; 102c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return false; 103c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 104c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 1052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) content::RenderViewHost* const rvh = target_contents->GetRenderViewHost(); 1062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int render_process_id = rvh->GetProcess()->GetID(); 1072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int routing_id = rvh->GetRoutingID(); 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Create a constraints vector. We will modify all the constraints in this 1102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // vector to append our chrome specific constraints. 1112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::vector<MediaStreamConstraint*> constraints; 1122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool has_audio = params->options.audio.get() && *params->options.audio.get(); 1132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool has_video = params->options.video.get() && *params->options.video.get(); 1142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!has_audio && !has_video) { 1162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) error_ = kNoAudioOrVideo; 1172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (has_audio) { 1212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!params->options.audio_constraints.get()) 1222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) params->options.audio_constraints.reset(new MediaStreamConstraint); 1232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) constraints.push_back(params->options.audio_constraints.get()); 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (has_video) { 1272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!params->options.video_constraints.get()) 1282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) params->options.video_constraints.reset(new MediaStreamConstraint); 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) constraints.push_back(params->options.video_constraints.get()); 1312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 1322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Device id we use for Tab Capture. 1342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::string device_id = 1352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::StringPrintf("%i:%i", render_process_id, routing_id); 1362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Append chrome specific tab constraints. 1382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for (std::vector<MediaStreamConstraint*>::iterator it = constraints.begin(); 1392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) it != constraints.end(); ++it) { 1402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::DictionaryValue* constraint = &(*it)->mandatory.additional_properties; 1412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) constraint->SetString(kMediaStreamSource, kMediaStreamSourceTab); 1422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) constraint->SetString(kMediaStreamSourceId, device_id); 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) extensions::TabCaptureRegistry* registry = 1461e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) extensions::TabCaptureRegistry::Get(GetProfile()); 1472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!registry->AddRequest(render_process_id, 1482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) routing_id, 149c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) extension_id, 1502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) tab_id, 1512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) tab_capture::TAB_CAPTURE_STATE_NONE)) { 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) error_ = kCapturingSameTab; 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Copy the result from our modified input parameters. This will be 1572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // intercepted by custom bindings which will build and send the special 1582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // WebRTC user media request. 1592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::DictionaryValue* result = new base::DictionaryValue(); 1602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) result->MergeDictionary(params->options.ToValue().get()); 1612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetResult(result); 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1665c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liubool TabCaptureGetCapturedTabsFunction::RunSync() { 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) extensions::TabCaptureRegistry* registry = 1681e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles) extensions::TabCaptureRegistry::Get(GetProfile()); 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const TabCaptureRegistry::RegistryCaptureInfo& captured_tabs = 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) registry->GetCapturedTabs(GetExtension()->id()); 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::ListValue *list = new base::ListValue(); 1742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for (TabCaptureRegistry::RegistryCaptureInfo::const_iterator it = 1752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) captured_tabs.begin(); it != captured_tabs.end(); ++it) { 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<tab_capture::CaptureInfo> info(new tab_capture::CaptureInfo()); 1772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) info->tab_id = it->first; 1782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) info->status = it->second; 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) list->Append(info->ToValue().release()); 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetResult(list); 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace extensions 187