Implement SUSPEND/RESUME like vDPA does, by automatically using it in
vhost_user_dev_start(). (Though our vDPA code does not implement RESUME
yet, so there, the device is reset when it is to be resumed.)
Signed-off-by: Hanna Czenczek
---
hw/virtio/vhost-user.c | 99 +-
1 file changed, 97 insertions(+), 2 deletions(-)
diff --git a/hw/virtio/vhost-user.c b/hw/virtio/vhost-user.c
index 8dcf049d42..4507de5a92 100644
--- a/hw/virtio/vhost-user.c
+++ b/hw/virtio/vhost-user.c
@@ -74,6 +74,8 @@ enum VhostUserProtocolFeature {
/* Feature 14 reserved for VHOST_USER_PROTOCOL_F_INBAND_NOTIFICATIONS. */
VHOST_USER_PROTOCOL_F_CONFIGURE_MEM_SLOTS = 15,
VHOST_USER_PROTOCOL_F_STATUS = 16,
+/* Feature 17 reserved for VHOST_USER_PROTOCOL_F_XEN_MMAP. */
+VHOST_USER_PROTOCOL_F_SUSPEND = 18,
VHOST_USER_PROTOCOL_F_MAX
};
@@ -121,6 +123,8 @@ typedef enum VhostUserRequest {
VHOST_USER_REM_MEM_REG = 38,
VHOST_USER_SET_STATUS = 39,
VHOST_USER_GET_STATUS = 40,
+VHOST_USER_SUSPEND = 41,
+VHOST_USER_RESUME = 42,
VHOST_USER_MAX
} VhostUserRequest;
@@ -1389,7 +1393,19 @@ static int vhost_user_set_u64(struct vhost_dev *dev, int
request, uint64_t u64,
static int vhost_user_set_status(struct vhost_dev *dev, uint8_t status)
{
-return vhost_user_set_u64(dev, VHOST_USER_SET_STATUS, status, false);
+int ret;
+
+ret = vhost_user_set_u64(dev, VHOST_USER_SET_STATUS, status, false);
+if (ret < 0) {
+return ret;
+}
+
+if (!status) {
+/* reset */
+dev->suspended = false;
+}
+
+return 0;
}
static int vhost_user_get_status(struct vhost_dev *dev, uint8_t *status)
@@ -1490,6 +1506,7 @@ static int vhost_user_get_max_memslots(struct vhost_dev
*dev,
static int vhost_user_reset_device(struct vhost_dev *dev)
{
+int ret;
VhostUserMsg msg = {
.hdr.flags = VHOST_USER_VERSION,
};
@@ -1499,7 +1516,13 @@ static int vhost_user_reset_device(struct vhost_dev *dev)
? VHOST_USER_RESET_DEVICE
: VHOST_USER_RESET_OWNER;
-return vhost_user_write(dev, , NULL, 0);
+ret = vhost_user_write(dev, , NULL, 0);
+if (ret < 0) {
+return ret;
+}
+
+dev->suspended = false;
+return 0;
}
static int vhost_user_backend_handle_config_change(struct vhost_dev *dev)
@@ -2707,8 +2730,80 @@ void vhost_user_async_close(DeviceState *d,
}
}
+static bool vhost_user_supports_suspend(struct vhost_dev *dev)
+{
+return virtio_has_feature(dev->protocol_features,
+ VHOST_USER_PROTOCOL_F_SUSPEND);
+}
+
+static int vhost_user_do_suspend_resume(struct vhost_dev *dev, bool suspend)
+{
+VhostUserMsg msg;
+bool reply_supported = virtio_has_feature(dev->protocol_features,
+ VHOST_USER_PROTOCOL_F_REPLY_ACK);
+VhostUserRequest request = suspend ? VHOST_USER_SUSPEND :
VHOST_USER_RESUME;
+int ret;
+
+if (dev->suspended == suspend) {
+/* Nothing to do */
+return 0;
+}
+
+if (!vhost_user_supports_suspend(dev)) {
+return -ENOTSUP;
+}
+
+msg = (VhostUserMsg) {
+.hdr = {
+.request = request,
+.size = 0,
+.flags = VHOST_USER_VERSION,
+},
+};
+if (reply_supported) {
+msg.hdr.flags |= VHOST_USER_NEED_REPLY_MASK;
+}
+
+ret = vhost_user_write(dev, , NULL, 0);
+if (ret < 0) {
+return ret;
+}
+
+if (reply_supported) {
+ret = process_message_reply(dev, );
+if (ret < 0) {
+return ret;
+}
+}
+
+dev->suspended = suspend;
+return 0;
+}
+
+static int vhost_user_suspend(struct vhost_dev *dev)
+{
+return vhost_user_do_suspend_resume(dev, true);
+}
+
+static int vhost_user_resume(struct vhost_dev *dev)
+{
+return vhost_user_do_suspend_resume(dev, false);
+}
+
static int vhost_user_dev_start(struct vhost_dev *dev, bool started)
{
+/*
+ * Ignore results. If the central vhost code cares, it will check
+ * dev->suspended. (These calls will fail if the back-end does not
+ * support suspend/resume, which callers that just want to start the
+ * device do not care about.)
+ */
+if (started) {
+vhost_user_resume(dev);
+} else {
+vhost_user_suspend(dev);
+}
+
if (!virtio_has_feature(dev->protocol_features,
VHOST_USER_PROTOCOL_F_STATUS)) {
return 0;
--
2.41.0
