blob: 9774eea32e59ff94c21336e825dab88bdbe19a60 [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
34#ifndef __ASSEMBLY__
35#define __ASSEMBLY__ /* Dirty trick to get only #defines */
36#endif
37#include <config.h>
38#undef __ASSEMBLY__
39
Heiko Schocher566a4942007-06-22 19:11:54 +020040extern void sha1_csum (unsigned char *input, int ilen, unsigned char output[20]);
41
42int main (int argc, char **argv)
43{
44 unsigned char output[20];
45 int i, len;
46
47 char *imagefile;
48 char *cmdname = *argv;
49 unsigned char *ptr;
50 unsigned char *data;
51 struct stat sbuf;
52 unsigned char *ptroff;
53 int ifd;
54 int off;
55
56 if (argc > 1) {
57 imagefile = argv[1];
58 ifd = open (imagefile, O_RDWR|O_BINARY);
59 if (ifd < 0) {
60 fprintf (stderr, "%s: Can't open %s: %s\n",
61 cmdname, imagefile, strerror(errno));
62 exit (EXIT_FAILURE);
63 }
64 if (fstat (ifd, &sbuf) < 0) {
65 fprintf (stderr, "%s: Can't stat %s: %s\n",
66 cmdname, imagefile, strerror(errno));
67 exit (EXIT_FAILURE);
68 }
69 len = sbuf.st_size;
70 ptr = (unsigned char *)mmap(0, len,
71 PROT_READ, MAP_SHARED, ifd, 0);
72 if (ptr == (unsigned char *)MAP_FAILED) {
73 fprintf (stderr, "%s: Can't read %s: %s\n",
74 cmdname, imagefile, strerror(errno));
75 exit (EXIT_FAILURE);
76 }
Wolfgang Denk4ef218f2007-07-10 00:01:28 +020077
Heiko Schocher566a4942007-06-22 19:11:54 +020078 /* create a copy, so we can blank out the sha1 sum */
79 data = malloc (len);
80 memcpy (data, ptr, len);
81 off = SHA1_SUM_POS;
82 ptroff = &data[len + off];
83 for (i = 0; i < SHA1_SUM_LEN; i++) {
84 ptroff[i] = 0;
85 }
Wolfgang Denk4ef218f2007-07-10 00:01:28 +020086
Heiko Schocher566a4942007-06-22 19:11:54 +020087 sha1_csum ((unsigned char *) data, len, (unsigned char *)output);
88
89 printf ("U-Boot sum:\n");
Wolfgang Denk4ef218f2007-07-10 00:01:28 +020090 for (i = 0; i < 20 ; i++) {
Heiko Schocher566a4942007-06-22 19:11:54 +020091 printf ("%02X ", output[i]);
92 }
93 printf ("\n");
94 /* overwrite the sum in the bin file, with the actual */
95 lseek (ifd, SHA1_SUM_POS, SEEK_END);
96 if (write (ifd, output, SHA1_SUM_LEN) != SHA1_SUM_LEN) {
97 fprintf (stderr, "%s: Can't write %s: %s\n",
98 cmdname, imagefile, strerror(errno));
99 exit (EXIT_FAILURE);
100 }
Wolfgang Denk4ef218f2007-07-10 00:01:28 +0200101
Heiko Schocher566a4942007-06-22 19:11:54 +0200102 free (data);
103 (void) munmap((void *)ptr, len);
104 (void) close (ifd);
105 }
106
107 return EXIT_SUCCESS;
108}