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
)
79 result
= strtoul(string
, &end
, 0);
80 switch (tolower(*end
)) {
82 case '\0' : exp
= 0; break;
83 case 'm' : exp
= 1; break;
84 case 'g' : exp
= 2; break;
92 fprintf(stderr
, "garbage after end of number\n");
96 /* result: number + 1024^(exp) */
99 return result
* (2 << ((10 * exp
) - 1));
102 /* convert the sector number into a CHS value for the partition table */
103 static void to_chs(long sect
, unsigned char chs
[3])
107 s
= (sect
% sectors
) + 1;
108 sect
= sect
/ sectors
;
114 chs
[1] = s
| ((c
>> 2) & 0xC0);
120 /* round the sector number up to the next cylinder */
121 static inline unsigned long round_to_cyl(long sect
)
123 int cyl_size
= heads
* sectors
;
125 return sect
+ cyl_size
- (sect
% cyl_size
);
128 /* round the sector number up to the kb_align boundary */
129 static inline unsigned long round_to_kb(long sect
) {
130 return ((sect
- 1) / kb_align
+ 1) * kb_align
;
133 /* check the partition sizes and write the partition table */
134 static int gen_ptable(uint32_t signature
, int nr
)
137 unsigned long sect
= 0;
138 int i
, fd
, ret
= -1, start
, len
;
140 memset(pte
, 0, sizeof(struct pte
) * 4);
141 for (i
= 0; i
< nr
; i
++) {
142 if (!parts
[i
].size
) {
143 fprintf(stderr
, "Invalid size in partition %d!\n", i
);
147 pte
[i
].active
= ((i
+ 1) == active
) ? 0x80 : 0;
148 pte
[i
].type
= parts
[i
].type
;
150 start
= sect
+ sectors
;
152 start
= round_to_kb(start
);
153 pte
[i
].start
= cpu_to_le32(start
);
155 sect
= start
+ parts
[i
].size
* 2;
157 sect
= round_to_cyl(sect
);
158 pte
[i
].length
= cpu_to_le32(len
= sect
- start
);
160 to_chs(start
, pte
[i
].chs_start
);
161 to_chs(start
+ len
- 1, pte
[i
].chs_end
);
164 fprintf(stderr
, "Partition %d: start=%ld, end=%ld, size=%ld\n", i
, (long)start
* 512, ((long)start
+ (long)len
) * 512, (long)len
* 512);
165 printf("%ld\n", (long)start
* 512);
166 printf("%ld\n", (long)len
* 512);
169 if ((fd
= open(filename
, O_WRONLY
|O_CREAT
|O_TRUNC
, 0644)) < 0) {
170 fprintf(stderr
, "Can't open output file '%s'\n",filename
);
174 lseek(fd
, 440, SEEK_SET
);
175 if (write(fd
, &signature
, sizeof(signature
)) != sizeof(signature
)) {
176 fprintf(stderr
, "write failed.\n");
180 lseek(fd
, 446, SEEK_SET
);
181 if (write(fd
, pte
, sizeof(struct pte
) * 4) != sizeof(struct pte
) * 4) {
182 fprintf(stderr
, "write failed.\n");
185 lseek(fd
, 510, SEEK_SET
);
186 if (write(fd
, "\x55\xaa", 2) != 2) {
187 fprintf(stderr
, "write failed.\n");
197 static void usage(char *prog
)
199 fprintf(stderr
, "Usage: %s [-v] -h <heads> -s <sectors> -o <outputfile> [-a 0..4] [-l <align kB>] [[-t <type>] -p <size>...] \n", prog
);
203 int main (int argc
, char **argv
)
208 uint32_t signature
= 0x5452574F; /* 'OWRT' */
210 while ((ch
= getopt(argc
, argv
, "h:s:p:a:t:o:vl:S:")) != -1) {
219 heads
= (int)strtoul(optarg
, NULL
, 0);
222 sectors
= (int)strtoul(optarg
, NULL
, 0);
226 fprintf(stderr
, "Too many partitions\n");
229 parts
[part
].size
= to_kbytes(optarg
);
230 parts
[part
++].type
= type
;
233 type
= (char)strtoul(optarg
, NULL
, 16);
236 active
= (int)strtoul(optarg
, NULL
, 0);
237 if ((active
< 0) || (active
> 4))
241 kb_align
= (int)strtoul(optarg
, NULL
, 0) * 2;
244 signature
= strtoul(optarg
, NULL
, 0);
252 if (argc
|| (heads
<= 0) || (sectors
<= 0) || !filename
)
255 return gen_ptable(signature
, part
) ? EXIT_FAILURE
: EXIT_SUCCESS
;