Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(535)

Unified Diff: remoting/protocol/message_reader.cc

Issue 1143443003: Fix MessageReader to pass errors to the channel (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 5 years, 7 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: remoting/protocol/message_reader.cc
diff --git a/remoting/protocol/message_reader.cc b/remoting/protocol/message_reader.cc
index 6e3d5369bf2ca690b8e4d4ce740b176dbf30a4a9..cd7fdedd4df8f995942f2958fd1515031e04833f 100644
--- a/remoting/protocol/message_reader.cc
+++ b/remoting/protocol/message_reader.cc
@@ -38,10 +38,15 @@ void MessageReader::SetMessageReceivedCallback(
message_received_callback_ = callback;
}
-void MessageReader::StartReading(net::Socket* socket) {
+void MessageReader::StartReading(
+ net::Socket* 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();
}
@@ -49,13 +54,15 @@ void MessageReader::DoRead() {
DCHECK(CalledOnValidThread());
// Don't try to read again if there is another read pending or we
// have messages that we haven't finished processing yet.
- while (!closed_ && !read_pending_ && pending_messages_ == 0) {
+ bool read_again = true;
+ while (read_again && !closed_ && !read_pending_ && pending_messages_ == 0) {
read_buffer_ = new net::IOBuffer(kReadBufferSize);
int result = socket_->Read(
read_buffer_.get(),
kReadBufferSize,
base::Bind(&MessageReader::OnRead, weak_factory_.GetWeakPtr()));
- HandleReadResult(result);
+
+ HandleReadResult(result, &read_again);
}
}
@@ -65,26 +72,33 @@ void MessageReader::OnRead(int result) {
read_pending_ = false;
if (!closed_) {
- HandleReadResult(result);
- DoRead();
+ bool read_again;
+ HandleReadResult(result, &read_again);
+ if (read_again)
+ DoRead();
}
}
-void MessageReader::HandleReadResult(int result) {
+void MessageReader::HandleReadResult(int result, bool* read_again) {
DCHECK(CalledOnValidThread());
if (closed_)
return;
+ *read_again = false;
+
if (result > 0) {
OnDataReceived(read_buffer_.get(), result);
+ *read_again = true;
} else if (result == net::ERR_IO_PENDING) {
read_pending_ = true;
} else {
- if (result != net::ERR_CONNECTION_CLOSED) {
- LOG(ERROR) << "Read() returned error " << result;
- }
+ DCHECK_LT(result, 0);
+
// Stop reading after any error.
closed_ = true;
+
+ LOG(ERROR) << "Read() returned error " << result;
+ read_failed_callback_.Run(result);
}
}

Powered by Google App Engine
This is Rietveld 408576698