HostMessageDispatcher to parse control messages

remoting/protocol/message_reader.cc

+// Copyright (c) 2010 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/message_loop.h"
+#include "net/base/io_buffer.h"
+#include "net/base/net_errors.h"
+#include "net/socket/socket.h"
+#include "remoting/base/multiple_array_input_stream.h"
+#include "remoting/proto/internal.pb.h"
+#include "remoting/protocol/chromoting_connection.h"
+
+namespace remoting {
+
+static const int kReadBufferSize = 4096;
+
+MessageReader::MessageReader()
+    : socket_(NULL),
+      closed_(false),
+      ALLOW_THIS_IN_INITIALIZER_LIST(
+          read_callback_(this, &MessageReader::OnRead)) {
+}
+
+MessageReader::~MessageReader() {
+  // Destroy MessageReaderPrivate if it was created.
+  if (destruction_callback_.get())
+    destruction_callback_->Run();
+}
+
+void MessageReader::Close() {
+  closed_ = true;
+}
+
+void MessageReader::Init(net::Socket* socket) {
+  DCHECK(socket);
+  socket_ = socket;
+  DoRead();
+}
+
+void MessageReader::DoRead() {
+  while (true) {
+    read_buffer_ = new net::IOBuffer(kReadBufferSize);
+    int result = socket_->Read(
+        read_buffer_, kReadBufferSize, &read_callback_);
+    HandleReadResult(result);
+    if (result < 0)
+      break;
+  }
+}
+
+void MessageReader::OnRead(int result) {
+  if (!closed_) {
+    HandleReadResult(result);
+    DoRead();
+  }
+}
+
+void MessageReader::HandleReadResult(int result) {
+  if (result > 0) {
+    data_received_callback_->Run(read_buffer_, result);
+  } else {
+    if (result != net::ERR_IO_PENDING)
+      LOG(ERROR) << "Read() returned error " << result;
+  }
+}
+
+}  // namespace remoting