qemu/qga/channel-posix.c

   1 #include <glib.h>
   2 #include <termios.h>
   3 #include <errno.h>
   4 #include <unistd.h>
   5 #include <fcntl.h>
   6 #include <stdlib.h>
   7 #include <string.h>
   8 #include "qemu/osdep.h"
   9 #include "qemu/sockets.h"
  10 #include "qga/channel.h"
  11
  12 #ifdef CONFIG_SOLARIS
  13 #include <stropts.h>
  14 #endif
  15
  16 #define GA_CHANNEL_BAUDRATE_DEFAULT B38400 /* for isa-serial channels */
  17
  18 struct GAChannel {
  19     GIOChannel *listen_channel;
  20     GIOChannel *client_channel;
  21     GAChannelMethod method;
  22     GAChannelCallback event_cb;
  23     gpointer user_data;
  24 };
  25
  26 static int ga_channel_client_add(GAChannel *c, int fd);
  27
  28 static gboolean ga_channel_listen_accept(GIOChannel *channel,
  29                                          GIOCondition condition, gpointer data)
  30 {
  31     GAChannel *c = data;
  32     int ret, client_fd;
  33     bool accepted = false;
  34     struct sockaddr_un addr;
  35     socklen_t addrlen = sizeof(addr);
  36
  37     g_assert(channel != NULL);
  38
  39     client_fd = qemu_accept(g_io_channel_unix_get_fd(channel),
  40                             (struct sockaddr *)&addr, &addrlen);
  41     if (client_fd == -1) {
  42         g_warning("error converting fd to gsocket: %s", strerror(errno));
  43         goto out;
  44     }
  45     qemu_set_nonblock(client_fd);
  46     ret = ga_channel_client_add(c, client_fd);
  47     if (ret) {
  48         g_warning("error setting up connection");
  49         close(client_fd);
  50         goto out;
  51     }
  52     accepted = true;
  53
  54 out:
  55     /* only accept 1 connection at a time */
  56     return !accepted;
  57 }
  58
  59 /* start polling for readable events on listen fd, new==true
  60  * indicates we should use the existing s->listen_channel
  61  */
  62 static void ga_channel_listen_add(GAChannel *c, int listen_fd, bool create)
  63 {
  64     if (create) {
  65         c->listen_channel = g_io_channel_unix_new(listen_fd);
  66     }
  67     g_io_add_watch(c->listen_channel, G_IO_IN, ga_channel_listen_accept, c);
  68 }
  69
  70 static void ga_channel_listen_close(GAChannel *c)
  71 {
  72     g_assert(c->method == GA_CHANNEL_UNIX_LISTEN);
  73     g_assert(c->listen_channel);
  74     g_io_channel_shutdown(c->listen_channel, true, NULL);
  75     g_io_channel_unref(c->listen_channel);
  76     c->listen_channel = NULL;
  77 }
  78
  79 /* cleanup state for closed connection/session, start accepting new
  80  * connections if we're in listening mode
  81  */
  82 static void ga_channel_client_close(GAChannel *c)
  83 {
  84     g_assert(c->client_channel);
  85     g_io_channel_shutdown(c->client_channel, true, NULL);
  86     g_io_channel_unref(c->client_channel);
  87     c->client_channel = NULL;
  88     if (c->method == GA_CHANNEL_UNIX_LISTEN && c->listen_channel) {
  89         ga_channel_listen_add(c, 0, false);
  90     }
  91 }
  92
  93 static gboolean ga_channel_client_event(GIOChannel *channel,
  94                                         GIOCondition condition, gpointer data)
  95 {
  96     GAChannel *c = data;
  97     gboolean client_cont;
  98
  99     g_assert(c);
 100     if (c->event_cb) {
 101         client_cont = c->event_cb(condition, c->user_data);
 102         if (!client_cont) {
 103             ga_channel_client_close(c);
 104             return false;
 105         }
 106     }
 107     return true;
 108 }
 109
 110 static int ga_channel_client_add(GAChannel *c, int fd)
 111 {
 112     GIOChannel *client_channel;
 113     GError *err = NULL;
 114
 115     g_assert(c && !c->client_channel);
 116     client_channel = g_io_channel_unix_new(fd);
 117     g_assert(client_channel);
 118     g_io_channel_set_encoding(client_channel, NULL, &err);
 119     if (err != NULL) {
 120         g_warning("error setting channel encoding to binary");
 121         g_error_free(err);
 122         return -1;
 123     }
 124     g_io_add_watch(client_channel, G_IO_IN | G_IO_HUP,
 125                    ga_channel_client_event, c);
 126     c->client_channel = client_channel;
 127     return 0;
 128 }
 129
 130 static gboolean ga_channel_open(GAChannel *c, const gchar *path, GAChannelMethod method)
 131 {
 132     int ret;
 133     c->method = method;
 134
 135     switch (c->method) {
 136     case GA_CHANNEL_VIRTIO_SERIAL: {
 137         int fd = qemu_open(path, O_RDWR | O_NONBLOCK
 138 #ifndef CONFIG_SOLARIS
 139                            | O_ASYNC
 140 #endif
 141                            );
 142         if (fd == -1) {
 143             g_critical("error opening channel: %s", strerror(errno));
 144             return false;
 145         }
 146 #ifdef CONFIG_SOLARIS
 147         ret = ioctl(fd, I_SETSIG, S_OUTPUT | S_INPUT | S_HIPRI);
 148         if (ret == -1) {
 149             g_critical("error setting event mask for channel: %s",
 150                        strerror(errno));
 151             close(fd);
 152             return false;
 153         }
 154 #endif
 155         ret = ga_channel_client_add(c, fd);
 156         if (ret) {
 157             g_critical("error adding channel to main loop");
 158             close(fd);
 159             return false;
 160         }
 161         break;
 162     }
 163     case GA_CHANNEL_ISA_SERIAL: {
 164         struct termios tio;
 165         int fd = qemu_open(path, O_RDWR | O_NOCTTY | O_NONBLOCK);
 166         if (fd == -1) {
 167             g_critical("error opening channel: %s", strerror(errno));
 168             return false;
 169         }
 170         tcgetattr(fd, &tio);
 171         /* set up serial port for non-canonical, dumb byte streaming */
 172         tio.c_iflag &= ~(IGNBRK | BRKINT | IGNPAR | PARMRK | INPCK | ISTRIP |
 173                          INLCR | IGNCR | ICRNL | IXON | IXOFF | IXANY |
 174                          IMAXBEL);
 175         tio.c_oflag = 0;
 176         tio.c_lflag = 0;
 177         tio.c_cflag |= GA_CHANNEL_BAUDRATE_DEFAULT;
 178         /* 1 available byte min or reads will block (we'll set non-blocking
 179          * elsewhere, else we have to deal with read()=0 instead)
 180          */
 181         tio.c_cc[VMIN] = 1;
 182         tio.c_cc[VTIME] = 0;
 183         /* flush everything waiting for read/xmit, it's garbage at this point */
 184         tcflush(fd, TCIFLUSH);
 185         tcsetattr(fd, TCSANOW, &tio);
 186         ret = ga_channel_client_add(c, fd);
 187         if (ret) {
 188             g_critical("error adding channel to main loop");
 189             close(fd);
 190             return false;
 191         }
 192         break;
 193     }
 194     case GA_CHANNEL_UNIX_LISTEN: {
 195         Error *local_err = NULL;
 196         int fd = unix_listen(path, NULL, strlen(path), &local_err);
 197         if (local_err != NULL) {
 198             g_critical("%s", error_get_pretty(local_err));
 199             error_free(local_err);
 200             return false;
 201         }
 202         ga_channel_listen_add(c, fd, true);
 203         break;
 204     }
 205     default:
 206         g_critical("error binding/listening to specified socket");
 207         return false;
 208     }
 209
 210     return true;
 211 }
 212
 213 GIOStatus ga_channel_write_all(GAChannel *c, const gchar *buf, gsize size)
 214 {
 215     GError *err = NULL;
 216     gsize written = 0;
 217     GIOStatus status = G_IO_STATUS_NORMAL;
 218
 219     while (size) {
 220         status = g_io_channel_write_chars(c->client_channel, buf, size,
 221                                           &written, &err);
 222         g_debug("sending data, count: %d", (int)size);
 223         if (err != NULL) {
 224             g_warning("error writing to channel: %s", err->message);
 225             return G_IO_STATUS_ERROR;
 226         }
 227         if (status != G_IO_STATUS_NORMAL) {
 228             break;
 229         }
 230         size -= written;
 231     }
 232
 233     if (status == G_IO_STATUS_NORMAL) {
 234         status = g_io_channel_flush(c->client_channel, &err);
 235         if (err != NULL) {
 236             g_warning("error flushing channel: %s", err->message);
 237             return G_IO_STATUS_ERROR;
 238         }
 239     }
 240
 241     return status;
 242 }
 243
 244 GIOStatus ga_channel_read(GAChannel *c, gchar *buf, gsize size, gsize *count)
 245 {
 246     return g_io_channel_read_chars(c->client_channel, buf, size, count, NULL);
 247 }
 248
 249 GAChannel *ga_channel_new(GAChannelMethod method, const gchar *path,
 250                           GAChannelCallback cb, gpointer opaque)
 251 {
 252     GAChannel *c = g_malloc0(sizeof(GAChannel));
 253     c->event_cb = cb;
 254     c->user_data = opaque;
 255
 256     if (!ga_channel_open(c, path, method)) {
 257         g_critical("error opening channel");
 258         ga_channel_free(c);
 259         return NULL;
 260     }
 261
 262     return c;
 263 }
 264
 265 void ga_channel_free(GAChannel *c)
 266 {
 267     if (c->method == GA_CHANNEL_UNIX_LISTEN
 268         && c->listen_channel) {
 269         ga_channel_listen_close(c);
 270     }
 271     if (c->client_channel) {
 272         ga_channel_client_close(c);
 273     }
 274     g_free(c);
 275 }