diff options
Diffstat (limited to 'src/afc.c')
| -rw-r--r-- | src/afc.c | 1285 |
1 files changed, 1285 insertions, 0 deletions
diff --git a/src/afc.c b/src/afc.c new file mode 100644 index 0000000..fc09287 --- /dev/null +++ b/src/afc.c @@ -0,0 +1,1285 @@ +/* + * afc.c + * Contains functions for the built-in AFC client. + * + * Copyright (c) 2008 Zach C. 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 + */ + +#include <stdio.h> +#include <stdlib.h> +#include <unistd.h> + +#include "afc.h" +#include "iphone.h" +#include "debug.h" + +// This is the maximum size an AFC data packet can be +static const int MAXIMUM_PACKET_SIZE = (2 << 15); + +/** Locks an AFC client, done for thread safety stuff + * + * @param client The AFC client connection to lock + */ +static void afc_lock(afc_client_t client) +{ + debug_info("Locked"); + g_mutex_lock(client->mutex); +} + +/** Unlocks an AFC client, done for thread safety stuff. + * + * @param client The AFC + */ +static void afc_unlock(afc_client_t client) +{ + debug_info("Unlocked"); + g_mutex_unlock(client->mutex); +} + +/** Makes a connection to the AFC service on the phone. + * + * @param device The device to connect to. + * @param port The destination port. + * @param client Pointer that will be set to a newly allocated afc_client_t + * upon successful return. + * + * @return AFC_E_SUCCESS on success, AFC_E_INVALID_ARGUMENT when device or port + * is invalid, AFC_E_MUX_ERROR when the connection failed, or AFC_E_NO_MEM + * when there's a memory allocation problem. + */ +afc_error_t afc_client_new(iphone_device_t device, uint16_t port, afc_client_t * client) +{ + /* makes sure thread environment is available */ + if (!g_thread_supported()) + g_thread_init(NULL); + + if (!device || port==0) + return AFC_E_INVALID_ARGUMENT; + + /* attempt connection */ + iphone_connection_t connection = NULL; + if (iphone_device_connect(device, port, &connection) != IPHONE_E_SUCCESS) { + return AFC_E_MUX_ERROR; + } + + afc_client_t client_loc = (afc_client_t) malloc(sizeof(struct afc_client_int)); + client_loc->connection = connection; + + /* allocate a packet */ + client_loc->afc_packet = (AFCPacket *) malloc(sizeof(AFCPacket)); + if (!client_loc->afc_packet) { + iphone_device_disconnect(client_loc->connection); + free(client_loc); + return AFC_E_NO_MEM; + } + + client_loc->afc_packet->packet_num = 0; + client_loc->afc_packet->entire_length = 0; + client_loc->afc_packet->this_length = 0; + memcpy(client_loc->afc_packet->magic, AFC_MAGIC, AFC_MAGIC_LEN); + client_loc->file_handle = 0; + client_loc->lock = 0; + client_loc->mutex = g_mutex_new(); + + *client = client_loc; + return AFC_E_SUCCESS; +} + +/** Disconnects an AFC client from the phone. + * + * @param client The client to disconnect. + */ +afc_error_t afc_client_free(afc_client_t client) +{ + if (!client || !client->connection || !client->afc_packet) + return AFC_E_INVALID_ARGUMENT; + + iphone_device_disconnect(client->connection); + free(client->afc_packet); + if (client->mutex) { + g_mutex_free(client->mutex); + } + free(client); + return AFC_E_SUCCESS; +} + +/** Dispatches an AFC packet over a client. + * + * @param client The client to send data through. + * @param data The data to send. + * @param length The length to send. + * @param bytes_sent The number of bytes actually sent. + * + * @return AFC_E_SUCCESS on success, or an AFC_E_* error value on error. + * + * @warning set client->afc_packet->this_length and + * client->afc_packet->entire_length to 0 before calling this. The + * reason is that if you set them to different values, it indicates + * you want to send the data as two packets. + */ +static afc_error_t afc_dispatch_packet(afc_client_t client, const char *data, uint32_t length, uint32_t *bytes_sent) +{ + uint32_t offset = 0; + uint32_t sent = 0; + + if (!client || !client->connection || !client->afc_packet) + return AFC_E_INVALID_ARGUMENT; + + *bytes_sent = 0; + + if (!data || !length) + length = 0; + + client->afc_packet->packet_num++; + if (!client->afc_packet->entire_length) { + client->afc_packet->entire_length = (length) ? sizeof(AFCPacket) + length : sizeof(AFCPacket); + client->afc_packet->this_length = client->afc_packet->entire_length; + } + if (!client->afc_packet->this_length) { + client->afc_packet->this_length = sizeof(AFCPacket); + } + // We want to send two segments; buffer+sizeof(AFCPacket) to + // this_length is the parameters + // And everything beyond that is the next packet. (for writing) + if (client->afc_packet->this_length != client->afc_packet->entire_length) { + offset = client->afc_packet->this_length - sizeof(AFCPacket); + + debug_info("Offset: %i", offset); + if ((length) < (client->afc_packet->entire_length - client->afc_packet->this_length)) { + debug_info("Length did not resemble what it was supposed to based on packet"); + debug_info("length minus offset: %i", length - offset); + debug_info("rest of packet: %i\n", client->afc_packet->entire_length - client->afc_packet->this_length); + return AFC_E_INTERNAL_ERROR; + } + + /* send AFC packet header */ + AFCPacket_to_LE(client->afc_packet); + sent = 0; + iphone_device_send(client->connection, (void*)client->afc_packet, sizeof(AFCPacket), &sent); + if (sent == 0) { + /* FIXME: should this be handled as success?! */ + return AFC_E_SUCCESS; + } + *bytes_sent += sent; + + /* send AFC packet data */ + sent = 0; + iphone_device_send(client->connection, data, offset, &sent); + if (sent == 0) { + return AFC_E_SUCCESS; + } + *bytes_sent += sent; + + debug_info("sent the first now go with the second"); + debug_info("Length: %i", length - offset); + debug_info("Buffer: "); + debug_buffer(data + offset, length - offset); + + sent = 0; + iphone_device_send(client->connection, data + offset, length - offset, &sent); + + *bytes_sent = sent; + return AFC_E_SUCCESS; + } else { + debug_info("doin things the old way"); + debug_info("packet length = %i", client->afc_packet->this_length); + + debug_buffer((char*)client->afc_packet, sizeof(AFCPacket)); + + /* send AFC packet header */ + AFCPacket_to_LE(client->afc_packet); + sent = 0; + iphone_device_send(client->connection, (void*)client->afc_packet, sizeof(AFCPacket), &sent); + if (sent == 0) { + return AFC_E_SUCCESS; + } + *bytes_sent += sent; + /* send AFC packet data (if there's data to send) */ + if (length > 0) { + debug_info("packet data follows"); + + debug_buffer(data, length); + iphone_device_send(client->connection, data, length, &sent); + *bytes_sent += sent; + } + return AFC_E_SUCCESS; + } + return AFC_E_INTERNAL_ERROR; +} + +/** Receives data through an AFC client and sets a variable to the received data. + * + * @param client The client to receive data on. + * @param dump_here The char* to point to the newly-received data. + * @param bytes_recv How much data was received. + * + * @return AFC_E_SUCCESS when data has been received, or an AFC_E_* error value + * when an error occured. + */ +static afc_error_t afc_receive_data(afc_client_t client, char **dump_here, uint32_t *bytes_recv) +{ + AFCPacket header; + uint32_t entire_len = 0; + uint32_t this_len = 0; + uint32_t current_count = 0; + uint64_t param1 = -1; + + *bytes_recv = 0; + + /* first, read the AFC header */ + iphone_device_recv(client->connection, (char*)&header, sizeof(AFCPacket), bytes_recv); + AFCPacket_from_LE(&header); + if (*bytes_recv == 0) { + debug_info("Just didn't get enough."); + *dump_here = NULL; + return AFC_E_MUX_ERROR; + } else if (*bytes_recv < sizeof(AFCPacket)) { + debug_info("Did not even get the AFCPacket header"); + *dump_here = NULL; + return AFC_E_MUX_ERROR; + } + + /* check if it's a valid AFC header */ + if (strncmp(header.magic, AFC_MAGIC, AFC_MAGIC_LEN)) { + debug_info("Invalid AFC packet received (magic != " AFC_MAGIC ")!"); + } + + /* check if it has the correct packet number */ + if (header.packet_num != client->afc_packet->packet_num) { + /* otherwise print a warning but do not abort */ + debug_info("ERROR: Unexpected packet number (%lld != %lld) aborting.", header.packet_num, client->afc_packet->packet_num); + *dump_here = NULL; + return AFC_E_OP_HEADER_INVALID; + } + + /* then, read the attached packet */ + if (header.this_length < sizeof(AFCPacket)) { + debug_info("Invalid AFCPacket header received!"); + *dump_here = NULL; + return AFC_E_OP_HEADER_INVALID; + } else if ((header.this_length == header.entire_length) + && header.entire_length == sizeof(AFCPacket)) { + debug_info("Empty AFCPacket received!"); + *dump_here = NULL; + *bytes_recv = 0; + if (header.operation == AFC_OP_DATA) { + return AFC_E_SUCCESS; + } else { + return AFC_E_IO_ERROR; + } + } + + debug_info("received AFC packet, full len=%lld, this len=%lld, operation=0x%llx", header.entire_length, header.this_length, header.operation); + + entire_len = (uint32_t)header.entire_length - sizeof(AFCPacket); + this_len = (uint32_t)header.this_length - sizeof(AFCPacket); + + /* this is here as a check (perhaps a different upper limit is good?) */ + if (entire_len > (uint32_t)MAXIMUM_PACKET_SIZE) { + fprintf(stderr, "%s: entire_len is larger than MAXIMUM_PACKET_SIZE, (%d > %d)!", __func__, entire_len, MAXIMUM_PACKET_SIZE); + } + + *dump_here = (char*)malloc(entire_len); + if (this_len > 0) { + iphone_device_recv(client->connection, *dump_here, this_len, bytes_recv); + if (*bytes_recv <= 0) { + free(*dump_here); + *dump_here = NULL; + debug_info("Did not get packet contents!"); + return AFC_E_NOT_ENOUGH_DATA; + } else if (*bytes_recv < this_len) { + free(*dump_here); + *dump_here = NULL; + debug_info("Could not receive this_len=%d bytes", this_len); + return AFC_E_NOT_ENOUGH_DATA; + } + } + + current_count = this_len; + + if (entire_len > this_len) { + while (current_count < entire_len) { + iphone_device_recv(client->connection, (*dump_here)+current_count, entire_len - current_count, bytes_recv); + if (*bytes_recv <= 0) { + debug_info("Error receiving data (recv returned %d)", *bytes_recv); + break; + } + current_count += *bytes_recv; + } + if (current_count < entire_len) { + debug_info("WARNING: could not receive full packet (read %s, size %d)", current_count, entire_len); + } + } + + if (current_count >= sizeof(uint64_t)) { + param1 = *(uint64_t*)(*dump_here); + } + + debug_info("packet data size = %i", current_count); + debug_info("packet data follows"); + debug_buffer(*dump_here, current_count); + + /* check operation types */ + if (header.operation == AFC_OP_STATUS) { + /* status response */ + debug_info("got a status response, code=%lld", param1); + + if (param1 != AFC_E_SUCCESS) { + /* error status */ + /* free buffer */ + free(*dump_here); + *dump_here = NULL; + return (afc_error_t)param1; + } + } else if (header.operation == AFC_OP_DATA) { + /* data response */ + debug_info("got a data response"); + } else if (header.operation == AFC_OP_FILE_OPEN_RES) { + /* file handle response */ + debug_info("got a file handle response, handle=%lld", param1); + } else if (header.operation == AFC_OP_FILE_TELL_RES) { + /* tell response */ + debug_info("got a tell response, position=%lld", param1); + } else { + /* unknown operation code received */ + free(*dump_here); + *dump_here = NULL; + *bytes_recv = 0; + + debug_info("WARNING: Unknown operation code received 0x%llx param1=%lld", header.operation, param1); + fprintf(stderr, "%s: WARNING: Unknown operation code received 0x%llx param1=%lld", __func__, (long long)header.operation, (long long)param1); + + return AFC_E_OP_NOT_SUPPORTED; + } + + *bytes_recv = current_count; + return AFC_E_SUCCESS; +} + +static uint32_t count_nullspaces(char *string, uint32_t number) +{ + uint32_t i = 0, nulls = 0; + + for (i = 0; i < number; i++) { + if (string[i] == '\0') + nulls++; + } + + return nulls; +} + +static char **make_strings_list(char *tokens, uint32_t true_length) +{ + uint32_t nulls = 0, i = 0, j = 0; + char **list = NULL; + + if (!tokens || !true_length) + return NULL; + + nulls = count_nullspaces(tokens, true_length); + list = (char **) malloc(sizeof(char *) * (nulls + 1)); + for (i = 0; i < nulls; i++) { + list[i] = strdup(tokens + j); + j += strlen(list[i]) + 1; + } + list[i] = NULL; + + return list; +} + +/** Gets a directory listing of the directory requested. + * + * @param client The client to get a directory listing from. + * @param dir The directory to list. (must be a fully-qualified path) + * + * @return A char ** list of files in that directory, terminated by an empty + * string for now or NULL if there was an error. + */ +afc_error_t afc_read_directory(afc_client_t client, const char *dir, char ***list) +{ + uint32_t bytes = 0; + char *data = NULL, **list_loc = NULL; + afc_error_t ret = AFC_E_UNKNOWN_ERROR; + + if (!client || !dir || !list || (list && *list)) + return AFC_E_INVALID_ARGUMENT; + + afc_lock(client); + + // Send the command + client->afc_packet->operation = AFC_OP_READ_DIR; + client->afc_packet->entire_length = 0; + client->afc_packet->this_length = 0; + ret = afc_dispatch_packet(client, dir, strlen(dir)+1, &bytes); + if (ret != AFC_E_SUCCESS) { + afc_unlock(client); + return AFC_E_NOT_ENOUGH_DATA; + } + // Receive the data + ret = afc_receive_data(client, &data, &bytes); + if (ret != AFC_E_SUCCESS) { + afc_unlock(client); + return ret; + } + // Parse the data + list_loc = make_strings_list(data, bytes); + if (data) + free(data); + + afc_unlock(client); + *list = list_loc; + + return ret; +} + +/** Get device info for a client connection to phone. (free space on disk, etc.) + * + * @param client The client to get device info for. + * + * @return A char ** list of parameters as given by AFC or NULL if there was an + * error. + */ +afc_error_t afc_get_device_info(afc_client_t client, char ***infos) +{ + uint32_t bytes = 0; + char *data = NULL, **list = NULL; + afc_error_t ret = AFC_E_UNKNOWN_ERROR; + + if (!client || !infos) + return AFC_E_INVALID_ARGUMENT; + + afc_lock(client); + + // Send the command + client->afc_packet->operation = AFC_OP_GET_DEVINFO; + client->afc_packet->entire_length = client->afc_packet->this_length = 0; + ret = afc_dispatch_packet(client, NULL, 0, &bytes); + if (ret != AFC_E_SUCCESS) { + afc_unlock(client); + return AFC_E_NOT_ENOUGH_DATA; + } + // Receive the data + ret = afc_receive_data(client, &data, &bytes); + if (ret != AFC_E_SUCCESS) { + afc_unlock(client); + return ret; + } + // Parse the data + list = make_strings_list(data, bytes); + if (data) + free(data); + + afc_unlock(client); + + *infos = list; + + return ret; +} + +/** Get a specific key of the device info list for a client connection. + * Known key values are: Model, FSTotalBytes, FSFreeBytes and FSBlockSize. + * This is a helper function for afc_get_device_info(). + * + * @param client The client to get device info for. + * @param key The key to get the value of. + * @param value The value for the key if successful or NULL otherwise. + * + * @return AFC_E_SUCCESS on success or an AFC_E_* error value. + */ +afc_error_t afc_get_device_info_key(afc_client_t client, const char *key, char **value) +{ + afc_error_t ret = AFC_E_INTERNAL_ERROR; + char **kvps, **ptr; + + *value = NULL; + if (key == NULL) + return AFC_E_INVALID_ARGUMENT; + + ret = afc_get_device_info(client, &kvps); + if (ret != AFC_E_SUCCESS) + return ret; + + for (ptr = kvps; *ptr; ptr++) { + if (!strcmp(*ptr, key)) { + *value = strdup(*(ptr+1)); + break; + } + } + + g_strfreev(kvps); + + return ret; +} + +/** Deletes a file or directory. + * + * @param client The client to use. + * @param path The path to delete. (must be a fully-qualified path) + * + * @return AFC_E_SUCCESS if everythong went well, AFC_E_INVALID_ARGUMENT + * if arguments are NULL or invalid, AFC_E_NOT_ENOUGH_DATA otherwise. + */ +afc_error_t afc_remove_path(afc_client_t client, const char *path) +{ + char *response = NULL; + uint32_t bytes = 0; + afc_error_t ret = AFC_E_UNKNOWN_ERROR; + + if (!client || !path || !client->afc_packet || !client->connection) + return AFC_E_INVALID_ARGUMENT; + + afc_lock(client); + + // Send command + client->afc_packet->this_length = client->afc_packet->entire_length = 0; + client->afc_packet->operation = AFC_OP_REMOVE_PATH; + ret = afc_dispatch_packet(client, path, strlen(path)+1, &bytes); + if (ret != AFC_E_SUCCESS) { + afc_unlock(client); + return AFC_E_NOT_ENOUGH_DATA; + } + // Receive response + ret = afc_receive_data(client, &response, &bytes); + if (response) + free(response); + + /* special case; unknown error actually means directory not empty */ + if (ret == AFC_E_UNKNOWN_ERROR) + ret = AFC_E_DIR_NOT_EMPTY; + + afc_unlock(client); + + return ret; +} + +/** Renames a file or directory on the phone. + * + * @param client The client to have rename. + * @param from The name to rename from. (must be a fully-qualified path) + * @param to The new name. (must also be a fully-qualified path) + * + * @return AFC_E_SUCCESS if everythong went well, AFC_E_INVALID_ARGUMENT + * if arguments are NULL or invalid, AFC_E_NOT_ENOUGH_DATA otherwise. + */ +afc_error_t afc_rename_path(afc_client_t client, const char *from, const char *to) +{ + char *response = NULL; + char *send = (char *) malloc(sizeof(char) * (strlen(from) + strlen(to) + 1 + sizeof(uint32_t))); + uint32_t bytes = 0; + afc_error_t ret = AFC_E_UNKNOWN_ERROR; + + if (!client || !from || !to || !client->afc_packet || !client->connection) + return AFC_E_INVALID_ARGUMENT; + + afc_lock(client); + + // Send command + memcpy(send, from, strlen(from) + 1); + memcpy(send + strlen(from) + 1, to, strlen(to) + 1); + client->afc_packet->entire_length = client->afc_packet->this_length = 0; + client->afc_packet->operation = AFC_OP_RENAME_PATH; + ret = afc_dispatch_packet(client, send, strlen(to)+1 + strlen(from)+1, &bytes); + free(send); + if (ret != AFC_E_SUCCESS) { + afc_unlock(client); + return AFC_E_NOT_ENOUGH_DATA; + } + // Receive response + ret = afc_receive_data(client, &response, &bytes); + if (response) + free(response); + + afc_unlock(client); + + return ret; +} + +/** Creates a directory on the phone. + * + * @param client The client to use to make a directory. + * @param dir The directory's path. (must be a fully-qualified path, I assume + * all other mkdir restrictions apply as well) + * + * @return AFC_E_SUCCESS if everythong went well, AFC_E_INVALID_ARGUMENT + * if arguments are NULL or invalid, AFC_E_NOT_ENOUGH_DATA otherwise. + */ +afc_error_t afc_make_directory(afc_client_t client, const char *dir) +{ + uint32_t bytes = 0; + char *response = NULL; + afc_error_t ret = AFC_E_UNKNOWN_ERROR; + + if (!client) + return AFC_E_INVALID_ARGUMENT; + + afc_lock(client); + + // Send command + client->afc_packet->operation = AFC_OP_MAKE_DIR; + client->afc_packet->this_length = client->afc_packet->entire_length = 0; + ret = afc_dispatch_packet(client, dir, strlen(dir)+1, &bytes); + if (ret != AFC_E_SUCCESS) { + afc_unlock(client); + return AFC_E_NOT_ENOUGH_DATA; + } + // Receive response + ret = afc_receive_data(client, &response, &bytes); + if (response) + free(response); + + afc_unlock(client); + + return ret; +} + +/** Gets information about a specific file. + * + * @param client The client to use to get the information of the file. + * @param path The fully-qualified path to the file. + * @param infolist Pointer to a buffer that will be filled with a NULL-terminated + * list of strings with the file information. + * Set to NULL before calling this function. + * + * @return AFC_E_SUCCESS on success or an AFC_E_* error value + * when something went wrong. + */ +afc_error_t afc_get_file_info(afc_client_t client, const char *path, char ***infolist) +{ + char *received = NULL; + uint32_t bytes = 0; + afc_error_t ret = AFC_E_UNKNOWN_ERROR; + + if (!client || !path || !infolist) + return AFC_E_INVALID_ARGUMENT; + + afc_lock(client); + + // Send command + client->afc_packet->operation = AFC_OP_GET_FILE_INFO; + client->afc_packet->entire_length = client->afc_packet->this_length = 0; + ret = afc_dispatch_packet(client, path, strlen(path)+1, &bytes); + if (ret != AFC_E_SUCCESS) { + afc_unlock(client); + return AFC_E_NOT_ENOUGH_DATA; + } + + // Receive data + ret = afc_receive_data(client, &received, &bytes); + if (received) { + *infolist = make_strings_list(received, bytes); + free(received); + } + + afc_unlock(client); + + return ret; +} + +/** Opens a file on the phone. + * + * @param client The client to use to open the file. + * @param filename The file to open. (must be a fully-qualified path) + * @param file_mode The mode to use to open the file. Can be AFC_FILE_READ or + * AFC_FILE_WRITE; the former lets you read and write, + * however, and the second one will *create* the file, + * destroying anything previously there. + * @param handle Pointer to a uint64_t that will hold the handle of the file + * + * @return AFC_E_SUCCESS on success or an AFC_E_* error on failure. + */ +iphone_error_t +afc_file_open(afc_client_t client, const char *filename, + afc_file_mode_t file_mode, uint64_t *handle) +{ + uint64_t file_mode_loc = GUINT64_TO_LE(file_mode); + uint32_t bytes = 0; + char *data = (char *) malloc(sizeof(char) * (8 + strlen(filename) + 1)); + afc_error_t ret = AFC_E_UNKNOWN_ERROR; + + // set handle to 0 so in case an error occurs, the handle is invalid + *handle = 0; + + if (!client || !client->connection || !client->afc_packet) + return AFC_E_INVALID_ARGUMENT; + + afc_lock(client); + + // Send command + memcpy(data, &file_mode_loc, 8); + memcpy(data + 8, filename, strlen(filename)); + data[8 + strlen(filename)] = '\0'; + client->afc_packet->operation = AFC_OP_FILE_OPEN; + client->afc_packet->entire_length = client->afc_packet->this_length = 0; + ret = afc_dispatch_packet(client, data, 8 + strlen(filename) + 1, &bytes); + free(data); + + if (ret != AFC_E_SUCCESS) { + debug_info("Didn't receive a response to the command"); + afc_unlock(client); + return AFC_E_NOT_ENOUGH_DATA; + } + // Receive the data + ret = afc_receive_data(client, &data, &bytes); + if ((ret == AFC_E_SUCCESS) && (bytes > 0) && data) { + afc_unlock(client); + + // Get the file handle + memcpy(handle, data, sizeof(uint64_t)); + free(data); + return ret; + } + + debug_info("Didn't get any further data"); + + afc_unlock(client); + + return ret; +} + +/** Attempts to the read the given number of bytes from the given file. + * + * @param client The relevant AFC client + * @param handle File handle of a previously opened file + * @param data The pointer to the memory region to store the read data + * @param length The number of bytes to read + * @param bytes_read The number of bytes actually read. + * + * @return AFC_E_SUCCESS on success or an AFC_E_* error value on error. + */ +iphone_error_t +afc_file_read(afc_client_t client, uint64_t handle, char *data, uint32_t length, uint32_t *bytes_read) +{ + char *input = NULL; + uint32_t current_count = 0, bytes_loc = 0; + const uint32_t MAXIMUM_READ_SIZE = 1 << 16; + afc_error_t ret = AFC_E_SUCCESS; + + if (!client || !client->afc_packet || !client->connection || handle == 0) + return AFC_E_INVALID_ARGUMENT; + debug_info("called for length %i", length); + + afc_lock(client); + + // Looping here to get around the maximum amount of data that + // afc_receive_data can handle + while (current_count < length) { + debug_info("current count is %i but length is %i", current_count, length); + + // Send the read command + AFCFilePacket *packet = (AFCFilePacket *) malloc(sizeof(AFCFilePacket)); + packet->filehandle = handle; + packet->size = GUINT64_TO_LE(((length - current_count) < MAXIMUM_READ_SIZE) ? (length - current_count) : MAXIMUM_READ_SIZE); + client->afc_packet->operation = AFC_OP_READ; + client->afc_packet->entire_length = client->afc_packet->this_length = 0; + ret = afc_dispatch_packet(client, (char *) packet, sizeof(AFCFilePacket), &bytes_loc); + free(packet); + + if (ret != AFC_E_SUCCESS) { + afc_unlock(client); + return AFC_E_NOT_ENOUGH_DATA; + } + // Receive the data + ret = afc_receive_data(client, &input, &bytes_loc); + debug_info("afc_receive_data returned error: %d", ret); + debug_info("bytes returned: %i", bytes_loc); + if (ret != AFC_E_SUCCESS) { + afc_unlock(client); + return ret; + } else if (bytes_loc == 0) { + if (input) + free(input); + afc_unlock(client); + *bytes_read = current_count; + /* FIXME: check that's actually a success */ + return ret; + } else { + if (input) { + debug_info("%d", bytes_loc); + memcpy(data + current_count, input, (bytes_loc > length) ? length : bytes_loc); + free(input); + input = NULL; + current_count += (bytes_loc > length) ? length : bytes_loc; + } + } + } + debug_info("returning current_count as %i", current_count); + + afc_unlock(client); + *bytes_read = current_count; + return ret; +} + +/** Writes a given number of bytes to a file. + * + * @param client The client to use to write to the file. + * @param handle File handle of previously opened file. + * @param data The data to write to the file. + * @param length How much data to write. + * @param bytes_written The number of bytes actually written to the file. + * + * @return AFC_E_SUCCESS on success, or an AFC_E_* error value on error. + */ +iphone_error_t +afc_file_write(afc_client_t client, uint64_t handle, const char *data, uint32_t length, uint32_t *bytes_written) +{ + char *acknowledgement = NULL; + const uint32_t MAXIMUM_WRITE_SIZE = 1 << 15; + uint32_t current_count = 0, i = 0; + uint32_t segments = (length / MAXIMUM_WRITE_SIZE); + uint32_t bytes_loc = 0; + char *out_buffer = NULL; + afc_error_t ret = AFC_E_SUCCESS; + + if (!client || !client->afc_packet || !client->connection || !bytes_written || (handle == 0)) + return AFC_E_INVALID_ARGUMENT; + + afc_lock(client); + + debug_info("Write length: %i", length); + + // Divide the file into segments. + for (i = 0; i < segments; i++) { + // Send the segment + client->afc_packet->this_length = sizeof(AFCPacket) + 8; + client->afc_packet->entire_length = client->afc_packet->this_length + MAXIMUM_WRITE_SIZE; + client->afc_packet->operation = AFC_OP_WRITE; + out_buffer = (char *) malloc(sizeof(char) * client->afc_packet->entire_length - sizeof(AFCPacket)); + memcpy(out_buffer, (char *)&handle, sizeof(uint64_t)); + memcpy(out_buffer + 8, data + current_count, MAXIMUM_WRITE_SIZE); + ret = afc_dispatch_packet(client, out_buffer, MAXIMUM_WRITE_SIZE + 8, &bytes_loc); + if (ret != AFC_E_SUCCESS) { + afc_unlock(client); + return AFC_E_NOT_ENOUGH_DATA; + } + free(out_buffer); + out_buffer = NULL; + + current_count += bytes_loc; + ret = afc_receive_data(client, &acknowledgement, &bytes_loc); + if (ret != AFC_E_SUCCESS) { + afc_unlock(client); + return ret; + } else { + free(acknowledgement); + } + } + + // By this point, we should be at the end. i.e. the last segment that + // didn't get sent in the for loop + // this length is fine because it's always sizeof(AFCPacket) + 8, but + // to be sure we do it again + if (current_count == length) { + afc_unlock(client); + *bytes_written = current_count; + return ret; + } + + client->afc_packet->this_length = sizeof(AFCPacket) + 8; + client->afc_packet->entire_length = client->afc_packet->this_length + (length - current_count); + client->afc_packet->operation = AFC_OP_WRITE; + out_buffer = (char *) malloc(sizeof(char) * client->afc_packet->entire_length - sizeof(AFCPacket)); + memcpy(out_buffer, (char *) &handle, sizeof(uint64_t)); + memcpy(out_buffer + 8, data + current_count, (length - current_count)); + ret = afc_dispatch_packet(client, out_buffer, (length - current_count) + 8, &bytes_loc); + free(out_buffer); + out_buffer = NULL; + + current_count += bytes_loc; + + if (ret != AFC_E_SUCCESS) { + afc_unlock(client); + *bytes_written = current_count; + return AFC_E_SUCCESS; + } + + ret = afc_receive_data(client, &acknowledgement, &bytes_loc); + afc_unlock(client); + if (ret != AFC_E_SUCCESS) { + debug_info("uh oh?"); + } else { + free(acknowledgement); + } + *bytes_written = current_count; + return ret; +} + +/** Closes a file on the phone. + * + * @param client The client to close the file with. + * @param handle File handle of a previously opened file. + */ +afc_error_t afc_file_close(afc_client_t client, uint64_t handle) +{ + char *buffer = malloc(sizeof(char) * 8); + uint32_t bytes = 0; + afc_error_t ret = AFC_E_UNKNOWN_ERROR; + + if (!client || (handle == 0)) + return AFC_E_INVALID_ARGUMENT; + + afc_lock(client); + + debug_info("File handle %i", handle); + + // Send command + memcpy(buffer, &handle, sizeof(uint64_t)); + client->afc_packet->operation = AFC_OP_FILE_CLOSE; + client->afc_packet->entire_length = client->afc_packet->this_length = 0; + ret = afc_dispatch_packet(client, buffer, 8, &bytes); + free(buffer); + buffer = NULL; + + if (ret != AFC_E_SUCCESS) { + afc_unlock(client); + return AFC_E_UNKNOWN_ERROR; + } + + // Receive the response + ret = afc_receive_data(client, &buffer, &bytes); + if (buffer) + free(buffer); + + afc_unlock(client); + + return ret; +} + +/** Locks or unlocks a file on the phone. + * + * makes use of flock on the device, see + * http://developer.apple.com/documentation/Darwin/Reference/ManPages/man2/flock.2.html + * + * @param client The client to lock the file with. + * @param handle File handle of a previously opened file. + * @param operation the lock or unlock operation to perform, this is one of + * AFC_LOCK_SH (shared lock), AFC_LOCK_EX (exclusive lock), + * or AFC_LOCK_UN (unlock). + */ +afc_error_t afc_file_lock(afc_client_t client, uint64_t handle, afc_lock_op_t operation) +{ + char *buffer = malloc(16); + uint32_t bytes = 0; + uint64_t op = GUINT64_TO_LE(operation); + afc_error_t ret = AFC_E_UNKNOWN_ERROR; + + if (!client || (handle == 0)) + return AFC_E_INVALID_ARGUMENT; + + afc_lock(client); + + debug_info("file handle %i", handle); + + // Send command + memcpy(buffer, &handle, sizeof(uint64_t)); + memcpy(buffer + 8, &op, 8); + + client->afc_packet->operation = AFC_OP_FILE_LOCK; + client->afc_packet->entire_length = client->afc_packet->this_length = 0; + ret = afc_dispatch_packet(client, buffer, 16, &bytes); + free(buffer); + buffer = NULL; + + if (ret != AFC_E_SUCCESS) { + afc_unlock(client); + debug_info("could not send lock command"); + return AFC_E_UNKNOWN_ERROR; + } + // Receive the response + ret = afc_receive_data(client, &buffer, &bytes); + if (buffer) { + debug_buffer(buffer, bytes); + free(buffer); + } + afc_unlock(client); + + return ret; +} + +/** Seeks to a given position of a pre-opened file on the phone. + * + * @param client The client to use to seek to the position. + * @param handle File handle of a previously opened. + * @param offset Seek offset. + * @param whence Seeking direction, one of SEEK_SET, SEEK_CUR, or SEEK_END. + * + * @return AFC_E_SUCCESS on success, AFC_E_NOT_ENOUGH_DATA on failure. + */ +afc_error_t afc_file_seek(afc_client_t client, uint64_t handle, int64_t offset, int whence) +{ + char *buffer = (char *) malloc(sizeof(char) * 24); + int64_t offset_loc = (int64_t)GUINT64_TO_LE(offset); + uint64_t whence_loc = GUINT64_TO_LE(whence); + uint32_t bytes = 0; + afc_error_t ret = AFC_E_UNKNOWN_ERROR; + + if (!client || (handle == 0)) + return AFC_E_INVALID_ARGUMENT; + + afc_lock(client); + + // Send the command + memcpy(buffer, &handle, sizeof(uint64_t)); // handle + memcpy(buffer + 8, &whence_loc, sizeof(uint64_t)); // fromwhere + memcpy(buffer + 16, &offset_loc, sizeof(uint64_t)); // offset + client->afc_packet->operation = AFC_OP_FILE_SEEK; + client->afc_packet->this_length = client->afc_packet->entire_length = 0; + ret = afc_dispatch_packet(client, buffer, 24, &bytes); + free(buffer); + buffer = NULL; + + if (ret != AFC_E_SUCCESS) { + afc_unlock(client); + return AFC_E_NOT_ENOUGH_DATA; + } + // Receive response + ret = afc_receive_data(client, &buffer, &bytes); + if (buffer) + free(buffer); + + afc_unlock(client); + + return ret; +} + +/** Returns current position in a pre-opened file on the phone. + * + * @param client The client to use. + * @param handle File handle of a previously opened file. + * @param position Position in bytes of indicator + * + * @return AFC_E_SUCCESS on success, AFC_E_NOT_ENOUGH_DATA on failure. + */ +afc_error_t afc_file_tell(afc_client_t client, uint64_t handle, uint64_t *position) +{ + char *buffer = (char *) malloc(sizeof(char) * 8); + uint32_t bytes = 0; + afc_error_t ret = AFC_E_UNKNOWN_ERROR; + + if (!client || (handle == 0)) + return AFC_E_INVALID_ARGUMENT; + + afc_lock(client); + + // Send the command + memcpy(buffer, &handle, sizeof(uint64_t)); // handle + client->afc_packet->operation = AFC_OP_FILE_TELL; + client->afc_packet->this_length = client->afc_packet->entire_length = 0; + ret = afc_dispatch_packet(client, buffer, 8, &bytes); + free(buffer); + buffer = NULL; + + if (ret != AFC_E_SUCCESS) { + afc_unlock(client); + return AFC_E_NOT_ENOUGH_DATA; + } + + // Receive the data + ret = afc_receive_data(client, &buffer, &bytes); + if (bytes > 0 && buffer) { + /* Get the position */ + memcpy(position, buffer, sizeof(uint64_t)); + *position = GUINT64_FROM_LE(*position); + } + if (buffer) + free(buffer); + + afc_unlock(client); + + return ret; +} + +/** Sets the size of a file on the phone. + * + * @param client The client to use to set the file size. + * @param handle File handle of a previously opened file. + * @param newsize The size to set the file to. + * + * @return 0 on success, -1 on failure. + * + * @note This function is more akin to ftruncate than truncate, and truncate + * calls would have to open the file before calling this, sadly. + */ +afc_error_t afc_file_truncate(afc_client_t client, uint64_t handle, uint64_t newsize) +{ + char *buffer = (char *) malloc(sizeof(char) * 16); + uint32_t bytes = 0; + uint64_t newsize_loc = GUINT64_TO_LE(newsize); + afc_error_t ret = AFC_E_UNKNOWN_ERROR; + + if (!client || (handle == 0)) + return AFC_E_INVALID_ARGUMENT; + + afc_lock(client); + + // Send command + memcpy(buffer, &handle, sizeof(uint64_t)); // handle + memcpy(buffer + 8, &newsize_loc, sizeof(uint64_t)); // newsize + client->afc_packet->operation = AFC_OP_FILE_SET_SIZE; + client->afc_packet->this_length = client->afc_packet->entire_length = 0; + ret = afc_dispatch_packet(client, buffer, 16, &bytes); + free(buffer); + buffer = NULL; + + if (ret != AFC_E_SUCCESS) { + afc_unlock(client); + return AFC_E_NOT_ENOUGH_DATA; + } + // Receive response + ret = afc_receive_data(client, &buffer, &bytes); + if (buffer) + free(buffer); + + afc_unlock(client); + + return ret; +} + +/** Sets the size of a file on the phone without prior opening it. + * + * @param client The client to use to set the file size. + * @param path The path of the file to be truncated. + * @param newsize The size to set the file to. + * + * @return AFC_E_SUCCESS if everything went well, AFC_E_INVALID_ARGUMENT + * if arguments are NULL or invalid, AFC_E_NOT_ENOUGH_DATA otherwise. + */ +afc_error_t afc_truncate(afc_client_t client, const char *path, uint64_t newsize) +{ + char *response = NULL; + char *send = (char *) malloc(sizeof(char) * (strlen(path) + 1 + 8)); + uint32_t bytes = 0; + uint64_t size_requested = GUINT64_TO_LE(newsize); + afc_error_t ret = AFC_E_UNKNOWN_ERROR; + + if (!client || !path || !client->afc_packet || !client->connection) + return AFC_E_INVALID_ARGUMENT; + + afc_lock(client); + + // Send command + memcpy(send, &size_requested, 8); + memcpy(send + 8, path, strlen(path) + 1); + client->afc_packet->entire_length = client->afc_packet->this_length = 0; + client->afc_packet->operation = AFC_OP_TRUNCATE; + ret = afc_dispatch_packet(client, send, 8 + strlen(path) + 1, &bytes); + free(send); + if (ret != AFC_E_SUCCESS) { + afc_unlock(client); + return AFC_E_NOT_ENOUGH_DATA; + } + // Receive response + ret = afc_receive_data(client, &response, &bytes); + if (response) + free(response); + + afc_unlock(client); + + return ret; +} + +/** Creates a hard link or symbolic link on the device. + * + * @param client The client to use for making a link + * @param type 1 = hard link, 2 = symlink + * @param target The file to be linked. + * @param linkname The name of link. + * + * @return AFC_E_SUCCESS if everything went well, AFC_E_INVALID_ARGUMENT + * if arguments are NULL or invalid, AFC_E_NOT_ENOUGH_DATA otherwise. + */ +afc_error_t afc_make_link(afc_client_t client, afc_link_type_t linktype, const char *target, const char *linkname) +{ + char *response = NULL; + char *send = (char *) malloc(sizeof(char) * (strlen(target)+1 + strlen(linkname)+1 + 8)); + uint32_t bytes = 0; + uint64_t type = GUINT64_TO_LE(linktype); + afc_error_t ret = AFC_E_UNKNOWN_ERROR; + + if (!client || !target || !linkname || !client->afc_packet || !client->connection) + return AFC_E_INVALID_ARGUMENT; + + afc_lock(client); + + debug_info("link type: %lld", type); + debug_info("target: %s, length:%d", target, strlen(target)); + debug_info("linkname: %s, length:%d", linkname, strlen(linkname)); + + // Send command + memcpy(send, &type, 8); + memcpy(send + 8, target, strlen(target) + 1); + memcpy(send + 8 + strlen(target) + 1, linkname, strlen(linkname) + 1); + client->afc_packet->entire_length = client->afc_packet->this_length = 0; + client->afc_packet->operation = AFC_OP_MAKE_LINK; + ret = afc_dispatch_packet(client, send, 8 + strlen(linkname) + 1 + strlen(target) + 1, &bytes); + free(send); + if (ret != AFC_E_SUCCESS) { + afc_unlock(client); + return AFC_E_NOT_ENOUGH_DATA; + } + // Receive response + ret = afc_receive_data(client, &response, &bytes); + if (response) + free(response); + + afc_unlock(client); + + return ret; +} + +/** Sets the modification time of a file on the phone. + * + * @param client The client to use to set the file size. + * @param path Path of the file for which the modification time should be set. + * @param mtime The modification time to set in nanoseconds since epoch. + * + * @return AFC_E_SUCCESS if everything went well, AFC_E_INVALID_ARGUMENT + * if arguments are NULL or invalid, AFC_E_NOT_ENOUGH_DATA otherwise. + */ +afc_error_t afc_set_file_time(afc_client_t client, const char *path, uint64_t mtime) +{ + char *response = NULL; + char *send = (char *) malloc(sizeof(char) * (strlen(path) + 1 + 8)); + uint32_t bytes = 0; + uint64_t mtime_loc = GUINT64_TO_LE(mtime); + afc_error_t ret = AFC_E_UNKNOWN_ERROR; + + if (!client || !path || !client->afc_packet || !client->connection) + return AFC_E_INVALID_ARGUMENT; + + afc_lock(client); + + // Send command + memcpy(send, &mtime_loc, 8); + memcpy(send + 8, path, strlen(path) + 1); + client->afc_packet->entire_length = client->afc_packet->this_length = 0; + client->afc_packet->operation = AFC_OP_SET_FILE_TIME; + ret = afc_dispatch_packet(client, send, 8 + strlen(path) + 1, &bytes); + free(send); + if (ret != AFC_E_SUCCESS) { + afc_unlock(client); + return AFC_E_NOT_ENOUGH_DATA; + } + // Receive response + ret = afc_receive_data(client, &response, &bytes); + if (response) + free(response); + + afc_unlock(client); + + return ret; +} + |