Signed-off-by: Marco Stornelli <marco.storne...@gmail.com>
---
diff --git a/tools/imls/Makefile b/tools/imls/Makefile
new file mode 100644
index 0000000..d654658
--- /dev/null
+++ b/tools/imls/Makefile
@@ -0,0 +1,106 @@
+#
+# (C) Copyright 2009 Marco Stornelli <marco.storne...@gmail.com>
+#
+# This program 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.
+#
+# This program 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 this program; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston,
+# MA 02111-1307 USA
+#
+
+include $(TOPDIR)/config.mk
+
+HOST_CFLAGS = -Wall -pedantic
+
+# Generated executable files
+BIN_FILES-y += imls
+
+# Source files which exist outside the tools directory
+EXT_OBJ_FILES-y += lib_generic/crc32.o
+EXT_OBJ_FILES-y += lib_generic/md5.o
+EXT_OBJ_FILES-y += lib_generic/sha1.o
+EXT_OBJ_FILES-y += common/image.o
+
+# Source files located in the tools directory
+OBJ_FILES-y += imls.o
+
+# Flattened device tree objects
+LIBFDT_OBJ_FILES-y += fdt.o
+LIBFDT_OBJ_FILES-y += fdt_ro.o
+LIBFDT_OBJ_FILES-y += fdt_rw.o
+LIBFDT_OBJ_FILES-y += fdt_strerror.o
+LIBFDT_OBJ_FILES-y += fdt_wip.o
+
+# now $(obj) is defined
+SRCS   += $(addprefix $(SRCTREE)/,$(EXT_OBJ_FILES-y:.o=.c))
+SRCS   += $(addprefix $(SRCTREE)/tools/,$(OBJ_FILES-y:.o=.c))
+SRCS   += $(addprefix $(SRCTREE)/libfdt/,$(LIBFDT_OBJ_FILES-y:.o=.c))
+BINS   := $(addprefix $(obj),$(sort $(BIN_FILES-y)))
+LIBFDT_OBJS    := $(addprefix $(obj),$(LIBFDT_OBJ_FILES-y))
+
+#
+# Use native tools and options
+# Define __KERNEL_STRICT_NAMES to prevent typedef overlaps
+#
+CPPFLAGS   = -idirafter $(SRCTREE)/include \
+               -idirafter $(OBJTREE)/include2 \
+               -idirafter $(OBJTREE)/include \
+               -I $(SRCTREE)/libfdt \
+               -I $(SRCTREE)/tools \
+               -DUSE_HOSTCC -D__KERNEL_STRICT_NAMES
+CFLAGS     = $(HOST_CFLAGS) $(CPPFLAGS) -O
+
+# No -pedantic switch to avoid libfdt compilation warnings
+FIT_CFLAGS = -Wall $(CPPFLAGS) -O
+
+CC        = $(CROSS_COMPILER)gcc
+STRIP     = $(CROSS_COMPILER)strip
+
+ifeq ($(MTD_VERSION),old)
+CPPFLAGS += -DMTD_OLD
+endif
+
+all:   $(BINS)
+
+$(obj)imls:    $(obj)imls.o $(obj)crc32.o $(obj)image.o $(obj)md5.o \
+               $(obj)sha1.o $(LIBFDT_OBJS)
+       $(CC) $(CFLAGS) -o $@ $^
+       $(STRIP) $@
+
+# Some files complain if compiled with -pedantic, use FIT_CFLAGS
+$(obj)image.o: $(SRCTREE)/common/image.c
+       $(CC) -g $(FIT_CFLAGS) -c -o $@ $<
+
+$(obj)imls.o: imls.c
+       $(CC) -g $(FIT_CFLAGS) -c -o $@ $<
+
+# Some of the tool objects need to be accessed from outside the tools
directory
+$(obj)%.o: $(SRCTREE)/common/%.c
+       $(CC) -g $(FIT_CFLAGS) -c -o $@ $<
+
+$(obj)%.o: $(SRCTREE)/lib_generic/%.c
+       $(CC) -g $(CFLAGS) -c -o $@ $<
+
+$(obj)%.o: $(SRCTREE)/libfdt/%.c
+       $(CC) -g $(FIT_CFLAGS) -c -o $@ $<
+
+clean:
+       rm -rf *.o imls
+
+#########################################################################
+
+# defines $(obj).depend target
+include $(SRCTREE)/rules.mk
+
+sinclude $(obj).depend
+
+#########################################################################
diff --git a/tools/imls/README b/tools/imls/README
new file mode 100644
index 0000000..63d639d
--- /dev/null
+++ b/tools/imls/README
@@ -0,0 +1,57 @@
+#
+# (C) Copyright 2009 Marco Stornelli <marco.storne...@gmail.com>
+#
+# See file CREDITS for list of people who contributed to this
+# project.
+#
+# The files in this directory are free software; you can redistribute
+# them and/or modify them 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.
+#
+# These files are distributed in the hope that they 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 this program; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston,
+# MA 02111-1307 USA
+#
+
+IMLS
+-------------
+
+imls is an implementation of a Linux command line tool to access
+to raw flash partitions and list images made with mkimage command.
+
+For building against older versions of the MTD headers (meaning before
+v2.6.8-rc1) it is required to pass the argument "MTD_VERSION=old" to
+make.
+
+Usage examples
+--------------
+
+1) Flash with sectors of 128KiB and 32 sectors:
+
+> imls -s 32 -c 131072 /dev/mtd0
+Searching...
+Image Name:   foo
+Created:      Fri Apr 10 18:11:30 2009
+Image Type:   Intel x86 Linux Standalone Program (uncompressed)
+Data Size:    10716 Bytes = 10.46 kB = 0.01 MB
+Load Address: 00000000
+Entry Point:  00000000
+
+2) Flash with sectors of 64KiB and 128 sectors and with a search offset
of one
+sector:
+
+> imls -o 1 -s 128 -c 65536 /dev/mtd0
+Searching...
+Image Name:   foo
+Created:      Fri Apr 10 18:11:30 2009
+Image Type:   Intel x86 Linux Standalone Program (uncompressed)
+Data Size:    10716 Bytes = 10.46 kB = 0.01 MB
+Load Address: 00000000
+Entry Point:  00000000
diff --git a/tools/imls/imls.c b/tools/imls/imls.c
new file mode 100644
index 0000000..583272a
--- /dev/null
+++ b/tools/imls/imls.c
@@ -0,0 +1,270 @@
+/*
+ * (C) Copyright 2009 Marco Stornelli <marco.storne...@gmail.com>
+ *
+ * This program 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.
+ *
+ * This program 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 this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
+ * MA 02111-1307 USA
+ */
+
+#include <errno.h>
+#include <fcntl.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <stddef.h>
+#include <string.h>
+#include <sys/types.h>
+#include <sys/ioctl.h>
+#include <sys/stat.h>
+#include <unistd.h>
+#include <asm/page.h>
+
+#ifdef MTD_OLD
+#include <stdint.h>
+#include <linux/mtd/mtd.h>
+#else
+#define  __user        /* nothing */
+#include <mtd/mtd-user.h>
+#endif
+
+#include <sha1.h>
+#include <fdt.h>
+#include <libfdt.h>
+#include <fdt_support.h>
+#include <image.h>
+
+#define MIN(a, b) (((a) < (b)) ? (a) : (b))
+
+extern unsigned long crc32(unsigned long crc, const char *buf, unsigned
int len);
+static void usage(void);
+static int image_verify_header(char *ptr, int fd);
+static int flash_bad_block(int fd, uint8_t mtd_type, loff_t start);
+
+char   *cmdname;
+char   *devicefile;
+
+unsigned int sectorcount = 0;
+int sflag = 0;
+unsigned int sectoroffset = 0;
+unsigned int sectorsize = 0;
+int cflag = 0;
+
+int main (int argc, char **argv)
+{
+       int fd = -1, err = 0, readbyte = 0, j;
+       struct mtd_info_user mtdinfo;
+       char buf[sizeof(image_header_t)];
+       int found = 0;
+
+       cmdname = *argv;
+
+       while (--argc > 0 && **++argv == '-') {
+               while (*++*argv) {
+                       switch (**argv) {
+                       case 's':
+                               if (--argc <= 0)
+                                       usage ();
+                               sectorcount = (unsigned int)atoi(*++argv);
+                               sflag = 1;
+                               goto NXTARG;
+                       case 'o':
+                               if (--argc <= 0)
+                                       usage ();
+                               sectoroffset = (unsigned int)atoi(*++argv);
+                               goto NXTARG;
+
+                       case 'c':
+                               if (--argc <= 0)
+                                       usage ();
+                               sectorsize = (unsigned int)atoi(*++argv);
+                               cflag = 1;
+                               goto NXTARG;
+                       default:
+                               usage ();
+                       }
+               }
+NXTARG:                ;
+       }
+
+       if (argc != 1 || cflag == 0 || sflag == 0)
+               usage();
+
+       devicefile = *argv;
+
+       fd = open(devicefile, O_RDONLY);
+       if (fd < 0) {
+               fprintf (stderr, "%s: Can't open %s: %s\n",
+                        cmdname, devicefile, strerror(errno));
+               exit(EXIT_FAILURE);
+       }
+
+       err = ioctl(fd, MEMGETINFO, &mtdinfo);
+       if (err < 0) {
+               fprintf(stderr, "%s: Cannot get MTD information: %s\n",cmdname,
+                       strerror(errno));
+               exit(EXIT_FAILURE);
+       }
+
+       if (mtdinfo.type != MTD_NORFLASH && mtdinfo.type != MTD_NANDFLASH) {
+               fprintf(stderr, "%s: Unsupported flash type %u\n",
+                       cmdname, mtdinfo.type);
+               exit(EXIT_FAILURE);
+       }
+
+       if (sectorsize * sectorcount != mtdinfo.size) {
+               fprintf(stderr, "%s: Partition size (%d) incompatible with "
+                       "sector size and count\n", cmdname, mtdinfo.size);
+               exit(EXIT_FAILURE);
+       }
+
+       if (sectorsize * sectoroffset >= mtdinfo.size) {
+               fprintf(stderr, "%s: Partition size (%d) incompatible with "
+                       "sector offset given\n", cmdname, mtdinfo.size);
+               exit(EXIT_FAILURE);
+       }
+
+       if (sectoroffset > sectorcount - 1) {
+               fprintf(stderr, "%s: Sector offset cannot be grater than "
+                       "sector count minus one\n", cmdname);
+               exit(EXIT_FAILURE);
+       }
+
+       printf("Searching....\n");
+
+       for (j = sectoroffset; j < sectorcount; ++j) {
+
+               if (lseek(fd, j*sectorsize, SEEK_SET) != j*sectorsize) {
+                       fprintf(stderr, "%s: lseek failure: %s\n",
+                       cmdname, strerror(errno));
+                       exit(EXIT_FAILURE);
+               }
+
+               err = flash_bad_block(fd, mtdinfo.type, j*sectorsize);
+               if (err < 0)
+                       exit(EXIT_FAILURE);
+               if (err)
+                       continue; /* Skip and jump to next */
+
+               readbyte = read(fd, buf, sizeof(image_header_t));
+               if (readbyte != sizeof(image_header_t)) {
+                       fprintf(stderr, "%s: Can't read from device: %s\n",
+                       cmdname, strerror(errno));
+                       exit(EXIT_FAILURE);
+               }
+
+               if (fdt_check_header(buf)) {
+                       /* old-style image */
+                       if (image_verify_header(buf, fd)) {
+                               found = 1;
+                               image_print_contents((image_header_t *)buf);
+                       }
+               } else {
+                       /* FIT image */
+                       fit_print_contents(buf);
+               }
+
+       }       
+
+       close(fd);
+
+       if(!found)
+               printf("No images found\n");
+
+       exit(EXIT_SUCCESS);
+}
+
+void usage()
+{
+       fprintf (stderr, "Usage:\n"
+                        "       %s [-o offset] -c size -s count device\n"
+                        "          -o ==> number of sectors to use as offset\n"
+                        "          -s ==> number of sectors\n"
+                        "          -c ==> size of sectors (byte)\n",
+               cmdname);
+
+       exit(EXIT_FAILURE);
+}
+
+static int image_verify_header(char *ptr, int fd)
+{
+       int len, nread;
+       char *data;
+       uint32_t checksum;
+       image_header_t *hdr = (image_header_t *)ptr;
+       char buf[PAGE_SIZE];
+
+       if (image_get_magic(hdr) != IH_MAGIC)
+               return 0;
+
+       data = (char *)hdr;
+       len  = image_get_header_size();
+
+       checksum = image_get_hcrc(hdr);
+       hdr->ih_hcrc = htonl(0);        /* clear for re-calculation */
+
+       if (crc32(0, data, len) != checksum) {
+               fprintf(stderr,
+                     "%s: Maybe image found but it has bad header checksum!\n",
+                     cmdname);
+               return 0;
+       }
+
+       len = image_get_size(hdr);
+       checksum = 0;
+
+       while (len > 0) {
+               nread = read(fd, buf, MIN(len,PAGE_SIZE));
+               if (nread != MIN(len,PAGE_SIZE)) {
+                       fprintf(stderr,
+                               "%s: Error while reading: %s\n",
+                               cmdname, strerror(errno));
+                       exit(EXIT_FAILURE);
+               }
+               checksum = crc32(checksum, buf, nread);
+               len -= nread;
+       }
+
+       if (checksum != image_get_dcrc(hdr)) {
+               fprintf (stderr,
+                       "%s: Maybe image found but it has corrupted data!\n",
+                       cmdname);
+               return 0;
+       }
+       
+       return 1;
+}
+
+/*
+ * Test for bad block on NAND, just returns 0 on NOR, on NAND:
+ * 0   - block is good
+ * > 0 - block is bad
+ * < 0 - failed to test
+ */
+static int flash_bad_block(int fd, uint8_t mtd_type, loff_t start)
+{
+       if (mtd_type == MTD_NANDFLASH) {
+               int badblock = ioctl(fd, MEMGETBADBLOCK, &start);
+
+               if (badblock < 0) {
+                       fprintf(stderr,"%s: Cannot read bad block mark: %s\n",
+                               cmdname, strerror(errno));
+                       return badblock;
+               }
+
+               if (badblock) {
+                       return badblock;
+               }
+       }
+
+       return 0;
+}
_______________________________________________
U-Boot mailing list
U-Boot@lists.denx.de
http://lists.denx.de/mailman/listinfo/u-boot

Reply via email to