Session.hpp

   1 #ifndef __DFTPD__SESSION_HPP__
   2 #define __DFTPD__SESSION_HPP__
   3
   4 #include "SessionPtr.hpp"
   5 #include "SessionControllerPtr.hpp"
   6 #include "String.hpp"
   7 #include "Telnet.hpp"
   8 #include "Auth.hpp"
   9
  10 typedef std::vector<std::string> Command;
  11
  12 class Session
  13 {
  14         enum State
  15         {
  16                 S_GREETING,
  17                 S_LOGIN,
  18                 S_PASSWORD,
  19                 S_READY
  20         };
  21
  22         enum PassState
  23         {
  24                 PS_NONE,
  25                 PS_LOGGEDIN,
  26                 PS_BADPASS
  27         };
  28
  29 public:
  30         static SessionPtr Create( int controlSock, const SessionControllerPtr& sessionController, const AuthPtr& auth );
  31         ~Session();
  32
  33         void Tick();
  34
  35 private:
  36         Session( int controlSock, const SessionControllerPtr& sessionController, const AuthPtr& auth );
  37
  38         void Remove();
  39
  40         void SendGreeting();
  41         void SendSyntaxError();
  42         void SendNotLoggedIn();
  43         bool AwaitLogin();
  44         PassState AwaitPassword();
  45         void AwaitReady();
  46
  47         void HandleMode( const Command& cmd );
  48         void HandleType( const Command& cmd );
  49         void HandleStru( const Command& cmd );
  50
  51         Command GetCommand() { return ParseCommand( m_control->GetBuf() ); }
  52
  53
  54         TelnetPtr m_control;
  55         int m_controlSock;
  56
  57         int m_id;
  58         static int m_counter;
  59
  60         State m_state;
  61
  62         SessionControllerWPtr m_sessionController;
  63         SessionWPtr m_this;
  64         AuthPtr m_auth;
  65
  66         std::string m_user;
  67 };
  68
  69 #endif