Subversion Repositories usb_nand_reader

#include <stdio.h>

#include <libusb-1.0/libusb.h>

#include <stdlib.h>

#include <string.h>

#include <unistd.h>

#include "include/opcodes.h"

#include "include/actions.h"

#include "include/nand_vendors.h"

int main(int argc, char **argv)

{

        FILE* dump;

        libusb_context* ctx;

        libusb_device** list;

        libusb_device_handle* hDevice = NULL;

        struct libusb_device_descriptor descr;

        int                     count, i;

        unsigned char   bufferOut[64];

        unsigned char   status;

        pnand_t                 nand;

        libusb_init(&ctx);

        count = libusb_get_device_list(ctx, &list);

        if(0 == count)

        {

                fprintf(stderr, "Could not get device list\n");

                exit(-1);

        }

        for(i = 0; i < count; i++)

        {

                libusb_device* tmpDevice = list[i];

                libusb_get_device_descriptor(tmpDevice, &descr);

                if(0x1234 == descr.idVendor && 0x1 == descr.idProduct)

                {

                        libusb_open(tmpDevice, &hDevice);

                        break;

                }

        }

        libusb_free_device_list(list,1);

        if(NULL == hDevice)

        {

                fprintf(stderr, "Could not find the device\n");

                exit(-1);

        }

        else

        {

                printf("Device opened\n");

        }

        if(libusb_kernel_driver_active(hDevice, 0) == 1)

        {

                printf("Detaching driver\n");

                libusb_detach_kernel_driver(hDevice, 0);

        }

        libusb_claim_interface(hDevice, 0);

        //libusb_reset_device(hDevice);

        //==============================

        // USB initialization completed.

        //==============================

        nand = (pnand_t)malloc(sizeof(nand_t));

        memset(nand, 0, sizeof(nand_t));

        memset(bufferOut, 0, 64);

        nand_enable(hDevice, 0);

        nand_reset(hDevice);

        nand_read_id(hDevice, nand->id);

        if(nand_is_onfi(hDevice, nand->onfiSignature))

        {

                nand->onfiParameterPage = (unsigned char*)malloc(sizeof(unsigned char) * 256);

                nand_read_onfi_param_page(hDevice, nand->onfiParameterPage);

        }

        if(1 != get_nand_configuration(nand))

        {

                printf("Could not get configuration for this chip\n");

                nand_disble(hDevice);

                libusb_close(hDevice);

                libusb_exit(ctx);

                return -1;

        }

        // Set configuration data if chip is not ONFI compliant or has corrupted ONFI parameter page/signature

        if(NULL == nand->onfiParameterPage)

        {

                printf("Setting configuration data\n");

                nand_set_config_data(hDevice, nand);

        }

        print_nand_configuration(nand);

        /*

        printf("Status: %02x\n", nand_read_status(hDevice, NULL));

        nand_toggle_wp(hDevice);

        fprintf(stderr, "Erasing block\n");

        nand_block_erase(hDevice, 0);

        //sleep(1);

        printf("Status: %02x\n", nand_read_status_enhanced(hDevice, 0, NULL));

        fprintf(stderr, "Programming page\n");

        {

                int x;

                for(x = 0; x < nand->bytesPerPage + nand->oobPerPage; x++)

                {

                        nand->pageBuffer[x] = (unsigned char)((x) & 0xf0);

                }

        }

        nand_page_program(hDevice, 0, nand);

        printf("Status: %02x\n", nand_read_status_enhanced(hDevice, 0, NULL));

        nand_toggle_wp(hDevice);

        */

        printf("Dumping flash\n");

        i = 0;

        dump = fopen("flash_dump.bin", "wb");

        nand_read_status_enhanced(hDevice, 0, &status);

        printf("Status: %02x\n", status);

        /*

        for(count = 0; count < nand->pagesPerBlock * nand->blocksPerPlane * nand->planesPerLun * nand->numLuns; count++)

        {

                if(count % 64 == 0)

                {

                        printf("\rBlock #%d", i++);

                        fflush(stdout);

                }

                nand_read_page(hDevice, (unsigned int)count, nand);

                store_page(dump, nand);

        }

        */

        printf("Dumping %d pages\n", nand->pagesPerBlock * nand->blocksPerPlane * nand->planesPerLun * nand->numLuns);

        //nand_read_page_cache(hDevice, 0, nand->pagesPerBlock * nand->blocksPerPlane * nand->planesPerLun * nand->numLuns, dump, nand);

        nand_read_page_cache(hDevice, 0, 640, dump, nand);

        nand_read_status(hDevice, &status);

        printf("Status: %02x\n", status);

        fclose(dump);

        printf("\rdone.\n");

        free_nand_info(&nand);

        nand_disble(hDevice);

        //=================

        // Close USB stuff.

        //=================

        libusb_close(hDevice);

        libusb_exit(ctx);

        return 0;

}