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
  14 Listener::Listener()
  15         : m_sock( 0 )
  16 {
  17         // Discover machine's IP address
  18         char buf[128];
  19         gethostname( buf, 128 );
  20
  21         hostent* h;
  22         if( ( h = gethostbyname( buf ) ) == (void*)-1 )
  23         {
  24                 throw strerror( errno );
  25         }
  26         m_ipaddr = inet_ntoa( *((in_addr*)h->h_addr) );
  27 }
  28
  29 Listener::~Listener()
  30 {
  31         if( m_sock != 0 )
  32         {
  33                 std::cout << "[Listener] Closing socket" << std::endl;
  34                 close( m_sock );
  35         }
  36 }
  37
  38 void Listener::Listen()
  39 {
  40         std::cout << "[Listener] Opening socket" << std::endl;
  41
  42         if( ( m_sock = socket( PF_INET, SOCK_STREAM, 0 ) ) == -1 )
  43         {
  44                 throw strerror( errno );
  45         }
  46
  47         // Set socket to non-blocking
  48         if( fcntl( m_sock, F_SETFL, O_NONBLOCK ) == -1 )
  49         {
  50                 throw strerror( errno );
  51         }
  52
  53         // Reuse port
  54         int yes = 1;
  55         if( setsockopt( m_sock, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof( int ) ) == -1 )
  56         {
  57                 throw strerror( errno );
  58         }
  59
  60         sockaddr_in addr;
  61         addr.sin_family = AF_INET;
  62         addr.sin_port = htons( 21 );
  63         addr.sin_addr.s_addr = inet_addr( m_ipaddr.c_str() );
  64         memset( addr.sin_zero, 0, sizeof( addr.sin_zero ) );
  65
  66         if( bind( m_sock, (sockaddr*)&addr, sizeof( addr ) ) == -1 )
  67         {
  68                 throw strerror( errno );
  69         }
  70
  71         if( listen( m_sock, 5 ) == -1 )
  72         {
  73                 throw strerror( errno );
  74         }
  75 }
  76
  77 void Listener::Tick()
  78 {
  79         // Check if there are some incoming connections
  80         sockaddr_in addr;
  81         socklen_t size = sizeof( addr );
  82
  83         int incoming = accept( m_sock, (sockaddr*)&addr, &size );
  84         if( incoming == -1 )
  85         {
  86                 if( errno == EAGAIN )
  87                 {
  88                         // No error, just no connections in queue
  89                         return;
  90                 }
  91                 throw strerror( errno );
  92         }
  93
  94         ServerPtr server = m_server.lock();
  95         if( !server )
  96         {
  97                 throw "Listener lost server";
  98         }
  99         server->IncomingConnection( incoming );
 100 }