From d33402036d563667a76910dbfec50a37b622d3a5 Mon Sep 17 00:00:00 2001
From: Nikias Bassen
Date: Sat, 15 Sep 2018 03:46:34 +0200
Subject: client: Send 'Paired' message when a device's pairing record has been
 stored successfully

This requires the SavePairRecord message sent from the client to contain a
usbmux device id so the device can be matched accordingly.
For the record: This is the original behavior.
---
 src/client.c        | 58 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 src/client.h        |  1 +
 src/usbmuxd-proto.h |  2 +-
 3 files changed, 60 insertions(+), 1 deletion(-)

diff --git a/src/client.c b/src/client.c
index a24233c..c566d8c 100644
--- a/src/client.c
+++ b/src/client.c
@@ -524,6 +524,24 @@ static int notify_device_remove(struct mux_client *client, uint32_t device_id)
 	return res;
 }
 
+static int notify_device_paired(struct mux_client *client, uint32_t device_id)
+{
+	int res = -1;
+	if (client->proto_version == 1) {
+		/* XML plist packet */
+		plist_t dict = plist_new_dict();
+		plist_dict_set_item(dict, "MessageType", plist_new_string("Paired"));
+		plist_dict_set_item(dict, "DeviceID", plist_new_uint(device_id));
+		res = send_plist_pkt(client, 0, dict);
+		plist_free(dict);
+	}
+	else {
+		/* binary packet */
+		res = send_pkt(client, 0, MESSAGE_DEVICE_PAIRED, &device_id, sizeof(uint32_t));
+	}
+	return res;
+}
+
 static int start_listen(struct mux_client *client)
 {
 	struct device_info *devs = NULL;
@@ -738,6 +756,34 @@ static int client_command(struct mux_client *client, struct usbmuxd_header *hdr)
 						res = config_set_device_record(record_id, record_data, record_size);
 						if (res < 0) {
 							rval = -res;
+						} else {
+							plist_t p_dev_id = plist_dict_get_item(dict, "DeviceID");
+							uint32_t dev_id = 0;
+							if (p_dev_id && plist_get_node_type(p_dev_id) == PLIST_UINT) {
+								uint64_t u_dev_id = 0;
+								plist_get_uint_val(p_dev_id, &u_dev_id);
+								dev_id = (uint32_t)u_dev_id;
+							}
+							if (dev_id > 0) {
+								struct device_info *devs = NULL;
+								struct device_info *dev;
+								int i;
+								int count = device_get_list(1, &devs);
+								int found = 0;
+								dev = devs;
+								for (i = 0; devs && i < count; i++, dev++) {
+									if ((uint32_t)dev->id == dev_id && (strcmp(dev->serial, record_id) == 0)) {
+										found++;
+										break;
+									}
+								}
+								if (!found) {
+									usbmuxd_log(LL_ERROR, "ERROR: SavePairRecord: DeviceID %d (%s) is not connected\n", dev_id, record_id);
+								} else {
+									client_device_paired(dev_id);
+								}
+								free(devs);
+							}
 						}
 						free(record_id);
 					} else {
@@ -937,6 +983,18 @@ void client_device_remove(int device_id)
 	pthread_mutex_unlock(&client_list_mutex);
 }
 
+void client_device_paired(int device_id)
+{
+	pthread_mutex_lock(&client_list_mutex);
+	uint32_t id = device_id;
+	usbmuxd_log(LL_DEBUG, "client_device_paired: id %d", device_id);
+	FOREACH(struct mux_client *client, &client_list) {
+		if (client->state == CLIENT_LISTEN)
+			notify_device_paired(client, id);
+	} ENDFOREACH
+	pthread_mutex_unlock(&client_list_mutex);
+}
+
 void client_init(void)
 {
 	usbmuxd_log(LL_DEBUG, "client_init");
diff --git a/src/client.h b/src/client.h
index bac563d..6cac4db 100644
--- a/src/client.h
+++ b/src/client.h
@@ -35,6 +35,7 @@ int client_notify_connect(struct mux_client *client, enum usbmuxd_result result)
 
 void client_device_add(struct device_info *dev);
 void client_device_remove(int device_id);
+void client_device_paired(int device_id);
 
 int client_accept(int fd);
 void client_get_fds(struct fdlist *list);
diff --git a/src/usbmuxd-proto.h b/src/usbmuxd-proto.h
index a344e60..9416416 100644
--- a/src/usbmuxd-proto.h
+++ b/src/usbmuxd-proto.h
@@ -52,7 +52,7 @@ enum usbmuxd_msgtype {
 	MESSAGE_LISTEN = 3,
 	MESSAGE_DEVICE_ADD = 4,
 	MESSAGE_DEVICE_REMOVE = 5,
-	//???
+	MESSAGE_DEVICE_PAIRED = 6,
 	//???
 	MESSAGE_PLIST = 8,
 };
-- 
cgit v1.1-32-gdbae