summaryrefslogtreecommitdiffstats
path: root/scripts/s5p_cksum.c
diff options
context:
space:
mode:
authorAlexey Galakhov <agalakhov@gmail.com>2012-05-18 15:43:28 +0600
committerSascha Hauer <s.hauer@pengutronix.de>2012-05-21 21:58:26 +0200
commitd1e782ed959b8d936102b1540f5f7513a0a84590 (patch)
tree0da0cb6b98f5e8f2c4bf62733628188c6d21e2e6 /scripts/s5p_cksum.c
parent77433a8a9adbdcb5c9012a82c6aa2741da635b1c (diff)
downloadbarebox-d1e782ed959b8d936102b1540f5f7513a0a84590.tar.gz
barebox-d1e782ed959b8d936102b1540f5f7513a0a84590.tar.xz
S5P boot header and image generator
Signed-off-by: Alexey Galakhov <agalakhov@gmail.com> Signed-off-by: Sascha Hauer <s.hauer@pengutronix.de>
Diffstat (limited to 'scripts/s5p_cksum.c')
-rw-r--r--scripts/s5p_cksum.c140
1 files changed, 140 insertions, 0 deletions
diff --git a/scripts/s5p_cksum.c b/scripts/s5p_cksum.c
new file mode 100644
index 0000000000..7142532875
--- /dev/null
+++ b/scripts/s5p_cksum.c
@@ -0,0 +1,140 @@
+/*
+ * Copyright (C) 2012 Alexey Galakhov
+ *
+ * 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
+ * version 2 as published by the Free Software Foundation.
+ *
+ * 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 <stdint.h>
+#include <stdlib.h>
+#include <string.h>
+#include <stddef.h>
+#include <errno.h>
+
+#define DEFAULT_BUF_SIZE 8192
+
+static int usage(const char* me)
+{
+ printf("Usage: %s <input> <output> [<bufsize]\n", me);
+ return 1;
+}
+
+static void put32(uint8_t *ptr, uint32_t value)
+{
+ ptr[0] = value & 0xFF;
+ ptr[1] = (value >> 8) & 0xFF;
+ ptr[2] = (value >> 16) & 0xFF;
+ ptr[3] = (value >> 24) & 0xFF;
+}
+
+static size_t safe_fread(void *buf, size_t len, FILE* file)
+{
+ size_t rd = fread(buf, 1, len, file);
+ if (! rd) {
+ if (ferror(file))
+ fprintf(stderr, "Error reading file: %s\n", strerror(errno));
+ else
+ fprintf(stderr, "Unexpected end of file\n");
+ }
+ return rd;
+}
+
+static size_t safe_fwrite(const void *buf, size_t len, FILE* file)
+{
+ size_t wr = fwrite(buf, 1, len, file);
+ if (wr != len) {
+ fprintf(stderr, "Error writing file: %s\n", strerror(errno));
+ return 0;
+ }
+ return wr;
+}
+
+static int process(FILE *input, FILE *output, uint8_t *buf, unsigned bufsize)
+{
+ size_t rd;
+ unsigned i;
+ uint32_t cksum;
+ /* Read first chunk */
+ rd = safe_fread(buf + 16, bufsize - 16, input);
+ if (! rd)
+ return 4;
+ /* Calculate header */
+ put32(buf + 0, bufsize);
+ cksum = 0;
+ for (i = 16; i < bufsize; ++i)
+ cksum += (uint32_t)buf[i];
+ put32(buf + 8, cksum);
+ if (! safe_fwrite(buf, bufsize, output))
+ return 4;
+ /* Copy the rest of file */
+ while (! feof(input)) {
+ rd = safe_fread(buf, bufsize, input);
+ if (! rd)
+ return 4;
+ if (! safe_fwrite(buf, rd, output))
+ return 4;
+ }
+ return 0;
+}
+
+static int work(const char* me, const char *infile, const char *outfile, unsigned bufsize)
+{
+ uint8_t *buf;
+ FILE *input;
+ FILE *output;
+ int ret;
+ if (bufsize < 512 || bufsize > 65536)
+ return usage(me);
+ buf = calloc(1, bufsize);
+ if (! buf) {
+ fprintf(stderr, "Unable to allocate %u bytes of memory\n", bufsize);
+ return 2;
+ }
+ input = fopen(infile, "r");
+ if (! input) {
+ fprintf(stderr, "Cannot open `%s' for reading\n", infile);
+ free(buf);
+ return 3;
+ }
+ output = fopen(outfile, "w");
+ if (! output) {
+ fprintf(stderr, "Cannot open `%s' for writing\n", outfile);
+ fclose(input);
+ free(buf);
+ return 3;
+ }
+
+ ret = process(input, output, buf, bufsize);
+
+ fclose(output);
+ fclose(input);
+ free(buf);
+ return ret;
+}
+
+int main(int argc, char** argv)
+{
+ switch (argc) {
+ case 3:
+ return work(argv[0], argv[1], argv[2], DEFAULT_BUF_SIZE);
+ case 4:
+ return work(argv[0], argv[1], argv[2], atoi(argv[3]));
+ default:
+ return usage(argv[0]);
+ }
+}