remoting/protocol/fake_datagram_socket.h

   1 // Copyright 2014 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 REMOTING_PROTOCOL_FAKE_DATAGRAM_SOCKET_H_
   6 #define REMOTING_PROTOCOL_FAKE_DATAGRAM_SOCKET_H_
   7
   8 #include <map>
   9 #include <string>
  10 #include <vector>
  11
  12 #include "base/memory/scoped_ptr.h"
  13 #include "base/memory/weak_ptr.h"
  14 #include "net/base/completion_callback.h"
  15 #include "remoting/protocol/datagram_channel_factory.h"
  16 #include "remoting/protocol/p2p_datagram_socket.h"
  17
  18 namespace base {
  19 class SingleThreadTaskRunner;
  20 }
  21
  22 namespace remoting {
  23 namespace protocol {
  24
  25 // FakeDatagramSocket implement P2PStreamSocket interface. All data written to
  26 // FakeDatagramSocket is stored in a buffer returned by written_packets().
  27 // Read() reads data from another buffer that can be set with
  28 // AppendInputPacket(). Pending reads are supported, so if there is a pending
  29 // read AppendInputPacket() calls the read callback.
  30 //
  31 // Two fake sockets can be connected to each other using the
  32 // PairWith() method, e.g.: a->PairWith(b). After this all data
  33 // written to |a| can be read from |b| and vice versa. Two connected
  34 // sockets |a| and |b| must be created and used on the same thread.
  35 class FakeDatagramSocket : public P2PDatagramSocket {
  36  public:
  37   FakeDatagramSocket();
  38   ~FakeDatagramSocket() override;
  39
  40   const std::vector<std::string>& written_packets() const {
  41     return written_packets_;
  42   }
  43
  44   void AppendInputPacket(const std::string& data);
  45
  46   // Current position in the input in number of packets, i.e. number of finished
  47   // Recv() calls.
  48   int input_pos() const { return input_pos_; }
  49
  50   // Pairs the socket with |peer_socket|. Deleting either of the paired sockets
  51   // unpairs them.
  52   void PairWith(FakeDatagramSocket* peer_socket);
  53
  54   base::WeakPtr<FakeDatagramSocket> GetWeakPtr();
  55
  56   // P2PDatagramSocket implementation.
  57   int Recv(const scoped_refptr<net::IOBuffer>& buf, int buf_len,
  58            const net::CompletionCallback& callback) override;
  59   int Send(const scoped_refptr<net::IOBuffer>& buf, int buf_len,
  60            const net::CompletionCallback& callback) override;
  61
  62  private:
  63   int CopyReadData(const scoped_refptr<net::IOBuffer>& buf, int buf_len);
  64
  65   base::WeakPtr<FakeDatagramSocket> peer_socket_;
  66
  67   scoped_refptr<net::IOBuffer> read_buffer_;
  68   int read_buffer_size_;
  69   net::CompletionCallback read_callback_;
  70
  71   std::vector<std::string> written_packets_;
  72   std::vector<std::string> input_packets_;
  73   int input_pos_;
  74
  75   scoped_refptr<base::SingleThreadTaskRunner> task_runner_;
  76   base::WeakPtrFactory<FakeDatagramSocket> weak_factory_;
  77
  78   DISALLOW_COPY_AND_ASSIGN(FakeDatagramSocket);
  79 };
  80
  81 class FakeDatagramChannelFactory : public DatagramChannelFactory {
  82  public:
  83   FakeDatagramChannelFactory();
  84   ~FakeDatagramChannelFactory() override;
  85
  86   void set_asynchronous_create(bool asynchronous_create) {
  87     asynchronous_create_ = asynchronous_create;
  88   }
  89
  90   void set_fail_create(bool fail_create) { fail_create_ = fail_create; }
  91
  92   // Pair with |peer_factory|. Once paired the factory will be automatically
  93   // pairing created sockets with the sockets with the same name from the peer
  94   // factory.
  95   void PairWith(FakeDatagramChannelFactory* peer_factory);
  96
  97   // Can be used to retrieve FakeDatagramSocket created by this factory, e.g. to
  98   // feed data into it. The caller doesn't get ownership of the result. Returns
  99   // nullptr if the socket doesn't exist.
 100   FakeDatagramSocket* GetFakeChannel(const std::string& name);
 101
 102   // DatagramChannelFactory interface.
 103   void CreateChannel(const std::string& name,
 104                      const ChannelCreatedCallback& callback) override;
 105   void CancelChannelCreation(const std::string& name) override;
 106
 107  private:
 108   typedef std::map<std::string, base::WeakPtr<FakeDatagramSocket> > ChannelsMap;
 109
 110   void NotifyChannelCreated(scoped_ptr<FakeDatagramSocket> owned_socket,
 111                             const std::string& name,
 112                             const ChannelCreatedCallback& callback);
 113
 114   base::WeakPtr<FakeDatagramChannelFactory> peer_factory_;
 115
 116   scoped_refptr<base::SingleThreadTaskRunner> task_runner_;
 117   bool asynchronous_create_;
 118   ChannelsMap channels_;
 119
 120   bool fail_create_;
 121
 122   base::WeakPtrFactory<FakeDatagramChannelFactory> weak_factory_;
 123
 124   DISALLOW_COPY_AND_ASSIGN(FakeDatagramChannelFactory);
 125 };
 126
 127 }  // namespace protocol
 128 }  // namespace remoting
 129
 130 #endif  // REMOTING_PROTOCOL_FAKE_DATAGRAM_SOCKET_H_