On Mon, Aug 20, 2018 at 11:25:45PM -0700, Andrey Smirnov wrote:
> Since imx_bbu_check_prereq() already uses file_detect_type() and we've
> extended it to understand i.MX boot image file type, we can simplify a
> bunch of repetitive code as follows:
> 
>     1. Convert all checks from IVT_BARKER to filetype_imx_image_v2
>        check
> 
>     2. Move all of the checking to be a part of imx_bbu_check_prereq()
> 
> Signed-off-by: Andrey Smirnov <andrew.smir...@gmail.com>
> ---
>  arch/arm/mach-imx/imx-bbu-internal.c | 64 +++++++++++++++++-----------
>  1 file changed, 39 insertions(+), 25 deletions(-)
> 
> diff --git a/arch/arm/mach-imx/imx-bbu-internal.c 
> b/arch/arm/mach-imx/imx-bbu-internal.c
> index 67ae2961c..ea57b2772 100644
> --- a/arch/arm/mach-imx/imx-bbu-internal.c
> +++ b/arch/arm/mach-imx/imx-bbu-internal.c
> @@ -106,11 +106,39 @@ err_close:
>       return ret;
>  }
>  
> -static int imx_bbu_check_prereq(const char *devicefile, struct bbu_data 
> *data)
> +static int imx_bbu_check_prereq(struct imx_internal_bbu_handler *imx_handler,
> +                             const char *devicefile, struct bbu_data *data,
> +                             enum filetype expected_type)
>  {
>       int ret;
> -
> -     if (file_detect_type(data->image, data->len) != filetype_arm_barebox) {
> +     const void *blob;
> +     size_t len;
> +     enum filetype type;
> +
> +     type = file_detect_type(data->image, data->len);
> +
> +     switch (type) {
> +     case filetype_arm_barebox:
> +             /*
> +              * Specifying expected_type as unknow will disable the
> +              * inner image type check
> +              */
> +             if (expected_type == filetype_unknown)
> +                     break;
> +
> +             blob = data->image + imx_handler->flash_header_offset;
> +             len  = data->len   - imx_handler->flash_header_offset;
> +             type = file_detect_type(blob, len);
> +
> +             if (type != expected_type) {
> +                     pr_err("Expected image type: %s, "
> +                            "detected image type: %s\n",
> +                            file_type_to_string(expected_type),
> +                            file_type_to_string(type));
> +                     return -EINVAL;
> +             }
> +             break;
> +     default:
>               if (!bbu_force(data, "Not an ARM barebox image"))
>                       return -EINVAL;
>       }
> @@ -137,7 +165,8 @@ static int imx_bbu_internal_v1_update(struct bbu_handler 
> *handler, struct bbu_da
>               container_of(handler, struct imx_internal_bbu_handler, handler);
>       int ret;
>  
> -     ret = imx_bbu_check_prereq(data->devicefile, data);
> +     ret = imx_bbu_check_prereq(imx_handler, data->devicefile, data,
> +                                filetype_unknown);

Why filetype_unknown here? in the v2 version we have
filetype_imx_image_v2. I would expect filetype_imx_image_v1 here.

Sascha

-- 
Pengutronix e.K.                           |                             |
Industrial Linux Solutions                 | http://www.pengutronix.de/  |
Peiner Str. 6-8, 31137 Hildesheim, Germany | Phone: +49-5121-206917-0    |
Amtsgericht Hildesheim, HRA 2686           | Fax:   +49-5121-206917-5555 |

_______________________________________________
barebox mailing list
barebox@lists.infradead.org
http://lists.infradead.org/mailman/listinfo/barebox

Reply via email to