Colony

/**

 * @author Jason Knichel

 * @date 7/22/07

 */

#ifndef CONNECTION_POOL_H

#define CONNECTION_POOL_H

#include <sys/select.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 MAX_CONNECTIONS 250

#define READ_BUFFER_SIZE 1024

#define WRITE_BUFFER_SIZE 1024

class ConnectionPool {

public:

  ConnectionPool();

  ~ConnectionPool();

  int add_client(int client_file_descriptor);

  int remove_client(int client_file_descriptor);

  int check_clients();

  void set_listen_socket_in_ready_set(int listen_socket);

  int perform_select(int listen_socket, struct timeval * select_timeout);

  int is_socket_ready_to_read(int socket);

  int is_socket_ready_to_write(int socket);

  int get_max_file_descriptor();

  void set_max_file_descriptor(int new_max_file_descriptor);

  int get_next_available_slot();

  int get_number_clients_ready();

  void set_number_clients_ready(int new_number_clients_ready);

  fd_set get_ready_set();

  fd_set get_read_set();

  void set_read_set(fd_set new_set);

  fd_set get_write_set();

  void set_write_set(fd_set new_set);

  int get_write_buffer_size(int pool_index);

  void set_write_buffer_size(int pool_index, int new_buffer_size);

  char * get_write_buffer(int pool_index);

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 numTokens);

};

#endif