Colony

/**

 * @author Jason Knichel

 * @date 7/22/07

 */

#ifndef CONNECTION_POOL_H

#define CONNECTION_POOL_H

#include <sys/select.h>

#include <colonet_wireless.h>

#define MAX_TOKENS 15

#define MAX_TOKEN_SIZE 30

#define ROBOT_COMMAND_OFFSET 1

#define ROBOT_COMMAND_LEN    3

#define ERROR_INVALID_CLIENT_DESCRIPTOR -1

#define ERROR_TOO_MANY_CLIENTS          -2

#define ERROR_ALLOCATING_MEMORY         -3

#define ERROR_NOT_ENOUGH_ROOM           -4

#define ERROR_INVALID_COMMAND           -5

#define ERROR_INVALID_CLIENT_ID         -6

#define ERROR_INVALID_MESSAGE           -7

#define ERROR_INVALID_MESSAGE_LENGTH    -8

#define MAX_CONNECTIONS 250

#define READ_BUFFER_SIZE 1024

#define WRITE_BUFFER_SIZE 4096

class ConnectionPool {

public:

  ConnectionPool();

  ~ConnectionPool();

  //TODO: restructure the class to get rid of exposing so much stuff

  int add_client(int client_file_descriptor);

  int remove_client(int pool_index);

  int check_clients();

  int write_to_client(int pool_index, char * message, int length);

  void set_listen_socket_in_ready_set(int listen_socket);

  int perform_select(int listen_socket);

  int is_socket_ready_to_read(int socket);

  int get_max_file_descriptor();

  void set_max_file_descriptor(int new_max_file_descriptor);

  int get_number_clients_ready();

private:

  int max_file_descriptor;

  int next_available_slot;

  int number_clients_ready;

  fd_set ready_set;

  fd_set read_set;

  fd_set write_set;

  int client_file_descriptor_array[MAX_CONNECTIONS];

  char * read_buffer[MAX_CONNECTIONS];

  int read_buffer_size[MAX_CONNECTIONS];

  char * write_buffer[MAX_CONNECTIONS];

  int write_buffer_size[MAX_CONNECTIONS];

  int parse_command(char* command, int pool_index);

  int tokenize_command(char* command, char tokens[MAX_TOKENS][MAX_TOKEN_SIZE]);

  int check_tokens(unsigned char* tokens, int number_tokens);

};

#endif