client: implemented ReadBUID, ReadPairRecord, SavePairRecord, and DeletePairRecord commonds

1 files changed, 107 insertions, 0 deletions

diff --git a/src/client.c b/src/client.c
index b81c11c..fdbea25 100644
--- a/src/client.c
+++ b/src/client.c
@@ -39,6 +39,7 @@ Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 #include "usb.h"
 #include "client.h"
 #include "device.h"
+#include "conf.h"
 
 #define CMD_BUF_SIZE	1024
 #define REPLY_BUF_SIZE	1024
@@ -285,6 +286,44 @@ static int send_device_list(struct mux_client *client, uint32_t tag)
 	return res;
 }
 
+static int send_system_buid(struct mux_client *client, uint32_t tag)
+{
+	int res = -1;
+	char* buid = NULL;
+
+	config_get_system_buid(&buid);
+
+	plist_t dict = plist_new_dict();
+	plist_dict_insert_item(dict, "BUID", plist_new_string(buid));
+	res = send_plist_pkt(client, tag, dict);
+	plist_free(dict);
+	return res;
+}
+
+static int send_pair_record(struct mux_client *client, uint32_t tag, const char* record_id)
+{
+	int res = -1;
+	char* record_data = NULL;
+	uint64_t record_size = 0;
+
+	if (!record_id) {
+		return send_result(client, tag, EINVAL);
+	}
+
+	config_get_device_record(record_id, &record_data, &record_size);
+	
+	if (record_data) {
+		plist_t dict = plist_new_dict();
+		plist_dict_insert_item(dict, "PairRecordData", plist_new_data(record_data, record_size));
+		free(record_data);
+		res = send_plist_pkt(client, tag, dict);
+		plist_free(dict);
+	} else {
+		res = send_result(client, tag, ENOENT);
+	}
+	return res;
+}
+
 static int notify_device_add(struct mux_client *client, struct device_info *dev)
 {
 	int res = -1;
@@ -355,6 +394,18 @@ static int start_listen(struct mux_client *client)
 	return count;
 }
 
+static char* plist_dict_get_string_val(plist_t dict, const char* key)
+{
+	if (!dict || plist_get_node_type(dict) != PLIST_DICT)
+		return NULL;
+	plist_t item = plist_dict_get_item(dict, key);
+	if (!item || plist_get_node_type(item) != PLIST_STRING)
+		return NULL;
+	char *str = NULL;
+	plist_get_string_val(item, &str);
+	return str;
+}
+
 static int client_command(struct mux_client *client, struct usbmuxd_header *hdr)
 {
 	int res;
@@ -450,6 +501,62 @@ static int client_command(struct mux_client *client, struct usbmuxd_header *hdr)
 					if (send_device_list(client, hdr->tag) < 0)
 						return -1;
 					return 0;
+				} else if (!strcmp(message, "ReadBUID")) {
+					if (send_system_buid(client, hdr->tag) < 0)
+						return -1;
+					return 0;
+				} else if (!strcmp(message, "ReadPairRecord")) {
+					free(message);
+					char* record_id = plist_dict_get_string_val(dict, "PairRecordID");
+					plist_free(dict);
+
+					res = send_pair_record(client, hdr->tag, record_id);
+					if (record_id)
+						free(record_id);
+					if (res < 0)
+						return -1;
+					return 0;
+				} else if (!strcmp(message, "SavePairRecord")) {
+					uint32_t rval = RESULT_OK;
+					free(message);
+					char* record_id = plist_dict_get_string_val(dict, "PairRecordID");
+					char* record_data = NULL;
+					uint64_t record_size = 0;
+					plist_t rdata = plist_dict_get_item(dict, "PairRecordData");
+					if (rdata && plist_get_node_type(rdata) == PLIST_DATA) {
+						plist_get_data_val(rdata, &record_data, &record_size);
+					}
+					plist_free(dict);
+
+					if (record_id && record_data) {
+						res = config_set_device_record(record_id, record_data, record_size);
+						if (res < 0) {
+							rval = -res;
+						}
+						free(record_id);
+					} else {
+						rval = EINVAL;
+					}
+					if (send_result(client, hdr->tag, rval) < 0)
+						return -1;
+					return 0;
+				} else if (!strcmp(message, "DeletePairRecord")) {
+					uint32_t rval = RESULT_OK;
+					free(message);
+					char* record_id = plist_dict_get_string_val(dict, "PairRecordID");
+					plist_free(dict);
+					if (record_id) {
+						res = config_remove_device_record(record_id);
+						if (res < 0) {
+							rval = -res;
+						}
+						free(record_id);
+					} else {
+						rval = EINVAL;
+					}
+					if (send_result(client, hdr->tag, rval) < 0)
+						return -1;
+					return 0;
 				} else {
 					usbmuxd_log(LL_ERROR, "Unexpected command '%s' received!", message);
 					free(message);