vdagent.c

   1 /*  vdagent.c xorg-client to vdagentd (daemon).
   2
   3     Copyright 2010 Red Hat, Inc.
   4
   5     Red Hat Authors:
   6     Hans de Goede <hdegoede@redhat.com>
   7
   8     This program is free software: you can redistribute it and/or modify
   9     it under the terms of the GNU General Public License as published by
  10     the Free Software Foundation, either version 3 of the License, or
  11     (at your option) any later version.
  12
  13     This program is distributed in the hope that it will be useful,
  14     but WITHOUT ANY WARRANTY; without even the implied warranty of
  15     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16     GNU General Public License for more details.
  17
  18     You should have received a copy of the GNU General Public License
  19     along with this program.  If not, see <http://www.gnu.org/licenses/>.
  20 */
  21
  22 #include <stdio.h>
  23 #include <stdlib.h>
  24 #include <string.h>
  25 #include <unistd.h>
  26 #include <fcntl.h>
  27 #include <errno.h>
  28 #include <sys/select.h>
  29 #include <sys/stat.h>
  30 #include <spice/vd_agent.h>
  31
  32 #include "udscs.h"
  33 #include "vdagentd-proto.h"
  34 #include "vdagentd-proto-strings.h"
  35 #include "vdagent-x11.h"
  36
  37 static int verbose = 0;
  38 static struct vdagent_x11 *x11 = NULL;
  39 static FILE *logfile = NULL;
  40
  41 void daemon_read_complete(struct udscs_connection **connp,
  42     struct udscs_message_header *header, const uint8_t *data)
  43 {
  44     switch (header->type) {
  45     case VDAGENTD_MONITORS_CONFIG:
  46         vdagent_x11_set_monitor_config(x11, (VDAgentMonitorsConfig *)data);
  47         break;
  48     case VDAGENTD_CLIPBOARD_REQUEST:
  49         vdagent_x11_clipboard_request(x11, header->opaque);
  50         break;
  51     case VDAGENTD_CLIPBOARD_GRAB:
  52         vdagent_x11_clipboard_grab(x11, (uint32_t *)data,
  53                                    header->size / sizeof(uint32_t));
  54         break;
  55     case VDAGENTD_CLIPBOARD_DATA:
  56         vdagent_x11_clipboard_data(x11, header->opaque, data, header->size);
  57         break;
  58     case VDAGENTD_CLIPBOARD_RELEASE:
  59         vdagent_x11_clipboard_release(x11);
  60         break;
  61     default:
  62         if (verbose)
  63             fprintf(logfile, "Unknown message from vdagentd type: %d\n",
  64                     header->type);
  65     }
  66 }
  67
  68 static void usage(FILE *fp)
  69 {
  70     fprintf(fp,
  71             "vdagent -- spice agent xorg client\n"
  72             "options:\n"
  73             "  -h    print this text\n"
  74             "  -d    log debug messages\n");
  75 }
  76
  77 int main(int argc, char *argv[])
  78 {
  79     struct udscs_connection *client = NULL;
  80     fd_set readfds, writefds;
  81     int c, n, nfds, x11_fd, retval = 0;
  82     char *home, filename[1024];
  83
  84     for (;;) {
  85         if (-1 == (c = getopt(argc, argv, "-dh")))
  86             break;
  87         switch (c) {
  88         case 'd':
  89             verbose++;
  90             break;
  91         case 'h':
  92             usage(stdout);
  93             return 0;
  94         default:
  95             usage(stderr);
  96             return 1;
  97         }
  98     }
  99
 100     home = getenv("HOME");
 101     if (home) {
 102         snprintf(filename, sizeof(filename), "%s/.spice-agent", home);
 103         n = mkdir(filename, 0755);
 104         snprintf(filename, sizeof(filename), "%s/.spice-agent/log", home);
 105         logfile = fopen(filename, "w");
 106         if (!logfile) {
 107             fprintf(stderr, "Error opening %s: %s\n", filename,
 108                     strerror(errno));
 109             logfile = stderr;
 110         }
 111     } else {
 112         fprintf(stderr, "Could not get home directory, logging to stderr\n");
 113         logfile = stderr;
 114     }
 115
 116     client = udscs_connect(VDAGENTD_SOCKET, daemon_read_complete, NULL,
 117                            vdagentd_messages, VDAGENTD_NO_MESSAGES,
 118                            verbose? logfile:NULL, logfile);
 119     if (!client)
 120         return 1;
 121
 122     x11 = vdagent_x11_create(client, logfile, verbose);
 123     if (!x11) {
 124         udscs_destroy_connection(&client);
 125         return 1;
 126     }
 127
 128     while (client) {
 129         FD_ZERO(&readfds);
 130         FD_ZERO(&writefds);
 131
 132         nfds = udscs_client_fill_fds(client, &readfds, &writefds);
 133         x11_fd = vdagent_x11_get_fd(x11);
 134         FD_SET(x11_fd, &readfds);
 135         if (x11_fd >= nfds)
 136             nfds = x11_fd + 1;
 137
 138         n = select(nfds, &readfds, &writefds, NULL, NULL);
 139         if (n == -1) {
 140             if (errno == EINTR)
 141                 continue;
 142             fprintf(logfile, "Fatal error select: %s\n", strerror(errno));
 143             retval = 1;
 144             break;
 145         }
 146
 147         if (FD_ISSET(x11_fd, &readfds))
 148             vdagent_x11_do_read(x11);
 149         udscs_client_handle_fds(&client, &readfds, &writefds);
 150         fflush(logfile);
 151     }
 152
 153     vdagent_x11_destroy(x11);
 154     udscs_destroy_connection(&client);
 155     fclose(logfile);
 156
 157     return retval;
 158 }