96 lines
2.3 KiB
C
96 lines
2.3 KiB
C
|
#ifndef _OASES_UTIL_H
|
||
|
|
#define _OASES_UTIL_H
|
||
|
|
|
||
|
|
#include <linux/kernel.h>
|
||
|
|
#include <linux/types.h>
|
||
|
|
#include <linux/printk.h>
|
||
|
|
|
||
|
|
/* general printk */
|
||
|
|
#define oases_debug(fmt, ...) \
|
||
|
|
do { \
|
||
|
|
pr_debug("[oases][%s:%d] " fmt, \
|
||
|
|
__func__, __LINE__, ## __VA_ARGS__); \
|
||
|
|
} while(0)
|
||
|
|
|
||
|
|
#define oases_info(fmt, ...) \
|
||
|
|
do { \
|
||
|
|
pr_info("[oases][%s:%d] " fmt, \
|
||
|
|
__func__, __LINE__, ## __VA_ARGS__); \
|
||
|
|
} while(0)
|
||
|
|
|
||
|
|
#define oases_error(fmt, ...) \
|
||
|
|
do { \
|
||
|
|
pr_err("[oases][%s:%d] " fmt, \
|
||
|
|
__func__, __LINE__, ## __VA_ARGS__); \
|
||
|
|
} while(0)
|
||
|
|
|
||
|
|
struct oases_patch_info;
|
||
|
|
|
||
|
|
int oases_is_null(const void *, int);
|
||
|
|
int oases_valid_name(const char *id, int maxlen);
|
||
|
|
|
||
|
|
void oases_module_lock(void);
|
||
|
|
void oases_module_unlock(void);
|
||
|
|
void *oases_ref_module(const char *name);
|
||
|
|
int oases_ref_module_ptr(void *module);
|
||
|
|
void oases_unref_module(void *module);
|
||
|
|
|
||
|
|
int oases_insn_patch(const char *id);
|
||
|
|
int oases_insn_unpatch(const char *id);
|
||
|
|
int oases_remove_patch(struct oases_patch_info *info);
|
||
|
|
|
||
|
|
#ifdef CONFIG_HISI_HHEE
|
||
|
|
|
||
|
|
#include <linux/arm-smccc.h>
|
||
|
|
#include <linux/of.h>
|
||
|
|
#include <asm/insn.h>
|
||
|
|
#include <asm/cacheflush.h>
|
||
|
|
#include <linux/hisi/hisi_hhee.h>
|
||
|
|
|
||
|
|
static inline bool is_hkip_enabled(void) {
|
||
|
|
bool ret = false;
|
||
|
|
if (HHEE_ENABLE == hhee_check_enable())
|
||
|
|
ret = true;
|
||
|
|
return ret;
|
||
|
|
}
|
||
|
|
|
||
|
|
static inline void aarch64_insn_patch_text_hkip(void *place, u32 insn)
|
||
|
|
{
|
||
|
|
extern unsigned long oases_hkip_token;
|
||
|
|
struct arm_smccc_res res;
|
||
|
|
|
||
|
|
arm_smccc_hvc(HHEE_HVC_LIVEPATCH, (unsigned long)place, AARCH64_INSN_SIZE, insn, 0, oases_hkip_token, 0, 0, &res);
|
||
|
|
flush_icache_range((uintptr_t)place, (uintptr_t)place + AARCH64_INSN_SIZE);
|
||
|
|
}
|
||
|
|
|
||
|
|
static inline void aarch64_insn_disable_pxn_hkip(void *place, u32 text_size)
|
||
|
|
{
|
||
|
|
extern unsigned long oases_hkip_token;
|
||
|
|
struct arm_smccc_res res;
|
||
|
|
|
||
|
|
arm_smccc_hvc(HHEE_LKM_UPDATE, (unsigned long)place, text_size, oases_hkip_token, 0, 0, 0, 0, &res);
|
||
|
|
oases_debug("%s, res=%lu, %lu, %lu, %lu\n",
|
||
|
|
__FUNCTION__, res.a0, res.a1, res.a2, res.a3);
|
||
|
|
|
||
|
|
flush_icache_range((uintptr_t)place, (uintptr_t)place + text_size);
|
||
|
|
}
|
||
|
|
|
||
|
|
static inline int get_oases_hkip_token(unsigned long *token)
|
||
|
|
{
|
||
|
|
struct arm_smccc_res res;
|
||
|
|
|
||
|
|
arm_smccc_hvc(HHEE_HVC_TOKEN, 0, 0, 0, 0, 0, 0, 0, &res);
|
||
|
|
oases_debug("%s, res.a0=%lu,\n", __FUNCTION__, res.a0);
|
||
|
|
|
||
|
|
if (res.a0 != 0) {
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
*token = res.a1;
|
||
|
|
|
||
|
|
return 0 ;
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#endif /* OASES_UTIL_H */
|