Listener.cpp

   1 #include <iostream>
   2 #include <string.h>
   3 #include <unistd.h>
   4 #include <errno.h>
   5 #include <netdb.h>
   6 #include <fcntl.h>
   7 #include <sys/types.h>
   8 #include <sys/socket.h>
   9 #include <netinet/in.h>
  10 #include <arpa/inet.h>
  11 #include "Listener.hpp"
  12 #include "Server.hpp"
  13 #include "Log.hpp"
  14 #include "Exceptions.hpp"
  15
  16 Listener::Listener()
  17         : m_sock( 0 )
  18 {
  19         // Discover machine's IP address
  20         char buf[128];
  21         gethostname( buf, 128 );
  22
  23         hostent* h;
  24         if( ( h = gethostbyname( buf ) ) == (void*)-1 )
  25         {
  26                 g_log->Print( strerror( errno ) );
  27                 throw ServerCrashException;
  28         }
  29         m_ipaddr = inet_ntoa( *((in_addr*)h->h_addr) );
  30 }
  31
  32 Listener::Listener( const std::string& ip )
  33         : m_sock( 0 )
  34         , m_ipaddr( ip )
  35 {
  36 }
  37
  38 Listener::~Listener()
  39 {
  40         if( m_sock != 0 )
  41         {
  42                 g_log->Print( "[Listener] Closing socket" );
  43                 close( m_sock );
  44         }
  45 }
  46
  47 void Listener::Listen()
  48 {
  49         g_log->Print( "[Listener] Opening socket" );
  50
  51         if( ( m_sock = socket( PF_INET, SOCK_STREAM, 0 ) ) == -1 )
  52         {
  53                 g_log->Print( strerror( errno ) );
  54                 throw ServerCrashException;
  55         }
  56
  57         // Reuse port
  58         int yes = 1;
  59         if( setsockopt( m_sock, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof( int ) ) == -1 )
  60         {
  61                 g_log->Print( strerror( errno ) );
  62                 throw ServerCrashException;
  63         }
  64
  65         sockaddr_in addr;
  66         addr.sin_family = AF_INET;
  67         addr.sin_port = htons( 21 );
  68         addr.sin_addr.s_addr = inet_addr( m_ipaddr.c_str() );
  69         memset( addr.sin_zero, 0, sizeof( addr.sin_zero ) );
  70
  71         if( bind( m_sock, (sockaddr*)&addr, sizeof( addr ) ) == -1 )
  72         {
  73                 g_log->Print( strerror( errno ) );
  74                 throw ServerCrashException;
  75         }
  76
  77         if( listen( m_sock, 5 ) == -1 )
  78         {
  79                 g_log->Print( strerror( errno ) );
  80                 throw ServerCrashException;
  81         }
  82 }
  83
  84 void Listener::Tick()
  85 {
  86         // Check if there are some incoming connections
  87         sockaddr_in addr;
  88         socklen_t size = sizeof( addr );
  89
  90         int incoming = accept( m_sock, (sockaddr*)&addr, &size );
  91         if( incoming == -1 )
  92         {
  93                 g_log->Print( strerror( errno ) );
  94                 throw ServerCrashException;
  95         }
  96
  97         ServerPtr server = m_server.lock();
  98         server->IncomingConnection( incoming );
  99 }