aboutsummaryrefslogtreecommitdiff
path: root/test
diff options
context:
space:
mode:
Diffstat (limited to 'test')
-rw-r--r--test/py/libvfio_user.py27
-rw-r--r--test/py/test_dma_map.py25
-rw-r--r--test/py/test_map_unmap_sg.py117
-rw-r--r--test/unit-tests.c68
4 files changed, 145 insertions, 92 deletions
diff --git a/test/py/libvfio_user.py b/test/py/libvfio_user.py
index 8cd1c5d..da7f1fb 100644
--- a/test/py/libvfio_user.py
+++ b/test/py/libvfio_user.py
@@ -568,6 +568,30 @@ def ext_cap_hdr(buf, offset):
cap_next >>= 4
return cap_id, cap_next
+@vfu_dma_register_cb_t
+def dma_register(ctx, info):
+ pass
+
+@vfu_dma_unregister_cb_t
+def dma_unregister(ctx, info):
+ pass
+ return 0
+
+def prepare_ctx_for_dma():
+ ctx = vfu_create_ctx(flags=LIBVFIO_USER_FLAG_ATTACH_NB)
+ assert ctx != None
+
+ ret = vfu_pci_init(ctx)
+ assert ret == 0
+
+ ret = vfu_setup_device_dma(ctx, dma_register, dma_unregister)
+ assert ret == 0
+
+ ret = vfu_realize_ctx(ctx)
+ assert ret == 0
+
+ return ctx
+
#
# Library wrappers
#
@@ -715,13 +739,12 @@ def vfu_addr_to_sg(ctx, dma_addr, length, max_sg=1,
prot=(mmap.PROT_READ | mmap.PROT_WRITE)):
assert ctx != None
- sg = dma_sg_t()
+ sg = (dma_sg_t * max_sg)()
return (lib.vfu_addr_to_sg(ctx, dma_addr, length, sg, max_sg, prot), sg)
def vfu_map_sg(ctx, sg, iovec, cnt=1, flags=0):
- # FIXME not sure wheter cnt != 1 will work because iovec is an array
return lib.vfu_map_sg(ctx, sg, iovec, cnt, flags)
def vfu_unmap_sg(ctx, sg, iovec, cnt=1):
diff --git a/test/py/test_dma_map.py b/test/py/test_dma_map.py
index fe48c45..e4ac2ba 100644
--- a/test/py/test_dma_map.py
+++ b/test/py/test_dma_map.py
@@ -36,31 +36,12 @@ import errno
ctx = None
-@vfu_dma_register_cb_t
-def dma_register(ctx, info):
- pass
-
-@vfu_dma_unregister_cb_t
-def dma_unregister(ctx, info):
- pass
- return 0
-
-def test_dma_map_setup():
+def test_dma_region_too_big():
global ctx
- ctx = vfu_create_ctx(flags=LIBVFIO_USER_FLAG_ATTACH_NB)
+ ctx = prepare_ctx_for_dma()
assert ctx != None
- ret = vfu_pci_init(ctx)
- assert ret == 0
-
- ret = vfu_setup_device_dma(ctx, dma_register, dma_unregister)
- assert ret == 0
-
- ret = vfu_realize_ctx(ctx)
- assert ret == 0
-
-def test_dma_region_too_big():
sock = connect_client(ctx)
payload = vfio_user_dma_map(argsz=len(vfio_user_dma_map()),
@@ -90,5 +71,5 @@ def test_dma_region_too_many():
disconnect_client(ctx, sock)
-def test_dirty_pages_cleanup():
+def test_dma_region_cleanup():
vfu_destroy_ctx(ctx)
diff --git a/test/py/test_map_unmap_sg.py b/test/py/test_map_unmap_sg.py
new file mode 100644
index 0000000..00d9b28
--- /dev/null
+++ b/test/py/test_map_unmap_sg.py
@@ -0,0 +1,117 @@
+#
+# Copyright (c) 2021 Nutanix Inc. All rights reserved.
+#
+# Authors: Swapnil Ingle <swapnil.ingle@nutanix.com>
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+# * Redistributions of source code must retain the above copyright
+# notice, this list of conditions and the following disclaimer.
+# * Redistributions in binary form must reproduce the above copyright
+# notice, this list of conditions and the following disclaimer in the
+# documentation and/or other materials provided with the distribution.
+# * Neither the name of Nutanix nor the names of its contributors may be
+# used to endorse or promote products derived from this software without
+# specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+# ARE DISCLAIMED. IN NO EVENT SHALL <COPYRIGHT HOLDER> BE LIABLE FOR ANY
+# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
+# DAMAGE.
+#
+
+import ctypes
+import errno
+from libvfio_user import *
+import tempfile
+
+ctx = None
+
+def test_map_sg_with_invalid_region():
+ global ctx
+
+ ctx = prepare_ctx_for_dma()
+ assert ctx != None
+
+ sg = dma_sg_t()
+ iovec = iovec_t()
+ ret = vfu_map_sg(ctx, sg, iovec)
+ assert ret == -1
+ assert ctypes.get_errno() == errno.EINVAL
+
+def test_map_sg_without_fd():
+ sock = connect_client(ctx)
+
+ payload = vfio_user_dma_map(argsz=len(vfio_user_dma_map()),
+ flags=(VFIO_USER_F_DMA_REGION_READ |
+ VFIO_USER_F_DMA_REGION_WRITE),
+ offset=0, addr=0x1000, size=4096)
+ msg(ctx, sock, VFIO_USER_DMA_MAP, payload)
+ sg = dma_sg_t()
+ iovec = iovec_t()
+ sg.region = 0
+ ret = vfu_map_sg(ctx, sg, iovec)
+ assert ret == -1
+ assert ctypes.get_errno() == errno.EFAULT
+
+ disconnect_client(ctx, sock)
+
+def test_map_multiple_sge():
+ sock = connect_client(ctx)
+ regions = 4
+ f = tempfile.TemporaryFile()
+ f.truncate(0x1000 * regions)
+ for i in range(1, regions):
+ payload = vfio_user_dma_map(argsz=len(vfio_user_dma_map()),
+ flags=(VFIO_USER_F_DMA_REGION_READ |
+ VFIO_USER_F_DMA_REGION_WRITE),
+ offset=0, addr=0x1000 * i, size=4096)
+ msg(ctx, sock, VFIO_USER_DMA_MAP, payload, fds=[f.fileno()])
+
+ ret, sg = vfu_addr_to_sg(ctx, dma_addr=0x1000, length=4096 * 3, max_sg=3,
+ prot=mmap.PROT_READ)
+ assert ret == 3
+
+ iovec = (iovec_t * 3)()
+ ret = vfu_map_sg(ctx, sg, iovec, cnt=3)
+ assert ret == 0
+ assert iovec[0].iov_len == 4096
+ assert iovec[1].iov_len == 4096
+ assert iovec[2].iov_len == 4096
+
+ disconnect_client(ctx, sock)
+
+def test_unmap_sg():
+ sock = connect_client(ctx)
+ regions = 4
+ f = tempfile.TemporaryFile()
+ f.truncate(0x1000 * regions)
+ for i in range(1, regions):
+ payload = vfio_user_dma_map(argsz=len(vfio_user_dma_map()),
+ flags=(VFIO_USER_F_DMA_REGION_READ |
+ VFIO_USER_F_DMA_REGION_WRITE),
+ offset=0, addr=0x1000 * i, size=4096)
+ msg(ctx, sock, VFIO_USER_DMA_MAP, payload, fds=[f.fileno()])
+
+ ret, sg = vfu_addr_to_sg(ctx, dma_addr=0x1000, length=4096 * 3, max_sg=3,
+ prot=mmap.PROT_READ)
+ assert ret == 3
+
+ iovec = (iovec_t * 3)()
+ ret = vfu_map_sg(ctx, sg, iovec, cnt=3)
+ assert ret == 0
+ vfu_unmap_sg(ctx, sg, iovec, cnt=3)
+
+ disconnect_client(ctx, sock)
+
+def test_map_unmap_sg_cleanup():
+ vfu_destroy_ctx(ctx)
+
+# ex: set tabstop=4 shiftwidth=4 softtabstop=4 expandtab:
diff --git a/test/unit-tests.c b/test/unit-tests.c
index b605728..9dc222e 100644
--- a/test/unit-tests.c
+++ b/test/unit-tests.c
@@ -358,72 +358,6 @@ test_dma_controller_remove_region_unmapped(void **state UNUSED)
}
static void
-test_dma_map_sg(void **state UNUSED)
-{
- dma_sg_t sg[] = { {.region = 0}, {.region = 1}, {.region = 2} };
- struct iovec iovec[] = { {0}, {0}, {0} };
-
- /* bad region */
- assert_int_equal(-1, dma_map_sg(vfu_ctx.dma, &sg[0], &iovec[0], 1));
- assert_int_equal(EINVAL, errno);
-
- vfu_ctx.dma->nregions = 1;
-
- /* w/o fd */
- sg[0].region = 0;
- assert_int_equal(-1, dma_map_sg(vfu_ctx.dma, &sg[0], &iovec[0], 1));
- assert_int_equal(EFAULT, errno);
-
- /* w/ fd */
- vfu_ctx.dma->regions[0].info.vaddr = (void *)0xdead0000;
-
- sg[0].offset = 0x0000beef;
- sg[0].length = 0xcafebabe;
- assert_int_equal(0, dma_map_sg(vfu_ctx.dma, &sg[0], &iovec[0], 1));
- assert_int_equal(0xdeadbeef, iovec[0].iov_base);
- assert_int_equal(0x00000000cafebabe, iovec[0].iov_len);
-
- /* multiple sg entries */
- vfu_ctx.dma->nregions = 3;
- vfu_ctx.dma->regions[0].info.vaddr = (void *)0xc0000;
- sg[0].offset = 0;
- sg[0].length = 0x20000;
- vfu_ctx.dma->regions[1].info.vaddr = (void *)0xe0000;
- sg[1].length = 0x20000;
- vfu_ctx.dma->regions[2].info.vaddr = (void *)0x100000;
- sg[2].length = 0x10000;
- assert_int_equal(0, dma_map_sg(vfu_ctx.dma, sg, iovec, 3));
- assert_int_equal(0xc0000, iovec[0].iov_base);
- assert_int_equal(0x20000, iovec[0].iov_len);
- assert_int_equal(0xe0000, iovec[1].iov_base);
- assert_int_equal(0x20000, iovec[1].iov_len);
- assert_int_equal(0x100000, iovec[2].iov_base);
- assert_int_equal(0x10000, iovec[2].iov_len);
-}
-
-static void
-test_dma_unmap_sg(void **state UNUSED)
-{
- dma_sg_t sg[] = { {.region = 0}, {.region = 1}, {.region = 2} };
- struct iovec iovec[] = { {0}, {0}, {0} };
-
- vfu_ctx.dma->nregions = 3;
- vfu_ctx.dma->regions[0].info.iova.iov_base = (void *)0xc0000;
- vfu_ctx.dma->regions[0].refcnt = 1;
- sg[0].dma_addr = (void *)0xc0000;
- vfu_ctx.dma->regions[1].info.iova.iov_base = (void *)0xe0000;
- vfu_ctx.dma->regions[1].refcnt = 1;
- sg[1].dma_addr = (void *)0xe0000;
- vfu_ctx.dma->regions[2].info.iova.iov_base = (void *)0x100000;
- vfu_ctx.dma->regions[2].refcnt = 1;
- sg[2].dma_addr = (void *)0x100000;
- dma_unmap_sg(vfu_ctx.dma, sg, iovec, 3);
- assert_int_equal(0, vfu_ctx.dma->regions[0].refcnt);
- assert_int_equal(0, vfu_ctx.dma->regions[1].refcnt);
- assert_int_equal(0, vfu_ctx.dma->regions[2].refcnt);
-}
-
-static void
test_dma_addr_to_sg(void **state UNUSED)
{
dma_memory_region_t *r, *r1;
@@ -787,8 +721,6 @@ main(void)
cmocka_unit_test_setup(test_dma_controller_add_region_no_fd, setup),
cmocka_unit_test_setup(test_dma_controller_remove_region_mapped, setup),
cmocka_unit_test_setup(test_dma_controller_remove_region_unmapped, setup),
- cmocka_unit_test_setup(test_dma_map_sg, setup),
- cmocka_unit_test_setup(test_dma_unmap_sg, setup),
cmocka_unit_test_setup(test_dma_addr_to_sg, setup),
cmocka_unit_test_setup(test_vfu_setup_device_dma, setup),
cmocka_unit_test_setup(test_migration_state_transitions, setup),
generated by cgit v1.1 at 2024-07-19 14:19:24 +0000