[Spice-devel] [PATCH vdagent v2 6/8] vdagent: Introduce VDAgent object

Jakub Janků janku.jakub.jj at gmail.com
Fri Sep 29 16:43:27 UTC 2017


Related to ongoing work to use GMainLoop and GTK integration

Signed-off-by: Victor Toso <victortoso at redhat.com>
---
 Makefile.am           |   3 ++
 configure.ac          |   1 +
 src/vdagent/vdagent.c | 140 ++++++++++++++++++++++++++++++--------------------
 src/vdagent/vdagent.h |  54 +++++++++++++++++++
 4 files changed, 142 insertions(+), 56 deletions(-)
 create mode 100644 src/vdagent/vdagent.h

diff --git a/Makefile.am b/Makefile.am
index 45f7177..efd058a 100644
--- a/Makefile.am
+++ b/Makefile.am
@@ -14,6 +14,7 @@ common_sources =				\
 src_spice_vdagent_CFLAGS =			\
 	$(X_CFLAGS)				\
 	$(SPICE_CFLAGS)				\
+	$(GOBJECT2_CFLAGS)			\
 	$(GLIB2_CFLAGS)				\
 	$(ALSA_CFLAGS)				\
 	-I$(srcdir)/src				\
@@ -23,6 +24,7 @@ src_spice_vdagent_CFLAGS =			\
 src_spice_vdagent_LDADD =			\
 	$(X_LIBS)				\
 	$(SPICE_LIBS)				\
+	$(GOBJECT2_LIBS)			\
 	$(GLIB2_LIBS)				\
 	$(ALSA_LIBS)				\
 	$(NULL)
@@ -38,6 +40,7 @@ src_spice_vdagent_SOURCES =			\
 	src/vdagent/x11.c			\
 	src/vdagent/x11.h			\
 	src/vdagent/vdagent.c			\
+	src/vdagent/vdagent.h			\
 	$(NULL)
 
 src_spice_vdagentd_CFLAGS =			\
diff --git a/configure.ac b/configure.ac
index d92b527..12b7d23 100644
--- a/configure.ac
+++ b/configure.ac
@@ -86,6 +86,7 @@ AC_ARG_ENABLE([static-uinput],
               [enable_static_uinput="no"])
 
 PKG_CHECK_MODULES([GLIB2], [glib-2.0 >= 2.34])
+PKG_CHECK_MODULES(GOBJECT2, [gobject-2.0])
 PKG_CHECK_MODULES(X, [xfixes xrandr >= 1.3 xinerama x11])
 PKG_CHECK_MODULES(SPICE, [spice-protocol >= 0.12.13])
 PKG_CHECK_MODULES(ALSA, [alsa >= 1.0.22])
diff --git a/src/vdagent/vdagent.c b/src/vdagent/vdagent.c
index ac6e732..fd2ba51 100644
--- a/src/vdagent/vdagent.c
+++ b/src/vdagent/vdagent.c
@@ -37,16 +37,13 @@
 #include <glib.h>
 #include <poll.h>
 
-#include "udscs.h"
 #include "vdagentd-proto.h"
 #include "vdagentd-proto-strings.h"
 #include "audio.h"
-#include "x11.h"
-#include "file-xfers.h"
+#include "vdagent.h"
+
+G_DEFINE_TYPE (VDAgent, vdagent, G_TYPE_OBJECT);
 
-static struct vdagent_x11 *x11 = NULL;
-static struct vdagent_file_xfers *vdagent_file_xfers = NULL;
-static struct udscs_connection *client = NULL;
 static int quit = 0;
 static int version_mismatch = 0;
 
@@ -90,7 +87,7 @@ static GOptionEntry entries[] = {
  * Return path where transferred files should be stored.
  * Returned path should not be freed or modified.
  **/
-static const gchar *xfer_get_download_directory(void)
+static const gchar *xfer_get_download_directory(VDAgent *agent)
 {
     if (fx_dir != NULL) {
         if (!strcmp(fx_dir, "xdg-desktop"))
@@ -101,7 +98,7 @@ static const gchar *xfer_get_download_directory(void)
         return fx_dir;
     }
 
-    return g_get_user_special_dir(vdagent_x11_has_icons_on_desktop(x11) ?
+    return g_get_user_special_dir(vdagent_x11_has_icons_on_desktop(agent->x11) ?
                                   G_USER_DIRECTORY_DESKTOP :
                                   G_USER_DIRECTORY_DOWNLOAD);
 }
@@ -110,63 +107,64 @@ static const gchar *xfer_get_download_directory(void)
  * vdagent_init_file_xfer
  *
  * Initialize handler for file xfer,
- * return TRUE on success (vdagent_file_xfers is not NULL).
+ * return TRUE on success (agent->xfers is not NULL).
  **/
-static gboolean vdagent_init_file_xfer(void)
+static gboolean vdagent_init_file_xfer(VDAgent *agent)
 {
     const gchar *xfer_dir;
 
-    if (vdagent_file_xfers != NULL) {
+    if (agent->xfers != NULL) {
         syslog(LOG_DEBUG, "File-xfer already initialized");
         return TRUE;
     }
 
-    xfer_dir = xfer_get_download_directory();
+    xfer_dir = xfer_get_download_directory(agent);
     if (xfer_dir == NULL) {
         syslog(LOG_WARNING,
                "warning could not get file xfer save dir, "
                "file transfers will be disabled");
-        vdagent_file_xfers = NULL;
         return FALSE;
     }
 
     if (fx_open_dir == -1)
-        fx_open_dir = !vdagent_x11_has_icons_on_desktop(x11);
+        fx_open_dir = !vdagent_x11_has_icons_on_desktop(agent->x11);
 
-    vdagent_file_xfers = vdagent_file_xfers_create(client, xfer_dir,
-                                                   fx_open_dir, debug);
-    return (vdagent_file_xfers != NULL);
+    agent->xfers = vdagent_file_xfers_create(agent->conn, xfer_dir,
+                                             fx_open_dir, debug);
+    return (agent->xfers != NULL);
 }
 
-static gboolean vdagent_finalize_file_xfer(void)
+static gboolean vdagent_finalize_file_xfer(VDAgent *agent)
 {
-    if (vdagent_file_xfers == NULL)
+    if (agent->xfers == NULL)
         return FALSE;
 
-    g_clear_pointer(&vdagent_file_xfers, vdagent_file_xfers_destroy);
+    g_clear_pointer(&agent->xfers, vdagent_file_xfers_destroy);
     return TRUE;
 }
 
 static void daemon_read_complete(struct udscs_connection **connp,
     struct udscs_message_header *header, uint8_t *data)
 {
+    VDAgent *agent = udscs_get_user_data(*connp);
+
     switch (header->type) {
     case VDAGENTD_MONITORS_CONFIG:
-        vdagent_x11_set_monitor_config(x11, (VDAgentMonitorsConfig *)data, 0);
+        vdagent_x11_set_monitor_config(agent->x11, (VDAgentMonitorsConfig *)data, 0);
         break;
     case VDAGENTD_CLIPBOARD_REQUEST:
-        vdagent_x11_clipboard_request(x11, header->arg1, header->arg2);
+        vdagent_x11_clipboard_request(agent->x11, header->arg1, header->arg2);
         break;
     case VDAGENTD_CLIPBOARD_GRAB:
-        vdagent_x11_clipboard_grab(x11, header->arg1, (uint32_t *)data,
+        vdagent_x11_clipboard_grab(agent->x11, header->arg1, (uint32_t *)data,
                                    header->size / sizeof(uint32_t));
         break;
     case VDAGENTD_CLIPBOARD_DATA:
-        vdagent_x11_clipboard_data(x11, header->arg1, header->arg2,
+        vdagent_x11_clipboard_data(agent->x11, header->arg1, header->arg2,
                                    data, header->size);
         break;
     case VDAGENTD_CLIPBOARD_RELEASE:
-        vdagent_x11_clipboard_release(x11, header->arg1);
+        vdagent_x11_clipboard_release(agent->x11, header->arg1);
         break;
     case VDAGENTD_VERSION:
         if (strcmp((char *)data, VERSION) != 0) {
@@ -177,8 +175,8 @@ static void daemon_read_complete(struct udscs_connection **connp,
         }
         break;
     case VDAGENTD_FILE_XFER_START:
-        if (vdagent_file_xfers != NULL) {
-            vdagent_file_xfers_start(vdagent_file_xfers,
+        if (agent->xfers != NULL) {
+            vdagent_file_xfers_start(agent->xfers,
                                      (VDAgentFileXferStartMessage *)data);
         } else {
             vdagent_file_xfers_error_disabled(*connp,
@@ -186,8 +184,8 @@ static void daemon_read_complete(struct udscs_connection **connp,
         }
         break;
     case VDAGENTD_FILE_XFER_STATUS:
-        if (vdagent_file_xfers != NULL) {
-            vdagent_file_xfers_status(vdagent_file_xfers,
+        if (agent->xfers != NULL) {
+            vdagent_file_xfers_status(agent->xfers,
                                       (VDAgentFileXferStatusMessage *)data);
         } else {
             vdagent_file_xfers_error_disabled(*connp,
@@ -198,7 +196,7 @@ static void daemon_read_complete(struct udscs_connection **connp,
         if (debug)
             syslog(LOG_DEBUG, "Disabling file-xfers");
 
-        vdagent_finalize_file_xfer();
+        vdagent_finalize_file_xfer(agent);
         break;
     case VDAGENTD_AUDIO_VOLUME_SYNC: {
         VDAgentAudioVolumeSync *avs = (VDAgentAudioVolumeSync *)data;
@@ -210,8 +208,8 @@ static void daemon_read_complete(struct udscs_connection **connp,
         break;
     }
     case VDAGENTD_FILE_XFER_DATA:
-        if (vdagent_file_xfers != NULL) {
-            vdagent_file_xfers_data(vdagent_file_xfers,
+        if (agent->xfers != NULL) {
+            vdagent_file_xfers_data(agent->xfers,
                                     (VDAgentFileXferDataMessage *)data);
         } else {
             vdagent_file_xfers_error_disabled(*connp,
@@ -219,9 +217,9 @@ static void daemon_read_complete(struct udscs_connection **connp,
         }
         break;
     case VDAGENTD_CLIENT_DISCONNECTED:
-        vdagent_x11_client_disconnected(x11);
-        if (vdagent_finalize_file_xfer()) {
-            vdagent_init_file_xfer();
+        vdagent_x11_client_disconnected(agent->x11);
+        if (vdagent_finalize_file_xfer(agent)) {
+            vdagent_init_file_xfer(agent);
         }
         break;
     default:
@@ -230,7 +228,7 @@ static void daemon_read_complete(struct udscs_connection **connp,
     }
 }
 
-static struct udscs_connection *client_setup_sync(void)
+static struct udscs_connection *client_setup_sync(VDAgent *agent)
 {
     struct udscs_connection *conn = NULL;
 
@@ -306,6 +304,43 @@ static int file_test(const char *path)
     return stat(path, &buffer);
 }
 
+static void vdagent_init(VDAgent *self)
+{
+}
+
+static void vdagent_finalize(GObject *gobject)
+{
+    VDAgent *self = VDAGENT(gobject);
+    vdagent_finalize_file_xfer(self);
+    vdagent_x11_destroy(self->x11, self->conn == NULL);
+    udscs_destroy_connection(&self->conn);
+
+    G_OBJECT_CLASS(vdagent_parent_class)->finalize(gobject);
+}
+
+static void vdagent_class_init(VDAgentClass  *klass)
+{
+    GObjectClass *gobject_class = G_OBJECT_CLASS(klass);
+    gobject_class->finalize = vdagent_finalize;
+}
+
+static gboolean vdagent_init_sync(VDAgent *agent)
+{
+    agent->conn = client_setup_sync(agent);
+    if (agent->conn == NULL)
+        return FALSE;
+    udscs_set_user_data(agent->conn, agent);
+
+    agent->x11 = vdagent_x11_create(agent->conn, debug, x11_sync);
+    if (agent->x11 == NULL)
+        return FALSE;
+
+    if (!vdagent_init_file_xfer(agent))
+        syslog(LOG_WARNING, "File transfer is disabled");
+
+    return TRUE;
+}
+
 int main(int argc, char *argv[])
 {
     fd_set readfds, writefds;
@@ -314,6 +349,7 @@ int main(int argc, char *argv[])
     struct sigaction act;
     GOptionContext *context;
     GError *error = NULL;
+    VDAgent *agent;
 
     context = g_option_context_new(NULL);
     g_option_context_add_main_entries(context, entries, NULL);
@@ -362,20 +398,13 @@ reconnect:
         execvp(argv[0], argv);
     }
 
-    client = client_setup_sync();
-    if (client == NULL) {
-        return 1;
+    agent = g_object_new(TYPE_VDAGENT, NULL);
+    agent->parent_socket = parent_socket;
+    if (!vdagent_init_sync(agent)) {
+        quit = 1;
+        goto end_main;
     }
 
-    x11 = vdagent_x11_create(client, debug, x11_sync);
-    if (!x11) {
-        udscs_destroy_connection(&client);
-        return 1;
-    }
-
-    if (!vdagent_init_file_xfer())
-        syslog(LOG_WARNING, "File transfer is disabled");
-
     if (parent_socket) {
         if (write(parent_socket, "OK", 2) != 2)
             syslog(LOG_WARNING, "Parent already gone.");
@@ -383,12 +412,12 @@ reconnect:
         parent_socket = 0;
     }
 
-    while (client && !quit) {
+    while (agent->conn && !quit) {
         FD_ZERO(&readfds);
         FD_ZERO(&writefds);
 
-        nfds = udscs_client_fill_fds(client, &readfds, &writefds);
-        x11_fd = vdagent_x11_get_fd(x11);
+        nfds = udscs_client_fill_fds(agent->conn, &readfds, &writefds);
+        x11_fd = vdagent_x11_get_fd(agent->x11);
         FD_SET(x11_fd, &readfds);
         if (x11_fd >= nfds)
             nfds = x11_fd + 1;
@@ -402,13 +431,12 @@ reconnect:
         }
 
         if (FD_ISSET(x11_fd, &readfds))
-            vdagent_x11_do_read(x11);
-        udscs_client_handle_fds(&client, &readfds, &writefds);
+            vdagent_x11_do_read(agent->x11);
+        udscs_client_handle_fds(&agent->conn, &readfds, &writefds);
     }
 
-    vdagent_finalize_file_xfer();
-    vdagent_x11_destroy(x11, client == NULL);
-    udscs_destroy_connection(&client);
+end_main:
+    g_clear_object(&agent);
     if (!quit && do_daemonize)
         goto reconnect;
 
diff --git a/src/vdagent/vdagent.h b/src/vdagent/vdagent.h
new file mode 100644
index 0000000..8376315
--- /dev/null
+++ b/src/vdagent/vdagent.h
@@ -0,0 +1,54 @@
+/*  vdagent.h
+
+    Copyright 2017 Red Hat, Inc.
+
+    This program is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#ifndef __VDAGENT_H_
+#define __VDAGENT_H_
+
+#include <glib-object.h>
+#include "udscs.h"
+#include "x11.h"
+#include "file-xfers.h"
+
+G_BEGIN_DECLS
+
+#define TYPE_VDAGENT            (vdagent_get_type ())
+#define VDAGENT(obj)            (G_TYPE_CHECK_INSTANCE_CAST ((obj), TYPE_VDAGENT, VDAgent))
+#define VDAGENT_CLASS(klass)    (G_TYPE_CHECK_CLASS_CAST ((klass), TYPE_VDAGENT, VDAgentClass))
+#define IS_VDAGENT(obj)         (G_TYPE_CHECK_INSTANCE_TYPE ((obj), TYPE_VDAGENT))
+#define IS_VDAGENT_CLASS(klass) (G_TYPE_CHECK_CLASS_TYPE ((klass), TYPE_VDAGENT))
+#define VDAGENT_GET_CLASS(obj)  (G_TYPE_INSTANCE_GET_CLASS ((obj), TYPE_VDAGENT, VDAgentClass))
+
+typedef struct VDAgent {
+    GObject parent;
+
+    struct vdagent_x11             *x11;
+    struct vdagent_file_xfers      *xfers;
+    struct udscs_connection        *conn;
+
+    int                             parent_socket;
+} VDAgent;
+
+typedef struct VDAgentClass {
+    GObjectClass parent;
+} VDAgentClass;
+
+GType vdagent_get_type(void);
+
+G_END_DECLS
+
+#endif
-- 
2.13.6



More information about the Spice-devel mailing list