[PATCH] fastboot: add possibility to register OEM commands

[Ahmad Fatoum]

Hello Thomas,

On 13.05.24 15:49, Thomas Haemmerle wrote:
> Currently it is not possible to add additional OEM commands without
> extending the array `cmd_oem_dispatch_info[]`.
> Add the possibility to register commands from e.g. board files by
> replacing the array with a list.

struct fastboot_opts has cmd_exec and cmd_flash callback for exactly the
purpose of board code handling custom OEM and flash commands.

To make use of this, you'd probably want to call usb_multi_register()
from board code with a suitable argument instead of having it done by
usbgadget command or autostart.

Would that work for you?

Cheers,
Ahmad

> 
> Signed-off-by: Thomas Haemmerle <thomas.haemmerle at leica-geosystems.com>
> ---
>  common/fastboot.c  | 53 +++++++++++++++++++++++++++++++++++-----------
>  include/fastboot.h |  9 ++++++++
>  2 files changed, 50 insertions(+), 12 deletions(-)
> 
> diff --git a/common/fastboot.c b/common/fastboot.c
> index f8a01dea7a..8f78b0e517 100644
> --- a/common/fastboot.c
> +++ b/common/fastboot.c
> @@ -885,24 +885,41 @@ static void cb_oem_exec(struct fastboot *fb, const char *cmd)
>  		fastboot_tx_print(fb, FASTBOOT_MSG_OKAY, "");
>  }
>  
> -static const struct cmd_dispatch_info cmd_oem_dispatch_info[] = {
> -	{
> -		.cmd = "getenv ",
> -		.cb = cb_oem_getenv,
> -	}, {
> -		.cmd = "setenv ",
> -		.cb = cb_oem_setenv,
> -	}, {
> -		.cmd = "exec ",
> -		.cb = cb_oem_exec,
> -	},
> +struct oem_command {
> +	char *cmd;
> +	void (*cb)(struct fastboot *fb, const char *opt);
> +	struct list_head list;
>  };
>  
> +LIST_HEAD(oem_cmd_list);
> +
> +int fastboot_register_oem_command(char *cmd, void (*cb)(struct fastboot *fb, const char *opt))
> +{
> +	struct oem_command *oem_cmd = xzalloc(sizeof(*oem_cmd));
> +
> +	oem_cmd->cmd = xstrdup(cmd);
> +	oem_cmd->cb = cb;
> +
> +	list_add_tail(&oem_cmd->list, &oem_cmd_list);
> +
> +	return 0;
> +}
> +
>  static void __maybe_unused cb_oem(struct fastboot *fb, const char *cmd)
>  {
> +	struct oem_command *oem_cmd;
> +
>  	pr_debug("%s: \"%s\"\n", __func__, cmd);
>  
> -	fb_run_command(fb, cmd, cmd_oem_dispatch_info, ARRAY_SIZE(cmd_oem_dispatch_info));
> +	list_for_each_entry(oem_cmd, &oem_cmd_list, list) {
> +		if (!strcmp_l1(oem_cmd->cmd, cmd)) {
> +			oem_cmd->cb(fb, cmd + strlen(oem_cmd->cmd));
> +			return;
> +		}
> +	}
> +
> +	fastboot_tx_print(fb, FASTBOOT_MSG_FAIL, "unknown OEM command %s",
> +			  cmd);
>  }
>  
>  static const struct cmd_dispatch_info cmd_dispatch_info[] = {
> @@ -969,6 +986,18 @@ struct file_list *get_fastboot_partitions(void)
>  	return system_partitions_get_null();
>  }
>  
> +static int fastboot_oem_cmds_init(void)
> +{
> +	if (!IS_ENABLED(CONFIG_FASTBOOT_CMD_OEM))
> +		return 0;
> +
> +	fastboot_register_oem_command("getenv ", cb_oem_getenv);
> +	fastboot_register_oem_command("setenv ", cb_oem_setenv);
> +	fastboot_register_oem_command("exec ", cb_oem_exec);
> +	return 0;
> +}
> +device_initcall(fastboot_oem_cmds_init);
> +
>  static int fastboot_globalvars_init(void)
>  {
>  	if (IS_ENABLED(CONFIG_FASTBOOT_SPARSE)) {
> diff --git a/include/fastboot.h b/include/fastboot.h
> index cd415847e3..38b0728a89 100644
> --- a/include/fastboot.h
> +++ b/include/fastboot.h
> @@ -78,6 +78,15 @@ static inline struct file_list *get_fastboot_partitions(void)
>  }
>  #endif
>  
> +#ifdef CONFIG_FASTBOOT_CMD_OEM
> +int fastboot_register_oem_command(char *cmd, void (*cb)(struct fastboot *fb, const char *opt));
> +#else
> +static inline int fastboot_register_oem_command(char *cmd, void (*cb)(struct fastboot *fb, const char *opt))
> +{
> +	return -ENOSYS;
> +}
> +#endif
> +
>  int fastboot_generic_init(struct fastboot *fb, bool export_bbu);
>  void fastboot_generic_close(struct fastboot *fb);
>  void fastboot_generic_free(struct fastboot *fb);
> 
> base-commit: e70a514bce4bc84429c46fa378622024ac6c6863

-- 
Pengutronix e.K.                           |                             |
Steuerwalder Str. 21                       | http://www.pengutronix.de/  |
31137 Hildesheim, Germany                  | Phone: +49-5121-206917-0    |
Amtsgericht Hildesheim, HRA 2686           | Fax:   +49-5121-206917-5555 |