Subversion Repositories ftdi_wb_bridge

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include "ftdi_hw.h"
 
//-----------------------------------------------------------------
// Defines:
//-----------------------------------------------------------------
#define CHUNK_SIZE          256
#define DEFAULT_FTDI_IFACE  1
 
//-----------------------------------------------------------------
// load_file_to_mem
//-----------------------------------------------------------------
static uint8_t* load_file_to_mem(const char *filename, long size_override, int *pSize)
{
    uint8_t *buf = NULL;
    FILE *f = fopen(filename, "rb");
 
    *pSize = 0;
 
    if (f)
    {
        long size;
 
        // Get size of file
        fseek(f, 0, SEEK_END);
        size = ftell(f);
        rewind(f);
 
        // User overriden file size
        if (size_override >= 0)
        {
            if (size > size_override)
                size = size_override;
        }
 
        buf = (uint8_t*)malloc(size);
        if (buf)
        {
            // Read file data into allocated memory
            int len = fread(buf, 1, size, f);
            if (len != size)
            {
                free(buf);
                buf = NULL;
            }
            else
                *pSize = size;
        }
        fclose(f);
    }
 
    return buf;
}
//-----------------------------------------------------------------
// compare
//-----------------------------------------------------------------
static int compare(uint32_t addr, uint8_t *data, int length)
{
    uint8_t buf[CHUNK_SIZE];
    int res = 1;
    int i;
    int size;
 
    for (i=0;i<length;i+=CHUNK_SIZE)
    {
        size = (length - i);
        if (size > CHUNK_SIZE)
            size = CHUNK_SIZE;
 
        if (ftdi_hw_mem_read(addr, buf, size) != size)
        {
            fprintf(stderr, "Compare: Error downloading file\n");
            res = -1;
            break;
        }
 
        // Check for differences
        if (memcmp(data, buf, size) != 0)
        {
            res = 0;
            break;
        }
 
        addr += CHUNK_SIZE;
        data += CHUNK_SIZE;
 
        printf("\r%d%%", (i * 100) / length);
        fflush(stdout);
    }
 
    return res;
}
//-----------------------------------------------------------------
// main
//-----------------------------------------------------------------
int main(int argc, char *argv[])
{
    int c;
    long size_override = -1;
    char *filename = NULL;
    int help = 0;
    int err = 1;
    int res;
    int size;
    uint32_t address = 0x0;
    uint8_t *buf;
    int ftdi_iface = DEFAULT_FTDI_IFACE;
 
    while ((c = getopt (argc, argv, "f:s:a:i:")) != -1)
    {
        switch(c)
        {
            case 'f':
                 filename = optarg;
                 break;
            case 's':
                 size_override = strtol(optarg, NULL, 0);
                 break;
            case 'a':
                address = strtoul(optarg, NULL, 0);
                break;
            case 'i':
                 ftdi_iface = (int)strtol(optarg, NULL, 0);
                 break;                
            default:
                help = 1;   
                break;
        }
    }
 
    if (help || filename == NULL)
    {
        fprintf (stderr,"Usage:\n");
        fprintf (stderr,"-f filename.bin   = Executable to compare (binary)\n");
        fprintf (stderr,"-a 0xnnnn         = Address to compare to (default to 0x0)\n");
        fprintf (stderr,"-i id             = FTDI interface ID (0 = A, 1 = B)\n");
        fprintf (stderr,"-s 0xnnnn         = Size override\n");
 
        exit(-1);
    }
 
    // Try and communicate with FTDI interface
    if (ftdi_hw_init(ftdi_iface) != 0)
    {
        fprintf(stderr, "ERROR: Could not open FTDI interface, try SUDOing / check connection\n");
        exit(-2);
    }
 
    // Read file into memory
    buf = load_file_to_mem(filename, size_override, &size);
    if (buf)
    {
        printf("Comparing %s (%dKB) to 0x%x:\n", filename, (size + 1023) / 1024, address);
 
        // Upload file to target
        res = compare(address, buf, size);
 
        // Free file memory
        free(buf);
        buf = NULL;
 
        if (res == 1)
            printf("\nMatches!\n");
        else if (res == 0)
            printf("\nDiffers!\n");
        else
        {
            printf("\n");
            err = 1;
        }
    }
    else
    {
        fprintf (stderr,"Error: Could not open image\n");
        err = 1;
    }
 
    ftdi_hw_close();
 
    return err;
}