src/remoting/base/socket_reader.h

   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
  12 namespace net {
  13 class IOBuffer;
  14 class Socket;
  15 }  // namespace net
  16
  17 namespace 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).
  23 class 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_