/**
* Hangs up the connection to the mobile_image_mounter service.
* This functions has to be called before freeing up a mobile_image_mounter
* instance. If not, errors appear in the device's syslog.
*
* @param client The client to hang up
*
* @return MOBILE_IMAGE_MOUNTER_E_SUCCESS on success,
* MOBILE_IMAGE_MOUNTER_E_INVALID_ARG if client is invalid,
* or another error code otherwise.
*/
mobile_image_mounter_error_t mobile_image_mounter_hangup(mobile_image_mounter_client_t client)
{
if (!client) {
return MOBILE_IMAGE_MOUNTER_E_INVALID_ARG;
}
mobile_image_mounter_lock(client);
plist_t dict = plist_new_dict();
plist_dict_insert_item(dict, "Command", plist_new_string("Hangup"));
mobile_image_mounter_error_t res = mobile_image_mounter_error(property_list_service_send_xml_plist(client->parent, dict));
plist_free(dict);
if (res != MOBILE_IMAGE_MOUNTER_E_SUCCESS) {
debug_info("%s: Error sending XML plist to device!", __func__);
goto leave_unlock;
}
dict = NULL;
res = mobile_image_mounter_error(property_list_service_receive_plist(client->parent, &dict));
if (res != MOBILE_IMAGE_MOUNTER_E_SUCCESS) {
debug_info("%s: Error receiving response from device!", __func__);
}
if (dict) {
debug_plist(dict);
plist_free(dict);
}
leave_unlock:
mobile_image_mounter_unlock(client);
return res;
}
/**
* Mounts an image on the device.
*
* @param client The connected mobile_image_mounter client.
* @param image_path The absolute path of the image to mount. The image must
* be present before calling this function.
* @param image_signature Pointer to a buffer holding the images' signature
* @param signature_length Length of the signature image_signature points to
* @param image_type Type of image to mount
* @param result Pointer to a plist that will receive the result of the
* operation.
*
* @note This function may return MOBILE_IMAGE_MOUNTER_E_SUCCESS even if the
* operation has failed. Check the resulting plist for further information.
* Note that there is no unmounting function. The mount persists until the
* device is rebooted.
*
* @return MOBILE_IMAGE_MOUNTER_E_SUCCESS on success,
* MOBILE_IMAGE_MOUNTER_E_INVALID_ARG if on ore more parameters are
* invalid, or another error code otherwise.
*/
mobile_image_mounter_error_t mobile_image_mounter_mount_image(mobile_image_mounter_client_t client, const char *image_path, const char *image_signature, uint16_t signature_length, const char *image_type, plist_t *result)
{
if (!client || !image_path || !image_signature || (signature_length == 0) || !image_type || !result) {
return MOBILE_IMAGE_MOUNTER_E_INVALID_ARG;
}
mobile_image_mounter_lock(client);
plist_t dict = plist_new_dict();
plist_dict_insert_item(dict, "Command", plist_new_string("MountImage"));
plist_dict_insert_item(dict, "ImagePath", plist_new_string(image_path));
plist_dict_insert_item(dict, "ImageSignature", plist_new_data(image_signature, signature_length));
plist_dict_insert_item(dict, "ImageType", plist_new_string(image_type));
mobile_image_mounter_error_t res = mobile_image_mounter_error(property_list_service_send_xml_plist(client->parent, dict));
plist_free(dict);
if (res != MOBILE_IMAGE_MOUNTER_E_SUCCESS) {
debug_info("%s: Error sending XML plist to device!", __func__);
goto leave_unlock;
}
res = mobile_image_mounter_error(property_list_service_receive_plist(client->parent, result));
if (res != MOBILE_IMAGE_MOUNTER_E_SUCCESS) {
debug_info("%s: Error receiving response from device!", __func__);
}
leave_unlock:
mobile_image_mounter_unlock(client);
return res;
}
/**
* Connects to the mobile_image_mounter service on the specified device.
*
* @param device The device to connect to.
* @param service The service descriptor returned by lockdownd_start_service.
* @param client Pointer that will be set to a newly allocated
* mobile_image_mounter_client_t upon successful return.
*
* @return MOBILE_IMAGE_MOUNTER_E_SUCCESS on success,
* MOBILE_IMAGE_MOUNTER_E_INVALID_ARG if device is NULL,
* or MOBILE_IMAGE_MOUNTER_E_CONN_FAILED if the connection to the
* device could not be established.
*/
mobile_image_mounter_error_t mobile_image_mounter_new(idevice_t device, lockdownd_service_descriptor_t service, mobile_image_mounter_client_t *client)
{
property_list_service_client_t plistclient = NULL;
mobile_image_mounter_error_t err = mobile_image_mounter_error(property_list_service_client_new(device, service, &plistclient));
if (err != MOBILE_IMAGE_MOUNTER_E_SUCCESS) {
return err;
}
mobile_image_mounter_client_t client_loc = (mobile_image_mounter_client_t) malloc(sizeof(struct mobile_image_mounter_client_private));
client_loc->parent = plistclient;
mutex_init(&client_loc->mutex);
*client = client_loc;
return MOBILE_IMAGE_MOUNTER_E_SUCCESS;
}
/**
* Uploads an image to the device.
*
* @param client The connected mobile_image_mounter client.
* @param image_type Type of image that is being uploaded.
* @param image_size Total size of the image.
* @param upload_cb Callback function that gets the data chunks for uploading
* the image.
* @param userdata User defined data for the upload callback function.
*
* @return MOBILE_IMAGE_MOUNTER_E_SUCCESS on succes, or a
* MOBILE_IMAGE_MOUNTER_E_* error code otherwise.
*/
mobile_image_mounter_error_t mobile_image_mounter_upload_image(mobile_image_mounter_client_t client, const char *image_type, size_t image_size, mobile_image_mounter_upload_cb_t upload_cb, void* userdata)
{
if (!client || !image_type || (image_size == 0) || !upload_cb) {
return MOBILE_IMAGE_MOUNTER_E_INVALID_ARG;
}
mobile_image_mounter_lock(client);
plist_t result = NULL;
plist_t dict = plist_new_dict();
plist_dict_set_item(dict, "Command", plist_new_string("ReceiveBytes"));
plist_dict_set_item(dict, "ImageSize", plist_new_uint(image_size));
plist_dict_set_item(dict, "ImageType", plist_new_string(image_type));
mobile_image_mounter_error_t res = mobile_image_mounter_error(property_list_service_send_xml_plist(client->parent, dict));
plist_free(dict);
if (res != MOBILE_IMAGE_MOUNTER_E_SUCCESS) {
debug_info("Error sending XML plist to device!");
goto leave_unlock;
}
res = mobile_image_mounter_error(property_list_service_receive_plist(client->parent, &result));
if (res != MOBILE_IMAGE_MOUNTER_E_SUCCESS) {
debug_info("Error receiving response from device!");
goto leave_unlock;
}
res = MOBILE_IMAGE_MOUNTER_E_COMMAND_FAILED;
char* strval = NULL;
plist_t node = plist_dict_get_item(result, "Status");
if (node && plist_get_node_type(node) == PLIST_STRING) {
plist_get_string_val(node, &strval);
}
if (!strval) {
debug_info("Error: Unexpected response received!");
goto leave_unlock;
}
if (strcmp(strval, "ReceiveBytesAck") != 0) {
debug_info("Error: didn't get ReceiveBytesAck but %s", strval);
free(strval);
goto leave_unlock;
}
free(strval);
size_t tx = 0;
size_t bufsize = 65536;
unsigned char *buf = (unsigned char*)malloc(bufsize);
if (!buf) {
debug_info("Out of memory");
res = MOBILE_IMAGE_MOUNTER_E_UNKNOWN_ERROR;
goto leave_unlock;
}
debug_info("uploading image (%d bytes)", (int)image_size);
while (tx < image_size) {
size_t remaining = image_size - tx;
size_t amount = (remaining < bufsize) ? remaining : bufsize;
ssize_t r = upload_cb(buf, amount, userdata);
if (r < 0) {
debug_info("upload_cb returned %d", (int)r);
break;
}
uint32_t sent = 0;
if (service_send(client->parent->parent, (const char*)buf, (uint32_t)r, &sent) != SERVICE_E_SUCCESS) {
debug_info("service_send failed");
break;
}
tx += r;
}
free(buf);
if (tx < image_size) {
debug_info("Error: failed to upload image");
goto leave_unlock;
}
debug_info("image uploaded");
res = mobile_image_mounter_error(property_list_service_receive_plist(client->parent, &result));
if (res != MOBILE_IMAGE_MOUNTER_E_SUCCESS) {
debug_info("Error receiving response from device!");
goto leave_unlock;
}
res = MOBILE_IMAGE_MOUNTER_E_COMMAND_FAILED;
strval = NULL;
node = plist_dict_get_item(result, "Status");
if (node && plist_get_node_type(node) == PLIST_STRING) {
plist_get_string_val(node, &strval);
}
if (!strval) {
debug_info("Error: Unexpected response received!");
goto leave_unlock;
}
if (strcmp(strval, "Complete") != 0) {
debug_info("Error: didn't get Complete but %s", strval);
free(strval);
goto leave_unlock;
} else {
res = MOBILE_IMAGE_MOUNTER_E_SUCCESS;
}
free(strval);
leave_unlock:
mobile_image_mounter_unlock(client);
if (result)
plist_free(result);
return res;
}