gfwleak/zhangyang-variable-monitor
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

do_dump_sa

Browse Source 
- module
- ucli
This commit is contained in:
zy
2023-12-12 04:05:56 -05:00
parent 96462422ce
commit 69bf4c8c69
7 changed files with 155 additions and 75 deletions
Download Patch File Download Diff File Expand all files Collapse all files

128
source/module/monitor_kernel_lib.c
View File

@@ -83,19 +83,31 @@ static void init_mm_tree(mm_tree *mm_tree) {
}
/**
* @brief init global variable load_monitor_variant_buffer
* @brief init buffer
*
* @param buf_size
* @param buffer
* @return int
*/
static int init_buffer(unsigned int buf_size) {
static int init_buffer(unsigned int buf_size,
struct diag_variant_buffer *buffer) {
init_mm_tree(&mm_tree_struct); // init mm_tree
init_diag_variant_buffer(&load_monitor_variant_buffer, buf_size);
init_diag_variant_buffer(buffer, buf_size);
int ret = 0;
ret = alloc_diag_variant_buffer(&load_monitor_variant_buffer);
ret = alloc_diag_variant_buffer(buffer);
return ret;
}
// init load_monitor_variant_buffer
static int init_global_buffer(void) {
return init_buffer(VARIABLE_MONITOR_BUFFER_SIZE,
&load_monitor_variant_buffer);
}
// init stand_alone_buffer
static int init_sa_buffer(void) {
return init_buffer(STAND_ALONE_BUFFER_SIZE, &stand_alone_buffer);
}
/**
* @brief diag task info | brief | user stack | kernel stack | proc chains | raw
* stack
@@ -114,37 +126,64 @@ static void diag_tsk(struct task_struct *p, variable_monitor_task *tsk_info) {
diag_task_raw_stack(p, &tsk_info->raw_stack); // raw stack
}
static void push_tskinfo_2_buffer_orig(variable_monitor_task *tsk_info,
unsigned long *flags,
struct diag_variant_buffer *buffer) {
// printk(KERN_INFO "push_tsk_info\n");
diag_variant_buffer_spin_lock(buffer, *flags);
diag_variant_buffer_reserve(buffer, sizeof(variable_monitor_task));
diag_variant_buffer_write_nolock(buffer, tsk_info,
sizeof(variable_monitor_task));
diag_variant_buffer_seal(buffer);
diag_variant_buffer_spin_unlock(buffer, *flags);
}
/**
* @brief push task info to global buffer
*
* @param tsk_info
* @param flags
*/
static void push_tskinfo_2_buffer(variable_monitor_task *tsk_info,
unsigned long *flags) {
// printk(KERN_INFO "push_tsk_info\n");
diag_variant_buffer_spin_lock(&load_monitor_variant_buffer, *flags);
diag_variant_buffer_reserve(&load_monitor_variant_buffer,
sizeof(variable_monitor_task));
diag_variant_buffer_write_nolock(&load_monitor_variant_buffer, tsk_info,
sizeof(variable_monitor_task));
diag_variant_buffer_seal(&load_monitor_variant_buffer);
diag_variant_buffer_spin_unlock(&load_monitor_variant_buffer, *flags);
// static void push_tskinfo_2_buffer(variable_monitor_task *tsk_info,
// unsigned long *flags) {
// push_tskinfo_2_buffer_orig(tsk_info, flags, &load_monitor_variant_buffer);
// }
/**
* @brief push task info to stand_alone_buffer
*
* @param tsk_info
* @param flags
*/
static void push_tskinfo_2_sa_buffer(variable_monitor_task *tsk_info,
unsigned long *flags) {
push_tskinfo_2_buffer_orig(tsk_info, flags, &stand_alone_buffer);
}
static void push_tskinfo_22_buffer(variable_monitor_task *tsk_info, unsigned long *flags){
/**
* @brief push user/sys task info to global buffer
*
* @param tsk_info
* @param flags
*/
static void push_tskinfo_22_buffer(variable_monitor_task *tsk_info,
unsigned long *flags) {
variable_monitor_task_system *tsk_info_system;
if (tsk_info->task.sys_task == 1) // system task
{
tsk_info_system = (variable_monitor_task_system *)tsk_info;
tsk_info_system->et_type = VARIABLE_MONITOR_TASK_TYPE_SYSTEM;
diag_variant_buffer_reserve(&load_monitor_variant_buffer, sizeof(variable_monitor_task_system));
diag_variant_buffer_write_nolock(&load_monitor_variant_buffer, tsk_info_system,
sizeof(variable_monitor_task_system));
diag_variant_buffer_reserve(&load_monitor_variant_buffer,
sizeof(variable_monitor_task_system));
diag_variant_buffer_write_nolock(&load_monitor_variant_buffer,
tsk_info_system,
sizeof(variable_monitor_task_system));
diag_variant_buffer_seal(&load_monitor_variant_buffer);
} else {
diag_variant_buffer_reserve(&load_monitor_variant_buffer, sizeof(variable_monitor_task));
diag_variant_buffer_write_nolock(&load_monitor_variant_buffer, tsk_info, sizeof(variable_monitor_task));
diag_variant_buffer_reserve(&load_monitor_variant_buffer,
sizeof(variable_monitor_task));
diag_variant_buffer_write_nolock(&load_monitor_variant_buffer, tsk_info,
sizeof(variable_monitor_task));
diag_variant_buffer_seal(&load_monitor_variant_buffer);
}
}
@@ -202,7 +241,8 @@ void diag_task_info_work(struct work_struct *work) {
// printk(KERN_INFO "threshold exceeded, Timestamp %lld:\n", vm_record.tv);
// for (i = 0; i < vm_record.threshold_num; i++) {
// printk(KERN_INFO "\t: pid: %d, name: %s, ptr: %p, threshold:%lld, true_value:%lld\n",
// printk(KERN_INFO "\t: pid: %d, name: %s, ptr: %p, threshold:%lld,
// true_value:%lld\n",
// vm_record.threshold_record[i].task_id,
// vm_record.threshold_record[i]
// .name, // Assuming name is a null-terminated string
@@ -247,7 +287,8 @@ void diag_task_info_work(struct work_struct *work) {
printk(KERN_INFO "threshold exceeded, Timestamp %lld:\n", vm_record.tv);
for (i = 0; i < vm_record.threshold_num; i++) {
printk(KERN_INFO "\t: pid: %d, name: %s, ptr: %p, threshold:%lld, true_value:%lld\n",
printk(KERN_INFO
"\t: pid: %d, name: %s, ptr: %p, threshold:%lld, true_value:%lld\n",
vm_record.threshold_record[i].task_id,
vm_record.threshold_record[i]
.name, // Assuming name is a null-terminated string
@@ -260,7 +301,7 @@ void diag_task_info_work(struct work_struct *work) {
return;
}
/**
* @brief all module function init. orig_X | buffer
* @brief all module function init. orig_X | buffer
*
* @return int
*/
@@ -270,7 +311,11 @@ int monitor_init(void) {
ret = init_orig_fun(); // init orig_X
if (ret)
return ret;
ret = init_buffer(VARIABLE_MONITOR_BUFFER_SIZE); // 256M
ret = init_global_buffer(); // 256M
if (ret)
return -1;
ret = init_sa_buffer(); // 50M
if (ret)
return -1;
return 0;
@@ -323,7 +368,9 @@ int start_watch_variable(watch_arg warg) {
INIT_WORK(&timer->wk, diag_task_info_work);
printk(KERN_INFO "Convert ptr to kptr: %p\n", kptr);
printk(KERN_INFO "Associated timer: %p , there are already %d variables, timer period %lld.\n", timer, timer->sentinel, timer->time_ns);
printk(KERN_INFO "Associated timer: %p , there are already %d variables, "
"timer period %lld.\n",
timer, timer->sentinel, timer->time_ns);
// printk(KERN_INFO "timer->hr_timer: %p\n", &timer->hr_timer);
TIMER_CANCEL(timer); // just in case
@@ -336,7 +383,7 @@ int start_watch_variable(watch_arg warg) {
/**
* @brief reinit all timer's work
*
*
*/
void init_work_all_hrTimer(void) {
int i = 0;
@@ -395,7 +442,7 @@ enum hrtimer_restart check_variable_cb(struct hrtimer *timer) {
queue_work(system_highpri_wq, &k_watch_timer->wk);
// restart timer after 5s
hrtimer_forward(timer, timer->base->get_time(),
ktime_set(dump_reset_sec, 0)); //! todo
ktime_set(dump_reset_sec, 0));
} else {
// keep frequency
hrtimer_forward(timer, timer->base->get_time(), k_watch_timer->kt);
@@ -405,9 +452,9 @@ enum hrtimer_restart check_variable_cb(struct hrtimer *timer) {
/**
* @brief for test only
*
* @param id
* @return int
*
* @param id
* @return int
*/
int diag_pid(int id) {
struct task_struct *tsk;
@@ -439,16 +486,16 @@ int diag_pid(int id) {
diag_tsk(tsk, &tsk_info);
put_task_struct(tsk); // count -1
push_tskinfo_2_buffer(&tsk_info, &flags); // push to buffer
push_tskinfo_2_sa_buffer(&tsk_info, &flags); // push to buffer
return 0;
}
/**
* @brief for test only
*
* @param id
* @return int
*
* @param id
* @return int
*/
int diag_tgid(int id) {
struct task_struct *tsk;
@@ -475,15 +522,6 @@ int diag_tgid(int id) {
struct task_struct *thread = tsk;
while_each_thread(tsk, thread) {
// save_task_info(thread, &detail);
// diag_variant_buffer_spin_lock(&pupil_variant_buffer, flags);
// diag_variant_buffer_reserve(&pupil_variant_buffer,
// sizeof(struct pupil_task_detail));
// diag_variant_buffer_write_nolock(&pupil_variant_buffer, &detail,
// sizeof(struct pupil_task_detail));
// diag_variant_buffer_seal(&pupil_variant_buffer);
// diag_variant_buffer_spin_unlock(&pupil_variant_buffer, flags);
get_task_struct(thread); // count +1
@@ -491,10 +529,10 @@ int diag_tgid(int id) {
tsk_info.id = event_id;
tsk_info.tv = vm_record.tv;
diag_tsk(thread, &tsk_info);
put_task_struct(thread); // count -1
push_tskinfo_2_buffer(&tsk_info, &flags); // push to buffer
push_tskinfo_2_sa_buffer(&tsk_info, &flags); // push to buffer
}
return 0;