@@ -66,6 +66,8 @@ struct CharDriverState {
const uint8_t *buf, int len);
GSource *(*chr_add_watch)(struct CharDriverState *s, GIOCondition cond);
void (*chr_update_read_handler)(struct CharDriverState *s);
+ void (*chr_update_read_handler_full)(struct CharDriverState *s,
+ GMainContext *context);
int (*chr_ioctl)(struct CharDriverState *s, int cmd, void *arg);
int (*get_msgfds)(struct CharDriverState *s, int* fds, int num);
int (*set_msgfds)(struct CharDriverState *s, int *fds, int num);
@@ -388,6 +390,14 @@ void qemu_chr_add_handlers(CharDriverState *s,
IOEventHandler *fd_event,
void *opaque);
+/* This API can make handler run in the context what you pass to. */
+void qemu_chr_add_handlers_full(CharDriverState *s,
+ IOCanReadHandler *fd_can_read,
+ IOReadHandler *fd_read,
+ IOEventHandler *fd_event,
+ void *opaque,
+ GMainContext *context);
+
void qemu_chr_be_generic_open(CharDriverState *s);
void qemu_chr_accept_input(CharDriverState *s);
int qemu_chr_add_client(CharDriverState *s, int fd);
@@ -480,6 +480,40 @@ void qemu_chr_add_handlers(CharDriverState *s,
}
}
+void qemu_chr_add_handlers_full(CharDriverState *s,
+ IOCanReadHandler *fd_can_read,
+ IOReadHandler *fd_read,
+ IOEventHandler *fd_event,
+ void *opaque,
+ GMainContext *context)
+{
+ int fe_open;
+
+ if (!opaque && !fd_can_read && !fd_read && !fd_event) {
+ fe_open = 0;
+ remove_fd_in_watch(s);
+ } else {
+ fe_open = 1;
+ }
+ s->chr_can_read = fd_can_read;
+ s->chr_read = fd_read;
+ s->chr_event = fd_event;
+ s->handler_opaque = opaque;
+ if (fe_open && s->chr_update_read_handler) {
+ s->chr_update_read_handler_full(s, context);
+ }
+
+ if (!s->explicit_fe_open) {
+ qemu_chr_fe_set_open(s, fe_open);
+ }
+
+ /* We're connecting to an already opened device, so let's make sure we
+ also get the open event */
+ if (fe_open && s->be_open) {
+ qemu_chr_be_generic_open(s);
+ }
+}
+
static int null_chr_write(CharDriverState *chr, const uint8_t *buf, int len)
{
return len;
@@ -840,6 +874,7 @@ typedef struct IOWatchPoll
IOCanReadHandler *fd_can_read;
GSourceFunc fd_read;
void *opaque;
+ GMainContext *context;
} IOWatchPoll;
static IOWatchPoll *io_watch_poll_from_source(GSource *source)
@@ -869,6 +904,29 @@ static gboolean io_watch_poll_prepare(GSource *source, gint *timeout_)
return FALSE;
}
+static gboolean io_watch_poll_prepare_full(GSource *source,
+ gint *timeout_)
+{
+ IOWatchPoll *iwp = io_watch_poll_from_source(source);
+ bool now_active = iwp->fd_can_read(iwp->opaque) > 0;
+ bool was_active = iwp->src != NULL;
+ if (was_active == now_active) {
+ return FALSE;
+ }
+
+ if (now_active) {
+ iwp->src = qio_channel_create_watch(
+ iwp->ioc, G_IO_IN | G_IO_ERR | G_IO_HUP | G_IO_NVAL);
+ g_source_set_callback(iwp->src, iwp->fd_read, iwp->opaque, NULL);
+ g_source_attach(iwp->src, iwp->context);
+ } else {
+ g_source_destroy(iwp->src);
+ g_source_unref(iwp->src);
+ iwp->src = NULL;
+ }
+ return FALSE;
+}
+
static gboolean io_watch_poll_check(GSource *source)
{
return FALSE;
@@ -903,6 +961,13 @@ static GSourceFuncs io_watch_poll_funcs = {
.finalize = io_watch_poll_finalize,
};
+static GSourceFuncs io_watch_poll_funcs_full = {
+ .prepare = io_watch_poll_prepare_full,
+ .check = io_watch_poll_check,
+ .dispatch = io_watch_poll_dispatch,
+ .finalize = io_watch_poll_finalize,
+};
+
/* Can only be used for read */
static guint io_add_watch_poll(QIOChannel *ioc,
IOCanReadHandler *fd_can_read,
@@ -924,6 +989,30 @@ static guint io_add_watch_poll(QIOChannel *ioc,
return tag;
}
+/* Can only be used for read */
+static guint io_add_watch_poll_full(QIOChannel *ioc,
+ IOCanReadHandler *fd_can_read,
+ QIOChannelFunc fd_read,
+ gpointer user_data,
+ GMainContext *context)
+{
+ IOWatchPoll *iwp;
+ int tag;
+
+ iwp = (IOWatchPoll *) g_source_new(&io_watch_poll_funcs_full,
+ sizeof(IOWatchPoll));
+ iwp->fd_can_read = fd_can_read;
+ iwp->opaque = user_data;
+ iwp->ioc = ioc;
+ iwp->fd_read = (GSourceFunc) fd_read;
+ iwp->src = NULL;
+ iwp->context = context;
+
+ tag = g_source_attach(&iwp->parent, context);
+ g_source_unref(&iwp->parent);
+ return tag;
+ }
+
static void io_remove_watch_poll(guint tag)
{
GSource *source;
@@ -1063,6 +1152,20 @@ static void fd_chr_update_read_handler(CharDriverState *chr)
}
}
+static void fd_chr_update_read_handler_full(CharDriverState *chr,
+ GMainContext *context)
+{
+ FDCharDriver *s = chr->opaque;
+
+ remove_fd_in_watch(chr);
+ if (s->ioc_in) {
+ chr->fd_in_tag = io_add_watch_poll_full(s->ioc_in,
+ fd_chr_read_poll,
+ fd_chr_read, chr,
+ context);
+ }
+}
+
static void fd_chr_close(struct CharDriverState *chr)
{
FDCharDriver *s = chr->opaque;
@@ -1099,6 +1202,7 @@ static CharDriverState *qemu_chr_open_fd(int fd_in, int fd_out,
chr->chr_add_watch = fd_chr_add_watch;
chr->chr_write = fd_chr_write;
chr->chr_update_read_handler = fd_chr_update_read_handler;
+ chr->chr_update_read_handler_full = fd_chr_update_read_handler_full;
chr->chr_close = fd_chr_close;
return chr;
@@ -1310,6 +1414,12 @@ static void pty_chr_update_read_handler(CharDriverState *chr)
qemu_mutex_unlock(&chr->chr_write_lock);
}
+static void pty_chr_update_read_handler_full(CharDriverState *chr,
+ GMainContext *context)
+{
+ pty_chr_update_read_handler(chr);
+}
+
/* Called with chr_write_lock held. */
static int pty_chr_write(CharDriverState *chr, const uint8_t *buf, int len)
{
@@ -1465,6 +1575,7 @@ static CharDriverState *qemu_chr_open_pty(const char *id,
chr->opaque = s;
chr->chr_write = pty_chr_write;
chr->chr_update_read_handler = pty_chr_update_read_handler;
+ chr->chr_update_read_handler_full = pty_chr_update_read_handler_full;
chr->chr_close = pty_chr_close;
chr->chr_add_watch = pty_chr_add_watch;
chr->explicit_be_open = true;
@@ -2558,6 +2669,20 @@ static void udp_chr_update_read_handler(CharDriverState *chr)
}
}
+static void udp_chr_update_read_handler_full(CharDriverState *chr,
+ GMainContext *context)
+{
+ NetCharDriver *s = chr->opaque;
+
+ remove_fd_in_watch(chr);
+ if (s->ioc) {
+ chr->fd_in_tag = io_add_watch_poll_full(s->ioc,
+ udp_chr_read_poll,
+ udp_chr_read, chr,
+ context);
+ }
+}
+
static void udp_chr_close(CharDriverState *chr)
{
NetCharDriver *s = chr->opaque;
@@ -2589,6 +2714,7 @@ static CharDriverState *qemu_chr_open_udp(QIOChannelSocket *sioc,
chr->opaque = s;
chr->chr_write = udp_chr_write;
chr->chr_update_read_handler = udp_chr_update_read_handler;
+ chr->chr_update_read_handler_full = udp_chr_update_read_handler_full;
chr->chr_close = udp_chr_close;
/* be isn't opened until we get a connection */
chr->explicit_be_open = true;
@@ -2952,6 +3078,24 @@ static void tcp_chr_update_read_handler(CharDriverState *chr)
}
}
+static void tcp_chr_update_read_handler_full(CharDriverState *chr,
+ GMainContext *context)
+{
+ TCPCharDriver *s = chr->opaque;
+
+ if (!s->connected) {
+ return;
+ }
+
+ remove_fd_in_watch(chr);
+ if (s->ioc) {
+ chr->fd_in_tag = io_add_watch_poll_full(s->ioc,
+ tcp_chr_read_poll,
+ tcp_chr_read, chr,
+ context);
+ }
+}
+
typedef struct {
CharDriverState *chr;
char buf[12];
@@ -4410,6 +4554,7 @@ static CharDriverState *qmp_chardev_open_socket(const char *id,
chr->chr_add_client = tcp_chr_add_client;
chr->chr_add_watch = tcp_chr_add_watch;
chr->chr_update_read_handler = tcp_chr_update_read_handler;
+ chr->chr_update_read_handler_full = tcp_chr_update_read_handler_full;
/* be isn't opened until we get a connection */
chr->explicit_be_open = true;