diff options
Diffstat (limited to 'test')
-rw-r--r-- | test/py/libvfio_user.py | 27 | ||||
-rw-r--r-- | test/py/test_dma_map.py | 25 | ||||
-rw-r--r-- | test/py/test_map_unmap_sg.py | 117 | ||||
-rw-r--r-- | test/unit-tests.c | 68 |
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), |