1 files changed, 35 insertions, 9 deletions
diff --git a/src/restore.h b/src/restore.h
index 644658a..9c11c34 100644
--- a/src/restore.h
+++ b/src/restore.h
@@ -1,4 +1,4 @@
-/*
+			/*
  * restore.h
  * Functions for handling idevices in restore mode
  *
@@ -19,18 +19,44 @@
  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
  */
 
-#ifndef RESTORED_H
-#define RESTORED_H
+#ifndef IDEVICERESTORE_RESTORE_H
+#define IDEVICERESTORE_RESTORE_H
 
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+#include <plist/plist.h>
 #include <libimobiledevice/restore.h>
+#include <libimobiledevice/libimobiledevice.h>
 
-#include "restore.h"
+struct restore_client_t {
+	plist_t tss;
+	idevice_t device;
+	const char* uuid;
+	unsigned int operation;
+	const char* filesystem;
+	restored_client_t client;
+};
 
-int restore_handle_progress_msg(restored_client_t client, plist_t msg);
-int restore_handle_status_msg(restored_client_t client, plist_t msg);
-int asr_send_system_image_data_from_file(idevice_t device, restored_client_t client, const char *filesystem);
-int restore_send_kernelcache(restored_client_t client, char *kernel_data, int len);
-int restore_send_nor_data(restored_client_t client, char* ipsw, plist_t tss);
+int restore_check_mode(const char* uuid);
+int restore_check_device(const char* uuid);
+int restore_client_new(struct idevicerestore_client_t* client);
+void restore_client_free(struct idevicerestore_client_t* client);
+int restore_reboot(struct idevicerestore_client_t* client);
 const char* restore_progress_string(unsigned int operation);
+int restore_handle_status_msg(restored_client_t client, plist_t msg);
+int restore_handle_progress_msg(restored_client_t client, plist_t msg);
+int restore_handle_data_request_msg(struct idevicerestore_client_t* client, idevice_t device, restored_client_t restore, plist_t message, plist_t build_identity, const char* filesystem);
+int restore_send_nor(restored_client_t restore, struct idevicerestore_client_t* client, plist_t build_identity);
+int restore_send_kernelcache(restored_client_t restore, struct idevicerestore_client_t* client, plist_t build_identity);
+int restore_device(struct idevicerestore_client_t* client, plist_t build_identity, const char* filesystem);
+int restore_open_with_timeout(struct idevicerestore_client_t* client);
+int restore_send_filesystem(idevice_t device, const char* filesystem);
+
+
+#ifdef __cplusplus
+}
+#endif
 
 #endif