From mboxrd@z Thu Jan 1 00:00:00 1970 Received: by passt.top (Postfix, from userid 1000) id C1AFF5A061D; Fri, 07 Feb 2025 23:54:15 +0100 (CET) From: Stefano Brivio To: passt-dev@passt.top Subject: [PATCH v12 4/6] vhost_user: Make source quit after reporting migration state Date: Fri, 7 Feb 2025 23:54:13 +0100 Message-ID: <20250207225415.1067296-5-sbrivio@redhat.com> X-Mailer: git-send-email 2.43.0 In-Reply-To: <20250207225415.1067296-1-sbrivio@redhat.com> References: <20250207225415.1067296-1-sbrivio@redhat.com> MIME-Version: 1.0 Content-Transfer-Encoding: 8bit Message-ID-Hash: 6M7UJ3XRR7E4KGDCMK7SZDZ3DWVSX434 X-Message-ID-Hash: 6M7UJ3XRR7E4KGDCMK7SZDZ3DWVSX434 X-MailFrom: sbrivio@passt.top X-Mailman-Rule-Misses: dmarc-mitigation; no-senders; approved; emergency; loop; banned-address; member-moderation; nonmember-moderation; administrivia; implicit-dest; max-recipients; max-size; news-moderation; no-subject; digests; suspicious-header CC: David Gibson X-Mailman-Version: 3.3.8 Precedence: list List-Id: Development discussion and patches for passt Archived-At: Archived-At: List-Archive: List-Archive: List-Help: List-Owner: List-Post: List-Subscribe: List-Unsubscribe: This will close all the sockets we currently have open in repair mode, and completes our migration tasks as source. If the hypervisor wants to have us back at this point, somebody needs to restart us. Signed-off-by: Stefano Brivio --- vhost_user.c | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/vhost_user.c b/vhost_user.c index 256c8ab..9115fb5 100644 --- a/vhost_user.c +++ b/vhost_user.c @@ -998,6 +998,9 @@ static bool vu_send_rarp_exec(struct vu_dev *vdev, return false; } +/* If set, quit when we get a VHOST_USER_CHECK_DEVICE_STATE, after replying */ +static bool quit_on_device_state = false; + /** * vu_set_device_state_fd_exec() - Set the device state migration channel * @vdev: vhost-user device @@ -1025,6 +1028,9 @@ static bool vu_set_device_state_fd_exec(struct vu_dev *vdev, migrate_request(vdev->context, msg->fds[0], direction == VHOST_USER_TRANSFER_STATE_DIRECTION_LOAD); + if (direction == VHOST_USER_TRANSFER_STATE_DIRECTION_SAVE) + quit_on_device_state = true; + /* We don't provide a new fd for the data transfer */ vmsg_set_reply_u64(msg, VHOST_USER_VRING_NOFD_MASK); @@ -1203,4 +1209,10 @@ void vu_control_handler(struct vu_dev *vdev, int fd, uint32_t events) if (reply_requested) vu_send_reply(fd, &msg); + + if (quit_on_device_state && + msg.hdr.request == VHOST_USER_CHECK_DEVICE_STATE) { + info("Migration complete, exiting"); + _exit(EXIT_SUCCESS); + } } -- 2.43.0