1 files changed, 102 insertions, 0 deletions
diff --git a/remoting/protocol/stream_reader.cc b/remoting/protocol/stream_reader.cc
new file mode 100644
index 0000000..bfceb3e
--- /dev/null
+++ b/remoting/protocol/stream_reader.cc
@@ -0,0 +1,102 @@
+// 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/stream_reader.h"
+
+#include "base/message_loop.h"
+#include "net/base/net_errors.h"
+#include "net/socket/socket.h"
+#include "remoting/protocol/chromoting_connection.h"
+
+namespace remoting {
+
+namespace {
+int kReadBufferSize = 4096;
+}  // namespace
+
+StreamReaderBase::StreamReaderBase()
+    : socket_(NULL),
+      closed_(false),
+      ALLOW_THIS_IN_INITIALIZER_LIST(
+          read_callback_(this, &StreamReaderBase::OnRead)) {
+}
+
+StreamReaderBase::~StreamReaderBase() { }
+
+void StreamReaderBase::Close() {
+  closed_ = true;
+}
+
+void StreamReaderBase::Init(net::Socket* socket) {
+  DCHECK(socket);
+  socket_ = socket;
+  DoRead();
+}
+
+void StreamReaderBase::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 StreamReaderBase::OnRead(int result) {
+  if (!closed_) {
+    HandleReadResult(result);
+    DoRead();
+  }
+}
+
+void StreamReaderBase::HandleReadResult(int result) {
+  if (result > 0) {
+    OnDataReceived(read_buffer_, result);
+  } else {
+    if (result != net::ERR_IO_PENDING)
+      LOG(ERROR) << "Read() returned error " << result;
+  }
+}
+
+// EventsStreamReader class.
+EventsStreamReader::EventsStreamReader() { }
+EventsStreamReader::~EventsStreamReader() { }
+
+void EventsStreamReader::Init(net::Socket* socket,
+                              OnMessageCallback* on_message_callback) {
+  on_message_callback_.reset(on_message_callback);
+  StreamReaderBase::Init(socket);
+}
+
+void EventsStreamReader::OnDataReceived(net::IOBuffer* buffer, int data_size) {
+  ClientMessageList messages_list;
+  messages_decoder_.ParseClientMessages(buffer, data_size, &messages_list);
+  for (ClientMessageList::iterator it = messages_list.begin();
+       it != messages_list.end(); ++it) {
+    on_message_callback_->Run(*it);
+  }
+}
+
+// VideoStreamReader class.
+VideoStreamReader::VideoStreamReader() { }
+VideoStreamReader::~VideoStreamReader() { }
+
+void VideoStreamReader::Init(net::Socket* socket,
+                             OnMessageCallback* on_message_callback) {
+  on_message_callback_.reset(on_message_callback);
+  StreamReaderBase::Init(socket);
+}
+
+void VideoStreamReader::OnDataReceived(net::IOBuffer* buffer, int data_size) {
+  HostMessageList messages_list;
+  messages_decoder_.ParseHostMessages(buffer, data_size, &messages_list);
+  for (HostMessageList::iterator it = messages_list.begin();
+       it != messages_list.end(); ++it) {
+    on_message_callback_->Run(*it);
+  }
+}
+
+}  // namespace remoting