[FIX] initialization uretprobe

diff --git a/us_manager/sspt/sspt.h b/us_manager/sspt/sspt.h
index 98bfa3b..984da5e 100644 (file)
--- a/us_manager/sspt/sspt.h
+++ b/us_manager/sspt/sspt.h
@@ -108,9 +108,16 @@ static inline int sspt_register_usprobe(struct us_ip *ip)
 {
        int ret = 0;
 
+       /* for juprobe */
        ip->jprobe.priv_arg = ip;
        ip->jprobe.up.task = ip->page->file->proc->task;
        ip->jprobe.up.sm = ip->page->file->proc->sm;
+
+       /* for retuprobe */
+       ip->retprobe.priv_arg = ip;
+       ip->retprobe.up.task = ip->page->file->proc->task;
+       ip->retprobe.up.sm = ip->page->file->proc->sm;
+
        ret = dbi_register_ujprobe(&ip->jprobe);
        if (ret) {
                if (ret == -ENOEXEC) {
@@ -124,9 +131,6 @@ static inline int sspt_register_usprobe(struct us_ip *ip)
        }
 
        if (ip->flag_retprobe) {
-               ip->retprobe.priv_arg = ip;
-               ip->retprobe.up.task = ip->page->file->proc->task;
-               ip->retprobe.up.sm = ip->page->file->proc->sm;
                ret = dbi_register_uretprobe(&ip->retprobe);
                if (ret) {
                        struct sspt_file *file = ip->page->file;