1// Copyright (c) 2012 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#ifndef REMOTING_BASE_SOCKET_READER_H_
6#define REMOTING_BASE_SOCKET_READER_H_
7
8#include "base/memory/ref_counted.h"
9#include "net/base/completion_callback.h"
10#include "base/memory/weak_ptr.h"
11
12namespace net {
13class IOBuffer;
14class Socket;
15}  // namespace net
16
17namespace remoting {
18
19// SocketReader reads data from a socket and then calls a callback for each
20// completed read. Note that when this object is destroyed it may leave a
21// pending read request for the socket, so the calling code should never try
22// reading from the same socket again (e.g. it may result in data being lost).
23class SocketReader {
24 public:
25  // Callback that is called for each finished read. |data| may be set to NULL
26  // in case of an error (result < 0).
27  typedef base::Callback<void(scoped_refptr<net::IOBuffer> data,
28                              int result)> ReadResultCallback;
29
30  SocketReader();
31  ~SocketReader();
32
33  // Starts reading from |socket|. |read_result_callback| is called for each
34  // completed read. Reading stops on the first error. Must not be called more
35  // than once.
36  void Init(net::Socket* socket, ReadResultCallback read_result_callback);
37
38 private:
39  void DoRead();
40  void OnRead(int result);
41  void HandleReadResult(int result);
42  void CallCallback(scoped_refptr<net::IOBuffer> data, int result);
43
44  net::Socket* socket_;
45  ReadResultCallback read_result_callback_;
46  scoped_refptr<net::IOBuffer> read_buffer_;
47
48  base::WeakPtrFactory<SocketReader> weak_factory_;
49
50  DISALLOW_COPY_AND_ASSIGN(SocketReader);
51};
52
53}  // namespace remoting
54
55#endif  // REMOTING_BASE_SOCKET_READER_H_
56