blob: 625e13f90c75c91afeba0eed29051a2ab920290e [file] [log] [blame]
Heiko Schocher566a4942007-06-22 19:11:54 +02001/*
2 * (C) Copyright 2007
3 * Heiko Schocher, DENX Software Engineering, <hs@denx.de>
4 *
5 * See file CREDITS for list of people who contributed to this
6 * project.
7 *
8 * This program is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU General Public License as
10 * published by the Free Software Foundation; either version 2 of
11 * the License, or (at your option) any later version.
12 *
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
17 *
18 * You should have received a copy of the GNU General Public License
19 * along with this program; if not, write to the Free Software
20 * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
21 * MA 02111-1307 USA
22 */
23
Peter Tyser2f8d3962009-03-13 18:54:51 -050024#include "os_support.h"
Heiko Schocher566a4942007-06-22 19:11:54 +020025#include <stdio.h>
26#include <stdlib.h>
27#include <unistd.h>
28#include <fcntl.h>
29#include <errno.h>
30#include <string.h>
Heiko Schocher566a4942007-06-22 19:11:54 +020031#include <sys/stat.h>
32#include "sha1.h"
33
Heiko Schocher566a4942007-06-22 19:11:54 +020034int main (int argc, char **argv)
35{
36 unsigned char output[20];
37 int i, len;
38
39 char *imagefile;
40 char *cmdname = *argv;
41 unsigned char *ptr;
42 unsigned char *data;
43 struct stat sbuf;
44 unsigned char *ptroff;
45 int ifd;
46 int off;
47
48 if (argc > 1) {
49 imagefile = argv[1];
50 ifd = open (imagefile, O_RDWR|O_BINARY);
51 if (ifd < 0) {
52 fprintf (stderr, "%s: Can't open %s: %s\n",
53 cmdname, imagefile, strerror(errno));
54 exit (EXIT_FAILURE);
55 }
56 if (fstat (ifd, &sbuf) < 0) {
57 fprintf (stderr, "%s: Can't stat %s: %s\n",
58 cmdname, imagefile, strerror(errno));
59 exit (EXIT_FAILURE);
60 }
61 len = sbuf.st_size;
62 ptr = (unsigned char *)mmap(0, len,
63 PROT_READ, MAP_SHARED, ifd, 0);
64 if (ptr == (unsigned char *)MAP_FAILED) {
65 fprintf (stderr, "%s: Can't read %s: %s\n",
66 cmdname, imagefile, strerror(errno));
67 exit (EXIT_FAILURE);
68 }
Wolfgang Denk4ef218f2007-07-10 00:01:28 +020069
Heiko Schocher566a4942007-06-22 19:11:54 +020070 /* create a copy, so we can blank out the sha1 sum */
71 data = malloc (len);
72 memcpy (data, ptr, len);
73 off = SHA1_SUM_POS;
74 ptroff = &data[len + off];
75 for (i = 0; i < SHA1_SUM_LEN; i++) {
76 ptroff[i] = 0;
77 }
Wolfgang Denk4ef218f2007-07-10 00:01:28 +020078
Heiko Schocher566a4942007-06-22 19:11:54 +020079 sha1_csum ((unsigned char *) data, len, (unsigned char *)output);
80
81 printf ("U-Boot sum:\n");
Wolfgang Denk4ef218f2007-07-10 00:01:28 +020082 for (i = 0; i < 20 ; i++) {
Heiko Schocher566a4942007-06-22 19:11:54 +020083 printf ("%02X ", output[i]);
84 }
85 printf ("\n");
86 /* overwrite the sum in the bin file, with the actual */
87 lseek (ifd, SHA1_SUM_POS, SEEK_END);
88 if (write (ifd, output, SHA1_SUM_LEN) != SHA1_SUM_LEN) {
89 fprintf (stderr, "%s: Can't write %s: %s\n",
90 cmdname, imagefile, strerror(errno));
91 exit (EXIT_FAILURE);
92 }
Wolfgang Denk4ef218f2007-07-10 00:01:28 +020093
Heiko Schocher566a4942007-06-22 19:11:54 +020094 free (data);
95 (void) munmap((void *)ptr, len);
96 (void) close (ifd);
97 }
98
99 return EXIT_SUCCESS;
100}