crcit.c 2.25 KB
Newer Older
Wolfgang Denk's avatar
Wolfgang Denk committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
/*
 * (C) Copyright 2005
 * 2N Telekomunikace, Ladislav Michl <michl@2n.cz>
 *
 * See file CREDITS for list of people who contributed to this
 * project.
 *
 * 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 <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <fcntl.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include "crcek.h"

Ladislav Michl's avatar
Ladislav Michl committed
34
extern uint32_t crc32(uint32_t, const unsigned char *, uint);
Wolfgang Denk's avatar
Wolfgang Denk committed
35

Ladislav Michl's avatar
Ladislav Michl committed
36
static uint32_t data[LOADER_SIZE/4 + 3];
Wolfgang Denk's avatar
Wolfgang Denk committed
37

Ladislav Michl's avatar
Ladislav Michl committed
38
static int do_crc(char *path, unsigned version)
Wolfgang Denk's avatar
Wolfgang Denk committed
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
{
	uint32_t *p;
	ssize_t size;
	int fd;

	fd = open(path, O_RDONLY);
	if (fd == -1) {
		perror("Error opening file");
		return EXIT_FAILURE;
	}
	p = data + 2;
	size = read(fd, p, LOADER_SIZE + 4);
	if (size == -1) {
		perror("Error reading file");
		return EXIT_FAILURE;
	}
	if (size > LOADER_SIZE) {
		fprintf(stderr, "File too large\n");
		return EXIT_FAILURE;
	}
Ladislav Michl's avatar
Ladislav Michl committed
59
	size = (size + 3) & ~3;	/* round up to 4 bytes */
Wolfgang Denk's avatar
Wolfgang Denk committed
60
61
	data[0] = size + 4;	/* add size of version field */
	data[1] = version;
Ladislav Michl's avatar
Ladislav Michl committed
62
	data[2 + (size >> 2)] = crc32(0, (unsigned char *)(data + 1), data[0]);
Wolfgang Denk's avatar
Wolfgang Denk committed
63
64
65
66
67
68
69
70
71
72
73
74
75
	close(fd);

	if (write(STDOUT_FILENO, data, size + 3*4) == -1) {
		perror("Error writing file");
		return EXIT_FAILURE;
	}

	return EXIT_SUCCESS;
}

int main(int argc, char **argv)
{
	if (argc == 2) {
Ladislav Michl's avatar
Ladislav Michl committed
76
		return do_crc(argv[1], 0);
Wolfgang Denk's avatar
Wolfgang Denk committed
77
78
79
	} else if ((argc == 4) && (strcmp(argv[1], "-v") == 0)) {
		char *endptr, *nptr = argv[2];
		unsigned ver = strtoul(nptr, &endptr, 0);
Wolfgang Denk's avatar
Wolfgang Denk committed
80
		if (*nptr != '\0' && *endptr == '\0')
Ladislav Michl's avatar
Ladislav Michl committed
81
			return do_crc(argv[3], ver);
Wolfgang Denk's avatar
Wolfgang Denk committed
82
83
84
85
86
	}
	fprintf(stderr, "Usage: crcit [-v version] <image>\n");

	return EXIT_FAILURE;
}