2 * ptgen - partition table generator
3 * Copyright (C) 2006 by Felix Fietkau <nbd@nbd.name>
6 * Copyright (C) 2002 by David Roetzel <david@roetzel.de>
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 2 of the License, or
11 * (at your option) any later version.
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
18 * You should have received a copy of the GNU General Public License
19 * along with this program; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
23 #include <sys/types.h>
34 #if __BYTE_ORDER == __BIG_ENDIAN
35 #define cpu_to_le32(x) bswap_32(x)
36 #elif __BYTE_ORDER == __LITTLE_ENDIAN
37 #define cpu_to_le32(x) (x)
39 #error unknown endianness!
42 /* Partition table entry */
62 struct partinfo parts
[4];
63 char *filename
= NULL
;
67 * parse the size argument, which is either
68 * a simple number (K assumed) or
71 * returns the size in KByte
73 static long to_kbytes(const char *string
) {
78 result
= strtoul(string
, &end
, 0);
79 switch (tolower(*end
)) {
81 case '\0' : exp
= 0; break;
82 case 'm' : exp
= 1; break;
83 case 'g' : exp
= 2; break;
91 fprintf(stderr
, "garbage after end of number\n");
95 /* result: number + 1024^(exp) */
98 return result
* (2 << ((10 * exp
) - 1));
101 /* convert the sector number into a CHS value for the partition table */
102 static void to_chs(long sect
, unsigned char chs
[3]) {
105 s
= (sect
% sectors
) + 1;
106 sect
= sect
/ sectors
;
112 chs
[1] = s
| ((c
>> 2) & 0xC0);
118 /* round the sector number up to the next cylinder */
119 static inline unsigned long round_to_cyl(long sect
) {
120 int cyl_size
= heads
* sectors
;
122 return sect
+ cyl_size
- (sect
% cyl_size
);
125 /* round the sector number up to the kb_align boundary */
126 static inline unsigned long round_to_kb(long sect
) {
127 return ((sect
- 1) / kb_align
+ 1) * kb_align
;
130 /* check the partition sizes and write the partition table */
131 static int gen_ptable(uint32_t signature
, int nr
)
134 unsigned long sect
= 0;
135 int i
, fd
, ret
= -1, start
, len
;
137 memset(pte
, 0, sizeof(struct pte
) * 4);
138 for (i
= 0; i
< nr
; i
++) {
139 if (!parts
[i
].size
) {
140 fprintf(stderr
, "Invalid size in partition %d!\n", i
);
143 pte
[i
].active
= ((i
+ 1) == active
) ? 0x80 : 0;
144 pte
[i
].type
= parts
[i
].type
;
145 start
= sect
+ sectors
;
147 start
= round_to_kb(start
);
148 pte
[i
].start
= cpu_to_le32(start
);
149 sect
= start
+ parts
[i
].size
* 2;
151 sect
= round_to_cyl(sect
);
152 pte
[i
].length
= cpu_to_le32(len
= sect
- start
);
153 to_chs(start
, pte
[i
].chs_start
);
154 to_chs(start
+ len
- 1, pte
[i
].chs_end
);
156 fprintf(stderr
, "Partition %d: start=%ld, end=%ld, size=%ld\n", i
, (long) start
* 512, ((long) start
+ (long) len
) * 512, (long) len
* 512);
157 printf("%ld\n", ((long) start
* 512));
158 printf("%ld\n", ((long) len
* 512));
161 if ((fd
= open(filename
, O_WRONLY
|O_CREAT
|O_TRUNC
, 0644)) < 0) {
162 fprintf(stderr
, "Can't open output file '%s'\n",filename
);
166 lseek(fd
, 440, SEEK_SET
);
167 if (write(fd
, &signature
, sizeof(signature
)) != sizeof(signature
)) {
168 fprintf(stderr
, "write failed.\n");
172 lseek(fd
, 446, SEEK_SET
);
173 if (write(fd
, pte
, sizeof(struct pte
) * 4) != sizeof(struct pte
) * 4) {
174 fprintf(stderr
, "write failed.\n");
177 lseek(fd
, 510, SEEK_SET
);
178 if (write(fd
, "\x55\xaa", 2) != 2) {
179 fprintf(stderr
, "write failed.\n");
189 static void usage(char *prog
)
191 fprintf(stderr
, "Usage: %s [-v] -h <heads> -s <sectors> -o <outputfile> [-a 0..4] [-l <align kB>] [[-t <type>] -p <size>...] \n", prog
);
195 int main (int argc
, char **argv
)
200 uint32_t signature
= 0x5452574F; /* 'OWRT' */
202 while ((ch
= getopt(argc
, argv
, "h:s:p:a:t:o:vl:S:")) != -1) {
211 heads
= (int) strtoul(optarg
, NULL
, 0);
214 sectors
= (int) strtoul(optarg
, NULL
, 0);
218 fprintf(stderr
, "Too many partitions\n");
221 parts
[part
].size
= to_kbytes(optarg
);
222 parts
[part
++].type
= type
;
225 type
= (char) strtoul(optarg
, NULL
, 16);
228 active
= (int) strtoul(optarg
, NULL
, 0);
229 if ((active
< 0) || (active
> 4))
233 kb_align
= (int) strtoul(optarg
, NULL
, 0) * 2;
236 signature
= strtoul(optarg
, NULL
, 0);
244 if (argc
|| (heads
<= 0) || (sectors
<= 0) || !filename
)
247 return gen_ptable(signature
, part
);