On 05/04/2017 05:24 AM, Naveen N. Rao wrote:
> Use safer string manipulation functions when dealing with a
> user-provided string in kprobe_lookup_name().
> 
> Reported-by: David Laight <david.lai...@aculab.com>
> Signed-off-by: Naveen N. Rao <naveen.n....@linux.vnet.ibm.com>
> ---
> Changed to ignore return value of 0 from strscpy(), as suggested by
> Masami.
> 
> - Naveen
> 
>  arch/powerpc/kernel/kprobes.c | 47 
> ++++++++++++++++++-------------------------
>  1 file changed, 20 insertions(+), 27 deletions(-)
> 
> diff --git a/arch/powerpc/kernel/kprobes.c b/arch/powerpc/kernel/kprobes.c
> index 160ae0fa7d0d..255d28d31ca1 100644
> --- a/arch/powerpc/kernel/kprobes.c
> +++ b/arch/powerpc/kernel/kprobes.c
> @@ -53,7 +53,7 @@ bool arch_within_kprobe_blacklist(unsigned long addr)
> 
>  kprobe_opcode_t *kprobe_lookup_name(const char *name, unsigned int offset)
>  {
> -     kprobe_opcode_t *addr;
> +     kprobe_opcode_t *addr = NULL;
> 
>  #ifdef PPC64_ELF_ABI_v2
>       /* PPC64 ABIv2 needs local entry point */
> @@ -85,36 +85,29 @@ kprobe_opcode_t *kprobe_lookup_name(const char *name, 
> unsigned int offset)
>        * Also handle <module:symbol> format.
>        */
>       char dot_name[MODULE_NAME_LEN + 1 + KSYM_NAME_LEN];
> -     const char *modsym;
>       bool dot_appended = false;
> -     if ((modsym = strchr(name, ':')) != NULL) {
> -             modsym++;
> -             if (*modsym != '\0' && *modsym != '.') {
> -                     /* Convert to <module:.symbol> */
> -                     strncpy(dot_name, name, modsym - name);
> -                     dot_name[modsym - name] = '.';
> -                     dot_name[modsym - name + 1] = '\0';
> -                     strncat(dot_name, modsym,
> -                             sizeof(dot_name) - (modsym - name) - 2);
> -                     dot_appended = true;
> -             } else {
> -                     dot_name[0] = '\0';
> -                     strncat(dot_name, name, sizeof(dot_name) - 1);
> -             }
> -     } else if (name[0] != '.') {
> -             dot_name[0] = '.';
> -             dot_name[1] = '\0';
> -             strncat(dot_name, name, KSYM_NAME_LEN - 2);
> +     const char *c;
> +     ssize_t ret = 0;
> +     int len = 0;
> +
> +     if ((c = strnchr(name, MODULE_NAME_LEN, ':')) != NULL) {

Shouldn't this be MODULE_NAME_LEN + 1, since the ':' can come after a module 
name of length MODULE_NAME_LEN?

> +             c++;
> +             len = c - name;
> +             memcpy(dot_name, name, len);
> +     } else
> +             c = name;
> +
> +     if (*c != '\0' && *c != '.') {
> +             dot_name[len++] = '.';
>               dot_appended = true;
> -     } else {
> -             dot_name[0] = '\0';
> -             strncat(dot_name, name, KSYM_NAME_LEN - 1);
>       }

PC

Reply via email to