X-Git-Url: https://git.rohieb.name/openwrt.git/blobdiff_plain/0c0407178e0f768de17a6c71dbb3057cdd270490..8838bdcc69e05bebc7fbbf019cdfdd405b8337f2:/tools/firmware-utils/src/trx.c

diff --git a/tools/firmware-utils/src/trx.c b/tools/firmware-utils/src/trx.c
index 1131238a4..44bd06a06 100644
--- a/tools/firmware-utils/src/trx.c
+++ b/tools/firmware-utils/src/trx.c
@@ -35,6 +35,12 @@
  * February 19, 2005 - mbm
  *
  * Add -a (align offset) and -b (absolute offset)
+ *
+ * March 24, 2010 - markus
+ *
+ * extend trx header struct for new version
+ * assume v1 for as default
+ * Add option -2 to allow v2 header
  */
 
 #include <stdio.h>
@@ -59,8 +65,7 @@ uint32_t crc32buf(char *buf, size_t len);
 /* from trxhdr.h */
 
 #define TRX_MAGIC	0x30524448	/* "HDR0" */
-#define TRX_VERSION	1
-#define TRX_MAX_LEN	0x5A0000
+#define TRX_MAX_LEN	0x720000
 #define TRX_NO_HEADER	1		/* Do not write TRX header */	
 
 struct trx_header {
@@ -68,7 +73,7 @@ struct trx_header {
 	uint32_t len;			/* Length of file including header */
 	uint32_t crc32;			/* 32-bit CRC from flag_version to end of file */
 	uint32_t flag_version;	/* 0:15 flags, 16:31 version */
-	uint32_t offsets[3];	/* Offsets of partitions from start of header */
+	uint32_t offsets[4];	/* Offsets of partitions from start of header */
 };
 
 /**********************************************************************/
@@ -77,7 +82,9 @@ void usage(void) __attribute__ (( __noreturn__ ));
 
 void usage(void)
 {
-	fprintf(stderr, "Usage: trx [-o outfile] [-m maxlen] [-a align] [-b offset] [-f file] [-f file [-f file]]\n");
+	fprintf(stderr, "Usage:\n");
+	fprintf(stderr, " trx [-2] [-o outfile] [-m maxlen] [-a align] [-b absolute offset] [-x relative offset]\n");
+	fprintf(stderr, "     [-f file] [-f file [-f file [-f file (v2 only)]]]\n");
 	exit(EXIT_FAILURE);
 }
 
@@ -88,11 +95,13 @@ int main(int argc, char **argv)
 	char *ofn = NULL;
 	char *buf;
 	char *e;
-	int c, i, append;
+	int c, i, append = 0;
 	size_t n;
+	ssize_t n2;
 	uint32_t cur_len;
 	unsigned long maxlen = TRX_MAX_LEN;
 	struct trx_header *p;
+	char trx_version = 1;
 
 	fprintf(stderr, "mjn3's trx replacement - v0.81.1\n");
 
@@ -104,14 +113,24 @@ int main(int argc, char **argv)
 	p = (struct trx_header *) buf;
 
 	p->magic = STORE32_LE(TRX_MAGIC);
-	cur_len = sizeof(struct trx_header);
-	p->flag_version = STORE32_LE((TRX_VERSION << 16));
+	cur_len = sizeof(struct trx_header) - 4; /* assume v1 header */
+	p->flag_version = STORE32_LE((trx_version << 16));
 
 	in = NULL;
 	i = 0;
 
-	while ((c = getopt(argc, argv, "-:o:m:a:b:f:A:")) != -1) {
+	while ((c = getopt(argc, argv, "-:2o:m:a:x:b:f:A:")) != -1) {
 		switch (c) {
+			case '2':
+				/* take care that nothing was written to buf so far */
+				if (cur_len != sizeof(struct trx_header) - 4) {
+					fprintf(stderr, "-2 has to be used before any other argument!\n");
+				}
+				else {
+					trx_version = 2;
+					cur_len += 4;
+				}
+				break;
 			case 'A':
 				append = 1;
 				/* fall through */
@@ -172,6 +191,7 @@ int main(int argc, char **argv)
 					fprintf(stderr, "realloc failed");
 					return EXIT_FAILURE;
 				}
+				p = (struct trx_header *) buf;
 				break;
 			case 'a':
 				errno = 0;
@@ -199,6 +219,25 @@ int main(int argc, char **argv)
 					memset(buf + cur_len, 0, n - cur_len);
 					cur_len = n;
 				}
+				break;
+			case 'x':
+				errno = 0;
+				n2 = strtol(optarg, &e, 0);
+				if (errno || (e == optarg) || *e) {
+					fprintf(stderr, "illegal numeric string\n");
+					usage();
+				}
+				if (n2 < 0) {
+					if (-n2 > cur_len) {
+						fprintf(stderr, "WARNING: current length smaller then -x %d offset\n",n2);
+						cur_len = 0;
+					} else
+						cur_len += n2;
+				} else {
+					memset(buf + cur_len, 0, n2);
+					cur_len += n2;
+				}
+
 				break;
 			default:
 				usage();
@@ -230,7 +269,7 @@ int main(int argc, char **argv)
 	}
 
 	fclose(out);
-
+	
 	return EXIT_SUCCESS;
 }