Server.cpp

#ifdef _WIN32
#include <winsock.h>
#undef min
#undef max
#else
#include <sys/select.h>
#endif

#include <iostream>
#include <string.h>
#include <errno.h>
#include "Server.hpp"
#include "Auth.hpp"
#include "Log.hpp"
#include "Exceptions.hpp"

Server::Server( const AuthPtr& auth )
	: m_listener( new Listener )
	, m_sessionController( new SessionController )
	, m_auth( auth )
{
	g_log->Print( "Dumb FTP server" );
	g_log->Print( std::string("IP: ") + m_listener->GetIPAddr() );

	LoadWelcomeMessage();

	m_listener->Listen();
}

Server::Server( const AuthPtr& auth, const std::string& ip )
	: m_listener( new Listener( ip ) )
	, m_sessionController( new SessionController )
	, m_auth( auth )
{
	g_log->Print( "Dumb FTP server" );
	g_log->Print( std::string("IP: ") + m_listener->GetIPAddr() );

	LoadWelcomeMessage();

	m_listener->Listen();
}

Server::~Server()
{
	g_log->Print( "[Server] Shutting down" );
}

ServerPtr Server::Create( const AuthPtr& auth )
{
	ServerPtr ret( new Server( auth ) );
	ret->m_this = ret;

	ret->InitListener();

	return ret;
}

ServerPtr Server::Create( const AuthPtr& auth, const std::string& ip )
{
	ServerPtr ret( new Server( auth, ip ) );
	ret->m_this = ret;

	ret->InitListener();

	return ret;
}

void Server::Tick()
{
	fd_set read, write;
	std::list<int> fds = m_sessionController->GetFds();
	int maxfd = m_listener->GetSock();

	FD_ZERO( &read );
	FD_ZERO( &write );

	FD_SET( maxfd, &read );

	for( std::list<int>::const_iterator it = fds.begin(); it != fds.end(); ++it )
	{
		maxfd = std::max( maxfd, abs( *it ) );

		if( *it < 0 )
		{
			// Hack
			FD_SET( -(*it), &write );
		}
		else
		{
			FD_SET( *it, &read );
		}
	}

	timeval tv;
	tv.tv_sec = 0;
	tv.tv_usec = 50000;

	int ret = select( maxfd + 1, &read, &write, NULL, &tv );

	if( ret == -1 )
	{
		g_log->Print( strerror( errno ) );
		throw ServerCrashException;
	}
	else if( ret == 0 )
	{
		// Descriptors not ready
		return;
	}

	std::list<int> activeFds;
	for( std::list<int>::const_iterator it = fds.begin(); it != fds.end(); ++it )
	{
		if( ( *it > 0 && FD_ISSET( *it, &read ) ) ||
		    ( *it < 0 && FD_ISSET( -(*it), &write ) ) )
		{
			activeFds.push_back( abs( *it ) );
		}
	}

	if( activeFds.size() != 0 )
	{
		m_sessionController->Tick( activeFds );
	}

	if( FD_ISSET( m_listener->GetSock(), &read ) )
	{
		m_listener->Tick();
	}
}

void Server::IncomingConnection( int sock )
{
	m_sessionController->Add( Session::Create( sock, m_sessionController, m_auth, m_listener->GetIPAddr(), m_this ) );
}

void Server::InitListener()
{
	m_listener->SetServer( m_this );
}

void Server::LoadWelcomeMessage()
{
	FILE *f = fopen( "welcome", "r" );
	if( !f )
	{
		return;
	}

	char buf[256];
	while( fgets( buf, 256, f ) )
	{
		if( buf[strlen( buf ) -  1] == '\n' )
		{
			buf[strlen( buf ) - 1] = 0;
		}

		m_welcome.push_back( buf );
	}

	fclose( f );
}