2 * ----------------------------------------------------------------------------
3 * "THE BEER-WARE LICENSE" (Revision 42):
4 * <phk@FreeBSD.org> wrote this file. As long as you retain this notice you
5 * can do whatever you want with this stuff. If we meet some day, and you think
6 * this stuff is worth it, you can buy me a beer in return. Poul-Henning Kamp
7 * ----------------------------------------------------------------------------
10 #include <sys/cdefs.h>
11 __FBSDID("$FreeBSD$");
20 #include <sys/sysctl.h>
21 #include <sys/stdint.h>
22 #include <sys/types.h>
24 #include <sys/ioctl.h>
25 #include <sys/disklabel.h>
36 const enum platform platform =
41 #elif defined(__i386__)
43 #elif defined(__alpha__)
45 #elif defined(__sparc64__)
47 #elif defined(__ia64__)
49 #elif defined(__ppc__)
51 #elif defined(__amd64__)
53 #elif defined(__arm__)
55 #elif defined(__mips__)
63 chunk_name(chunk_e type)
66 case unused: return ("unused");
67 case mbr: return ("mbr");
68 case part: return ("part");
69 case gpt: return ("gpt");
70 case pc98: return ("pc98");
71 case sun: return ("sun");
72 case freebsd: return ("freebsd");
73 case fat: return ("fat");
74 case spare: return ("spare");
75 case efi: return ("efi");
76 case apple: return ("apple");
77 default: return ("??");
82 Open_Disk(const char *name)
89 error = sysctlbyname("kern.geom.conftxt", NULL, &txtsize, NULL, 0);
91 warn("kern.geom.conftxt sysctl not available, giving up!");
94 conftxt = malloc(txtsize+1);
95 if (conftxt == NULL) {
96 warn("cannot malloc memory for conftxt");
99 error = sysctlbyname("kern.geom.conftxt", conftxt, &txtsize, NULL, 0);
101 warn("error reading kern.geom.conftxt from the system");
105 conftxt[txtsize] = '\0'; /* in case kernel bug is still there */
106 d = Int_Open_Disk(name, conftxt);
113 Debug_Disk(struct disk *d)
116 printf("Debug_Disk(%s)", d->name);
119 printf(" bios_geom=%lu/%lu/%lu = %lu\n",
120 d->bios_cyl, d->bios_hd, d->bios_sect,
121 d->bios_cyl * d->bios_hd * d->bios_sect);
123 printf(" boot1=%p, boot2=%p, bootipl=%p, bootmenu=%p\n",
124 d->boot1, d->boot2, d->bootipl, d->bootmenu);
125 #elif defined(__i386__) || defined(__amd64__)
126 printf(" boot1=%p, boot2=%p, bootmgr=%p\n",
127 d->boot1, d->boot2, d->bootmgr);
128 #elif defined(__alpha__)
129 printf(" boot1=%p, bootmgr=%p\n",
130 d->boot1, d->bootmgr);
132 /* Should be: error "Debug_Disk: unknown arch"; */
135 printf(" media size=%lu, sector size=%lu\n", d->media_size,
139 Debug_Chunk(d->chunks);
143 Free_Disk(struct disk *d)
146 Free_Chunk(d->chunks);
155 #if !defined(__ia64__)
160 #if !defined(__ia64__)
164 #if defined(__i386__) || defined(__amd64__)
173 Collapse_Disk(struct disk *d)
176 while (Collapse_Chunk(d, d->chunks))
182 qstrcmp(const void* a, const void* b)
184 const char *str1 = *(char* const*)a;
185 const char *str2 = *(char* const*)b;
187 return strcmp(str1, str2);
197 char *disklist, *disk1, *disk2;
199 error = sysctlbyname("kern.disks", NULL, &listsize, NULL, 0);
201 warn("kern.disks sysctl not available");
208 disks = malloc(sizeof *disks * (1 + MAX_NO_DISKS));
211 disk1 = disklist = (char *)malloc(listsize + 1);
212 if (disklist == NULL) {
216 memset(disks,0,sizeof *disks * (1 + MAX_NO_DISKS));
217 memset(disklist, 0, listsize + 1);
218 error = sysctlbyname("kern.disks", disklist, &listsize, NULL, 0);
219 if (error || disklist[0] == 0) {
224 for (disk_cnt = 0; disk_cnt < MAX_NO_DISKS; disk_cnt++) {
225 disk2 = strsep(&disk1, " ");
228 disks[disk_cnt] = strdup(disk2);
229 if (disks[disk_cnt] == NULL) {
230 for (disk_cnt--; disk_cnt >= 0; disk_cnt--)
231 free(disks[disk_cnt]);
237 qsort(disks, disk_cnt, sizeof(char*), qstrcmp);
244 Set_Boot_Mgr(struct disk *d, const u_char *bootipl, const size_t bootipl_size,
245 const u_char *bootmenu, const size_t bootmenu_size)
248 Set_Boot_Mgr(struct disk *d, const u_char *b, const size_t s)
251 #if !defined(__ia64__)
253 if (d->sector_size == 0)
255 if (bootipl_size % d->sector_size != 0)
262 d->bootipl_size = bootipl_size;
263 d->bootipl = malloc(bootipl_size);
266 memcpy(d->bootipl, bootipl, bootipl_size);
269 if (bootmenu_size % d->sector_size != 0)
276 d->bootmenu_size = bootmenu_size;
277 d->bootmenu = malloc(bootmenu_size);
280 memcpy(d->bootmenu, bootmenu, bootmenu_size);
283 if (d->sector_size == 0)
285 if (s % d->sector_size != 0)
293 d->bootmgr = malloc(s);
296 memcpy(d->bootmgr, b, s);
303 Set_Boot_Blocks(struct disk *d, const u_char *b1, const u_char *b2)
305 #if defined(__i386__) || defined(__amd64__)
308 d->boot1 = malloc(512);
311 memcpy(d->boot1, b1, 512);
314 d->boot2 = malloc(15 * 512);
317 memcpy(d->boot2, b2, 15 * 512);
318 #elif defined(__alpha__)
321 d->boot1 = malloc(15 * 512);
324 memcpy(d->boot1, b1, 15 * 512);
325 #elif defined(__sparc64__)
326 if (d->boot1 != NULL)
328 d->boot1 = malloc(16 * 512);
329 if (d->boot1 == NULL)
331 memcpy(d->boot1, b1, 16 * 512);
332 #elif defined(__ia64__)
335 /* Should be: #error "Set_Boot_Blocks: unknown arch"; */
341 slice_type_name( int type, int subtype )
349 case 1: return "fat (12-bit)";
350 case 2: return "XENIX /";
351 case 3: return "XENIX /usr";
352 case 4: return "fat (16-bit,<=32Mb)";
353 case 5: return "extended DOS";
354 case 6: return "fat (16-bit,>32Mb)";
355 case 7: return "NTFS/HPFS/QNX";
356 case 8: return "AIX bootable";
357 case 9: return "AIX data";
358 case 10: return "OS/2 bootmgr";
359 case 11: return "fat (32-bit)";
360 case 12: return "fat (32-bit,LBA)";
361 case 14: return "fat (16-bit,>32Mb,LBA)";
362 case 15: return "extended DOS, LBA";
363 case 18: return "Compaq Diagnostic";
364 case 57: return "Plan 9";
365 case 77: return "QNX 4.X";
366 case 78: return "QNX 4.X 2nd part";
367 case 79: return "QNX 4.X 3rd part";
368 case 84: return "OnTrack diskmgr";
369 case 100: return "Netware 2.x";
370 case 101: return "Netware 3.x";
371 case 115: return "SCO UnixWare";
372 case 128: return "Minix 1.1";
373 case 129: return "Minix 1.5";
374 case 130: return "linux_swap";
375 case 131: return "ext2fs";
376 case 133: return "linux extended";
377 case 166: return "OpenBSD FFS"; /* 0xA6 */
378 case 168: return "Mac OS-X";
379 case 169: return "NetBSD FFS"; /* 0xA9 */
380 case 171: return "Mac OS-X Boot";
381 case 182: return "OpenBSD"; /* dedicated */
382 case 183: return "bsd/os";
383 case 184: return "bsd/os swap";
384 case 191: return "Solaris (new)";
385 case 238: return "EFI GPT";
386 case 239: return "EFI Sys. Part.";
387 default: return "unknown";
394 case 0xc494: return "freebsd";
396 case 165: return "freebsd";
398 default: return "unknown";