1 files changed, 97 insertions, 0 deletions
diff --git a/remoting/protocol/stream_reader.h b/remoting/protocol/stream_reader.h
new file mode 100644
index 0000000..b71f4b0
--- /dev/null
+++ b/remoting/protocol/stream_reader.h
@@ -0,0 +1,97 @@
+// 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.
+
+#ifndef REMOTING_PROTOCOL_MESSAGES_STREAM_READER_H_
+#define REMOTING_PROTOCOL_MESSAGES_STREAM_READER_H_
+
+#include "base/callback.h"
+#include "base/ref_counted.h"
+#include "base/scoped_ptr.h"
+#include "net/base/completion_callback.h"
+#include "remoting/protocol/messages_decoder.h"
+
+namespace net {
+class Socket;
+}  // namespace net
+
+namespace remoting {
+
+class ChromotingConnection;
+
+class StreamReaderBase {
+ public:
+  StreamReaderBase();
+  virtual ~StreamReaderBase();
+
+  // Stops reading. Must be called on the same thread as Init().
+  void Close();
+
+ protected:
+  void Init(net::Socket* socket);
+  virtual void OnDataReceived(net::IOBuffer* buffer, int data_size) = 0;
+
+ private:
+  void DoRead();
+  void OnRead(int result);
+  void HandleReadResult(int result);
+
+  net::Socket* socket_;
+  bool closed_;
+  scoped_refptr<net::IOBuffer> read_buffer_;
+  net::CompletionCallbackImpl<StreamReaderBase> read_callback_;
+};
+
+class EventsStreamReader : public StreamReaderBase {
+ public:
+  EventsStreamReader();
+  ~EventsStreamReader();
+
+  // The OnMessageCallback is called whenever a new message is received.
+  // Ownership of the message is passed the callback.
+  typedef Callback1<ChromotingClientMessage*>::Type OnMessageCallback;
+
+  // Initialize the reader and start reading. Must be called on the thread
+  // |socket| belongs to. The callback will be called when a new message is
+  // received. EventsStreamReader owns |on_message_callback|, doesn't own
+  // |socket|.
+  void Init(net::Socket* socket, OnMessageCallback* on_message_callback);
+
+ protected:
+  virtual void OnDataReceived(net::IOBuffer* buffer, int data_size);
+
+ private:
+  MessagesDecoder messages_decoder_;
+  scoped_ptr<OnMessageCallback> on_message_callback_;
+
+  DISALLOW_COPY_AND_ASSIGN(EventsStreamReader);
+};
+
+class  VideoStreamReader : public StreamReaderBase {
+ public:
+  VideoStreamReader();
+  ~VideoStreamReader();
+
+  // The OnMessageCallback is called whenever a new message is received.
+  // Ownership of the message is passed the callback.
+  typedef Callback1<ChromotingHostMessage*>::Type OnMessageCallback;
+
+  // Initialize the reader and start reading. Must be called on the thread
+  // |socket| belongs to. The callback will be called when a new message is
+  // received. VideoStreamReader owns |on_message_callback|, doesn't own
+  // |socket|.
+  void Init(net::Socket* socket, OnMessageCallback* on_message_callback);
+
+ protected:
+  virtual void OnDataReceived(net::IOBuffer* buffer, int data_size);
+
+ private:
+  MessagesDecoder messages_decoder_;
+  scoped_ptr<OnMessageCallback> on_message_callback_;
+
+  DISALLOW_COPY_AND_ASSIGN(VideoStreamReader);
+};
+
+}  // namespace remoting
+
+#endif  // REMOTING_PROTOCOL_MESSAGES_STREAM_READER_H_