Index: remoting/protocol/channel_dispatcher_base.cc |
diff --git a/remoting/protocol/channel_dispatcher_base.cc b/remoting/protocol/channel_dispatcher_base.cc |
index 575b1e57913206218e774cd82976e21ce29f2745..b5c78c2f9510e99744bcb43e5771e9666e024d55 100644 |
--- a/remoting/protocol/channel_dispatcher_base.cc |
+++ b/remoting/protocol/channel_dispatcher_base.cc |
@@ -5,7 +5,7 @@ |
#include "remoting/protocol/channel_dispatcher_base.h" |
#include "base/bind.h" |
-#include "net/socket/stream_socket.h" |
+#include "remoting/protocol/p2p_stream_socket.h" |
#include "remoting/protocol/session.h" |
#include "remoting/protocol/session_config.h" |
#include "remoting/protocol/stream_channel_factory.h" |
@@ -49,7 +49,7 @@ void ChannelDispatcherBase::Init(Session* session, |
} |
void ChannelDispatcherBase::OnChannelReady( |
- scoped_ptr<net::StreamSocket> socket) { |
+ scoped_ptr<P2PStreamSocket> socket) { |
if (!socket.get()) { |
event_handler_->OnChannelError(this, CHANNEL_CONNECTION_ERROR); |
return; |
@@ -57,9 +57,10 @@ void ChannelDispatcherBase::OnChannelReady( |
channel_factory_ = nullptr; |
channel_ = socket.Pass(); |
- writer_.Init(channel_.get(), |
- base::Bind(&ChannelDispatcherBase::OnReadWriteFailed, |
- base::Unretained(this))); |
+ writer_.Init( |
+ base::Bind(&P2PStreamSocket::Write, base::Unretained(channel_.get())), |
+ base::Bind(&ChannelDispatcherBase::OnReadWriteFailed, |
+ base::Unretained(this))); |
reader_.StartReading(channel_.get(), |
base::Bind(&ChannelDispatcherBase::OnReadWriteFailed, |
base::Unretained(this))); |