1 files changed, 18 insertions, 11 deletions

diff --git a/dev/lckdclient.c b/dev/lckdclient.c
index c96f052..e197002 100644
--- a/dev/lckdclient.c
+++ b/dev/lckdclient.c
@@ -27,12 +27,11 @@
 #include <readline/history.h>
 
 #include <libiphone/libiphone.h>
-
+#include <libiphone/lockdown.h>
 
 int main(int argc, char *argv[])
 {
-        int bytes = 0, port = 0, i = 0;
-        iphone_lckd_client_t control = NULL;
+        lockdownd_client_t client = NULL;
         iphone_device_t phone = NULL;
 
         iphone_set_debug(1);
@@ -42,13 +41,13 @@ int main(int argc, char *argv[])
                 return -1;
         }
 
-        if (IPHONE_E_SUCCESS != iphone_lckd_new_client(phone, &control)) {
+        if (IPHONE_E_SUCCESS != lockdownd_new_client(phone, &client)) {
                 iphone_free_device(phone);
                 return -1;
         }
 
         char *uid = NULL;
-        if (IPHONE_E_SUCCESS == lockdownd_get_device_uid(control, &uid)) {
+        if (IPHONE_E_SUCCESS == lockdownd_get_device_uid(client, &uid)) {
                 printf("DeviceUniqueID : %s\n", uid);
                 free(uid);
         }
@@ -74,17 +73,25 @@ int main(int argc, char *argv[])
                                 if (!strcmp(*args, "quit"))
                                         loop = FALSE;
 
-                                if (!strcmp(*args, "get") && len == 3) {
-                                        char *value = NULL;
-                                        if (IPHONE_E_SUCCESS == lockdownd_generic_get_value(control, *(args + 1), *(args + 2), &value))
-                                                printf("Success : value = %s\n", value);
+                                if (!strcmp(*args, "get") && len >= 2) {
+                                        plist_t *value = NULL;
+                                        if (IPHONE_E_SUCCESS == lockdownd_get_value(client, len == 3 ? *(args + 1):NULL,  len == 3 ? *(args + 2):*(args + 1), &value))
+                                        {
+                                                char *xml = NULL;
+                                                uint32_t length;
+                                                plist_to_xml(value, &xml, &length);
+                                                printf("Success : value = %s\n", xml);
+
+                                                free(xml);
+                                                free(value);
+                                        }
                                         else
                                                 printf("Error\n");
                                 }
 
                                 if (!strcmp(*args, "start") && len == 2) {
                                         int port = 0;
-                                        iphone_lckd_start_service(control, *(args + 1), &port);
+                                        lockdownd_start_service(client, *(args + 1), &port);
                                         printf("%i\n", port);
                                 }
                         }
@@ -94,7 +101,7 @@ int main(int argc, char *argv[])
                 cmd = NULL;
         }
         clear_history();
-        iphone_lckd_free_client(control);
+        lockdownd_free_client(client);
         iphone_free_device(phone);
 
         return 0;