From d9e7ebde2fad1dc4a872d40ebfbe449ee9288947 Mon Sep 17 00:00:00 2001
From: Michael Jumper <mike.jumper@guac-dev.org>
Date: Sat, 5 Jul 2014 13:26:27 -0700
Subject: [PATCH] GUAC-743: Store and maintain connections within the client
 map structure.

---
 src/guacd/daemon.c | 19 +++++++++++++++++--
 1 file changed, 17 insertions(+), 2 deletions(-)

diff --git a/src/guacd/daemon.c b/src/guacd/daemon.c
index c854edaf..7265af60 100644
--- a/src/guacd/daemon.c
+++ b/src/guacd/daemon.c
@@ -23,6 +23,7 @@
 #include "config.h"
 
 #include "client.h"
+#include "client-map.h"
 #include "log.h"
 
 #include <guacamole/client.h>
@@ -56,7 +57,11 @@
 #define GUACD_DEV_NULL "/dev/null"
 #define GUACD_ROOT     "/"
 
-void guacd_handle_connection(guac_socket* socket) {
+/**
+ * Creates a new guac_client for the connection on the given socket, adding
+ * it to the client map based on its ID.
+ */
+static void guacd_handle_connection(guacd_client_map* map, guac_socket* socket) {
 
     guac_client* client;
     guac_client_plugin* plugin;
@@ -216,6 +221,10 @@ void guacd_handle_connection(guac_socket* socket) {
             sizeof(char*) * video->argc);
     client->info.video_mimetypes[video->argc] = NULL;
 
+    /* Store client */
+    if (guacd_client_map_add(map, client))
+        guacd_log_error("Unable to add client. Internal client storage has failed");
+
     /* Send connection ID */
     guacd_log_info("Connection ID is \"%s\"", client->connection_id);
     guac_protocol_send_ready(socket, client->connection_id);
@@ -247,6 +256,10 @@ void guacd_handle_connection(guac_socket* socket) {
     else
         guacd_log_info("Client finished normally");
 
+    /* Remove client */
+    if (guacd_client_map_remove(map, client->connection_id) == NULL)
+        guacd_log_error("Unable to remove client. Internal client storage has failed");
+
     /* Free mimetype lists */
     free(client->info.audio_mimetypes);
     free(client->info.video_mimetypes);
@@ -376,6 +389,8 @@ int main(int argc, char* argv[]) {
     SSL_CTX* ssl_context = NULL;
 #endif
 
+    guacd_client_map* map = guacd_client_map_alloc();
+
     /* General */
     int retval;
 
@@ -645,7 +660,7 @@ int main(int argc, char* argv[]) {
             socket = guac_socket_open(connected_socket_fd);
 #endif
 
-            guacd_handle_connection(socket);
+            guacd_handle_connection(map, socket);
             close(connected_socket_fd);
             return 0;
         }