1 /* SPDX-License-Identifier: GPL-2.0+ */
2 /*
3  * Copyright (c) 2022 Google, Inc.
4  * Written by Andrew Scull <ascull@google.com>
5  */
6 
7 #ifndef __TEST_FUZZ_H
8 #define __TEST_FUZZ_H
9 
10 #include <linker_lists.h>
11 #include <linux/types.h>
12 
13 /**
14  * struct fuzz_test - Information about a fuzz test
15  *
16  * @name: Name of fuzz test
17  * @func: Function to call to perform fuzz test on an input
18  * @flags: Flags indicate pre-conditions for fuzz test
19  */
20 struct fuzz_test {
21 	const char *name;
22 	int (*func)(const uint8_t * data, size_t size);
23 	int flags;
24 };
25 
26 /**
27  * FUZZ_TEST() - register a fuzz test
28  *
29  * The fuzz test function must return 0 as other values are reserved for future
30  * use.
31  *
32  * @_name:	the name of the fuzz test function
33  * @_flags:	an integer field that can be evaluated by the fuzzer
34  * 		implementation
35  */
36 #define FUZZ_TEST(_name, _flags)					\
37 	ll_entry_declare(struct fuzz_test, _name, fuzz_tests) = {	\
38 		.name = #_name,						\
39 		.func = _name,						\
40 		.flags = _flags,					\
41 	}
42 
43 /** Get the start of the list of fuzz tests */
44 #define FUZZ_TEST_START() \
45 	ll_entry_start(struct fuzz_test, fuzz_tests)
46 
47 /** Get the number of elements in the list of fuzz tests */
48 #define FUZZ_TEST_COUNT() \
49 	ll_entry_count(struct fuzz_test, fuzz_tests)
50 
51 #endif /* __TEST_FUZZ_H */
52