Merge branch 'master' of git://github.com/MattColyer/libiphone into martin

1 files changed, 22 insertions, 53 deletions

diff --git a/src/lockdown.c b/src/lockdown.c
index 6f9c922..80a89dc 100644
--- a/src/lockdown.c
+++ b/src/lockdown.c
@@ -60,11 +60,7 @@ static int lockdown_check_result(plist_t dict, const char *query_match)
 {
 	int ret = -1;
 
-	plist_t query_key = plist_find_node_by_key(dict, "Request");
-	if (!query_key) {
-		return ret;
-	}
-	plist_t query_node = plist_get_next_sibling(query_key);
+	plist_t query_node = plist_dict_get_item(dict, "Request");
 	if (!query_node) {
 		return ret;
 	}
@@ -83,40 +79,30 @@ static int lockdown_check_result(plist_t dict, const char *query_match)
 		free(query_value);
 	}
 
-	plist_t result_node = plist_get_next_sibling(query_node);
+	plist_t result_node = plist_dict_get_item(dict, "Result");
 	if (!result_node) {
 		return ret;
 	}
 
-	plist_t value_node = plist_get_next_sibling(result_node);
-	if (!value_node) {
-		return ret;
-	}
-
 	plist_type result_type = plist_get_node_type(result_node);
-	plist_type value_type = plist_get_node_type(value_node);
 
-	if (result_type == PLIST_KEY && value_type == PLIST_STRING) {
+	if (result_type == PLIST_STRING) {
 
 		char *result_value = NULL;
-		char *value_value = NULL;
 
-		plist_get_key_val(result_node, &result_value);
-		plist_get_string_val(value_node, &value_value);
+		plist_get_string_val(result_node, &result_value);
 
-		if (result_value && value_value && !strcmp(result_value, "Result")) {
-			if (!strcmp(value_value, "Success")) {
+		if (result_value) {
+			if (!strcmp(result_value, "Success")) {
 				ret = RESULT_SUCCESS;
-			} else if (!strcmp(value_value, "Failure")) {
+			} else if (!strcmp(result_value, "Failure")) {
 				ret = RESULT_FAILURE;
 			} else {
-				log_dbg_msg(DBGMASK_LOCKDOWND, "%s: ERROR: unknown result value '%s'\n", __func__, value_value);
+				log_dbg_msg(DBGMASK_LOCKDOWND, "%s: ERROR: unknown result value '%s'\n", __func__, result_value);
 			}
 		}
 		if (result_value)
 			free(result_value);
-		if (value_value)
-			free(value_value);
 	}
 	return ret;
 }
@@ -437,20 +423,11 @@ lockdownd_error_t lockdownd_get_value(lockdownd_client_t client, const char *dom
 		return ret;
 	}
 
-	plist_t value_key_node = plist_find_node_by_key(dict, "Value");
-	plist_t value_value_node = plist_get_next_sibling(value_key_node);
-
-	plist_type value_key_type = plist_get_node_type(value_key_node);
+	plist_t value_node = plist_dict_get_item(dict, "Value");
 
-	if (value_key_type == PLIST_KEY) {
-		char *result_key = NULL;
-		plist_get_key_val(value_key_node, &result_key);
-
-		if (!strcmp(result_key, "Value")) {
-			log_dbg_msg(DBGMASK_LOCKDOWND, "%s: has a value\n", __func__);
-			*value = plist_copy(value_value_node);
-		}
-		free(result_key);
+	if (value_node) {
+		log_dbg_msg(DBGMASK_LOCKDOWND, "%s: has a value\n", __func__);
+		*value = plist_copy(value_node);
 	}
 
 	plist_free(dict);
@@ -1024,7 +1001,7 @@ lockdownd_error_t lockdownd_start_ssl_session(lockdownd_client_t client, const c
 		return LOCKDOWN_E_PLIST_ERROR;
 
 	if (lockdown_check_result(dict, "StartSession") == RESULT_FAILURE) {
-		plist_t error_node = plist_get_dict_el_from_key(dict, "Error");
+		plist_t error_node = plist_dict_get_item(dict, "Error");
 		if (error_node && PLIST_STRING == plist_get_node_type(error_node)) {
 			char *error = NULL;
 			plist_get_string_val(error_node, &error);
@@ -1108,18 +1085,17 @@ lockdownd_error_t lockdownd_start_ssl_session(lockdownd_client_t client, const c
 		}
 	}
 	/* store session id */
-	plist_t session_node = plist_find_node_by_key(dict, "SessionID");
+	plist_t session_node = plist_dict_get_item(dict, "SessionID");
 	if (session_node) {
 
-		plist_t session_node_val = plist_get_next_sibling(session_node);
-		plist_type session_node_val_type = plist_get_node_type(session_node_val);
+		plist_type session_node_type = plist_get_node_type(session_node);
 
-		if (session_node_val_type == PLIST_STRING) {
+		if (session_node_type == PLIST_STRING) {
 
 			char *session_id = NULL;
-			plist_get_string_val(session_node_val, &session_id);
+			plist_get_string_val(session_node, &session_id);
 
-			if (session_node_val_type == PLIST_STRING && session_id) {
+			if (session_node_type == PLIST_STRING && session_id) {
 				/* we need to store the session ID for StopSession */
 				strcpy(client->session_id, session_id);
 				log_dbg_msg(DBGMASK_LOCKDOWND, "%s: SessionID: %s\n", __func__, client->session_id);
@@ -1260,23 +1236,16 @@ lockdownd_error_t lockdownd_start_service(lockdownd_client_t client, const char
 
 	ret = LOCKDOWN_E_UNKNOWN_ERROR;
 	if (lockdown_check_result(dict, "StartService") == RESULT_SUCCESS) {
-		plist_t port_key_node = plist_find_node_by_key(dict, "Port");
-		plist_t port_value_node = plist_get_next_sibling(port_key_node);
+		plist_t port_value_node = plist_dict_get_item(dict, "Port");
 
-		if ((plist_get_node_type(port_key_node) == PLIST_KEY)
-			&& (plist_get_node_type(port_value_node) == PLIST_UINT)) {
-			char *port_key = NULL;
+		if (port_value_node && (plist_get_node_type(port_value_node) == PLIST_UINT)) {
 			uint64_t port_value = 0;
-
-			plist_get_key_val(port_key_node, &port_key);
 			plist_get_uint_val(port_value_node, &port_value);
-			if (port_key && !strcmp(port_key, "Port")) {
+
+			if (port_value) {
 				port_loc = port_value;
 				ret = LOCKDOWN_E_SUCCESS;
 			}
-			if (port_key)
-				free(port_key);
-
 			if (port && ret == LOCKDOWN_E_SUCCESS)
 				*port = port_loc;
 		}