Subversion Repositories freemyipod

#include <stdio.h>

#include <libusb-1.0/libusb.h>

#include "misc.h"
#include "usb.h"

libusb_context *usb_ctx = NULL;
libusb_device_handle *usb_handle = NULL;

int usb_init(void) {
        int res;
        
        printf("Initialising USB library... ");
        
        res = libusb_init(&usb_ctx);
        
        if (LIBUSB_SUCCESS != res) {
                return res;
        }
        
        printf("OK\n");
        
#ifdef DEBUG
        libusb_set_debug(usb_ctx, 3);
#endif

        return LIBUSB_SUCCESS;
}

int usb_find(unsigned char *reattach) {
        libusb_device **devs, *dev;
        ssize_t devs_cnt;
        int res, i;
        struct libusb_device_descriptor dev_desc;
        unsigned char found = 0;
        struct libusb_config_descriptor *cfg_desc;
        const struct libusb_interface *iface;
        const struct libusb_interface_descriptor *iface_desc;
        
        printf("Getting USB device list... ");
        
        devs_cnt = libusb_get_device_list(usb_ctx, &devs);
                
        if (devs_cnt < 0) {
                return devs_cnt;
        }
        
        printf("Found %d USB devices!\n", devs_cnt);
                        
        for (i = 0; i < devs_cnt; ++i) {
                dev = devs[i];
                
                printf("Getting device descriptor of USB device %d...\n", i);

                res = libusb_get_device_descriptor(dev, &dev_desc);

                if (LIBUSB_SUCCESS != res) {
                        fprintf(stderr, "Unable to get device descriptor of device %d!\n", i);
                        
                        continue;
                }
                
                printf("[%04x:%04x] bus %d, device %d, USB ver. %04x\n", dev_desc.idVendor,
                        dev_desc.idProduct, libusb_get_bus_number(dev),
                        libusb_get_device_address(dev), dev_desc.bcdUSB);

                if (0x05ac == dev_desc.idVendor && (
                        /* DFU */
                        0x1220 == dev_desc.idProduct || /* iPod Nano 2G */
                        0x1223 == dev_desc.idProduct || /* iPod Classic 1G */
                        0x1224 == dev_desc.idProduct || /* iPod Nano 3G */
                        0x1225 == dev_desc.idProduct || /* iPod Nano 4G */
                        0x1231 == dev_desc.idProduct || /* iPod Nano 5G */
                        0x1232 == dev_desc.idProduct || /* iPod Nano 6G */
                        0x1233 == dev_desc.idProduct || /* iPod Shuffle 4G */
                        /* WTF */
                        0x1240 == dev_desc.idProduct || /* iPod Nano 2G */
                        0x1241 == dev_desc.idProduct || /* iPod Classic 1G */
                        0x1242 == dev_desc.idProduct || /* iPod Nano 3G */
                        0x1243 == dev_desc.idProduct || /* iPod Nano 4G */
                        0x1245 == dev_desc.idProduct || /* iPod Classic 2G */
                        0x1246 == dev_desc.idProduct || /* iPod Nano 5G */
                        0x1247 == dev_desc.idProduct || /* iPod Classic 3G */
                        0x1248 == dev_desc.idProduct    /* iPod Nano 6G */
                )) {
                        printf("Found DFU USB device!\n");
                        
                        if (1 != dev_desc.bNumConfigurations) {
                                fprintf(stderr, "Number of configs is different than 1, not the right device...\n");
                                
                                continue;
                        }
                        
                        printf("Getting config descriptor 0 of device...\n");

                        res = libusb_get_config_descriptor(dev, 0, &cfg_desc);

                        if (LIBUSB_SUCCESS != res) {
                                return res;
                        }
                        
                        if (1 != cfg_desc->bNumInterfaces) {
                                fprintf(stderr, "Wrong USB device, it should have exactly 1 interface\n");
                                
                                continue;
                        }
                        
                        iface = &cfg_desc->interface[0];

                        if (1 != iface->num_altsetting) {
                                fprintf(stderr, "Wrong USB device, it should have exactly 1 altsetting\n");
                                
                                continue;
                        }
                        
                        iface_desc = &iface->altsetting[0];
                        
                        if (0 != iface_desc->bNumEndpoints) {
                                fprintf(stderr, "Wrong USB device, it should have no endpoints\n");
                                
                                continue;
                        }
                        
                        found = 1;
                }
        }

        if (found) {
                res = usb_open(dev, reattach);
        }
        else {
                fprintf(stderr, "DFU USB device not found!\n");
                
                res = 1; // not found
        }
        
        printf("Freeing device list...\n");

        libusb_free_device_list(devs, 1);

        return res;
}

int usb_open(libusb_device *dev, unsigned char *reattach) {
        int res;
        
        printf("Opening USB device... ");

        res = libusb_open(dev, &usb_handle);

        if (LIBUSB_SUCCESS != res) {
                return res;
        }
        
        printf("OK\n");

        printf("Setting USB configuration 1... ");

        res = libusb_set_configuration(usb_handle, 1);

        if (LIBUSB_SUCCESS != res) {
                return res;
        }
        
        printf("OK\n");

        res = libusb_kernel_driver_active(usb_handle, 0);

        if (1 == res) {
                printf("Kernel driver active, detaching... ");
                
                *reattach = 1;

                res = libusb_detach_kernel_driver(usb_handle, 0);
                
                if (LIBUSB_SUCCESS == res) {
                        printf("OK\n");
                }
        }

        if (LIBUSB_SUCCESS != res) {
                return res;
        }
        
        printf("Claiming interface 0... ");

        res = libusb_claim_interface(usb_handle, 0);

        if (LIBUSB_SUCCESS != res) {
                return res;
        }
        
        printf("OK\n");
        
        return LIBUSB_SUCCESS;
}

int usb_bulk_transfer(const unsigned char endpoint, unsigned char *data, const unsigned int length) {
        int transferred;
        int res;
        
        res = libusb_bulk_transfer(usb_handle, endpoint, data, length, &transferred, 30000);
        
        if (LIBUSB_SUCCESS != res) {
                return res;
        }
        
        if ((unsigned int) transferred != length) {
                return 2; // incomplete
        }
        
        return LIBUSB_SUCCESS;
}

int usb_control_transfer(uint8_t bmRequestType, uint8_t bRequest, uint16_t wValue, uint16_t wIndex, unsigned char *data, uint16_t wLength) {
        int res;
        
        res = libusb_control_transfer(usb_handle, bmRequestType, bRequest, wValue, wIndex, data, wLength, 30000);
        
        if (LIBUSB_SUCCESS != res) {
                return res;
        }
        
        return LIBUSB_SUCCESS;
}

int usb_close(const unsigned char reattach) {
        int res;
        
        printf("Releasing USB interface... ");
        
        res = libusb_release_interface(usb_handle, 0);
        
        if (LIBUSB_SUCCESS != res) {
                return res;
        }
        
        printf("OK\n");
        
        if (reattach) {
                printf("Reattaching kernel driver... ");
                
                res = libusb_attach_kernel_driver(usb_handle, 0);
                
                if (LIBUSB_SUCCESS == res) {
                        printf("OK\n");
                }
                else {
                        printf("\n");
                        print_error(res);
                        
                        res = LIBUSB_SUCCESS;
                }
        }
        
        printf("Closing USB device handle...\n");
        
        libusb_close(usb_handle);
        
        return res;
}

void usb_exit(void) {
        printf("Deinitializing USB library...\n");

        libusb_exit(usb_ctx);
}