Index: remoting/protocol/message_reader.h |
diff --git a/remoting/protocol/message_reader.h b/remoting/protocol/message_reader.h |
index 97023b6bdceff3287ad73d5e659cffc291015028..30e425de15a7d5ae4398e7f1408d915846117192 100644 |
--- a/remoting/protocol/message_reader.h |
+++ b/remoting/protocol/message_reader.h |
@@ -14,12 +14,13 @@ |
namespace net { |
class IOBuffer; |
-class Socket; |
} // namespace net |
namespace remoting { |
namespace protocol { |
+class P2PStreamSocket; |
+ |
// MessageReader reads data from the socket asynchronously and calls |
// callback for each message it receives. It stops calling the |
// callback as soon as the socket is closed, so the socket should |
@@ -44,7 +45,7 @@ class MessageReader : public base::NonThreadSafe { |
void SetMessageReceivedCallback(const MessageReceivedCallback& callback); |
// Starts reading from |socket|. |
- void StartReading(net::Socket* socket, |
+ void StartReading(P2PStreamSocket* socket, |
const ReadFailedCallback& read_failed_callback); |
private: |
@@ -57,7 +58,7 @@ class MessageReader : public base::NonThreadSafe { |
ReadFailedCallback read_failed_callback_; |
- net::Socket* socket_; |
+ P2PStreamSocket* socket_; |
// Set to true, when we have a socket read pending, and expecting |
// OnRead() to be called when new data is received. |