firware-utils/mktplinkfw: add TL-WR2543N/ND support

[openwrt.git] / tools / firmware-utils / src / mkdniimg.c

diff --git a/tools/firmware-utils/src/mkdniimg.c b/tools/firmware-utils/src/mkdniimg.c
index 31b3210..7230f09 100644 (file)
--- a/tools/firmware-utils/src/mkdniimg.c
+++ b/tools/firmware-utils/src/mkdniimg.c
@@ -27,7 +27,8 @@ static char *ifname;
 static char *progname;
 static char *ofname;
 static char *version = "1.00.00";
 static char *progname;
 static char *ofname;
 static char *version = "1.00.00";

-static char *region ="NA";
+static char *region = "";
+static char *hd_id;

 
 static char *board_id;
 /*
 
 static char *board_id;
 /*


@@ -60,6 +61,7 @@ void usage(int status)
 "  -o <file>       write output to the file <file>\n"
 "  -v <version>    set image version to <version>\n"
 "  -r <region>     set image region to <region>\n"
 "  -o <file>       write output to the file <file>\n"
 "  -v <version>    set image version to <version>\n"
 "  -r <region>     set image region to <region>\n"

+"  -H <hd_id>      set image hardware id to <hd_id>\n"

 "  -h              show this screen\n"
        );
 
 "  -h              show this screen\n"
        );
 


@@ -73,7 +75,7 @@ int main(int argc, char *argv[])
        int err;
        struct stat st;
        char *buf;
        int err;
        struct stat st;
        char *buf;

-       int i;
+       int pos, rem, i;

        uint8_t csum;
 
        FILE *outfile, *infile;
        uint8_t csum;
 
        FILE *outfile, *infile;


@@ -83,7 +85,7 @@ int main(int argc, char *argv[])
        while ( 1 ) {
                int c;
 
        while ( 1 ) {
                int c;
 

-               c = getopt(argc, argv, "B:i:o:v:r:h");
+               c = getopt(argc, argv, "B:i:o:v:r:H:h");

                if (c == -1)
                        break;
 
                if (c == -1)
                        break;
 


@@ -103,6 +105,9 @@ int main(int argc, char *argv[])
                case 'r':
                        region = optarg;
                        break;
                case 'r':
                        region = optarg;
                        break;

+               case 'H':
+                       hd_id = optarg;
+                       break;

                case 'h':
                        usage(EXIT_SUCCESS);
                        break;
                case 'h':
                        usage(EXIT_SUCCESS);
                        break;


@@ -141,8 +146,12 @@ int main(int argc, char *argv[])
        }
 
        memset(buf, 0, DNI_HDR_LEN);
        }
 
        memset(buf, 0, DNI_HDR_LEN);

-       snprintf(buf, DNI_HDR_LEN, "device:%s\nversion:V%s\nregion:%s\n",
-                board_id, version, region);
+       pos = snprintf(buf, DNI_HDR_LEN, "device:%s\nversion:V%s\nregion:%s\n",
+                      board_id, version, region);
+       rem = DNI_HDR_LEN - pos;
+       if (pos >= 0 && rem > 1 && hd_id) {
+               snprintf(buf + pos, rem, "hd_id:%s\n", hd_id);
+       }

 
        infile = fopen(ifname, "r");
        if (infile == NULL) {
 
        infile = fopen(ifname, "r");
        if (infile == NULL) {