blob: cde96b57ef9d40fe1c19b1620271881ae302efb9 [file] [log] [blame] [edit]
/*
* mobilebackup.c
* Contains functions for the built-in MobileBackup client.
*
* Copyright (c) 2010-2019 Nikias Bassen, All Rights Reserved.
* Copyright (c) 2009 Martin Szulecki All Rights Reserved.
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include <plist/plist.h>
#include <string.h>
#include <stdlib.h>
#include "mobilebackup.h"
#include "device_link_service.h"
#include "common/debug.h"
#define MBACKUP_VERSION_INT1 100
#define MBACKUP_VERSION_INT2 0
#define IS_FLAG_SET(x, y) ((x & y) == y)
/**
* Convert an device_link_service_error_t value to an mobilebackup_error_t value.
* Used internally to get correct error codes when using device_link_service stuff.
*
* @param err An device_link_service_error_t error code
*
* @return A matching mobilebackup_error_t error code,
* MOBILEBACKUP_E_UNKNOWN_ERROR otherwise.
*/
static mobilebackup_error_t mobilebackup_error(device_link_service_error_t err)
{
switch (err) {
case DEVICE_LINK_SERVICE_E_SUCCESS:
return MOBILEBACKUP_E_SUCCESS;
case DEVICE_LINK_SERVICE_E_INVALID_ARG:
return MOBILEBACKUP_E_INVALID_ARG;
case DEVICE_LINK_SERVICE_E_PLIST_ERROR:
return MOBILEBACKUP_E_PLIST_ERROR;
case DEVICE_LINK_SERVICE_E_MUX_ERROR:
return MOBILEBACKUP_E_MUX_ERROR;
case DEVICE_LINK_SERVICE_E_SSL_ERROR:
return MOBILEBACKUP_E_SSL_ERROR;
case DEVICE_LINK_SERVICE_E_RECEIVE_TIMEOUT:
return MOBILEBACKUP_E_RECEIVE_TIMEOUT;
case DEVICE_LINK_SERVICE_E_BAD_VERSION:
return MOBILEBACKUP_E_BAD_VERSION;
default:
break;
}
return MOBILEBACKUP_E_UNKNOWN_ERROR;
}
LIBIMOBILEDEVICE_API mobilebackup_error_t mobilebackup_client_new(idevice_t device, lockdownd_service_descriptor_t service, mobilebackup_client_t * client)
{
if (!device || !service || service->port == 0 || !client || *client)
return MOBILEBACKUP_E_INVALID_ARG;
device_link_service_client_t dlclient = NULL;
mobilebackup_error_t ret = mobilebackup_error(device_link_service_client_new(device, service, &dlclient));
if (ret != MOBILEBACKUP_E_SUCCESS) {
return ret;
}
mobilebackup_client_t client_loc = (mobilebackup_client_t) malloc(sizeof(struct mobilebackup_client_private));
client_loc->parent = dlclient;
/* perform handshake */
ret = mobilebackup_error(device_link_service_version_exchange(dlclient, MBACKUP_VERSION_INT1, MBACKUP_VERSION_INT2));
if (ret != MOBILEBACKUP_E_SUCCESS) {
debug_info("version exchange failed, error %d", ret);
mobilebackup_client_free(client_loc);
return ret;
}
*client = client_loc;
return ret;
}
LIBIMOBILEDEVICE_API mobilebackup_error_t mobilebackup_client_start_service(idevice_t device, mobilebackup_client_t * client, const char* label)
{
mobilebackup_error_t err = MOBILEBACKUP_E_UNKNOWN_ERROR;
service_client_factory_start_service(device, MOBILEBACKUP_SERVICE_NAME, (void**)client, label, SERVICE_CONSTRUCTOR(mobilebackup_client_new), &err);
return err;
}
LIBIMOBILEDEVICE_API mobilebackup_error_t mobilebackup_client_free(mobilebackup_client_t client)
{
if (!client)
return MOBILEBACKUP_E_INVALID_ARG;
mobilebackup_error_t err = MOBILEBACKUP_E_SUCCESS;
if (client->parent) {
device_link_service_disconnect(client->parent, NULL);
err = mobilebackup_error(device_link_service_client_free(client->parent));
}
free(client);
return err;
}
LIBIMOBILEDEVICE_API mobilebackup_error_t mobilebackup_receive(mobilebackup_client_t client, plist_t * plist)
{
if (!client)
return MOBILEBACKUP_E_INVALID_ARG;
mobilebackup_error_t ret = mobilebackup_error(device_link_service_receive(client->parent, plist));
return ret;
}
LIBIMOBILEDEVICE_API mobilebackup_error_t mobilebackup_send(mobilebackup_client_t client, plist_t plist)
{
if (!client || !plist)
return MOBILEBACKUP_E_INVALID_ARG;
return mobilebackup_error(device_link_service_send(client->parent, plist));
}
/**
* Sends a backup message plist.
*
* @param client The connected MobileBackup client to use.
* @param message The message to send. This will be inserted into the request
* plist as value for BackupMessageTypeKey. If this parameter is NULL,
* the plist passed in the options parameter will be sent directly.
* @param options Additional options as PLIST_DICT to add to the request.
* The BackupMessageTypeKey with the value passed in the message parameter
* will be inserted into this plist before sending it. This parameter
* can be NULL if message is not NULL.
*/
static mobilebackup_error_t mobilebackup_send_message(mobilebackup_client_t client, const char *message, plist_t options)
{
if (!client || !client->parent || (!message && !options))
return MOBILEBACKUP_E_INVALID_ARG;
if (options && (plist_get_node_type(options) != PLIST_DICT)) {
return MOBILEBACKUP_E_INVALID_ARG;
}
mobilebackup_error_t err;
if (message) {
plist_t dict = NULL;
if (options) {
dict = plist_copy(options);
} else {
dict = plist_new_dict();
}
plist_dict_set_item(dict, "BackupMessageTypeKey", plist_new_string(message));
/* send it as DLMessageProcessMessage */
err = mobilebackup_error(device_link_service_send_process_message(client->parent, dict));
plist_free(dict);
} else {
err = mobilebackup_error(device_link_service_send_process_message(client->parent, options));
}
if (err != MOBILEBACKUP_E_SUCCESS) {
debug_info("ERROR: Could not send message '%s' (%d)!", message, err);
}
return err;
}
/**
* Receives a plist from the device and checks if the value for the
* BackupMessageTypeKey matches the value passed in the message parameter.
*
* @param client The connected MobileBackup client to use.
* @param message The expected message to check.
* @param result Pointer to a plist_t that will be set to the received plist
* for further processing. The caller has to free it using plist_free().
* Note that it will be set to NULL if the operation itself fails due to
* a communication or plist error.
* If this parameter is NULL, it will be ignored.
*
* @return MOBILEBACKUP_E_SUCCESS on success, MOBILEBACKUP_E_INVALID_ARG if
* client or message is invalid, MOBILEBACKUP_E_REPLY_NOT_OK if the
* expected message could not be received, MOBILEBACKUP_E_PLIST_ERROR if
* the received message is not a valid backup message plist (i.e. the
* BackupMessageTypeKey is not present), or MOBILEBACKUP_E_MUX_ERROR
* if a communication error occurs.
*/
static mobilebackup_error_t mobilebackup_receive_message(mobilebackup_client_t client, const char *message, plist_t *result)
{
if (!client || !client->parent || !message)
return MOBILEBACKUP_E_INVALID_ARG;
if (result)
*result = NULL;
mobilebackup_error_t err;
plist_t dict = NULL;
/* receive DLMessageProcessMessage */
err = mobilebackup_error(device_link_service_receive_process_message(client->parent, &dict));
if (err != MOBILEBACKUP_E_SUCCESS) {
goto leave;
}
plist_t node = plist_dict_get_item(dict, "BackupMessageTypeKey");
if (!node) {
debug_info("ERROR: BackupMessageTypeKey not found in plist!");
err = MOBILEBACKUP_E_PLIST_ERROR;
goto leave;
}
char *str = NULL;
plist_get_string_val(node, &str);
if (str && (strcmp(str, message) == 0)) {
err = MOBILEBACKUP_E_SUCCESS;
} else {
debug_info("ERROR: BackupMessageTypeKey value does not match '%s'!", message);
err = MOBILEBACKUP_E_REPLY_NOT_OK;
}
if (str)
free(str);
if (result) {
*result = dict;
dict = NULL;
}
leave:
if (dict) {
plist_free(dict);
}
return err;
}
LIBIMOBILEDEVICE_API mobilebackup_error_t mobilebackup_request_backup(mobilebackup_client_t client, plist_t backup_manifest, const char *base_path, const char *proto_version)
{
if (!client || !client->parent || !base_path || !proto_version)
return MOBILEBACKUP_E_INVALID_ARG;
if (backup_manifest && (plist_get_node_type(backup_manifest) != PLIST_DICT))
return MOBILEBACKUP_E_PLIST_ERROR;
mobilebackup_error_t err;
/* construct request plist */
plist_t dict = plist_new_dict();
if (backup_manifest)
plist_dict_set_item(dict, "BackupManifestKey", plist_copy(backup_manifest));
plist_dict_set_item(dict, "BackupComputerBasePathKey", plist_new_string(base_path));
plist_dict_set_item(dict, "BackupMessageTypeKey", plist_new_string("BackupMessageBackupRequest"));
plist_dict_set_item(dict, "BackupProtocolVersion", plist_new_string(proto_version));
/* send request */
err = mobilebackup_send_message(client, NULL, dict);
plist_free(dict);
dict = NULL;
if (err != MOBILEBACKUP_E_SUCCESS) {
debug_info("ERROR: Could not send backup request message (%d)!", err);
goto leave;
}
/* now receive and hopefully get a BackupMessageBackupReplyOK */
err = mobilebackup_receive_message(client, "BackupMessageBackupReplyOK", &dict);
if (err != MOBILEBACKUP_E_SUCCESS) {
debug_info("ERROR: Could not receive BackupReplyOK message (%d)!", err);
goto leave;
}
plist_t node = plist_dict_get_item(dict, "BackupProtocolVersion");
if (node) {
char *str = NULL;
plist_get_string_val(node, &str);
if (str) {
if (strcmp(str, proto_version) != 0) {
err = MOBILEBACKUP_E_BAD_VERSION;
}
free(str);
}
}
if (err != MOBILEBACKUP_E_SUCCESS)
goto leave;
/* BackupMessageBackupReplyOK received, send it back */
err = mobilebackup_send_message(client, NULL, dict);
if (err != MOBILEBACKUP_E_SUCCESS) {
debug_info("ERROR: Could not send BackupReplyOK ACK (%d)", err);
}
leave:
if (dict)
plist_free(dict);
return err;
}
LIBIMOBILEDEVICE_API mobilebackup_error_t mobilebackup_send_backup_file_received(mobilebackup_client_t client)
{
return mobilebackup_send_message(client, "kBackupMessageBackupFileReceived", NULL);
}
LIBIMOBILEDEVICE_API mobilebackup_error_t mobilebackup_request_restore(mobilebackup_client_t client, plist_t backup_manifest, mobilebackup_flags_t flags, const char *proto_version)
{
if (!client || !client->parent || !backup_manifest || !proto_version)
return MOBILEBACKUP_E_INVALID_ARG;
if (backup_manifest && (plist_get_node_type(backup_manifest) != PLIST_DICT))
return MOBILEBACKUP_E_PLIST_ERROR;
mobilebackup_error_t err;
/* construct request plist */
plist_t dict = plist_new_dict();
plist_dict_set_item(dict, "BackupManifestKey", plist_copy(backup_manifest));
plist_dict_set_item(dict, "BackupMessageTypeKey", plist_new_string("kBackupMessageRestoreRequest"));
plist_dict_set_item(dict, "BackupProtocolVersion", plist_new_string(proto_version));
/* add flags */
plist_dict_set_item(dict, "BackupNotifySpringBoard", plist_new_bool(IS_FLAG_SET(flags, MB_RESTORE_NOTIFY_SPRINGBOARD)));
plist_dict_set_item(dict, "BackupPreserveSettings", plist_new_bool(IS_FLAG_SET(flags, MB_RESTORE_PRESERVE_SETTINGS)));
plist_dict_set_item(dict, "BackupPreserveCameraRoll", plist_new_bool(IS_FLAG_SET(flags, MB_RESTORE_PRESERVE_CAMERA_ROLL)));
/* send request */
err = mobilebackup_send_message(client, NULL, dict);
plist_free(dict);
dict = NULL;
if (err != MOBILEBACKUP_E_SUCCESS) {
debug_info("ERROR: Could not send restore request message (%d)!", err);
goto leave;
}
/* now receive and hopefully get a BackupMessageRestoreReplyOK */
err = mobilebackup_receive_message(client, "BackupMessageRestoreReplyOK", &dict);
if (err != MOBILEBACKUP_E_SUCCESS) {
debug_info("ERROR: Could not receive RestoreReplyOK message (%d)!", err);
goto leave;
}
plist_t node = plist_dict_get_item(dict, "BackupProtocolVersion");
if (node) {
char *str = NULL;
plist_get_string_val(node, &str);
if (str) {
if (strcmp(str, proto_version) != 0) {
err = MOBILEBACKUP_E_BAD_VERSION;
}
free(str);
}
}
leave:
if (dict)
plist_free(dict);
return err;
}
LIBIMOBILEDEVICE_API mobilebackup_error_t mobilebackup_receive_restore_file_received(mobilebackup_client_t client, plist_t *result)
{
return mobilebackup_receive_message(client, "BackupMessageRestoreFileReceived", result);
}
LIBIMOBILEDEVICE_API mobilebackup_error_t mobilebackup_receive_restore_application_received(mobilebackup_client_t client, plist_t *result)
{
return mobilebackup_receive_message(client, "BackupMessageRestoreApplicationReceived", result);
}
LIBIMOBILEDEVICE_API mobilebackup_error_t mobilebackup_send_restore_complete(mobilebackup_client_t client)
{
mobilebackup_error_t err = mobilebackup_send_message(client, "BackupMessageRestoreComplete", NULL);
if (err != MOBILEBACKUP_E_SUCCESS) {
return err;
}
plist_t dlmsg = NULL;
err = mobilebackup_receive(client, &dlmsg);
if ((err != MOBILEBACKUP_E_SUCCESS) || !dlmsg || (plist_get_node_type(dlmsg) != PLIST_ARRAY) || (plist_array_get_size(dlmsg) != 2)) {
if (dlmsg) {
debug_info("ERROR: Did not receive DLMessageDisconnect:");
debug_plist(dlmsg);
plist_free(dlmsg);
}
if (err == MOBILEBACKUP_E_SUCCESS) {
err = MOBILEBACKUP_E_PLIST_ERROR;
}
return err;
}
plist_t node = plist_array_get_item (dlmsg, 0);
char *msg = NULL;
if (node && (plist_get_node_type(node) == PLIST_STRING)) {
plist_get_string_val(node, &msg);
}
if (msg && !strcmp(msg, "DLMessageDisconnect")) {
err = MOBILEBACKUP_E_SUCCESS;
/* we need to do this here, otherwise mobilebackup_client_free
will fail */
device_link_service_client_free(client->parent);
client->parent = NULL;
} else {
debug_info("ERROR: Malformed plist received:");
debug_plist(dlmsg);
err = MOBILEBACKUP_E_PLIST_ERROR;
}
plist_free(dlmsg);
if (msg)
free(msg);
return err;
}
LIBIMOBILEDEVICE_API mobilebackup_error_t mobilebackup_send_error(mobilebackup_client_t client, const char *reason)
{
if (!client || !client->parent || !reason)
return MOBILEBACKUP_E_INVALID_ARG;
mobilebackup_error_t err;
/* construct error plist */
plist_t dict = plist_new_dict();
plist_dict_set_item(dict, "BackupErrorReasonKey", plist_new_string(reason));
err = mobilebackup_send_message(client, "BackupMessageError", dict);
plist_free(dict);
return err;
}