#include <libusb.h>
#include <glib.h>


static void dump_line (const guchar *data, gsize line_len)
{
    unsigned int i;

    for (i = 0; i < line_len; i++) {
        g_print ("%02x ", data[i]);
    } 
    for (i = 0; i < line_len; i++) {
        if (data[i] >= 0x20) {
            g_print ("%c", data[i]);
        } else {
            g_print (".");
        }
    } 
    g_print ("\n");
}

static void hexdump (const guchar *data, gsize len)
{
    unsigned int line;
    const unsigned int BYTES_PER_LINE = 16;

    for (line = 0; line < len/BYTES_PER_LINE; line++) {
        dump_line (data, BYTES_PER_LINE);
        data += BYTES_PER_LINE;
    }
    dump_line (data, len % BYTES_PER_LINE);
}


static int send_command (libusb_device_handle *handle, const guint value, const guint index, guchar *data, gsize len)
{
    return libusb_control_transfer (handle, LIBUSB_ENDPOINT_IN | LIBUSB_REQUEST_TYPE_VENDOR | LIBUSB_RECIPIENT_DEVICE, 0x40, value, index, data, len, 0);
}

static libusb_device_handle *open_handle (libusb_context *ctx, const guint bus_number, const guint device_address)
{
    libusb_device **list;
    libusb_device *dev;
    libusb_device_handle *handle;
    unsigned int i;

    int success;
    int device_count;

    device_count = libusb_get_device_list (ctx, &list);
    if (device_count < 0) {
        g_print ("Couldn't get device list\n");
        return NULL;
    }

    dev = NULL;
    for (i = 0; i < device_count; i++) {
        if (libusb_get_bus_number (list[i]) != bus_number) {
            continue;
        }
        if (libusb_get_device_address (list[i]) != device_address) {
            continue;
        }
        dev = list[i];
        break;
    }

    if (dev == NULL) {
        g_print ("Couldn't find iPod\n");
        return NULL;
    }

    success = libusb_open (dev, &handle);
    libusb_free_device_list (list, 1);
    if (success < 0) {
        g_print ("Couldn't open usb device: %d\n", success);
        return NULL;
    }

    return handle;
}

static const guint USB_BUS_NUMBER = 2;
static const guint USB_DEVICE_ADDRESS = 7;

gchar *get_sysinfo_extended (libusb_device_handle *handle)
{
    GString *sysinfo_extended;
    guchar data[0x1000];
    unsigned int i; 

    sysinfo_extended = g_string_sized_new (0x8000);

    for (i = 0; i < 0xffff; i++) {
        int bytes_read;
        bytes_read = send_command (handle, 0x02, i, data, sizeof (data));
        if (bytes_read < 0) {
            g_print ("error reading: %d (i = %d)\n", bytes_read, i);
            return NULL;
        }
        g_print ("Read %d bytes (i = %d)\n", bytes_read, i);
        g_string_append_len (sysinfo_extended, (gchar*)data, bytes_read);
        if (bytes_read != sizeof (data)) {
            /* assume short reads mean "end of file" */
            break;
        }
    }
    hexdump ((guchar *)sysinfo_extended->str, sysinfo_extended->len);
    return g_string_free (sysinfo_extended, FALSE);
}

int main (int argc, char **argv)
{
    int success;
    libusb_context *ctx;
    libusb_device_handle *handle;
    gchar *sysinfo_extended;

    success = libusb_init (&ctx);
    if (success < 0) {
        g_print ("Couldn't get a libusb context\n");
        return 1;

    }
    libusb_set_debug (ctx, 3);

    handle = open_handle (ctx, USB_BUS_NUMBER, USB_DEVICE_ADDRESS);
    if (handle == NULL) {
        return 1;
    }

    sysinfo_extended = get_sysinfo_extended (handle);
    g_print ("%s\n", sysinfo_extended);

    libusb_close (handle);
    libusb_exit (ctx);

    return 0;
}