afc_dispatch_packet optimization: remove additional buffer

This patch removes the additional buffer that was used inside
afc_dispatch_packet. So instead of 'alloc, copy header, copy data, send,
free' it will now simply do 'send header, send data'. This should reduce
cpu usage.

1 files changed, 29 insertions, 20 deletions

diff --git a/src/AFC.c b/src/AFC.c
index 0a24a15..fe6fc00 100644
--- a/src/AFC.c
+++ b/src/AFC.c
@@ -130,7 +130,7 @@ afc_error_t afc_client_free(afc_client_t client)
 static int afc_dispatch_packet(afc_client_t client, const char *data, uint64_t length)
 {
 	int bytes = 0, offset = 0;
-	char *buffer;
+	uint32_t sent = 0;
 
 	if (!client || !client->connection || !client->afc_packet)
 		return 0;
@@ -150,8 +150,6 @@ static int afc_dispatch_packet(afc_client_t client, const char *data, uint64_t l
 	// this_length is the parameters
 	// And everything beyond that is the next packet. (for writing)
 	if (client->afc_packet->this_length != client->afc_packet->entire_length) {
-		buffer = (char *) malloc(client->afc_packet->this_length);
-		memcpy(buffer, (char *) client->afc_packet, sizeof(AFCPacket));
 		offset = client->afc_packet->this_length - sizeof(AFCPacket);
 
 		log_debug_msg("%s: Offset: %i\n", __func__, offset);
@@ -160,39 +158,50 @@ static int afc_dispatch_packet(afc_client_t client, const char *data, uint64_t l
 			log_debug_msg("to based on the packet.\n");
 			log_debug_msg("%s: length minus offset: %i\n", __func__, length - offset);
 			log_debug_msg("%s: rest of packet: %i\n", __func__, client->afc_packet->entire_length - client->afc_packet->this_length);
-			free(buffer);
 			return -1;
 		}
-		memcpy(buffer + sizeof(AFCPacket), data, offset);
-		iphone_device_send(client->connection, buffer, client->afc_packet->this_length, (uint32_t*)&bytes);
-		free(buffer);
-		if (bytes <= 0) {
+
+		iphone_device_send(client->connection, (void*)client->afc_packet, sizeof(AFCPacket), &sent);
+		if (sent == 0) {
 			return bytes;
 		}
+		bytes += sent;
+
+		iphone_device_send(client->connection, data, offset, &sent);
+		if (sent == 0) {
+			return bytes;
+		}
+		bytes += sent;
 
 		log_debug_msg("%s: sent the first now go with the second\n", __func__);
 		log_debug_msg("%s: Length: %i\n", __func__, length - offset);
 		log_debug_msg("%s: Buffer: \n", __func__);
 		log_debug_buffer(data + offset, length - offset);
 
-		iphone_device_send(client->connection, data + offset, length - offset, (uint32_t*)&bytes);
+		sent = 0;
+		iphone_device_send(client->connection, data + offset, length - offset, &sent);
+
+		bytes = sent;
 		return bytes;
 	} else {
 		log_debug_msg("%s: doin things the old way\n", __func__);
-		buffer = (char *) malloc(sizeof(char) * client->afc_packet->this_length);
 		log_debug_msg("%s: packet length = %i\n", __func__, client->afc_packet->this_length);
-		memcpy(buffer, (char *) client->afc_packet, sizeof(AFCPacket));
-		log_debug_msg("%s: packet data follows\n", __func__);
-		if (length > 0) {
-			memcpy(buffer + sizeof(AFCPacket), data, length);
-		}
-		log_debug_buffer(buffer, client->afc_packet->this_length);
+
+		log_debug_buffer((char*)client->afc_packet, sizeof(AFCPacket));
 		log_debug_msg("\n");
-		iphone_device_send(client->connection, buffer, client->afc_packet->this_length, (uint32_t*)&bytes);
 
-		if (buffer) {
-			free(buffer);
-			buffer = NULL;
+		iphone_device_send(client->connection, (void*)client->afc_packet, sizeof(AFCPacket), &sent);
+		if (sent == 0) {
+			return bytes;
+		}
+		bytes += sent;
+		if (length > 0) {
+			log_debug_msg("%s: packet data follows\n", __func__);	
+
+			log_debug_buffer(data, length);
+			log_debug_msg("\n");
+			iphone_device_send(client->connection, data, length, &sent);
+			bytes += sent;
 		}
 		return bytes;
 	}