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
 
//-----------------------------------------------------------------
// download
//-----------------------------------------------------------------
static int download(FILE *f, uint32_t addr, int length)
{
    uint8_t buf[CHUNK_SIZE];
    int err = 0;
    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, "Download: Error downloading file\n");
            err = 1;
            break;
        }
 
        if (fwrite(buf, 1, size, f) != size)
        {
            fprintf(stderr, "Download: Error writing file\n");
            err = 1;
            break;
        }
 
        addr += CHUNK_SIZE;
 
        printf("\r%d%%", (i * 100) / length);
        fflush(stdout);
    }
 
    return err;
}
//-----------------------------------------------------------------
// main
//-----------------------------------------------------------------
int main(int argc, char *argv[])
{
    int c;
    int size = -1;
    char *filename = NULL;
    int help = 0;
    int err = 1;
    uint32_t address = 0x0;
    int ftdi_iface = DEFAULT_FTDI_IFACE;
    FILE *f = NULL;
 
    while ((c = getopt (argc, argv, "o:s:a:i:")) != -1)
    {
        switch(c)
        {
            case 'o':
                 filename = optarg;
                 break;
            case 's':
                 size = (int)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 || size < 0)
    {
        fprintf (stderr,"Usage:\n");
        fprintf (stderr,"-o filename.bin   = Output filename\n");
        fprintf (stderr,"-s n              = Size to dump\n");
        fprintf (stderr,"-a 0xnnnn         = Address to dump from (default to 0x0)\n");
        fprintf (stderr,"-i id             = FTDI interface ID (0 = A, 1 = B)\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);
    }
 
    // Try and create new file
    f = fopen(filename, "wb");
    if (f)
    {
        printf("Downloading %s (%dKB) from 0x%x:\n", filename, (size + 1023) / 1024, address);
 
        err = download(f, address, size);
 
        if (!err)
            printf("\rDone!\n");
        else
            printf("\rFailed!\n");
 
        fclose(f);
        f = NULL;
    }
    else
    {
        fprintf (stderr,"Error: Could not create file\n");
        err = 1;
    }
 
    ftdi_hw_close();
 
    return err;
}