From 551b01a5a6cf75c9e76b86ea085393cf5d58aff9 Mon Sep 17 00:00:00 2001
From: Gregor Haas <gregorhaas1997@gmail.com>
Date: Sat, 7 Dec 2019 14:53:39 -0500
Subject: [PATCH] Don't re-exploit device

---
 .gitignore        | 1 +
 exploit/exploit.c | 5 ++++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/.gitignore b/.gitignore
index ebeeab3..6cbb503 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1 +1,2 @@
+.idea/*
 cmake-build-debug/*
diff --git a/exploit/exploit.c b/exploit/exploit.c
index 42762d3..6d8fbff 100644
--- a/exploit/exploit.c
+++ b/exploit/exploit.c
@@ -108,7 +108,10 @@ int check_function(struct libusb_device_bundle *bundle)
 
 int exploit_device()
 {
-    int ret = complete_stage(stage1_function);
+    int ret = complete_stage(check_function);
+    if(ret == 0) return ret;
+
+    ret = complete_stage(stage1_function);
     if(ret == 0)
     {
         ret = complete_stage(stage2_function);