include/nntp/nntpconnection.h

   1 #ifndef _nntpconnection_\r
   2 #define _nntpconnection_\r
   3 \r
   4 #include "../socketdefines.h"\r
   5 #include "../ilogger.h"\r
   6 #include "../message.h"\r
   7 \r
   8 #include <string>\r
   9 #include <vector>\r
  10 //#include <zthread/Runnable.h>\r
  11 #include "../pthreadwrapper/runnable.h"\r
  12 \r
  13 #ifdef _WIN32\r
  14 \r
  15 #else\r
  16         #include <sys/socket.h>\r
  17         #include <sys/select.h>\r
  18         #include <sys/types.h>\r
  19         #include <netdb.h>\r
  20         #include <netinet/in.h>\r
  21         #include <arpa/inet.h>\r
  22 #endif\r
  23 \r
  24 class NNTPConnection:public PThread::Runnable,public ILogger\r
  25 {\r
  26 public:\r
  27         NNTPConnection(SOCKET sock);\r
  28         ~NNTPConnection();\r
  29 \r
  30         void Disconnect();\r
  31         const bool Disconnected()               { return m_socket==INVALID_SOCKET; }\r
  32 \r
  33         void Run();\r
  34 \r
  35 private:\r
  36 \r
  37         typedef enum ClientMode\r
  38         {\r
  39                 MODE_NONE=0,\r
  40                 MODE_READER\r
  41         };\r
  42         struct NNTPCommand\r
  43         {\r
  44                 std::string m_command;\r
  45                 std::vector<std::string> m_arguments;\r
  46         };\r
  47         struct ClientStatus\r
  48         {\r
  49                 ClientMode m_mode;\r
  50                 bool m_allowpost;\r
  51                 bool m_isposting;\r
  52                 long m_boardid;\r
  53                 long m_messageid;\r
  54         };\r
  55 \r
  56         void SendBuffered(const std::string &data);\r
  57         void SendBufferedLine(const std::string &data)  { SendBuffered(data+"\r\n"); }\r
  58         void SocketSend();                      // immediately send buffered data - will block if send if no ready\r
  59         void SocketReceive();           // immediately recv data on socket - will block if no data is waiting\r
  60         void HandleReceivedData();\r
  61         std::vector<char>::iterator Find(std::vector<char> &buffer, const std::string &val);\r
  62         const bool HandleCommand(const NNTPCommand &command);\r
  63         void HandlePostedMessage(const std::string &message);\r
  64 \r
  65         void SendArticleParts(const NNTPCommand &command);\r
  66         void SendArticleOverInfo(Message &message);\r
  67 \r
  68         // various NNTP commands to handle\r
  69         const bool HandleQuitCommand(const NNTPCommand &command);\r
  70         const bool HandleModeCommand(const NNTPCommand &command);\r
  71         const bool HandleCapabilitiesCommand(const NNTPCommand &command);\r
  72         const bool HandleHelpCommand(const NNTPCommand &command);\r
  73         const bool HandleDateCommand(const NNTPCommand &command);\r
  74         const bool HandleListCommand(const NNTPCommand &command);\r
  75         const bool HandleGroupCommand(const NNTPCommand &command);\r
  76         const bool HandleListGroupCommand(const NNTPCommand &command);\r
  77         const bool HandleNextCommand(const NNTPCommand &command);\r
  78         const bool HandleLastCommand(const NNTPCommand &command);\r
  79         const bool HandleArticleCommand(const NNTPCommand &command);\r
  80         const bool HandleHeadCommand(const NNTPCommand &command);\r
  81         const bool HandleBodyCommand(const NNTPCommand &command);\r
  82         const bool HandleStatCommand(const NNTPCommand &command);\r
  83         const bool HandleNewGroupsCommand(const NNTPCommand &command);\r
  84         const bool HandlePostCommand(const NNTPCommand &command);\r
  85         const bool HandleOverCommand(const NNTPCommand &command);\r
  86 \r
  87         SOCKET m_socket;\r
  88         ClientStatus m_status;\r
  89         std::vector<char> m_sendbuffer;\r
  90         std::vector<char> m_receivebuffer;\r
  91         std::vector<char> m_tempbuffer;\r
  92 \r
  93 };\r
  94 \r
  95 #endif  // _nntpconnection_\r