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

diag_task_info_work: try to fix pust buffer bug

Browse Source
This commit is contained in:
zy
2023-12-06 03:37:24 -05:00
parent 73ac7f86ad
commit fcb09347ea
1 changed files with 16 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
source/module/monitor_kernel_lib.c
View File

@@ -132,6 +132,15 @@ static void push_tskinfo_2_buffer(variable_monitor_task *tsk_info,
diag_variant_buffer_spin_unlock(&load_monitor_variant_buffer, *flags);
}
static void push_tskinfo_22_buffer(variable_monitor_task *tsk_info,
unsigned long *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);
}
/// @brief clear all watch and reset kernel_wtimer_list/kernel_wtimer_num
/// @param
static void clear_all_watch(void) {
@@ -193,16 +202,15 @@ void diag_task_info_work(struct work_struct *work) {
}
rcu_read_lock();
diag_variant_buffer_spin_lock(&load_monitor_variant_buffer, flags);
diag_variant_buffer_reserve(&load_monitor_variant_buffer,
sizeof(variable_monitor_record));
diag_variant_buffer_write_nolock(&load_monitor_variant_buffer, &vm_record,
sizeof(variable_monitor_record));
diag_variant_buffer_seal(&load_monitor_variant_buffer);
diag_variant_buffer_spin_unlock(&load_monitor_variant_buffer, flags);
// diag_variant_buffer_spin_unlock(&load_monitor_variant_buffer, flags);
rcu_read_unlock();
// for task info
do_each_thread(g, p) {
if (p->__state == TASK_RUNNING || __task_contributes_to_load(p) ||
@@ -217,10 +225,14 @@ void diag_task_info_work(struct work_struct *work) {
put_task_struct(p); // count -1
push_tskinfo_2_buffer(&tsk_info, &flags); // push to buffer
push_tskinfo_22_buffer(&tsk_info, &flags); // push to buffer
}
}
while_each_thread(g, p);
diag_variant_buffer_spin_unlock(&load_monitor_variant_buffer, flags);
rcu_read_unlock();
printk(KERN_INFO "-------------------------------------\n");
return;
}