diff options
author | Swapnil Ingle <swapnil.ingle@nutanix.com> | 2021-08-20 16:21:07 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-08-20 16:21:07 +0200 |
commit | 647c9341d2e06266a710ddd075f69c95dd3b8446 (patch) | |
tree | dbdaf047ce03bc7876860064ba0a2d5c2182bbe3 /test/py | |
parent | 66216488980a40d3cdde892d633a3deeb1d4786f (diff) | |
download | libvfio-user-647c9341d2e06266a710ddd075f69c95dd3b8446.zip libvfio-user-647c9341d2e06266a710ddd075f69c95dd3b8446.tar.gz libvfio-user-647c9341d2e06266a710ddd075f69c95dd3b8446.tar.bz2 |
pytests for vfu_dma_{map, unmap}_sg (#596)
* pytests for vfu_dma_{map, unmap}_sg
Signed-off-by: Swapnil Ingle <swapnil.ingle@nutanix.com>
Reviewed-by: John Levon <john.levon@nutanix.com>
Diffstat (limited to 'test/py')
-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 |
3 files changed, 145 insertions, 24 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: |