Home | History | Annotate | Download | only in membarrier 
      1 // SPDX-License-Identifier: GPL-2.0
      2 #define _GNU_SOURCE
      3 #include <linux/membarrier.h>
      4 #include <syscall.h>
      5 #include <stdio.h>
      6 #include <errno.h>
      7 #include <string.h>
      8 
      9 #include "../kselftest.h"
     10 
     11 static int sys_membarrier(int cmd, int flags)
     12 {
     13 	return syscall(__NR_membarrier, cmd, flags);
     14 }
     15 
     16 static int test_membarrier_cmd_fail(void)
     17 {
     18 	int cmd = -1, flags = 0;
     19 
     20 	if (sys_membarrier(cmd, flags) != -1) {
     21 		ksft_exit_fail_msg(
     22 			"sys membarrier invalid command test: command = %d, flags = %d. Should fail, but passed\n",
     23 			cmd, flags);
     24 	}
     25 
     26 	ksft_test_result_pass(
     27 		"sys membarrier invalid command test: command = %d, flags = %d. Failed as expected\n",
     28 		cmd, flags);
     29 	return 0;
     30 }
     31 
     32 static int test_membarrier_flags_fail(void)
     33 {
     34 	int cmd = MEMBARRIER_CMD_QUERY, flags = 1;
     35 
     36 	if (sys_membarrier(cmd, flags) != -1) {
     37 		ksft_exit_fail_msg(
     38 			"sys membarrier MEMBARRIER_CMD_QUERY invalid flags test: flags = %d. Should fail, but passed\n",
     39 			flags);
     40 	}
     41 
     42 	ksft_test_result_pass(
     43 		"sys membarrier MEMBARRIER_CMD_QUERY invalid flags test: flags = %d. Failed as expected\n",
     44 		flags);
     45 	return 0;
     46 }
     47 
     48 static int test_membarrier_success(void)
     49 {
     50 	int cmd = MEMBARRIER_CMD_SHARED, flags = 0;
     51 	const char *test_name = "sys membarrier MEMBARRIER_CMD_SHARED\n";
     52 
     53 	if (sys_membarrier(cmd, flags) != 0) {
     54 		ksft_exit_fail_msg(
     55 			"sys membarrier MEMBARRIER_CMD_SHARED test: flags = %d\n",
     56 			flags);
     57 	}
     58 
     59 	ksft_test_result_pass(
     60 		"sys membarrier MEMBARRIER_CMD_SHARED test: flags = %d\n",
     61 		flags);
     62 	return 0;
     63 }
     64 
     65 static int test_membarrier(void)
     66 {
     67 	int status;
     68 
     69 	status = test_membarrier_cmd_fail();
     70 	if (status)
     71 		return status;
     72 	status = test_membarrier_flags_fail();
     73 	if (status)
     74 		return status;
     75 	status = test_membarrier_success();
     76 	if (status)
     77 		return status;
     78 	return 0;
     79 }
     80 
     81 static int test_membarrier_query(void)
     82 {
     83 	int flags = 0, ret;
     84 
     85 	ret = sys_membarrier(MEMBARRIER_CMD_QUERY, flags);
     86 	if (ret < 0) {
     87 		if (errno == ENOSYS) {
     88 			/*
     89 			 * It is valid to build a kernel with
     90 			 * CONFIG_MEMBARRIER=n. However, this skips the tests.
     91 			 */
     92 			ksft_exit_skip(
     93 				"sys membarrier (CONFIG_MEMBARRIER) is disabled.\n");
     94 		}
     95 		ksft_exit_fail_msg("sys_membarrier() failed\n");
     96 	}
     97 	if (!(ret & MEMBARRIER_CMD_SHARED))
     98 		ksft_exit_fail_msg("sys_membarrier is not supported.\n");
     99 
    100 	ksft_test_result_pass("sys_membarrier available\n");
    101 	return 0;
    102 }
    103 
    104 int main(int argc, char **argv)
    105 {
    106 	ksft_print_header();
    107 
    108 	test_membarrier_query();
    109 	test_membarrier();
    110 
    111 	ksft_exit_pass();
    112 }
    113