webrtc_connection_to_client.h 4.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112
  1. // Copyright 2015 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. #ifndef REMOTING_PROTOCOL_WEBRTC_CONNECTION_TO_CLIENT_H_
  5. #define REMOTING_PROTOCOL_WEBRTC_CONNECTION_TO_CLIENT_H_
  6. #include <stdint.h>
  7. #include <memory>
  8. #include <string>
  9. #include "base/memory/raw_ptr.h"
  10. #include "base/memory/weak_ptr.h"
  11. #include "base/task/single_thread_task_runner.h"
  12. #include "base/threading/thread_checker.h"
  13. #include "remoting/protocol/channel_dispatcher_base.h"
  14. #include "remoting/protocol/connection_to_client.h"
  15. #include "remoting/protocol/host_video_stats_dispatcher.h"
  16. #include "remoting/protocol/session.h"
  17. #include "remoting/protocol/webrtc_transport.h"
  18. namespace remoting {
  19. namespace protocol {
  20. class WebrtcVideoEncoderFactory;
  21. class HostControlDispatcher;
  22. class HostEventDispatcher;
  23. class WebrtcConnectionToClient : public ConnectionToClient,
  24. public Session::EventHandler,
  25. public WebrtcTransport::EventHandler,
  26. public ChannelDispatcherBase::EventHandler {
  27. public:
  28. WebrtcConnectionToClient(
  29. std::unique_ptr<Session> session,
  30. scoped_refptr<protocol::TransportContext> transport_context,
  31. scoped_refptr<base::SingleThreadTaskRunner> audio_task_runner);
  32. WebrtcConnectionToClient(const WebrtcConnectionToClient&) = delete;
  33. WebrtcConnectionToClient& operator=(const WebrtcConnectionToClient&) = delete;
  34. ~WebrtcConnectionToClient() override;
  35. // ConnectionToClient interface.
  36. void SetEventHandler(
  37. ConnectionToClient::EventHandler* event_handler) override;
  38. Session* session() override;
  39. void Disconnect(ErrorCode error) override;
  40. std::unique_ptr<VideoStream> StartVideoStream(
  41. const std::string& stream_name,
  42. std::unique_ptr<webrtc::DesktopCapturer> desktop_capturer) override;
  43. std::unique_ptr<AudioStream> StartAudioStream(
  44. std::unique_ptr<AudioSource> audio_source) override;
  45. ClientStub* client_stub() override;
  46. void set_clipboard_stub(ClipboardStub* clipboard_stub) override;
  47. void set_host_stub(HostStub* host_stub) override;
  48. void set_input_stub(InputStub* input_stub) override;
  49. void ApplySessionOptions(const SessionOptions& options) override;
  50. PeerConnectionControls* peer_connection_controls() override;
  51. WebrtcEventLogData* rtc_event_log() override;
  52. // Session::EventHandler interface.
  53. void OnSessionStateChange(Session::State state) override;
  54. // WebrtcTransport::EventHandler interface
  55. void OnWebrtcTransportConnecting() override;
  56. void OnWebrtcTransportConnected() override;
  57. void OnWebrtcTransportError(ErrorCode error) override;
  58. void OnWebrtcTransportProtocolChanged() override;
  59. void OnWebrtcTransportIncomingDataChannel(
  60. const std::string& name,
  61. std::unique_ptr<MessagePipe> pipe) override;
  62. void OnWebrtcTransportMediaStreamAdded(
  63. rtc::scoped_refptr<webrtc::MediaStreamInterface> stream) override;
  64. void OnWebrtcTransportMediaStreamRemoved(
  65. rtc::scoped_refptr<webrtc::MediaStreamInterface> stream) override;
  66. void OnWebrtcTransportRouteChanged(const TransportRoute& route) override;
  67. // ChannelDispatcherBase::EventHandler interface.
  68. void OnChannelInitialized(ChannelDispatcherBase* channel_dispatcher) override;
  69. void OnChannelClosed(ChannelDispatcherBase* channel_dispatcher) override;
  70. private:
  71. bool allChannelsConnected();
  72. base::ThreadChecker thread_checker_;
  73. // Event handler for handling events sent from this object.
  74. raw_ptr<ConnectionToClient::EventHandler> event_handler_ = nullptr;
  75. std::unique_ptr<WebrtcTransport> transport_;
  76. std::unique_ptr<Session> session_;
  77. raw_ptr<WebrtcVideoEncoderFactory> video_encoder_factory_;
  78. HostVideoStatsDispatcher video_stats_dispatcher_;
  79. scoped_refptr<base::SingleThreadTaskRunner> audio_task_runner_;
  80. SessionOptions session_options_;
  81. std::unique_ptr<HostControlDispatcher> control_dispatcher_;
  82. std::unique_ptr<HostEventDispatcher> event_dispatcher_;
  83. base::WeakPtrFactory<WebrtcConnectionToClient> weak_factory_{this};
  84. };
  85. } // namespace protocol
  86. } // namespace remoting
  87. #endif // REMOTING_PROTOCOL_WEBRTC_CONNECTION_TO_CLIENT_H_