Commit cf4303b7 authored by Menglong Dong's avatar Menglong Dong Committed by Alexei Starovoitov
Browse files

bpf: use rcu_read_lock_dont_migrate() for bpf_task_storage_free() 



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

Signed-off-by: default avatarMenglong Dong <dongml2@chinatelecom.cn>
Link: https://lore.kernel.org/r/20250821090609.42508-6-dongml2@chinatelecom.cn


Signed-off-by: default avatarAlexei Starovoitov <ast@kernel.org>
parent 68748f03

kernel/bpf/bpf_task_storage.c

+2 −4
Original line number Diff line number Diff line
@@ -70,8 +70,7 @@ void bpf_task_storage_free(struct task_struct *task) 
{

	struct bpf_local_storage *local_storage;



	migrate_disable();

	rcu_read_lock();

	rcu_read_lock_dont_migrate();



	local_storage = rcu_dereference(task->bpf_storage);

	if (!local_storage)
@@ -81,8 +80,7 @@ void bpf_task_storage_free(struct task_struct *task) 
	bpf_local_storage_destroy(local_storage);

	bpf_task_storage_unlock();

out:

	rcu_read_unlock();

	migrate_enable();

	rcu_read_unlock_migrate();

}



static void *bpf_pid_task_storage_lookup_elem(struct bpf_map *map, void *key)