libcast_socket_service.h 4.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130
  1. // Copyright 2021 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 COMPONENTS_CAST_CHANNEL_LIBCAST_SOCKET_SERVICE_H_
  5. #define COMPONENTS_CAST_CHANNEL_LIBCAST_SOCKET_SERVICE_H_
  6. #include <map>
  7. #include <memory>
  8. #include "base/observer_list.h"
  9. #include "base/sequence_checker.h"
  10. #include "base/task/single_thread_task_runner.h"
  11. #include "base/time/time.h"
  12. #include "components/cast_channel/cast_socket.h"
  13. #include "components/cast_channel/cast_socket_service.h"
  14. #include "components/openscreen_platform/task_runner.h"
  15. #include "third_party/openscreen/src/cast/common/public/cast_socket.h"
  16. #include "third_party/openscreen/src/cast/sender/public/sender_socket_factory.h"
  17. #include "third_party/openscreen/src/platform/api/tls_connection_factory.h"
  18. namespace cast_channel {
  19. using LibcastSocket = openscreen::cast::CastSocket;
  20. class CastSocketWrapper;
  21. class LibcastSocketService final
  22. : public CastSocketService,
  23. public openscreen::cast::CastSocket::Client,
  24. public openscreen::cast::SenderSocketFactory::Client {
  25. public:
  26. using CastSocketService::NetworkContextGetter;
  27. LibcastSocketService(const LibcastSocketService&) = delete;
  28. LibcastSocketService& operator=(const LibcastSocketService&) = delete;
  29. ~LibcastSocketService() override;
  30. // CastSocketService overrides.
  31. std::unique_ptr<CastSocket> RemoveSocket(int channel_id) override;
  32. CastSocket* GetSocket(int channel_id) const override;
  33. CastSocket* GetSocket(const net::IPEndPoint& ip_endpoint) const override;
  34. void OpenSocket(NetworkContextGetter network_context_getter,
  35. const CastSocketOpenParams& open_params,
  36. CastSocket::OnOpenCallback open_cb) override;
  37. void AddObserver(CastSocket::Observer* observer) override;
  38. void RemoveObserver(CastSocket::Observer* observer) override;
  39. // openscreen::cast::CastSocket::Client overrides.
  40. void OnError(LibcastSocket* socket, openscreen::Error error) override;
  41. void OnMessage(LibcastSocket* socket,
  42. ::cast::channel::CastMessage message) override;
  43. // openscreen::cast::SenderSocketFactory::Client overrides.
  44. void OnConnected(openscreen::cast::SenderSocketFactory* factory,
  45. const openscreen::IPEndpoint& endpoint,
  46. std::unique_ptr<LibcastSocket> socket) override;
  47. void OnError(openscreen::cast::SenderSocketFactory* factory,
  48. const openscreen::IPEndpoint& endpoint,
  49. openscreen::Error error) override;
  50. void SetLibcastSocketForTest(std::unique_ptr<LibcastSocket> socket_for_test) {
  51. libcast_socket_for_test_ = std::move(socket_for_test);
  52. }
  53. private:
  54. friend class CastSocketService;
  55. friend class LibcastSocketServiceTest;
  56. struct ConnectTimer {
  57. ConnectTimer(std::unique_ptr<base::CancelableOnceClosure> callback,
  58. std::unique_ptr<base::OneShotTimer> timer);
  59. ConnectTimer(ConnectTimer&&);
  60. ~ConnectTimer();
  61. ConnectTimer& operator=(ConnectTimer&&);
  62. std::unique_ptr<base::CancelableOnceClosure> callback;
  63. std::unique_ptr<base::OneShotTimer> timer;
  64. };
  65. struct SavedOpenParams {
  66. base::TimeDelta ping_interval;
  67. base::TimeDelta liveness_timeout;
  68. };
  69. LibcastSocketService();
  70. bool EndpointPending(const net::IPEndPoint& ip_endpoint) const;
  71. void OnErrorSocketIOThread(LibcastSocket* socket, openscreen::Error error);
  72. void OnMessageIOThread(LibcastSocket* socket,
  73. ::cast::channel::CastMessage message);
  74. void OnConnectedIOThread(openscreen::cast::SenderSocketFactory* factory,
  75. const openscreen::IPEndpoint& endpoint,
  76. std::unique_ptr<LibcastSocket> socket);
  77. void OnErrorIOThread(openscreen::cast::SenderSocketFactory* factory,
  78. const openscreen::IPEndpoint& endpoint,
  79. openscreen::Error error);
  80. void OnErrorBounce(LibcastSocket* socket, ChannelError error);
  81. // Used to generate CastSocket IDs on error, since the socket factory doesn't
  82. // provide us one in that case.
  83. static int last_channel_id_;
  84. // List of socket observers.
  85. base::ObserverList<CastSocket::Observer>::Unchecked observers_;
  86. openscreen_platform::TaskRunner openscreen_task_runner_;
  87. openscreen::cast::SenderSocketFactory socket_factory_;
  88. std::unique_ptr<openscreen::TlsConnectionFactory> tls_factory_;
  89. std::map<int, std::unique_ptr<CastSocketWrapper>> sockets_;
  90. std::map<openscreen::IPEndpoint, int> socket_endpoints_;
  91. // Data for pending connections.
  92. std::map<openscreen::IPEndpoint, ConnectTimer> pending_endpoints_;
  93. std::map<openscreen::IPEndpoint, std::vector<CastSocket::OnOpenCallback>>
  94. open_callbacks_;
  95. std::map<openscreen::IPEndpoint, SavedOpenParams> open_params_;
  96. std::unique_ptr<LibcastSocket> libcast_socket_for_test_;
  97. };
  98. } // namespace cast_channel
  99. #endif // COMPONENTS_CAST_CHANNEL_LIBCAST_SOCKET_SERVICE_H_