108 lines
2.9 KiB
Diff
108 lines
2.9 KiB
Diff
diff --git a/src/toa.c b/src/toa.c
|
|
index 2942674..82382f9 100644
|
|
--- a/src/toa.c
|
|
+++ b/src/toa.c
|
|
@@ -1,3 +1,4 @@
|
|
+#include <linux/kprobes.h>
|
|
#include "toa.h"
|
|
|
|
/*
|
|
@@ -7,6 +8,9 @@
|
|
|
|
unsigned long sk_data_ready_addr = 0;
|
|
|
|
+typedef unsigned long (*kallsyms_lookup_name_fun_t)(const char *name);
|
|
+static kallsyms_lookup_name_fun_t kallsyms_lookup_name_fun;
|
|
+
|
|
/*
|
|
* Statistics of toa in proc /proc/net/toa_stats
|
|
*/
|
|
@@ -23,6 +27,23 @@ struct toa_stats_entry toa_stats[] = {
|
|
|
|
struct toa_stat_mib *ext_stats;
|
|
|
|
+static int toa_get_kallsyms_lookup_name_addr(void)
|
|
+{
|
|
+ static struct kprobe kp = {
|
|
+ .symbol_name = "kallsyms_lookup_name",
|
|
+ };
|
|
+
|
|
+ if (register_kprobe(&kp) < 0)
|
|
+ return -1;
|
|
+
|
|
+ if (!kallsyms_lookup_name_fun) {
|
|
+ kallsyms_lookup_name_fun = (kallsyms_lookup_name_fun_t)kp.addr;
|
|
+ }
|
|
+ unregister_kprobe(&kp);
|
|
+
|
|
+ return 0;
|
|
+}
|
|
+
|
|
#if defined(CONFIG_ARM64)
|
|
/*
|
|
* ARM64 interface
|
|
@@ -41,7 +62,7 @@ static pte_t *lookup_address(unsigned long address, unsigned int *level)
|
|
pte_t *ptep;
|
|
unsigned long addr = address;
|
|
|
|
- unsigned long init_mm = kallsyms_lookup_name("init_mm");
|
|
+ unsigned long init_mm = kallsyms_lookup_name_fun("init_mm");
|
|
if (!init_mm) {
|
|
TOA_INFO("CPU [%u] lookup_address init_mm null\n", smp_processor_id());
|
|
return NULL;
|
|
@@ -52,7 +73,7 @@ static pte_t *lookup_address(unsigned long address, unsigned int *level)
|
|
if (pgd_none(READ_ONCE(*pgdp)))
|
|
return NULL;
|
|
|
|
- pudp = pud_offset(pgdp, addr);
|
|
+ pudp = pud_offset((p4d_t *)pgdp, addr);
|
|
pud = READ_ONCE(*pudp);
|
|
if (pud_none(pud))
|
|
return NULL;
|
|
@@ -522,6 +543,14 @@ static int toa_stats_seq_open(struct inode *inode, struct file *file)
|
|
return single_open(file, toa_stats_show, NULL);
|
|
}
|
|
|
|
+#if LINUX_VERSION_CODE >= KERNEL_VERSION(5,5,0)
|
|
+static const struct proc_ops toa_stats_fops = {
|
|
+ .proc_open = toa_stats_seq_open,
|
|
+ .proc_read = seq_read,
|
|
+ .proc_lseek = seq_lseek,
|
|
+ .proc_release = single_release,
|
|
+};
|
|
+#else
|
|
static const struct file_operations toa_stats_fops = {
|
|
.owner = THIS_MODULE,
|
|
.open = toa_stats_seq_open,
|
|
@@ -529,6 +558,7 @@ static const struct file_operations toa_stats_fops = {
|
|
.llseek = seq_lseek,
|
|
.release = single_release,
|
|
};
|
|
+#endif
|
|
|
|
/*
|
|
* TOA module init and destory
|
|
@@ -538,6 +568,13 @@ static const struct file_operations toa_stats_fops = {
|
|
static int __init
|
|
toa_init(void)
|
|
{
|
|
+ int ret = -1;
|
|
+
|
|
+ ret = toa_get_kallsyms_lookup_name_addr();
|
|
+ if (ret < 0) {
|
|
+ return 1;
|
|
+ }
|
|
+
|
|
/* alloc statistics array for toa */
|
|
ext_stats = alloc_percpu(struct toa_stat_mib);
|
|
if (NULL == ext_stats)
|
|
@@ -547,7 +584,7 @@ toa_init(void)
|
|
/* get the address of function sock_def_readable
|
|
* so later we can know whether the sock is for rpc, tux or others
|
|
*/
|
|
- sk_data_ready_addr = kallsyms_lookup_name("sock_def_readable");
|
|
+ sk_data_ready_addr = kallsyms_lookup_name_fun("sock_def_readable");
|
|
TOA_INFO("CPU [%u] sk_data_ready_addr = "
|
|
"kallsyms_lookup_name(sock_def_readable) = %lu\n",
|
|
smp_processor_id(), sk_data_ready_addr);
|