summaryrefslogtreecommitdiffstats
path: root/src/lockdown.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/lockdown.c')
-rw-r--r--src/lockdown.c921
1 files changed, 448 insertions, 473 deletions
diff --git a/src/lockdown.c b/src/lockdown.c
index 2d85a03..2f48dfd 100644
--- a/src/lockdown.c
+++ b/src/lockdown.c
@@ -31,6 +31,9 @@
31#include <libtasn1.h> 31#include <libtasn1.h>
32#include <gnutls/x509.h> 32#include <gnutls/x509.h>
33 33
34#include <plist/plist.h>
35
36
34const ASN1_ARRAY_TYPE pkcs1_asn1_tab[] = { 37const ASN1_ARRAY_TYPE pkcs1_asn1_tab[] = {
35 {"PKCS1", 536872976, 0}, 38 {"PKCS1", 536872976, 0},
36 {0, 1073741836, 0}, 39 {0, 1073741836, 0},
@@ -40,35 +43,6 @@ const ASN1_ARRAY_TYPE pkcs1_asn1_tab[] = {
40 {0, 0, 0} 43 {0, 0, 0}
41}; 44};
42 45
43static int get_rand(int min, int max)
44{
45 int retval = (rand() % (max - min)) + min;
46 return retval;
47}
48
49/** Generates a valid HostID (which is actually a UUID).
50 *
51 * @param A null terminated string containing a valid HostID.
52 */
53char *lockdownd_generate_hostid(void)
54{
55 char *hostid = (char *) malloc(sizeof(char) * 37); // HostID's are just UUID's, and UUID's are 36 characters long
56 const char *chars = "ABCDEF0123456789";
57 srand(time(NULL));
58 int i = 0;
59
60 for (i = 0; i < 36; i++) {
61 if (i == 8 || i == 13 || i == 18 || i == 23) {
62 hostid[i] = '-';
63 continue;
64 } else {
65 hostid[i] = chars[get_rand(0, 16)];
66 }
67 }
68 hostid[36] = '\0'; // make it a real string
69 return hostid;
70}
71
72/** Creates a lockdownd client for the give iPhone. 46/** Creates a lockdownd client for the give iPhone.
73 * 47 *
74 * @param phone The iPhone to create a lockdownd client for 48 * @param phone The iPhone to create a lockdownd client for
@@ -101,57 +75,56 @@ iphone_lckd_client_t new_lockdownd_client(iphone_device_t phone)
101static void iphone_lckd_stop_session(iphone_lckd_client_t control) 75static void iphone_lckd_stop_session(iphone_lckd_client_t control)
102{ 76{
103 if (!control) 77 if (!control)
104 return; // IPHONE_E_INVALID_ARG; 78 return; //IPHONE_E_INVALID_ARG;
105 xmlDocPtr plist = new_plist(); 79
106 xmlNode *dict, *key;
107 char **dictionary;
108 int bytes = 0, i = 0;
109 iphone_error_t ret = IPHONE_E_UNKNOWN_ERROR; 80 iphone_error_t ret = IPHONE_E_UNKNOWN_ERROR;
110 81
111 log_debug_msg("lockdownd_stop_session() called\n"); 82 plist_t dict = plist_new_dict();
112 dict = add_child_to_plist(plist, "dict", "\n", NULL, 0); 83 plist_add_sub_key_el(dict, "Request");
113 key = add_key_str_dict_element(plist, dict, "Request", "StopSession", 1); 84 plist_add_sub_string_el(dict, "StopSession");
114 key = add_key_str_dict_element(plist, dict, "SessionID", control->session_id, 1); 85 plist_add_sub_key_el(dict, "SessionID");
86 plist_add_sub_string_el(dict, control->session_id);
115 87
116 char *XML_content; 88 log_dbg_msg(DBGMASK_LOCKDOWND, "iphone_lckd_stop_session() called\n");
117 uint32 length;
118 89
119 xmlDocDumpMemory(plist, (xmlChar **) & XML_content, &length); 90 ret = iphone_lckd_send(control, dict);
120 ret = iphone_lckd_send(control, XML_content, length, &bytes);
121 91
122 xmlFree(XML_content); 92 plist_free(dict);
123 xmlFreeDoc(plist); 93 dict = NULL;
124 plist = NULL; 94
125 ret = iphone_lckd_recv(control, &XML_content, &bytes); 95 ret = iphone_lckd_recv(control, &dict);
126 96
127 plist = xmlReadMemory(XML_content, bytes, NULL, NULL, 0);
128 if (!plist) {
129 log_debug_msg("lockdownd_stop_session(): IPHONE_E_PLIST_ERROR\n");
130 return; //IPHONE_E_PLIST_ERROR;
131 }
132 dict = xmlDocGetRootElement(plist);
133 for (dict = dict->children; dict; dict = dict->next) {
134 if (!xmlStrcmp(dict->name, "dict"))
135 break;
136 }
137 if (!dict) { 97 if (!dict) {
138 log_debug_msg("lockdownd_stop_session(): IPHONE_E_DICT_ERROR\n"); 98 log_dbg_msg(DBGMASK_LOCKDOWND, "lockdownd_stop_session(): IPHONE_E_PLIST_ERROR\n");
139 return; //IPHONE_E_DICT_ERROR; 99 return; // IPHONE_E_PLIST_ERROR;
140 } 100 }
141 dictionary = read_dict_element_strings(dict);
142 xmlFreeDoc(plist);
143 free(XML_content);
144 101
145 for (i = 0; dictionary[i]; i += 2) { 102 plist_t query_node = plist_find_node_by_string(dict, "StopSession");
146 if (!strcmp(dictionary[i], "Result") && !strcmp(dictionary[i + 1], "Success")) { 103 plist_t result_node = plist_get_next_sibling(query_node);
147 log_debug_msg("lockdownd_stop_session(): success\n"); 104 plist_t value_node = plist_get_next_sibling(result_node);
105
106 plist_type result_type = plist_get_node_type(result_node);
107 plist_type value_type = plist_get_node_type(value_node);
108
109 if (result_type == PLIST_KEY && value_type == PLIST_STRING) {
110
111 char *result_value = NULL;
112 char *value_value = NULL;
113
114 plist_get_key_val(result_node, &result_value);
115 plist_get_string_val(value_node, &value_value);
116
117 if (!strcmp(result_value, "Result") && !strcmp(value_value, "Success")) {
118 log_dbg_msg(DBGMASK_LOCKDOWND, "lockdownd_stop_session(): success\n");
148 ret = IPHONE_E_SUCCESS; 119 ret = IPHONE_E_SUCCESS;
149 break;
150 } 120 }
121 free(result_value);
122 free(value_value);
151 } 123 }
124 plist_free(dict);
125 dict = NULL;
152 126
153 free_dictionary(dictionary); 127 return; // ret;
154 return; //ret;
155} 128}
156 129
157/** 130/**
@@ -164,14 +137,14 @@ static void iphone_lckd_stop_session(iphone_lckd_client_t control)
164static void iphone_lckd_stop_SSL_session(iphone_lckd_client_t client) 137static void iphone_lckd_stop_SSL_session(iphone_lckd_client_t client)
165{ 138{
166 if (!client) { 139 if (!client) {
167 log_debug_msg("lockdownd_stop_SSL_session(): invalid argument!\n"); 140 log_dbg_msg(DBGMASK_LOCKDOWND, "lockdownd_stop_SSL_session(): invalid argument!\n");
168 return; 141 return;
169 } 142 }
170 143
171 if (client->in_SSL) { 144 if (client->in_SSL) {
172 log_debug_msg("Stopping SSL Session\n"); 145 log_dbg_msg(DBGMASK_LOCKDOWND, "Stopping SSL Session\n");
173 iphone_lckd_stop_session(client); 146 iphone_lckd_stop_session(client);
174 log_debug_msg("Sending SSL close notify\n"); 147 log_dbg_msg(DBGMASK_LOCKDOWND, "Sending SSL close notify\n");
175 gnutls_bye(*client->ssl_session, GNUTLS_SHUT_RDWR); 148 gnutls_bye(*client->ssl_session, GNUTLS_SHUT_RDWR);
176 } 149 }
177 if (client->ssl_session) { 150 if (client->ssl_session) {
@@ -217,13 +190,13 @@ iphone_error_t iphone_lckd_free_client(iphone_lckd_client_t client)
217 * 190 *
218 * @return The number of bytes received 191 * @return The number of bytes received
219 */ 192 */
220iphone_error_t iphone_lckd_recv(iphone_lckd_client_t client, char **dump_data, uint32_t * recv_bytes) 193iphone_error_t iphone_lckd_recv(iphone_lckd_client_t client, plist_t * plist)
221{ 194{
222 if (!client || !dump_data || !recv_bytes) 195 if (!client || !plist || (plist && *plist))
223 return IPHONE_E_INVALID_ARG; 196 return IPHONE_E_INVALID_ARG;
224 iphone_error_t ret = IPHONE_E_UNKNOWN_ERROR; 197 iphone_error_t ret = IPHONE_E_UNKNOWN_ERROR;
225 char *receive; 198 char *receive;
226 uint32 datalen = 0, bytes = 0; 199 uint32_t datalen = 0, bytes = 0;
227 200
228 if (!client->in_SSL) 201 if (!client->in_SSL)
229 ret = iphone_mux_recv(client->connection, (char *) &datalen, sizeof(datalen), &bytes); 202 ret = iphone_mux_recv(client->connection, (char *) &datalen, sizeof(datalen), &bytes);
@@ -242,8 +215,18 @@ iphone_error_t iphone_lckd_recv(iphone_lckd_client_t client, char **dump_data, u
242 if (bytes > 0) 215 if (bytes > 0)
243 ret = IPHONE_E_SUCCESS; 216 ret = IPHONE_E_SUCCESS;
244 } 217 }
245 *dump_data = receive; 218
246 *recv_bytes = bytes; 219 if (bytes <= 0) {
220 free(receive);
221 return IPHONE_E_NOT_ENOUGH_DATA;
222 }
223
224 plist_from_xml(receive, bytes, plist);
225 free(receive);
226
227 if (!*plist)
228 ret = IPHONE_E_PLIST_ERROR;
229
247 return ret; 230 return ret;
248} 231}
249 232
@@ -252,26 +235,31 @@ iphone_error_t iphone_lckd_recv(iphone_lckd_client_t client, char **dump_data, u
252 * @note This function is low-level and should only be used if you need to send 235 * @note This function is low-level and should only be used if you need to send
253 * a new type of message. 236 * a new type of message.
254 * 237 *
255 * @param control The lockdownd client 238 * @param client The lockdownd client
256 * @param raw_data The null terminated string buffer to send 239 * @param plist The plist to send
257 * @param length The length of data to send
258 * 240 *
259 * @return The number of bytes sent 241 * @return an error code (IPHONE_E_SUCCESS on success)
260 */ 242 */
261iphone_error_t iphone_lckd_send(iphone_lckd_client_t client, char *raw_data, uint32_t length, uint32_t * sent_bytes) 243iphone_error_t iphone_lckd_send(iphone_lckd_client_t client, plist_t plist)
262{ 244{
263 if (!client || !raw_data || length == 0 || !sent_bytes) 245 if (!client || !plist)
264 return IPHONE_E_INVALID_ARG; 246 return IPHONE_E_INVALID_ARG;
265 char *real_query; 247 char *real_query;
266 int bytes; 248 int bytes;
249 char *XMLContent = NULL;
250 uint32_t length = 0;
267 iphone_error_t ret = IPHONE_E_UNKNOWN_ERROR; 251 iphone_error_t ret = IPHONE_E_UNKNOWN_ERROR;
268 252
253 plist_to_xml(plist, &XMLContent, &length);
254 log_dbg_msg(DBGMASK_LOCKDOWND, "Send msg :\nsize : %i\nbuffer :\n%s\n", length, XMLContent);
255
256
269 real_query = (char *) malloc(sizeof(char) * (length + 4)); 257 real_query = (char *) malloc(sizeof(char) * (length + 4));
270 length = htonl(length); 258 length = htonl(length);
271 memcpy(real_query, &length, sizeof(length)); 259 memcpy(real_query, &length, sizeof(length));
272 memcpy(real_query + 4, raw_data, ntohl(length)); 260 memcpy(real_query + 4, XMLContent, ntohl(length));
273 log_debug_msg("lockdownd_send(): made the query, sending it along\n"); 261 free(XMLContent);
274 dump_debug_buffer("grpkt", real_query, ntohl(length) + 4); 262 log_dbg_msg(DBGMASK_LOCKDOWND, "lockdownd_send(): made the query, sending it along\n");
275 263
276 if (!client->in_SSL) 264 if (!client->in_SSL)
277 ret = iphone_mux_send(client->connection, real_query, ntohl(length) + sizeof(length), &bytes); 265 ret = iphone_mux_send(client->connection, real_query, ntohl(length) + sizeof(length), &bytes);
@@ -279,9 +267,9 @@ iphone_error_t iphone_lckd_send(iphone_lckd_client_t client, char *raw_data, uin
279 gnutls_record_send(*client->ssl_session, real_query, ntohl(length) + sizeof(length)); 267 gnutls_record_send(*client->ssl_session, real_query, ntohl(length) + sizeof(length));
280 ret = IPHONE_E_SUCCESS; 268 ret = IPHONE_E_SUCCESS;
281 } 269 }
282 log_debug_msg("lockdownd_send(): sent it!\n"); 270 log_dbg_msg(DBGMASK_LOCKDOWND, "lockdownd_send(): sent it!\n");
283 free(real_query); 271 free(real_query);
284 *sent_bytes = bytes; 272
285 return ret; 273 return ret;
286} 274}
287 275
@@ -297,49 +285,50 @@ iphone_error_t lockdownd_hello(iphone_lckd_client_t control)
297{ 285{
298 if (!control) 286 if (!control)
299 return IPHONE_E_INVALID_ARG; 287 return IPHONE_E_INVALID_ARG;
300 xmlDocPtr plist = new_plist(); 288
301 xmlNode *dict, *key;
302 char **dictionary;
303 int bytes = 0, i = 0;
304 iphone_error_t ret = IPHONE_E_UNKNOWN_ERROR; 289 iphone_error_t ret = IPHONE_E_UNKNOWN_ERROR;
305 290
306 log_debug_msg("lockdownd_hello() called\n"); 291 plist_t dict = plist_new_dict();
307 dict = add_child_to_plist(plist, "dict", "\n", NULL, 0); 292 plist_add_sub_key_el(dict, "Request");
308 key = add_key_str_dict_element(plist, dict, "Request", "QueryType", 1); 293 plist_add_sub_string_el(dict, "QueryType");
309 char *XML_content;
310 uint32 length;
311 294
312 xmlDocDumpMemory(plist, (xmlChar **) & XML_content, &length); 295 log_dbg_msg(DBGMASK_LOCKDOWND, "lockdownd_hello() called\n");
313 ret = iphone_lckd_send(control, XML_content, length, &bytes); 296 ret = iphone_lckd_send(control, dict);
314 297
315 xmlFree(XML_content); 298 plist_free(dict);
316 xmlFreeDoc(plist); 299 dict = NULL;
317 plist = NULL;
318 ret = iphone_lckd_recv(control, &XML_content, &bytes);
319 300
320 plist = xmlReadMemory(XML_content, bytes, NULL, NULL, 0); 301 ret = iphone_lckd_recv(control, &dict);
321 if (!plist) 302
322 return IPHONE_E_PLIST_ERROR; 303 if (IPHONE_E_SUCCESS != ret)
323 dict = xmlDocGetRootElement(plist); 304 return ret;
324 for (dict = dict->children; dict; dict = dict->next) { 305
325 if (!xmlStrcmp(dict->name, "dict")) 306 plist_t query_node = plist_find_node_by_string(dict, "QueryType");
326 break; 307 plist_t result_node = plist_get_next_sibling(query_node);
327 } 308 plist_t value_node = plist_get_next_sibling(result_node);
328 if (!dict) 309
329 return IPHONE_E_DICT_ERROR; 310 plist_type result_type = plist_get_node_type(result_node);
330 dictionary = read_dict_element_strings(dict); 311 plist_type value_type = plist_get_node_type(value_node);
331 xmlFreeDoc(plist); 312
332 free(XML_content); 313 if (result_type == PLIST_KEY && value_type == PLIST_STRING) {
333 314
334 for (i = 0; dictionary[i]; i += 2) { 315 char *result_value = NULL;
335 if (!strcmp(dictionary[i], "Result") && !strcmp(dictionary[i + 1], "Success")) { 316 char *value_value = NULL;
336 log_debug_msg("lockdownd_hello(): success\n"); 317
318 plist_get_key_val(result_node, &result_value);
319 plist_get_string_val(value_node, &value_value);
320
321 if (!strcmp(result_value, "Result") && !strcmp(value_value, "Success")) {
322 log_dbg_msg(DBGMASK_LOCKDOWND, "lockdownd_hello(): success\n");
337 ret = IPHONE_E_SUCCESS; 323 ret = IPHONE_E_SUCCESS;
338 break;
339 } 324 }
325 free(result_value);
326 free(value_value);
340 } 327 }
341 328
342 free_dictionary(dictionary); 329 plist_free(dict);
330 dict = NULL;
331
343 return ret; 332 return ret;
344} 333}
345 334
@@ -351,74 +340,91 @@ iphone_error_t lockdownd_hello(iphone_lckd_client_t control)
351 * 340 *
352 * @return IPHONE_E_SUCCESS on success. 341 * @return IPHONE_E_SUCCESS on success.
353 */ 342 */
354iphone_error_t lockdownd_generic_get_value(iphone_lckd_client_t control, const char *req_key, const char *req_string, 343iphone_error_t lockdownd_generic_get_value(iphone_lckd_client_t control, const char *req_key, char *req_string,
355 char **value) 344 gnutls_datum_t * value)
356{ 345{
357 if (!control || !req_key || !value || (value && *value)) 346 if (!control || !req_key || !value || value->data)
358 return IPHONE_E_INVALID_ARG; 347 return IPHONE_E_INVALID_ARG;
359 xmlDocPtr plist = new_plist(); 348
360 xmlNode *dict = NULL; 349 plist_t dict = NULL;
361 xmlNode *key = NULL;;
362 char **dictionary = NULL;
363 int bytes = 0, i = 0;
364 char *XML_content = NULL;
365 uint32 length = 0;
366 iphone_error_t ret = IPHONE_E_UNKNOWN_ERROR; 350 iphone_error_t ret = IPHONE_E_UNKNOWN_ERROR;
367 351
368 /* Setup DevicePublicKey request plist */ 352 /* Setup DevicePublicKey request plist */
369 dict = add_child_to_plist(plist, "dict", "\n", NULL, 0); 353 dict = plist_new_dict();
370 key = add_key_str_dict_element(plist, dict, req_key, req_string, 1); 354 plist_add_sub_key_el(dict, req_key);
371 key = add_key_str_dict_element(plist, dict, "Request", "GetValue", 1); 355 plist_add_sub_string_el(dict, req_string);
372 xmlDocDumpMemory(plist, (xmlChar **) & XML_content, &length); 356 plist_add_sub_key_el(dict, "Request");
357 plist_add_sub_string_el(dict, "GetValue");
373 358
374 /* send to iPhone */ 359 /* send to iPhone */
375 ret = iphone_lckd_send(control, XML_content, length, &bytes); 360 ret = iphone_lckd_send(control, dict);
376 361
377 xmlFree(XML_content); 362 plist_free(dict);
378 xmlFreeDoc(plist); 363 dict = NULL;
379 plist = NULL;
380 364
381 if (ret != IPHONE_E_SUCCESS) 365 if (ret != IPHONE_E_SUCCESS)
382 return ret; 366 return ret;
383 367
384 /* Now get iPhone's answer */ 368 /* Now get iPhone's answer */
385 ret = iphone_lckd_recv(control, &XML_content, &bytes); 369 ret = iphone_lckd_recv(control, &dict);
386 370
387 if (ret != IPHONE_E_SUCCESS) 371 if (ret != IPHONE_E_SUCCESS)
388 return ret; 372 return ret;
389 373
390 plist = xmlReadMemory(XML_content, bytes, NULL, NULL, 0); 374 plist_t query_node = plist_find_node_by_string(dict, "GetValue");
391 if (!plist) 375 plist_t result_key_node = plist_get_next_sibling(query_node);
392 return IPHONE_E_PLIST_ERROR; 376 plist_t result_value_node = plist_get_next_sibling(result_key_node);
393 dict = xmlDocGetRootElement(plist);
394 for (dict = dict->children; dict; dict = dict->next) {
395 if (!xmlStrcmp(dict->name, "dict"))
396 break;
397 }
398 if (!dict)
399 return IPHONE_E_DICT_ERROR;
400 377
401 /* Parse xml to check success and to find public key */ 378 plist_type result_key_type = plist_get_node_type(result_key_node);
402 dictionary = read_dict_element_strings(dict); 379 plist_type result_value_type = plist_get_node_type(result_value_node);
403 xmlFreeDoc(plist);
404 free(XML_content);
405 380
406 int success = 0; 381 if (result_key_type == PLIST_KEY && result_value_type == PLIST_STRING) {
407 for (i = 0; dictionary[i]; i += 2) { 382
408 if (!strcmp(dictionary[i], "Result") && !strcmp(dictionary[i + 1], "Success")) { 383 char *result_key = NULL;
409 success = 1; 384 char *result_value = NULL;
410 } 385 ret = IPHONE_E_DICT_ERROR;
411 if (!strcmp(dictionary[i], "Value")) { 386
412 *value = strdup(dictionary[i + 1]); 387 plist_get_key_val(result_key_node, &result_key);
388 plist_get_string_val(result_value_node, &result_value);
389
390 if (!strcmp(result_key, "Result") && !strcmp(result_value, "Success")) {
391 log_dbg_msg(DBGMASK_LOCKDOWND, "lockdownd_generic_get_value(): success\n");
392 ret = IPHONE_E_SUCCESS;
413 } 393 }
394 free(result_key);
395 free(result_value);
396 }
397 if (ret != IPHONE_E_SUCCESS) {
398 return ret;
414 } 399 }
415 400
416 if (dictionary) { 401 plist_t value_key_node = plist_get_next_sibling(result_key_node);
417 free_dictionary(dictionary); 402 plist_t value_value_node = plist_get_next_sibling(value_key_node);
418 dictionary = NULL; 403
404 plist_type value_key_type = plist_get_node_type(value_key_node);
405
406 if (value_key_type == PLIST_KEY) {
407
408 char *result_key = NULL;
409 plist_get_key_val(value_key_node, &result_key);
410
411 if (!strcmp(result_key, "Value")) {
412 log_dbg_msg(DBGMASK_LOCKDOWND, "lockdownd_generic_get_value(): success\n");
413
414 plist_type value_value_type = plist_get_node_type(value_value_node);
415 if (PLIST_STRING == value_value_type) {
416 char *value_value = NULL;
417 plist_get_string_val(value_value_node, &value_value);
418
419 value->data = value_value;
420 value->size = strlen(value_value);
421 ret = IPHONE_E_SUCCESS;
422 }
423 }
424 free(result_key);
419 } 425 }
420 if (success) 426
421 ret = IPHONE_E_SUCCESS; 427 plist_free(dict);
422 return ret; 428 return ret;
423} 429}
424 430
@@ -430,7 +436,9 @@ iphone_error_t lockdownd_generic_get_value(iphone_lckd_client_t control, const c
430 */ 436 */
431iphone_error_t lockdownd_get_device_uid(iphone_lckd_client_t control, char **uid) 437iphone_error_t lockdownd_get_device_uid(iphone_lckd_client_t control, char **uid)
432{ 438{
433 return lockdownd_generic_get_value(control, "Key", "UniqueDeviceID", uid); 439 gnutls_datum_t temp = { NULL, 0 };
440 return lockdownd_generic_get_value(control, "Key", "UniqueDeviceID", &temp);
441 *uid = temp.data;
434} 442}
435 443
436/** Askes for the device's public key. Part of the lockdownd handshake. 444/** Askes for the device's public key. Part of the lockdownd handshake.
@@ -439,7 +447,7 @@ iphone_error_t lockdownd_get_device_uid(iphone_lckd_client_t control, char **uid
439 * 447 *
440 * @return 1 on success and 0 on failure. 448 * @return 1 on success and 0 on failure.
441 */ 449 */
442iphone_error_t lockdownd_get_device_public_key(iphone_lckd_client_t control, char **public_key) 450iphone_error_t lockdownd_get_device_public_key(iphone_lckd_client_t control, gnutls_datum_t * public_key)
443{ 451{
444 return lockdownd_generic_get_value(control, "Key", "DevicePublicKey", public_key); 452 return lockdownd_generic_get_value(control, "Key", "DevicePublicKey", public_key);
445} 453}
@@ -511,107 +519,91 @@ iphone_error_t iphone_lckd_new_client(iphone_device_t device, iphone_lckd_client
511iphone_error_t lockdownd_pair_device(iphone_lckd_client_t control, char *uid, char *host_id) 519iphone_error_t lockdownd_pair_device(iphone_lckd_client_t control, char *uid, char *host_id)
512{ 520{
513 iphone_error_t ret = IPHONE_E_UNKNOWN_ERROR; 521 iphone_error_t ret = IPHONE_E_UNKNOWN_ERROR;
514 xmlDocPtr plist = new_plist(); 522 plist_t dict = NULL;
515 xmlNode *dict = NULL; 523 plist_t dict_record = NULL;
516 xmlNode *dictRecord = NULL; 524
517 char **dictionary = NULL; 525 gnutls_datum_t device_cert = { NULL, 0 };
518 int bytes = 0, i = 0; 526 gnutls_datum_t host_cert = { NULL, 0 };
519 char *XML_content = NULL; 527 gnutls_datum_t root_cert = { NULL, 0 };
520 uint32 length = 0; 528 gnutls_datum_t public_key = { NULL, 0 };
521 529
522 char *device_cert_b64 = NULL; 530 ret = lockdownd_get_device_public_key(control, &public_key);
523 char *host_cert_b64 = NULL;
524 char *root_cert_b64 = NULL;
525 char *public_key_b64 = NULL;
526
527 ret = lockdownd_get_device_public_key(control, &public_key_b64);
528 if (ret != IPHONE_E_SUCCESS) { 531 if (ret != IPHONE_E_SUCCESS) {
529 log_debug_msg("Device refused to send public key.\n"); 532 log_debug_msg("Device refused to send public key.\n");
530 return ret; 533 return ret;
531 } 534 }
532 535
533 ret = lockdownd_gen_pair_cert(public_key_b64, &device_cert_b64, &host_cert_b64, &root_cert_b64); 536 ret = lockdownd_gen_pair_cert(public_key, &device_cert, &host_cert, &root_cert);
534 if (ret != IPHONE_E_SUCCESS) { 537 if (ret != IPHONE_E_SUCCESS) {
535 free(public_key_b64); 538 free(public_key.data);
536 return ret; 539 return ret;
537 } 540 }
538 541
539 /* Setup Pair request plist */ 542 /* Setup Pair request plist */
540 dict = add_child_to_plist(plist, "dict", "\n", NULL, 0); 543 dict = plist_new_dict();
541 dictRecord = add_key_dict_node(plist, dict, "PairRecord", "\n", 1); 544 plist_add_sub_key_el(dict, "PairRecord");
542 //dictRecord = add_child_to_plist(plist, "dict", "\n", NULL, 1); 545 dict_record = plist_new_dict();
543 add_key_data_dict_element(plist, dictRecord, "DeviceCertificate", device_cert_b64, 2); 546 plist_add_sub_node(dict, dict_record);
544 add_key_data_dict_element(plist, dictRecord, "HostCertificate", host_cert_b64, 2); 547 plist_add_sub_key_el(dict_record, "DeviceCertificate");
545 add_key_str_dict_element(plist, dictRecord, "HostID", host_id, 2); 548 plist_add_sub_data_el(dict_record, device_cert.data, device_cert.size);
546 add_key_data_dict_element(plist, dictRecord, "RootCertificate", root_cert_b64, 2); 549 plist_add_sub_key_el(dict_record, "HostCertificate");
547 add_key_str_dict_element(plist, dict, "Request", "Pair", 1); 550 plist_add_sub_data_el(dict_record, host_cert.data, host_cert.size);
548 551 plist_add_sub_key_el(dict_record, "HostID");
549 xmlDocDumpMemory(plist, (xmlChar **) & XML_content, &length); 552 plist_add_sub_string_el(dict_record, host_id);
550 553 plist_add_sub_key_el(dict_record, "RootCertificate");
551 printf("XML Pairing request : %s\n", XML_content); 554 plist_add_sub_data_el(dict_record, root_cert.data, root_cert.size);
555 plist_add_sub_key_el(dict_record, "Request");
556 plist_add_sub_string_el(dict_record, "Pair");
552 557
553 /* send to iPhone */ 558 /* send to iPhone */
554 ret = iphone_lckd_send(control, XML_content, length, &bytes); 559 ret = iphone_lckd_send(control, dict);
555 560 plist_free(dict);
556 xmlFree(XML_content); 561 dict = NULL;
557 xmlFreeDoc(plist);
558 plist = NULL;
559 562
560 if (ret != IPHONE_E_SUCCESS) 563 if (ret != IPHONE_E_SUCCESS)
561 return ret; 564 return ret;
562 565
563 /* Now get iPhone's answer */ 566 /* Now get iPhone's answer */
564 ret = iphone_lckd_recv(control, &XML_content, &bytes); 567 ret = iphone_lckd_recv(control, &dict);
565 568
566 if (ret != IPHONE_E_SUCCESS) 569 if (ret != IPHONE_E_SUCCESS)
567 return ret; 570 return ret;
568 571
569 log_debug_msg("lockdown_pair_device: iPhone's response to our pair request:\n"); 572 plist_t query_node = plist_find_node_by_string(dict, "Pair");
570 log_debug_msg(XML_content); 573 plist_t result_key_node = plist_get_next_sibling(query_node);
571 log_debug_msg("\n\n"); 574 plist_t result_value_node = plist_get_next_sibling(result_key_node);
572 575
573 plist = xmlReadMemory(XML_content, bytes, NULL, NULL, 0); 576 plist_type result_key_type = plist_get_node_type(result_key_node);
574 if (!plist) { 577 plist_type result_value_type = plist_get_node_type(result_value_node);
575 free(public_key_b64); 578
576 return IPHONE_E_PLIST_ERROR; 579 if (result_key_type == PLIST_KEY && result_value_type == PLIST_STRING) {
577 } 580
578 dict = xmlDocGetRootElement(plist); 581 char *result_key = NULL;
579 for (dict = dict->children; dict; dict = dict->next) { 582 char *result_value = NULL;
580 if (!xmlStrcmp(dict->name, "dict"))
581 break;
582 }
583 if (!dict) {
584 free(public_key_b64);
585 return IPHONE_E_DICT_ERROR;
586 }
587 583
588 /* Parse xml to check success and to find public key */ 584 plist_get_key_val(result_key_node, &result_key);
589 dictionary = read_dict_element_strings(dict); 585 plist_get_string_val(result_value_node, &result_value);
590 xmlFreeDoc(plist);
591 free(XML_content);
592 586
593 int success = 0; 587 if (!strcmp(result_key, "Result") && !strcmp(result_value, "Success")) {
594 for (i = 0; dictionary[i]; i += 2) { 588 ret = IPHONE_E_SUCCESS;
595 if (!strcmp(dictionary[i], "Result") && !strcmp(dictionary[i + 1], "Success")) {
596 success = 1;
597 } 589 }
598 }
599 590
600 if (dictionary) { 591 free(result_key);
601 free_dictionary(dictionary); 592 free(result_value);
602 dictionary = NULL;
603 } 593 }
594 plist_free(dict);
595 dict = NULL;
604 596
605 /* store public key in config if pairing succeeded */ 597 /* store public key in config if pairing succeeded */
606 if (success) { 598 if (ret == IPHONE_E_SUCCESS) {
607 log_debug_msg("lockdownd_pair_device: pair success\n"); 599 log_dbg_msg(DBGMASK_LOCKDOWND, "lockdownd_pair_device: pair success\n");
608 store_device_public_key(uid, public_key_b64); 600 store_device_public_key(uid, public_key);
609 ret = IPHONE_E_SUCCESS; 601 ret = IPHONE_E_SUCCESS;
610 } else { 602 } else {
611 log_debug_msg("lockdownd_pair_device: pair failure\n"); 603 log_dbg_msg(DBGMASK_LOCKDOWND, "lockdownd_pair_device: pair failure\n");
612 ret = IPHONE_E_PAIRING_FAILED; 604 ret = IPHONE_E_PAIRING_FAILED;
613 } 605 }
614 free(public_key_b64); 606 free(public_key.data);
615 return ret; 607 return ret;
616} 608}
617 609
@@ -624,55 +616,51 @@ iphone_error_t lockdownd_pair_device(iphone_lckd_client_t control, char *uid, ch
624void lockdownd_close(iphone_lckd_client_t control) 616void lockdownd_close(iphone_lckd_client_t control)
625{ 617{
626 if (!control) 618 if (!control)
627 return; // IPHONE_E_INVALID_ARG; 619 return; //IPHONE_E_INVALID_ARG;
628 xmlDocPtr plist = new_plist(); 620
629 xmlNode *dict, *key;
630 char **dictionary;
631 int bytes = 0, i = 0;
632 iphone_error_t ret = IPHONE_E_UNKNOWN_ERROR; 621 iphone_error_t ret = IPHONE_E_UNKNOWN_ERROR;
633 622
634 log_debug_msg("lockdownd_close() called\n"); 623 plist_t dict = plist_new_dict();
635 dict = add_child_to_plist(plist, "dict", "\n", NULL, 0); 624 plist_add_sub_key_el(dict, "Request");
636 key = add_key_str_dict_element(plist, dict, "Request", "Goodbye", 1); 625 plist_add_sub_string_el(dict, "Goodbye");
637 char *XML_content;
638 uint32 length;
639 626
640 xmlDocDumpMemory(plist, (xmlChar **) & XML_content, &length); 627 log_dbg_msg(DBGMASK_LOCKDOWND, "lockdownd_close() called\n");
641 ret = iphone_lckd_send(control, XML_content, length, &bytes);
642 628
643 xmlFree(XML_content); 629 ret = iphone_lckd_send(control, dict);
644 xmlFreeDoc(plist); 630 plist_free(dict);
645 plist = NULL; 631 dict = NULL;
646 ret = iphone_lckd_recv(control, &XML_content, &bytes); 632
633 ret = iphone_lckd_recv(control, &dict);
647 634
648 plist = xmlReadMemory(XML_content, bytes, NULL, NULL, 0);
649 if (!plist) {
650 log_debug_msg("lockdownd_close(): IPHONE_E_PLIST_ERROR\n");
651 return; //IPHONE_E_PLIST_ERROR;
652 }
653 dict = xmlDocGetRootElement(plist);
654 for (dict = dict->children; dict; dict = dict->next) {
655 if (!xmlStrcmp(dict->name, "dict"))
656 break;
657 }
658 if (!dict) { 635 if (!dict) {
659 log_debug_msg("lockdownd_close(): IPHONE_E_DICT_ERROR\n"); 636 log_dbg_msg(DBGMASK_LOCKDOWND, "lockdownd_close(): IPHONE_E_PLIST_ERROR\n");
660 return; //IPHONE_E_DICT_ERROR; 637 return; // IPHONE_E_PLIST_ERROR;
661 } 638 }
662 dictionary = read_dict_element_strings(dict);
663 xmlFreeDoc(plist);
664 free(XML_content);
665 639
666 for (i = 0; dictionary[i]; i += 2) { 640 plist_t query_node = plist_find_node_by_string(dict, "Goodbye");
667 if (!strcmp(dictionary[i], "Result") && !strcmp(dictionary[i + 1], "Success")) { 641 plist_t result_node = plist_get_next_sibling(query_node);
668 log_debug_msg("lockdownd_close(): success\n"); 642 plist_t value_node = plist_get_next_sibling(result_node);
643
644 plist_type result_type = plist_get_node_type(result_node);
645 plist_type value_type = plist_get_node_type(value_node);
646
647 if (result_type == PLIST_KEY && value_type == PLIST_STRING) {
648 char *result_value = NULL;
649 char *value_value = NULL;
650
651 plist_get_key_val(result_node, &result_value);
652 plist_get_string_val(value_node, &value_value);
653
654 if (!strcmp(result_value, "Result") && !strcmp(value_value, "Success")) {
655 log_dbg_msg(DBGMASK_LOCKDOWND, "lockdownd_close(): success\n");
669 ret = IPHONE_E_SUCCESS; 656 ret = IPHONE_E_SUCCESS;
670 break;
671 } 657 }
658 free(result_value);
659 free(value_value);
672 } 660 }
673 661 plist_free(dict);
674 free_dictionary(dictionary); 662 dict = NULL;
675 return; //ret; 663 return; // ret;
676} 664}
677 665
678/** Generates the device certificate from the public key as well as the host 666/** Generates the device certificate from the public key as well as the host
@@ -680,25 +668,19 @@ void lockdownd_close(iphone_lckd_client_t control)
680 * 668 *
681 * @return IPHONE_E_SUCCESS on success. 669 * @return IPHONE_E_SUCCESS on success.
682 */ 670 */
683iphone_error_t lockdownd_gen_pair_cert(char *public_key_b64, char **device_cert_b64, char **host_cert_b64, 671iphone_error_t lockdownd_gen_pair_cert(gnutls_datum_t public_key, gnutls_datum_t * odevice_cert,
684 char **root_cert_b64) 672 gnutls_datum_t * ohost_cert, gnutls_datum_t * oroot_cert)
685{ 673{
686 if (!public_key_b64 || !device_cert_b64 || !host_cert_b64 || !root_cert_b64) 674 if (!public_key.data || !odevice_cert || !ohost_cert || !oroot_cert)
687 return IPHONE_E_INVALID_ARG; 675 return IPHONE_E_INVALID_ARG;
688 iphone_error_t ret = IPHONE_E_UNKNOWN_ERROR; 676 iphone_error_t ret = IPHONE_E_UNKNOWN_ERROR;
689 677
690 gnutls_datum_t modulus = { NULL, 0 }; 678 gnutls_datum_t modulus = { NULL, 0 };
691 gnutls_datum_t exponent = { NULL, 0 }; 679 gnutls_datum_t exponent = { NULL, 0 };
692 680
693 /* first decode base64 public_key */
694 gnutls_datum_t pem_pub_key;
695 gsize decoded_size;
696 pem_pub_key.data = g_base64_decode(public_key_b64, &decoded_size);
697 pem_pub_key.size = decoded_size;
698
699 /* now decode the PEM encoded key */ 681 /* now decode the PEM encoded key */
700 gnutls_datum_t der_pub_key; 682 gnutls_datum_t der_pub_key;
701 if (GNUTLS_E_SUCCESS == gnutls_pem_base64_decode_alloc("RSA PUBLIC KEY", &pem_pub_key, &der_pub_key)) { 683 if (GNUTLS_E_SUCCESS == gnutls_pem_base64_decode_alloc("RSA PUBLIC KEY", &public_key, &der_pub_key)) {
702 684
703 /* initalize asn.1 parser */ 685 /* initalize asn.1 parser */
704 ASN1_TYPE pkcs1 = ASN1_TYPE_EMPTY; 686 ASN1_TYPE pkcs1 = ASN1_TYPE_EMPTY;
@@ -782,10 +764,18 @@ iphone_error_t lockdownd_gen_pair_cert(char *public_key_b64, char **device_cert_
782 dev_pem.data = gnutls_malloc(dev_pem.size); 764 dev_pem.data = gnutls_malloc(dev_pem.size);
783 gnutls_x509_crt_export(dev_cert, GNUTLS_X509_FMT_PEM, dev_pem.data, &dev_pem.size); 765 gnutls_x509_crt_export(dev_cert, GNUTLS_X509_FMT_PEM, dev_pem.data, &dev_pem.size);
784 766
785 /* now encode certificates for output */ 767 /* copy buffer for output */
786 *device_cert_b64 = g_base64_encode(dev_pem.data, dev_pem.size); 768 odevice_cert->data = malloc(dev_pem.size);
787 *host_cert_b64 = g_base64_encode(pem_host_cert.data, pem_host_cert.size); 769 memcpy(odevice_cert->data, dev_pem.data, dev_pem.size);
788 *root_cert_b64 = g_base64_encode(pem_root_cert.data, pem_root_cert.size); 770 odevice_cert->size = dev_pem.size;
771
772 ohost_cert->data = malloc(pem_host_cert.size);
773 memcpy(ohost_cert->data, pem_host_cert.data, pem_host_cert.size);
774 ohost_cert->size = pem_host_cert.size;
775
776 oroot_cert->data = malloc(pem_root_cert.size);
777 memcpy(oroot_cert->data, pem_root_cert.data, pem_root_cert.size);
778 oroot_cert->size = pem_root_cert.size;
789 } 779 }
790 gnutls_free(pem_root_priv.data); 780 gnutls_free(pem_root_priv.data);
791 gnutls_free(pem_root_cert.data); 781 gnutls_free(pem_root_cert.data);
@@ -797,7 +787,6 @@ iphone_error_t lockdownd_gen_pair_cert(char *public_key_b64, char **device_cert_
797 gnutls_free(exponent.data); 787 gnutls_free(exponent.data);
798 788
799 gnutls_free(der_pub_key.data); 789 gnutls_free(der_pub_key.data);
800 g_free(pem_pub_key.data);
801 790
802 return ret; 791 return ret;
803} 792}
@@ -811,129 +800,127 @@ iphone_error_t lockdownd_gen_pair_cert(char *public_key_b64, char **device_cert_
811 */ 800 */
812iphone_error_t lockdownd_start_SSL_session(iphone_lckd_client_t control, const char *HostID) 801iphone_error_t lockdownd_start_SSL_session(iphone_lckd_client_t control, const char *HostID)
813{ 802{
814 xmlDocPtr plist = new_plist(); 803 plist_t dict = NULL;
815 xmlNode *dict = add_child_to_plist(plist, "dict", "\n", NULL, 0); 804 uint32_t return_me = 0;
816 xmlNode *key;
817 char *what2send = NULL, **dictionary = NULL;
818 uint32 len = 0, bytes = 0, return_me = 0, i = 0;
819 iphone_error_t ret = IPHONE_E_UNKNOWN_ERROR;
820 // end variables
821 805
806 iphone_error_t ret = IPHONE_E_UNKNOWN_ERROR;
822 control->session_id[0] = '\0'; 807 control->session_id[0] = '\0';
823 808
824 key = add_key_str_dict_element(plist, dict, "HostID", HostID, 1); 809 /* Setup DevicePublicKey request plist */
825 if (!key) { 810 dict = plist_new_dict();
826 log_debug_msg("Couldn't add a key.\n"); 811 plist_add_sub_key_el(dict, "HostID");
827 xmlFreeDoc(plist); 812 plist_add_sub_string_el(dict, HostID);
828 return IPHONE_E_DICT_ERROR; 813 plist_add_sub_key_el(dict, "Request");
829 } 814 plist_add_sub_string_el(dict, "StartSession");
830 key = add_key_str_dict_element(plist, dict, "Request", "StartSession", 1);
831 if (!key) {
832 log_debug_msg("Couldn't add a key.\n");
833 xmlFreeDoc(plist);
834 return IPHONE_E_DICT_ERROR;
835 }
836
837 xmlDocDumpMemory(plist, (xmlChar **) & what2send, &len);
838 ret = iphone_lckd_send(control, what2send, len, &bytes);
839 815
840 xmlFree(what2send); 816 ret = iphone_lckd_send(control, dict);
841 xmlFreeDoc(plist); 817 plist_free(dict);
818 dict = NULL;
842 819
843 if (ret != IPHONE_E_SUCCESS) 820 if (ret != IPHONE_E_SUCCESS)
844 return ret; 821 return ret;
845 822
846 if (bytes > 0) { 823 ret = iphone_lckd_recv(control, &dict);
847 ret = iphone_lckd_recv(control, &what2send, &len); 824
848 plist = xmlReadMemory(what2send, len, NULL, NULL, 0); 825 if (!dict)
849 dict = xmlDocGetRootElement(plist); 826 return IPHONE_E_PLIST_ERROR;
850 if (!dict) 827
851 return IPHONE_E_DICT_ERROR; 828 plist_t query_node = plist_find_node_by_string(dict, "StartSession");
852 for (dict = dict->children; dict; dict = dict->next) { 829 plist_t result_key_node = plist_get_next_sibling(query_node);
853 if (!xmlStrcmp(dict->name, "dict")) 830 plist_t result_value_node = plist_get_next_sibling(result_key_node);
854 break; 831
855 } 832 plist_type result_key_type = plist_get_node_type(result_key_node);
856 dictionary = read_dict_element_strings(dict); 833 plist_type result_value_type = plist_get_node_type(result_value_node);
857 xmlFreeDoc(plist); 834
858 free(what2send); 835 if (result_key_type == PLIST_KEY && result_value_type == PLIST_STRING) {
836 char *result_key = NULL;
837 char *result_value = NULL;
838
839 plist_get_key_val(result_key_node, &result_key);
840 plist_get_string_val(result_value_node, &result_value);
841
859 ret = IPHONE_E_SSL_ERROR; 842 ret = IPHONE_E_SSL_ERROR;
860 for (i = 0; dictionary[i]; i += 2) { 843 if (!strcmp(result_key, "Result") && !strcmp(result_value, "Success")) {
861 if (!strcmp(dictionary[i], "Result") && !strcmp(dictionary[i + 1], "Success")) { 844 // Set up GnuTLS...
862 // Set up GnuTLS... 845 //gnutls_anon_client_credentials_t anoncred;
863 //gnutls_anon_client_credentials_t anoncred; 846 gnutls_certificate_credentials_t xcred;
864 gnutls_certificate_credentials_t xcred; 847
865 848 log_dbg_msg(DBGMASK_LOCKDOWND, "We started the session OK, now trying GnuTLS\n");
866 log_debug_msg("We started the session OK, now trying GnuTLS\n"); 849 errno = 0;
867 errno = 0; 850 gnutls_global_init();
868 gnutls_global_init(); 851 //gnutls_anon_allocate_client_credentials(&anoncred);
869 //gnutls_anon_allocate_client_credentials(&anoncred); 852 gnutls_certificate_allocate_credentials(&xcred);
870 gnutls_certificate_allocate_credentials(&xcred); 853 gnutls_certificate_set_x509_trust_file(xcred, "hostcert.pem", GNUTLS_X509_FMT_PEM);
871 gnutls_certificate_set_x509_trust_file(xcred, "hostcert.pem", GNUTLS_X509_FMT_PEM); 854 gnutls_init(control->ssl_session, GNUTLS_CLIENT);
872 gnutls_init(control->ssl_session, GNUTLS_CLIENT); 855 {
873 { 856 int protocol_priority[16] = { GNUTLS_SSL3, 0 };
874 int protocol_priority[16] = { GNUTLS_SSL3, 0 }; 857 int kx_priority[16] = { GNUTLS_KX_ANON_DH, GNUTLS_KX_RSA, 0 };
875 int kx_priority[16] = { GNUTLS_KX_ANON_DH, GNUTLS_KX_RSA, 0 }; 858 int cipher_priority[16] = { GNUTLS_CIPHER_AES_128_CBC, GNUTLS_CIPHER_AES_256_CBC, 0 };
876 int cipher_priority[16] = { GNUTLS_CIPHER_AES_128_CBC, GNUTLS_CIPHER_AES_256_CBC, 0 }; 859 int mac_priority[16] = { GNUTLS_MAC_SHA1, GNUTLS_MAC_MD5, 0 };
877 int mac_priority[16] = { GNUTLS_MAC_SHA1, GNUTLS_MAC_MD5, 0 }; 860 int comp_priority[16] = { GNUTLS_COMP_NULL, 0 };
878 int comp_priority[16] = { GNUTLS_COMP_NULL, 0 }; 861
879 862 gnutls_cipher_set_priority(*control->ssl_session, cipher_priority);
880 gnutls_cipher_set_priority(*control->ssl_session, cipher_priority); 863 gnutls_compression_set_priority(*control->ssl_session, comp_priority);
881 gnutls_compression_set_priority(*control->ssl_session, comp_priority); 864 gnutls_kx_set_priority(*control->ssl_session, kx_priority);
882 gnutls_kx_set_priority(*control->ssl_session, kx_priority); 865 gnutls_protocol_set_priority(*control->ssl_session, protocol_priority);
883 gnutls_protocol_set_priority(*control->ssl_session, protocol_priority); 866 gnutls_mac_set_priority(*control->ssl_session, mac_priority);
884 gnutls_mac_set_priority(*control->ssl_session, mac_priority); 867
885 868 }
886 } 869 gnutls_credentials_set(*control->ssl_session, GNUTLS_CRD_CERTIFICATE, xcred); // this part is killing me.
887 gnutls_credentials_set(*control->ssl_session, GNUTLS_CRD_CERTIFICATE, xcred); // this part is killing me. 870
888 871 log_dbg_msg(DBGMASK_LOCKDOWND, "GnuTLS step 1...\n");
889 log_debug_msg("GnuTLS step 1...\n"); 872 gnutls_transport_set_ptr(*control->ssl_session, (gnutls_transport_ptr_t) control);
890 gnutls_transport_set_ptr(*control->ssl_session, (gnutls_transport_ptr_t) control); 873 log_dbg_msg(DBGMASK_LOCKDOWND, "GnuTLS step 2...\n");
891 log_debug_msg("GnuTLS step 2...\n"); 874 gnutls_transport_set_push_function(*control->ssl_session, (gnutls_push_func) & lockdownd_secuwrite);
892 gnutls_transport_set_push_function(*control->ssl_session, (gnutls_push_func) & lockdownd_secuwrite); 875 log_dbg_msg(DBGMASK_LOCKDOWND, "GnuTLS step 3...\n");
893 log_debug_msg("GnuTLS step 3...\n"); 876 gnutls_transport_set_pull_function(*control->ssl_session, (gnutls_pull_func) & lockdownd_securead);
894 gnutls_transport_set_pull_function(*control->ssl_session, (gnutls_pull_func) & lockdownd_securead); 877 log_dbg_msg(DBGMASK_LOCKDOWND, "GnuTLS step 4 -- now handshaking...\n");
895 log_debug_msg("GnuTLS step 4 -- now handshaking...\n"); 878
896 879 if (errno)
897 if (errno) 880 log_dbg_msg(DBGMASK_LOCKDOWND, "WARN: errno says %s before handshake!\n", strerror(errno));
898 log_debug_msg("WARN: errno says %s before handshake!\n", strerror(errno)); 881 return_me = gnutls_handshake(*control->ssl_session);
899 return_me = gnutls_handshake(*control->ssl_session); 882 log_dbg_msg(DBGMASK_LOCKDOWND, "GnuTLS handshake done...\n");
900 log_debug_msg("GnuTLS handshake done...\n"); 883
901 884 if (return_me != GNUTLS_E_SUCCESS) {
902 if (return_me != GNUTLS_E_SUCCESS) { 885 log_dbg_msg(DBGMASK_LOCKDOWND, "GnuTLS reported something wrong.\n");
903 log_debug_msg("GnuTLS reported something wrong.\n"); 886 gnutls_perror(return_me);
904 gnutls_perror(return_me); 887 log_dbg_msg(DBGMASK_LOCKDOWND, "oh.. errno says %s\n", strerror(errno));
905 log_debug_msg("oh.. errno says %s\n", strerror(errno)); 888 return IPHONE_E_SSL_ERROR;
906 return IPHONE_E_SSL_ERROR; 889 } else {
907 } else { 890 control->in_SSL = 1;
908 control->in_SSL = 1; 891 ret = IPHONE_E_SUCCESS;
909 ret = IPHONE_E_SUCCESS;
910 }
911 } else if (!strcmp(dictionary[i], "SessionID")) {
912 // we need to store the session ID for StopSession
913 strcpy(control->session_id, dictionary[i + 1]);
914 log_debug_msg("SessionID: %s\n", control->session_id);
915 free_dictionary(dictionary);
916 return ret;
917 } 892 }
918 } 893 }
919 if (ret == IPHONE_E_SUCCESS) { 894 }
920 log_debug_msg("Failed to get SessionID!\n"); 895 //store session id
921 return ret; 896 plist_t session_node = plist_find_node_by_key(dict, "SessionID");
922 } 897 if (session_node) {
898
899 plist_t session_node_val = plist_get_next_sibling(session_node);
900 plist_type session_node_val_type = plist_get_node_type(session_node_val);
923 901
924 log_debug_msg("Apparently failed negotiating with lockdownd.\n"); 902 if (session_node_val_type == PLIST_STRING) {
925 log_debug_msg("Responding dictionary: \n"); 903
926 for (i = 0; dictionary[i]; i += 2) { 904 char *session_id = NULL;
927 log_debug_msg("\t%s: %s\n", dictionary[i], dictionary[i + 1]); 905 plist_get_string_val(session_node_val, &session_id);
906
907 if (session_node_val_type == PLIST_STRING && session_id) {
908 // we need to store the session ID for StopSession
909 strcpy(control->session_id, session_id);
910 log_dbg_msg(DBGMASK_LOCKDOWND, "SessionID: %s\n", control->session_id);
911 }
912 free(session_id);
928 } 913 }
914 } else
915 log_dbg_msg(DBGMASK_LOCKDOWND, "Failed to get SessionID!\n");
916 plist_free(dict);
917 dict = NULL;
929 918
919 if (ret == IPHONE_E_SUCCESS)
920 return ret;
930 921
931 free_dictionary(dictionary); 922 log_dbg_msg(DBGMASK_LOCKDOWND, "Apparently failed negotiating with lockdownd.\n");
932 return IPHONE_E_SSL_ERROR; 923 return IPHONE_E_SSL_ERROR;
933 } else {
934 log_debug_msg("Didn't get enough bytes.\n");
935 return IPHONE_E_NOT_ENOUGH_DATA;
936 }
937} 924}
938 925
939/** gnutls callback for writing data to the iPhone. 926/** gnutls callback for writing data to the iPhone.
@@ -949,10 +936,10 @@ ssize_t lockdownd_secuwrite(gnutls_transport_ptr_t transport, char *buffer, size
949 int bytes = 0; 936 int bytes = 0;
950 iphone_lckd_client_t control; 937 iphone_lckd_client_t control;
951 control = (iphone_lckd_client_t) transport; 938 control = (iphone_lckd_client_t) transport;
952 log_debug_msg("lockdownd_secuwrite() called\n"); 939 log_dbg_msg(DBGMASK_LOCKDOWND, "lockdownd_secuwrite() called\n");
953 log_debug_msg("pre-send\nlength = %zi\n", length); 940 log_dbg_msg(DBGMASK_LOCKDOWND, "pre-send\nlength = %zi\n", length);
954 iphone_mux_send(control->connection, buffer, length, &bytes); 941 iphone_mux_send(control->connection, buffer, length, &bytes);
955 log_debug_msg("post-send\nsent %i bytes\n", bytes); 942 log_dbg_msg(DBGMASK_LOCKDOWND, "post-send\nsent %i bytes\n", bytes);
956 943
957 dump_debug_buffer("sslpacketwrite.out", buffer, length); 944 dump_debug_buffer("sslpacketwrite.out", buffer, length);
958 return bytes; 945 return bytes;
@@ -972,7 +959,7 @@ ssize_t lockdownd_securead(gnutls_transport_ptr_t transport, char *buffer, size_
972 char *hackhackhack = NULL; 959 char *hackhackhack = NULL;
973 iphone_lckd_client_t control; 960 iphone_lckd_client_t control;
974 control = (iphone_lckd_client_t) transport; 961 control = (iphone_lckd_client_t) transport;
975 log_debug_msg("lockdownd_securead() called\nlength = %zi\n", length); 962 log_dbg_msg(DBGMASK_LOCKDOWND, "lockdownd_securead() called\nlength = %zi\n", length);
976 // Buffering hack! Throw what we've got in our "buffer" into the stream first, then get more. 963 // Buffering hack! Throw what we've got in our "buffer" into the stream first, then get more.
977 if (control->gtls_buffer_hack_len > 0) { 964 if (control->gtls_buffer_hack_len > 0) {
978 if (length > control->gtls_buffer_hack_len) { // If it's asking for more than we got 965 if (length > control->gtls_buffer_hack_len) { // If it's asking for more than we got
@@ -981,7 +968,7 @@ ssize_t lockdownd_securead(gnutls_transport_ptr_t transport, char *buffer, size_
981 memcpy(buffer, control->gtls_buffer_hack, control->gtls_buffer_hack_len); // Fill their buffer partially 968 memcpy(buffer, control->gtls_buffer_hack, control->gtls_buffer_hack_len); // Fill their buffer partially
982 free(control->gtls_buffer_hack); // free our memory, it's not chained anymore 969 free(control->gtls_buffer_hack); // free our memory, it's not chained anymore
983 control->gtls_buffer_hack_len = 0; // we don't have a hack buffer anymore 970 control->gtls_buffer_hack_len = 0; // we don't have a hack buffer anymore
984 log_debug_msg("Did a partial fill to help quench thirst for data\n"); 971 log_dbg_msg(DBGMASK_LOCKDOWND, "Did a partial fill to help quench thirst for data\n");
985 } else if (length < control->gtls_buffer_hack_len) { // If it's asking for less... 972 } else if (length < control->gtls_buffer_hack_len) { // If it's asking for less...
986 control->gtls_buffer_hack_len -= length; // subtract what they're asking for 973 control->gtls_buffer_hack_len -= length; // subtract what they're asking for
987 memcpy(buffer, control->gtls_buffer_hack, length); // fill their buffer 974 memcpy(buffer, control->gtls_buffer_hack, length); // fill their buffer
@@ -990,33 +977,34 @@ ssize_t lockdownd_securead(gnutls_transport_ptr_t transport, char *buffer, size_
990 free(control->gtls_buffer_hack); // Free the old one 977 free(control->gtls_buffer_hack); // Free the old one
991 control->gtls_buffer_hack = hackhackhack; // And make it the new one. 978 control->gtls_buffer_hack = hackhackhack; // And make it the new one.
992 hackhackhack = NULL; 979 hackhackhack = NULL;
993 log_debug_msg("Quenched the thirst for data; new hack length is %i\n", control->gtls_buffer_hack_len); 980 log_dbg_msg(DBGMASK_LOCKDOWND, "Quenched the thirst for data; new hack length is %i\n",
981 control->gtls_buffer_hack_len);
994 return length; // hand it over. 982 return length; // hand it over.
995 } else { // length == hack length 983 } else { // length == hack length
996 memcpy(buffer, control->gtls_buffer_hack, length); // copy our buffer into theirs 984 memcpy(buffer, control->gtls_buffer_hack, length); // copy our buffer into theirs
997 free(control->gtls_buffer_hack); // free our "obligation" 985 free(control->gtls_buffer_hack); // free our "obligation"
998 control->gtls_buffer_hack_len = 0; // free our "obligation" 986 control->gtls_buffer_hack_len = 0; // free our "obligation"
999 log_debug_msg("Satiated the thirst for data; now we have to eventually receive again.\n"); 987 log_dbg_msg(DBGMASK_LOCKDOWND, "Satiated the thirst for data; now we have to eventually receive again.\n");
1000 return length; // hand it over 988 return length; // hand it over
1001 } 989 }
1002 } 990 }
1003 // End buffering hack! 991 // End buffering hack!
1004 char *recv_buffer = (char *) malloc(sizeof(char) * (length * 1000)); // ensuring nothing stupid happens 992 char *recv_buffer = (char *) malloc(sizeof(char) * (length * 1000)); // ensuring nothing stupid happens
1005 993
1006 log_debug_msg("pre-read\nclient wants %zi bytes\n", length); 994 log_dbg_msg(DBGMASK_LOCKDOWND, "pre-read\nclient wants %zi bytes\n", length);
1007 iphone_mux_recv(control->connection, recv_buffer, (length * 1000), &bytes); 995 iphone_mux_recv(control->connection, recv_buffer, (length * 1000), &bytes);
1008 log_debug_msg("post-read\nwe got %i bytes\n", bytes); 996 log_dbg_msg(DBGMASK_LOCKDOWND, "post-read\nwe got %i bytes\n", bytes);
1009 if (bytes < 0) { 997 if (bytes < 0) {
1010 log_debug_msg("lockdownd_securead(): uh oh\n"); 998 log_dbg_msg(DBGMASK_LOCKDOWND, "lockdownd_securead(): uh oh\n");
1011 log_debug_msg 999 log_dbg_msg(DBGMASK_LOCKDOWND,
1012 ("I believe what we have here is a failure to communicate... libusb says %s but strerror says %s\n", 1000 "I believe what we have here is a failure to communicate... libusb says %s but strerror says %s\n",
1013 usb_strerror(), strerror(errno)); 1001 usb_strerror(), strerror(errno));
1014 return bytes + 28; // an errno 1002 return bytes + 28; // an errno
1015 } 1003 }
1016 if (bytes >= length) { 1004 if (bytes >= length) {
1017 if (bytes > length) { 1005 if (bytes > length) {
1018 log_debug_msg 1006 log_dbg_msg(DBGMASK_LOCKDOWND,
1019 ("lockdownd_securead: Client deliberately read less data than was there; resorting to GnuTLS buffering hack.\n"); 1007 "lockdownd_securead: Client deliberately read less data than was there; resorting to GnuTLS buffering hack.\n");
1020 if (!control->gtls_buffer_hack_len) { // if there's no hack buffer yet 1008 if (!control->gtls_buffer_hack_len) { // if there's no hack buffer yet
1021 //control->gtls_buffer_hack = strndup(recv_buffer+length, bytes-length); // strndup is NOT a good solution! 1009 //control->gtls_buffer_hack = strndup(recv_buffer+length, bytes-length); // strndup is NOT a good solution!
1022 control->gtls_buffer_hack_len += bytes - length; 1010 control->gtls_buffer_hack_len += bytes - length;
@@ -1032,10 +1020,11 @@ ssize_t lockdownd_securead(gnutls_transport_ptr_t transport, char *buffer, size_
1032 memcpy(buffer + pos_start_fill, recv_buffer, length); 1020 memcpy(buffer + pos_start_fill, recv_buffer, length);
1033 free(recv_buffer); 1021 free(recv_buffer);
1034 if (bytes == length) { 1022 if (bytes == length) {
1035 log_debug_msg("Returning how much we received.\n"); 1023 log_dbg_msg(DBGMASK_LOCKDOWND, "Returning how much we received.\n");
1036 return bytes; 1024 return bytes;
1037 } else { 1025 } else {
1038 log_debug_msg("Returning what they want to hear.\nHack length: %i\n", control->gtls_buffer_hack_len); 1026 log_dbg_msg(DBGMASK_LOCKDOWND, "Returning what they want to hear.\nHack length: %i\n",
1027 control->gtls_buffer_hack_len);
1039 return length; 1028 return length;
1040 } 1029 }
1041 } 1030 }
@@ -1060,86 +1049,72 @@ iphone_error_t iphone_lckd_start_service(iphone_lckd_client_t client, const char
1060 if (!client->in_SSL && !lockdownd_start_SSL_session(client, host_id)) 1049 if (!client->in_SSL && !lockdownd_start_SSL_session(client, host_id))
1061 return IPHONE_E_SSL_ERROR; 1050 return IPHONE_E_SSL_ERROR;
1062 1051
1063 char *XML_query, **dictionary; 1052 plist_t dict = NULL;
1064 uint32 length, i = 0, port_loc = 0, bytes = 0; 1053 uint32_t port_loc = 0;
1065 uint8 result = 0;
1066 iphone_error_t ret = IPHONE_E_UNKNOWN_ERROR; 1054 iphone_error_t ret = IPHONE_E_UNKNOWN_ERROR;
1067 1055
1068 free(host_id); 1056 free(host_id);
1069 host_id = NULL; 1057 host_id = NULL;
1070 1058
1071 xmlDocPtr plist = new_plist(); 1059 dict = plist_new_dict();
1072 xmlNode *dict = add_child_to_plist(plist, "dict", "\n", NULL, 0); 1060 plist_add_sub_key_el(dict, "Request");
1073 xmlNode *key; 1061 plist_add_sub_string_el(dict, "StartService");
1074 key = add_key_str_dict_element(plist, dict, "Request", "StartService", 1); 1062 plist_add_sub_key_el(dict, "Service");
1075 if (!key) { 1063 plist_add_sub_string_el(dict, service);
1076 xmlFreeDoc(plist);
1077 return IPHONE_E_UNKNOWN_ERROR;
1078 }
1079 key = add_key_str_dict_element(plist, dict, "Service", service, 1);
1080 if (!key) {
1081 xmlFreeDoc(plist);
1082 return IPHONE_E_UNKNOWN_ERROR;
1083 }
1084 1064
1085 xmlDocDumpMemory(plist, (xmlChar **) & XML_query, &length); 1065 /* send to iPhone */
1066 ret = iphone_lckd_send(client, dict);
1067 plist_free(dict);
1068 dict = NULL;
1086 1069
1087 ret = iphone_lckd_send(client, XML_query, length, &bytes);
1088 free(XML_query);
1089 if (IPHONE_E_SUCCESS != ret) 1070 if (IPHONE_E_SUCCESS != ret)
1090 return ret; 1071 return ret;
1091 1072
1092 ret = iphone_lckd_recv(client, &XML_query, &bytes); 1073 ret = iphone_lckd_recv(client, &dict);
1093 xmlFreeDoc(plist); 1074
1094 if (IPHONE_E_SUCCESS != ret) 1075 if (IPHONE_E_SUCCESS != ret)
1095 return ret; 1076 return ret;
1096 1077
1097 if (bytes <= 0) 1078 if (!dict)
1098 return IPHONE_E_NOT_ENOUGH_DATA; 1079 return IPHONE_E_PLIST_ERROR;
1099 else {
1100 plist = xmlReadMemory(XML_query, bytes, NULL, NULL, 0);
1101 if (!plist)
1102 return IPHONE_E_UNKNOWN_ERROR;
1103 dict = xmlDocGetRootElement(plist);
1104 if (!dict)
1105 return IPHONE_E_UNKNOWN_ERROR;
1106 for (dict = dict->children; dict; dict = dict->next) {
1107 if (!xmlStrcmp(dict->name, "dict"))
1108 break;
1109 }
1110 1080
1111 if (!dict) 1081 plist_t query_node = plist_find_node_by_string(dict, "StartService");
1112 return IPHONE_E_UNKNOWN_ERROR; 1082 plist_t result_key_node = plist_get_next_sibling(query_node);
1113 dictionary = read_dict_element_strings(dict); 1083 plist_t result_value_node = plist_get_next_sibling(result_key_node);
1114 1084
1115 for (i = 0; dictionary[i]; i += 2) { 1085 plist_t port_key_node = plist_find_node_by_key(dict, "Port");
1116 log_debug_msg("lockdownd_start_service() dictionary %s: %s\n", dictionary[i], dictionary[i + 1]); 1086 plist_t port_value_node = plist_get_next_sibling(port_key_node);
1117 1087
1118 if (!xmlStrcmp(dictionary[i], "Port")) { 1088 plist_type result_key_type = plist_get_node_type(result_key_node);
1119 port_loc = atoi(dictionary[i + 1]); 1089 plist_type result_value_type = plist_get_node_type(result_value_node);
1120 log_debug_msg("lockdownd_start_service() atoi'd port: %i\n", port); 1090 plist_type port_key_type = plist_get_node_type(port_key_node);
1121 } 1091 plist_type port_value_type = plist_get_node_type(port_value_node);
1122 1092
1123 if (!xmlStrcmp(dictionary[i], "Result")) { 1093 if (result_key_type == PLIST_KEY && result_value_type == PLIST_STRING && port_key_type == PLIST_KEY
1124 if (!xmlStrcmp(dictionary[i + 1], "Success")) { 1094 && port_value_type == PLIST_UINT) {
1125 result = 1; 1095
1126 } 1096 char *result_key = NULL;
1127 } 1097 char *result_value = NULL;
1128 } 1098 char *port_key = NULL;
1099 uint64_t port_value = 0;
1129 1100
1130 log_debug_msg("lockdownd_start_service(): DATA RECEIVED:\n\n"); 1101 plist_get_key_val(result_key_node, &result_key);
1131 log_debug_msg(XML_query); 1102 plist_get_string_val(result_value_node, &result_value);
1132 log_debug_msg("end data received by lockdownd_start_service()\n"); 1103 plist_get_key_val(port_key_node, &port_key);
1104 plist_get_uint_val(port_value_node, &port_value);
1133 1105
1134 free(XML_query); 1106 if (!strcmp(result_key, "Result") && !strcmp(result_value, "Success") && !strcmp(port_key, "Port")) {
1135 xmlFreeDoc(plist); 1107 port_loc = port_value;
1136 free_dictionary(dictionary); 1108 ret = IPHONE_E_SUCCESS;
1137 if (port && result) { 1109 }
1110
1111 if (port && ret == IPHONE_E_SUCCESS)
1138 *port = port_loc; 1112 *port = port_loc;
1139 return IPHONE_E_SUCCESS; 1113 else
1140 } else 1114 ret = IPHONE_E_UNKNOWN_ERROR;
1141 return IPHONE_E_UNKNOWN_ERROR;
1142 } 1115 }
1143 1116
1144 return IPHONE_E_UNKNOWN_ERROR; 1117 plist_free(dict);
1118 dict = NULL;
1119 return ret;
1145} 1120}