allow shadow memory offset per shadow ioeventfd (#703)

Signed-off-by: Thanos Makatos <thanos.makatos@nutanix.com>
Reviewed-by: John Levon <john.levon@nutanix.com>

3 files changed, 17 insertions, 15 deletions

diff --git a/test/py/libvfio_user.py b/test/py/libvfio_user.py
index 2bcd1bc..fd09fa8 100644
--- a/test/py/libvfio_user.py
+++ b/test/py/libvfio_user.py
@@ -410,13 +410,14 @@ class vfio_user_region_io_fds_request(Structure):
 class vfio_user_sub_region_ioeventfd(Structure):
     _pack_ = 1
     _fields_ = [
-        ("offset", c.c_uint64),
+        ("gpa_offset", c.c_uint64),
         ("size", c.c_uint64),
         ("fd_index", c.c_uint32),
         ("type", c.c_uint32),
         ("flags", c.c_uint32),
-        ("padding", c.c_uint32),
-        ("datamatch", c.c_uint64)
+        ("shadow_mem_fd_index", c.c_uint32),
+        ("datamatch", c.c_uint64),
+        ("shadow_offset", c.c_uint64)
     ]
 
 
@@ -633,7 +634,7 @@ lib.vfu_sgl_put.argtypes = (c.c_void_p, c.POINTER(dma_sg_t),
 
 lib.vfu_create_ioeventfd.argtypes = (c.c_void_p, c.c_uint32, c.c_int,
                                      c.c_size_t, c.c_uint32, c.c_uint32,
-                                     c.c_uint64, c.c_int32)
+                                     c.c_uint64, c.c_int32, c.c_uint64)
 
 lib.vfu_device_quiesced.argtypes = (c.c_void_p, c.c_int)
 
@@ -1207,12 +1208,12 @@ def vfu_sgl_put(ctx, sg, iovec, cnt=1):
     return lib.vfu_sgl_put(ctx, sg, iovec, cnt)
 
 
-def vfu_create_ioeventfd(ctx, region_idx, fd, offset, size, flags, datamatch,
-                         shadow_fd=-1):
+def vfu_create_ioeventfd(ctx, region_idx, fd, gpa_offset, size, flags,
+                         datamatch, shadow_fd=-1, shadow_offset=0):
     assert ctx is not None
 
-    return lib.vfu_create_ioeventfd(ctx, region_idx, fd, offset, size,
-                                    flags, datamatch, shadow_fd)
+    return lib.vfu_create_ioeventfd(ctx, region_idx, fd, gpa_offset, size,
+                                    flags, datamatch, shadow_fd, shadow_offset)
 
 
 def vfu_device_quiesced(ctx, err):
diff --git a/test/py/test_device_get_region_io_fds.py b/test/py/test_device_get_region_io_fds.py
index 924f462..b9c8d93 100644
--- a/test/py/test_device_get_region_io_fds.py
+++ b/test/py/test_device_get_region_io_fds.py
@@ -225,7 +225,7 @@ def test_device_get_region_io_fds_full():
         [out] = struct.unpack("@Q", out)
         assert out == 1
         assert ioevents[i].size == IOEVENT_SIZE
-        assert ioevents[i].offset == 40 - (IOEVENT_SIZE * i)
+        assert ioevents[i].gpa_offset == 40 - (IOEVENT_SIZE * i)
         assert ioevents[i].type == VFIO_USER_IO_FD_TYPE_IOEVENTFD
 
     for i in newfds:
@@ -284,11 +284,11 @@ def test_device_get_region_io_fds_fds_read_write_dupe_fd():
         [out] = struct.unpack("@Q", out)
         assert out == 1
         assert ioevents[i].size == IOEVENT_SIZE
-        assert ioevents[i].offset == 56 - (IOEVENT_SIZE * i)
+        assert ioevents[i].gpa_offset == 56 - (IOEVENT_SIZE * i)
         assert ioevents[i].type == VFIO_USER_IO_FD_TYPE_IOEVENTFD
 
     assert ioevents[0].fd_index == ioevents[1].fd_index
-    assert ioevents[0].offset != ioevents[1].offset
+    assert ioevents[0].gpa_offset != ioevents[1].gpa_offset
 
     os.write(newfds[ioevents[0].fd_index], c.c_ulonglong(1))
 
diff --git a/test/py/test_shadow_ioeventfd.py b/test/py/test_shadow_ioeventfd.py
index 642ad0e..c7cc4e5 100644
--- a/test/py/test_shadow_ioeventfd.py
+++ b/test/py/test_shadow_ioeventfd.py
@@ -54,7 +54,7 @@ def test_shadow_ioeventfd():
 
     efd = eventfd(flags=EFD_NONBLOCK)
     ret = vfu_create_ioeventfd(ctx, VFU_PCI_DEV_BAR0_REGION_IDX, efd, 0x8,
-                               0x16, 0, 0, shadow_fd=fo.fileno())
+                               0x16, 0, 0, shadow_fd=fo.fileno(), 0x10)
     assert ret == 0
     ret = vfu_realize_ctx(ctx)
     assert ret == 0
@@ -70,12 +70,13 @@ def test_shadow_ioeventfd():
     reply, ret = vfio_user_region_io_fds_reply.pop_from_buffer(ret)
     assert reply.count == 1  # 1 eventfd
     ioevent, _ = vfio_user_sub_region_ioeventfd.pop_from_buffer(ret)
-    assert ioevent.offset == 0x8
+    assert ioevent.gpa_offset == 0x8
     assert ioevent.size == 0x16
     assert ioevent.fd_index == 0
     assert ioevent.type == VFIO_USER_IO_FD_TYPE_IOEVENTFD_SHADOW
     assert ioevent.flags == 0
     assert ioevent.datamatch == 0
+    assert ioevent.shadow_offset = 0x10
 
     assert len(newfds) == 2  # 2 FDs: eventfd plus shadow FD
     cefd = newfds[0]
@@ -93,13 +94,13 @@ def test_shadow_ioeventfd():
 
     # Client writes to the I/O region. The write to the eventfd would be done
     # by KVM and the value would be the same in both cases.
-    cmem.seek(0x8)
+    cmem.seek(ioevent.shadow_offset)
     cmem.write(c.c_ulonglong(0xdeadbeef))
     os.write(cefd, c.c_ulonglong(0xcafebabe))
 
     # vfio-user app reads eventfd
     assert os.read(efd, IOEVENT_SIZE) == to_bytes_le(0xcafebabe, 8)
-    fo.seek(0x8)
+    fo.seek(ioevent.shadow_offset)
     assert fo.read(0x8) == to_bytes_le(0xdeadbeef, 8)
 
     vfu_destroy_ctx(ctx)