tests/sys/aio/local.h

   1 /*-
   2  * Copyright (c) 2016 Chelsio Communications, Inc.
   3  * All rights reserved.
   4  * Written by: John Baldwin <jhb@FreeBSD.org>
   5  *
   6  * Redistribution and use in source and binary forms, with or without
   7  * modification, are permitted provided that the following conditions
   8  * are met:
   9  * 1. Redistributions of source code must retain the above copyright
  10  *    notice, this list of conditions and the following disclaimer.
  11  * 2. Redistributions in binary form must reproduce the above copyright
  12  *    notice, this list of conditions and the following disclaimer in the
  13  *    documentation and/or other materials provided with the distribution.
  14  *
  15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  16  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  17  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  18  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  19  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  20  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  21  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  22  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  23  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  24  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  25  * SUCH DAMAGE.
  26  *
  27  * $FreeBSD$
  28  */
  29
  30 #ifndef _AIO_TEST_LOCAL_H_
  31 #define _AIO_TEST_LOCAL_H_
  32
  33 #include <sys/types.h>
  34 #include <sys/sysctl.h>
  35 #include <errno.h>
  36 #include <stdio.h>
  37 #include <string.h>
  38 #include <unistd.h>
  39
  40 #include <atf-c.h>
  41
  42 static const char       *sysctl_oid_name = "vfs.aio.enable_unsafe";
  43
  44 static int
  45 is_unsafe_aio_enabled(void)
  46 {
  47         size_t len;
  48         int unsafe;
  49
  50         len = sizeof(unsafe);
  51         if (sysctlbyname(sysctl_oid_name, &unsafe, &len, NULL, 0) < 0) {
  52                 if (errno == ENOENT)
  53                         return (-1);
  54                 return (0);
  55         }
  56         return (unsafe == 0 ? 0 : 1);
  57 }
  58
  59 #define ATF_REQUIRE_UNSAFE_AIO() do {                                           \
  60         switch (is_unsafe_aio_enabled()) {                                      \
  61         case -1:                                                                \
  62                 atf_libc_error(errno, "Failed to read %s", sysctl_oid_name);    \
  63                 break;                                                          \
  64         case 0:                                                                 \
  65                 atf_tc_skip("Unsafe AIO is disabled");                          \
  66                 break;                                                          \
  67         default:                                                                \
  68                 printf("Unsafe AIO is enabled\n");                              \
  69                 break;                                                          \
  70         }                                                                       \
  71 } while (0)
  72
  73 #define PLAIN_REQUIRE_UNSAFE_AIO(_exit_code) do {                               \
  74         switch (is_unsafe_aio_enabled()) {                                      \
  75         case -1:                                                                \
  76                 printf("Failed to read %s", sysctl_oid_name);                   \
  77                 _exit(_exit_code);                                              \
  78                 break;                                                          \
  79         case 0:                                                                 \
  80                 printf("Unsafe AIO is disabled\n");                             \
  81                 _exit(_exit_code);                                              \
  82                 break;                                                          \
  83         default:                                                                \
  84                 printf("Unsafe AIO is enabled\n");                              \
  85                 break;                                                          \
  86         }                                                                       \
  87 } while (0)
  88
  89 #endif /* !_AIO_TEST_LOCAL_H_ */