file-xfer: Add file-xfer support for linux agent

The patch makes linux agent support file-xfer feature.

Signed-off-by: Dunrong Huang <riegamaths@gmail.com>

2 files changed, 216 insertions, 0 deletions

diff --git a/src/vdagent.c b/src/vdagent.c
index 430407e..78c7910 100644
--- a/src/vdagent.c
+++ b/src/vdagent.c
@@ -34,12 +34,23 @@
 #include <sys/select.h>
 #include <sys/stat.h>
 #include <spice/vd_agent.h>
+#include <glib.h>
 
 #include "udscs.h"
 #include "vdagentd-proto.h"
 #include "vdagentd-proto-strings.h"
 #include "vdagent-x11.h"
 
+typedef struct AgentFileXferTask {
+    uint32_t                       id;
+    int                            file_fd;
+    uint64_t                       read_bytes;
+    char                           *file_name;
+    uint64_t                       file_size;
+} AgentFileXferTask;
+
+GHashTable *agent_file_xfer_tasks = NULL;
+
 static const char *portdev = "/dev/virtio-ports/com.redhat.spice.0";
 static int debug = 0;
 static struct vdagent_x11 *x11 = NULL;
@@ -47,6 +58,154 @@ static struct udscs_connection *client = NULL;
 static int quit = 0;
 static int version_mismatch = 0;
 
+static void agent_file_xfer_task_free(gpointer data)
+{
+    AgentFileXferTask *task = data;
+
+    g_return_if_fail(task != NULL);
+
+    if (task->file_fd > 0) {
+        close(task->file_fd);
+    }
+    g_free(task->file_name);
+    g_free(task);
+}
+
+/* Parse start messag then create a new file xfer task */
+static AgentFileXferTask *vdagent_parse_start_msg(
+    VDAgentFileXferStartMessage *msg)
+{
+    GKeyFile *keyfile = NULL;
+    AgentFileXferTask *task = NULL;
+    GError *error = NULL;
+
+    keyfile = g_key_file_new();
+    if (g_key_file_load_from_data(keyfile,
+                                  (const gchar *)msg->data,
+                                  -1,
+                                  G_KEY_FILE_NONE, &error) == FALSE) {
+        syslog(LOG_ERR, "failed to load keyfile from data, error:%s\n",
+               error->message);
+        goto error;
+    }
+    task = g_new0(AgentFileXferTask, 1);
+    task->id = msg->id;
+    task->file_name = g_key_file_get_string(
+        keyfile, "vdagent-file-xfer", "name", &error);
+    if (error) {
+        syslog(LOG_ERR, "failed to parse filename, error:%s\n", error->message);
+        goto error;
+    }
+    task->file_size = g_key_file_get_uint64(
+        keyfile, "vdagent-file-xfer", "size", &error);
+    if (error) {
+        syslog(LOG_ERR, "failed to parse filesize, error:%s\n", error->message);
+        goto error;
+    }
+
+    g_key_file_free(keyfile);
+    return task;
+
+error:
+    g_clear_error(&error);
+    agent_file_xfer_task_free(task);
+    if (keyfile) {
+        g_key_file_free(keyfile);
+    }
+    return NULL;
+}
+
+static void vdagent_file_xfer_start(VDAgentFileXferStartMessage *msg)
+{
+    AgentFileXferTask *new;
+    char *file_path;
+    const gchar *desktop;
+
+    new = vdagent_parse_start_msg(msg);
+    if (new == NULL) {
+        goto error;
+    }
+
+    desktop = g_get_user_special_dir(G_USER_DIRECTORY_DESKTOP);
+    if (desktop == NULL) {
+        goto error;
+    }
+    file_path = g_build_filename(desktop, new->file_name, NULL);
+    new->file_fd = open(file_path, O_CREAT | O_WRONLY, 0644);
+    g_free(file_path);
+    if (new->file_fd == -1) {
+        syslog(LOG_ERR, "Create file error:%s\n", strerror(errno));
+        goto error;
+    }
+
+    if (ftruncate(new->file_fd, new->file_size) < 0) {
+        goto error;
+    }
+
+    g_hash_table_insert(agent_file_xfer_tasks, GINT_TO_POINTER(msg->id), new);
+
+    udscs_write(client, VDAGENTD_FILE_XFER_STATUS,
+                msg->id, VD_AGENT_FILE_XFER_STATUS_CAN_SEND_DATA, NULL, 0);
+    return ;
+
+error:
+    udscs_write(client, VDAGENTD_FILE_XFER_STATUS,
+                msg->id, VD_AGENT_FILE_XFER_STATUS_ERROR, NULL, 0);
+    agent_file_xfer_task_free(new);
+}
+
+static void vdagent_file_xfer_status(VDAgentFileXferStatusMessage *msg)
+{
+    syslog(LOG_INFO, "task %d received response %d", msg->id, msg->result);
+
+    if (msg->result == VD_AGENT_FILE_XFER_STATUS_CAN_SEND_DATA) {
+        /* Do nothing */
+    } else {
+        /* Error, remove this task */
+        gboolean found;
+        found = g_hash_table_remove(agent_file_xfer_tasks, GINT_TO_POINTER(msg->id));
+        if (found) {
+            syslog(LOG_DEBUG, "remove task %d", msg->id);
+        } else {
+            syslog(LOG_ERR, "can not find task %d", msg->id);
+        }
+    }
+}
+
+static void vdagent_file_xfer_data(VDAgentFileXferDataMessage *msg)
+{
+    AgentFileXferTask *task;
+    int len;
+
+    task = g_hash_table_lookup(agent_file_xfer_tasks, GINT_TO_POINTER(msg->id));
+    if (task == NULL) {
+        syslog(LOG_INFO, "Can not find task:%d", msg->id);
+        return ;
+    }
+
+    len = write(task->file_fd, msg->data, msg->size);
+    if (len == -1) {
+        syslog(LOG_ERR, "write file error:%s\n", strerror(errno));
+        /* TODO: close, cancel dnd */
+        return ;
+    }
+
+    task->read_bytes += msg->size;
+    if (task->read_bytes >= task->file_size) {
+        gboolean found;
+        if (task->read_bytes > task->file_size) {
+            syslog(LOG_ERR, "error: received too much data");
+        }
+        syslog(LOG_DEBUG, "task %d have been finished", task->id);
+        found = g_hash_table_remove(agent_file_xfer_tasks, GINT_TO_POINTER(msg->id));
+        if (found) {
+            syslog(LOG_DEBUG, "remove task %d", msg->id);
+        } else {
+            syslog(LOG_ERR, "can not find task %d", msg->id);
+        }
+    }
+}
+
 void daemon_read_complete(struct udscs_connection **connp,
     struct udscs_message_header *header, uint8_t *data)
 {
@@ -82,6 +241,18 @@ void daemon_read_complete(struct udscs_connection **connp,
             version_mismatch = 1;
         }
         break;
+    case VDAGENTD_FILE_XFER_START:
+        vdagent_file_xfer_start((VDAgentFileXferStartMessage *)data);
+        free(data);
+        break;
+    case VDAGENTD_FILE_XFER_STATUS:
+        vdagent_file_xfer_status((VDAgentFileXferStatusMessage *)data);
+        free(data);
+        break;
+    case VDAGENTD_FILE_XFER_DATA:
+        vdagent_file_xfer_data((VDAgentFileXferDataMessage *)data);
+        free(data);
+        break;
     default:
         syslog(LOG_ERR, "Unknown message from vdagentd type: %d, ignoring",
                header->type);
@@ -199,6 +370,8 @@ int main(int argc, char *argv[])
     if (do_daemonize)
         daemonize();
 
+    agent_file_xfer_tasks = g_hash_table_new_full(g_direct_hash, g_direct_equal, NULL, agent_file_xfer_task_free);
+
 reconnect:
     if (version_mismatch) {
         syslog(LOG_INFO, "Version mismatch, restarting");
diff --git a/src/vdagentd.c b/src/vdagentd.c
index 27925f5..2fa19e9 100644
--- a/src/vdagentd.c
+++ b/src/vdagentd.c
@@ -216,6 +216,34 @@ static void do_client_clipboard(struct vdagent_virtio_port *vport,
                 data, size);
 }
 
+static void do_client_file_xfer(struct vdagent_virtio_port *vport,
+                                VDAgentMessage *message_header,
+                                uint8_t *data)
+{
+    uint32_t msg_type;
+
+    if (!active_session_conn) {
+        syslog(LOG_WARNING,
+               "Could not find an agent connnection belonging to the "
+               "active session, ignoring client clipboard request");
+        return;
+    }
+
+    switch (message_header->type) {
+    case VD_AGENT_FILE_XFER_START:
+        msg_type = VDAGENTD_FILE_XFER_START;
+        break;
+    case VD_AGENT_FILE_XFER_STATUS:
+        msg_type = VDAGENTD_FILE_XFER_STATUS;
+        break;
+    case VD_AGENT_FILE_XFER_DATA:
+        msg_type = VDAGENTD_FILE_XFER_DATA;
+        break;
+    }
+
+    udscs_write(active_session_conn, msg_type, 0, 0, data, message_header->size);
+}
+
 int virtio_port_read_complete(
         struct vdagent_virtio_port *vport,
         int port_nr,
@@ -285,6 +313,11 @@ int virtio_port_read_complete(
         }
         do_client_clipboard(vport, message_header, data);
         break;
+    case VD_AGENT_FILE_XFER_START:
+    case VD_AGENT_FILE_XFER_STATUS:
+    case VD_AGENT_FILE_XFER_DATA:
+        do_client_file_xfer(vport, message_header, data);
+        break;
     default:
         syslog(LOG_WARNING, "unknown message type %d, ignoring",
                message_header->type);
@@ -615,6 +648,16 @@ void agent_read_complete(struct udscs_connection **connp,
             return;
         }
         break;
+    case VDAGENTD_FILE_XFER_STATUS:{
+        VDAgentFileXferStatusMessage status;
+        status.id = header->arg1;
+        status.result = header->arg2;
+        vdagent_virtio_port_write(virtio_port, VDP_CLIENT_PORT,
+                                  VD_AGENT_FILE_XFER_STATUS, 0,
+                                  (uint8_t *)&status, sizeof(status));
+        break;
+    }
+
     default:
         syslog(LOG_ERR, "unknown message from vdagent: %u, ignoring",
                header->type);