net/socket/stream_listen_socket.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 // Stream-based listen socket implementation that handles reading and writing
   6 // to the socket, but does not handle creating the socket nor connecting
   7 // sockets, which are handled by subclasses on creation and in Accept,
   8 // respectively.
   9
  10 // StreamListenSocket handles IO asynchronously in the specified MessageLoop.
  11 // This class is NOT thread safe. It uses WSAEVENT handles to monitor activity
  12 // in a given MessageLoop. This means that callbacks will happen in that loop's
  13 // thread always and that all other methods (including constructor and
  14 // destructor) should also be called from the same thread.
  15
  16 #ifndef NET_SOCKET_STREAM_LISTEN_SOCKET_H_
  17 #define NET_SOCKET_STREAM_LISTEN_SOCKET_H_
  18
  19 #include "build/build_config.h"
  20
  21 #if defined(OS_WIN)
  22 #include <winsock2.h>
  23 #endif
  24 #include <string>
  25 #if defined(OS_WIN)
  26 #include "base/win/object_watcher.h"
  27 #elif defined(OS_POSIX)
  28 #include "base/message_loop/message_loop.h"
  29 #endif
  30
  31 #include "base/basictypes.h"
  32 #include "base/compiler_specific.h"
  33 #include "base/memory/scoped_ptr.h"
  34 #include "net/base/net_export.h"
  35 #include "net/socket/socket_descriptor.h"
  36
  37 namespace net {
  38
  39 class IPEndPoint;
  40
  41 class NET_EXPORT StreamListenSocket
  42     :
  43 #if defined(OS_WIN)
  44       public base::win::ObjectWatcher::Delegate {
  45 #elif defined(OS_POSIX)
  46       public base::MessageLoopForIO::Watcher {
  47 #endif
  48
  49  public:
  50   virtual ~StreamListenSocket();
  51
  52   // TODO(erikkay): this delegate should really be split into two parts
  53   // to split up the listener from the connected socket.  Perhaps this class
  54   // should be split up similarly.
  55   class Delegate {
  56    public:
  57     // |server| is the original listening Socket, connection is the new
  58     // Socket that was created.
  59     virtual void DidAccept(StreamListenSocket* server,
  60                            scoped_ptr<StreamListenSocket> connection) = 0;
  61     virtual void DidRead(StreamListenSocket* connection,
  62                          const char* data,
  63                          int len) = 0;
  64     virtual void DidClose(StreamListenSocket* sock) = 0;
  65
  66    protected:
  67     virtual ~Delegate() {}
  68   };
  69
  70   // Send data to the socket.
  71   void Send(const char* bytes, int len, bool append_linefeed = false);
  72   void Send(const std::string& str, bool append_linefeed = false);
  73
  74   // Copies the local address to |address|. Returns a network error code.
  75   int GetLocalAddress(IPEndPoint* address);
  76
  77   static const int kSocketError;
  78
  79  protected:
  80   enum WaitState {
  81     NOT_WAITING      = 0,
  82     WAITING_ACCEPT   = 1,
  83     WAITING_READ     = 2
  84   };
  85
  86   StreamListenSocket(SocketDescriptor s, Delegate* del);
  87
  88   SocketDescriptor AcceptSocket();
  89   virtual void Accept() = 0;
  90
  91   void Listen();
  92   void Read();
  93   void Close();
  94   void CloseSocket(SocketDescriptor s);
  95
  96   // Pass any value in case of Windows, because in Windows
  97   // we are not using state.
  98   void WatchSocket(WaitState state);
  99   void UnwatchSocket();
 100
 101   Delegate* const socket_delegate_;
 102
 103  private:
 104   friend class TransportClientSocketTest;
 105
 106   void SendInternal(const char* bytes, int len);
 107
 108 #if defined(OS_WIN)
 109   // ObjectWatcher delegate.
 110   virtual void OnObjectSignaled(HANDLE object);
 111   base::win::ObjectWatcher watcher_;
 112   HANDLE socket_event_;
 113 #elif defined(OS_POSIX)
 114   // Called by MessagePumpLibevent when the socket is ready to do I/O.
 115   virtual void OnFileCanReadWithoutBlocking(int fd) OVERRIDE;
 116   virtual void OnFileCanWriteWithoutBlocking(int fd) OVERRIDE;
 117   WaitState wait_state_;
 118   // The socket's libevent wrapper.
 119   base::MessageLoopForIO::FileDescriptorWatcher watcher_;
 120 #endif
 121
 122   // NOTE: This is for unit test use only!
 123   // Pause/Resume calling Read(). Note that ResumeReads() will also call
 124   // Read() if there is anything to read.
 125   void PauseReads();
 126   void ResumeReads();
 127
 128   const SocketDescriptor socket_;
 129   bool reads_paused_;
 130   bool has_pending_reads_;
 131
 132   DISALLOW_COPY_AND_ASSIGN(StreamListenSocket);
 133 };
 134
 135 // Abstract factory that must be subclassed for each subclass of
 136 // StreamListenSocket.
 137 class NET_EXPORT StreamListenSocketFactory {
 138  public:
 139   virtual ~StreamListenSocketFactory() {}
 140
 141   // Returns a new instance of StreamListenSocket or NULL if an error occurred.
 142   virtual scoped_ptr<StreamListenSocket> CreateAndListen(
 143       StreamListenSocket::Delegate* delegate) const = 0;
 144 };
 145
 146 }  // namespace net
 147
 148 #endif  // NET_SOCKET_STREAM_LISTEN_SOCKET_H_