mirror of
				https://github.com/torvalds/linux.git
				synced 2025-11-04 02:30:34 +02:00 
			
		
		
		
	bpf: Allow local kptrs to be exchanged via bpf_kptr_xchg
The previous patch added necessary plumbing for verifier and runtime to know what to do with non-kernel PTR_TO_BTF_IDs in map values, but didn't provide any way to get such local kptrs into a map value. This patch modifies verifier handling of bpf_kptr_xchg to allow MEM_ALLOC kptr types. check_reg_type is modified accept MEM_ALLOC-flagged input to bpf_kptr_xchg despite such types not being in btf_ptr_types. This could have been done with a MAYBE_MEM_ALLOC equivalent to MAYBE_NULL, but bpf_kptr_xchg is the only helper that I can forsee using MAYBE_MEM_ALLOC, so keep it special-cased for now. The verifier tags bpf_kptr_xchg retval MEM_ALLOC if and only if the BTF associated with the retval is not kernel BTF. Signed-off-by: Dave Marchevsky <davemarchevsky@fb.com> Link: https://lore.kernel.org/r/20230310230743.2320707-3-davemarchevsky@fb.com Signed-off-by: Alexei Starovoitov <ast@kernel.org>
This commit is contained in:
		
							parent
							
								
									c8e1875409
								
							
						
					
					
						commit
						738c96d5e2
					
				
					 1 changed files with 7 additions and 1 deletions
				
			
		| 
						 | 
				
			
			@ -7123,6 +7123,9 @@ static int check_reg_type(struct bpf_verifier_env *env, u32 regno,
 | 
			
		|||
	if (arg_type & PTR_MAYBE_NULL)
 | 
			
		||||
		type &= ~PTR_MAYBE_NULL;
 | 
			
		||||
 | 
			
		||||
	if (meta->func_id == BPF_FUNC_kptr_xchg && type & MEM_ALLOC)
 | 
			
		||||
		type &= ~MEM_ALLOC;
 | 
			
		||||
 | 
			
		||||
	for (i = 0; i < ARRAY_SIZE(compatible->types); i++) {
 | 
			
		||||
		expected = compatible->types[i];
 | 
			
		||||
		if (expected == NOT_INIT)
 | 
			
		||||
| 
						 | 
				
			
			@ -7185,7 +7188,8 @@ static int check_reg_type(struct bpf_verifier_env *env, u32 regno,
 | 
			
		|||
		break;
 | 
			
		||||
	}
 | 
			
		||||
	case PTR_TO_BTF_ID | MEM_ALLOC:
 | 
			
		||||
		if (meta->func_id != BPF_FUNC_spin_lock && meta->func_id != BPF_FUNC_spin_unlock) {
 | 
			
		||||
		if (meta->func_id != BPF_FUNC_spin_lock && meta->func_id != BPF_FUNC_spin_unlock &&
 | 
			
		||||
		    meta->func_id != BPF_FUNC_kptr_xchg) {
 | 
			
		||||
			verbose(env, "verifier internal error: unimplemented handling of MEM_ALLOC\n");
 | 
			
		||||
			return -EFAULT;
 | 
			
		||||
		}
 | 
			
		||||
| 
						 | 
				
			
			@ -9151,6 +9155,8 @@ static int check_helper_call(struct bpf_verifier_env *env, struct bpf_insn *insn
 | 
			
		|||
		if (func_id == BPF_FUNC_kptr_xchg) {
 | 
			
		||||
			ret_btf = meta.kptr_field->kptr.btf;
 | 
			
		||||
			ret_btf_id = meta.kptr_field->kptr.btf_id;
 | 
			
		||||
			if (!btf_is_kernel(ret_btf))
 | 
			
		||||
				regs[BPF_REG_0].type |= MEM_ALLOC;
 | 
			
		||||
		} else {
 | 
			
		||||
			if (fn->ret_btf_id == BPF_PTR_POISON) {
 | 
			
		||||
				verbose(env, "verifier internal error:");
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
		Reference in a new issue