1 // Copyright 2013 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 #ifndef CHROME_BROWSER_MEDIA_DESKTOP_STREAMS_REGISTRY_H_ 6 #define CHROME_BROWSER_MEDIA_DESKTOP_STREAMS_REGISTRY_H_ 7 8 #include <map> 9 #include <string> 10 11 #include "chrome/browser/media/desktop_media_list.h" 12 #include "url/gurl.h" 13 14 // DesktopStreamsRegistry is used to store accepted desktop media streams for 15 // Desktop Capture API. Single instance of this class is created per browser in 16 // MediaCaptureDevicesDispatcher. 17 class DesktopStreamsRegistry { 18 public: 19 DesktopStreamsRegistry(); 20 ~DesktopStreamsRegistry(); 21 22 // Adds new stream to the registry. Called by the implementation of 23 // desktopCapture.chooseDesktopMedia() API after user has approved access to 24 // |source| for the |origin|. Returns identifier of the new stream. 25 std::string RegisterStream(int render_process_id, 26 int render_view_id, 27 const GURL& origin, 28 const content::DesktopMediaID& source, 29 const std::string& extension_name); 30 31 // Validates stream identifier specified in getUserMedia(). Returns null 32 // DesktopMediaID if the specified |id| is invalid, i.e. wasn't generated 33 // using RegisterStream() or if it was generated for a different origin. 34 // Otherwise returns ID of the source and removes it from the registry. 35 content::DesktopMediaID RequestMediaForStreamId(const std::string& id, 36 int render_process_id, 37 int render_view_id, 38 const GURL& origin, 39 std::string* extension_name); 40 41 private: 42 // Type used to store list of accepted desktop media streams. 43 struct ApprovedDesktopMediaStream { 44 ApprovedDesktopMediaStream(); 45 46 int render_process_id; 47 int render_view_id; 48 GURL origin; 49 content::DesktopMediaID source; 50 std::string extension_name; 51 }; 52 typedef std::map<std::string, ApprovedDesktopMediaStream> StreamsMap; 53 54 // Helper function that removes an expired stream from the registry. 55 void CleanupStream(const std::string& id); 56 57 StreamsMap approved_streams_; 58 59 DISALLOW_COPY_AND_ASSIGN(DesktopStreamsRegistry); 60 }; 61 62 #endif // CHROME_BROWSER_MEDIA_DESKTOP_STREAMS_REGISTRY_H_ 63