int restore_send_filesystem(idevice_t device, const char* filesystem) {
int i = 0;
FILE* file = NULL;
plist_t data = NULL;
idevice_connection_t asr = NULL;
idevice_error_t device_error = IDEVICE_E_UNKNOWN_ERROR;
if (asr_open_with_timeout(device, &asr) < 0) {
error("ERROR: Unable to connect to ASR\n");
return -1;
}
info("Connected to ASR\n");
/* receive Initiate command message */
if (asr_receive(asr, &data) < 0) {
error("ERROR: Unable to receive data from ASR\n");
asr_close(asr);
return -1;
}
plist_free(data);
// this step sends requested chunks of data from various offsets to asr so
// it can validate the filesystem before installing it
info("Validating the filesystem\n");
if (asr_perform_validation(asr, filesystem) < 0) {
error("ERROR: ASR was unable to validate the filesystem\n");
asr_close(asr);
return -1;
}
info("Filesystem validated\n");
// once the target filesystem has been validated, ASR then requests the
// entire filesystem to be sent.
info("Sending filesystem now...\n");
if (asr_send_payload(asr, filesystem) < 0) {
error("ERROR: Unable to send payload to ASR\n");
asr_close(asr);
return -1;
}
info("Filesystem sent\n");
asr_close(asr);
return 0;
}
int asr_open_with_timeout(idevice_t device, asr_client_t* asr) {
int i = 0;
int attempts = 10;
idevice_connection_t connection = NULL;
idevice_error_t device_error = IDEVICE_E_SUCCESS;
*asr = NULL;
if (device == NULL) {
return -1;
}
debug("Connecting to ASR\n");
for (i = 1; i <= attempts; i++) {
device_error = idevice_connect(device, ASR_PORT, &connection);
if (device_error == IDEVICE_E_SUCCESS) {
break;
}
if (i >= attempts) {
error("ERROR: Unable to connect to ASR client\n");
return -1;
}
sleep(2);
debug("Retrying connection...\n");
}
asr_client_t asr_loc = (asr_client_t)malloc(sizeof(struct asr_client));
memset(asr_loc, '\0', sizeof(struct asr_client));
asr_loc->connection = connection;
/* receive Initiate command message */
plist_t data = NULL;
asr_loc->checksum_chunks = 0;
if (asr_receive(asr_loc, &data) < 0) {
error("ERROR: Unable to receive data from ASR\n");
asr_free(asr_loc);
plist_free(data);
return -1;
}
plist_t node;
node = plist_dict_get_item(data, "Command");
if (node && (plist_get_node_type(node) == PLIST_STRING)) {
char* strval = NULL;
plist_get_string_val(node, &strval);
if (strval && (strcmp(strval, "Initiate") != 0)) {
error("ERROR: unexpected ASR plist received:\n");
debug_plist(data);
plist_free(data);
asr_free(asr_loc);
return -1;
}
}
node = plist_dict_get_item(data, "Checksum Chunks");
if (node && (plist_get_node_type(node) == PLIST_BOOLEAN)) {
plist_get_bool_val(node, &(asr_loc->checksum_chunks));
}
plist_free(data);
*asr = asr_loc;
return 0;
}
int asr_perform_validation(asr_client_t asr, const char* filesystem) {
FILE* file = NULL;
uint64_t length = 0;
char* command = NULL;
plist_t node = NULL;
plist_t packet = NULL;
plist_t packet_info = NULL;
plist_t payload_info = NULL;
int attempts = 0;
file = fopen(filesystem, "rb");
if (file == NULL) {
return -1;
}
fseek(file, 0, SEEK_END);
length = ftell(file);
fseek(file, 0, SEEK_SET);
payload_info = plist_new_dict();
plist_dict_set_item(payload_info, "Port", plist_new_uint(1));
plist_dict_set_item(payload_info, "Size", plist_new_uint(length));
packet_info = plist_new_dict();
if (asr->checksum_chunks) {
plist_dict_set_item(packet_info, "Checksum Chunk Size", plist_new_uint(ASR_CHECKSUM_CHUNK_SIZE));
}
plist_dict_set_item(packet_info, "FEC Slice Stride", plist_new_uint(ASR_FEC_SLICE_STRIDE));
plist_dict_set_item(packet_info, "Packet Payload Size", plist_new_uint(ASR_PAYLOAD_PACKET_SIZE));
plist_dict_set_item(packet_info, "Packets Per FEC", plist_new_uint(ASR_PACKETS_PER_FEC));
plist_dict_set_item(packet_info, "Payload", payload_info);
plist_dict_set_item(packet_info, "Stream ID", plist_new_uint(ASR_STREAM_ID));
plist_dict_set_item(packet_info, "Version", plist_new_uint(ASR_VERSION));
if (asr_send(asr, packet_info)) {
error("ERROR: Unable to sent packet information to ASR\n");
plist_free(packet_info);
return -1;
}
plist_free(packet_info);
while (1) {
if (asr_receive(asr, &packet) < 0) {
error("ERROR: Unable to receive validation packet\n");
return -1;
}
if (packet == NULL) {
if (attempts < 5) {
info("Retrying to receive validation packet... %d\n", attempts);
attempts++;
sleep(1);
continue;
}
}
attempts = 0;
node = plist_dict_get_item(packet, "Command");
if (!node || plist_get_node_type(node) != PLIST_STRING) {
error("ERROR: Unable to find command node in validation request\n");
return -1;
}
plist_get_string_val(node, &command);
if (!strcmp(command, "OOBData")) {
asr_handle_oob_data_request(asr, packet, file);
plist_free(packet);
} else if(!strcmp(command, "Payload")) {
plist_free(packet);
break;
} else {
error("ERROR: Unknown command received from ASR\n");
plist_free(packet);
return -1;
}
}
return 0;
}
