chrome/browser/media/desktop_streams_registry.h

   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_