LCOV - code coverage report
Current view: top level - spdk/test/unit/lib/sock/uring.c - uring_ut.c (source / functions) Hit Total Coverage
Test: Combined Lines: 156 166 94.0 %
Date: 2024-07-13 02:07:07 Functions: 6 13 46.2 %
Legend: Lines: hit not hit | Branches: + taken - not taken # not executed Branches: 19 38 50.0 %

           Branch data     Line data    Source code
       1                 :            : /*   SPDX-License-Identifier: BSD-3-Clause
       2                 :            :  *   Copyright (C) 2019 Intel Corporation.
       3                 :            :  *   All rights reserved.
       4                 :            :  */
       5                 :            : 
       6                 :            : #include "spdk/stdinc.h"
       7                 :            : #include "spdk/util.h"
       8                 :            : 
       9                 :            : #include "spdk_internal/mock.h"
      10                 :            : 
      11                 :            : #include "spdk_internal/cunit.h"
      12                 :            : 
      13                 :            : #include "common/lib/test_env.c"
      14                 :            : #include "sock/uring/uring.c"
      15                 :            : 
      16                 :          0 : DEFINE_STUB(spdk_sock_map_insert, int, (struct spdk_sock_map *map, int placement_id,
      17                 :            :                                         struct spdk_sock_group_impl *group), 0);
      18                 :          0 : DEFINE_STUB_V(spdk_sock_map_release, (struct spdk_sock_map *map, int placement_id));
      19                 :          0 : DEFINE_STUB(spdk_sock_map_lookup, int, (struct spdk_sock_map *map, int placement_id,
      20                 :            :                                         struct spdk_sock_group_impl **group, struct spdk_sock_group_impl *hint), 0);
      21                 :          0 : DEFINE_STUB(spdk_sock_map_find_free, int, (struct spdk_sock_map *map), -1);
      22                 :          2 : DEFINE_STUB_V(spdk_sock_map_cleanup, (struct spdk_sock_map *map));
      23                 :            : 
      24                 :          2 : DEFINE_STUB_V(spdk_net_impl_register, (struct spdk_net_impl *impl, int priority));
      25                 :          0 : DEFINE_STUB(spdk_sock_close, int, (struct spdk_sock **s), 0);
      26                 :          0 : DEFINE_STUB(io_uring_submit, int, (struct io_uring *ring), 0);
      27                 :          0 : DEFINE_STUB(io_uring_queue_init, int, (unsigned entries, struct io_uring *ring, unsigned flags), 0);
      28                 :          0 : DEFINE_STUB_V(io_uring_queue_exit, (struct io_uring *ring));
      29                 :          0 : DEFINE_STUB(spdk_sock_group_provide_buf, int, (struct spdk_sock_group *group, void *buf,
      30                 :            :                 size_t len, void *ctx), 0);
      31                 :          0 : DEFINE_STUB(spdk_sock_group_get_buf, size_t, (struct spdk_sock_group *group, void **buf,
      32                 :            :                 void **ctx), 0);
      33                 :            : 
      34                 :            : static void
      35                 :         18 : _req_cb(void *cb_arg, int len)
      36                 :            : {
      37                 :         18 :         *(bool *)cb_arg = true;
      38                 :         18 :         CU_ASSERT(len == 0);
      39                 :         18 : }
      40                 :            : 
      41                 :            : static void
      42                 :          2 : flush_client(void)
      43                 :            : {
      44                 :          2 :         struct spdk_uring_sock_group_impl group = {};
      45                 :          2 :         struct spdk_uring_sock usock = {};
      46                 :          2 :         struct spdk_sock *sock = &usock.base;
      47                 :            :         struct spdk_sock_request *req1, *req2;
      48                 :          1 :         bool cb_arg1, cb_arg2;
      49                 :            :         int rc;
      50                 :            : 
      51                 :            :         /* Set up data structures */
      52                 :          2 :         TAILQ_INIT(&sock->queued_reqs);
      53                 :          2 :         TAILQ_INIT(&sock->pending_reqs);
      54                 :          2 :         sock->group_impl = &group.base;
      55                 :            : 
      56                 :          2 :         req1 = calloc(1, sizeof(struct spdk_sock_request) + 3 * sizeof(struct iovec));
      57         [ -  + ]:          2 :         SPDK_CU_ASSERT_FATAL(req1 != NULL);
      58                 :          2 :         SPDK_SOCK_REQUEST_IOV(req1, 0)->iov_base = (void *)100;
      59                 :          2 :         SPDK_SOCK_REQUEST_IOV(req1, 0)->iov_len = 64;
      60                 :          2 :         SPDK_SOCK_REQUEST_IOV(req1, 1)->iov_base = (void *)200;
      61                 :          2 :         SPDK_SOCK_REQUEST_IOV(req1, 1)->iov_len = 64;
      62                 :          2 :         SPDK_SOCK_REQUEST_IOV(req1, 2)->iov_base = (void *)300;
      63                 :          2 :         SPDK_SOCK_REQUEST_IOV(req1, 2)->iov_len = 64;
      64                 :          2 :         req1->iovcnt = 3;
      65                 :          2 :         req1->cb_fn = _req_cb;
      66                 :          2 :         req1->cb_arg = &cb_arg1;
      67                 :            : 
      68                 :          2 :         req2 = calloc(1, sizeof(struct spdk_sock_request) + 2 * sizeof(struct iovec));
      69         [ -  + ]:          2 :         SPDK_CU_ASSERT_FATAL(req2 != NULL);
      70                 :          2 :         SPDK_SOCK_REQUEST_IOV(req2, 0)->iov_base = (void *)100;
      71                 :          2 :         SPDK_SOCK_REQUEST_IOV(req2, 0)->iov_len = 32;
      72                 :          2 :         SPDK_SOCK_REQUEST_IOV(req2, 1)->iov_base = (void *)200;
      73                 :          2 :         SPDK_SOCK_REQUEST_IOV(req2, 1)->iov_len = 32;
      74                 :          2 :         req2->iovcnt = 2;
      75                 :          2 :         req2->cb_fn = _req_cb;
      76                 :          2 :         req2->cb_arg = &cb_arg2;
      77                 :            : 
      78                 :            :         /* Simple test - a request with a 3 element iovec
      79                 :            :          * that gets submitted in a single sendmsg. */
      80                 :          2 :         spdk_sock_request_queue(sock, req1);
      81                 :          2 :         MOCK_SET(sendmsg, 192);
      82                 :          2 :         cb_arg1 = false;
      83                 :          2 :         rc = uring_sock_flush(sock);
      84                 :          2 :         CU_ASSERT(rc == 192);
      85         [ -  + ]:          2 :         CU_ASSERT(cb_arg1 == true);
      86                 :          2 :         CU_ASSERT(TAILQ_EMPTY(&sock->queued_reqs));
      87                 :            : 
      88                 :            :         /* Two requests, where both can fully send. */
      89                 :          2 :         spdk_sock_request_queue(sock, req1);
      90                 :          2 :         spdk_sock_request_queue(sock, req2);
      91                 :          2 :         MOCK_SET(sendmsg, 256);
      92                 :          2 :         cb_arg1 = false;
      93                 :          2 :         cb_arg2 = false;
      94                 :          2 :         rc = uring_sock_flush(sock);
      95                 :          2 :         CU_ASSERT(rc == 256);
      96         [ -  + ]:          2 :         CU_ASSERT(cb_arg1 == true);
      97         [ -  + ]:          2 :         CU_ASSERT(cb_arg2 == true);
      98                 :          2 :         CU_ASSERT(TAILQ_EMPTY(&sock->queued_reqs));
      99                 :            : 
     100                 :            :         /* Two requests. Only first one can send */
     101                 :          2 :         spdk_sock_request_queue(sock, req1);
     102                 :          2 :         spdk_sock_request_queue(sock, req2);
     103                 :          2 :         MOCK_SET(sendmsg, 192);
     104                 :          2 :         cb_arg1 = false;
     105                 :          2 :         cb_arg2 = false;
     106                 :          2 :         rc = uring_sock_flush(sock);
     107                 :          2 :         CU_ASSERT(rc == 192);
     108         [ -  + ]:          2 :         CU_ASSERT(cb_arg1 == true);
     109         [ -  + ]:          2 :         CU_ASSERT(cb_arg2 == false);
     110                 :          2 :         CU_ASSERT(TAILQ_FIRST(&sock->queued_reqs) == req2);
     111         [ -  + ]:          2 :         TAILQ_REMOVE(&sock->queued_reqs, req2, internal.link);
     112                 :          2 :         CU_ASSERT(TAILQ_EMPTY(&sock->queued_reqs));
     113                 :            : 
     114                 :            :         /* One request. Partial send. */
     115                 :          2 :         spdk_sock_request_queue(sock, req1);
     116                 :          2 :         MOCK_SET(sendmsg, 10);
     117                 :          2 :         cb_arg1 = false;
     118                 :          2 :         rc = uring_sock_flush(sock);
     119                 :          2 :         CU_ASSERT(rc == 10);
     120         [ -  + ]:          2 :         CU_ASSERT(cb_arg1 == false);
     121                 :          2 :         CU_ASSERT(TAILQ_FIRST(&sock->queued_reqs) == req1);
     122                 :            : 
     123                 :            :         /* Do a second flush that partial sends again. */
     124                 :          2 :         MOCK_SET(sendmsg, 52);
     125                 :          2 :         cb_arg1 = false;
     126                 :          2 :         rc = uring_sock_flush(sock);
     127                 :          2 :         CU_ASSERT(rc == 52);
     128         [ -  + ]:          2 :         CU_ASSERT(cb_arg1 == false);
     129                 :          2 :         CU_ASSERT(TAILQ_FIRST(&sock->queued_reqs) == req1);
     130                 :            : 
     131                 :            :         /* Flush the rest of the data */
     132                 :          2 :         MOCK_SET(sendmsg, 130);
     133                 :          2 :         cb_arg1 = false;
     134                 :          2 :         rc = uring_sock_flush(sock);
     135                 :          2 :         CU_ASSERT(rc == 130);
     136         [ -  + ]:          2 :         CU_ASSERT(cb_arg1 == true);
     137                 :          2 :         CU_ASSERT(TAILQ_EMPTY(&sock->queued_reqs));
     138                 :            : 
     139                 :          2 :         free(req1);
     140                 :          2 :         free(req2);
     141                 :          2 : }
     142                 :            : 
     143                 :            : static void
     144                 :          2 : flush_server(void)
     145                 :            : {
     146                 :          2 :         struct spdk_uring_sock_group_impl group = {};
     147                 :          2 :         struct spdk_uring_sock usock = {};
     148                 :          2 :         struct spdk_sock *sock = &usock.base;
     149                 :            :         struct spdk_sock_request *req1, *req2;
     150                 :          1 :         bool cb_arg1, cb_arg2;
     151                 :            :         int rc;
     152                 :            : 
     153                 :            :         /* Set up data structures */
     154                 :          2 :         TAILQ_INIT(&sock->queued_reqs);
     155                 :          2 :         TAILQ_INIT(&sock->pending_reqs);
     156                 :          2 :         sock->group_impl = &group.base;
     157                 :          2 :         usock.write_task.sock = &usock;
     158                 :          2 :         usock.group = &group;
     159                 :            : 
     160                 :          2 :         req1 = calloc(1, sizeof(struct spdk_sock_request) + 2 * sizeof(struct iovec));
     161         [ -  + ]:          2 :         SPDK_CU_ASSERT_FATAL(req1 != NULL);
     162                 :          2 :         SPDK_SOCK_REQUEST_IOV(req1, 0)->iov_base = (void *)100;
     163                 :          2 :         SPDK_SOCK_REQUEST_IOV(req1, 0)->iov_len = 64;
     164                 :          2 :         SPDK_SOCK_REQUEST_IOV(req1, 1)->iov_base = (void *)200;
     165                 :          2 :         SPDK_SOCK_REQUEST_IOV(req1, 1)->iov_len = 64;
     166                 :          2 :         req1->iovcnt = 2;
     167                 :          2 :         req1->cb_fn = _req_cb;
     168                 :          2 :         req1->cb_arg = &cb_arg1;
     169                 :            : 
     170                 :          2 :         req2 = calloc(1, sizeof(struct spdk_sock_request) + 2 * sizeof(struct iovec));
     171         [ -  + ]:          2 :         SPDK_CU_ASSERT_FATAL(req2 != NULL);
     172                 :          2 :         SPDK_SOCK_REQUEST_IOV(req2, 0)->iov_base = (void *)100;
     173                 :          2 :         SPDK_SOCK_REQUEST_IOV(req2, 0)->iov_len = 32;
     174                 :          2 :         SPDK_SOCK_REQUEST_IOV(req2, 1)->iov_base = (void *)200;
     175                 :          2 :         SPDK_SOCK_REQUEST_IOV(req2, 1)->iov_len = 32;
     176                 :          2 :         req2->iovcnt = 2;
     177                 :          2 :         req2->cb_fn = _req_cb;
     178                 :          2 :         req2->cb_arg = &cb_arg2;
     179                 :            : 
     180                 :            :         /* we should not call _sock_flush directly, since it will finally
     181                 :            :          * call liburing related functions  */
     182                 :            : 
     183                 :            :         /* Simple test - a request with a 2 element iovec
     184                 :            :          * that is fully completed. */
     185                 :          2 :         spdk_sock_request_queue(sock, req1);
     186                 :          2 :         cb_arg1 = false;
     187                 :          2 :         rc = spdk_sock_prep_reqs(sock, usock.write_task.iovs, 0, NULL, NULL);
     188                 :          2 :         CU_ASSERT(rc == 2);
     189                 :          2 :         sock_complete_write_reqs(sock, 128, 0);
     190         [ -  + ]:          2 :         CU_ASSERT(cb_arg1 == true);
     191                 :          2 :         CU_ASSERT(TAILQ_EMPTY(&sock->queued_reqs));
     192                 :            : 
     193                 :            :         /* Two requests, where both can be fully completed. */
     194                 :          2 :         spdk_sock_request_queue(sock, req1);
     195                 :          2 :         spdk_sock_request_queue(sock, req2);
     196                 :          2 :         cb_arg1 = false;
     197                 :          2 :         cb_arg2 = false;
     198                 :          2 :         rc = spdk_sock_prep_reqs(sock, usock.write_task.iovs, 0, NULL, NULL);
     199                 :          2 :         CU_ASSERT(rc == 4);
     200                 :          2 :         sock_complete_write_reqs(sock, 192, 0);
     201         [ -  + ]:          2 :         CU_ASSERT(cb_arg1 == true);
     202         [ -  + ]:          2 :         CU_ASSERT(cb_arg2 == true);
     203                 :          2 :         CU_ASSERT(TAILQ_EMPTY(&sock->queued_reqs));
     204                 :            : 
     205                 :            : 
     206                 :            :         /* One request that is partially sent. */
     207                 :          2 :         spdk_sock_request_queue(sock, req1);
     208                 :          2 :         cb_arg1 = false;
     209                 :          2 :         rc = spdk_sock_prep_reqs(sock, usock.write_task.iovs, 0, NULL, NULL);
     210                 :          2 :         CU_ASSERT(rc == 2);
     211                 :          2 :         sock_complete_write_reqs(sock, 92, 0);
     212                 :          2 :         CU_ASSERT(rc == 2);
     213         [ -  + ]:          2 :         CU_ASSERT(cb_arg1 == false);
     214                 :          2 :         CU_ASSERT(TAILQ_FIRST(&sock->queued_reqs) == req1);
     215                 :            : 
     216                 :            :         /* Get the second time partial sent result. */
     217                 :          2 :         sock_complete_write_reqs(sock, 10, 0);
     218         [ -  + ]:          2 :         CU_ASSERT(cb_arg1 == false);
     219                 :          2 :         CU_ASSERT(TAILQ_FIRST(&sock->queued_reqs) == req1);
     220                 :            : 
     221                 :            :         /* Data is finally sent. */
     222                 :          2 :         sock_complete_write_reqs(sock, 26, 0);
     223         [ -  + ]:          2 :         CU_ASSERT(cb_arg1 == true);
     224                 :          2 :         CU_ASSERT(TAILQ_EMPTY(&sock->queued_reqs));
     225                 :            : 
     226                 :          2 :         free(req1);
     227                 :          2 :         free(req2);
     228                 :          2 : }
     229                 :            : 
     230                 :            : int
     231                 :          2 : main(int argc, char **argv)
     232                 :            : {
     233                 :          2 :         CU_pSuite       suite = NULL;
     234                 :            :         unsigned int    num_failures;
     235                 :            : 
     236                 :          2 :         CU_initialize_registry();
     237                 :            : 
     238                 :          2 :         suite = CU_add_suite("uring", NULL, NULL);
     239                 :            : 
     240                 :            : 
     241                 :          2 :         CU_ADD_TEST(suite, flush_client);
     242                 :          2 :         CU_ADD_TEST(suite, flush_server);
     243                 :            : 
     244                 :            : 
     245                 :          2 :         num_failures = spdk_ut_run_tests(argc, argv, NULL);
     246                 :            : 
     247                 :          2 :         CU_cleanup_registry();
     248                 :            : 
     249                 :          2 :         return num_failures;
     250                 :            : }

Generated by: LCOV version 1.14