content/renderer/p2p/port_allocator.h

   1 // Copyright (c) 2012 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 CONTENT_RENDERER_P2P_PORT_ALLOCATOR_H_
   6 #define CONTENT_RENDERER_P2P_PORT_ALLOCATOR_H_
   7
   8 #include "third_party/webrtc/p2p/client/basicportallocator.h"
   9
  10 namespace content {
  11
  12 class P2PPortAllocatorSession;
  13 class P2PSocketDispatcher;
  14
  15 class P2PPortAllocator : public cricket::BasicPortAllocator {
  16  public:
  17   struct Config {
  18     Config();
  19     ~Config();
  20
  21     struct RelayServerConfig {
  22       RelayServerConfig();
  23       ~RelayServerConfig();
  24
  25       std::string username;
  26       std::string password;
  27       std::string server_address;
  28       int port;
  29       std::string transport_type;
  30       bool secure;
  31     };
  32
  33     std::set<rtc::SocketAddress> stun_servers;
  34
  35     std::vector<RelayServerConfig> relays;
  36
  37     // Disable TCP-based transport when set to true.
  38     bool disable_tcp_transport;
  39
  40     // Disable binding to individual NICs when set to false.
  41     bool enable_multiple_routes;
  42   };
  43
  44   P2PPortAllocator(P2PSocketDispatcher* socket_dispatcher,
  45                    rtc::NetworkManager* network_manager,
  46                    rtc::PacketSocketFactory* socket_factory,
  47                    const Config& config);
  48   ~P2PPortAllocator() override;
  49
  50   cricket::PortAllocatorSession* CreateSessionInternal(
  51       const std::string& content_name,
  52       int component,
  53       const std::string& ice_username_fragment,
  54       const std::string& ice_password) override;
  55
  56  private:
  57   friend class P2PPortAllocatorSession;
  58
  59   P2PSocketDispatcher* socket_dispatcher_;
  60   Config config_;
  61
  62   DISALLOW_COPY_AND_ASSIGN(P2PPortAllocator);
  63 };
  64
  65 class P2PPortAllocatorSession : public cricket::BasicPortAllocatorSession {
  66  public:
  67   P2PPortAllocatorSession(
  68       P2PPortAllocator* allocator,
  69       const std::string& content_name,
  70       int component,
  71       const std::string& ice_username_fragment,
  72       const std::string& ice_password);
  73   ~P2PPortAllocatorSession() override;
  74
  75  protected:
  76   // Overrides for cricket::BasicPortAllocatorSession.
  77   void GetPortConfigurations() override;
  78
  79  private:
  80   P2PPortAllocator* allocator_;
  81
  82   DISALLOW_COPY_AND_ASSIGN(P2PPortAllocatorSession);
  83 };
  84
  85 }  // namespace content
  86
  87 #endif  // CONTENT_RENDERER_P2P_PORT_ALLOCATOR_H_