2005-11-24 Carl Worth <cworth@cworth.org>
[ttt] / src / ttt-client.c
1 /* ttt-client.c - client handling code for tic-tac-toe game server
2  *
3  * Copyright © 2005 Carl Worth
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License as published by
7  * the Free Software Foundation; either version 2, or (at your option)
8  * any later version.
9  *
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public License
16  * along with this program; if not, write to the Free Software Foundation,
17  * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
18  *
19  * Author: Carl Worth <cworth@cworth.org>
20  */
21
22 #include "ttt-client.h"
23
24 #include "ttt-error.h"
25 #include "ttt-lex.h"
26 #include "ttt-server.h"
27 #include "ttt-socket.h"
28 #include "ttt-token.h"
29
30 struct _ttt_client {
31     pthread_mutex_t mutex;
32     pthread_t thread;
33
34     ttt_server_t *server;
35     int socket;
36     yyscan_t scanner;
37
38     char *name;
39
40     char **request_strings;
41     int num_request_strings;
42 };
43
44 typedef ttt_error_t (*ttt_command_func_t) (ttt_client_t *client,
45                                            char **args,
46                                            int num_args);
47
48 static ttt_error_t
49 _ttt_client_execute_helo (ttt_client_t *client,
50                           char         **args,
51                           int          num_args);
52
53 typedef struct _ttt_command_description {
54     const char *command;
55     int args_min;
56     int args_max;
57     ttt_command_func_t execute;
58 } ttt_command_description_t;
59
60 ttt_command_description_t command_descriptions[] = {
61     {"HELO", 1, 1, _ttt_client_execute_helo}
62 };
63
64 static ttt_error_t
65 _ttt_client_execute_helo (ttt_client_t *client,
66                           char         **args,
67                           int          num_args)
68 {
69     ttt_error_t error;
70     char *response;
71
72     if (num_args == 1)
73         ttt_client_set_name (client, args[0]);
74
75     error = ttt_server_register_client (client->server, client);
76     if (error)
77         return error;
78
79     xasprintf (&response, "HELO %s %s %s\n", client->name,
80                ttt_server_get_host (client->server),
81                ttt_server_get_port (client->server));
82
83     ttt_client_send (client, response);
84
85     free (response);
86
87     return TTT_ERROR_NONE;
88 }
89
90 #define ARRAY_SIZE(arr) (sizeof(arr)/sizeof(arr[0]))
91
92 static void
93 _free_request (ttt_client_t *client);
94
95 static void
96 _ttt_client_init (ttt_client_t  *client,
97                   ttt_server_t  *server,
98                   int            socket)
99 {
100     FILE *file;
101
102     pthread_mutex_init (&client->mutex, NULL);
103
104     client->server = server;
105     client->socket = socket;
106
107     file = xfdopen (socket, "r");
108     yylex_init(&client->scanner);
109     yyset_in (file, client->scanner);
110
111     client->request_strings = NULL;
112     client->num_request_strings = 0;
113
114     client->name = NULL;
115 }
116
117 static void
118 _ttt_client_fini (ttt_client_t *client)
119 {
120     pthread_mutex_lock (&client->mutex);
121
122     ttt_server_unregister_client (client->server, client);
123
124     free (client->name);
125     client->name = NULL;
126
127     yylex_destroy (client->scanner);
128     shutdown (client->socket, SHUT_RDWR);
129
130     _free_request (client);
131
132     pthread_mutex_unlock (&client->mutex);
133
134     pthread_mutex_destroy (&client->mutex);
135 }
136
137 /* XXX: The memory management for the request strings is pretty cheesy. */
138 static void
139 _append_to_request (ttt_client_t        *client,
140                     const char          *string)
141 {
142     client->num_request_strings++;
143     client->request_strings =
144         xrealloc (client->request_strings,
145                   client->num_request_strings * sizeof (char *));
146
147     client->request_strings[client->num_request_strings - 1] = xstrdup (string);
148 }
149
150 static void
151 _free_request (ttt_client_t *client)
152 {
153     int i;
154
155     for (i = 0; i < client->num_request_strings; i++)
156         free (client->request_strings[i]);
157
158     free (client->request_strings);
159
160     client->request_strings = NULL;
161     client->num_request_strings = 0;
162 }
163
164 static ttt_status_t
165 _read_request (ttt_client_t *client)
166 {
167     ttt_token_t token;
168
169     _free_request (client);
170
171     while (1) {
172         token = yylex (client->scanner);
173         /* Yes, EOF in two different enums is pretty ugly. */
174         if (token == TTT_TOKEN_EOF)
175             return TTT_STATUS_EOF;
176
177         if (token == TTT_TOKEN_NEWLINE) {
178             if (client->num_request_strings)
179                 return TTT_STATUS_SUCCESS;
180             else
181                 continue;
182         }
183
184         assert (token == TTT_TOKEN_STRING);
185
186         _append_to_request (client, yyget_text (client->scanner));
187     }
188 }
189
190 static ttt_error_t
191 _execute_request (ttt_client_t *client)
192 {
193     int i;
194
195     char *command = client->request_strings[0];
196     int num_args = client->num_request_strings-1;
197     ttt_command_description_t *desc;
198
199     for (i=0; i < strlen (command); i++)
200         command[i] = toupper (command[i]);
201
202     for (i=0; i < ARRAY_SIZE(command_descriptions); i++) {
203         desc = &command_descriptions[i];
204         if (strcmp(command, desc->command) == 0) {
205             if ((num_args < desc->args_min) || (num_args > desc->args_max))
206                 return TTT_ERROR_SYNTAX;
207             return (desc->execute) (client, &client->request_strings[1], num_args);
208             }
209         }
210
211     return TTT_ERROR_COMMAND;
212 }
213
214 static void *
215 _handle_requests_thread (void *closure)
216 {
217     ttt_status_t status;
218     ttt_error_t error;
219     ttt_client_t *client = closure;
220
221     while (1) {
222
223         status = _read_request (client);
224         if (status == TTT_STATUS_EOF)
225             break;
226         if (status)
227             ASSERT_NOT_REACHED;
228
229         error = _execute_request (client);
230         if (error)
231             ttt_client_send (client, ttt_error_string (error));
232     }
233
234     _ttt_client_fini (client);
235     free (client);
236
237     return (void *) 0;
238 }
239
240 /* Exported: See ttt-client.h for documentation. */
241 void
242 ttt_client_new (void *closure, int client_socket)
243 {
244     ttt_server_t *server = closure;
245     ttt_client_t *client;
246     int err;
247
248     client = xmalloc (sizeof (ttt_client_t));
249     
250     _ttt_client_init (client, server, client_socket);
251
252     err = pthread_create (&client->thread, NULL,
253                           _handle_requests_thread, client);
254     if (err != 0) {
255         fprintf (stderr, "Error: pthread_create failed: %s. Aborting.\n",
256                  strerror (err));
257         exit (1);
258     }
259 }
260
261 /* Exported: See ttt-client.h for documentation. */
262 void
263 ttt_client_send (ttt_client_t *client, const char *message)
264 {
265     pthread_mutex_lock (&client->mutex);
266
267     ttt_socket_write (client->socket, message, strlen (message));
268
269     pthread_mutex_unlock (&client->mutex);
270 }
271
272 /* Exported: See ttt-client.h for documentation. */
273 const char*
274 ttt_client_get_name (ttt_client_t *client)
275 {
276     return client->name;
277 }
278
279 /* Exported: See ttt-client.h for documentation. */
280 void
281 ttt_client_set_name (ttt_client_t *client, const char *name)
282 {
283     free (client->name);
284     client->name = xstrdup (name);
285 }