1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63

content / browser / media / capture / desktop_streams_registry_impl.h [blame]

// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CONTENT_BROWSER_MEDIA_CAPTURE_DESKTOP_STREAMS_REGISTRY_IMPL_H_
#define CONTENT_BROWSER_MEDIA_CAPTURE_DESKTOP_STREAMS_REGISTRY_IMPL_H_

#include <map>
#include <string>

#include "content/public/browser/child_process_host.h"
#include "content/public/browser/desktop_media_id.h"
#include "content/public/browser/desktop_streams_registry.h"
#include "url/origin.h"

namespace content {

class DesktopStreamsRegistryImpl : public DesktopStreamsRegistry {
 public:
  DesktopStreamsRegistryImpl();

  DesktopStreamsRegistryImpl(const DesktopStreamsRegistryImpl&) = delete;
  DesktopStreamsRegistryImpl& operator=(const DesktopStreamsRegistryImpl&) =
      delete;

  ~DesktopStreamsRegistryImpl() override;

  // Returns the DesktopStreamRegistryImpl singleton.
  static DesktopStreamsRegistryImpl* GetInstance();

  // DesktopStreamsRegistry:
  std::string RegisterStream(int render_process_id,
                             std::optional<int> restrict_to_render_frame_id,
                             const url::Origin& origin,
                             const DesktopMediaID& source,
                             const DesktopStreamRegistryType type) override;
  DesktopMediaID RequestMediaForStreamId(
      const std::string& id,
      int render_process_id,
      int render_frame_id,
      const url::Origin& origin,
      const DesktopStreamRegistryType type) override;

 private:
  // Type used to store list of accepted desktop media streams.
  struct ApprovedDesktopMediaStream {
    int render_process_id = content::ChildProcessHost::kInvalidUniqueID;
    std::optional<int> restrict_to_render_frame_id;
    url::Origin origin;
    DesktopMediaID source;
    DesktopStreamRegistryType type;
  };
  typedef std::map<std::string, ApprovedDesktopMediaStream> StreamsMap;

  // Helper function that removes an expired stream from the registry.
  void CleanupStream(const std::string& id);

  StreamsMap approved_streams_;
};

}  // namespace content

#endif  // CONTENT_BROWSER_MEDIA_CAPTURE_DESKTOP_STREAMS_REGISTRY_IMPL_H_