diagnostics_relay: Implement query functions for MobileGestalt and IORegistry

1 files changed, 111 insertions, 0 deletions

diff --git a/src/diagnostics_relay.c b/src/diagnostics_relay.c
index f289fd9..72a5fd7 100644
--- a/src/diagnostics_relay.c
+++ b/src/diagnostics_relay.c
@@ -364,3 +364,114 @@ diagnostics_relay_error_t diagnostics_relay_request_diagnostics(diagnostics_rela
 	plist_free(dict);
 	return ret;
 }
+
+diagnostics_relay_error_t diagnostics_relay_query_mobilegestalt(diagnostics_relay_client_t client, plist_t keys, plist_t* result)
+{
+	if (!client || plist_get_node_type(keys) != PLIST_ARRAY || result == NULL)
+		return DIAGNOSTICS_RELAY_E_INVALID_ARG;
+
+	diagnostics_relay_error_t ret = DIAGNOSTICS_RELAY_E_UNKNOWN_ERROR;
+
+	plist_t dict = plist_new_dict();
+	plist_dict_insert_item(dict,"MobileGestaltKeys", plist_copy(keys));
+	plist_dict_insert_item(dict,"Request", plist_new_string("MobileGestalt"));
+	ret = diagnostics_relay_send(client, dict);
+	plist_free(dict);
+	dict = NULL;
+
+	ret = diagnostics_relay_receive(client, &dict);
+	if (!dict) {
+		return DIAGNOSTICS_RELAY_E_PLIST_ERROR;
+	}
+
+	if (diagnostics_relay_check_result(dict) == RESULT_SUCCESS) {
+		ret = DIAGNOSTICS_RELAY_E_SUCCESS;
+	}
+	if (ret != DIAGNOSTICS_RELAY_E_SUCCESS) {
+		plist_free(dict);
+		return ret;
+	}
+
+	plist_t value_node = plist_dict_get_item(dict, "Diagnostics");
+	if (value_node) {
+		*result = plist_copy(value_node);
+	}
+
+	plist_free(dict);
+	return ret;
+}
+
+diagnostics_relay_error_t diagnostics_relay_query_ioregistry_entry(diagnostics_relay_client_t client, const char* name, const char* class, plist_t* result)
+{
+	if (!client || (name == NULL && class == NULL) || result == NULL)
+		return DIAGNOSTICS_RELAY_E_INVALID_ARG;
+
+	diagnostics_relay_error_t ret = DIAGNOSTICS_RELAY_E_UNKNOWN_ERROR;
+
+	plist_t dict = plist_new_dict();
+	if (name)
+		plist_dict_insert_item(dict,"EntryName", plist_new_string(name));
+	if (class)
+		plist_dict_insert_item(dict,"EntryClass", plist_new_string(class));
+	plist_dict_insert_item(dict,"Request", plist_new_string("IORegistry"));
+	ret = diagnostics_relay_send(client, dict);
+	plist_free(dict);
+	dict = NULL;
+
+	ret = diagnostics_relay_receive(client, &dict);
+	if (!dict) {
+		return DIAGNOSTICS_RELAY_E_PLIST_ERROR;
+	}
+
+	if (diagnostics_relay_check_result(dict) == RESULT_SUCCESS) {
+		ret = DIAGNOSTICS_RELAY_E_SUCCESS;
+	}
+	if (ret != DIAGNOSTICS_RELAY_E_SUCCESS) {
+		plist_free(dict);
+		return ret;
+	}
+
+	plist_t value_node = plist_dict_get_item(dict, "Diagnostics");
+	if (value_node) {
+		*result = plist_copy(value_node);
+	}
+
+	plist_free(dict);
+	return ret;
+}
+
+diagnostics_relay_error_t diagnostics_relay_query_ioregistry_plane(diagnostics_relay_client_t client, const char* plane, plist_t* result)
+{
+	if (!client || plane == NULL || result == NULL)
+		return DIAGNOSTICS_RELAY_E_INVALID_ARG;
+
+	diagnostics_relay_error_t ret = DIAGNOSTICS_RELAY_E_UNKNOWN_ERROR;
+
+	plist_t dict = plist_new_dict();
+	plist_dict_insert_item(dict,"CurrentPlane", plist_new_string(plane));
+	plist_dict_insert_item(dict,"Request", plist_new_string("IORegistry"));
+	ret = diagnostics_relay_send(client, dict);
+	plist_free(dict);
+	dict = NULL;
+
+	ret = diagnostics_relay_receive(client, &dict);
+	if (!dict) {
+		return DIAGNOSTICS_RELAY_E_PLIST_ERROR;
+	}
+
+	if (diagnostics_relay_check_result(dict) == RESULT_SUCCESS) {
+		ret = DIAGNOSTICS_RELAY_E_SUCCESS;
+	}
+	if (ret != DIAGNOSTICS_RELAY_E_SUCCESS) {
+		plist_free(dict);
+		return ret;
+	}
+
+	plist_t value_node = plist_dict_get_item(dict, "Diagnostics");
+	if (value_node) {
+		*result = plist_copy(value_node);
+	}
+
+	plist_free(dict);
+	return ret;
+}