From: Sergey Filippov <sergey.filippov@outlook.com>
Date: Sat, 15 Jun 2024 17:14:35 +0000 (+0200)
Subject: linksys: add magic header generation tool for e8350 v1
X-Git-Url: http://git.openwrt.org/?a=commitdiff_plain;h=6ac44974185a3e7dc7848e97b964339948e817a7;p=project%2Ffirmware-utils.git

linksys: add magic header generation tool for e8350 v1

Add support for upgrade from vendor firmware for Linksys e8350.
As part of adding support for Linksys e8350 to OpenWrt it is required to have this tool.

Signed-off-by: Sergey Filippov <sergey.filippov@outlook.com>
[ fix formal code problem, calloc instead of malloc + memset ]
Link: https://github.com/openwrt/firmware-utils/pull/30
Signed-off-by: Christian Marangi <ansuelsmth@gmail.com>
---

diff --git a/CMakeLists.txt b/CMakeLists.txt
index f0f608a..078c49a 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -50,6 +50,7 @@ FW_UTIL(imagetag "src/imagetag_cmdline.c;src/cyg_crc32.c" "" "")
 FW_UTIL(iptime-crc32 src/cyg_crc32.c "" "")
 FW_UTIL(iptime-naspkg "" "" "")
 FW_UTIL(jcgimage "" "" "${ZLIB_LIBRARIES}")
+FW_UTIL(linksys-addfwhdr src/cyg_crc32.c "" "")
 FW_UTIL(lxlfw "" "" "")
 FW_UTIL(lzma2eva "" "" "${ZLIB_LIBRARIES}")
 FW_UTIL(makeamitbin "" "" "")
diff --git a/src/linksys-addfwhdr.c b/src/linksys-addfwhdr.c
new file mode 100644
index 0000000..56a2ae2
--- /dev/null
+++ b/src/linksys-addfwhdr.c
@@ -0,0 +1,172 @@
+/*
+ * Linksys e8350 v1 firmware header generator
+ */
+
+#include <stdbool.h>
+#include <stdio.h>
+#include <string.h>
+#include <stdlib.h>
+#include <fcntl.h>
+#include <unistd.h>
+#include <sys/stat.h>
+#include <stdarg.h>
+#include <getopt.h>
+#include <time.h>
+#include <sys/time.h>
+#include "cyg_crc.h"
+
+#define AC2350          		20
+#define CYBERTAN_VERSION		"v1.0.03"
+#define SERIAL_NUMBER   		"003"
+#define MINOR_VERSION   		""
+#define BUILD_KEYWORD   		" B"
+#define BUILD_NUMBER    		SERIAL_NUMBER
+#define BETA_VERSION    		" "
+#define CYBERTAN_UBOOT_VERSION		"v1.0"
+
+/* add for AC2350 F/W header */
+#define FWHDR_MAGIC_STR			"CHDR"
+#define FWHDR_MAGIC			0X52444843
+
+struct cbt_fw_header {
+	unsigned int magic;             /* "CHDR" */
+	unsigned int len;               /* Length of file including header */
+	unsigned int crc32;             /* 32-bit CRC */
+	unsigned int res;
+};
+
+#define MAX_BUF				1024
+/* Initial CRC32 checksum value */
+#define CRC32_INIT_VALUE		0xffffffff
+
+int fd, fd_w;
+
+void die(const char * str, ...)
+{
+	va_list args;
+	va_start(args, str);
+	vfprintf(stderr, str, args);
+	fputc('\n', stderr);
+	exit(1);
+}
+
+int fill_null0(int size)
+{
+	unsigned char buf[1];
+	int i;
+
+	fprintf(stderr,"Fill null\n");
+
+	buf[0] = 0xff;
+	for (i = 0 ; i < size; i++)
+		if (write(fd_w, buf, 1) != 1)
+			return 0;
+
+	return 1;
+}
+
+long file_open(const char *name)
+{
+	struct stat sb;
+	if ((fd = open(name, O_RDONLY, 0)) < 0) 
+		die("Unable to open `%s' : %m", name);
+
+	if (fstat (fd, &sb))
+		die("Unable to stat `%s' : %m", name);
+
+	return sb.st_size;
+}
+
+void usage(void)
+{
+	die("Usage: addfwhdr [-i|--input] sysupgrade.o [-o|--output] code.bin\n");
+}
+
+int main(int argc, char ** argv)
+{
+	char *input_file = NULL, *output_file = NULL;
+	extern int optind, opterr, optopt;
+	unsigned int input_size,c;
+	int option_index = 0;
+	extern char *optarg;
+	char *buf = NULL;
+	int garbage = 0;
+	int opt;
+	
+	struct cbt_fw_header *fwhdr;
+	unsigned int crc;	
+
+	static struct option long_options[] = {
+		{"input", 1, 0, 'i'},
+		{"output", 1, 0, 'o'},
+		{"garbage", 0, 0, 'g'},
+		{0, 0, 0, 0}
+	};
+
+	while(true) {
+		opt = getopt_long(argc, argv, "i:o:g",long_options, &option_index);
+		if (opt == -1)
+			break;
+		switch(opt){
+			case 'h' : 
+				usage(); 
+				break;
+			case 'i' :
+				input_file = optarg;
+				printf("input file is [%s]\n", input_file); 
+				break;
+			case 'o' :
+				output_file = optarg;
+				printf("output file is [%s]\n", output_file); 
+				break;
+			case 'g' :
+				garbage = 1; 
+				break;
+			default :
+				usage();
+		}
+	}
+
+	if (!input_file || !output_file) {
+		printf("You must specify the input and output file!\n");
+		usage();
+	}
+	
+	unlink(output_file);
+	if ((fd_w = open(output_file, O_RDWR|O_CREAT, S_IREAD | S_IWRITE)) < 0)
+		die("Unable to open `%s' : %m", output_file);
+
+	printf("\n---------- add fw header --------\n");
+	
+	fwhdr = calloc(1, sizeof(struct cbt_fw_header));
+	memcpy((char *)&fwhdr->magic, FWHDR_MAGIC_STR, sizeof(fwhdr->magic));
+	
+	input_size = file_open(input_file);
+	if (!(buf = malloc(input_size))){
+		perror("malloc");
+		goto fail;
+	}
+	c = read(fd, buf, input_size);
+	fwhdr->len = input_size + sizeof(struct cbt_fw_header);
+	fwhdr->res = fwhdr->res | 0x1;
+
+	crc = cyg_crc32_accumulate(CRC32_INIT_VALUE, (uint8_t *)&fwhdr->res, 4);
+	crc = cyg_crc32_accumulate(crc, (uint8_t *)&buf[0], input_size);
+	
+	fwhdr->crc32 = crc;
+
+	/* write code pattern header */
+	write(fd_w, fwhdr, sizeof(struct cbt_fw_header));
+
+	if (write(fd_w, buf, c) != c)
+		die("Write call failed!\n");
+	
+fail:
+	free(fwhdr);
+	if (buf)
+		free(buf);
+	close(fd);
+	close(fd_w);
+	
+	return 0;
+}