From 40b7451502fcad7041521346d701f77c9dc2deaf Mon Sep 17 00:00:00 2001
From: 怯花颜
Date: Tue, 1 Mar 2022 09:08:13 +0800
Subject: diagnostics_relay: Check for errors after sending requests to service
 connection

---
 src/diagnostics_relay.c | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'src')

diff --git a/src/diagnostics_relay.c b/src/diagnostics_relay.c
index b6cf4d9..5444fcc 100644
--- a/src/diagnostics_relay.c
+++ b/src/diagnostics_relay.c
@@ -299,6 +299,9 @@ LIBIMOBILEDEVICE_API diagnostics_relay_error_t diagnostics_relay_request_diagnos
 	ret = diagnostics_relay_send(client, dict);
 	plist_free(dict);
 	dict = NULL;
+	if (ret != DIAGNOSTICS_RELAY_E_SUCCESS) {
+		return ret;
+	}
 
 	ret = diagnostics_relay_receive(client, &dict);
 	if (!dict) {
@@ -341,6 +344,9 @@ LIBIMOBILEDEVICE_API diagnostics_relay_error_t diagnostics_relay_query_mobileges
 	ret = diagnostics_relay_send(client, dict);
 	plist_free(dict);
 	dict = NULL;
+	if (ret != DIAGNOSTICS_RELAY_E_SUCCESS) {
+		return ret;
+	}
 
 	ret = diagnostics_relay_receive(client, &dict);
 	if (!dict) {
@@ -386,6 +392,9 @@ LIBIMOBILEDEVICE_API diagnostics_relay_error_t diagnostics_relay_query_ioregistr
 	ret = diagnostics_relay_send(client, dict);
 	plist_free(dict);
 	dict = NULL;
+	if (ret != DIAGNOSTICS_RELAY_E_SUCCESS) {
+		return ret;
+	}
 
 	ret = diagnostics_relay_receive(client, &dict);
 	if (!dict) {
-- 
cgit v1.1-32-gdbae