[IMPROVE] Add swap_put_uarg()

Change-Id: I87057e7b496c655b582aef86484403c0ff6a85e7
Signed-off-by: Alexander Aksenov <a.aksenov@samsung.com>

diff --git a/uprobe/arch/arm/swap-asm/swap_uprobes.h b/uprobe/arch/arm/swap-asm/swap_uprobes.h
index 826e308..c519d12 100644 (file)
--- a/uprobe/arch/arm/swap-asm/swap_uprobes.h
+++ b/uprobe/arch/arm/swap-asm/swap_uprobes.h
@@ -109,6 +109,27 @@ static inline unsigned long swap_get_uarg(struct pt_regs *regs, unsigned long n)
        return addr;
 }
 
+static inline void swap_put_uarg(struct pt_regs *regs, unsigned long n,
+                                 unsigned long val)
+{
+       u32 *ptr;
+
+       switch (n) {
+       case 0:
+               regs->ARM_r0 = val;
+       case 1:
+               regs->ARM_r1 = val;
+       case 2:
+               regs->ARM_r2 = val;
+       case 3:
+               regs->ARM_r3 = val;
+       }
+
+       ptr = (u32 *)regs->ARM_sp + n - 4;
+       if (put_user(val, ptr))
+               printk("failed to dereference a pointer, ptr=%p\n", ptr);
+}
+
 int swap_arch_init_uprobes(void);
 void swap_arch_exit_uprobes(void);
 

diff --git a/uprobe/arch/x86/swap-asm/swap_uprobes.h b/uprobe/arch/x86/swap-asm/swap_uprobes.h
index 3ca6334..360c033 100644 (file)
--- a/uprobe/arch/x86/swap-asm/swap_uprobes.h
+++ b/uprobe/arch/x86/swap-asm/swap_uprobes.h
@@ -103,6 +103,17 @@ static inline unsigned long swap_get_uarg(struct pt_regs *regs, unsigned long n)
        return addr;
 }
 
+static inline void swap_put_uarg(struct pt_regs *regs, unsigned long n,
+                                 unsigned long val)
+{
+       u32 *ptr;
+
+       /* 1 - return address saved on top of the stack */
+       ptr = (u32 *)regs->sp + n + 1;
+       if (put_user(val, ptr))
+               printk("failed to dereference a pointer, ptr=%p\n", ptr);
+}
+
 int swap_arch_init_uprobes(void);
 void swap_arch_exit_uprobes(void);