Server.cpp

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