[PATCH] lib: sbi: Hang when next mode not support

[Xiang W]

The original code may let the hart that does not support next mode
enter init_warmboot. Fix this.

Signed-off-by: Xiang W <wxjstz at 126.com>
---
 lib/sbi/sbi_init.c | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/lib/sbi/sbi_init.c b/lib/sbi/sbi_init.c
index 423e6d8..d148bd2 100644
--- a/lib/sbi/sbi_init.c
+++ b/lib/sbi/sbi_init.c
@@ -532,10 +532,11 @@ void __noreturn sbi_init(struct sbi_scratch *scratch)
 		if (misa_extension('U'))
 			next_mode_supported = true;
 		break;
-	default:
-		sbi_hart_hang();
 	}
 
+	if (next_mode_supported == false)
+		sbi_hart_hang();
+
 	/*
 	 * Only the HART supporting privilege mode specified in the
 	 * scratch->next_mode should be allowed to become the coldboot
@@ -547,8 +548,7 @@ void __noreturn sbi_init(struct sbi_scratch *scratch)
 	 */
 
 	if (sbi_platform_cold_boot_allowed(plat, hartid)) {
-		if (next_mode_supported &&
-		    atomic_xchg(&coldboot_lottery, 1) == 0)
+		if (atomic_xchg(&coldboot_lottery, 1) == 0)
 			coldboot = true;
 	}
 
-- 
2.39.2