From 6d40d0ab626eb0ffee4f005b7fdc915bc561deb9 Mon Sep 17 00:00:00 2001
From: Nikias Bassen
Date: Thu, 4 Apr 2024 09:30:06 +0200
Subject: dfu: A little code optimization

---
 src/dfu.c | 20 +++++---------------
 1 file changed, 5 insertions(+), 15 deletions(-)

diff --git a/src/dfu.c b/src/dfu.c
index 62a3dc0..cc8e1fb 100644
--- a/src/dfu.c
+++ b/src/dfu.c
@@ -103,13 +103,13 @@ irecv_device_t dfu_get_irecv_device(struct idevicerestore_client_t* client)
 	return device;
 }
 
-int dfu_send_buffer(struct idevicerestore_client_t* client, unsigned char* buffer, unsigned int size)
+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, 1);
+	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;
@@ -118,19 +118,9 @@ 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)
+int dfu_send_buffer(struct idevicerestore_client_t* client, unsigned char* buffer, unsigned int size)
 {
-	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;
+	return dfu_send_buffer_with_options(client, buffer, size, IRECV_SEND_OPT_DFU_NOTIFY_FINISH);
 }
 
 int dfu_send_component(struct idevicerestore_client_t* client, plist_t build_identity, const char* component)
@@ -208,7 +198,7 @@ int dfu_send_component(struct idevicerestore_client_t* client, plist_t build_ide
 
 	info("Sending %s (%d bytes)...\n", component, size);
 
-	irecv_error_t err = irecv_send_buffer(client->dfu->client, data, size, 1);
+	irecv_error_t err = irecv_send_buffer(client->dfu->client, data, size, IRECV_SEND_OPT_DFU_NOTIFY_FINISH);
 	if (err != IRECV_E_SUCCESS) {
 		error("ERROR: Unable to send %s component: %s\n", component, irecv_strerror(err));
 		free(data);
-- 
cgit v1.1-32-gdbae