[FIX] Change order of probe register

diff --git a/kprobe/dbi_uprobes.c b/kprobe/dbi_uprobes.c
index 14779d3..286de1f 100644 (file)
--- a/kprobe/dbi_uprobes.c
+++ b/kprobe/dbi_uprobes.c
@@ -252,16 +252,16 @@ int dbi_register_uretprobe(struct task_struct *task, struct kretprobe *rp, int a
 
        rp->nmissed = 0;
 
-       /* Establish function exit probe point */
-       ret = arch_prepare_uretprobe(rp, task);
+       /* Establish function entry probe point */
+       ret = __register_uprobe(&rp->kp, task, atomic);
        if (ret) {
+               free_rp_inst(rp);
                goto out;
        }
 
-       /* Establish function entry probe point */
-       ret = __register_uprobe(&rp->kp, task, atomic);
+       /* Establish function exit probe point */
+       ret = arch_prepare_uretprobe(rp, task);
        if (ret) {
-               free_rp_inst(rp);
                goto out;
        }