src/jingle/glue/fake_socket_factory.h

   1 // Copyright (c) 2011 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 JINGLE_GLUE_FAKE_SOCKET_FACTORY_H_
   6 #define JINGLE_GLUE_FAKE_SOCKET_FACTORY_H_
   7
   8 #include <map>
   9 #include <vector>
  10
  11 #include "base/basictypes.h"
  12 #include "base/compiler_specific.h"
  13 #include "base/memory/ref_counted.h"
  14 #include "base/threading/non_thread_safe.h"
  15 #include "net/base/ip_endpoint.h"
  16 #include "third_party/libjingle/source/talk/base/asyncpacketsocket.h"
  17 #include "third_party/libjingle/source/talk/p2p/base/packetsocketfactory.h"
  18
  19 namespace base {
  20 class MessageLoop;
  21 }
  22
  23 namespace jingle_glue {
  24
  25 class FakeSocketManager;
  26
  27 class FakeUDPPacketSocket : public talk_base::AsyncPacketSocket,
  28                             public base::NonThreadSafe {
  29  public:
  30   FakeUDPPacketSocket(FakeSocketManager* fake_socket_manager,
  31                       const net::IPEndPoint& address);
  32   virtual ~FakeUDPPacketSocket();
  33
  34   const net::IPEndPoint& endpoint() const { return endpoint_; }
  35   void DeliverPacket(const net::IPEndPoint& from,
  36                      const std::vector<char>& data);
  37
  38   // talk_base::AsyncPacketSocket implementation.
  39   virtual talk_base::SocketAddress GetLocalAddress() const OVERRIDE;
  40   virtual talk_base::SocketAddress GetRemoteAddress() const OVERRIDE;
  41   virtual int Send(const void *pv, size_t cb,
  42                    const talk_base::PacketOptions& options) OVERRIDE;
  43   virtual int SendTo(const void *pv, size_t cb,
  44                      const talk_base::SocketAddress& addr,
  45                      const talk_base::PacketOptions& options) OVERRIDE;
  46   virtual int Close() OVERRIDE;
  47   virtual State GetState() const OVERRIDE;
  48   virtual int GetOption(talk_base::Socket::Option opt, int* value) OVERRIDE;
  49   virtual int SetOption(talk_base::Socket::Option opt, int value) OVERRIDE;
  50   virtual int GetError() const OVERRIDE;
  51   virtual void SetError(int error) OVERRIDE;
  52
  53  private:
  54   enum InternalState {
  55     IS_OPEN,
  56     IS_CLOSED,
  57   };
  58
  59   scoped_refptr<FakeSocketManager> fake_socket_manager_;
  60   net::IPEndPoint endpoint_;
  61   talk_base::SocketAddress local_address_;
  62   talk_base::SocketAddress remote_address_;
  63   InternalState state_;
  64   int error_;
  65
  66   DISALLOW_COPY_AND_ASSIGN(FakeUDPPacketSocket);
  67 };
  68
  69 class FakeSocketManager : public base::RefCountedThreadSafe<FakeSocketManager> {
  70  public:
  71   FakeSocketManager();
  72
  73   void SendPacket(const net::IPEndPoint& from,
  74                   const net::IPEndPoint& to,
  75                   const std::vector<char>& data);
  76
  77   void AddSocket(FakeUDPPacketSocket* socket_factory);
  78   void RemoveSocket(FakeUDPPacketSocket* socket_factory);
  79
  80  private:
  81   friend class base::RefCountedThreadSafe<FakeSocketManager>;
  82
  83   ~FakeSocketManager();
  84
  85   void DeliverPacket(const net::IPEndPoint& from,
  86                      const net::IPEndPoint& to,
  87                      const std::vector<char>& data);
  88
  89   base::MessageLoop* message_loop_;
  90   std::map<net::IPEndPoint, FakeUDPPacketSocket*> endpoints_;
  91
  92   DISALLOW_COPY_AND_ASSIGN(FakeSocketManager);
  93 };
  94
  95 class FakeSocketFactory : public talk_base::PacketSocketFactory {
  96  public:
  97   FakeSocketFactory(FakeSocketManager* socket_manager,
  98                     const net::IPAddressNumber& address);
  99   virtual ~FakeSocketFactory();
 100
 101   // talk_base::PacketSocketFactory implementation.
 102   virtual talk_base::AsyncPacketSocket* CreateUdpSocket(
 103       const talk_base::SocketAddress& local_address,
 104       int min_port, int max_port) OVERRIDE;
 105   virtual talk_base::AsyncPacketSocket* CreateServerTcpSocket(
 106       const talk_base::SocketAddress& local_address, int min_port, int max_port,
 107       int opts) OVERRIDE;
 108   virtual talk_base::AsyncPacketSocket* CreateClientTcpSocket(
 109       const talk_base::SocketAddress& local_address,
 110       const talk_base::SocketAddress& remote_address,
 111       const talk_base::ProxyInfo& proxy_info,
 112       const std::string& user_agent,
 113       int opts) OVERRIDE;
 114
 115  private:
 116   scoped_refptr<FakeSocketManager> socket_manager_;
 117   net::IPAddressNumber address_;
 118   int last_allocated_port_;
 119
 120   DISALLOW_COPY_AND_ASSIGN(FakeSocketFactory);
 121 };
 122
 123 }  // namespace jingle_glue
 124
 125 #endif  // JINGLE_GLUE_FAKE_SOCKET_FACTORY_H_