[tools] tools/firmware-utils/ptgen option to use MB rounding

[openwrt/svn-archive/archive.git] / tools / firmware-utils / src / ptgen.c

diff --git a/tools/firmware-utils/src/ptgen.c b/tools/firmware-utils/src/ptgen.c
index 2d78eae53514cdcfa6f1d9f54b45a8f351ff8d6a..0247fd0131cbd6b9899b94e017bcf2a6d0afa8bb 100644 (file)
--- a/tools/firmware-utils/src/ptgen.c
+++ b/tools/firmware-utils/src/ptgen.c
@@ -56,6 +56,7 @@ int verbose = 0;
 int active = 1;
 int heads = -1;
 int sectors = -1;
+int kb_align = 0;
 struct partinfo parts[4];
 char *filename = NULL;
 
@@ -114,7 +115,12 @@ static void to_chs(long sect, unsigned char chs[3]) {
 static inline unsigned long round_to_cyl(long sect) {
        int cyl_size = heads * sectors;
 
-       return sect + cyl_size - ((sect % cyl_size) ?: cyl_size); 
+       return sect + cyl_size - (sect % cyl_size); 
+}
+
+/* round the sector number up to the kb_align boundary */
+static inline unsigned long round_to_kb(long sect) {
+        return ((sect - 1) / kb_align + 1) * kb_align;
 }
 
 /* check the partition sizes and write the partition table */
@@ -132,17 +138,23 @@ static int gen_ptable(int nr)
                }
                pte[i].active = ((i + 1) == active) ? 0x80 : 0;
                pte[i].type = parts[i].type;
-               pte[i].start = cpu_to_le16(start = sect + sectors);
-               sect = round_to_cyl(start + parts[i].size * 2);
+               start = sect + sectors;
+               if (kb_align != 0)
+                       start = round_to_kb(start);
+               pte[i].start = cpu_to_le16(start);
+               sect = start + parts[i].size * 2;
+               if (kb_align == 0)
+                       sect = round_to_cyl(sect);
                pte[i].length = cpu_to_le16(len = sect - start);
                to_chs(start, pte[i].chs_start);
                to_chs(start + len - 1, pte[i].chs_end);
                if (verbose)
                        fprintf(stderr, "Partition %d: start=%ld, end=%ld, size=%ld\n", i, (long) start * 512, ((long) start + (long) len) * 512, (long) len * 512);
                printf("%ld\n", ((long) start * 512));
+               printf("%ld\n", ((long) len * 512));
        }
 
-       if ((fd = open(filename, O_WRONLY|O_CREAT, 0644)) < 0) {
+       if ((fd = open(filename, O_WRONLY|O_CREAT|O_TRUNC, 0644)) < 0) {
                fprintf(stderr, "Can't open output file '%s'\n",filename);
                return -1;
        }
@@ -166,7 +178,7 @@ fail:
 
 static void usage(char *prog)
 {
-       fprintf(stderr, "Usage: %s [-v] -h <heads> -s <sectors> -o <outputfile> [-a 0..4] [[-t <type>] -p <size>...] \n", prog);
+       fprintf(stderr, "Usage: %s [-v] -h <heads> -s <sectors> -o <outputfile> [-a 0..4] [-l <align kB>] [[-t <type>] -p <size>...] \n", prog);
        exit(1);
 }
 
@@ -176,7 +188,7 @@ int main (int argc, char **argv)
        int ch;
        int part = 0;
 
-       while ((ch = getopt(argc, argv, "h:s:p:a:t:o:v")) != -1) {
+       while ((ch = getopt(argc, argv, "h:s:p:a:t:o:vl:")) != -1) {
                switch (ch) {
                case 'o':
                        filename = optarg;
@@ -206,6 +218,9 @@ int main (int argc, char **argv)
                        if ((active < 0) || (active > 4))
                                active = 0;
                        break;
+               case 'l':
+                       kb_align = (int) strtoul(optarg, NULL, 0) * 2;
+                       break;
                case '?':
                default:
                        usage(argv[0]);