diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/irecovery.c | 55 | ||||
| -rw-r--r-- | src/libirecovery.c | 109 |
2 files changed, 164 insertions, 0 deletions
diff --git a/src/irecovery.c b/src/irecovery.c new file mode 100644 index 0000000..dfcf422 --- /dev/null +++ b/src/irecovery.c @@ -0,0 +1,55 @@ +/** + * 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 <libirecovery.h> + +int main(int argc, char** argv) { + irecv_device* device = NULL; + if(irecv_init(&device) < 0) { + fprintf(stderr, "Unable to initialize libirecovery\n"); + return -1; + } + + if(irecv_open(device) < 0) { + fprintf(stderr, "Unable to open device\n"); + return -1; + } + + switch (device->mode) { + case kRecoveryMode: + printf("Found device in recovery mode\n"); + break; + + case kDfuMode: + printf("Found device in DFU mode\n"); + break; + + case kKernelMode: + printf("Found device in kernel mode\n"); + break; + + default: + printf("No device found\n"); + break; + } + + irecv_exit(device); + return 0; +} + diff --git a/src/libirecovery.c b/src/libirecovery.c new file mode 100644 index 0000000..dfc0aea --- /dev/null +++ b/src/libirecovery.c @@ -0,0 +1,109 @@ +/** + * 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 <libusb-1.0/libusb.h> + +#include "libirecovery.h" + +static unsigned int debug = 1; + +int irecv_init(irecv_device** p_device) { + struct libusb_context* usb_context = NULL; + + libusb_init(&usb_context); + if (debug) + libusb_set_debug(usb_context, 3); + + irecv_device* device = (irecv_device*) malloc(sizeof(irecv_device)); + if (device == NULL) { + *p_device = NULL; + return IRECV_ERROR_OUT_OF_MEMORY; + } + memset(device, '\0', sizeof(irecv_device)); + device->context = usb_context; + + *p_device = device; + return IRECV_SUCCESS; +} + +int irecv_open(irecv_device* device) { + int i = 0; + int usb_device_count = 0; + 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; + + if (device == NULL || device->context == NULL) { + return IRECV_ERROR_NO_DEVICE; + } + + usb_device_count = libusb_get_device_list(device->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 == kAppleVendorId) { + + libusb_open(usb_device, &usb_handle); + if (usb_handle == NULL) { + libusb_free_device_list(usb_device_list, 1); + return IRECV_ERROR_UNABLE_TO_CONNECT; + } + + libusb_free_device_list(usb_device_list, 1); + device->mode = usb_descriptor.idProduct; + device->handle = usb_handle; + return IRECV_SUCCESS; + } + } + + return IRECV_ERROR_NO_DEVICE; +} + +int irecv_close(irecv_device* device) { + if (device != NULL) { + if (device->handle != NULL) { + libusb_close(device->handle); + device->handle = NULL; + } + } + + return IRECV_SUCCESS; +} + +int irecv_exit(irecv_device* device) { + if (device != NULL) { + if (device->handle != NULL) { + libusb_close(device->handle); + device->handle = NULL; + } + + if (device->context != NULL) { + libusb_exit(device->context); + device->context = NULL; + } + + free(device); + device = NULL; + } + + return IRECV_SUCCESS; +} |