2 * ptgen - partition table generator
3 * Copyright (C) 2006 by Felix Fietkau <nbd@openwrt.org>
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>
33 #if __BYTE_ORDER == __BIG_ENDIAN
34 #define cpu_to_le16(x) bswap_16(x)
35 #elif __BYTE_ORDER == __LITTLE_ENDIAN
36 #define cpu_to_le16(x) (x)
38 #error unknown endianness!
41 /* Partition table entry */
44 unsigned char chs_start
[3];
46 unsigned char chs_end
[3];
61 struct partinfo parts
[4];
62 char *filename
= NULL
;
66 * parse the size argument, which is either
67 * a simple number (K assumed) or
70 * returns the size in KByte
72 static long to_kbytes(const char *string
) {
77 result
= strtoul(string
, &end
, 0);
78 switch (tolower(*end
)) {
80 case '\0' : exp
= 0; break;
81 case 'm' : exp
= 1; break;
82 case 'g' : exp
= 2; break;
90 fprintf(stderr
, "garbage after end of number\n");
94 /* result: number + 1024^(exp) */
95 return result
* ((2 << ((10 * exp
) - 1)) ?: 1);
98 /* convert the sector number into a CHS value for the partition table */
99 static void to_chs(long sect
, unsigned char chs
[3]) {
102 s
= (sect
% sectors
) + 1;
103 sect
= sect
/ sectors
;
109 chs
[1] = s
| ((c
>> 2) & 0xC0);
115 /* round the sector number up to the next cylinder */
116 static inline unsigned long round_to_cyl(long sect
) {
117 int cyl_size
= heads
* sectors
;
119 return sect
+ cyl_size
- (sect
% cyl_size
);
122 /* round the sector number up to the kb_align boundary */
123 static inline unsigned long round_to_kb(long sect
) {
124 return ((sect
- 1) / kb_align
+ 1) * kb_align
;
127 /* check the partition sizes and write the partition table */
128 static int gen_ptable(uint32_t signature
, int nr
)
131 unsigned long sect
= 0;
132 int i
, fd
, ret
= -1, start
, len
;
134 memset(pte
, 0, sizeof(struct pte
) * 4);
135 for (i
= 0; i
< nr
; i
++) {
136 if (!parts
[i
].size
) {
137 fprintf(stderr
, "Invalid size in partition %d!\n", i
);
140 pte
[i
].active
= ((i
+ 1) == active
) ? 0x80 : 0;
141 pte
[i
].type
= parts
[i
].type
;
142 start
= sect
+ sectors
;
144 start
= round_to_kb(start
);
145 pte
[i
].start
= cpu_to_le16(start
);
146 sect
= start
+ parts
[i
].size
* 2;
148 sect
= round_to_cyl(sect
);
149 pte
[i
].length
= cpu_to_le16(len
= sect
- start
);
150 to_chs(start
, pte
[i
].chs_start
);
151 to_chs(start
+ len
- 1, pte
[i
].chs_end
);
153 fprintf(stderr
, "Partition %d: start=%ld, end=%ld, size=%ld\n", i
, (long) start
* 512, ((long) start
+ (long) len
) * 512, (long) len
* 512);
154 printf("%ld\n", ((long) start
* 512));
155 printf("%ld\n", ((long) len
* 512));
158 if ((fd
= open(filename
, O_WRONLY
|O_CREAT
|O_TRUNC
, 0644)) < 0) {
159 fprintf(stderr
, "Can't open output file '%s'\n",filename
);
163 lseek(fd
, 440, SEEK_SET
);
164 if (write(fd
, &signature
, sizeof(signature
)) != sizeof(signature
)) {
165 fprintf(stderr
, "write failed.\n");
169 lseek(fd
, 446, SEEK_SET
);
170 if (write(fd
, pte
, sizeof(struct pte
) * 4) != sizeof(struct pte
) * 4) {
171 fprintf(stderr
, "write failed.\n");
174 lseek(fd
, 510, SEEK_SET
);
175 if (write(fd
, "\x55\xaa", 2) != 2) {
176 fprintf(stderr
, "write failed.\n");
186 static void usage(char *prog
)
188 fprintf(stderr
, "Usage: %s [-v] -h <heads> -s <sectors> -o <outputfile> [-a 0..4] [-l <align kB>] [[-t <type>] -p <size>...] \n", prog
);
192 int main (int argc
, char **argv
)
197 uint32_t signature
= 0x5452574F; /* 'OWRT' */
199 while ((ch
= getopt(argc
, argv
, "h:s:p:a:t:o:vl:S:")) != -1) {
208 heads
= (int) strtoul(optarg
, NULL
, 0);
211 sectors
= (int) strtoul(optarg
, NULL
, 0);
215 fprintf(stderr
, "Too many partitions\n");
218 parts
[part
].size
= to_kbytes(optarg
);
219 parts
[part
++].type
= type
;
222 type
= (char) strtoul(optarg
, NULL
, 16);
225 active
= (int) strtoul(optarg
, NULL
, 0);
226 if ((active
< 0) || (active
> 4))
230 kb_align
= (int) strtoul(optarg
, NULL
, 0) * 2;
233 signature
= strtoul(optarg
, NULL
, 0);
241 if (argc
|| (heads
<= 0) || (sectors
<= 0) || !filename
)
244 return gen_ptable(signature
, part
);