Add P2PDatagramSocket and P2PStreamSocket interfaces.

remoting/protocol/fake_datagram_socket.h

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef REMOTING_PROTOCOL_FAKE_DATAGRAM_SOCKET_H_
 #define REMOTING_PROTOCOL_FAKE_DATAGRAM_SOCKET_H_
 
 #include <map>
 #include <string>
 #include <vector>
 
 #include "base/memory/scoped_ptr.h"
 #include "base/memory/weak_ptr.h"
 #include "net/base/completion_callback.h"
-#include "net/socket/socket.h"
 #include "remoting/protocol/datagram_channel_factory.h"
+#include "remoting/protocol/p2p_datagram_socket.h"
 
 namespace base {
 class SingleThreadTaskRunner;
 }
 
 namespace remoting {
 namespace protocol {
 
-// FakeDatagramSocket implement net::StreamSocket interface. All data written to
+// FakeDatagramSocket implement P2PStreamSocket interface. All data written to
 // FakeDatagramSocket is stored in a buffer returned by written_packets().
 // Read() reads data from another buffer that can be set with
 // AppendInputPacket(). Pending reads are supported, so if there is a pending
 // read AppendInputPacket() calls the read callback.
 //
 // Two fake sockets can be connected to each other using the
 // PairWith() method, e.g.: a->PairWith(b). After this all data
 // written to |a| can be read from |b| and vice versa. Two connected
 // sockets |a| and |b| must be created and used on the same thread.
-class FakeDatagramSocket : public net::Socket {
+class FakeDatagramSocket : public P2PDatagramSocket {
  public:
   FakeDatagramSocket();
   ~FakeDatagramSocket() override;
 
   const std::vector<std::string>& written_packets() const {
     return written_packets_;
   }
 
   void AppendInputPacket(const std::string& data);
 
   // Current position in the input in number of packets, i.e. number of finished
-  // Read() calls.
+  // Recv() calls.
   int input_pos() const { return input_pos_; }
 
   // Pairs the socket with |peer_socket|. Deleting either of the paired sockets
   // unpairs them.
   void PairWith(FakeDatagramSocket* peer_socket);
 
   base::WeakPtr<FakeDatagramSocket> GetWeakPtr();
 
-  // net::Socket implementation.
-  int Read(net::IOBuffer* buf,
-           int buf_len,
+  // P2PDatagramSocket implementation.
+  int Recv(const scoped_refptr<net::IOBuffer>& buf, int buf_len,
            const net::CompletionCallback& callback) override;
-  int Write(net::IOBuffer* buf,
-            int buf_len,
-            const net::CompletionCallback& callback) override;
-  int SetReceiveBufferSize(int32 size) override;
-  int SetSendBufferSize(int32 size) override;
+  int Send(const scoped_refptr<net::IOBuffer>& buf, int buf_len,
+           const net::CompletionCallback& callback) override;
 
  private:
-  int CopyReadData(net::IOBuffer* buf, int buf_len);
+  int CopyReadData(const scoped_refptr<net::IOBuffer>& buf, int buf_len);
 
   base::WeakPtr<FakeDatagramSocket> peer_socket_;
 
   scoped_refptr<net::IOBuffer> read_buffer_;
   int read_buffer_size_;
   net::CompletionCallback read_callback_;
 
   std::vector<std::string> written_packets_;
   std::vector<std::string> input_packets_;
   int input_pos_;
@@ skipping 47 matching lines @@
 
   base::WeakPtrFactory<FakeDatagramChannelFactory> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(FakeDatagramChannelFactory);
 };
 
 }  // namespace protocol
 }  // namespace remoting
 
 #endif  // REMOTING_PROTOCOL_FAKE_DATAGRAM_SOCKET_H_