include/nntp/nntpconnection.h

   1 #ifndef _nntpconnection_\r
   2 #define _nntpconnection_\r
   3 \r
   4 #include "../threadwrapper/cancelablerunnable.h"\r
   5 #include "../socketdefines.h"\r
   6 #include "../ilogger.h"\r
   7 #include "../message.h"\r
   8 #include "../localidentity.h"\r
   9 \r
  10 #include <string>\r
  11 #include <vector>\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 CancelableRunnable,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                 LocalIdentity m_authuser;               // -1 if user not authenticated, otherwise id of user from tblLocalIdentity\r
  55                 bool m_authenticated;\r
  56         };\r
  57 \r
  58         void SendBuffered(const std::string &data);\r
  59         void SendBufferedLine(const std::string &data)  { SendBuffered(data+"\r\n"); }\r
  60         void SocketSend();                      // immediately send buffered data - will block if send if no ready\r
  61         void SocketReceive();           // immediately recv data on socket - will block if no data is waiting\r
  62         void HandleReceivedData();\r
  63         std::vector<char>::iterator Find(std::vector<char> &buffer, const std::string &val);\r
  64         const bool HandleCommand(const NNTPCommand &command);\r
  65         void HandlePostedMessage(const std::string &message);\r
  66 \r
  67         void SendArticleParts(const NNTPCommand &command);\r
  68         void SendArticleOverInfo(Message &message);\r
  69 \r
  70         // various NNTP commands to handle\r
  71         const bool HandleQuitCommand(const NNTPCommand &command);\r
  72         const bool HandleModeCommand(const NNTPCommand &command);\r
  73         const bool HandleCapabilitiesCommand(const NNTPCommand &command);\r
  74         const bool HandleHelpCommand(const NNTPCommand &command);\r
  75         const bool HandleDateCommand(const NNTPCommand &command);\r
  76         const bool HandleListCommand(const NNTPCommand &command);\r
  77         const bool HandleGroupCommand(const NNTPCommand &command);\r
  78         const bool HandleListGroupCommand(const NNTPCommand &command);\r
  79         const bool HandleNextCommand(const NNTPCommand &command);\r
  80         const bool HandleLastCommand(const NNTPCommand &command);\r
  81         const bool HandleArticleCommand(const NNTPCommand &command);\r
  82         const bool HandleHeadCommand(const NNTPCommand &command);\r
  83         const bool HandleBodyCommand(const NNTPCommand &command);\r
  84         const bool HandleStatCommand(const NNTPCommand &command);\r
  85         const bool HandleNewGroupsCommand(const NNTPCommand &command);\r
  86         const bool HandlePostCommand(const NNTPCommand &command);\r
  87         const bool HandleOverCommand(const NNTPCommand &command);\r
  88         const bool HandleAuthInfoCommand(const NNTPCommand &command);\r
  89         const bool HandleGetTrustCommand(const NNTPCommand &command);\r
  90         const bool HandleSetTrustCommand(const NNTPCommand &command);\r
  91         const bool HandleGetTrustListCommand(const NNTPCommand &command);\r
  92 \r
  93         SOCKET m_socket;\r
  94         ClientStatus m_status;\r
  95         std::vector<char> m_sendbuffer;\r
  96         std::vector<char> m_receivebuffer;\r
  97         std::vector<char> m_tempbuffer;\r
  98 \r
  99 };\r
 100 \r
 101 #endif  // _nntpconnection_\r