aboutsummaryrefslogtreecommitdiffstats
path: root/package/utils/otrx/src/otrx.c
diff options
context:
space:
mode:
Diffstat (limited to 'package/utils/otrx/src/otrx.c')
-rw-r--r--package/utils/otrx/src/otrx.c124
1 files changed, 69 insertions, 55 deletions
diff --git a/package/utils/otrx/src/otrx.c b/package/utils/otrx/src/otrx.c
index e42aacaf0e..25e0592737 100644
--- a/package/utils/otrx/src/otrx.c
+++ b/package/utils/otrx/src/otrx.c
@@ -14,6 +14,7 @@
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
+#include <string.h>
#include <unistd.h>
#if __BYTE_ORDER == __BIG_ENDIAN
@@ -39,14 +40,6 @@ struct trx_header {
uint32_t offset[3];
};
-enum mode {
- MODE_UNKNOWN,
- MODE_CHECK,
- MODE_EXTRACT,
-};
-
-enum mode mode = MODE_UNKNOWN;
-
char *trx_path;
size_t trx_offset = 0;
char *partition[TRX_MAX_PARTS] = {};
@@ -137,7 +130,19 @@ uint32_t otrx_crc32(uint8_t *buf, size_t len) {
* Check
**************************************************/
-static int otrx_check() {
+static void otrx_check_parse_options(int argc, char **argv) {
+ int c;
+
+ while ((c = getopt(argc, argv, "o:")) != -1) {
+ switch (c) {
+ case 'o':
+ trx_offset = atoi(optarg);
+ break;
+ }
+ }
+}
+
+static int otrx_check(int argc, char **argv) {
FILE *trx;
struct trx_header hdr;
size_t bytes, length;
@@ -145,6 +150,16 @@ static int otrx_check() {
uint32_t crc32;
int err = 0;
+ if (argc < 3) {
+ fprintf(stderr, "No TRX file passed\n");
+ err = -EINVAL;
+ goto out;
+ }
+ trx_path = argv[2];
+
+ optind = 3;
+ otrx_check_parse_options(argc, argv);
+
trx = fopen(trx_path, "r");
if (!trx) {
fprintf(stderr, "Couldn't open %s\n", trx_path);
@@ -209,6 +224,27 @@ out:
* Extract
**************************************************/
+static void otrx_extract_parse_options(int argc, char **argv) {
+ int c;
+
+ while ((c = getopt(argc, argv, "c:e:o:1:2:3:")) != -1) {
+ switch (c) {
+ case 'o':
+ trx_offset = atoi(optarg);
+ break;
+ case '1':
+ partition[0] = optarg;
+ break;
+ case '2':
+ partition[1] = optarg;
+ break;
+ case '3':
+ partition[2] = optarg;
+ break;
+ }
+ }
+}
+
static int otrx_extract_copy(FILE *trx, size_t offset, size_t length, char *out_path) {
FILE *out;
size_t bytes;
@@ -254,13 +290,23 @@ out:
return err;
}
-static int otrx_extract() {
+static int otrx_extract(int argc, char **argv) {
FILE *trx;
struct trx_header hdr;
size_t bytes;
int i;
int err = 0;
+ if (argc < 3) {
+ fprintf(stderr, "No TRX file passed\n");
+ err = -EINVAL;
+ goto out;
+ }
+ trx_path = argv[2];
+
+ optind = 3;
+ otrx_extract_parse_options(argc, argv);
+
trx = fopen(trx_path, "r");
if (!trx) {
fprintf(stderr, "Couldn't open %s\n", trx_path);
@@ -310,61 +356,29 @@ out:
* Start
**************************************************/
-static void parse_options(int argc, char **argv) {
- int c;
-
- while ((c = getopt(argc, argv, "c:e:o:1:2:3:")) != -1) {
- switch (c) {
- case 'c':
- mode = MODE_CHECK;
- trx_path = optarg;
- break;
- case 'e':
- mode = MODE_EXTRACT;
- trx_path = optarg;
- break;
- case 'o':
- trx_offset = atoi(optarg);
- break;
- case '1':
- partition[0] = optarg;
- break;
- case '2':
- partition[1] = optarg;
- break;
- case '3':
- partition[2] = optarg;
- break;
- }
- }
-}
-
static void usage() {
printf("Usage:\n");
printf("\n");
printf("Checking TRX file:\n");
- printf("\t-c file\t\tcheck if file is a valid TRX\n");
- printf("\t-o offset\toffset of TRX data in file (default: 0)\n");
+ printf("\totrx check <file> [options]\tcheck if file is a valid TRX\n");
+ printf("\t-o offset\t\t\toffset of TRX data in file (default: 0)\n");
printf("\n");
printf("Extracting from TRX file:\n");
- printf("\t-e file\t\tfile with TRX to extract from\n");
- printf("\t-o offset\toffset of TRX data in file (default: 0)\n");
- printf("\t-1 file\t\tfile to extract 1st partition to (optional)\n");
- printf("\t-2 file\t\tfile to extract 2nd partition to (optional)\n");
- printf("\t-3 file\t\tfile to extract 3rd partition to (optional)\n");
+ printf("\totrx extract <file> [options]\textract partitions from TRX file\n");
+ printf("\t-o offset\t\t\toffset of TRX data in file (default: 0)\n");
+ printf("\t-1 file\t\t\t\tfile to extract 1st partition to (optional)\n");
+ printf("\t-2 file\t\t\t\tfile to extract 2nd partition to (optional)\n");
+ printf("\t-3 file\t\t\t\tfile to extract 3rd partition to (optional)\n");
}
int main(int argc, char **argv) {
- parse_options(argc, argv);
-
- switch (mode) {
- case MODE_CHECK:
- return otrx_check();
- case MODE_EXTRACT:
- return otrx_extract();
- default:
- usage();
+ if (argc > 1) {
+ if (!strcmp(argv[1], "check"))
+ return otrx_check(argc, argv);
+ else if (!strcmp(argv[1], "extract"))
+ return otrx_extract(argc, argv);
}
+ usage();
return 0;
}