From e4a5ac4114177293e3a1b555ee767377b21d4432 Mon Sep 17 00:00:00 2001
From: Nikias Bassen
Date: Sat, 23 Mar 2024 04:25:21 +0100
Subject: Add support for Port DFU device restore

---
 src/dfu.c | 83 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 83 insertions(+)

(limited to 'src/dfu.c')

diff --git a/src/dfu.c b/src/dfu.c
index 8602bc0..62a3dc0 100644
--- a/src/dfu.c
+++ b/src/dfu.c
@@ -118,6 +118,21 @@ int dfu_send_buffer(struct idevicerestore_client_t* client, unsigned char* buffe
 	return 0;
 }
 
+int dfu_send_buffer_with_options(struct idevicerestore_client_t* client, unsigned char* buffer, unsigned int size, unsigned int irecv_options)
+{
+	irecv_error_t err = 0;
+
+	info("Sending data (%d bytes)...\n", size);
+
+	err = irecv_send_buffer(client->dfu->client, buffer, size, irecv_options);
+	if (err != IRECV_E_SUCCESS) {
+		error("ERROR: Unable to send data: %s\n", irecv_strerror(err));
+		return -1;
+	}
+
+	return 0;
+}
+
 int dfu_send_component(struct idevicerestore_client_t* client, plist_t build_identity, const char* component)
 {
 	char* path = NULL;
@@ -204,6 +219,24 @@ int dfu_send_component(struct idevicerestore_client_t* client, plist_t build_ide
 	return 0;
 }
 
+int dfu_get_bdid(struct idevicerestore_client_t* client, unsigned int* bdid)
+{
+	if(client->dfu == NULL) {
+		if (dfu_client_new(client) < 0) {
+			return -1;
+		}
+	}
+
+	const struct irecv_device_info *device_info = irecv_get_device_info(client->dfu->client);
+	if (!device_info) {
+		return -1;
+	}
+
+	*bdid = device_info->bdid;
+
+	return 0;
+}
+
 int dfu_get_cpid(struct idevicerestore_client_t* client, unsigned int* cpid)
 {
 	if(client->dfu == NULL) {
@@ -222,6 +255,27 @@ int dfu_get_cpid(struct idevicerestore_client_t* client, unsigned int* cpid)
 	return 0;
 }
 
+int dfu_get_prev(struct idevicerestore_client_t* client, unsigned int* prev)
+{
+	if(client->dfu == NULL) {
+		if (dfu_client_new(client) < 0) {
+			return -1;
+		}
+	}
+
+	const struct irecv_device_info *device_info = irecv_get_device_info(client->dfu->client);
+	if (!device_info) {
+		return -1;
+	}
+	char* ptr = strstr(device_info->serial_string, "PREV:");
+	if (ptr) {
+		sscanf(ptr, "PREV:%x", prev);
+		return 0;
+	}
+	return -1;
+}
+
+
 int dfu_is_image4_supported(struct idevicerestore_client_t* client)
 {
 	if(client->dfu == NULL) {
@@ -238,6 +292,35 @@ int dfu_is_image4_supported(struct idevicerestore_client_t* client)
 	return (device_info->ibfl & IBOOT_FLAG_IMAGE4_AWARE);
 }
 
+int dfu_get_portdfu_nonce(struct idevicerestore_client_t* client, unsigned char** nonce, unsigned int* nonce_size)
+{
+	if(client->dfu == NULL) {
+		if (dfu_client_new(client) < 0) {
+			return -1;
+		}
+	}
+
+	const struct irecv_device_info *device_info = irecv_get_device_info(client->dfu->client);
+	if (!device_info) {
+		return -1;
+	}
+
+	if (device_info->ap_nonce && device_info->ap_nonce_size > 0) {
+		*nonce = (unsigned char*)malloc(device_info->ap_nonce_size);
+		if (!*nonce) {
+			return -1;
+		}
+		*nonce_size = device_info->ap_nonce_size;
+		// The nonce is backwards, so we have to swap the bytes
+		unsigned int i = 0;
+		for (i = 0; i < *nonce_size; i++) {
+			(*nonce)[(*nonce_size)-1-i] = device_info->ap_nonce[i];
+		}
+	}
+
+	return 0;
+}
+
 int dfu_get_ap_nonce(struct idevicerestore_client_t* client, unsigned char** nonce, unsigned int* nonce_size)
 {
 	if(client->dfu == NULL) {
-- 
cgit v1.1-32-gdbae