Make sure we can pass NULL as lockdown label to disable sending one

1 files changed, 10 insertions, 4 deletions

diff --git a/src/lockdown.c b/src/lockdown.c
index 5c51120..c78d691 100644
--- a/src/lockdown.c
+++ b/src/lockdown.c
@@ -251,13 +251,17 @@ lockdownd_error_t lockdownd_client_free(lockdownd_client_t client)
  * Sets the label to send for requests to lockdownd.
  *
  * @param client The lockdown client
- * @param label The label to set or NULL to disable
+ * @param label The label to set or NULL to disable sending a label
  *
  */
 void lockdownd_client_set_label(lockdownd_client_t client, const char *label)
 {
-	if (client)
-		client->label = strdup(label);
+	if (client) {
+		if (client->label)
+			free(client->label);
+
+		client->label = (label != NULL) ? strdup(label): NULL;
+	}
 }
 
 /** Polls the iPhone for lockdownd data.
@@ -644,7 +648,9 @@ lockdownd_error_t lockdownd_client_new(iphone_device_t device, lockdownd_client_
 	client_loc->in_SSL = 0;
 	client_loc->session_id = NULL;
 	client_loc->uuid = NULL;
-	client_loc->label = strdup(label);
+	client_loc->label = NULL;
+	if (label != NULL)
+		strdup(label);
 
 	if (LOCKDOWN_E_SUCCESS != lockdownd_query_type(client_loc, &type)) {
 		log_debug_msg("%s: QueryType failed in the lockdownd client.\n", __func__);