include/socket.h

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  Inspire is copyright (C) 2002-2004 ChatSpike-Dev.
   6  *                       E-mail:
   7  *                <brain@chatspike.net>
   8  *                <Craig@chatspike.net>
   9  *
  10  * Written by Craig Edwards, Craig McLure, and others.
  11  * This program is free but copyrighted software; see
  12  *            the file COPYING for details.
  13  *
  14  * ---------------------------------------------------
  15  */
  16
  17 #ifndef __INSP_SOCKET_H__
  18 #define __INSP_SOCKET_H__
  19
  20 #include <sys/types.h>
  21 #include <sys/socket.h>
  22 #include <netinet/in.h>
  23 #include <poll.h>
  24 #include <sstream>
  25 #include <string>
  26
  27 enum InspSocketState { I_DISCONNECTED, I_CONNECTING, I_CONNECTED, I_LISTENING, I_ERROR };
  28 enum InspSocketError { I_ERR_TIMEOUT, I_ERR_SOCKET, I_ERR_CONNECT, I_ERR_BIND };
  29
  30 class InspSocket
  31 {
  32 private:
  33         int fd;
  34         std::string host;
  35         int port;
  36         InspSocketState state;
  37         sockaddr_in addr;
  38         in_addr addy;
  39         time_t timeout_end;
  40         bool timeout;
  41         pollfd polls;
  42         char ibuf[1024];
  43         std::string IP;
  44         sockaddr_in client;
  45         sockaddr_in server;
  46         socklen_t length;
  47 public:
  48         InspSocket();
  49         InspSocket(int newfd);
  50         InspSocket(std::string host, int port, bool listening, unsigned long maxtime);
  51         virtual bool OnConnected();
  52         virtual void OnError(InspSocketError e);
  53         virtual int OnDisconnect();
  54         virtual bool OnDataReady();
  55         virtual void OnTimeout();
  56         virtual void OnClose();
  57         virtual char* Read();
  58         std::string GetIP();
  59         virtual int Write(std::string data);
  60         virtual int OnIncomingConnection(int newfd, char* ip);
  61         void SetState(InspSocketState s);
  62         InspSocketState GetState();
  63         bool Poll();
  64         virtual void Close();
  65         virtual ~InspSocket();
  66 };
  67
  68 #endif