LXR linux/tools/testing/selftests/futex/functional/futex

   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
  17volatile futex_t *f1;
  18
  19void 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
  28void *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
  41int 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}
 137