Hi Sjur,

Sorry for the (very) late reply.

On Sun, Feb 10, 2013 at 12:39:05PM +0100, sjur.brandel...@stericsson.com wrote:
> From: Sjur Brændeland <sjur.brandel...@stericsson.com>
> 
> Refactor rproc_elf_find_rsc_table and split out the scanning
> for the section header named resource table. This is done to
> prepare for loading firmware once.
> 
> Signed-off-by: Sjur Brændeland <sjur.brandel...@stericsson.com>
> ---
>  drivers/remoteproc/remoteproc_elf_loader.c |   79 
> ++++++++++++++++++----------
>  1 files changed, 52 insertions(+), 27 deletions(-)
> 
> diff --git a/drivers/remoteproc/remoteproc_elf_loader.c 
> b/drivers/remoteproc/remoteproc_elf_loader.c
> index 0d36f94..a958950 100644
> --- a/drivers/remoteproc/remoteproc_elf_loader.c
> +++ b/drivers/remoteproc/remoteproc_elf_loader.c
> @@ -208,38 +208,19 @@ rproc_elf_load_segments(struct rproc *rproc, const 
> struct firmware *fw)
>       return ret;
>  }
>  
> -/**
> - * rproc_elf_find_rsc_table() - find the resource table
> - * @rproc: the rproc handle
> - * @fw: the ELF firmware image
> - * @tablesz: place holder for providing back the table size
> - *
> - * This function finds the resource table inside the remote processor's
> - * firmware. It is used both upon the registration of @rproc (in order
> - * to look for and register the supported virito devices), and when the
> - * @rproc is booted.
> - *
> - * Returns the pointer to the resource table if it is found, and write its
> - * size into @tablesz. If a valid table isn't found, NULL is returned
> - * (and @tablesz isn't set).
> - */
> -static struct resource_table *
> -rproc_elf_find_rsc_table(struct rproc *rproc, const struct firmware *fw,
> -                                                     int *tablesz)
> +static struct elf32_shdr *
> +find_rsc_shdr(struct device *dev, struct elf32_hdr *ehdr, size_t fw_size)
>  {
> -     struct elf32_hdr *ehdr;
>       struct elf32_shdr *shdr;
> +     int i;
>       const char *name_table;
> -     struct device *dev = &rproc->dev;
>       struct resource_table *table = NULL;
> -     int i;
> -     const u8 *elf_data = fw->data;
> +     const u8 *elf_data = (void *)ehdr;
>  
> -     ehdr = (struct elf32_hdr *)elf_data;
> +     /* look for the resource table and handle it */
>       shdr = (struct elf32_shdr *)(elf_data + ehdr->e_shoff);
>       name_table = elf_data + shdr[ehdr->e_shstrndx].sh_offset;
>  
> -     /* look for the resource table and handle it */
>       for (i = 0; i < ehdr->e_shnum; i++, shdr++) {
>               int size = shdr->sh_size;
>               int offset = shdr->sh_offset;
> @@ -250,7 +231,7 @@ rproc_elf_find_rsc_table(struct rproc *rproc, const 
> struct firmware *fw,
>               table = (struct resource_table *)(elf_data + offset);
>  
>               /* make sure we have the entire table */
> -             if (offset + size > fw->size) {
> +             if (offset + size > fw_size) {

While we're at it, perhaps also verify that there aren't any integer overflows
here?

>                       dev_err(dev, "resource table truncated\n");
>                       return NULL;
>               }
> @@ -280,10 +261,54 @@ rproc_elf_find_rsc_table(struct rproc *rproc, const 
> struct firmware *fw,
>                       return NULL;
>               }
>  
> -             *tablesz = shdr->sh_size;
> -             break;
> +             return shdr;
>       }
>  
> +     return NULL;
> +}
> +
> +/**
> + * rproc_elf_find_rsc_table() - find the resource table
> + * @rproc: the rproc handle
> + * @fw: the ELF firmware image
> + * @tablesz: place holder for providing back the table size
> + *
> + * This function finds the resource table inside the remote processor's
> + * firmware. It is used both upon the registration of @rproc (in order
> + * to look for and register the supported virito devices), and when the
> + * @rproc is booted.
> + *
> + * Returns the pointer to the resource table if it is found, and write its
> + * size into @tablesz. If a valid table isn't found, NULL is returned
> + * (and @tablesz isn't set).
> + */
> +static struct resource_table *
> +rproc_elf_find_rsc_table(struct rproc *rproc, const struct firmware *fw,
> +                                                     int *tablesz)
> +{
> +     struct elf32_hdr *ehdr;
> +     struct elf32_shdr *shdr;
> +
> +     struct device *dev = &rproc->dev;
> +     struct resource_table *table = NULL;
> +
> +     const u8 *elf_data = fw->data;
> +

Mind removing empty lines here?

> +     ehdr = (struct elf32_hdr *)elf_data;
> +
> +     shdr = find_rsc_shdr(dev, ehdr, fw->size);
> +     if (!shdr)
> +             return NULL;
> +
> +     /* make sure we have the entire table */
> +     if (shdr->sh_offset + shdr->sh_size > fw->size) {
> +             dev_err(dev, "resource table truncated\n");
> +             return NULL;
> +     }

Any reason for this explicit assertion? It seemed to be checked in
find_rsc_shdr.

> +
> +     table = (struct resource_table *)(elf_data + shdr->sh_offset);
> +     *tablesz = shdr->sh_size;
> +
>       return table;
>  }
>  
> -- 
> 1.7.5.4
> 

Thanks,
Ido.
--
To unsubscribe from this list: send the line "unsubscribe linux-kernel" in
the body of a message to majord...@vger.kernel.org
More majordomo info at  http://vger.kernel.org/majordomo-info.html
Please read the FAQ at  http://www.tux.org/lkml/

Reply via email to