Hi Peter,

On Mon, 25 Feb 2019 16:06:03 +0100
Peter Zijlstra <pet...@infradead.org> wrote:

> On Mon, Feb 25, 2019 at 11:05:41PM +0900, Masami Hiramatsu wrote:
> 
> > +static __always_inline long __strncpy_from_unsafe_user(char *dst,
> > +                    const char __user *unsafe_addr, long count)
> > +{
> > +   if (!access_ok(unsafe_addr, count))
> > +           return -EFAULT;
> > +
> > +   return strncpy_from_unsafe_common(dst, unsafe_addr, count);
> > +}
> 
> Would something like so work for people?
> 
> ---
>  arch/x86/include/asm/uaccess.h |  8 +++++++-
>  include/linux/uaccess.h        | 18 ++++++++++++++++++
>  2 files changed, 25 insertions(+), 1 deletion(-)
> 
> diff --git a/arch/x86/include/asm/uaccess.h b/arch/x86/include/asm/uaccess.h
> index 780f2b42c8ef..3125d129d3b6 100644
> --- a/arch/x86/include/asm/uaccess.h
> +++ b/arch/x86/include/asm/uaccess.h
> @@ -92,12 +92,18 @@ static inline bool __chk_range_not_ok(unsigned long addr, 
> unsigned long size, un
>   * checks that the pointer is in the user space range - after calling
>   * this function, memory access functions may still return -EFAULT.
>   */
> -#define access_ok(addr, size)                                        \
> +#define access_ok(addr, size)                                                
> \
>  ({                                                                   \
>       WARN_ON_IN_IRQ();                                               \
>       likely(!__range_not_ok(addr, size, user_addr_max()));           \
>  })
>  
> +#define user_access_ok(addr, size)                                   \
> +({                                                                   \
> +     WARN_ON_ONCE(!segment_eq(get_fs(), USER_DS));                   \
> +     likely(!__range_not_ok(addr, size, user_addr_max()));           \
> +})
> +
>  /*
>   * These are the main single-value transfer routines.  They automatically
>   * use the right size if we just have the right pointer type.
> diff --git a/include/linux/uaccess.h b/include/linux/uaccess.h
> index 37b226e8df13..088f2ae09e14 100644
> --- a/include/linux/uaccess.h
> +++ b/include/linux/uaccess.h
> @@ -10,6 +10,24 @@
>  
>  #include <asm/uaccess.h>
>  
> +/**
> + * user_access_ok: Checks if a user space pointer is valid
> + * @addr: User space pointer to start of block to check
> + * @size: Size of block to check
> + *
> + * Context: User context or explicit set_fs(USER_DS).
> + *
> + * This function is very much like access_ok(), except it (may) have 
> different
> + * context validation. In general we must be very careful when using this.
> + */
> +#ifndef user_access_ok
> +#define user_access_ok(addr, size)                                   \
> +({                                                                   \
> +     WARN_ON_ONCE(!segment_eq(get_fs(), USER_DS));                   \
> +     access_ok(addr, size);                                          \
> +})
> +#endif
> +
>  /*
>   * Architectures should provide two primitives (raw_copy_{to,from}_user())
>   * and get rid of their private instances of copy_{to,from}_user() and

Yeah, looks good to me. 

Thank you,


-- 
Masami Hiramatsu <mhira...@kernel.org>

Reply via email to