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);