Server.cpp

   1 #include <iostream>
   2 #include <sys/select.h>
   3 #include <string.h>
   4 #include <errno.h>
   5 #include "Server.hpp"
   6 #include "Auth.hpp"
   7 #include "Log.hpp"
   8 #include "Exceptions.hpp"
   9
  10 Server::Server( const AuthPtr& auth )
  11         : m_listener( new Listener )
  12         , m_sessionController( new SessionController )
  13         , m_auth( auth )
  14 {
  15         g_log->Print( "Dumb FTP server" );
  16         g_log->Print( std::string("IP: ") + m_listener->GetIPAddr() );
  17
  18         LoadWelcomeMessage();
  19
  20         m_listener->Listen();
  21 }
  22
  23 Server::Server( const AuthPtr& auth, const std::string& ip )
  24         : m_listener( new Listener( ip ) )
  25         , m_sessionController( new SessionController )
  26         , m_auth( auth )
  27 {
  28         g_log->Print( "Dumb FTP server" );
  29         g_log->Print( std::string("IP: ") + m_listener->GetIPAddr() );
  30
  31         LoadWelcomeMessage();
  32
  33         m_listener->Listen();
  34 }
  35
  36 Server::~Server()
  37 {
  38         g_log->Print( "[Server] Shutting down" );
  39 }
  40
  41 ServerPtr Server::Create( const AuthPtr& auth )
  42 {
  43         ServerPtr ret( new Server( auth ) );
  44         ret->m_this = ret;
  45
  46         ret->InitListener();
  47
  48         return ret;
  49 }
  50
  51 ServerPtr Server::Create( const AuthPtr& auth, const std::string& ip )
  52 {
  53         ServerPtr ret( new Server( auth, ip ) );
  54         ret->m_this = ret;
  55
  56         ret->InitListener();
  57
  58         return ret;
  59 }
  60
  61 void Server::Tick()
  62 {
  63         fd_set read, write;
  64         std::list<int> fds = m_sessionController->GetFds();
  65         int maxfd = m_listener->GetSock();
  66
  67         FD_ZERO( &read );
  68         FD_ZERO( &write );
  69
  70         FD_SET( maxfd, &read );
  71
  72         for( std::list<int>::const_iterator it = fds.begin(); it != fds.end(); ++it )
  73         {
  74                 maxfd = std::max( maxfd, abs( *it ) );
  75
  76                 if( *it < 0 )
  77                 {
  78                         // Hack
  79                         FD_SET( -(*it), &write );
  80                 }
  81                 else
  82                 {
  83                         FD_SET( *it, &read );
  84                 }
  85         }
  86
  87         timeval tv;
  88         tv.tv_sec = 0;
  89         tv.tv_usec = 50000;
  90
  91         int ret = select( maxfd + 1, &read, &write, NULL, &tv );
  92
  93         if( ret == -1 )
  94         {
  95                 g_log->Print( strerror( errno ) );
  96                 throw ServerCrashException;
  97         }
  98         else if( ret == 0 )
  99         {
 100                 // Descriptors not ready
 101                 return;
 102         }
 103
 104         std::list<int> activeFds;
 105         for( std::list<int>::const_iterator it = fds.begin(); it != fds.end(); ++it )
 106         {
 107                 if( ( *it > 0 && FD_ISSET( *it, &read ) ) ||
 108                     ( *it < 0 && FD_ISSET( -(*it), &write ) ) )
 109                 {
 110                         activeFds.push_back( abs( *it ) );
 111                 }
 112         }
 113
 114         if( activeFds.size() != 0 )
 115         {
 116                 m_sessionController->Tick( activeFds );
 117         }
 118
 119         if( FD_ISSET( m_listener->GetSock(), &read ) )
 120         {
 121                 m_listener->Tick();
 122         }
 123 }
 124
 125 void Server::IncomingConnection( int sock )
 126 {
 127         m_sessionController->Add( Session::Create( sock, m_sessionController, m_auth, m_listener->GetIPAddr(), m_this ) );
 128 }
 129
 130 void Server::InitListener()
 131 {
 132         m_listener->SetServer( m_this );
 133 }
 134
 135 void Server::LoadWelcomeMessage()
 136 {
 137         FILE *f = fopen( "welcome", "r" );
 138         if( !f )
 139         {
 140                 return;
 141         }
 142
 143         char buf[256];
 144         while( fgets( buf, 256, f ) )
 145         {
 146                 if( buf[strlen( buf ) -  1] == '\n' )
 147                 {
 148                         buf[strlen( buf ) - 1] = 0;
 149                 }
 150
 151                 m_welcome.push_back( buf );
 152         }
 153
 154         fclose( f );
 155 }