webrtc_connection_to_host.h 4.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122
  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_HOST_H_
  5. #define REMOTING_PROTOCOL_WEBRTC_CONNECTION_TO_HOST_H_
  6. #include <memory>
  7. #include <string>
  8. #include "base/memory/raw_ptr.h"
  9. #include "base/task/single_thread_task_runner.h"
  10. #include "remoting/protocol/channel_dispatcher_base.h"
  11. #include "remoting/protocol/clipboard_filter.h"
  12. #include "remoting/protocol/connection_to_host.h"
  13. #include "remoting/protocol/errors.h"
  14. #include "remoting/protocol/input_filter.h"
  15. #include "remoting/protocol/session.h"
  16. #include "remoting/protocol/webrtc_transport.h"
  17. namespace remoting {
  18. namespace protocol {
  19. class ClientControlDispatcher;
  20. class ClientEventDispatcher;
  21. class SessionConfig;
  22. class WebrtcVideoRendererAdapter;
  23. class WebrtcAudioSinkAdapter;
  24. class WebrtcConnectionToHost : public ConnectionToHost,
  25. public Session::EventHandler,
  26. public WebrtcTransport::EventHandler,
  27. public ChannelDispatcherBase::EventHandler {
  28. public:
  29. WebrtcConnectionToHost();
  30. WebrtcConnectionToHost(const WebrtcConnectionToHost&) = delete;
  31. WebrtcConnectionToHost& operator=(const WebrtcConnectionToHost&) = delete;
  32. ~WebrtcConnectionToHost() override;
  33. // ConnectionToHost interface.
  34. void set_client_stub(ClientStub* client_stub) override;
  35. void set_clipboard_stub(ClipboardStub* clipboard_stub) override;
  36. void set_video_renderer(VideoRenderer* video_renderer) override;
  37. void InitializeAudio(
  38. scoped_refptr<base::SingleThreadTaskRunner> audio_decode_task_runner,
  39. base::WeakPtr<AudioStub> audio_consumer) override;
  40. void Connect(std::unique_ptr<Session> session,
  41. scoped_refptr<TransportContext> transport_context,
  42. HostEventCallback* event_callback) override;
  43. void Disconnect(ErrorCode error) override;
  44. const SessionConfig& config() override;
  45. ClipboardStub* clipboard_forwarder() override;
  46. HostStub* host_stub() override;
  47. InputStub* input_stub() override;
  48. State state() const override;
  49. private:
  50. // Session::EventHandler interface.
  51. void OnSessionStateChange(Session::State state) override;
  52. // WebrtcTransport::EventHandler interface.
  53. void OnWebrtcTransportConnecting() override;
  54. void OnWebrtcTransportConnected() override;
  55. void OnWebrtcTransportError(ErrorCode error) override;
  56. void OnWebrtcTransportProtocolChanged() override;
  57. void OnWebrtcTransportIncomingDataChannel(
  58. const std::string& name,
  59. std::unique_ptr<MessagePipe> pipe) override;
  60. void OnWebrtcTransportMediaStreamAdded(
  61. rtc::scoped_refptr<webrtc::MediaStreamInterface> stream) override;
  62. void OnWebrtcTransportMediaStreamRemoved(
  63. rtc::scoped_refptr<webrtc::MediaStreamInterface> stream) override;
  64. void OnWebrtcTransportRouteChanged(const TransportRoute& route) override;
  65. // ChannelDispatcherBase::EventHandler interface.
  66. void OnChannelInitialized(ChannelDispatcherBase* channel_dispatcher) override;
  67. void OnChannelClosed(ChannelDispatcherBase* channel_dispatcher) override;
  68. void NotifyIfChannelsReady();
  69. WebrtcVideoRendererAdapter* GetOrCreateVideoAdapter(const std::string& label);
  70. void CloseChannels();
  71. void OnFrameRendered(uint32_t frame_id,
  72. base::TimeTicks event_timestamp,
  73. base::TimeTicks frame_rendered_time);
  74. void SetState(State state, ErrorCode error);
  75. raw_ptr<HostEventCallback> event_callback_ = nullptr;
  76. scoped_refptr<base::SingleThreadTaskRunner> audio_decode_task_runner_;
  77. // Stub for incoming messages.
  78. raw_ptr<ClientStub> client_stub_ = nullptr;
  79. raw_ptr<VideoRenderer> video_renderer_ = nullptr;
  80. base::WeakPtr<AudioStub> audio_consumer_;
  81. raw_ptr<ClipboardStub> clipboard_stub_ = nullptr;
  82. std::unique_ptr<Session> session_;
  83. std::unique_ptr<WebrtcTransport> transport_;
  84. std::unique_ptr<ClientControlDispatcher> control_dispatcher_;
  85. std::unique_ptr<ClientEventDispatcher> event_dispatcher_;
  86. ClipboardFilter clipboard_forwarder_;
  87. InputFilter event_forwarder_;
  88. std::unique_ptr<WebrtcVideoRendererAdapter> video_adapter_;
  89. std::unique_ptr<WebrtcAudioSinkAdapter> audio_adapter_;
  90. // Internal state of the connection.
  91. State state_ = INITIALIZING;
  92. ErrorCode error_ = OK;
  93. };
  94. } // namespace protocol
  95. } // namespace remoting
  96. #endif // REMOTING_PROTOCOL_WEBRTC_CONNECTION_TO_HOST_H_