Make device mode specific code more consistent

1 files changed, 13 insertions, 6 deletions

diff --git a/src/common.h b/src/common.h
index 5d23299..3a4c069 100644
--- a/src/common.h
+++ b/src/common.h
@@ -41,12 +41,19 @@ extern "C" {
 #include "idevicerestore.h"
 #include "thread.h"
 
-#define MODE_UNKNOWN        -1
-#define MODE_WTF             0
-#define MODE_DFU             1
-#define MODE_RECOVERY        2
-#define MODE_RESTORE         3
-#define MODE_NORMAL          4
+#define _MODE_UNKNOWN         0
+#define _MODE_WTF             1
+#define _MODE_DFU             2
+#define _MODE_RECOVERY        3
+#define _MODE_RESTORE         4
+#define _MODE_NORMAL          5
+
+#define MODE_UNKNOWN  &idevicerestore_modes[_MODE_UNKNOWN]
+#define MODE_WTF      &idevicerestore_modes[_MODE_WTF]
+#define MODE_DFU      &idevicerestore_modes[_MODE_DFU]
+#define MODE_RECOVERY &idevicerestore_modes[_MODE_RECOVERY]
+#define MODE_RESTORE  &idevicerestore_modes[_MODE_RESTORE]
+#define MODE_NORMAL   &idevicerestore_modes[_MODE_NORMAL]
 
 #define FLAG_QUIT            1