From feb987d33c322fb32a859e287271601ecb1a86a4 Mon Sep 17 00:00:00 2001
From: LukeZGD <26163116+LukeZGD@users.noreply.github.com>
Date: Sun, 3 Sep 2023 21:12:34 +0800
Subject: [PATCH] Set device_mode to DFU

---
 restore.sh | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/restore.sh b/restore.sh
index 75387bd..fb700b6 100755
--- a/restore.sh
+++ b/restore.sh
@@ -929,6 +929,7 @@ device_dfuhelper() {
     opt=$?
     if [[ $opt == 1 ]]; then
         log "Found device in DFU mode."
+        device_mode="DFU"
         return
     fi
     print "* Get ready..."
@@ -943,6 +944,7 @@ device_dfuhelper() {
         opt=$?
         if [[ $opt == 1 ]]; then
             echo -e "\n$(log 'Found device in DFU mode.')"
+            device_mode="DFU"
             return
         fi
         sleep 1
@@ -954,6 +956,7 @@ device_dfuhelper() {
         opt=$?
         if [[ $opt == 1 ]]; then
             echo -e "\n$(log 'Found device in DFU mode.')"
+            device_mode="DFU"
             return
         fi
         sleep 1