lockdown: Only query ProductVersion and DeviceClass if we are talking to lockdownd

In restore mode these requests fail so we need to make sure we are
actually talking to lockdownd

1 files changed, 7 insertions, 4 deletions

diff --git a/src/lockdown.c b/src/lockdown.c
index 1edb99b..32389c9 100644
--- a/src/lockdown.c
+++ b/src/lockdown.c
@@ -643,16 +643,19 @@ lockdownd_error_t lockdownd_client_new(idevice_t device, lockdownd_client_t *cli
 
 	client_loc->label = label ? strdup(label) : NULL;
 
+	int is_lockdownd = 0;
 	if (lockdownd_query_type(client_loc, &type) != LOCKDOWN_E_SUCCESS) {
 		debug_info("QueryType failed in the lockdownd client.");
-	} else if (strcmp("com.apple.mobile.lockdown", type) != 0) {
-		debug_info("Warning QueryType request returned \"%s\".", type);
+	} else if (!strcmp("com.apple.mobile.lockdown", type)) {
+		is_lockdownd = 1;
+	} else {
+		debug_info("QueryType request returned \"%s\"", type);
 	}
 	free(type);
 
 	*client = client_loc;
 
-	if (device->version == 0) {
+	if (is_lockdownd && device->version == 0) {
 		plist_t p_version = NULL;
 		if (lockdownd_get_value(client_loc, NULL, "ProductVersion", &p_version) == LOCKDOWN_E_SUCCESS) {
 			int vers[3] = {0, 0, 0};
@@ -665,7 +668,7 @@ lockdownd_error_t lockdownd_client_new(idevice_t device, lockdownd_client_t *cli
 		}
 		plist_free(p_version);
 	}
-	if (device->device_class == 0) {
+	if (is_lockdownd && device->device_class == 0) {
 		plist_t p_device_class = NULL;
 		if (lockdownd_get_value(client_loc, NULL, "DeviceClass", &p_device_class) == LOCKDOWN_E_SUCCESS) {
 			char* s_device_class = NULL;