From e79d34f974942b63e503edf661e736ebd86d96fb Mon Sep 17 00:00:00 2001
From: Michael Jumper <zhangmaike@users.sourceforge.net>
Date: Sun, 27 Mar 2011 19:04:05 -0700
Subject: [PATCH] Using cairo instead of libpng (as required by libguac), now
 handling RDP messages.

---
 protocols/rdp/configure.in     |  2 +-
 protocols/rdp/src/rdp_client.c | 91 ++++++++++++++++++++++++++++++++--
 2 files changed, 89 insertions(+), 4 deletions(-)

diff --git a/protocols/rdp/configure.in b/protocols/rdp/configure.in
index 4939ce6d..968cd7cd 100644
--- a/protocols/rdp/configure.in
+++ b/protocols/rdp/configure.in
@@ -44,7 +44,7 @@ AC_PROG_LIBTOOL
 
 # Checks for libraries.
 AC_CHECK_LIB([guac], [guac_get_client],, AC_MSG_ERROR("libguac is required for communication via the guacamole protocol"))
-AC_CHECK_LIB([png], [png_write_png],, AC_MSG_ERROR("libpng is required for writing png messages"))
+AC_CHECK_LIB([cairo], [cairo_create],, AC_MSG_ERROR("cairo is required for drawing instructions"))
 AC_CHECK_LIB([freerdp], [freerdp_new],, AC_MSG_ERROR("libfreerdp is required"))
 AC_CHECK_LIB([freerdpchanman], [freerdp_chanman_new],, AC_MSG_ERROR("libfreerdp is required"))
 
diff --git a/protocols/rdp/src/rdp_client.c b/protocols/rdp/src/rdp_client.c
index 0159d516..f67b4409 100644
--- a/protocols/rdp/src/rdp_client.c
+++ b/protocols/rdp/src/rdp_client.c
@@ -38,6 +38,9 @@
 #include <stdlib.h>
 #include <string.h>
 
+#include <sys/select.h>
+#include <errno.h>
+
 #include <freerdp/freerdp.h>
 #include <freerdp/chanman.h>
 
@@ -84,6 +87,88 @@ int rdp_guac_client_free_handler(guac_client* client) {
 
 }
 
+int rdp_guac_client_handle_messages(guac_client* client) {
+
+    rdp_guac_client_data* guac_client_data = (rdp_guac_client_data*) client->data;
+    rdpInst* rdp_inst = guac_client_data->rdp_inst;
+    rdpChanMan* chanman = guac_client_data->chanman;
+
+    int index;
+    int max_fd, fd;
+    void* read_fds[32];
+    void* write_fds[32];
+    int read_count = 0;
+    int write_count = 0;
+
+    fd_set rfds, wfds;
+
+    /* get rdp fds */
+    if (rdp_inst->rdp_get_fds(rdp_inst, read_fds, &read_count, write_fds, &write_count) != 0) {
+        guac_log_error("Unable to read RDP file descriptors.");
+        return 1;
+    }
+
+    /* get channel fds */
+    if (freerdp_chanman_get_fds(chanman, rdp_inst, read_fds, &read_count, write_fds, &write_count) != 0) {
+        guac_log_error("Unable to read RDP channel file descriptors.");
+        return 1;
+    }
+
+    /* Construct read fd_set */
+    max_fd = 0;
+    FD_ZERO(&rfds);
+    for (index = 0; index < read_count; index++) {
+        fd = (int)(long) (read_fds[index]);
+        if (fd > max_fd)
+            max_fd = fd;
+        FD_SET(fd, &rfds);
+    }
+
+    /* Construct write fd_set */
+    FD_ZERO(&wfds);
+    for (index = 0; index < write_count; index++) {
+        fd = (int)(long) (write_fds[index]);
+        if (fd > max_fd)
+            max_fd = fd;
+        FD_SET(fd, &wfds);
+    }
+
+    /* If no file descriptors, error */
+    if (max_fd == 0) {
+        guac_log_error("No file descriptors");
+        return 1;
+    }
+
+    /* Otherwise, wait for file descriptors given */
+    if (select(max_fd + 1, &rfds, &wfds, NULL, NULL) == -1) {
+        /* these are not really errors */
+        if (!((errno == EAGAIN) ||
+            (errno == EWOULDBLOCK) ||
+            (errno == EINPROGRESS) ||
+            (errno == EINTR))) /* signal occurred */
+        {
+            guac_log_error("Error waiting for file descriptor.");
+            return 1;
+        }
+    }
+
+    /* check the libfreerdp fds */
+    if (rdp_inst->rdp_check_fds(rdp_inst) != 0) {
+        guac_log_error("Error handling RDP file descriptors.");
+        return 1;
+    }
+
+    /* check channel fds */
+    if (freerdp_chanman_check_fds(chanman, rdp_inst) != 0) {
+        guac_log_error("Error handling RDP channel file descriptors.");
+        return 1;
+    }
+
+    /* Success */
+    return 0;
+
+}
+
 int guac_client_init(guac_client* client, int argc, char** argv) {
 
     rdp_guac_client_data* guac_client_data;
@@ -223,10 +308,10 @@ int guac_client_init(guac_client* client, int argc, char** argv) {
 
     /* Client handlers */
     client->free_handler = rdp_guac_client_free_handler;
+    client->handle_messages = rdp_guac_client_handle_messages;
 
-    /* STUB */
-    guac_send_error(client->io, "STUB");
-    return 1;
+    /* Success */
+    return 0;
 
 }