Colony

/** @file ColonetServer.cpp

 *

 * @brief colonet_server - primary server application for Colonet

 *

 * @author Jason Knichel

 * @author Eugene Marinelli

 * @date 10/31/06

 */

#include <colonet_wireless.h>

#include <sys/select.h>

#include <sys/socket.h>

#include <netinet/in.h>

#include <errno.h>

#include <arpa/inet.h>

#include "includes/initialization.h"

#include "includes/ConnectionPool.h"

#include "includes/client.h"

#include "includes/options.h"

#include "includes/Logging.h"

#define LISTEN_BACKLOG 5

#define LOG_BUFFER_LENGTH 128

int listenSocket = 0;

ConnectionPool connection_pool;

Log * logger;

int main(int argc, char** argv) {

  if (initializeServer(argc, argv) < 0) {

    printf("\t\nThere was an error initializing the server. "

           "Terminating server...\n");

    return -1;

  }

  if (listen(listenSocket, LISTEN_BACKLOG) < 0) {

    printf("\t\nThere was an error telling the socket to "

           "listen for connections from clients.  Terminating Server...\n");

    return -1;

  }

  connection_pool.set_listen_socket_in_ready_set(listenSocket);

  //TODO: why is all of this in that if statement and not just conn_pool.maxfd = listenSocket ?

  if (listenSocket > connection_pool.get_max_file_descriptor()) {

    connection_pool.set_max_file_descriptor(listenSocket);

    int acceptSocket = 0;

    struct sockaddr_in clientAddr;

    socklen_t clen = sizeof(clientAddr);

    struct timeval selectTimeout;

    memset(&selectTimeout,0,sizeof(selectTimeout));

    logger->logMessage(LOG_TYPE_MESSAGE, "Server initialized.  About to start listening for connections");

    while(1) {

      connection_pool.perform_select(listenSocket, &selectTimeout);

      //either no descriptors are ready or there was an error

      //TODO: check for specific errors

      if (connection_pool.get_number_clients_ready() <= 0) {

        continue;

      }

      if (connection_pool.is_socket_ready_to_read(listenSocket)) {

        printf("Something is trying to connect...\n");

        if ((acceptSocket = accept(listenSocket, (struct sockaddr*) &clientAddr, &clen)) < 0) {

          if (errno == EMFILE) {

            printf("\tWhen attempting to accept a connection, "

                   "reached the per process limit of file descriptors."

                   "  Dropping the new connection.\n");

            continue;

          } else {

            printf("\tThere was an error when attempting to accept a "

                   "connection");

          }

          continue;

        }

        char logBuffer[LOG_BUFFER_LENGTH];

        snprintf(logBuffer, LOG_BUFFER_LENGTH, "Client at address %s attempting to connect.", inet_ntoa(clientAddr.sin_addr));

        logger->logMessage(LOG_TYPE_CONNECT, logBuffer);

        //TODO: remove this

        //printf("Attempting to add a client.\n");

        if (connection_pool.add_client(acceptSocket) < 0) {

          printf("\tThere was an error when trying to add a client to the "

                 "connection pool.");

          continue;

        }

        snprintf(logBuffer, LOG_BUFFER_LENGTH, "Client at address %s successfully added to connection pool.", inet_ntoa(clientAddr.sin_addr));

        logger->logMessage(LOG_TYPE_CONNECT, logBuffer);

      }

      if (connection_pool.check_clients() < 0) {

        printf("\tThere was an error trying to update the clients.");

        continue;

      }

    }

  }

   return 0;

}