mirror of
https://github.com/cirosantilli/linux-kernel-module-cheat.git
synced 2026-01-23 10:15:57 +01:00
c8c4f89854
Previously had wonky line pointer in asm_main. New interface simpler and more portable. Add tests for ASSERT_EQ_ and family in arm and aarch64, previously on x86_64. ASSERT_EQ_ and family in ARM can now either take =123, =addr or var, before this the = was added on macros so var was not possible. Define the main function directly in assembly, the C driver was useless.
63 lines
1.5 KiB
C
63 lines
1.5 KiB
C
/* https://github.com/cirosantilli/linux-kernel-module-cheat#lkmc-c */
|
|
|
|
#ifndef LKMC_H
|
|
#define LKMC_H
|
|
|
|
#if !defined(__ASSEMBLER__)
|
|
#include <inttypes.h>
|
|
#include <stdbool.h>
|
|
#include <stdint.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
#define LKMC_ASSERT_EQ_DECLARE(bits) \
|
|
void lkmc_assert_eq_ ## bits( \
|
|
uint ## bits ## _t val1, \
|
|
uint ## bits ## _t val2, \
|
|
uint32_t line \
|
|
)
|
|
LKMC_ASSERT_EQ_DECLARE(32);
|
|
LKMC_ASSERT_EQ_DECLARE(64);
|
|
void lkmc_assert_fail(uint32_t line);
|
|
void lkmc_assert_memcmp(const void *s1, const void *s2, size_t n, uint32_t line);
|
|
void lkmc_baremetal_on_exit_callback(int status, void *arg);
|
|
#endif
|
|
|
|
/* Assert that the given branch instruction is taken. */
|
|
#define LKMC_ASSERT(branch_if_pass) \
|
|
branch_if_pass 1f; \
|
|
LKMC_ASSERT_FAIL; \
|
|
1: \
|
|
;
|
|
|
|
/* https://stackoverflow.com/questions/1489932/how-to-concatenate-twice-with-the-c-preprocessor-and-expand-a-macro-as-in-arg */
|
|
#define LKMC_CONCAT_EVAL(a,b) a ## b
|
|
#define LKMC_CONCAT(a,b) LKMC_CONCAT_EVAL(a, b)
|
|
|
|
#define LKMC_GLOBAL(name) \
|
|
.global name; \
|
|
name:
|
|
|
|
/* Common C definitions. */
|
|
#define LKMC_UNUSED(x) (void)x
|
|
|
|
/* Weak means that if any other file defines it as a non-weak global,
|
|
* that one will take precedence:
|
|
* https://stackoverflow.com/questions/274753/how-to-make-weak-linking-work-with-gcc/54601464#54601464
|
|
*/
|
|
#define LKMC_WEAK(name) \
|
|
.weak name; \
|
|
name:
|
|
|
|
#if defined(__x86_64__)
|
|
#include <lkmc/x86_64.h>
|
|
#elif defined(__arm__)
|
|
#include <lkmc/arm.h>
|
|
#elif defined(__aarch64__)
|
|
#include <lkmc/aarch64.h>
|
|
#else
|
|
#error
|
|
#endif
|
|
|
|
#endif
|