tools/testing/selftests/futex/functional/futex_requeue.c

   1 // SPDX-License-Identifier: GPL-2.0-or-later
   2 /*
   3  * Copyright Collabora Ltd., 2021
   4  *
   5  * futex cmp requeue test by André Almeida <andrealmeid@collabora.com>
   6  */
   7
   8 #include <pthread.h>
   9 #include <limits.h>
  10 #include "logging.h"
  11 #include "futextest.h"
  12
  13 #define TEST_NAME "futex-requeue"
  14 #define timeout_ns  30000000
  15 #define WAKE_WAIT_US 10000
  16
  17 volatile futex_t *f1;
  18
  19 void usage(char *prog)
  20 {
  21         printf("Usage: %s\n", prog);
  22         printf("  -c    Use color\n");
  23         printf("  -h    Display this help message\n");
  24         printf("  -v L  Verbosity level: %d=QUIET %d=CRITICAL %d=INFO\n",
  25                VQUIET, VCRITICAL, VINFO);
  26 }
  27
  28 void *waiterfn(void *arg)
  29 {
  30         struct timespec to;
  31
  32         to.tv_sec = 0;
  33         to.tv_nsec = timeout_ns;
  34
  35         if (futex_wait(f1, *f1, &to, 0))
  36                 printf("waiter failed errno %d\n", errno);
  37
  38         return NULL;
  39 }
  40
  41 int main(int argc, char *argv[])
  42 {
  43         pthread_t waiter[10];
  44         int res, ret = RET_PASS;
  45         int c, i;
  46         volatile futex_t _f1 = 0;
  47         volatile futex_t f2 = 0;
  48
  49         f1 = &_f1;
  50
  51         while ((c = getopt(argc, argv, "cht:v:")) != -1) {
  52                 switch (c) {
  53                 case 'c':
  54                         log_color(1);
  55                         break;
  56                 case 'h':
  57                         usage(basename(argv[0]));
  58                         exit(0);
  59                 case 'v':
  60                         log_verbosity(atoi(optarg));
  61                         break;
  62                 default:
  63                         usage(basename(argv[0]));
  64                         exit(1);
  65                 }
  66         }
  67
  68         ksft_print_header();
  69         ksft_set_plan(2);
  70         ksft_print_msg("%s: Test futex_requeue\n",
  71                        basename(argv[0]));
  72
  73         /*
  74          * Requeue a waiter from f1 to f2, and wake f2.
  75          */
  76         if (pthread_create(&waiter[0], NULL, waiterfn, NULL))
  77                 error("pthread_create failed\n", errno);
  78
  79         usleep(WAKE_WAIT_US);
  80
  81         info("Requeuing 1 futex from f1 to f2\n");
  82         res = futex_cmp_requeue(f1, 0, &f2, 0, 1, 0);
  83         if (res != 1) {
  84                 ksft_test_result_fail("futex_requeue simple returned: %d %s\n",
  85                                       res ? errno : res,
  86                                       res ? strerror(errno) : "");
  87                 ret = RET_FAIL;
  88         }
  89
  90
  91         info("Waking 1 futex at f2\n");
  92         res = futex_wake(&f2, 1, 0);
  93         if (res != 1) {
  94                 ksft_test_result_fail("futex_requeue simple returned: %d %s\n",
  95                                       res ? errno : res,
  96                                       res ? strerror(errno) : "");
  97                 ret = RET_FAIL;
  98         } else {
  99                 ksft_test_result_pass("futex_requeue simple succeeds\n");
 100         }
 101
 102
 103         /*
 104          * Create 10 waiters at f1. At futex_requeue, wake 3 and requeue 7.
 105          * At futex_wake, wake INT_MAX (should be exactly 7).
 106          */
 107         for (i = 0; i < 10; i++) {
 108                 if (pthread_create(&waiter[i], NULL, waiterfn, NULL))
 109                         error("pthread_create failed\n", errno);
 110         }
 111
 112         usleep(WAKE_WAIT_US);
 113
 114         info("Waking 3 futexes at f1 and requeuing 7 futexes from f1 to f2\n");
 115         res = futex_cmp_requeue(f1, 0, &f2, 3, 7, 0);
 116         if (res != 10) {
 117                 ksft_test_result_fail("futex_requeue many returned: %d %s\n",
 118                                       res ? errno : res,
 119                                       res ? strerror(errno) : "");
 120                 ret = RET_FAIL;
 121         }
 122
 123         info("Waking INT_MAX futexes at f2\n");
 124         res = futex_wake(&f2, INT_MAX, 0);
 125         if (res != 7) {
 126                 ksft_test_result_fail("futex_requeue many returned: %d %s\n",
 127                                       res ? errno : res,
 128                                       res ? strerror(errno) : "");
 129                 ret = RET_FAIL;
 130         } else {
 131                 ksft_test_result_pass("futex_requeue many succeeds\n");
 132         }
 133
 134         ksft_print_cnts();
 135         return ret;
 136 }