Home | History | Annotate | Download | only in glue 
      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                    talk_base::DiffServCodePoint dscp) OVERRIDE;
     43   virtual int SendTo(const void *pv, size_t cb,
     44                      const talk_base::SocketAddress& addr,
     45                      talk_base::DiffServCodePoint dscp) 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_
    126