From: Marek Szyprowski <m.szyprowski@samsung.com>
Date: Fri, 27 Nov 2020 15:10:34 +0000 (+0100)
Subject: thor: add support for dfu_alt_info reintialization from the downloaded script
X-Git-Tag: submit/tizen/20201208.071016~14
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=e7487bba0aed4a23994d357833bfbe039076dffe;p=platform%2Fkernel%2Fu-boot.git

thor: add support for dfu_alt_info reintialization from the downloaded script

Reinitialize dfu_env_entities after flashing the 'SCRIPT' entity to
ensure that the potential changes to the 'dfu_alt_info' environment
variable are applied.

Signed-off-by: Marek Szyprowski <m.szyprowski@samsung.com>
Change-Id: Iad98f348d25ee37fbb7493d4e567d1a4de55dc2c
---

diff --git a/cmd/thordown.c b/cmd/thordown.c
index a817e89a32..9c96cf45b6 100644
--- a/cmd/thordown.c
+++ b/cmd/thordown.c
@@ -75,13 +75,18 @@ int do_thor_down(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[])
 		goto exit;
 	}
 
-	ret = thor_handle();
-	if (ret) {
-		pr_err("THOR failed: %d\n", ret);
-		ret = CMD_RET_FAILURE;
-		goto exit;
-	}
-
+	do {
+		ret = thor_handle();
+		if (ret == THOR_DFU_REINIT_NEEDED) {
+			dfu_free_entities();
+			ret = dfu_init_env_entities(interface, devstring);
+		}
+		if (ret) {
+			pr_err("THOR failed: %d\n", ret);
+			ret = CMD_RET_FAILURE;
+			goto exit;
+		}
+	} while (ret == 0);
 exit:
 	g_dnl_unregister();
 	usb_gadget_release(controller_index);
diff --git a/drivers/usb/gadget/f_thor.c b/drivers/usb/gadget/f_thor.c
index 1368684a58..54c326144d 100644
--- a/drivers/usb/gadget/f_thor.c
+++ b/drivers/usb/gadget/f_thor.c
@@ -35,6 +35,7 @@
 #include <samsung/misc.h>
 #include <linux/input.h>
 #include <usb.h>
+#include <thor.h>
 
 #include "f_thor.h"
 
@@ -805,6 +806,8 @@ int thor_handle(void)
 			printf("%s: No data received!\n", __func__);
 			return ret;
 		}
+		if (dfu_reinit_needed)
+			return THOR_DFU_REINIT_NEEDED;
 	}
 
 	return 0;
diff --git a/include/thor.h b/include/thor.h
index 62501bda17..ee67ab0a27 100644
--- a/include/thor.h
+++ b/include/thor.h
@@ -12,6 +12,8 @@
 
 #include <linux/usb/composite.h>
 
+#define THOR_DFU_REINIT_NEEDED	0xFFFFFFFE
+
 int thor_handle(void);
 int thor_init(void);
 int thor_add(struct usb_configuration *c);