bpf: use rcu_read_lock_dont_migrate() for bpf_inode_storage_free()

Use rcu_read_lock_dont_migrate() and rcu_read_unlock_migrate() in
bpf_inode_storage_free to obtain better performance when PREEMPT_RCU is
not enabled.

Signed-off-by: Menglong Dong <dongml2@chinatelecom.cn>
Link: https://lore.kernel.org/r/20250821090609.42508-4-dongml2@chinatelecom.cn
Signed-off-by: Alexei Starovoitov <ast@kernel.org>

diff --git a/kernel/bpf/bpf_inode_storage.c b/kernel/bpf/bpf_inode_storage.c
index 15a3eb9b02d94c1bdd9bb6ca1600a86ab85efc79..e54cce2b91754c9a03eb840f0bec2247c3c9480e 100644 (file)
--- a/kernel/bpf/bpf_inode_storage.c
+++ b/kernel/bpf/bpf_inode_storage.c
@@ -62,8 +62,7 @@ void bpf_inode_storage_free(struct inode *inode)
        if (!bsb)
                return;
 
-       migrate_disable();
-       rcu_read_lock();
+       rcu_read_lock_dont_migrate();
 
        local_storage = rcu_dereference(bsb->storage);
        if (!local_storage)
@@ -71,8 +70,7 @@ void bpf_inode_storage_free(struct inode *inode)
 
        bpf_local_storage_destroy(local_storage);
 out:
-       rcu_read_unlock();
-       migrate_enable();
+       rcu_read_unlock_migrate();
 }
 
 static void *bpf_fd_inode_storage_lookup_elem(struct bpf_map *map, void *key)