qga/channel-posix.c

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