summaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
authorGravatar Joshua Hill2010-11-06 18:09:53 -0400
committerGravatar Joshua Hill2010-11-06 18:09:53 -0400
commitd82a73f1180a5f201397b92e6f6db33411cbe686 (patch)
treebcc7c4ba2e103770f905972854392d8c3e448f16 /src
parente89c9418e7072d5a1491055c8f78748bf82de49c (diff)
downloadlibirecovery-d82a73f1180a5f201397b92e6f6db33411cbe686.tar.gz
libirecovery-d82a73f1180a5f201397b92e6f6db33411cbe686.tar.bz2
updated libirecovery to the latest version from greenpois0n. it at least compiles on osx now, I need to test on other systems
Diffstat (limited to 'src')
-rw-r--r--src/irecovery.c343
-rw-r--r--src/libirecovery.c672
2 files changed, 0 insertions, 1015 deletions
diff --git a/src/irecovery.c b/src/irecovery.c
deleted file mode 100644
index 0e981cd..0000000
--- a/src/irecovery.c
+++ /dev/null
@@ -1,343 +0,0 @@
-/**
- * iRecovery - Utility for DFU 2.0, WTF and Recovery Mode
- * Copyright (C) 2008 - 2009 westbaer
- *
- * This program is free software: you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation, either version 3 of the License, or
- * (at your option) any later version.
- *
- * This program 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 General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program. If not, see <http://www.gnu.org/licenses/>.
- **/
-
-#include <stdio.h>
-#include <stdlib.h>
-#include <unistd.h>
-#include <libirecovery.h>
-#include <readline/readline.h>
-#include <readline/history.h>
-
-#define FILE_HISTORY_PATH ".irecovery"
-#define debug(...) if(verbose) fprintf(stderr, __VA_ARGS__)
-
-enum {
- kResetDevice, kStartShell, kSendCommand, kSendFile, kSendExploit, kSendScript
-};
-
-static unsigned int quit = 0;
-static unsigned int verbose = 0;
-
-void print_progress_bar(double progress);
-int received_cb(irecv_client_t client, const irecv_event_t* event);
-int progress_cb(irecv_client_t client, const irecv_event_t* event);
-int precommand_cb(irecv_client_t client, const irecv_event_t* event);
-int postcommand_cb(irecv_client_t client, const irecv_event_t* event);
-
-void shell_usage() {
- printf("Usage:\n");
- printf("\t/upload <file>\tSend file to client.\n");
- printf("\t/exploit [file]\tSend usb exploit with optional payload\n");
- printf("\t/help\t\tShow this help.\n");
- printf("\t/exit\t\tExit interactive shell.\n");
-}
-
-void parse_command(irecv_client_t client, unsigned char* command, unsigned int size) {
- char* cmd = strdup(command);
- char* action = strtok(cmd, " ");
- debug("Executing %s\n", action);
- if (!strcmp(cmd, "/exit")) {
- quit = 1;
- } else
-
- if (!strcmp(cmd, "/help")) {
- shell_usage();
- } else
-
- if (!strcmp(cmd, "/upload")) {
- char* filename = strtok(NULL, " ");
- debug("Uploading files %s\n", filename);
- if (filename != NULL) {
- irecv_send_file(client, filename);
- }
- } else
-
- if (!strcmp(cmd, "/exploit")) {
- char* filename = strtok(NULL, " ");
- debug("Sending exploit %s\n", filename);
- if (filename != NULL) {
- irecv_send_file(client, filename);
- }
- irecv_send_exploit(client);
- } else
-
- if (!strcmp(cmd, "/execute")) {
- char* filename = strtok(NULL, " ");
- debug("Executing script %s\n", filename);
- if (filename != NULL) {
- irecv_execute_script(client, filename);
- }
- }
-
-
- free(action);
-}
-
-void load_command_history() {
- read_history(FILE_HISTORY_PATH);
-}
-
-void append_command_to_history(char* cmd) {
- add_history(cmd);
- write_history(FILE_HISTORY_PATH);
-}
-
-void init_shell(irecv_client_t client) {
- irecv_error_t error = 0;
- load_command_history();
- irecv_event_subscribe(client, IRECV_PROGRESS, &progress_cb, NULL);
- irecv_event_subscribe(client, IRECV_RECEIVED, &received_cb, NULL);
- irecv_event_subscribe(client, IRECV_PRECOMMAND, &precommand_cb, NULL);
- irecv_event_subscribe(client, IRECV_POSTCOMMAND, &postcommand_cb, NULL);
- while (!quit) {
- error = irecv_receive(client);
- if (error != IRECV_E_SUCCESS) {
- debug("%s\n", irecv_strerror(error));
- break;
- }
-
- char* cmd = readline("> ");
- if (cmd && *cmd) {
- error = irecv_send_command(client, cmd);
- if (error != IRECV_E_SUCCESS) {
- quit = 1;
- }
-
- append_command_to_history(cmd);
- free(cmd);
- }
- }
-}
-
-int received_cb(irecv_client_t client, const irecv_event_t* event) {
- if (event->type == IRECV_RECEIVED) {
- int i = 0;
- int size = event->size;
- char* data = event->data;
- for (i = 0; i < size; i++) {
- printf("%c", data[i]);
- }
- }
- return 0;
-}
-
-int precommand_cb(irecv_client_t client, const irecv_event_t* event) {
- if (event->type == IRECV_PRECOMMAND) {
- irecv_error_t error = 0;
- if (event->data[0] == '/') {
- parse_command(client, event->data, event->size);
- return -1;
- }
- }
- return 0;
-}
-
-int postcommand_cb(irecv_client_t client, const irecv_event_t* event) {
- char* value = NULL;
- char* action = NULL;
- char* command = NULL;
- char* argument = NULL;
- irecv_error_t error = IRECV_E_SUCCESS;
-
- if (event->type == IRECV_POSTCOMMAND) {
- command = strdup(event->data);
- action = strtok(command, " ");
- if (!strcmp(action, "getenv")) {
- argument = strtok(NULL, " ");
- error = irecv_getenv(client, argument, &value);
- if (error != IRECV_E_SUCCESS) {
- debug("%s\n", irecv_strerror(error));
- free(command);
- return error;
- }
- printf("%s\n", value);
- free(value);
- }
-
- if (!strcmp(action, "reboot")) {
- quit = 1;
- }
- }
-
- if (command) free(command);
- return 0;
-}
-
-int progress_cb(irecv_client_t client, const irecv_event_t* event) {
- if (event->type == IRECV_PROGRESS) {
- print_progress_bar(event->progress);
- }
- return 0;
-}
-
-void print_progress_bar(double progress) {
- int i = 0;
- if(progress < 0) {
- return;
- }
-
- if(progress > 100) {
- progress = 100;
- }
-
- printf("\r[");
- for(i = 0; i < 50; i++) {
- if(i < progress / 2) {
- printf("=");
- } else {
- printf(" ");
- }
- }
-
- printf("] %3.1f%%", progress);
- fflush(stdout);
- if(progress == 100) {
- printf("\n");
- }
-}
-
-void print_usage() {
- printf("iRecovery - iDevice Recovery Utility\n");
- printf("Usage: ./irecovery [args]\n");
- printf("\t-v\t\tStart irecovery in verbose mode.\n");
- printf("\t-c <cmd>\tSend command to client.\n");
- printf("\t-f <file>\tSend file to client.\n");
- printf("\t-k [payload]\tSend usb exploit to client.\n");
- printf("\t-h\t\tShow this help.\n");
- printf("\t-r\t\tReset client.\n");
- printf("\t-s\t\tStart interactive shell.\n");
- printf("\t-e <script>\tExecutes recovery shell script.\n");
- exit(1);
-}
-
-int main(int argc, char** argv) {
- int i = 0;
- int opt = 0;
- int action = 0;
- char* argument = NULL;
- irecv_error_t error = 0;
- if (argc == 1) print_usage();
- while ((opt = getopt(argc, argv, "vhrsc:f:e:k::")) > 0) {
- switch (opt) {
- case 'v':
- verbose += 1;
- break;
-
- case 'h':
- print_usage();
- break;
-
- case 'r':
- action = kResetDevice;
- break;
-
- case 's':
- action = kStartShell;
- break;
-
- case 'f':
- action = kSendFile;
- argument = optarg;
- break;
-
- case 'c':
- action = kSendCommand;
- argument = optarg;
- break;
-
- case 'k':
- action = kSendExploit;
- argument = optarg;
- break;
-
- case 'e':
- action = kSendScript;
- argument = optarg;
- break;
-
- default:
- fprintf(stderr, "Unknown argument\n");
- return -1;
- }
- }
-
- irecv_client_t client = NULL;
- for (i = 0; i <= 5; i++) {
- debug("Attempting to connect... \n");
-
- if (irecv_open(&client) != IRECV_E_SUCCESS)
- sleep(1);
- else
- break;
-
- if (i == 5) {
- return -1;
- }
- }
-
- if (verbose) irecv_set_debug_level(verbose);
-
- switch (action) {
- case kResetDevice:
- irecv_reset(client);
- break;
-
- case kSendFile:
- irecv_event_subscribe(client, IRECV_PROGRESS, &progress_cb, NULL);
- error = irecv_send_file(client, argument);
- debug("%s\n", irecv_strerror(error));
- break;
-
- case kSendCommand:
- error = irecv_send_command(client, argument);
- debug("%s\n", irecv_strerror(error));
- break;
-
- case kSendExploit:
- if (argument != NULL) {
- irecv_event_subscribe(client, IRECV_PROGRESS, &progress_cb, NULL);
- error = irecv_send_file(client, argument);
- if (error != IRECV_E_SUCCESS) {
- debug("%s\n", irecv_strerror(error));
- break;
- }
- }
- error = irecv_send_exploit(client);
- debug("%s\n", irecv_strerror(error));
- break;
-
- case kStartShell:
- init_shell(client);
- break;
-
- case kSendScript:
- error = irecv_execute_script(client, argument);
- if(error != IRECV_E_SUCCESS) {
- debug("%s\n", irecv_strerror(error));
- }
- break;
-
- default:
- fprintf(stderr, "Unknown action\n");
- break;
- }
-
- irecv_close(client);
- return 0;
-}
-
diff --git a/src/libirecovery.c b/src/libirecovery.c
deleted file mode 100644
index dd5c734..0000000
--- a/src/libirecovery.c
+++ /dev/null
@@ -1,672 +0,0 @@
-/**
- * iRecovery - Utility for DFU 2.0, WTF and Recovery Mode
- * Copyright (C) 2008 - 2009 westbaer
- *
- * This program is free software: you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation, either version 3 of the License, or
- * (at your option) any later version.
- *
- * This program 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 General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program. If not, see <http://www.gnu.org/licenses/>.
- **/
-
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
-#include <unistd.h>
-#include <libusb-1.0/libusb.h>
-
-#include "libirecovery.h"
-
-#define BUFFER_SIZE 0x1000
-#define debug(...) if(libirecovery_debug) fprintf(stderr, __VA_ARGS__)
-
-static int libirecovery_debug = 0;
-static libusb_context* libirecovery_context = NULL;
-
-int irecv_write_file(const char* filename, const void* data, size_t size);
-int irecv_read_file(const char* filename, char** data, uint32_t* size);
-
-irecv_error_t irecv_open(irecv_client_t* pclient) {
- int i = 0;
- char serial[256];
- struct libusb_device* usb_device = NULL;
- struct libusb_device** usb_device_list = NULL;
- struct libusb_device_handle* usb_handle = NULL;
- struct libusb_device_descriptor usb_descriptor;
-
- *pclient = NULL;
- libusb_init(&libirecovery_context);
- if(libirecovery_debug) {
- irecv_set_debug_level(libirecovery_debug);
- }
-
- irecv_error_t error = IRECV_E_SUCCESS;
- int usb_device_count = libusb_get_device_list(libirecovery_context, &usb_device_list);
- for (i = 0; i < usb_device_count; i++) {
- usb_device = usb_device_list[i];
- libusb_get_device_descriptor(usb_device, &usb_descriptor);
- if (usb_descriptor.idVendor == APPLE_VENDOR_ID) {
- /* verify this device is in a mode we understand */
- if (usb_descriptor.idProduct == kRecoveryMode1 ||
- usb_descriptor.idProduct == kRecoveryMode2 ||
- usb_descriptor.idProduct == kRecoveryMode3 ||
- usb_descriptor.idProduct == kRecoveryMode4 ||
- usb_descriptor.idProduct == kDfuMode) {
-
- libusb_open(usb_device, &usb_handle);
- if (usb_handle == NULL) {
- libusb_free_device_list(usb_device_list, 1);
- libusb_close(usb_handle);
- libusb_exit(libirecovery_context);
- return IRECV_E_UNABLE_TO_CONNECT;
- }
- libusb_free_device_list(usb_device_list, 1);
-
- irecv_client_t client = (irecv_client_t) malloc(sizeof(struct irecv_client));
- if (client == NULL) {
- libusb_close(usb_handle);
- libusb_exit(libirecovery_context);
- return IRECV_E_OUT_OF_MEMORY;
- }
-
- memset(client, '\0', sizeof(struct irecv_client));
- client->interface = 0;
- client->handle = usb_handle;
- client->mode = usb_descriptor.idProduct;
-
- error = irecv_set_configuration(client, 1);
- if (error != IRECV_E_SUCCESS) {
- return error;
- }
-
- error = irecv_set_interface(client, 1, 1);
- if (error != IRECV_E_SUCCESS) {
- return error;
- }
-
- *pclient = client;
- return IRECV_E_SUCCESS;
- }
- }
- }
-
- return IRECV_E_UNABLE_TO_CONNECT;
-}
-
-irecv_error_t irecv_set_configuration(irecv_client_t client, int configuration) {
- if (client == NULL || client->handle == NULL) {
- return IRECV_E_NO_DEVICE;
- }
-
- debug("Setting to configuration %d", configuration);
-
- int current = 0;
- libusb_get_configuration(client->handle, &current);
- if (current != configuration) {
- if (libusb_set_configuration(client->handle, configuration) < 0) {
- return IRECV_E_USB_CONFIGURATION;
- }
- }
-
- client->config = configuration;
- return IRECV_E_SUCCESS;
-}
-
-irecv_error_t irecv_set_interface(irecv_client_t client, int interface, int alt_interface) {
- if (client == NULL || client->handle == NULL) {
- return IRECV_E_NO_DEVICE;
- }
-
- if (client->interface == interface) {
- return IRECV_E_SUCCESS;
- }
-
- debug("Setting to interface %d:%d", interface, alt_interface);
- if (libusb_claim_interface(client->handle, interface) < 0) {
- return IRECV_E_USB_INTERFACE;
- }
-
- if (libusb_set_interface_alt_setting(client->handle, interface, alt_interface) < 0) {
- return IRECV_E_USB_INTERFACE;
- }
-
- client->interface = interface;
- client->alt_interface = alt_interface;
- return IRECV_E_SUCCESS;
-}
-
-irecv_error_t irecv_reset(irecv_client_t client) {
- if (client == NULL || client->handle == NULL) {
- return IRECV_E_NO_DEVICE;
- }
-
- libusb_reset_device(client->handle);
-
- return IRECV_E_SUCCESS;
-}
-
-irecv_error_t irecv_event_subscribe(irecv_client_t client, irecv_event_type type, irecv_event_cb_t callback, void* user_data) {
- switch(type) {
- case IRECV_RECEIVED:
- client->received_callback = callback;
- break;
-
- case IRECV_PROGRESS:
- client->progress_callback = callback;
-
- case IRECV_CONNECTED:
- client->connected_callback = callback;
-
- case IRECV_PRECOMMAND:
- client->precommand_callback = callback;
- break;
-
- case IRECV_POSTCOMMAND:
- client->postcommand_callback = callback;
- break;
-
- case IRECV_DISCONNECTED:
- client->disconnected_callback = callback;
-
- default:
- return IRECV_E_UNKNOWN_ERROR;
- }
-
- return IRECV_E_SUCCESS;
-}
-
-irecv_error_t irecv_event_unsubscribe(irecv_client_t client, irecv_event_type type) {
- switch(type) {
- case IRECV_RECEIVED:
- client->received_callback = NULL;
- break;
-
- case IRECV_PROGRESS:
- client->progress_callback = NULL;
-
- case IRECV_CONNECTED:
- client->connected_callback = NULL;
-
- case IRECV_PRECOMMAND:
- client->precommand_callback = NULL;
- break;
-
- case IRECV_POSTCOMMAND:
- client->postcommand_callback = NULL;
- break;
-
- case IRECV_DISCONNECTED:
- client->disconnected_callback = NULL;
-
- default:
- return IRECV_E_UNKNOWN_ERROR;
- }
-
- return IRECV_E_SUCCESS;
-}
-
-irecv_error_t irecv_close(irecv_client_t client) {
- if (client != NULL) {
- if(client->disconnected_callback != NULL) {
- irecv_event_t event;
- event.size = 0;
- event.data = NULL;
- event.progress = 0;
- event.type = IRECV_DISCONNECTED;
- client->disconnected_callback(client, &event);
- }
-
- if (client->handle != NULL) {
- libusb_release_interface(client->handle, client->interface);
- libusb_close(client->handle);
- client->handle = NULL;
- }
-
- if (libirecovery_context != NULL) {
- libusb_exit(libirecovery_context);
- libirecovery_context = NULL;
- }
-
- free(client);
- client = NULL;
- }
-
- return IRECV_E_SUCCESS;
-}
-
-void irecv_set_debug_level(int level) {
- libirecovery_debug = level;
- if(libirecovery_context) {
- libusb_set_debug(libirecovery_context, libirecovery_debug);
- }
-}
-
-irecv_error_t irecv_send_command(irecv_client_t client, unsigned char* command) {
- if (client == NULL || client->handle == NULL) {
- return IRECV_E_NO_DEVICE;
- }
-
- unsigned int length = strlen(command);
- if (length >= 0x100) {
- length = 0xFF;
- }
-
- irecv_event_t event;
- if(client->precommand_callback != NULL) {
- event.size = length;
- event.data = command;
- event.type = IRECV_PRECOMMAND;
- if(client->precommand_callback(client, &event)) {
- return IRECV_E_SUCCESS;
- }
- }
-
- if (length > 0) {
- libusb_control_transfer(client->handle, 0x40, 0, 0, 0, command, length + 1, 100);
- }
-
- if(client->postcommand_callback != NULL) {
- event.size = length;
- event.data = command;
- event.type = IRECV_POSTCOMMAND;
- if(client->postcommand_callback(client, &event)) {
- return IRECV_E_SUCCESS;
- }
- }
-
- return IRECV_E_SUCCESS;
-}
-
-irecv_error_t irecv_send_file(irecv_client_t client, const char* filename) {
- if (client == NULL || client->handle == NULL) {
- return IRECV_E_NO_DEVICE;
- }
-
- FILE* file = fopen(filename, "rb");
- if (file == NULL) {
- return IRECV_E_FILE_NOT_FOUND;
- }
-
- fseek(file, 0, SEEK_END);
- int length = ftell(file);
- fseek(file, 0, SEEK_SET);
-
- unsigned char* buffer = (unsigned char*) malloc(length);
- if (buffer == NULL) {
- fclose(file);
- return IRECV_E_OUT_OF_MEMORY;
- }
-
- int bytes = fread(buffer, 1, length, file);
- fclose(file);
-
- if (bytes != length) {
- free(buffer);
- return IRECV_E_UNKNOWN_ERROR;
- }
-
- irecv_error_t error = irecv_send_buffer(client, buffer, length);
- free(buffer);
- return error;
-}
-
-irecv_error_t irecv_get_status(irecv_client_t client, unsigned int* status) {
- if (client == NULL || client->handle == NULL) {
- *status = 0;
- return IRECV_E_NO_DEVICE;
- }
-
- unsigned char buffer[6];
- memset(buffer, '\0', 6);
- if (libusb_control_transfer(client->handle, 0xA1, 3, 0, 0, buffer, 6, 1000) != 6) {
- *status = 0;
- return IRECV_E_USB_STATUS;
- }
-
- debug("status: %d\n", (unsigned int) buffer[4]);
- *status = (unsigned int) buffer[4];
- return IRECV_E_SUCCESS;
-}
-
-irecv_error_t irecv_send_buffer(irecv_client_t client, unsigned char* buffer, unsigned int length) {
- irecv_error_t error = 0;
- if (client == NULL || client->handle == NULL) {
- return IRECV_E_NO_DEVICE;
- }
-
- int last = length % 0x800;
- int packets = length / 0x800;
- if (last != 0) {
- packets++;
- }
-
- int i = 0;
- double progress = 0;
- unsigned int count = 0;
- unsigned int status = 0;
- for (i = 0; i < packets; i++) {
- int size = i + 1 < packets ? 0x800 : last;
- int bytes = libusb_control_transfer(client->handle, 0x21, 1, 0, 0, &buffer[i * 0x800], size, 1000);
- if (bytes != size) {
- return IRECV_E_USB_UPLOAD;
- }
-
- error = irecv_get_status(client, &status);
- if (error != IRECV_E_SUCCESS) {
- return error;
- }
-
- if (status != 5) {
- return IRECV_E_USB_UPLOAD;
- }
-
- count += size;
- if(client->progress_callback != NULL) {
- irecv_event_t event;
- event.progress = ((double) count/ (double) length) * 100.0;
- event.type = IRECV_PROGRESS;
- event.data = "Uploading";
- event.size = count;
- client->progress_callback(client, &event);
- } else {
- debug("Sent: %d bytes - %d of %d\n", bytes, count, length);
- }
- }
-
- libusb_control_transfer(client->handle, 0x21, 1, 0, 0, buffer, 0, 1000);
- for (i = 0; i < 3; i++) {
- error = irecv_get_status(client, &status);
- if (error != IRECV_E_SUCCESS) {
- return error;
- }
- }
-
- return IRECV_E_SUCCESS;
-}
-
-irecv_error_t irecv_receive(irecv_client_t client) {
- unsigned char buffer[BUFFER_SIZE];
- memset(buffer, '\0', BUFFER_SIZE);
- if (client == NULL || client->handle == NULL) {
- return IRECV_E_NO_DEVICE;
- }
-
- int bytes = 0;
- while (libusb_bulk_transfer(client->handle, 0x81, buffer, BUFFER_SIZE, &bytes, 100) == 0) {
- if (bytes > 0) {
- if (client->received_callback != NULL) {
- irecv_event_t event;
- event.size = bytes;
- event.data = buffer;
- event.type = IRECV_RECEIVED;
- if (client->received_callback(client, &event) != 0) {
- return IRECV_E_SUCCESS;
- }
- }
- } else break;
- }
-
- return IRECV_E_SUCCESS;
-}
-
-irecv_error_t irecv_getenv(irecv_client_t client, const char* variable, char** value) {
- char command[256];
- if (client == NULL || client->handle == NULL) {
- return IRECV_E_NO_DEVICE;
- }
-
- *value = NULL;
-
- if(variable == NULL) {
- return IRECV_E_UNKNOWN_ERROR;
- }
-
- memset(command, '\0', sizeof(command));
- snprintf(command, sizeof(command)-1, "getenv %s", variable);
- irecv_error_t error = irecv_send_command(client, command);
- if(error != IRECV_E_SUCCESS) {
- return error;
- }
-
- unsigned char* response = (unsigned char*) malloc(256);
- if (response == NULL) {
- return IRECV_E_OUT_OF_MEMORY;
- }
-
- memset(response, '\0', 256);
- int ret = libusb_control_transfer(client->handle, 0xC0, 0, 0, 0, response, 255, 500);
- if (ret < 0) {
- return IRECV_E_UNKNOWN_ERROR;
- }
-
- *value = response;
- return IRECV_E_SUCCESS;
-}
-
-irecv_error_t irecv_get_cpid(irecv_client_t client, unsigned int* cpid) {
- char info[256];
- memset(info, '\0', 256);
-
- if (client == NULL || client->handle == NULL) {
- return IRECV_E_NO_DEVICE;
- }
-
- libusb_get_string_descriptor_ascii(client->handle, 3, info, 255);
-
- unsigned char* cpid_string = strstr(info, "CPID:");
- if (cpid_string == NULL) {
- *cpid = 0;
- return IRECV_E_UNKNOWN_ERROR;
- }
- sscanf(cpid_string, "CPID:%d", cpid);
-
- return IRECV_E_SUCCESS;
-}
-
-irecv_error_t irecv_get_bdid(irecv_client_t client, unsigned int* bdid) {
- char info[256];
- memset(info, '\0', 256);
-
- if (client == NULL || client->handle == NULL) {
- return IRECV_E_NO_DEVICE;
- }
-
- libusb_get_string_descriptor_ascii(client->handle, 3, info, 255);
-
- unsigned char* bdid_string = strstr(info, "BDID:");
- if (bdid_string == NULL) {
- *bdid = 0;
- return IRECV_E_UNKNOWN_ERROR;
- }
- sscanf(bdid_string, "BDID:%d", bdid);
-
- return IRECV_E_SUCCESS;
-}
-
-irecv_error_t irecv_get_ecid(irecv_client_t client, unsigned long long* ecid) {
- char info[256];
- memset(info, '\0', 256);
-
- if (client == NULL || client->handle == NULL) {
- return IRECV_E_NO_DEVICE;
- }
-
- libusb_get_string_descriptor_ascii(client->handle, 3, info, 255);
-
- unsigned char* ecid_string = strstr(info, "ECID:");
- if (ecid_string == NULL) {
- *ecid = 0;
- return IRECV_E_UNKNOWN_ERROR;
- }
- sscanf(ecid_string, "ECID:%qX", ecid);
-
- return IRECV_E_SUCCESS;
-}
-
-irecv_error_t irecv_send_exploit(irecv_client_t client) {
- if (client == NULL || client->handle == NULL) {
- return IRECV_E_NO_DEVICE;
- }
-
- libusb_control_transfer(client->handle, 0x21, 2, 0, 0, NULL, 0, 100);
- return IRECV_E_SUCCESS;
-}
-
-irecv_error_t irecv_execute_script(irecv_client_t client, const char* filename) {
- irecv_error_t error = IRECV_E_SUCCESS;
- if (client == NULL || client->handle == NULL) {
- return IRECV_E_NO_DEVICE;
- }
-
- int file_size = 0;
- char* file_data = NULL;
- if(irecv_read_file(filename, &file_data, &file_size) < 0) {
- return IRECV_E_FILE_NOT_FOUND;
- }
-
- char* line = strtok(file_data, "\n");
- while(line != NULL) {
- if(line[0] != '#') {
- error = irecv_send_command(client, line);
- if(error != IRECV_E_SUCCESS) {
- return error;
- }
-
- error = irecv_receive(client);
- if(error != IRECV_E_SUCCESS) {
- return error;
- }
- }
- line = strtok(NULL, "\n");
- }
-
- return IRECV_E_SUCCESS;
-}
-
-irecv_error_t irecv_setenv(irecv_client_t client, const char* variable, const char* value) {
- char command[256];
- if (client == NULL || client->handle == NULL) {
- return IRECV_E_NO_DEVICE;
- }
-
- if(variable == NULL || value == NULL) {
- return IRECV_E_UNKNOWN_ERROR;
- }
-
- memset(command, '\0', sizeof(command));
- snprintf(command, sizeof(command)-1, "setenv %s %s", variable, value);
- irecv_error_t error = irecv_send_command(client, command);
- if(error != IRECV_E_SUCCESS) {
- return error;
- }
-
- return IRECV_E_SUCCESS;
-}
-
-const char* irecv_strerror(irecv_error_t error) {
- switch (error) {
- case IRECV_E_SUCCESS:
- return "Command completed successfully";
-
- case IRECV_E_NO_DEVICE:
- return "Unable to find device";
-
- case IRECV_E_OUT_OF_MEMORY:
- return "Out of memory";
-
- case IRECV_E_UNABLE_TO_CONNECT:
- return "Unable to connect to device";
-
- case IRECV_E_INVALID_INPUT:
- return "Invalid input";
-
- case IRECV_E_FILE_NOT_FOUND:
- return "File not found";
-
- case IRECV_E_USB_UPLOAD:
- return "Unable to upload data to device";
-
- case IRECV_E_USB_STATUS:
- return "Unable to get device status";
-
- case IRECV_E_USB_INTERFACE:
- return "Unable to set device interface";
-
- case IRECV_E_USB_CONFIGURATION:
- return "Unable to set device configuration";
-
- default:
- return "Unknown error";
- }
-
- return NULL;
-}
-
-int irecv_write_file(const char* filename, const void* data, size_t size) {
- size_t bytes = 0;
- FILE* file = NULL;
-
- debug("Writing data to %s\n", filename);
- file = fopen(filename, "wb");
- if (file == NULL) {
- error("read_file: Unable to open file %s\n", filename);
- return -1;
- }
-
- bytes = fwrite(data, 1, size, file);
- fclose(file);
-
- if (bytes != size) {
- error("ERROR: Unable to write entire file: %s: %d of %d\n", filename, bytes, size);
- return -1;
- }
-
- return size;
-}
-
-int irecv_read_file(const char* filename, char** data, uint32_t* size) {
- size_t bytes = 0;
- size_t length = 0;
- FILE* file = NULL;
- char* buffer = NULL;
- debug("Reading data from %s\n", filename);
-
- *size = 0;
- *data = NULL;
-
- file = fopen(filename, "rb");
- if (file == NULL) {
- error("read_file: File %s not found\n", filename);
- return -1;
- }
-
- fseek(file, 0, SEEK_END);
- length = ftell(file);
- rewind(file);
-
- buffer = (char*) malloc(length);
- if(buffer == NULL) {
- error("ERROR: Out of memory\n");
- fclose(file);
- return -1;
- }
- bytes = fread(buffer, 1, length, file);
- fclose(file);
-
- if(bytes != length) {
- error("ERROR: Unable to read entire file\n");
- free(buffer);
- return -1;
- }
-
- *size = length;
- *data = buffer;
- return 0;
-}