]> git.cworth.org Git - ttt/blobdiff - src/ttt-client.c
Add a dependency of ttt-client.c on ttt-lex.h to fix the build.
[ttt] / src / ttt-client.c
index 7b37fad3c5ecade9e6acc9ec6bd98a78c5f9b6d6..c198d3a15cfdb6779bbeea9f40b2ea7ddcfe3dae 100644 (file)
  */
 
 #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 <username>         ", "Register."},
+
+    {"HELP",    0, 1, _ttt_client_execute_help,
+     "HELP <command>          ", "Display help for a command."},
+
+    {"INVITE",  1, 1, _ttt_client_execute_invite,
+     "INVITE <username>       ", "Invite a player to play a game."},
+
+    {"ACCEPT",  1, 1, _ttt_client_execute_accept,
+     "ACCEPT <username>       ", "Accept a game invitation."},
+
+    {"RETRACT", 1, 1, _ttt_client_execute_retract,
+     "RETRACT <username>      ", "Retract a game invitation."},
+
+    {"DECLINE", 1, 1, _ttt_client_execute_decline,
+     "DECLINE <username>      ", "Decline a game invitation."},
+
+    {"MESSAGE", 1, 1, _ttt_client_execute_message,
+     "MESSAGE <message>       ", "Send a message to everyone."},
+
+    {"STATISTICS", 1, 1, _ttt_client_execute_statistics,
+     "STATISTICS <username>   ", "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 <client-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 (&notice, "NOTICE USER %s\r\n",
+                  ttt_client_get_username (client));
+       ttt_server_broadcast (client->server, notice);
 
-    xasprintf (&response, "HELO %s %s %s\n",
-              client->name,
+       free (notice);
+    }
+
+    xasprintf (&response, "HELO %s %s %s\r\n",
+              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\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 (&notice, "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 (&notice,"NOTICE QUIT %s\n",client->name);
-    ttt_server_broadcast(client->server,notice);
+
+    if (!client->registered)
+       return TTT_ERROR_QUIT_REQUESTED;
+    
+    xasprintf (&notice, "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\n");
+    xasprintf (&response, "INVITE\r\n");
     ttt_client_send (client, response);
+
+    xasprintf (&notice, "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 (&notice, "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; i<num_args; i++){
-       xasprintf(&response, "%s %s",response,args[i]);
+    xasprintf (&response, "RETRACT\r\n");
+    ttt_client_send (client, response);
 
-    }
-    xasprintf(&response, "%s\n",response);
-    ttt_server_broadcast (client->server,response);
+    xasprintf (&notice, "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 (&notice, "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;
@@ -344,7 +728,7 @@ ttt_client_new (void *closure, int client_socket)
     err = pthread_create (&client->thread, NULL,
                          _handle_requests_thread, client);
     if (err != 0) {
-       fprintf (stderr, "Error: pthread_create failed: %s. Aborting.\n",
+       fprintf (stderr, "Error: pthread_create failed: %s. Aborting.\r\n",
                 strerror (err));
        exit (1);
     }
@@ -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);}