@@ -124,6 +124,9 @@ struct QIOChannelClass {
int (*io_set_blocking)(QIOChannel *ioc,
bool enabled,
Error **errp);
+ int (*io_set_keepalive)(QIOChannel *ioc,
+ bool enabled,
+ Error **errp);
/* Optional callbacks */
int (*io_shutdown)(QIOChannel *ioc,
@@ -490,6 +493,18 @@ int qio_channel_set_blocking(QIOChannel *ioc,
bool enabled,
Error **errp);
+/*
+ * qio_channel_set_keepalive:
+ * @ioc: the channel object
+ * @enabled: the keepalive flag state
+ * @errp: pointer to a NULL-initialized error object
+ *
+ * Returns 0 on success, -1 on error.
+ */
+int qio_channel_set_keepalive(QIOChannel *ioc,
+ bool enabled,
+ Error **errp);
+
/**
* qio_channel_close:
* @ioc: the channel object
@@ -656,6 +656,25 @@ qio_channel_socket_set_blocking(QIOChannel *ioc,
}
+static int
+qio_channel_socket_set_keepalive(QIOChannel *ioc,
+ bool enabled,
+ Error **errp)
+{
+ QIOChannelSocket *sioc = QIO_CHANNEL_SOCKET(ioc);
+ int val = enabled;
+ int ret = qemu_setsockopt(sioc->fd, SOL_SOCKET, SO_KEEPALIVE,
+ &val, sizeof(val));
+
+ if (ret < 0) {
+ error_setg_errno(errp, errno, "Unable to set KEEPALIVE");
+ return -1;
+ }
+
+ return 0;
+}
+
+
static void
qio_channel_socket_set_delay(QIOChannel *ioc,
bool enabled)
@@ -762,6 +781,7 @@ static void qio_channel_socket_class_init(ObjectClass *klass,
ioc_klass->io_writev = qio_channel_socket_writev;
ioc_klass->io_readv = qio_channel_socket_readv;
ioc_klass->io_set_blocking = qio_channel_socket_set_blocking;
+ ioc_klass->io_set_keepalive = qio_channel_socket_set_keepalive;
ioc_klass->io_close = qio_channel_socket_close;
ioc_klass->io_shutdown = qio_channel_socket_shutdown;
ioc_klass->io_set_cork = qio_channel_socket_set_cork;
@@ -265,6 +265,20 @@ int qio_channel_set_blocking(QIOChannel *ioc,
return klass->io_set_blocking(ioc, enabled, errp);
}
+int qio_channel_set_keepalive(QIOChannel *ioc,
+ bool enabled,
+ Error **errp)
+{
+ QIOChannelClass *klass = QIO_CHANNEL_GET_CLASS(ioc);
+
+ if (!klass->io_set_keepalive) {
+ error_setg(errp, "KEEPALIVE is not supported by IO channel");
+ return -1;
+ }
+
+ return klass->io_set_keepalive(ioc, enabled, errp);
+}
+
int qio_channel_close(QIOChannel *ioc,
Error **errp)
Signed-off-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> --- include/io/channel.h | 15 +++++++++++++++ io/channel-socket.c | 20 ++++++++++++++++++++ io/channel.c | 14 ++++++++++++++ 3 files changed, 49 insertions(+)