+#include "ttt-error.h"
+#include "ttt-lex.h"
+#include "ttt-server.h"
+#include "ttt-socket.h"
+#include "ttt-token.h"
+
+struct _ttt_client {
+ pthread_mutex_t mutex;
+ pthread_t thread;
+
+ ttt_server_t *server;
+ int socket;
+ yyscan_t scanner;
+
+ char *name;
+
+ char **request_strings;
+ int num_request_strings;
+};
+
+typedef ttt_error_t (*ttt_command_func_t) (ttt_client_t *client,
+ char **args,
+ int num_args);
+
+static ttt_error_t
+_ttt_client_execute_helo (ttt_client_t *client,
+ char **args,
+ int num_args);
+
+typedef struct _ttt_command_description {
+ const char *command;
+ int args_min;
+ int args_max;
+ ttt_command_func_t execute;
+} ttt_command_description_t;
+
+ttt_command_description_t command_descriptions[] = {
+ {"HELO", 1, 1, _ttt_client_execute_helo}
+};
+
+static ttt_error_t
+_ttt_client_execute_helo (ttt_client_t *client,
+ char **args,
+ int num_args)
+{
+ ttt_error_t error;
+ char *response;
+
+ if (num_args == 1)
+ ttt_client_set_name (client, args[0]);
+
+ error = ttt_server_register_client (client->server, client);
+ if (error)
+ return error;
+
+ xasprintf (&response, "HELO %s %s %s\n", client->name,
+ ttt_server_get_host (client->server),
+ ttt_server_get_port (client->server));
+
+ ttt_client_send (client, response);
+
+ free (response);
+
+ return TTT_ERROR_NONE;
+}
+
+#define ARRAY_SIZE(arr) (sizeof(arr)/sizeof(arr[0]))
+
+static void
+_free_request (ttt_client_t *client);
+
+static void
+_ttt_client_init (ttt_client_t *client,
+ ttt_server_t *server,
+ int socket)
+{
+ FILE *file;
+
+ pthread_mutex_init (&client->mutex, NULL);
+
+ client->server = server;
+ client->socket = socket;
+
+ file = xfdopen (socket, "r");
+ yylex_init(&client->scanner);
+ yyset_in (file, client->scanner);
+
+ client->request_strings = NULL;
+ client->num_request_strings = 0;
+
+ client->name = NULL;
+}
+
+static void
+_ttt_client_fini (ttt_client_t *client)
+{
+ pthread_mutex_lock (&client->mutex);
+
+ ttt_server_unregister_client (client->server, client);
+
+ free (client->name);
+ client->name = NULL;
+
+ yylex_destroy (client->scanner);
+ shutdown (client->socket, SHUT_RDWR);
+
+ _free_request (client);
+
+ pthread_mutex_unlock (&client->mutex);
+
+ pthread_mutex_destroy (&client->mutex);
+}
+
+/* XXX: The memory management for the request strings is pretty cheesy. */
+static void
+_append_to_request (ttt_client_t *client,
+ const char *string)
+{
+ client->num_request_strings++;
+ client->request_strings =
+ xrealloc (client->request_strings,
+ client->num_request_strings * sizeof (char *));
+
+ client->request_strings[client->num_request_strings - 1] = xstrdup (string);
+}
+
+static void
+_free_request (ttt_client_t *client)
+{
+ int i;
+
+ for (i = 0; i < client->num_request_strings; i++)
+ free (client->request_strings[i]);
+
+ free (client->request_strings);
+
+ client->request_strings = NULL;
+ client->num_request_strings = 0;
+}
+
+static ttt_status_t
+_read_request (ttt_client_t *client)
+{
+ ttt_token_t token;
+
+ _free_request (client);
+
+ while (1) {
+ token = yylex (client->scanner);
+ /* Yes, EOF in two different enums is pretty ugly. */
+ if (token == TTT_TOKEN_EOF)
+ return TTT_STATUS_EOF;
+
+ if (token == TTT_TOKEN_NEWLINE) {
+ if (client->num_request_strings)
+ return TTT_STATUS_SUCCESS;
+ else
+ continue;
+ }
+
+ assert (token == TTT_TOKEN_STRING);
+
+ _append_to_request (client, yyget_text (client->scanner));
+ }
+}
+
+static ttt_error_t
+_execute_request (ttt_client_t *client)
+{
+ int i;
+
+ char *command = client->request_strings[0];
+ int num_args = client->num_request_strings-1;
+ ttt_command_description_t *desc;
+
+ for (i=0; i < strlen (command); i++)
+ command[i] = toupper (command[i]);
+
+ for (i=0; i < ARRAY_SIZE(command_descriptions); i++) {
+ desc = &command_descriptions[i];
+ if (strcmp(command, desc->command) == 0) {
+ if ((num_args < desc->args_min) || (num_args > desc->args_max))
+ return TTT_ERROR_SYNTAX;
+ return (desc->execute) (client, &client->request_strings[1], num_args);
+ }
+ }
+
+ return TTT_ERROR_COMMAND;
+}
+
+static void *
+_handle_requests_thread (void *closure)
+{
+ ttt_status_t status;
+ ttt_error_t error;
+ ttt_client_t *client = closure;
+
+ while (1) {
+
+ status = _read_request (client);
+ if (status == TTT_STATUS_EOF)
+ break;
+ if (status)
+ ASSERT_NOT_REACHED;
+
+ error = _execute_request (client);
+ if (error)
+ ttt_client_send (client, ttt_error_string (error));
+ }
+
+ _ttt_client_fini (client);
+ free (client);
+
+ return (void *) 0;
+}
+
+/* Exported: See ttt-client.h for documentation. */
+void
+ttt_client_new (void *closure, int client_socket)