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
42static const char	*sysctl_oid_name = "vfs.aio.enable_unsafe";
43
44static int
45is_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_ */
90