Enable the videoconsole during the stm32prog command execution
to have information without UART.

Signed-off-by: Patrick Delaunay <[email protected]>
---

 .../cmd_stm32prog/cmd_stm32prog.c             | 28 +++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/arch/arm/mach-stm32mp/cmd_stm32prog/cmd_stm32prog.c 
b/arch/arm/mach-stm32mp/cmd_stm32prog/cmd_stm32prog.c
index 1769ba05f2..15bbdc2cb6 100644
--- a/arch/arm/mach-stm32mp/cmd_stm32prog/cmd_stm32prog.c
+++ b/arch/arm/mach-stm32mp/cmd_stm32prog/cmd_stm32prog.c
@@ -11,6 +11,32 @@
 
 struct stm32prog_data *stm32prog_data;
 
+static void enable_vidconsole(void)
+{
+#ifdef CONFIG_DM_VIDEO
+       char *stdname;
+       char buf[64];
+
+       stdname = env_get("stdout");
+       if (!stdname || !strstr(stdname, "vidconsole")) {
+               if (!stdname)
+                       snprintf(buf, sizeof(buf), "serial,vidconsole");
+               else
+                       snprintf(buf, sizeof(buf), "%s,vidconsole", stdname);
+               env_set("stdout", buf);
+       }
+
+       stdname = env_get("stderr");
+       if (!stdname || !strstr(stdname, "vidconsole")) {
+               if (!stdname)
+                       snprintf(buf, sizeof(buf), "serial,vidconsole");
+               else
+                       snprintf(buf, sizeof(buf), "%s,vidconsole", stdname);
+               env_set("stderr", buf);
+       }
+#endif
+}
+
 static int do_stm32prog(cmd_tbl_t *cmdtp, int flag, int argc,
                        char * const argv[])
 {
@@ -45,6 +71,8 @@ static int do_stm32prog(cmd_tbl_t *cmdtp, int flag, int argc,
        if (argc > 4)
                size = simple_strtoul(argv[4], NULL, 16);
 
+       enable_vidconsole();
+
        data = (struct stm32prog_data *)malloc(sizeof(*data));
 
        if (!data) {
-- 
2.17.1

Reply via email to