On Mon, Nov 16, 2020 at 11:48 PM KP Singh <kpsi...@chromium.org> wrote:
>
> [...]
>
> > >
> > > +BPF_CALL_2(bpf_lsm_set_bprm_opts, struct linux_binprm *, bprm, u64, 
> > > flags)
> > > +{
> >
> > This should also reject invalid flags. I'd rather change this helper from 
> > RET_VOID
> > to RET_INTEGER and throw -EINVAL for everything other than 
> > BPF_LSM_F_BPRM_SECUREEXEC
> > passed in here including zero so it can be extended in future.
>
> Sounds good, I added:
>
>  enum {
>         BPF_LSM_F_BPRM_SECUREEXEC       = (1ULL << 0),
> +       /* Mask for all the currently supported BPRM options */
> +       BPF_LSM_F_BRPM_OPTS_MASK        = 0x1ULL,
>  };
>
> changed the return type to RET_INTEGER as suggested checking for
> invalid flags as:
>
>  BPF_CALL_2(bpf_lsm_set_bprm_opts, struct linux_binprm *, bprm, u64, flags)
>  {
> +
> +       if (flags & !BPF_LSM_F_BRPM_OPTS_MASK)
> +               return -EINVAL;
>
> Do let me know if this is okay and I can spin up a v2 with these changes.

Oops this should have been:

      if (flags & ~BPF_LSM_F_BRPM_OPTS_MASK)
               return -EINVAL;

>
> - KP
>
> >
> > > +     bprm->secureexec = (flags & BPF_LSM_F_BPRM_SECUREEXEC);
> > > +     return 0;
> > > +}
> > > +
> > > +BTF_ID_LIST_SINGLE(bpf_lsm_set_bprm_opts_btf_ids, struct, linux_binprm)
> > > +
> > > +const static struct bpf_func_proto bpf_lsm_set_bprm_opts_proto = {
> > > +     .func           = bpf_lsm_set_bprm_opts,
> > > +     .gpl_only       = false,
> > > +     .ret_type       = RET_VOID,
> > > +     .arg1_type      = ARG_PTR_TO_BTF_ID,
> > > +     .arg1_btf_id    = &bpf_lsm_set_bprm_opts_btf_ids[0],
> > > +     .arg2_type      = ARG_ANYTHING,
> > > +};
> > > +

Reply via email to