return nr_args + 1;
 }
 
-static bool prog_type_args_trusted(enum bpf_prog_type prog_type)
+static bool prog_args_trusted(const struct bpf_prog *prog)
 {
-       return prog_type == BPF_PROG_TYPE_TRACING || prog_type == BPF_PROG_TYPE_STRUCT_OPS;
+       enum bpf_attach_type atype = prog->expected_attach_type;
+
+       switch (prog->type) {
+       case BPF_PROG_TYPE_TRACING:
+               return atype == BPF_TRACE_RAW_TP || atype == BPF_TRACE_ITER;
+       case BPF_PROG_TYPE_LSM:
+       case BPF_PROG_TYPE_STRUCT_OPS:
+               return true;
+       default:
+               return false;
+       }
 }
 
 bool btf_ctx_access(int off, int size, enum bpf_access_type type,
        }
 
        info->reg_type = PTR_TO_BTF_ID;
-       if (prog_type_args_trusted(prog->type))
+       if (prog_args_trusted(prog))
                info->reg_type |= PTR_TRUSTED;
 
        if (tgt_prog) {
 
                        break;
                case PTR_TO_BTF_ID:
                case PTR_TO_BTF_ID | PTR_UNTRUSTED:
-               case PTR_TO_BTF_ID | PTR_TRUSTED:
                /* PTR_TO_BTF_ID | MEM_ALLOC always has a valid lifetime, unlike
                 * PTR_TO_BTF_ID, and an active ref_obj_id, but the same cannot
                 * be said once it is marked PTR_UNTRUSTED, hence we must handle
                 * for this case.
                 */
                case PTR_TO_BTF_ID | MEM_ALLOC | PTR_UNTRUSTED:
-               case PTR_TO_BTF_ID | PTR_UNTRUSTED | PTR_TRUSTED:
-               case PTR_TO_BTF_ID | PTR_UNTRUSTED | MEM_ALLOC | PTR_TRUSTED:
                        if (type == BPF_READ) {
                                insn->code = BPF_LDX | BPF_PROBE_MEM |
                                        BPF_SIZE((insn)->code);