Subversion Repositories freemyipod

//
//
//    Copyright 2011 user890104
//
//
//    This file is part of ipoddfu.
//
//    ipoddfu 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 2 of the
//    License, or (at your option) any later version.
//
//    ipoddfu 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 ipoddfu.  If not, see <http://www.gnu.org/licenses/>.
//
//


#include <stdio.h>
#include <string.h>

#include <libusb-1.0/libusb.h>

#include "usb.h"

int dfu_write(const unsigned char i, const unsigned int length, const unsigned char *data)
{
    return usb_control_transfer(0x21, 1, i, 0, (unsigned char *) data, length);
}

int dfu_read(unsigned char result[6])
{
    return usb_control_transfer(0xa1, 3, 0, 0, result, 6);
}

int dfu_send(const unsigned long int size, const unsigned char *data)
{
    unsigned char result[6];
    unsigned int i;
    int res;
    
    printf("Uploading... ");
    fflush(stdout);

    for (i = 0; i < (size + 4 + 2047) / 2048; ++i)
    {
        res = dfu_write(i, ((i + 1) * 2048 > size + 4) ? (size + 4) - (i * 2048) : 2048,
            (unsigned char *) (data + (i * 2048)));
        
        if (LIBUSB_SUCCESS > res)
        {
            return res;
        }
        
        memset(result, 0, sizeof(result));
        
        while ('\x05' != result[4])
        {
            res = dfu_read(result);
        
            if (LIBUSB_SUCCESS > res)
            {
                return res;
            }
        }
        
        printf("#");
        fflush(stdout);
    }
    
    res = dfu_write(i, 0, NULL);

    if (LIBUSB_SUCCESS > res)
    {
        return res;
    }
    
    memset(result, 0, sizeof(result));

    i = 0;

    while ('\x02' != result[4] && i++ < 1000)
    {
        dfu_read(result);
        
        if (LIBUSB_SUCCESS > res)
        {
            return res;
        }
    }

    if (1000 == i || '\x02' == result[4])
    {
        printf(" failed: %d / %d\n", result[4], result[0]);
    }
    else
    {
        printf(" OK\n");
    }
    
    return 0;
}