414 lines
12 KiB
C
414 lines
12 KiB
C
/****************************************************************************
|
|
* examples/media/hello_main.c
|
|
*
|
|
* Copyright (C) 2015 Gregory Nutt. All rights reserved.
|
|
* Author: Gregory Nutt <gnutt@nuttx.org>
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
*
|
|
* 1. Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in
|
|
* the documentation and/or other materials provided with the
|
|
* distribution.
|
|
* 3. Neither the name NuttX nor the names of its contributors may be
|
|
* used to endorse or promote products derived from this software
|
|
* without specific prior written permission.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
|
|
* FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
|
|
* COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
|
|
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
|
|
* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
|
|
* OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
|
|
* AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
|
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
|
|
* ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
* POSSIBILITY OF SUCH DAMAGE.
|
|
*
|
|
****************************************************************************/
|
|
|
|
/****************************************************************************
|
|
* Included Files
|
|
****************************************************************************/
|
|
|
|
#include <nuttx/config.h>
|
|
|
|
#include <sys/ioctl.h>
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <fcntl.h>
|
|
#include <errno.h>
|
|
|
|
#include <nuttx/fs/fs.h>
|
|
#include <nuttx/mtd/mtd.h>
|
|
|
|
/****************************************************************************
|
|
* Pre-processor Definitions
|
|
****************************************************************************/
|
|
|
|
#define START_VALUE 0x20
|
|
#define END_VALUE 0x7f
|
|
|
|
/****************************************************************************
|
|
* Private types
|
|
****************************************************************************/
|
|
|
|
struct media_info_s
|
|
{
|
|
off_t blocksize;
|
|
off_t nblocks;
|
|
};
|
|
|
|
/****************************************************************************
|
|
* Private Functions
|
|
****************************************************************************/
|
|
|
|
/****************************************************************************
|
|
* get_blocksize
|
|
****************************************************************************/
|
|
|
|
static void get_blocksize(int fd, FAR struct media_info_s *info)
|
|
{
|
|
struct mtd_geometry_s mtdgeo;
|
|
int ret;
|
|
|
|
/* If the underlying driver is an MTD driver, then we should able to get
|
|
* the erase block size from an MTD ioctl call.
|
|
*/
|
|
|
|
ret = ioctl(fd, MTDIOC_GEOMETRY, (unsigned long)((uintptr_t)&mtdgeo));
|
|
if (ret >= 0)
|
|
{
|
|
/* Its an MTD device. Use its geometry */
|
|
|
|
printf("MTD Geometry:\n");
|
|
printf(" blocksize: %u\n", (unsigned int)mtdgeo.blocksize);
|
|
printf(" erasesize: %lu\n", (unsigned long)mtdgeo.erasesize);
|
|
printf(" neraseblocks: %lu\n", (unsigned long)mtdgeo.neraseblocks);
|
|
|
|
info->blocksize = mtdgeo.erasesize;
|
|
info->nblocks = mtdgeo.neraseblocks;
|
|
|
|
/* Attempt to erase the entire MTD device */
|
|
|
|
ret = ioctl(fd, MTDIOC_BULKERASE, 0);
|
|
if (ret < 0)
|
|
{
|
|
fprintf(stderr, "ERROR: Failed erase the MTD device\n");
|
|
}
|
|
}
|
|
|
|
/* Otherwise, use the configured default. We have no idea of the size
|
|
* of the media in this case.
|
|
*/
|
|
|
|
else
|
|
{
|
|
info->blocksize = CONFIG_EXAMPLES_MEDIA_BLOCKSIZE;
|
|
info->nblocks = 0;
|
|
}
|
|
}
|
|
|
|
/****************************************************************************
|
|
* Public Functions
|
|
****************************************************************************/
|
|
|
|
/****************************************************************************
|
|
* media_main
|
|
****************************************************************************/
|
|
|
|
#ifdef CONFIG_BUILD_KERNEL
|
|
int main(int argc, FAR char *argv[])
|
|
#else
|
|
int media_main(int argc, FAR char *argv[])
|
|
#endif
|
|
{
|
|
FAR uint8_t *txbuffer;
|
|
FAR uint8_t *rxbuffer;
|
|
FAR char *devpath;
|
|
struct media_info_s info;
|
|
ssize_t nwritten;
|
|
ssize_t nread;
|
|
off_t blockno;
|
|
off_t pos;
|
|
off_t seekpos;
|
|
off_t i;
|
|
uint32_t nerrors;
|
|
uint8_t value;
|
|
int fd;
|
|
|
|
/* Open the character driver that wraps the media */
|
|
|
|
if (argc < 2)
|
|
{
|
|
devpath = CONFIG_EXAMPLES_MEDIA_DEVPATH;
|
|
}
|
|
else
|
|
{
|
|
devpath = argv[1];
|
|
}
|
|
|
|
fd = open(argv[1], O_RDWR);
|
|
|
|
if (fd < 0)
|
|
{
|
|
fprintf(stderr, "ERROR: failed to open %s: %d\n",
|
|
devpath, errno);
|
|
return 1;
|
|
}
|
|
/* Get the block size to use */
|
|
|
|
get_blocksize(fd, &info);
|
|
|
|
printf("Using:\n");
|
|
printf(" blocksize: %lu\n", (unsigned long)info.blocksize);
|
|
printf(" nblocks: %lu\n", (unsigned long)info.nblocks);
|
|
|
|
/* Allocate I/O buffers of the correct block size */
|
|
|
|
txbuffer = (FAR uint8_t *)malloc((size_t)info.blocksize);
|
|
if (txbuffer == NULL)
|
|
{
|
|
fprintf(stderr, "ERROR: failed to allocate TX I/O buffer of size %lu\n",
|
|
(unsigned long)info.blocksize);
|
|
close(fd);
|
|
return 1;
|
|
}
|
|
|
|
rxbuffer = (FAR uint8_t *)malloc((size_t)info.blocksize);
|
|
if (rxbuffer == NULL)
|
|
{
|
|
fprintf(stderr, "ERROR: failed to allocate IRX /O buffer of size %lu\n",
|
|
(unsigned long)info.blocksize);
|
|
free(txbuffer);
|
|
close(fd);
|
|
return 1;
|
|
}
|
|
|
|
/* Write and verify each sector */
|
|
|
|
pos = 0;
|
|
nerrors = 0;
|
|
value = START_VALUE;
|
|
|
|
for (blockno = 0; info.nblocks == 0 || blockno < info.nblocks; blockno++)
|
|
{
|
|
printf("Write/Verify: Block %lu\n", (unsigned long)blockno);
|
|
|
|
/* Fill buffer with a (possibly) unique pattern */
|
|
|
|
for (i = 0; i < info.blocksize; i++)
|
|
{
|
|
txbuffer[i] = value;
|
|
if (++value >= END_VALUE)
|
|
{
|
|
value = START_VALUE;
|
|
}
|
|
}
|
|
|
|
seekpos = lseek(fd, pos, SEEK_SET);
|
|
if (seekpos == (off_t)-1)
|
|
{
|
|
int errcode = errno;
|
|
|
|
fprintf(stderr, "ERROR: lseek to %lu failed: %d\n",
|
|
(unsigned long)pos, errcode);
|
|
fprintf(stderr, "ERROR: Aborting at block: %lu\n", blockno);
|
|
info.nblocks = blockno;
|
|
break;
|
|
}
|
|
else if (seekpos != pos)
|
|
{
|
|
fprintf(stderr, "ERROR: lseek failed: %lu vs %lu\n",
|
|
(unsigned)seekpos, (unsigned long) pos);
|
|
fprintf(stderr, "ERROR: Aborting at block: %lu\n", blockno);
|
|
info.nblocks = blockno;
|
|
break;
|
|
}
|
|
|
|
nwritten = write(fd, txbuffer, info.blocksize);
|
|
if (nwritten < 0)
|
|
{
|
|
int errcode = errno;
|
|
|
|
fprintf(stderr, "ERROR: write failed: %d\n", errcode);
|
|
if (errno != EINTR)
|
|
{
|
|
fprintf(stderr, "ERROR: Aborting at block: %lu\n", blockno);
|
|
info.nblocks = blockno;
|
|
break;
|
|
}
|
|
}
|
|
else if (nwritten != info.blocksize)
|
|
{
|
|
fprintf(stderr, "ERROR: Unexpected write size: %lu vs. %lu\n",
|
|
(unsigned long)nwritten, (unsigned long)info.blocksize);
|
|
fprintf(stderr, "ERROR: Aborting at block: %lu\n", blockno);
|
|
info.nblocks = blockno;
|
|
break;
|
|
}
|
|
|
|
seekpos = lseek(fd, pos, SEEK_SET);
|
|
if (seekpos == (off_t)-1)
|
|
{
|
|
int errcode = errno;
|
|
|
|
fprintf(stderr, "ERROR: lseek to %lu failed: %d\n",
|
|
(unsigned long)pos, errcode);
|
|
fprintf(stderr, "ERROR: Aborting at block: %lu\n", blockno);
|
|
info.nblocks = blockno;
|
|
break;
|
|
}
|
|
else if (seekpos != pos)
|
|
{
|
|
fprintf(stderr, "ERROR: lseek failed: %lu vs %lu\n",
|
|
(unsigned)seekpos, (unsigned long) pos);
|
|
fprintf(stderr, "ERROR: Aborting at block: %lu\n", blockno);
|
|
info.nblocks = blockno;
|
|
break;
|
|
}
|
|
|
|
nread = read(fd, rxbuffer, info.blocksize);
|
|
if (nread < 0)
|
|
{
|
|
int errcode = errno;
|
|
|
|
fprintf(stderr, "ERROR: read failed: %d\n", errcode);
|
|
if (errno != EINTR)
|
|
{
|
|
fprintf(stderr, "ERROR: Aborting at block: %lu\n", blockno);
|
|
info.nblocks = blockno;
|
|
break;
|
|
}
|
|
}
|
|
else if (nread != info.blocksize)
|
|
{
|
|
fprintf(stderr, "ERROR: Unexpected read size: %lu vs. %lu\n",
|
|
(unsigned long)nread, (unsigned long)info.blocksize);
|
|
fprintf(stderr, "ERROR: Aborting at block: %lu\n", blockno);
|
|
info.nblocks = blockno;
|
|
break;
|
|
}
|
|
else
|
|
{
|
|
for (i = 0; i < info.blocksize && nerrors <= 100; i++)
|
|
{
|
|
if (txbuffer[i] != rxbuffer[i])
|
|
{
|
|
fprintf(stderr,
|
|
"ERROR: block=%lu offset=%lu. Unexpected value: %02x vs. %02x\n",
|
|
blockno, i, rxbuffer[i], txbuffer[i]);
|
|
nerrors++;
|
|
}
|
|
}
|
|
|
|
if (nerrors > 100)
|
|
{
|
|
fprintf(stderr, "ERROR: Too many errors\n");
|
|
fprintf(stderr, "ERROR: Aborting at block: %lu\n", blockno);
|
|
info.nblocks = blockno;
|
|
break;
|
|
}
|
|
}
|
|
|
|
pos += info.blocksize;
|
|
}
|
|
|
|
/* Set the number of blocks if it was unknown before */
|
|
|
|
if (info.nblocks == 0)
|
|
{
|
|
info.nblocks = blockno;
|
|
}
|
|
|
|
/* Seek to the beginnin of the file */
|
|
|
|
seekpos = lseek(fd, 0, SEEK_SET);
|
|
if (seekpos == (off_t)-1)
|
|
{
|
|
int errcode = errno;
|
|
|
|
fprintf(stderr, "ERROR: lseek to 0 failed: %d\n", errcode);
|
|
}
|
|
else if (seekpos != 0)
|
|
{
|
|
fprintf(stderr, "ERROR: lseek to 0 failed: %lu\n",
|
|
(unsigned)seekpos);
|
|
}
|
|
|
|
/* Re-read and verify each sector */
|
|
|
|
value = START_VALUE;
|
|
|
|
for (blockno = 0; blockno < info.nblocks; blockno++)
|
|
{
|
|
printf("Re-read/Verify: Block %lu\n", (unsigned long)blockno);
|
|
|
|
/* Fill buffer with a (possibly) unique pattern */
|
|
|
|
for (i = 0; i < info.blocksize; i++)
|
|
{
|
|
txbuffer[i] = value;
|
|
if (++value >= END_VALUE)
|
|
{
|
|
value = START_VALUE;
|
|
}
|
|
}
|
|
|
|
nread = read(fd, rxbuffer, info.blocksize);
|
|
if (nread < 0)
|
|
{
|
|
int errcode = errno;
|
|
|
|
fprintf(stderr, "ERROR: read failed: %d\n", errcode);
|
|
if (errno != EINTR)
|
|
{
|
|
fprintf(stderr, "ERROR: Aborting at block: %lu\n", blockno);
|
|
break;
|
|
}
|
|
}
|
|
else if (nread != info.blocksize)
|
|
{
|
|
fprintf(stderr, "ERROR: Unexpected read size: %lu vs. %lu\n",
|
|
(unsigned long)nread, (unsigned long)info.blocksize);
|
|
fprintf(stderr, "ERROR: Aborting at block: %lu\n", blockno);
|
|
break;
|
|
}
|
|
else
|
|
{
|
|
for (i = 0; i < info.blocksize && nerrors <= 100; i++)
|
|
{
|
|
if (txbuffer[i] != rxbuffer[i])
|
|
{
|
|
fprintf(stderr,
|
|
"ERROR: block=%lu offset=%lu. Unexpected value: %02x vs. %02x\n",
|
|
blockno, i, rxbuffer[i], txbuffer[i]);
|
|
nerrors++;
|
|
}
|
|
}
|
|
|
|
if (nerrors > 100)
|
|
{
|
|
fprintf(stderr, "ERROR: Too many errors\n");
|
|
fprintf(stderr, "ERROR: Aborting at block: %lu\n", blockno);
|
|
info.nblocks = blockno;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
/* Clean-up and exit */
|
|
|
|
free(rxbuffer);
|
|
free(txbuffer);
|
|
close(fd);
|
|
return 0;
|
|
}
|