samples/client: also receive region_access response for region write

Signed-off-by: Thanos Makatos <thanos.makatos@nutanix.com>

1 files changed, 9 insertions, 3 deletions

diff --git a/samples/client.c b/samples/client.c
index 7daa209..9a793dd 100644
--- a/samples/client.c
+++ b/samples/client.c
@@ -296,19 +296,23 @@ access_bar0(int sock)
         .t = time(NULL)
     };
     uint16_t msg_id = 1;
+    const int sleep_time = 1;
+    struct vfio_user_region_access region_access = {};
     int ret = send_recv_vfio_user_msg(sock, msg_id, VFIO_USER_REGION_WRITE,
-                                      &data, sizeof data, NULL, 0, NULL, NULL, 0);
+                                      &data, sizeof data, NULL, 0, NULL,
+                                      &region_access, sizeof region_access);
     if (ret < 0) {
         fprintf(stderr, "failed to write to BAR0: %s\n", strerror(-ret));
         return ret;
     }
+    assert(region_access.count == sizeof data.t);
 
     printf("wrote to BAR0: %ld\n", data.t);
 
-    sleep(2);
-
     msg_id++;
 
+    sleep(sleep_time);
+
     ret = send_recv_vfio_user_msg(sock, msg_id, VFIO_USER_REGION_READ,
                                   &data.region_access, sizeof data.region_access,
                                   NULL, 0, NULL, &data, sizeof data);
@@ -320,6 +324,8 @@ access_bar0(int sock)
 
     printf("read from BAR0: %ld\n", data.t);
 
+    assert(data.t >= sleep_time);
+
     return 0;
 }