]>
Commit | Line | Data |
---|---|---|
566a494f HS |
1 | /* |
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 | ||
2f8d396b | 24 | #include "os_support.h" |
566a494f HS |
25 | #include <stdio.h> |
26 | #include <stdlib.h> | |
27 | #include <unistd.h> | |
28 | #include <fcntl.h> | |
29 | #include <errno.h> | |
30 | #include <string.h> | |
566a494f HS |
31 | #include <sys/stat.h> |
32 | #include "sha1.h" | |
33 | ||
566a494f HS |
34 | int 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 | } | |
4ef218f6 | 69 | |
566a494f HS |
70 | /* 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 | } | |
4ef218f6 | 78 | |
566a494f HS |
79 | sha1_csum ((unsigned char *) data, len, (unsigned char *)output); |
80 | ||
81 | printf ("U-Boot sum:\n"); | |
4ef218f6 | 82 | for (i = 0; i < 20 ; i++) { |
566a494f HS |
83 | 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 | } | |
4ef218f6 | 93 | |
566a494f HS |
94 | free (data); |
95 | (void) munmap((void *)ptr, len); | |
96 | (void) close (ifd); | |
97 | } | |
98 | ||
99 | return EXIT_SUCCESS; | |
100 | } |