^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 1) // SPDX-License-Identifier: GPL-2.0
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 2) /*
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 3) * This is for all the tests relating directly to Control Flow Integrity.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 4) */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 5) #include "lkdtm.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 6)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 7) static int called_count;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 8)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 9) /* Function taking one argument, without a return value. */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 10) static noinline void lkdtm_increment_void(int *counter)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 11) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 12) (*counter)++;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 13) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 14)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 15) /* Function taking one argument, returning int. */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 16) static noinline int lkdtm_increment_int(int *counter)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 17) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 18) (*counter)++;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 19)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 20) return *counter;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 21) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 22) /*
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 23) * This tries to call an indirect function with a mismatched prototype.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 24) */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 25) void lkdtm_CFI_FORWARD_PROTO(void)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 26) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 27) /*
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 28) * Matches lkdtm_increment_void()'s prototype, but not
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 29) * lkdtm_increment_int()'s prototype.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 30) */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 31) void (*func)(int *);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 32)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 33) pr_info("Calling matched prototype ...\n");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 34) func = lkdtm_increment_void;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 35) func(&called_count);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 36)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 37) pr_info("Calling mismatched prototype ...\n");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 38) func = (void *)lkdtm_increment_int;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 39) func(&called_count);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 40)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 41) pr_info("Fail: survived mismatched prototype function call!\n");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 42) }