rxsocket.hh

説明を見る。

#ifndef RXSOCKET_HH_
# define RXSOCKET_HH_

# include <winsock2.h>
# include <windows.h>
# include <map>
# include <iostream>
# include <sstream>

# define RXSOCKET_WND "RxSocket"
# define RXSOCKET_EVENTMSG 0x7FFF
# define RXSOCKET_EVENTS (FD_READ | FD_WRITE | FD_ACCEPT | FD_CONNECT | FD_CLOSE)
# define RXSOCKET_BUFFER_SIZE 1024

namespace rxsocket
{
  class RxSocketInit
  {
  public:
    enum Status
    {
      status_uninitialized,
      status_ok
    };

  public:
    RxSocketInit();
    ~RxSocketInit();

  private:
    Status _status;
    WSADATA _wsa_data;
    Status init();
  };

  class RxSocket
  {
  public:
    ~RxSocket();

    virtual void process_event(int code, int err);
    virtual void open() = 0;
    virtual void close();

    virtual int data_available() = 0;
    virtual bool is_connected();
    void print(std::ostream &os);

    static void set_instance(HINSTANCE instance);
  protected:
    RxSocket();
  
  protected:
    HWND _wnd;
    SOCKET _socket;

    virtual void event_connect();
    virtual void event_close();
    virtual void can_accept();
    virtual void can_write() = 0;
    virtual void can_read() = 0;

  public:
  };

  class RxTCPClient : public RxSocket
  {
  public:
    RxTCPClient(const std::string &host, u_short port,
      int buffer_size = RXSOCKET_BUFFER_SIZE);
  
    virtual void open();
    virtual int data_available() const;
    virtual bool can_send() const;
    virtual void send(const std::string &str);
    virtual void receive(std::string &str, int size);

    virtual const std::string &get_send_buf() const;
    virtual const std::string &get_receive_buf() const;

  protected:
    std::string _host;
    u_short _port;
    std::string _send_buf;
    std::string _receive_buf;
    bool _can_send;
    int _buffer_size;

    void send_now();
    void def_can_write();
    void def_can_read();
    virtual void can_write();
    virtual void can_read();
  };
}

#endif

---