Add P2PDatagramSocket and P2PStreamSocket interfaces.

remoting/protocol/message_reader.cc

 // Copyright (c) 2012 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.
 
 #include "remoting/protocol/message_reader.h"
 
 #include "base/bind.h"
 #include "base/callback.h"
 #include "base/compiler_specific.h"
 #include "base/location.h"
 #include "base/thread_task_runner_handle.h"
 #include "base/single_thread_task_runner.h"
 #include "net/base/io_buffer.h"
 #include "net/base/net_errors.h"
-#include "net/socket/socket.h"
 #include "remoting/base/compound_buffer.h"
 #include "remoting/proto/internal.pb.h"
+#include "remoting/protocol/p2p_stream_socket.h"
 
 namespace remoting {
 namespace protocol {
 
 static const int kReadBufferSize = 4096;
 
 MessageReader::MessageReader()
     : socket_(nullptr),
       read_pending_(false),
       pending_messages_(0),
       closed_(false),
       weak_factory_(this) {
 }
 
 MessageReader::~MessageReader() {
 }
 
 void MessageReader::SetMessageReceivedCallback(
     const MessageReceivedCallback& callback) {
   DCHECK(CalledOnValidThread());
   message_received_callback_ = callback;
 }
 
 void MessageReader::StartReading(
-    net::Socket* socket,
+    P2PStreamSocket* socket,
     const ReadFailedCallback& read_failed_callback) {
   DCHECK(CalledOnValidThread());
   DCHECK(socket);
   DCHECK(!read_failed_callback.is_null());
 
   socket_ = socket;
   read_failed_callback_ = read_failed_callback;
   DoRead();
 }
 
@@ skipping 82 matching lines @@
   DCHECK(CalledOnValidThread());
   pending_messages_--;
   DCHECK_GE(pending_messages_, 0);
 
   // Start next read if necessary.
   DoRead();
 }
 
 }  // namespace protocol
 }  // namespace remoting