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_ 56