* TODO: checked off WHO, QUIT, ERROR SYNTAX

[ttt] / src / ttt-server.c

diff --git a/src/ttt-server.c b/src/ttt-server.c
index c50a797b259ec8bf1dbb6b74886bba2468ca4131..d98fc24e2891dafd8de6f34efe4fdffac3536edb 100644 (file)
--- a/src/ttt-server.c
+++ b/src/ttt-server.c
@@ -23,12 +23,14 @@
 
 #include "ttt-args.h"
 #include "ttt-client.h"
+#include "ttt-error.h"
 #include "ttt-socket.h"
 
 struct _ttt_server {
     pthread_mutex_t mutex;
 
-    int next_client_id;
+    const char *host;
+    const char *port;
 
     ttt_client_t **clients;
     int clients_size;
@@ -36,11 +38,12 @@ struct _ttt_server {
 };
 
 static void
-ttt_server_init (ttt_server_t *server)
+ttt_server_init (ttt_server_t *server, const char *host, const char *port)
 {
     pthread_mutex_init (&server->mutex, NULL);
 
-    server->next_client_id = 0;
+    server->host = host;
+    server->port = port;
 
     server->clients = NULL;
     server->clients_size = 0;
@@ -48,16 +51,30 @@ ttt_server_init (ttt_server_t *server)
 }
 
 /* Exported: See ttt-server.h for documentation. */
-int
+ttt_error_t
 ttt_server_register_client (ttt_server_t *server, ttt_client_t *client)
 {
-    int id;
+    int i;
+    ttt_error_t error = TTT_ERROR_NONE;
+    const char *name;
 
     pthread_mutex_lock (&server->mutex);
 
-    id = server->next_client_id++;
+    name = ttt_client_get_name (client);
+
+    assert (name != NULL);
 
-    printf ("Client %d has joined.\n", id);
+    if (name[0] == '\0')
+       return TTT_ERROR_INVALID_NAME;
+
+    for (i = 0; i < server->num_clients; i++) {
+       if (strcmp (ttt_client_get_name (server->clients[i]), name) == 0) {
+           error = TTT_ERROR_INVALID_NAME;
+           goto CLEANUP_LOCK;
+       }
+    }
+
+    printf ("Client %s has joined.\n", name);
 
     server->num_clients++;
 
@@ -73,9 +90,10 @@ ttt_server_register_client (ttt_server_t *server, ttt_client_t *client)
 
     server->clients [server->num_clients - 1] = client;
 
+ CLEANUP_LOCK:
     pthread_mutex_unlock (&server->mutex);
 
-    return id;
+    return error;
 }
 
 /* Exported: See ttt-server.h for documentation. */
@@ -92,7 +110,7 @@ ttt_server_unregister_client (ttt_server_t *server, ttt_client_t *client)
 
     assert (i < server->num_clients);
 
-    printf ("Client %d has left.\n", ttt_client_get_id (client));
+    printf ("Client %s has left.\n", ttt_client_get_name (client));
 
     memmove (&server->clients[i], &server->clients[i+1],
             (server->num_clients - i - 1) * sizeof (ttt_client_t *));
@@ -102,6 +120,7 @@ ttt_server_unregister_client (ttt_server_t *server, ttt_client_t *client)
     pthread_mutex_unlock (&server->mutex);
 }
 
+/* Exported: See ttt-server.h for documentation. */
 void
 ttt_server_broadcast (ttt_server_t *server, const char *message)
 {
@@ -115,6 +134,44 @@ ttt_server_broadcast (ttt_server_t *server, const char *message)
     pthread_mutex_unlock (&server->mutex);
 }
 
+/* Exported: See ttt-server.h for documentation. */
+const char*
+ttt_server_who (ttt_server_t *server)
+{
+    int i;
+    char *response;
+
+    pthread_mutex_lock (&server->mutex);
+
+    xasprintf (&response, "WHO");
+
+    for (i = 0; i < server->num_clients; i++)
+       xasprintf (&response, "%s %s %d",
+                  response,
+                  ttt_client_get_name (server->clients[i]),
+                  ttt_client_get_num_wins (server->clients[i]));
+
+    xasprintf (&response, "%s\n", response);
+
+    pthread_mutex_unlock (&server->mutex);
+
+    return response;
+}
+
+/* Exported: See ttt-server.h for documentation. */
+const char*
+ttt_server_get_host (ttt_server_t *server)
+{
+    return server->host;
+}
+
+/* Exported: See ttt-server.h for documentation. */
+const char*
+ttt_server_get_port (ttt_server_t *server)
+{
+    return server->port;
+}
+
 static const char *WELCOME_MESSAGE = 
 "Welcome to ttt-server. So far, this program is still a demonstration\n"
 "TCP/IP server, acting something like a rather braindead chat server.\n"
@@ -161,7 +218,7 @@ main (int argc, char **argv)
 
     printf (WELCOME_MESSAGE, args.host, args.port, args.host, args.port);
 
-    ttt_server_init (&server);
+    ttt_server_init (&server, args.host, args.port);
 
     while (1)
        ttt_socket_accept (socket, _ttt_server_accept, &server);