1/* SPDX-License-Identifier: GPL-2.0 */
2#ifndef __LKDTM_H
3#define __LKDTM_H
4
5#define pr_fmt(fmt) "lkdtm: " fmt
6
7#include <linux/kernel.h>
8
9extern char *lkdtm_kernel_info;
10
11#define pr_expected_config(kconfig)				\
12do {								\
13	if (IS_ENABLED(kconfig)) 				\
14		pr_err("Unexpected! This %s was built with " #kconfig "=y\n", \
15			lkdtm_kernel_info);			\
16	else							\
17		pr_warn("This is probably expected, since this %s was built *without* " #kconfig "=y\n", \
18			lkdtm_kernel_info);			\
19} while (0)
20
21#ifndef MODULE
22int lkdtm_check_bool_cmdline(const char *param);
23#define pr_expected_config_param(kconfig, param)		\
24do {								\
25	if (IS_ENABLED(kconfig)) {				\
26		switch (lkdtm_check_bool_cmdline(param)) {	\
27		case 0:						\
28			pr_warn("This is probably expected, since this %s was built with " #kconfig "=y but booted with '" param "=N'\n", \
29				lkdtm_kernel_info);		\
30			break;					\
31		case 1:						\
32			pr_err("Unexpected! This %s was built with " #kconfig "=y and booted with '" param "=Y'\n", \
33				lkdtm_kernel_info);		\
34			break;					\
35		default:					\
36			pr_err("Unexpected! This %s was built with " #kconfig "=y (and booted without '" param "' specified)\n", \
37				lkdtm_kernel_info);		\
38		}						\
39	} else {						\
40		switch (lkdtm_check_bool_cmdline(param)) {	\
41		case 0:						\
42			pr_warn("This is probably expected, as this %s was built *without* " #kconfig "=y and booted with '" param "=N'\n", \
43				lkdtm_kernel_info);		\
44			break;					\
45		case 1:						\
46			pr_err("Unexpected! This %s was built *without* " #kconfig "=y but booted with '" param "=Y'\n", \
47				lkdtm_kernel_info);		\
48			break;					\
49		default:					\
50			pr_err("This is probably expected, since this %s was built *without* " #kconfig "=y (and booted without '" param "' specified)\n", \
51				lkdtm_kernel_info);		\
52			break;					\
53		}						\
54	}							\
55} while (0)
56#else
57#define pr_expected_config_param(kconfig, param) pr_expected_config(kconfig)
58#endif
59
60/* Crash types. */
61struct crashtype {
62	const char *name;
63	void (*func)(void);
64};
65
66#define CRASHTYPE(_name)			\
67	{					\
68		.name = __stringify(_name),	\
69		.func = lkdtm_ ## _name,	\
70	}
71
72/* Category's collection of crashtypes. */
73struct crashtype_category {
74	struct crashtype *crashtypes;
75	size_t len;
76};
77
78/* Each category's crashtypes list. */
79extern struct crashtype_category bugs_crashtypes;
80extern struct crashtype_category heap_crashtypes;
81extern struct crashtype_category perms_crashtypes;
82extern struct crashtype_category refcount_crashtypes;
83extern struct crashtype_category usercopy_crashtypes;
84extern struct crashtype_category stackleak_crashtypes;
85extern struct crashtype_category cfi_crashtypes;
86extern struct crashtype_category fortify_crashtypes;
87extern struct crashtype_category powerpc_crashtypes;
88
89/* Each category's init/exit routines. */
90void __init lkdtm_bugs_init(int *recur_param);
91void __init lkdtm_heap_init(void);
92void __exit lkdtm_heap_exit(void);
93void __init lkdtm_perms_init(void);
94void __init lkdtm_usercopy_init(void);
95void __exit lkdtm_usercopy_exit(void);
96
97/* Special declaration for function-in-rodata. */
98void lkdtm_rodata_do_nothing(void);
99
100#endif
101