asyncsocket.h

本人收集整理的一份c/c++跨平台网络库

#ifndef UTILS_BASE_ASYNCSOCKET_H_
#define UTILS_BASE_ASYNCSOCKET_H_

#include "sigslot.h"
#include "socket.h"

namespace utils_base {

// Provides the ability to perform socket I/O asynchronously.
class AsyncSocket : public Socket, public sigslot::has_slots<>  {
public:
  virtual ~AsyncSocket() {}

  sigslot::signal1<AsyncSocket*> SignalReadEvent;  // ready to read
  sigslot::signal1<AsyncSocket*> SignalWriteEvent; // ready to write
  sigslot::signal1<AsyncSocket*> SignalConnectEvent; // connected
  sigslot::signal2<AsyncSocket*,int> SignalCloseEvent; // closed
  // TODO: error
};

class AsyncSocketAdapter : public AsyncSocket {
public:
  AsyncSocketAdapter(Socket * socket) : socket_(socket) {
  }
  AsyncSocketAdapter(AsyncSocket * socket) : socket_(socket) {
    socket->SignalConnectEvent.connect(this, &AsyncSocketAdapter::OnConnectEvent);
    socket->SignalReadEvent.connect(this, &AsyncSocketAdapter::OnReadEvent);
    socket->SignalWriteEvent.connect(this, &AsyncSocketAdapter::OnWriteEvent);
    socket->SignalCloseEvent.connect(this, &AsyncSocketAdapter::OnCloseEvent);
  }
  virtual ~AsyncSocketAdapter() { delete socket_; }

  virtual SocketAddress GetLocalAddress() const { return socket_->GetLocalAddress(); }
  virtual SocketAddress GetRemoteAddress() const { return socket_->GetRemoteAddress(); }

  virtual int Bind(const SocketAddress& addr) { return socket_->Bind(addr); }
  virtual int Connect(const SocketAddress& addr) {return socket_->Connect(addr); }
  virtual int Send(const void *pv, size_t cb) { return socket_->Send(pv, cb); }
  virtual int SendTo(const void *pv, size_t cb, const SocketAddress& addr) { return socket_->SendTo(pv, cb, addr); }
  virtual int Recv(void *pv, size_t cb) { return socket_->Recv(pv, cb); }
  virtual int RecvFrom(void *pv, size_t cb, SocketAddress *paddr) { return socket_->RecvFrom(pv, cb, paddr); }
  virtual int Listen(int backlog) { return socket_->Listen(backlog); }
  virtual Socket *Accept(SocketAddress *paddr) { return socket_->Accept(paddr); }
  virtual int Close() { return socket_->Close(); }
  virtual int GetError() const { return socket_->GetError(); }
  virtual void SetError(int error) { return socket_->SetError(error); }

  virtual ConnState GetState() const { return socket_->GetState(); }

  virtual int EstimateMTU(uint16* mtu) { return socket_->EstimateMTU(mtu); }
  virtual int SetOption(Option opt, int value) { return socket_->SetOption(opt, value); }

protected:
  virtual void OnConnectEvent(AsyncSocket * socket) { SignalConnectEvent(this); }
  virtual void OnReadEvent(AsyncSocket * socket) { SignalReadEvent(this); }
  virtual void OnWriteEvent(AsyncSocket * socket) { SignalWriteEvent(this); }
  virtual void OnCloseEvent(AsyncSocket * socket, int err) { SignalCloseEvent(this, err); }

  Socket * socket_;
};

} // namespace utils_base

#endif // UTILS_BASE_ASYNCSOCKET_H_