mirror of https://github.com/dynup/kpatch
76 lines
2.4 KiB
Diff
76 lines
2.4 KiB
Diff
diff -Nupr src.orig/fs/proc/array.c src/fs/proc/array.c
|
|
--- src.orig/fs/proc/array.c 2022-04-29 16:08:21.408780547 -0400
|
|
+++ src/fs/proc/array.c 2022-04-29 16:09:15.255972171 -0400
|
|
@@ -372,12 +372,19 @@ static inline void task_seccomp(struct s
|
|
seq_putc(m, '\n');
|
|
}
|
|
|
|
+#include <linux/livepatch.h>
|
|
static inline void task_context_switch_counts(struct seq_file *m,
|
|
struct task_struct *p)
|
|
{
|
|
+ int *newpid;
|
|
+
|
|
seq_put_decimal_ull(m, "voluntary_ctxt_switches:\t", p->nvcsw);
|
|
seq_put_decimal_ull(m, "\nnonvoluntary_ctxt_switches:\t", p->nivcsw);
|
|
seq_putc(m, '\n');
|
|
+
|
|
+ newpid = klp_shadow_get(p, 0);
|
|
+ if (newpid)
|
|
+ seq_printf(m, "newpid:\t%d\n", *newpid);
|
|
}
|
|
|
|
static void task_cpus_allowed(struct seq_file *m, struct task_struct *task)
|
|
diff -Nupr src.orig/kernel/exit.c src/kernel/exit.c
|
|
--- src.orig/kernel/exit.c 2022-04-29 16:08:21.572781130 -0400
|
|
+++ src/kernel/exit.c 2022-04-29 16:09:15.256972174 -0400
|
|
@@ -703,6 +703,7 @@ static void check_stack_usage(void)
|
|
static inline void check_stack_usage(void) {}
|
|
#endif
|
|
|
|
+#include <linux/livepatch.h>
|
|
void __noreturn do_exit(long code)
|
|
{
|
|
struct task_struct *tsk = current;
|
|
@@ -803,6 +804,8 @@ void __noreturn do_exit(long code)
|
|
exit_task_work(tsk);
|
|
exit_thread(tsk);
|
|
|
|
+ klp_shadow_free(tsk, 0, NULL);
|
|
+
|
|
/*
|
|
* Flush inherited counters to the parent - before the parent
|
|
* gets woken up by child-exit notifications.
|
|
diff -Nupr src.orig/kernel/fork.c src/kernel/fork.c
|
|
--- src.orig/kernel/fork.c 2022-04-29 16:08:21.572781130 -0400
|
|
+++ src/kernel/fork.c 2022-04-29 16:09:15.256972174 -0400
|
|
@@ -2364,6 +2364,7 @@ struct mm_struct *copy_init_mm(void)
|
|
* It copies the process, and if successful kick-starts
|
|
* it and waits for it to finish using the VM if required.
|
|
*/
|
|
+#include <linux/livepatch.h>
|
|
long _do_fork(unsigned long clone_flags,
|
|
unsigned long stack_start,
|
|
unsigned long stack_size,
|
|
@@ -2376,6 +2377,8 @@ long _do_fork(unsigned long clone_flags,
|
|
struct task_struct *p;
|
|
int trace = 0;
|
|
long nr;
|
|
+ int *newpid;
|
|
+ static int ctr = 0;
|
|
|
|
/*
|
|
* Determine whether and which event to report to ptracer. When
|
|
@@ -2402,6 +2405,11 @@ long _do_fork(unsigned long clone_flags,
|
|
if (IS_ERR(p))
|
|
return PTR_ERR(p);
|
|
|
|
+ newpid = klp_shadow_get_or_alloc(p, 0, sizeof(*newpid), GFP_KERNEL,
|
|
+ NULL, NULL);
|
|
+ if (newpid)
|
|
+ *newpid = ctr++;
|
|
+
|
|
/*
|
|
* Do this prior waking up the new thread - the thread pointer
|
|
* might get invalid after that point, if the thread exits quickly.
|