Home | History | Annotate | Download | only in media 
      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   // |render_frame_id| refers to the RenderFrame requesting the stream.
     26   std::string RegisterStream(int render_process_id,
     27                              int render_frame_id,
     28                              const GURL& origin,
     29                              const content::DesktopMediaID& source,
     30                              const std::string& extension_name);
     31 
     32   // Validates stream identifier specified in getUserMedia(). Returns null
     33   // DesktopMediaID if the specified |id| is invalid, i.e. wasn't generated
     34   // using RegisterStream() or if it was generated for a different
     35   // RenderFrame/origin. Otherwise returns ID of the source and removes it from
     36   // the registry.
     37   content::DesktopMediaID RequestMediaForStreamId(const std::string& id,
     38                                                   int render_process_id,
     39                                                   int render_frame_id,
     40                                                   const GURL& origin,
     41                                                   std::string* extension_name);
     42 
     43  private:
     44   // Type used to store list of accepted desktop media streams.
     45   struct ApprovedDesktopMediaStream {
     46     ApprovedDesktopMediaStream();
     47 
     48     int render_process_id;
     49     int render_frame_id;
     50     GURL origin;
     51     content::DesktopMediaID source;
     52     std::string extension_name;
     53   };
     54   typedef std::map<std::string, ApprovedDesktopMediaStream> StreamsMap;
     55 
     56   // Helper function that removes an expired stream from the registry.
     57   void CleanupStream(const std::string& id);
     58 
     59   StreamsMap approved_streams_;
     60 
     61   DISALLOW_COPY_AND_ASSIGN(DesktopStreamsRegistry);
     62 };
     63 
     64 #endif  // CHROME_BROWSER_MEDIA_DESKTOP_STREAMS_REGISTRY_H_
     65