[U-Boot] [PATCH v3 1/5] mtd: Use default mtdparts/mtids when not defined in the environment

[Marek Vasut]

On 11/13/2018 12:43 PM, Boris Brezillon wrote:
> U-boot provides a mean to define default values for mtdids and mtdparts
> when they're not defined in the environment. Patch mtd_probe_devices()
> to use those default values when env_get("mtdparts") or
> env_get("mtdids") return NULL.
> 
> This implementation is based on the logic found in cmd/mtdparts.c.
> 
> Fixes: 5db66b3aee6f ("cmd: mtd: add 'mtd' command")
> Reported-by: Stefan Roese <sr at denx.de>
> Signed-off-by: Boris Brezillon <boris.brezillon at bootlin.com>
> Tested-by: Stefan Roese <sr at denx.de>
> Reviewed-by: Lukasz Majewski <lukma at denx.de>
> ---
> Changes in v3:
> - Fix env_get_f() call
> - Add Lukasz R-b
> 
> Changes in v2:
> - none
> ---
>  drivers/mtd/mtd_uboot.c | 62 +++++++++++++++++++++++++++++++++++++++--
>  1 file changed, 60 insertions(+), 2 deletions(-)
> 
> diff --git a/drivers/mtd/mtd_uboot.c b/drivers/mtd/mtd_uboot.c
> index 7d7a11c990d6..5ca560c96879 100644
> --- a/drivers/mtd/mtd_uboot.c
> +++ b/drivers/mtd/mtd_uboot.c
> @@ -92,12 +92,70 @@ static void mtd_probe_uclass_mtd_devs(void) { }
>  #endif
>  
>  #if defined(CONFIG_MTD_PARTITIONS)
> +extern void board_mtdparts_default(const char **mtdids,
> +				   const char **mtdparts);

Why is there this extern ? This should use a prototype of function
defined in a header file. Once someone changes the prototype and forgets
to update this location, it'll be a disaster.

> +static const char *get_mtdids(void)
> +{
> +	__maybe_unused const char *mtdparts = NULL;
> +	const char *mtdids = env_get("mtdids");
> +
> +	if (mtdids)
> +		return mtdids;
> +
> +#if defined(CONFIG_SYS_MTDPARTS_RUNTIME)
> +	board_mtdparts_default(&mtdids, &mtdparts);
> +#elif defined(MTDIDS_DEFAULT)
> +	mtdids = MTDIDS_DEFAULT;
> +#elif defined(CONFIG_MTDIDS_DEFAULT)
> +	mtdids = CONFIG_MTDIDS_DEFAULT;
> +#endif
> +
> +	if (mtdids)
> +		env_set("mtdids", mtdids);
> +
> +	return mtdids;
> +}
> +
> +#define MTDPARTS_MAXLEN         512
> +
> +static const char *get_mtdparts(void)
> +{
> +	__maybe_unused const char *mtdids = NULL;
> +	static char tmp_parts[MTDPARTS_MAXLEN];
> +	static bool use_defaults = true;
> +	const char *mtdparts = NULL;
> +
> +	if (gd->flags & GD_FLG_ENV_READY)
> +		mtdparts = env_get("mtdparts");
> +	else if (env_get_f("mtdparts", tmp_parts, sizeof(tmp_parts)) != -1)
> +		mtdparts = tmp_parts;
> +
> +	if (mtdparts || !use_defaults)
> +		return mtdparts;
> +
> +#if defined(CONFIG_SYS_MTDPARTS_RUNTIME)
> +	board_mtdparts_default(&mtdids, &mtdparts);
> +#elif defined(MTDPARTS_DEFAULT)
> +	mtdparts = MTDPARTS_DEFAULT;
> +#elif defined(CONFIG_MTDPARTS_DEFAULT)
> +	mtdparts = CONFIG_MTDPARTS_DEFAULT;
> +#endif
> +
> +	if (mtdparts)
> +		env_set("mtdparts", mtdparts);
> +
> +	use_defaults = false;
> +
> +	return mtdparts;
> +}
> +
>  int mtd_probe_devices(void)
>  {
>  	static char *old_mtdparts;
>  	static char *old_mtdids;
> -	const char *mtdparts = env_get("mtdparts");
> -	const char *mtdids = env_get("mtdids");
> +	const char *mtdparts = get_mtdparts();
> +	const char *mtdids = get_mtdids();
>  	bool remaining_partitions = true;
>  	struct mtd_info *mtd;
>  
> 


-- 
Best regards,
Marek Vasut