diff options
author | Jo-Philipp Wich <jow@openwrt.org> | 2014-10-16 16:16:47 +0000 |
---|---|---|
committer | Jo-Philipp Wich <jow@openwrt.org> | 2014-10-16 16:16:47 +0000 |
commit | 872c274a3a64fb31343a2401d369e9acb895569a (patch) | |
tree | 93a19c5d8203b0564068ac8430821730d0385e4a /tools/firmware-utils/src | |
parent | 41f42651018420470e57ee4483db06ccad8ec0c6 (diff) | |
download | master-187ad058-872c274a3a64fb31343a2401d369e9acb895569a.tar.gz master-187ad058-872c274a3a64fb31343a2401d369e9acb895569a.tar.bz2 master-187ad058-872c274a3a64fb31343a2401d369e9acb895569a.zip |
firmware-utils: allow passing a specific MBR signature to ptgen
Signed-off-by: Jo-Philipp Wich <jow@openwrt.org>
git-svn-id: svn://svn.openwrt.org/openwrt/trunk@42934 3c298f89-4303-0410-b956-a3cf2f4a3e73
Diffstat (limited to 'tools/firmware-utils/src')
-rw-r--r-- | tools/firmware-utils/src/ptgen.c | 19 |
1 files changed, 15 insertions, 4 deletions
diff --git a/tools/firmware-utils/src/ptgen.c b/tools/firmware-utils/src/ptgen.c index 0247fd0131..6379ed7a59 100644 --- a/tools/firmware-utils/src/ptgen.c +++ b/tools/firmware-utils/src/ptgen.c @@ -28,6 +28,7 @@ #include <stdio.h> #include <ctype.h> #include <fcntl.h> +#include <stdint.h> #if __BYTE_ORDER == __BIG_ENDIAN #define cpu_to_le16(x) bswap_16(x) @@ -124,7 +125,7 @@ static inline unsigned long round_to_kb(long sect) { } /* check the partition sizes and write the partition table */ -static int gen_ptable(int nr) +static int gen_ptable(uint32_t signature, int nr) { struct pte pte[4]; unsigned long sect = 0; @@ -159,6 +160,12 @@ static int gen_ptable(int nr) return -1; } + lseek(fd, 440, SEEK_SET); + if (write(fd, &signature, sizeof(signature)) != sizeof(signature)) { + fprintf(stderr, "write failed.\n"); + goto fail; + } + lseek(fd, 446, SEEK_SET); if (write(fd, pte, sizeof(struct pte) * 4) != sizeof(struct pte) * 4) { fprintf(stderr, "write failed.\n"); @@ -187,8 +194,9 @@ int main (int argc, char **argv) char type = 0x83; int ch; int part = 0; + uint32_t signature = 0x5452574F; /* 'OWRT' */ - while ((ch = getopt(argc, argv, "h:s:p:a:t:o:vl:")) != -1) { + while ((ch = getopt(argc, argv, "h:s:p:a:t:o:vl:S:")) != -1) { switch (ch) { case 'o': filename = optarg; @@ -221,6 +229,9 @@ int main (int argc, char **argv) case 'l': kb_align = (int) strtoul(optarg, NULL, 0) * 2; break; + case 'S': + signature = strtoul(optarg, NULL, 0); + break; case '?': default: usage(argv[0]); @@ -229,6 +240,6 @@ int main (int argc, char **argv) argc -= optind; if (argc || (heads <= 0) || (sectors <= 0) || !filename) usage(argv[0]); - - return gen_ptable(part); + + return gen_ptable(signature, part); } |