git.cworth.org Git - ttt/blob - src/ttt-server.c

   1 /* ttt-server.c - 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-server.h"
  23
  24 #include "ttt-args.h"
  25 #include "ttt-client.h"
  26 #include "ttt-error.h"
  27 #include "ttt-socket.h"
  28
  29 struct _ttt_server {
  30     pthread_mutex_t mutex;
  31
  32     const char *host;
  33     const char *port;
  34
  35     ttt_client_t **clients;
  36     int clients_size;
  37     int num_clients;
  38 };
  39
  40 static void
  41 ttt_server_init (ttt_server_t *server, const char *host, const char *port)
  42 {
  43     pthread_mutex_init (&server->mutex, NULL);
  44
  45     server->host = host;
  46     server->port = port;
  47
  48     server->clients = NULL;
  49     server->clients_size = 0;
  50     server->num_clients = 0;
  51 }
  52
  53 /* Exported: See ttt-server.h for documentation. */
  54 ttt_error_t
  55 ttt_server_register_client (ttt_server_t *server, ttt_client_t *client)
  56 {
  57     int i;
  58     ttt_error_t error = TTT_ERROR_NONE;
  59     const char *name;
  60
  61     pthread_mutex_lock (&server->mutex);
  62
  63     name = ttt_client_get_name (client);
  64
  65     assert (name != NULL);
  66
  67     if (name[0] == '\0')
  68         return TTT_ERROR_INVALID_NAME;
  69
  70     for (i = 0; i < server->num_clients; i++) {
  71         if (strcmp (ttt_client_get_name (server->clients[i]), name) == 0) {
  72             error = TTT_ERROR_INVALID_NAME;
  73             goto CLEANUP_LOCK;
  74         }
  75     }
  76
  77     printf ("Client %s has joined.\r\n", name);
  78
  79     server->num_clients++;
  80
  81     if (server->num_clients > server->clients_size) {
  82         if (server->clients_size == 0)
  83             server->clients_size = 1;
  84         else
  85             server->clients_size *= 2;
  86
  87         server->clients = xrealloc (server->clients,
  88                                     server->clients_size * sizeof (ttt_client_t *));
  89     }
  90
  91     server->clients [server->num_clients - 1] = client;
  92
  93  CLEANUP_LOCK:
  94     pthread_mutex_unlock (&server->mutex);
  95
  96     return error;
  97 }
  98
  99 /* Exported: See ttt-server.h for documentation. */
 100 void
 101 ttt_server_unregister_client (ttt_server_t *server, ttt_client_t *client)
 102 {
 103     int i;
 104
 105     pthread_mutex_lock (&server->mutex);
 106
 107     for (i = 0; i < server->num_clients; i++)
 108         if (server->clients[i] == client)
 109             break;
 110
 111     assert (i < server->num_clients);
 112
 113     printf ("Client %s has left.\r\n", ttt_client_get_name (client));
 114
 115     memmove (&server->clients[i], &server->clients[i+1],
 116              (server->num_clients - i - 1) * sizeof (ttt_client_t *));
 117
 118     server->num_clients--;
 119
 120     pthread_mutex_unlock (&server->mutex);
 121 }
 122
 123 /* Exported: See ttt-server.h for documentation. */
 124 void
 125 ttt_server_broadcast (ttt_server_t *server, const char *message)
 126 {
 127     int i;
 128
 129     pthread_mutex_lock (&server->mutex);
 130
 131     for (i = 0; i < server->num_clients; i++)
 132         ttt_client_send (server->clients[i], message);
 133
 134     pthread_mutex_unlock (&server->mutex);
 135 }
 136
 137 /* Exported: See ttt-server.h for documentation. */
 138 const char*
 139 ttt_server_who (ttt_server_t *server)
 140 {
 141     int i;
 142     char *response;
 143
 144     pthread_mutex_lock (&server->mutex);
 145
 146     xasprintf (&response, "WHO");
 147
 148     for (i = 0; i < server->num_clients; i++)
 149         xasprintf (&response, "%s %s %d",
 150                    response,
 151                    ttt_client_get_name (server->clients[i]),
 152                    ttt_client_get_num_wins (server->clients[i]));
 153
 154     xasprintf (&response, "%s\r\n", response);
 155
 156     pthread_mutex_unlock (&server->mutex);
 157
 158     return response;
 159 }
 160
 161 /* Exported: See ttt-server.h for documentation. */
 162 const char*
 163 ttt_server_get_host (ttt_server_t *server)
 164 {
 165     return server->host;
 166 }
 167
 168 /* Exported: See ttt-server.h for documentation. */
 169 const char*
 170 ttt_server_get_port (ttt_server_t *server)
 171 {
 172     return server->port;
 173 }
 174
 175 static const char *WELCOME_MESSAGE =
 176 "Welcome to ttt-server. The server is currently listening on:\r\n"
 177 "\r\n"
 178 "       %s:%s\r\n"
 179 "\r\n"
 180 "To test this, simply connect one or more clients to that host and port.\r\n"
 181 "For example:\r\n"
 182 "\r\n"
 183 "       telnet %s %s\r\n"
 184 "\r\n"
 185 "The TTTP (tic-tac-toe protocol) has been partially implemented.\r\n"
 186 "The following commands should work: HELO, HELP, MESSAGE, VERSION, QUIT, WHO.\r\n"
 187 "\r\n";
 188
 189 static void
 190 _ttt_server_accept (void *closure, int client_socket)
 191 {
 192     ttt_server_t *server = closure;
 193
 194     ttt_client_new (server, client_socket);
 195 }
 196
 197 int
 198 main (int argc, char **argv)
 199 {
 200     ttt_args_t args;
 201     ttt_server_t server;
 202     int socket;
 203
 204     ttt_args_parse (&args, argc, argv);
 205
 206     if (args.log_file)
 207         xfreopen (args.log_file, "a", stderr);
 208
 209     socket = ttt_socket_create_server (args.host, args.port);
 210
 211     printf (WELCOME_MESSAGE, args.host, args.port, args.host, args.port);
 212
 213     ttt_server_init (&server, args.host, args.port);
 214
 215     while (1)
 216         ttt_socket_accept (socket, _ttt_server_accept, &server);
 217
 218     /* We only reach here if something bad happened. */
 219     return 1;
 220 }