1 /* SPDX-License-Identifier: GPL-2.0-only */
3 * arch/arm/probes/kprobes/test-core.h
5 * Copyright (C) 2011 Jon Medhurst <tixy@yxit.co.uk>.
8 #define VERBOSE 0 /* Set to '1' for more logging of test cases */
10 #ifdef CONFIG_THUMB2_KERNEL
11 #define NORMAL_ISA "16"
13 #define NORMAL_ISA "32"
17 /* Flags used in kprobe_test_flags */
18 #define TEST_FLAG_NO_ITBLOCK (1<<0)
19 #define TEST_FLAG_FULL_ITBLOCK (1<<1)
20 #define TEST_FLAG_NARROW_INSTR (1<<2)
22 extern int kprobe_test_flags
;
23 extern int kprobe_test_cc_position
;
26 #define TEST_MEMORY_SIZE 256
30 * Test case structures.
32 * The arguments given to test cases can be one of three types.
35 * Load a register with the given value.
38 * Load a register with a pointer into the stack buffer (SP + given value).
41 * Store the given value into the stack buffer at [SP+index].
45 #define ARG_TYPE_END 0
46 #define ARG_TYPE_REG 1
47 #define ARG_TYPE_PTR 2
48 #define ARG_TYPE_MEM 3
49 #define ARG_TYPE_REG_MASKED 4
51 #define ARG_FLAG_UNSUPPORTED 0x01
52 #define ARG_FLAG_SUPPORTED 0x02
53 #define ARG_FLAG_THUMB 0x10 /* Must be 16 so TEST_ISA can be used */
54 #define ARG_FLAG_ARM 0x20 /* Must be 32 so TEST_ISA can be used */
57 u8 type
; /* ARG_TYPE_x */
61 struct test_arg_regptr
{
62 u8 type
; /* ARG_TYPE_REG or ARG_TYPE_PTR or ARG_TYPE_REG_MASKED */
69 u8 type
; /* ARG_TYPE_MEM */
76 u8 type
; /* ARG_TYPE_END */
77 u8 flags
; /* ARG_FLAG_x */
85 * Building blocks for test cases.
87 * Each test case is wrapped between TESTCASE_START and TESTCASE_END.
89 * To specify arguments for a test case the TEST_ARG_{REG,PTR,MEM} macros are
90 * used followed by a terminating TEST_ARG_END.
92 * After this, the instruction to be tested is defined with TEST_INSTRUCTION.
93 * Or for branches, TEST_BRANCH_B and TEST_BRANCH_F (branch forwards/backwards).
95 * Some specific test cases may make use of other custom constructs.
99 #define verbose(fmt, ...) pr_info(fmt, ##__VA_ARGS__)
101 #define verbose(fmt, ...)
104 #define TEST_GROUP(title) \
106 verbose(title"\n"); \
107 verbose("---------------------------------------------------------\n");
109 #define TESTCASE_START(title) \
110 __asm__ __volatile__ ( \
111 "bl __kprobes_test_case_start \n\t" \
112 ".pushsection .rodata \n\t" \
114 /* don't use .asciz here as 'title' may be */ \
115 /* multiple strings to be concatenated. */ \
116 ".ascii "#title" \n\t" \
121 #define TEST_ARG_REG(reg, val) \
122 ".byte "__stringify(ARG_TYPE_REG)" \n\t" \
123 ".byte "#reg" \n\t" \
127 #define TEST_ARG_PTR(reg, val) \
128 ".byte "__stringify(ARG_TYPE_PTR)" \n\t" \
129 ".byte "#reg" \n\t" \
133 #define TEST_ARG_MEM(index, val) \
134 ".byte "__stringify(ARG_TYPE_MEM)" \n\t" \
135 ".byte "#index" \n\t" \
139 #define TEST_ARG_REG_MASKED(reg, val) \
140 ".byte "__stringify(ARG_TYPE_REG_MASKED)" \n\t" \
141 ".byte "#reg" \n\t" \
145 #define TEST_ARG_END(flags) \
146 ".byte "__stringify(ARG_TYPE_END)" \n\t" \
147 ".byte "TEST_ISA flags" \n\t" \
148 ".short 50f-0f \n\t" \
149 ".short 2f-0f \n\t" \
150 ".short 99f-0f \n\t" \
151 ".code "TEST_ISA" \n\t" \
154 #define TEST_INSTRUCTION(instruction) \
156 "1: "instruction" \n\t" \
159 #define TEST_BRANCH_F(instruction) \
160 TEST_INSTRUCTION(instruction) \
164 #define TEST_BRANCH_B(instruction) \
169 TEST_INSTRUCTION(instruction)
171 #define TEST_BRANCH_FX(instruction, codex) \
172 TEST_INSTRUCTION(instruction) \
178 #define TEST_BRANCH_BX(instruction, codex) \
184 TEST_INSTRUCTION(instruction)
186 #define TESTCASE_END \
189 " bl __kprobes_test_case_end_"TEST_ISA" \n\t" \
190 ".code "NORMAL_ISA" \n\t" \
192 : "r0", "r1", "r2", "r3", "ip", "lr", "memory", "cc" \
197 * Macros to define test cases.
199 * Those of the form TEST_{R,P,M}* can be used to define test cases
200 * which take combinations of the three basic types of arguments. E.g.
202 * TEST_R One register argument
203 * TEST_RR Two register arguments
204 * TEST_RPR A register, a pointer, then a register argument
206 * For testing instructions which may branch, there are macros TEST_BF_*
207 * and TEST_BB_* for branching forwards and backwards.
209 * TEST_SUPPORTED and TEST_UNSUPPORTED don't cause the code to be executed,
210 * the just verify that a kprobe is or is not allowed on the given instruction.
214 TESTCASE_START(code) \
216 TEST_INSTRUCTION(code) \
219 #define TEST_UNSUPPORTED(code) \
220 TESTCASE_START(code) \
221 TEST_ARG_END("|"__stringify(ARG_FLAG_UNSUPPORTED)) \
222 TEST_INSTRUCTION(code) \
225 #define TEST_SUPPORTED(code) \
226 TESTCASE_START(code) \
227 TEST_ARG_END("|"__stringify(ARG_FLAG_SUPPORTED)) \
228 TEST_INSTRUCTION(code) \
231 #define TEST_R(code1, reg, val, code2) \
232 TESTCASE_START(code1 #reg code2) \
233 TEST_ARG_REG(reg, val) \
235 TEST_INSTRUCTION(code1 #reg code2) \
238 #define TEST_RR(code1, reg1, val1, code2, reg2, val2, code3) \
239 TESTCASE_START(code1 #reg1 code2 #reg2 code3) \
240 TEST_ARG_REG(reg1, val1) \
241 TEST_ARG_REG(reg2, val2) \
243 TEST_INSTRUCTION(code1 #reg1 code2 #reg2 code3) \
246 #define TEST_RRR(code1, reg1, val1, code2, reg2, val2, code3, reg3, val3, code4)\
247 TESTCASE_START(code1 #reg1 code2 #reg2 code3 #reg3 code4) \
248 TEST_ARG_REG(reg1, val1) \
249 TEST_ARG_REG(reg2, val2) \
250 TEST_ARG_REG(reg3, val3) \
252 TEST_INSTRUCTION(code1 #reg1 code2 #reg2 code3 #reg3 code4) \
255 #define TEST_RRRR(code1, reg1, val1, code2, reg2, val2, code3, reg3, val3, code4, reg4, val4) \
256 TESTCASE_START(code1 #reg1 code2 #reg2 code3 #reg3 code4 #reg4) \
257 TEST_ARG_REG(reg1, val1) \
258 TEST_ARG_REG(reg2, val2) \
259 TEST_ARG_REG(reg3, val3) \
260 TEST_ARG_REG(reg4, val4) \
262 TEST_INSTRUCTION(code1 #reg1 code2 #reg2 code3 #reg3 code4 #reg4) \
265 #define TEST_P(code1, reg1, val1, code2) \
266 TESTCASE_START(code1 #reg1 code2) \
267 TEST_ARG_PTR(reg1, val1) \
269 TEST_INSTRUCTION(code1 #reg1 code2) \
272 #define TEST_PR(code1, reg1, val1, code2, reg2, val2, code3) \
273 TESTCASE_START(code1 #reg1 code2 #reg2 code3) \
274 TEST_ARG_PTR(reg1, val1) \
275 TEST_ARG_REG(reg2, val2) \
277 TEST_INSTRUCTION(code1 #reg1 code2 #reg2 code3) \
280 #define TEST_RP(code1, reg1, val1, code2, reg2, val2, code3) \
281 TESTCASE_START(code1 #reg1 code2 #reg2 code3) \
282 TEST_ARG_REG(reg1, val1) \
283 TEST_ARG_PTR(reg2, val2) \
285 TEST_INSTRUCTION(code1 #reg1 code2 #reg2 code3) \
288 #define TEST_PRR(code1, reg1, val1, code2, reg2, val2, code3, reg3, val3, code4)\
289 TESTCASE_START(code1 #reg1 code2 #reg2 code3 #reg3 code4) \
290 TEST_ARG_PTR(reg1, val1) \
291 TEST_ARG_REG(reg2, val2) \
292 TEST_ARG_REG(reg3, val3) \
294 TEST_INSTRUCTION(code1 #reg1 code2 #reg2 code3 #reg3 code4) \
297 #define TEST_RPR(code1, reg1, val1, code2, reg2, val2, code3, reg3, val3, code4)\
298 TESTCASE_START(code1 #reg1 code2 #reg2 code3 #reg3 code4) \
299 TEST_ARG_REG(reg1, val1) \
300 TEST_ARG_PTR(reg2, val2) \
301 TEST_ARG_REG(reg3, val3) \
303 TEST_INSTRUCTION(code1 #reg1 code2 #reg2 code3 #reg3 code4) \
306 #define TEST_RRP(code1, reg1, val1, code2, reg2, val2, code3, reg3, val3, code4)\
307 TESTCASE_START(code1 #reg1 code2 #reg2 code3 #reg3 code4) \
308 TEST_ARG_REG(reg1, val1) \
309 TEST_ARG_REG(reg2, val2) \
310 TEST_ARG_PTR(reg3, val3) \
312 TEST_INSTRUCTION(code1 #reg1 code2 #reg2 code3 #reg3 code4) \
315 #define TEST_BF_P(code1, reg1, val1, code2) \
316 TESTCASE_START(code1 #reg1 code2) \
317 TEST_ARG_PTR(reg1, val1) \
319 TEST_BRANCH_F(code1 #reg1 code2) \
322 #define TEST_BF(code) \
323 TESTCASE_START(code) \
325 TEST_BRANCH_F(code) \
328 #define TEST_BB(code) \
329 TESTCASE_START(code) \
331 TEST_BRANCH_B(code) \
334 #define TEST_BF_R(code1, reg, val, code2) \
335 TESTCASE_START(code1 #reg code2) \
336 TEST_ARG_REG(reg, val) \
338 TEST_BRANCH_F(code1 #reg code2) \
341 #define TEST_BB_R(code1, reg, val, code2) \
342 TESTCASE_START(code1 #reg code2) \
343 TEST_ARG_REG(reg, val) \
345 TEST_BRANCH_B(code1 #reg code2) \
348 #define TEST_BF_RR(code1, reg1, val1, code2, reg2, val2, code3) \
349 TESTCASE_START(code1 #reg1 code2 #reg2 code3) \
350 TEST_ARG_REG(reg1, val1) \
351 TEST_ARG_REG(reg2, val2) \
353 TEST_BRANCH_F(code1 #reg1 code2 #reg2 code3) \
356 #define TEST_BF_X(code, codex) \
357 TESTCASE_START(code) \
359 TEST_BRANCH_FX(code, codex) \
362 #define TEST_BB_X(code, codex) \
363 TESTCASE_START(code) \
365 TEST_BRANCH_BX(code, codex) \
368 #define TEST_BF_RX(code1, reg, val, code2, codex) \
369 TESTCASE_START(code1 #reg code2) \
370 TEST_ARG_REG(reg, val) \
372 TEST_BRANCH_FX(code1 #reg code2, codex) \
375 #define TEST_X(code, codex) \
376 TESTCASE_START(code) \
378 TEST_INSTRUCTION(code) \
383 #define TEST_RX(code1, reg, val, code2, codex) \
384 TESTCASE_START(code1 #reg code2) \
385 TEST_ARG_REG(reg, val) \
387 TEST_INSTRUCTION(code1 __stringify(reg) code2) \
392 #define TEST_RRX(code1, reg1, val1, code2, reg2, val2, code3, codex) \
393 TESTCASE_START(code1 #reg1 code2 #reg2 code3) \
394 TEST_ARG_REG(reg1, val1) \
395 TEST_ARG_REG(reg2, val2) \
397 TEST_INSTRUCTION(code1 __stringify(reg1) code2 __stringify(reg2) code3) \
402 #define TEST_RMASKED(code1, reg, mask, code2) \
403 TESTCASE_START(code1 #reg code2) \
404 TEST_ARG_REG_MASKED(reg, mask) \
406 TEST_INSTRUCTION(code1 #reg code2) \
410 * We ignore the state of the imprecise abort disable flag (CPSR.A) because this
411 * can change randomly as the kernel doesn't take care to preserve or initialise
412 * this across context switches. Also, with Security Extensions, the flag may
413 * not be under control of the kernel; for this reason we ignore the state of
414 * the FIQ disable flag CPSR.F as well.
416 #define PSR_IGNORE_BITS (PSR_A_BIT | PSR_F_BIT)
420 * Macros for defining space directives spread over multiple lines.
421 * These are required so the compiler guesses better the length of inline asm
422 * code and will spill the literal pool early enough to avoid generating PC
423 * relative loads with out of range offsets.
426 #define SPACE_0x8 TWICE(".space 4\n\t")
427 #define SPACE_0x10 TWICE(SPACE_0x8)
428 #define SPACE_0x20 TWICE(SPACE_0x10)
429 #define SPACE_0x40 TWICE(SPACE_0x20)
430 #define SPACE_0x80 TWICE(SPACE_0x40)
431 #define SPACE_0x100 TWICE(SPACE_0x80)
432 #define SPACE_0x200 TWICE(SPACE_0x100)
433 #define SPACE_0x400 TWICE(SPACE_0x200)
434 #define SPACE_0x800 TWICE(SPACE_0x400)
435 #define SPACE_0x1000 TWICE(SPACE_0x800)
438 /* Various values used in test cases... */
439 #define N(val) (val ^ 0xffffffff)
440 #define VAL1 0x12345678
442 #define VAL3 0xa5f801
444 #define VALM 0x456789ab
445 #define VALR 0xdeaddead
446 #define HH1 0x0123fecb
447 #define HH2 0xa9874567
450 #ifdef CONFIG_THUMB2_KERNEL
451 void kprobe_thumb16_test_cases(void);
452 void kprobe_thumb32_test_cases(void);
454 void kprobe_arm_test_cases(void);