Commit 8e77a794 authored by Zheng Yejian's avatar Zheng Yejian
Browse files

livepatch/arm64: Implement arch_klp_check_task_calltrace() 

hulk inclusion
category: feature
bugzilla: https://gitee.com/openeuler/kernel/issues/I9R2TB



--------------------------------

Signed-off-by: default avatarZheng Yejian <zhengyejian1@huawei.com>
parent bcf8c806

arch/arm64/kernel/livepatch.c

+53 −24
Original line number Diff line number Diff line
@@ -205,13 +205,12 @@ static bool klp_check_jump_func(void *ws_args, unsigned long pc) 
	return args->check_func(args->data, &args->ret, pc);

}



static int do_check_calltrace(struct walk_stackframe_args *args,

static int check_task_calltrace(struct task_struct *t,

				struct walk_stackframe_args *args,

				bool (*fn)(void *, unsigned long))

{

	struct task_struct *g, *t;

	struct stackframe frame;



	for_each_process_thread(g, t) {

	/*

	 * Handle the current carefully on each CPUs, we shouldn't

	 * use saved FP and PC when backtrace current. It's difficult
@@ -222,9 +221,7 @@ static int do_check_calltrace(struct walk_stackframe_args *args, 
	if (t == current) {

		/* current on this CPU */

		frame.fp = (unsigned long)__builtin_frame_address(0);

			frame.pc = (unsigned long)do_check_calltrace;

		} else if (klp_is_migration_thread(t->comm)) {

			continue;

		frame.pc = (unsigned long)check_task_calltrace;

	} else {

		frame.fp = thread_saved_fp(t);

		frame.pc = thread_saved_pc(t);
@@ -236,10 +233,42 @@ static int do_check_calltrace(struct walk_stackframe_args *args, 
		show_stack(t, NULL, KERN_INFO);

		return args->ret;

	}

	return 0;

}



static int do_check_calltrace(struct walk_stackframe_args *args,

			      bool (*fn)(void *, unsigned long))

{

	int ret;

	struct task_struct *g, *t;



	for_each_process_thread(g, t) {

		if (klp_is_migration_thread(t->comm))

			continue;

		ret = check_task_calltrace(t, args, fn);

		if (ret)

			return ret;

	}

	return 0;

}



#ifdef CONFIG_LIVEPATCH_BREAKPOINT_NO_STOP_MACHINE

int arch_klp_check_task_calltrace(struct task_struct *t,

				  bool (*check_func)(void *, int *, unsigned long),

				  void *data)

{

	struct walk_stackframe_args args = {

		.data = data,

		.ret = 0,

		.check_func = check_func,

	};



	if (t == NULL)

		return -EINVAL;

	return check_task_calltrace(t, &args, klp_check_jump_func);

}

#endif



int arch_klp_check_calltrace(bool (*check_func)(void *, int *, unsigned long), void *data)

{

	struct walk_stackframe_args args = {