We are going to remove "ftrace_ops->private == bpf_trampoline" setup
in following changes.

Adding ip argument to ftrace_ops_func_t callback function, so we can
use it to look up the trampoline.

Signed-off-by: Jiri Olsa <[email protected]>
---
 include/linux/ftrace.h  | 2 +-
 kernel/bpf/trampoline.c | 3 ++-
 kernel/trace/ftrace.c   | 6 +++---
 3 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/include/linux/ftrace.h b/include/linux/ftrace.h
index c27b7381c5f1..515cf5b723dd 100644
--- a/include/linux/ftrace.h
+++ b/include/linux/ftrace.h
@@ -402,7 +402,7 @@ enum ftrace_ops_cmd {
  *        Negative on failure. The return value is dependent on the
  *        callback.
  */
-typedef int (*ftrace_ops_func_t)(struct ftrace_ops *op, enum ftrace_ops_cmd 
cmd);
+typedef int (*ftrace_ops_func_t)(struct ftrace_ops *op, unsigned long ip, enum 
ftrace_ops_cmd cmd);
 
 #ifdef CONFIG_DYNAMIC_FTRACE
 
diff --git a/kernel/bpf/trampoline.c b/kernel/bpf/trampoline.c
index f298bafab76e..17af2aad8382 100644
--- a/kernel/bpf/trampoline.c
+++ b/kernel/bpf/trampoline.c
@@ -33,7 +33,8 @@ static DEFINE_MUTEX(trampoline_mutex);
 #ifdef CONFIG_DYNAMIC_FTRACE_WITH_DIRECT_CALLS
 static int bpf_trampoline_update(struct bpf_trampoline *tr, bool 
lock_direct_mutex);
 
-static int bpf_tramp_ftrace_ops_func(struct ftrace_ops *ops, enum 
ftrace_ops_cmd cmd)
+static int bpf_tramp_ftrace_ops_func(struct ftrace_ops *ops, unsigned long ip,
+                                    enum ftrace_ops_cmd cmd)
 {
        struct bpf_trampoline *tr = ops->private;
        int ret = 0;
diff --git a/kernel/trace/ftrace.c b/kernel/trace/ftrace.c
index c77f620b3eb3..9582cb374d4f 100644
--- a/kernel/trace/ftrace.c
+++ b/kernel/trace/ftrace.c
@@ -2049,7 +2049,7 @@ static int __ftrace_hash_update_ipmodify(struct 
ftrace_ops *ops,
                                 */
                                if (!ops->ops_func)
                                        return -EBUSY;
-                               ret = ops->ops_func(ops, 
FTRACE_OPS_CMD_ENABLE_SHARE_IPMODIFY_SELF);
+                               ret = ops->ops_func(ops, rec->ip, 
FTRACE_OPS_CMD_ENABLE_SHARE_IPMODIFY_SELF);
                                if (ret)
                                        return ret;
                        } else if (is_ipmodify) {
@@ -8984,7 +8984,7 @@ static int prepare_direct_functions_for_ipmodify(struct 
ftrace_ops *ops)
                                if (!op->ops_func)
                                        return -EBUSY;
 
-                               ret = op->ops_func(op, 
FTRACE_OPS_CMD_ENABLE_SHARE_IPMODIFY_PEER);
+                               ret = op->ops_func(op, ip, 
FTRACE_OPS_CMD_ENABLE_SHARE_IPMODIFY_PEER);
                                if (ret)
                                        return ret;
                        }
@@ -9031,7 +9031,7 @@ static void 
cleanup_direct_functions_after_ipmodify(struct ftrace_ops *ops)
 
                        /* The cleanup is optional, ignore any errors */
                        if (found_op && op->ops_func)
-                               op->ops_func(op, 
FTRACE_OPS_CMD_DISABLE_SHARE_IPMODIFY_PEER);
+                               op->ops_func(op, ip, 
FTRACE_OPS_CMD_DISABLE_SHARE_IPMODIFY_PEER);
                }
        }
        mutex_unlock(&direct_mutex);
-- 
2.52.0


Reply via email to