1 files changed, 20 insertions, 4 deletions
diff --git a/src/dfu.h b/src/dfu.h
index 3577888..0e20448 100644
--- a/src/dfu.h
+++ b/src/dfu.h
@@ -22,10 +22,26 @@
 #ifndef IDEVICERESTORE_DFU_H
 #define IDEVICERESTORE_DFU_H
 
-#include <stdint.h>
-#include <plist/plist.h>
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+#include <libirecovery.h>
+#include "common.h"
+
+struct dfu_client_t {
+	irecv_client_t client;
+	const char* ipsw;
+	plist_t tss;
+};
 
-int dfu_check_mode();
-int dfu_enter_recovery(const char* ipsw, plist_t tss);
+int dfu_client_new(struct idevicerestore_client_t* client, uint32_t timeout);
+void dfu_client_free(struct idevicerestore_client_t* client);
+int dfu_enter_recovery(struct idevicerestore_client_t* client);
+
+
+#ifdef __cplusplus
+}
+#endif
 
 #endif