X-Git-Url: https://git.cworth.org/git?p=ttt;a=blobdiff_plain;f=src%2Fttt-client.c;h=c198d3a15cfdb6779bbeea9f40b2ea7ddcfe3dae;hp=5f2a193179a0c644c410ed534549a186b06fd4cc;hb=HEAD;hpb=226be118d65765ba95dcd84d14b4162b5837e1a9 diff --git a/src/ttt-client.c b/src/ttt-client.c index 5f2a193..c198d3a 100644 --- a/src/ttt-client.c +++ b/src/ttt-client.c @@ -20,32 +20,34 @@ */ #include "ttt-client.h" - #include "ttt-error.h" -#include "ttt-lex.h" #include "ttt-server.h" #include "ttt-socket.h" #include "ttt-token.h" +#define YY_DECL int yylex (yyscan_t yyscanner, ttt_token_t *token) +#include "ttt-lex.h" +YY_DECL; + struct _ttt_client { pthread_mutex_t mutex; pthread_t thread; - + ttt_server_t *server; int socket; yyscan_t scanner; - + char **request_strings; int num_request_strings; - - char *name; + + char *username; ttt_bool_t registered; int num_wins; }; typedef ttt_error_t (*ttt_command_func_t) (ttt_client_t *client, - char **args, - int num_args); + char **args, + int num_args); static ttt_error_t _ttt_client_execute_helo (ttt_client_t *client, @@ -57,29 +59,93 @@ _ttt_client_execute_who (ttt_client_t *client, char **args, int num_args); +static ttt_error_t +_ttt_client_execute_statistics (ttt_client_t *client, + char **args, + int num_args); + +static ttt_error_t +_ttt_client_execute_message (ttt_client_t *client, + char **args, + int num_args); + +static ttt_error_t +_ttt_client_execute_help (ttt_client_t *client, + char **args, + int num_args); + +static ttt_error_t +_ttt_client_execute_version (ttt_client_t *client, + char **args, + int num_args); + static ttt_error_t _ttt_client_execute_quit (ttt_client_t *client, char **args, int num_args); static ttt_error_t -_ttt_client_execute_message (ttt_client_t *client, +_ttt_client_execute_invite (ttt_client_t *client, + char **args, + int num_args); + +static ttt_error_t +_ttt_client_execute_accept (ttt_client_t *client, + char **args, + int num_args); + +static ttt_error_t +_ttt_client_execute_retract (ttt_client_t *client, char **args, int num_args); - +static ttt_error_t +_ttt_client_execute_decline (ttt_client_t *client, + char **args, + int num_args); typedef struct _ttt_command_description { const char *command; - int args_required; + int args_min; + int args_max; ttt_command_func_t execute; + const char *usage; + const char *description; } ttt_command_description_t; ttt_command_description_t command_descriptions[] = { - {"HELO", 1, _ttt_client_execute_helo}, - {"WHO", 0, _ttt_client_execute_who }, - {"QUIT", 0, _ttt_client_execute_quit}, - {"MESSAGE", 1, _ttt_client_execute_message} + {"HELO", 1, 1, _ttt_client_execute_helo, + "HELO ", "Register."}, + + {"HELP", 0, 1, _ttt_client_execute_help, + "HELP ", "Display help for a command."}, + + {"INVITE", 1, 1, _ttt_client_execute_invite, + "INVITE ", "Invite a player to play a game."}, + + {"ACCEPT", 1, 1, _ttt_client_execute_accept, + "ACCEPT ", "Accept a game invitation."}, + + {"RETRACT", 1, 1, _ttt_client_execute_retract, + "RETRACT ", "Retract a game invitation."}, + + {"DECLINE", 1, 1, _ttt_client_execute_decline, + "DECLINE ", "Decline a game invitation."}, + + {"MESSAGE", 1, 1, _ttt_client_execute_message, + "MESSAGE ", "Send a message to everyone."}, + + {"STATISTICS", 1, 1, _ttt_client_execute_statistics, + "STATISTICS ", "Lists the statistics for the specified user."}, + + {"QUIT", 0, 0, _ttt_client_execute_quit, + "QUIT ", "Quit session."}, + + {"VERSION", 1, 1, _ttt_client_execute_version, + "VERSION ", "Negotiate protocol version."}, + + {"WHO", 0, 0, _ttt_client_execute_who, + "WHO ", "List registered users."} }; #define ARRAY_SIZE(arr) (sizeof(arr)/sizeof(arr[0])) @@ -89,29 +155,34 @@ _ttt_client_execute_helo (ttt_client_t *client, char **args, int num_args) { + char *response; + char *notice; ttt_error_t error; - char *response; assert (num_args == 1); - ttt_client_set_name (client, args[0]); + if (!(client->registered)) + { + ttt_client_set_username (client, args[0]); - error = ttt_server_register_client (client->server, client); - if (error) - return error; + error = ttt_server_register_client (client->server, client); + if (error) + return error; + client->registered = TRUE; - client->registered = TRUE; + xasprintf (¬ice, "NOTICE USER %s\r\n", + ttt_client_get_username (client)); + ttt_server_broadcast (client->server, notice); + + free (notice); + } xasprintf (&response, "HELO %s %s %s\r\n", - client->name, + ttt_client_get_username (client), ttt_server_get_host (client->server), ttt_server_get_port (client->server)); - ttt_client_send (client, response); - xasprintf (&response, "NOTICE USER %s\r\n",client->name); - ttt_server_broadcast(client->server,response); - free (response); return TTT_ERROR_NONE; @@ -126,8 +197,10 @@ _ttt_client_execute_who (ttt_client_t *client, assert (num_args == 0); - response = ttt_server_who (client->server); + if (!client->registered) + return TTT_ERROR_NO_NAME_SET; + response = xstrdup (ttt_server_who (client->server)); ttt_client_send (client, response); free (response); @@ -135,6 +208,158 @@ _ttt_client_execute_who (ttt_client_t *client, return TTT_ERROR_NONE; } +static ttt_error_t +_ttt_client_execute_statistics (ttt_client_t *client, + char **args, + int num_args) +{ + const char *argusername; + ttt_client_t *argclient; + char *response; + ttt_error_t error; + + assert (num_args == 1); + + argusername = args[0]; + + if (!client->registered) + return TTT_ERROR_NO_NAME_SET; + + error = ttt_server_get_client_from_username (client->server, + argusername, + &argclient); + if (error) + return error; + + xasprintf (&response, "STATISTICS %s \"\r\n" + "TICTACTOE WINS %d\r\n" + "\"\r\n", + ttt_client_get_username(argclient), + ttt_client_get_num_wins(argclient)); + + ttt_client_send (client, response); + + free (response); + + return TTT_ERROR_NONE; +} + +static ttt_error_t +_ttt_client_execute_message (ttt_client_t *client, + char **args, + int num_args) +{ + const char *argmessage; + char *response; + char *notice; + + assert (num_args == 1); + + argmessage = args[0]; + + if (!client->registered) + return TTT_ERROR_NO_NAME_SET; + + xasprintf (&response, "MESSAGE\r\n"); + ttt_client_send (client, response); + + xasprintf (¬ice, "NOTICE MESSAGE %s \"%s\"\r\n", + ttt_client_get_username(client), + argmessage); + ttt_server_broadcast (client->server, notice); + + free (notice); + free (response); + + return TTT_ERROR_NONE; +} + +static ttt_error_t +_ttt_client_execute_help (ttt_client_t *client, + char **args, + int num_args) +{ + char *response; + char *argcommand; + ttt_command_description_t *desc; + ttt_bool_t is_command = FALSE; + int i; + + if (num_args == 0) { + xasprintf (&response, "HELP \"\r\n" + "Available Commands:\r\n"); + for (i = 0; i < ARRAY_SIZE(command_descriptions); i++) { + desc = &command_descriptions[i]; + xasprintf (&response, "%s\r\n %s - %s\r\n", + response, + desc->usage, + desc->description); + } + xasprintf (&response, "%s\"\r\n", response); + } else { + argcommand = args[0]; + for (i = 0; i < ARRAY_SIZE(command_descriptions); i++) { + desc = &command_descriptions[i]; + if (strcasecmp (desc->command, argcommand) == 0) { + is_command = TRUE; + xasprintf (&response, "HELP %s \"\r\n" + "%s\r\n" + "\r\n" + "Usage:\r\n" + " %s\r\n" + "\"\r\n", + desc->command, + desc->description, + desc->usage); + /* XXX: Add detailed help. */ + } + } + } + + if ((num_args == 1) && (!is_command)) + return TTT_ERROR_SYNTAX; + + ttt_client_send (client, response); + + free (response); + return TTT_ERROR_NONE; +} + +static ttt_error_t +_ttt_client_execute_version (ttt_client_t *client, + char **args, + int num_args) +{ + char *response; + char *argversion; + int version; + int i; + + assert (num_args == 1); + + argversion = args[0]; + + /* Verify that provided version arg is a positive integer */ + for (i = 0; i < strlen(argversion); i++) + if (!isdigit (argversion[i])) + return TTT_ERROR_SYNTAX; + + version = atoi (argversion); + + if (version < 1) + return TTT_ERROR_SYNTAX; + + if (version > TTT_SERVER_PROTOCOL_VERSION) + version = TTT_SERVER_PROTOCOL_VERSION; + + xasprintf (&response, "VERSION %d\r\n", + version); + ttt_client_send (client, response); + + free (response); + return TTT_ERROR_NONE; +} + static ttt_error_t _ttt_client_execute_quit (ttt_client_t *client, char **args, @@ -143,37 +368,193 @@ _ttt_client_execute_quit (ttt_client_t *client, char *notice; assert (num_args == 0); - xasprintf (¬ice,"NOTICE QUIT %s\r\n",client->name); - ttt_server_broadcast(client->server,notice); + + if (!client->registered) + return TTT_ERROR_QUIT_REQUESTED; + + xasprintf (¬ice, "NOTICE QUIT %s\r\n", + ttt_client_get_username(client)); + ttt_server_broadcast (client->server, notice); + free (notice); + return TTT_ERROR_QUIT_REQUESTED; } static ttt_error_t -_ttt_client_execute_message (ttt_client_t *client, - char **args, - int num_args) +_ttt_client_execute_invite (ttt_client_t *client, + char **args, + int num_args) { - char *response; - int i; + const char *invitee_username; + ttt_client_t *invitee; + char *response; + char *notice; + ttt_error_t error; + + assert (num_args == 1); + + invitee_username = args[0]; + + if (!client->registered) + return TTT_ERROR_NO_NAME_SET; + + error = ttt_server_get_client_from_username (client->server, + invitee_username, + &invitee); + if (error) + return error; - assert (num_args > 0); + ttt_server_add_invite (client->server, + client, + invitee); - xasprintf(&response, "MESSAGE\r\n"); + xasprintf (&response, "INVITE\r\n"); ttt_client_send (client, response); + + xasprintf (¬ice, "NOTICE INVITE %s %s\r\n", + ttt_client_get_username(client), + ttt_client_get_username(invitee)); + ttt_server_broadcast (client->server, notice); + + free (notice); + free (response); + + return TTT_ERROR_NONE; +} + +static ttt_error_t +_ttt_client_execute_accept (ttt_client_t *client, + char **args, + int num_args) +{ + const char *actor_username; + ttt_client_t *actor; + char *response; + char *notice; + ttt_error_t error; + + assert (num_args == 1); + + actor_username = args[0]; + + if (!client->registered) + return TTT_ERROR_NO_NAME_SET; + + error = ttt_server_get_client_from_username (client->server, + actor_username, + &actor); + if (error) + return error; + + error = ttt_server_remove_invite (client->server, + actor, + client); + if (error) + return error; + + xasprintf (&response, "ACCEPT\r\n"); + ttt_client_send (client, response); + + xasprintf (¬ice, "NOTICE ACCEPT %s %s\r\n", + ttt_client_get_username(client), + ttt_client_get_username(actor)); + ttt_server_broadcast (client->server, notice); + + /* XXX: Start a new game */ + + free (notice); + free (response); + + return TTT_ERROR_NONE; +} + +static ttt_error_t +_ttt_client_execute_retract (ttt_client_t *client, + char **args, + int num_args) +{ + const char *invitee_username; + ttt_client_t *invitee; + char *response; + char *notice; + ttt_error_t error; + + assert (num_args == 1); + + invitee_username = args[0]; + + if (!client->registered) + return TTT_ERROR_NO_NAME_SET; + + error = ttt_server_get_client_from_username (client->server, + invitee_username, + &invitee); + if (error) + return error; + + error = ttt_server_remove_invite (client->server, + client, + invitee); + if (error) + return error; - xasprintf(&response, "NOTICE MESSAGE %s", client->name); - for (i=0; iserver,response); + xasprintf (¬ice, "NOTICE RETRACT %s %s\r\n", + ttt_client_get_username(client), + ttt_client_get_username(invitee)); + ttt_server_broadcast (client->server, notice); + free (notice); free (response); + return TTT_ERROR_NONE; +} + +static ttt_error_t +_ttt_client_execute_decline (ttt_client_t *client, + char **args, + int num_args) +{ + const char *actor_username; + ttt_client_t *actor; + char *response; + char *notice; + ttt_error_t error; + + assert (num_args == 1); + + actor_username = args[0]; + + if (!client->registered) + return TTT_ERROR_NO_NAME_SET; + + error = ttt_server_get_client_from_username (client->server, + actor_username, + &actor); + if (error) + return error; + + error = ttt_server_remove_invite (client->server, + actor, + client); + if (error) + return error; + + xasprintf (&response, "DECLINE\r\n"); + ttt_client_send (client, response); + xasprintf (¬ice, "NOTICE DECLINE %s %s\r\n", + ttt_client_get_username(client), + ttt_client_get_username(actor)); + ttt_server_broadcast (client->server, notice); + free (notice); + free (response); + + return TTT_ERROR_NONE; } static void @@ -192,13 +573,13 @@ _ttt_client_init (ttt_client_t *client, client->socket = socket; file = xfdopen (socket, "r"); - yylex_init(&client->scanner); + yylex_init (&client->scanner); yyset_in (file, client->scanner); client->request_strings = NULL; client->num_request_strings = 0; - client->name = NULL; + client->username = NULL; client->registered = FALSE; client->num_wins = 0; } @@ -211,8 +592,8 @@ _ttt_client_fini (ttt_client_t *client) if (client->registered) ttt_server_unregister_client (client->server, client); - free (client->name); - client->name = NULL; + free (client->username); + client->username = NULL; yylex_destroy (client->scanner); shutdown (client->socket, SHUT_RDWR); @@ -254,49 +635,53 @@ _free_request (ttt_client_t *client) static ttt_status_t _read_request (ttt_client_t *client) { - ttt_token_t token; + ttt_token_t token; + ttt_token_type_t token_type; _free_request (client); while (1) { - token = yylex (client->scanner); + token_type = yylex (client->scanner, &token); /* Yes, EOF in two different enums is pretty ugly. */ - if (token == TTT_TOKEN_EOF) + if (token_type == TTT_TOKEN_TYPE_EOF) return TTT_STATUS_EOF; - if (token == TTT_TOKEN_NEWLINE) { + if (token_type == TTT_TOKEN_TYPE_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)); + assert (token_type == TTT_TOKEN_TYPE_STRING); + + _append_to_request (client, token.u.string); + + free (token.u.string); } } 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; + char *command = client->request_strings[0]; + int num_args = client->num_request_strings - 1; ttt_command_description_t *desc; + int i; - for (i=0; i < strlen (command); i++) + for (i = 0; i < strlen (command); i++) command[i] = toupper (command[i]); - for (i=0; i < ARRAY_SIZE(command_descriptions); 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_required) + 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 (desc->execute) (client, + &client->request_strings[1], + num_args); } + } return TTT_ERROR_COMMAND; } @@ -305,11 +690,10 @@ static void * _handle_requests_thread (void *closure) { ttt_status_t status; - ttt_error_t error; ttt_client_t *client = closure; + ttt_error_t error; while (1) { - status = _read_request (client); if (status == TTT_STATUS_EOF) break; @@ -363,17 +747,17 @@ ttt_client_send (ttt_client_t *client, const char *message) /* Exported: See ttt-client.h for documentation. */ const char* -ttt_client_get_name (ttt_client_t *client) +ttt_client_get_username (ttt_client_t *client) { - return client->name; + return client->username; } /* Exported: See ttt-client.h for documentation. */ void -ttt_client_set_name (ttt_client_t *client, const char *name) +ttt_client_set_username (ttt_client_t *client, const char *username) { - free (client->name); - client->name = xstrdup (name); + free (client->username); + client->username = xstrdup (username); } /* Exported: See ttt-client.h for documentation. */ @@ -382,3 +766,9 @@ ttt_client_get_num_wins (ttt_client_t *client) { return client->num_wins; } + +/* This is just to keep the compiler quiet about a function declared + * static but never defined. Just an annoying bug in flex's output. */ +static int yy_init_globals (yyscan_t yyscanner) {return 0;} +void use_yy_init_globals (void); +void use_yy_init_globals (void) {yyscan_t scan; yy_init_globals(scan);}