33 static int bpf_trampoline_update(struct bpf_trampoline *tr, bool lock_direct_mutex);
37 	struct bpf_trampoline *tr = ops->private;  in bpf_tramp_ftrace_ops_func()  local
44 		lockdep_assert_held_once(&tr->mutex);  in bpf_tramp_ftrace_ops_func()
51 		if ((tr->flags & BPF_TRAMP_F_CALL_ORIG) &&  in bpf_tramp_ftrace_ops_func()
52 		    !(tr->flags & BPF_TRAMP_F_ORIG_STACK)) {  in bpf_tramp_ftrace_ops_func()
53 			if (WARN_ON_ONCE(tr->flags & BPF_TRAMP_F_SHARE_IPMODIFY))  in bpf_tramp_ftrace_ops_func()
56 			tr->flags |= BPF_TRAMP_F_SHARE_IPMODIFY;  in bpf_tramp_ftrace_ops_func()
75 	if (!mutex_trylock(&tr->mutex)) {  in bpf_tramp_ftrace_ops_func()
85 		tr->flags |= BPF_TRAMP_F_SHARE_IPMODIFY;  in bpf_tramp_ftrace_ops_func()
87 		if ((tr->flags & BPF_TRAMP_F_CALL_ORIG) &&  in bpf_tramp_ftrace_ops_func()
88 		    !(tr->flags & BPF_TRAMP_F_ORIG_STACK))  in bpf_tramp_ftrace_ops_func()
89 			ret = bpf_trampoline_update(tr, false /* lock_direct_mutex */);  in bpf_tramp_ftrace_ops_func()
92 		tr->flags &= ~BPF_TRAMP_F_SHARE_IPMODIFY;  in bpf_tramp_ftrace_ops_func()
94 		if (tr->flags & BPF_TRAMP_F_ORIG_STACK)  in bpf_tramp_ftrace_ops_func()
95 			ret = bpf_trampoline_update(tr, false /* lock_direct_mutex */);  in bpf_tramp_ftrace_ops_func()
102 	mutex_unlock(&tr->mutex);  in bpf_tramp_ftrace_ops_func()
140 	struct bpf_trampoline *tr;  in bpf_trampoline_lookup()  local
146 	hlist_for_each_entry(tr, head, hlist) {  in bpf_trampoline_lookup()
147 		if (tr->key == key) {  in bpf_trampoline_lookup()
148 			refcount_inc(&tr->refcnt);  in bpf_trampoline_lookup()
152 	tr = kzalloc(sizeof(*tr), GFP_KERNEL);  in bpf_trampoline_lookup()
153 	if (!tr)  in bpf_trampoline_lookup()
156 	tr->fops = kzalloc(sizeof(struct ftrace_ops), GFP_KERNEL);  in bpf_trampoline_lookup()
157 	if (!tr->fops) {  in bpf_trampoline_lookup()
158 		kfree(tr);  in bpf_trampoline_lookup()
159 		tr = NULL;  in bpf_trampoline_lookup()
162 	tr->fops->private = tr;  in bpf_trampoline_lookup()
163 	tr->fops->ops_func = bpf_tramp_ftrace_ops_func;  in bpf_trampoline_lookup()
166 	tr->key = key;  in bpf_trampoline_lookup()
167 	INIT_HLIST_NODE(&tr->hlist);  in bpf_trampoline_lookup()
168 	hlist_add_head(&tr->hlist, head);  in bpf_trampoline_lookup()
169 	refcount_set(&tr->refcnt, 1);  in bpf_trampoline_lookup()
170 	mutex_init(&tr->mutex);  in bpf_trampoline_lookup()
172 		INIT_HLIST_HEAD(&tr->progs_hlist[i]);  in bpf_trampoline_lookup()
175 	return tr;  in bpf_trampoline_lookup()
178 static int unregister_fentry(struct bpf_trampoline *tr, void *old_addr)  in unregister_fentry()  argument
180 	void *ip = tr->func.addr;  in unregister_fentry()
183 	if (tr->func.ftrace_managed)  in unregister_fentry()
184 		ret = unregister_ftrace_direct(tr->fops, (long)old_addr, false);  in unregister_fentry()
191 static int modify_fentry(struct bpf_trampoline *tr, void *old_addr, void *new_addr,  in modify_fentry()  argument
194 	void *ip = tr->func.addr;  in modify_fentry()
197 	if (tr->func.ftrace_managed) {  in modify_fentry()
199 			ret = modify_ftrace_direct(tr->fops, (long)new_addr);  in modify_fentry()
201 			ret = modify_ftrace_direct_nolock(tr->fops, (long)new_addr);  in modify_fentry()
209 static int register_fentry(struct bpf_trampoline *tr, void *new_addr)  in register_fentry()  argument
211 	void *ip = tr->func.addr;  in register_fentry()
217 		if (!tr->fops)  in register_fentry()
219 		tr->func.ftrace_managed = true;  in register_fentry()
222 	if (tr->func.ftrace_managed) {  in register_fentry()
223 		ftrace_set_filter_ip(tr->fops, (unsigned long)ip, 0, 1);  in register_fentry()
224 		ret = register_ftrace_direct(tr->fops, (long)new_addr);  in register_fentry()
233 bpf_trampoline_get_progs(const struct bpf_trampoline *tr, int *total, bool *ip_arg)  in bpf_trampoline_get_progs()  argument
246 		tlinks[kind].nr_links = tr->progs_cnt[kind];  in bpf_trampoline_get_progs()
247 		*total += tr->progs_cnt[kind];  in bpf_trampoline_get_progs()
250 		hlist_for_each_entry(link, &tr->progs_hlist[kind], tramp_hlist) {  in bpf_trampoline_get_progs()
398 static int bpf_trampoline_update(struct bpf_trampoline *tr, bool lock_direct_mutex)  in bpf_trampoline_update()  argument
402 	u32 orig_flags = tr->flags;  in bpf_trampoline_update()
406 	tlinks = bpf_trampoline_get_progs(tr, &total, &ip_arg);  in bpf_trampoline_update()
411 		err = unregister_fentry(tr, tr->cur_image->image);  in bpf_trampoline_update()
412 		bpf_tramp_image_put(tr->cur_image);  in bpf_trampoline_update()
413 		tr->cur_image = NULL;  in bpf_trampoline_update()
418 	tr->flags &= (BPF_TRAMP_F_SHARE_IPMODIFY | BPF_TRAMP_F_TAIL_CALL_CTX);  in bpf_trampoline_update()
425 		tr->flags |= BPF_TRAMP_F_CALL_ORIG | BPF_TRAMP_F_SKIP_FRAME;  in bpf_trampoline_update()
427 		tr->flags |= BPF_TRAMP_F_RESTORE_REGS;  in bpf_trampoline_update()
431 		tr->flags |= BPF_TRAMP_F_IP_ARG;  in bpf_trampoline_update()
435 	if ((tr->flags & BPF_TRAMP_F_SHARE_IPMODIFY) &&  in bpf_trampoline_update()
436 	    (tr->flags & BPF_TRAMP_F_CALL_ORIG))  in bpf_trampoline_update()
437 		tr->flags |= BPF_TRAMP_F_ORIG_STACK;  in bpf_trampoline_update()
440 	size = arch_bpf_trampoline_size(&tr->func.model, tr->flags,  in bpf_trampoline_update()
441 					tlinks, tr->func.addr);  in bpf_trampoline_update()
452 	im = bpf_tramp_image_alloc(tr->key, size);  in bpf_trampoline_update()
459 					  &tr->func.model, tr->flags, tlinks,  in bpf_trampoline_update()
460 					  tr->func.addr);  in bpf_trampoline_update()
468 	WARN_ON(tr->cur_image && total == 0);  in bpf_trampoline_update()
469 	if (tr->cur_image)  in bpf_trampoline_update()
471 		err = modify_fentry(tr, tr->cur_image->image, im->image, lock_direct_mutex);  in bpf_trampoline_update()
474 		err = register_fentry(tr, im->image);  in bpf_trampoline_update()
483 		tr->fops->func = NULL;  in bpf_trampoline_update()
484 		tr->fops->trampoline = 0;  in bpf_trampoline_update()
494 	if (tr->cur_image)  in bpf_trampoline_update()
495 		bpf_tramp_image_put(tr->cur_image);  in bpf_trampoline_update()
496 	tr->cur_image = im;  in bpf_trampoline_update()
500 		tr->flags = orig_flags;  in bpf_trampoline_update()
550 				      struct bpf_trampoline *tr,  in __bpf_trampoline_link_prog()  argument
559 	if (tr->extension_prog)  in __bpf_trampoline_link_prog()
566 		cnt += tr->progs_cnt[i];  in __bpf_trampoline_link_prog()
575 		tr->extension_prog = link->link.prog;  in __bpf_trampoline_link_prog()
576 		return bpf_arch_text_poke(tr->func.addr, BPF_MOD_JUMP, NULL,  in __bpf_trampoline_link_prog()
584 	hlist_for_each_entry(link_exiting, &tr->progs_hlist[kind], tramp_hlist) {  in __bpf_trampoline_link_prog()
591 	hlist_add_head(&link->tramp_hlist, &tr->progs_hlist[kind]);  in __bpf_trampoline_link_prog()
592 	tr->progs_cnt[kind]++;  in __bpf_trampoline_link_prog()
593 	err = bpf_trampoline_update(tr, true /* lock_direct_mutex */);  in __bpf_trampoline_link_prog()
596 		tr->progs_cnt[kind]--;  in __bpf_trampoline_link_prog()
602 			     struct bpf_trampoline *tr,  in bpf_trampoline_link_prog()  argument
607 	mutex_lock(&tr->mutex);  in bpf_trampoline_link_prog()
608 	err = __bpf_trampoline_link_prog(link, tr, tgt_prog);  in bpf_trampoline_link_prog()
609 	mutex_unlock(&tr->mutex);  in bpf_trampoline_link_prog()
614 					struct bpf_trampoline *tr,  in __bpf_trampoline_unlink_prog()  argument
622 		WARN_ON_ONCE(!tr->extension_prog);  in __bpf_trampoline_unlink_prog()
623 		err = bpf_arch_text_poke(tr->func.addr, BPF_MOD_JUMP,  in __bpf_trampoline_unlink_prog()
624 					 tr->extension_prog->bpf_func, NULL);  in __bpf_trampoline_unlink_prog()
625 		tr->extension_prog = NULL;  in __bpf_trampoline_unlink_prog()
631 	tr->progs_cnt[kind]--;  in __bpf_trampoline_unlink_prog()
632 	return bpf_trampoline_update(tr, true /* lock_direct_mutex */);  in __bpf_trampoline_unlink_prog()
637 			       struct bpf_trampoline *tr,  in bpf_trampoline_unlink_prog()  argument
642 	mutex_lock(&tr->mutex);  in bpf_trampoline_unlink_prog()
643 	err = __bpf_trampoline_unlink_prog(link, tr, tgt_prog);  in bpf_trampoline_unlink_prog()
644 	mutex_unlock(&tr->mutex);  in bpf_trampoline_unlink_prog()
710 static struct bpf_shim_tramp_link *cgroup_shim_find(struct bpf_trampoline *tr,  in cgroup_shim_find()  argument
717 		hlist_for_each_entry(link, &tr->progs_hlist[kind], tramp_hlist) {  in cgroup_shim_find()
733 	struct bpf_trampoline *tr;  in bpf_trampoline_link_cgroup_shim()  local
748 	tr = bpf_trampoline_get(key, &tgt_info);  in bpf_trampoline_link_cgroup_shim()
749 	if (!tr)  in bpf_trampoline_link_cgroup_shim()
752 	mutex_lock(&tr->mutex);  in bpf_trampoline_link_cgroup_shim()
754 	shim_link = cgroup_shim_find(tr, bpf_func);  in bpf_trampoline_link_cgroup_shim()
759 		mutex_unlock(&tr->mutex);  in bpf_trampoline_link_cgroup_shim()
760 		bpf_trampoline_put(tr); /* bpf_trampoline_get above */  in bpf_trampoline_link_cgroup_shim()
772 	err = __bpf_trampoline_link_prog(&shim_link->link, tr, NULL);  in bpf_trampoline_link_cgroup_shim()
776 	shim_link->trampoline = tr;  in bpf_trampoline_link_cgroup_shim()
779 	mutex_unlock(&tr->mutex);  in bpf_trampoline_link_cgroup_shim()
783 	mutex_unlock(&tr->mutex);  in bpf_trampoline_link_cgroup_shim()
789 	bpf_trampoline_put(tr); /* bpf_trampoline_get above */  in bpf_trampoline_link_cgroup_shim()
797 	struct bpf_trampoline *tr;  in bpf_trampoline_unlink_cgroup_shim()  local
805 	tr = bpf_trampoline_lookup(key);  in bpf_trampoline_unlink_cgroup_shim()
806 	if (WARN_ON_ONCE(!tr))  in bpf_trampoline_unlink_cgroup_shim()
809 	mutex_lock(&tr->mutex);  in bpf_trampoline_unlink_cgroup_shim()
810 	shim_link = cgroup_shim_find(tr, bpf_func);  in bpf_trampoline_unlink_cgroup_shim()
811 	mutex_unlock(&tr->mutex);  in bpf_trampoline_unlink_cgroup_shim()
816 	bpf_trampoline_put(tr); /* bpf_trampoline_lookup above */  in bpf_trampoline_unlink_cgroup_shim()
823 	struct bpf_trampoline *tr;  in bpf_trampoline_get()  local
825 	tr = bpf_trampoline_lookup(key);  in bpf_trampoline_get()
826 	if (!tr)  in bpf_trampoline_get()
829 	mutex_lock(&tr->mutex);  in bpf_trampoline_get()
830 	if (tr->func.addr)  in bpf_trampoline_get()
833 	memcpy(&tr->func.model, &tgt_info->fmodel, sizeof(tgt_info->fmodel));  in bpf_trampoline_get()
834 	tr->func.addr = (void *)tgt_info->tgt_addr;  in bpf_trampoline_get()
836 	mutex_unlock(&tr->mutex);  in bpf_trampoline_get()
837 	return tr;  in bpf_trampoline_get()
840 void bpf_trampoline_put(struct bpf_trampoline *tr)  in bpf_trampoline_put()  argument
844 	if (!tr)  in bpf_trampoline_put()
847 	if (!refcount_dec_and_test(&tr->refcnt))  in bpf_trampoline_put()
849 	WARN_ON_ONCE(mutex_is_locked(&tr->mutex));  in bpf_trampoline_put()
852 		if (WARN_ON_ONCE(!hlist_empty(&tr->progs_hlist[i])))  in bpf_trampoline_put()
861 	hlist_del(&tr->hlist);  in bpf_trampoline_put()
862 	if (tr->fops) {  in bpf_trampoline_put()
863 		ftrace_free_filter(tr->fops);  in bpf_trampoline_put()
864 		kfree(tr->fops);  in bpf_trampoline_put()
866 	kfree(tr);  in bpf_trampoline_put()
1048 void notrace __bpf_tramp_enter(struct bpf_tramp_image *tr)  in __bpf_tramp_enter()  argument
1050 	percpu_ref_get(&tr->pcref);  in __bpf_tramp_enter()
1053 void notrace __bpf_tramp_exit(struct bpf_tramp_image *tr)  in __bpf_tramp_exit()  argument
1055 	percpu_ref_put(&tr->pcref);  in __bpf_tramp_exit()