disks.c revision 1.7 1 /* $NetBSD: disks.c,v 1.7 2015/01/02 19:43:13 abs Exp $ */
2
3 /*
4 * Copyright 1997 Piermont Information Systems Inc.
5 * All rights reserved.
6 *
7 * Written by Philip A. Nelson for Piermont Information Systems Inc.
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions
11 * are met:
12 * 1. Redistributions of source code must retain the above copyright
13 * notice, this list of conditions and the following disclaimer.
14 * 2. Redistributions in binary form must reproduce the above copyright
15 * notice, this list of conditions and the following disclaimer in the
16 * documentation and/or other materials provided with the distribution.
17 * 3. The name of Piermont Information Systems Inc. may not be used to endorse
18 * or promote products derived from this software without specific prior
19 * written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY PIERMONT INFORMATION SYSTEMS INC. ``AS IS''
22 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED. IN NO EVENT SHALL PIERMONT INFORMATION SYSTEMS INC. BE
25 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
26 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
27 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
28 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
29 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
30 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
31 * THE POSSIBILITY OF SUCH DAMAGE.
32 *
33 */
34
35 /* disks.c -- routines to deal with finding disks and labeling disks. */
36
37
38 #include <errno.h>
39 #include <stdio.h>
40 #include <stdlib.h>
41 #include <unistd.h>
42 #include <fcntl.h>
43 #include <util.h>
44 #include <uuid.h>
45
46 #include <sys/param.h>
47 #include <sys/sysctl.h>
48 #include <sys/swap.h>
49 #include <ufs/ufs/dinode.h>
50 #include <ufs/ffs/fs.h>
51 #define FSTYPENAMES
52 #include <sys/disklabel.h>
53 #include <sys/disklabel_gpt.h>
54
55 #include <dev/scsipi/scsipi_all.h>
56 #include <sys/scsiio.h>
57
58 #include <dev/ata/atareg.h>
59 #include <sys/ataio.h>
60
61 #include "defs.h"
62 #include "md.h"
63 #include "msg_defs.h"
64 #include "menu_defs.h"
65 #include "txtwalk.h"
66
67 /* Disk descriptions */
68 struct disk_desc {
69 char dd_name[SSTRSIZE];
70 char dd_descr[70];
71 uint dd_no_mbr;
72 uint dd_cyl;
73 uint dd_head;
74 uint dd_sec;
75 uint dd_secsize;
76 uint dd_totsec;
77 };
78
79 /* gpt(8) use different filesystem names.
80 So, we cant use ./common/lib/libutil/getfstypename.c */
81 struct gptfs_t {
82 const char *name;
83 int id;
84 uuid_t uuid;
85 };
86 static const struct gptfs_t gpt_filesystems[] = {
87 { "swap", FS_SWAP, GPT_ENT_TYPE_NETBSD_SWAP, },
88 { "ffs", FS_BSDFFS, GPT_ENT_TYPE_NETBSD_FFS, },
89 { "lfs", FS_BSDLFS, GPT_ENT_TYPE_NETBSD_LFS, },
90 { "linux", FS_EX2FS, GPT_ENT_TYPE_LINUX_DATA, },
91 { "windows,", FS_MSDOS, GPT_ENT_TYPE_MS_BASIC_DATA, },
92 { "hfs", FS_HFS, GPT_ENT_TYPE_APPLE_HFS, },
93 { "ufs", FS_OTHER, GPT_ENT_TYPE_APPLE_UFS, },
94 { "ccd", FS_CCD, GPT_ENT_TYPE_NETBSD_CCD, },
95 { "raid", FS_RAID, GPT_ENT_TYPE_NETBSD_RAIDFRAME, },
96 { "cgd", FS_CGD, GPT_ENT_TYPE_NETBSD_CGD, },
97 { "efi", FS_OTHER, GPT_ENT_TYPE_EFI, },
98 { "bios", FS_OTHER, GPT_ENT_TYPE_BIOS, },
99 { NULL, -1, GPT_ENT_TYPE_UNUSED, },
100 };
101
102 /* Local prototypes */
103 static int foundffs(struct data *, size_t);
104 #ifdef USE_SYSVBFS
105 static int foundsysvbfs(struct data *, size_t);
106 #endif
107 static int fsck_preen(const char *, int, const char *);
108 static void fixsb(const char *, const char *, char);
109 static bool is_gpt(const char *);
110 static int incoregpt(pm_devs_t *, partinfo *);
111 static bool have_gpt_binary(void);
112
113 #ifndef DISK_NAMES
114 #define DISK_NAMES "wd", "sd", "ld", "raid"
115 #endif
116
117 static const char *disk_names[] = { DISK_NAMES, "vnd", "cgd", NULL };
118
119 static bool tmpfs_on_var_shm(void);
120
121 const char *
122 getfslabelname(uint8_t f)
123 {
124 if (f >= __arraycount(fstypenames) || fstypenames[f] == NULL)
125 return "invalid";
126 return fstypenames[f];
127 }
128
129 /*
130 * Decide wether we want to mount a tmpfs on /var/shm: we do this always
131 * when the machine has more than 16 MB of user memory. On smaller machines,
132 * shm_open() and friends will not perform well anyway.
133 */
134 static bool
135 tmpfs_on_var_shm()
136 {
137 uint64_t ram;
138 size_t len;
139
140 len = sizeof(ram);
141 if (sysctlbyname("hw.usermem64", &ram, &len, NULL, 0))
142 return false;
143
144 return ram > 16UL*1024UL*1024UL;
145 }
146
147 /* from src/sbin/atactl/atactl.c
148 * extract_string: copy a block of bytes out of ataparams and make
149 * a proper string out of it, truncating trailing spaces and preserving
150 * strict typing. And also, not doing unaligned accesses.
151 */
152 static void
153 ata_extract_string(char *buf, size_t bufmax,
154 uint8_t *bytes, unsigned numbytes,
155 int needswap)
156 {
157 unsigned i;
158 size_t j;
159 unsigned char ch1, ch2;
160
161 for (i = 0, j = 0; i < numbytes; i += 2) {
162 ch1 = bytes[i];
163 ch2 = bytes[i+1];
164 if (needswap && j < bufmax-1) {
165 buf[j++] = ch2;
166 }
167 if (j < bufmax-1) {
168 buf[j++] = ch1;
169 }
170 if (!needswap && j < bufmax-1) {
171 buf[j++] = ch2;
172 }
173 }
174 while (j > 0 && buf[j-1] == ' ') {
175 j--;
176 }
177 buf[j] = '\0';
178 }
179
180 /*
181 * from src/sbin/scsictl/scsi_subr.c
182 */
183 #define STRVIS_ISWHITE(x) ((x) == ' ' || (x) == '\0' || (x) == (u_char)'\377')
184
185 static void
186 scsi_strvis(char *sdst, size_t dlen, const char *ssrc, size_t slen)
187 {
188 u_char *dst = (u_char *)sdst;
189 const u_char *src = (const u_char *)ssrc;
190
191 /* Trim leading and trailing blanks and NULs. */
192 while (slen > 0 && STRVIS_ISWHITE(src[0]))
193 ++src, --slen;
194 while (slen > 0 && STRVIS_ISWHITE(src[slen - 1]))
195 --slen;
196
197 while (slen > 0) {
198 if (*src < 0x20 || *src >= 0x80) {
199 /* non-printable characters */
200 dlen -= 4;
201 if (dlen < 1)
202 break;
203 *dst++ = '\\';
204 *dst++ = ((*src & 0300) >> 6) + '0';
205 *dst++ = ((*src & 0070) >> 3) + '0';
206 *dst++ = ((*src & 0007) >> 0) + '0';
207 } else if (*src == '\\') {
208 /* quote characters */
209 dlen -= 2;
210 if (dlen < 1)
211 break;
212 *dst++ = '\\';
213 *dst++ = '\\';
214 } else {
215 /* normal characters */
216 if (--dlen < 1)
217 break;
218 *dst++ = *src;
219 }
220 ++src, --slen;
221 }
222
223 *dst++ = 0;
224 }
225
226
227 static int
228 get_descr_scsi(struct disk_desc *dd, int fd)
229 {
230 struct scsipi_inquiry_data inqbuf;
231 struct scsipi_inquiry cmd;
232 scsireq_t req;
233 /* x4 in case every character is escaped, +1 for NUL. */
234 char vendor[(sizeof(inqbuf.vendor) * 4) + 1],
235 product[(sizeof(inqbuf.product) * 4) + 1],
236 revision[(sizeof(inqbuf.revision) * 4) + 1];
237 char size[5];
238 int error;
239
240 memset(&inqbuf, 0, sizeof(inqbuf));
241 memset(&cmd, 0, sizeof(cmd));
242 memset(&req, 0, sizeof(req));
243
244 cmd.opcode = INQUIRY;
245 cmd.length = sizeof(inqbuf);
246 memcpy(req.cmd, &cmd, sizeof(cmd));
247 req.cmdlen = sizeof(cmd);
248 req.databuf = &inqbuf;
249 req.datalen = sizeof(inqbuf);
250 req.timeout = 10000;
251 req.flags = SCCMD_READ;
252 req.senselen = SENSEBUFLEN;
253
254 error = ioctl(fd, SCIOCCOMMAND, &req);
255 if (error == -1 || req.retsts != SCCMD_OK)
256 return 0;
257
258 scsi_strvis(vendor, sizeof(vendor), inqbuf.vendor,
259 sizeof(inqbuf.vendor));
260 scsi_strvis(product, sizeof(product), inqbuf.product,
261 sizeof(inqbuf.product));
262 scsi_strvis(revision, sizeof(revision), inqbuf.revision,
263 sizeof(inqbuf.revision));
264
265 humanize_number(size, sizeof(size),
266 (uint64_t)dd->dd_secsize * (uint64_t)dd->dd_totsec,
267 "", HN_AUTOSCALE, HN_B | HN_NOSPACE | HN_DECIMAL);
268
269 snprintf(dd->dd_descr, sizeof(dd->dd_descr),
270 "%s (%s, %s %s)",
271 dd->dd_name, size, vendor, product);
272
273 return 1;
274 }
275
276 static int
277 get_descr_ata(struct disk_desc *dd, int fd)
278 {
279 struct atareq req;
280 static union {
281 unsigned char inbuf[DEV_BSIZE];
282 struct ataparams inqbuf;
283 } inbuf;
284 struct ataparams *inqbuf = &inbuf.inqbuf;
285 char model[sizeof(inqbuf->atap_model)+1];
286 char size[5];
287 int error, needswap = 0;
288
289 memset(&inbuf, 0, sizeof(inbuf));
290 memset(&req, 0, sizeof(req));
291
292 req.flags = ATACMD_READ;
293 req.command = WDCC_IDENTIFY;
294 req.databuf = (void *)&inbuf;
295 req.datalen = sizeof(inbuf);
296 req.timeout = 1000;
297
298 error = ioctl(fd, ATAIOCCOMMAND, &req);
299 if (error == -1 || req.retsts != ATACMD_OK)
300 return 0;
301
302 #if BYTE_ORDER == LITTLE_ENDIAN
303 /*
304 * On little endian machines, we need to shuffle the string
305 * byte order. However, we don't have to do this for NEC or
306 * Mitsumi ATAPI devices
307 */
308
309 if (!(inqbuf->atap_config != WDC_CFG_CFA_MAGIC &&
310 (inqbuf->atap_config & WDC_CFG_ATAPI) &&
311 ((inqbuf->atap_model[0] == 'N' &&
312 inqbuf->atap_model[1] == 'E') ||
313 (inqbuf->atap_model[0] == 'F' &&
314 inqbuf->atap_model[1] == 'X')))) {
315 needswap = 1;
316 }
317 #endif
318
319 ata_extract_string(model, sizeof(model),
320 inqbuf->atap_model, sizeof(inqbuf->atap_model), needswap);
321 humanize_number(size, sizeof(size),
322 (uint64_t)dd->dd_secsize * (uint64_t)dd->dd_totsec,
323 "", HN_AUTOSCALE, HN_B | HN_NOSPACE | HN_DECIMAL);
324
325 snprintf(dd->dd_descr, sizeof(dd->dd_descr), "%s (%s, %s)",
326 dd->dd_name, size, model);
327
328 return 1;
329 }
330
331 static void
332 get_descr(struct disk_desc *dd)
333 {
334 char diskpath[MAXPATHLEN];
335 int fd = -1;
336
337 fd = opendisk(dd->dd_name, O_RDONLY, diskpath, sizeof(diskpath), 0);
338 if (fd < 0)
339 goto done;
340
341 dd->dd_descr[0] = '\0';
342
343 /* try ATA */
344 if (get_descr_ata(dd, fd))
345 goto done;
346 /* try SCSI */
347 if (get_descr_scsi(dd, fd))
348 goto done;
349 /* XXX: get description from raid, cgd, vnd... */
350
351 done:
352 if (fd >= 0)
353 close(fd);
354 if (strlen(dd->dd_descr) == 0)
355 strcpy(dd->dd_descr, dd->dd_name);
356 }
357
358 /* disknames - contains device names without partition letters
359 * cdrom_devices - contains devices including partition letters
360 * returns the first entry in hw.disknames matching a cdrom_device, or
361 * first entry on error or no match
362 */
363 const char *
364 get_default_cdrom(void)
365 {
366 static const char *cdrom_devices[] = { CD_NAMES, 0};
367 static const char mib_name[] = "hw.disknames";
368 size_t len;
369 char *disknames;
370 char *last;
371 char *name;
372 const char **arg;
373 const char *cd_dev;
374
375 /* On error just use first entry in cdrom_devices */
376 if (sysctlbyname(mib_name, NULL, &len, NULL, 0) == -1)
377 return cdrom_devices[0];
378 if ((disknames = malloc(len + 2)) == 0) /* skip on malloc fail */
379 return cdrom_devices[0];
380
381 (void)sysctlbyname(mib_name, disknames, &len, NULL, 0);
382 for ((name = strtok_r(disknames, " ", &last)); name;
383 (name = strtok_r(NULL, " ", &last))) {
384 for (arg = cdrom_devices; *arg; ++arg) {
385 cd_dev = *arg;
386 /* skip unit and partition */
387 if (strncmp(cd_dev, name, strlen(cd_dev) - 2) != 0)
388 continue;
389 if (name != disknames)
390 strcpy(disknames, name);
391 strcat(disknames, "a");
392 /* XXX: leaks, but so what? */
393 return disknames;
394 }
395 }
396 free(disknames);
397 return cdrom_devices[0];
398 }
399
400 static int
401 get_disks(struct disk_desc *dd)
402 {
403 const char **xd;
404 char *cp;
405 struct disklabel l;
406 int i;
407 int numdisks;
408
409 /* initialize */
410 numdisks = 0;
411
412 for (xd = disk_names; *xd != NULL; xd++) {
413 for (i = 0; i < MAX_DISKS; i++) {
414 strlcpy(dd->dd_name, *xd, sizeof dd->dd_name - 2);
415 cp = strchr(dd->dd_name, ':');
416 if (cp != NULL)
417 dd->dd_no_mbr = !strcmp(cp, ":no_mbr");
418 else {
419 dd->dd_no_mbr = 0;
420 cp = strchr(dd->dd_name, 0);
421 }
422
423 snprintf(cp, 2 + 1, "%d", i);
424 if (!get_geom(dd->dd_name, &l)) {
425 if (errno == ENOENT)
426 break;
427 continue;
428 }
429
430 /*
431 * Exclude a disk mounted as root partition,
432 * in case of install-image on a USB memstick.
433 */
434 if (is_active_rootpart(dd->dd_name, 0))
435 continue;
436
437 dd->dd_cyl = l.d_ncylinders;
438 dd->dd_head = l.d_ntracks;
439 dd->dd_sec = l.d_nsectors;
440 dd->dd_secsize = l.d_secsize;
441 dd->dd_totsec = l.d_secperunit;
442 get_descr(dd);
443 dd++;
444 numdisks++;
445 if (numdisks >= MAX_DISKS)
446 return numdisks;
447 }
448 }
449 return numdisks;
450 }
451
452 int
453 find_disks(const char *doingwhat)
454 {
455 struct disk_desc disks[MAX_DISKS];
456 menu_ent dsk_menu[nelem(disks) + 1]; // + 1 for extended partitioning entry
457 struct disk_desc *disk;
458 int i, already_found;
459 int numdisks, selected_disk = -1;
460 int menu_no;
461 pm_devs_t *pm_i, *pm_last = NULL;
462
463 /* Find disks. */
464 numdisks = get_disks(disks);
465
466 /* need a redraw here, kernel messages hose everything */
467 touchwin(stdscr);
468 refresh();
469 /* Kill typeahead, it won't be what the user had in mind */
470 fpurge(stdin);
471
472 /* partman_go: <0 - we want to see menu with extended partitioning
473 ==0 - we want to see simple select disk menu
474 >0 - we do not want to see any menus, just detect all disks */
475 if (partman_go <= 0) {
476 if (numdisks == 0) {
477 /* No disks found! */
478 msg_display(MSG_nodisk);
479 process_menu(MENU_ok, NULL);
480 /*endwin();*/
481 return -1;
482 } else {
483 /* One or more disks found! */
484 for (i = 0; i < numdisks; i++) {
485 dsk_menu[i].opt_name = disks[i].dd_descr;
486 dsk_menu[i].opt_menu = OPT_NOMENU;
487 dsk_menu[i].opt_flags = OPT_EXIT;
488 dsk_menu[i].opt_action = set_menu_select;
489 }
490 if (partman_go < 0) {
491 dsk_menu[i].opt_name = MSG_partman;
492 dsk_menu[i].opt_menu = OPT_NOMENU;
493 dsk_menu[i].opt_flags = OPT_EXIT;
494 dsk_menu[i].opt_action = set_menu_select;
495 }
496 menu_no = new_menu(MSG_Available_disks,
497 dsk_menu, numdisks + ((partman_go<0)?1:0), -1, 4, 0, 0, MC_SCROLL,
498 NULL, NULL, NULL, NULL, NULL);
499 if (menu_no == -1)
500 return -1;
501 msg_display(MSG_ask_disk, doingwhat);
502 process_menu(menu_no, &selected_disk);
503 free_menu(menu_no);
504 }
505 if (partman_go < 0 && selected_disk == numdisks) {
506 partman_go = 1;
507 return -2;
508 } else
509 partman_go = 0;
510 if (selected_disk < 0 || selected_disk >= numdisks)
511 return -1;
512 }
513
514 /* Fill pm struct with device(s) info */
515 for (i = 0; i < numdisks; i++) {
516 if (! partman_go)
517 disk = disks + selected_disk;
518 else {
519 disk = disks + i;
520 already_found = 0;
521 SLIST_FOREACH(pm_i, &pm_head, l) {
522 pm_last = pm_i;
523 if (!already_found &&
524 strcmp(pm_i->diskdev, disk->dd_name) == 0) {
525 pm_i->found = 1;
526 break;
527 }
528 }
529 if (pm_i != NULL && pm_i->found)
530 /* We already added this device, skipping */
531 continue;
532 }
533 pm = pm_new;
534 pm->found = 1;
535 pm->bootable = 0;
536 pm->pi.menu_no = -1;
537 pm->disktype = "unknown";
538 pm->doessf = "";
539 strlcpy(pm->diskdev, disk->dd_name, sizeof pm->diskdev);
540 strlcpy(pm->diskdev_descr, disk->dd_descr, sizeof pm->diskdev_descr);
541 /* Use as a default disk if the user has the sets on a local disk */
542 strlcpy(localfs_dev, disk->dd_name, sizeof localfs_dev);
543
544 pm->gpt = is_gpt(pm->diskdev);
545 pm->no_mbr = disk->dd_no_mbr || pm->gpt;
546 pm->sectorsize = disk->dd_secsize;
547 pm->dlcyl = disk->dd_cyl;
548 pm->dlhead = disk->dd_head;
549 pm->dlsec = disk->dd_sec;
550 pm->dlsize = disk->dd_totsec;
551 if (pm->dlsize == 0)
552 pm->dlsize = disk->dd_cyl * disk->dd_head * disk->dd_sec;
553 if (pm->dlsize > UINT32_MAX && ! partman_go) {
554 if (logfp)
555 fprintf(logfp, "Cannot process disk %s: too big size (%d)\n",
556 pm->diskdev, (int)pm->dlsize);
557 msg_display(MSG_toobigdisklabel);
558 process_menu(MENU_ok, NULL);
559 return -1;
560 }
561 pm->dlcylsize = pm->dlhead * pm->dlsec;
562
563 label_read();
564 if (partman_go) {
565 pm_getrefdev(pm_new);
566 if (SLIST_EMPTY(&pm_head) || pm_last == NULL)
567 SLIST_INSERT_HEAD(&pm_head, pm_new, l);
568 else
569 SLIST_INSERT_AFTER(pm_last, pm_new, l);
570 pm_new = malloc(sizeof (pm_devs_t));
571 memset(pm_new, 0, sizeof *pm_new);
572 } else
573 /* We is not in partman and do not want to process all devices, exit */
574 break;
575 }
576
577 return numdisks;
578 }
579
580 static bool
581 have_gpt_binary(void)
582 {
583 static bool did_test = false;
584 static bool have_gpt;
585
586 if (!did_test) {
587 have_gpt = binary_available("gpt");
588 did_test = true;
589 }
590
591 return have_gpt;
592 }
593
594 void
595 label_read(void)
596 {
597 /* Get existing/default label */
598 memset(&pm->oldlabel, 0, sizeof pm->oldlabel);
599 if (!have_gpt_binary() || !pm->gpt)
600 incorelabel(pm->diskdev, pm->oldlabel);
601 else
602 incoregpt(pm, pm->oldlabel);
603 /* Set 'target' label to current label in case we don't change it */
604 memcpy(&pm->bsdlabel, &pm->oldlabel, sizeof pm->bsdlabel);
605 #ifndef NO_DISKLABEL
606 if (! pm->gpt)
607 savenewlabel(pm->oldlabel, getmaxpartitions());
608 #endif
609 }
610
611 void
612 fmt_fspart(menudesc *m, int ptn, void *arg)
613 {
614 unsigned int poffset, psize, pend;
615 const char *desc;
616 static const char *Yes;
617 partinfo *p = pm->bsdlabel + ptn;
618
619 if (Yes == NULL)
620 Yes = msg_string(MSG_Yes);
621
622 poffset = p->pi_offset / sizemult;
623 psize = p->pi_size / sizemult;
624 if (psize == 0)
625 pend = 0;
626 else
627 pend = (p->pi_offset + p->pi_size) / sizemult - 1;
628
629 if (p->pi_fstype == FS_BSDFFS)
630 if (p->pi_flags & PIF_FFSv2)
631 desc = "FFSv2";
632 else
633 desc = "FFSv1";
634 else
635 desc = getfslabelname(p->pi_fstype);
636
637 #ifdef PART_BOOT
638 if (ptn == PART_BOOT)
639 desc = msg_string(MSG_Boot_partition_cant_change);
640 #endif
641 if (ptn == getrawpartition())
642 desc = msg_string(MSG_Whole_disk_cant_change);
643 else {
644 if (ptn == PART_C)
645 desc = msg_string(MSG_NetBSD_partition_cant_change);
646 }
647
648 wprintw(m->mw, msg_string(MSG_fspart_row),
649 poffset, pend, psize, desc,
650 p->pi_flags & PIF_NEWFS ? Yes : "",
651 p->pi_flags & PIF_MOUNT ? Yes : "",
652 p->pi_mount);
653 }
654
655 /*
656 * Label a disk using an MD-specific string DISKLABEL_CMD for
657 * to invoke disklabel.
658 * if MD code does not define DISKLABEL_CMD, this is a no-op.
659 *
660 * i386 port uses "/sbin/disklabel -w -r", just like i386
661 * miniroot scripts, though this may leave a bogus incore label.
662 *
663 * Sun ports should use DISKLABEL_CMD "/sbin/disklabel -w"
664 * to get incore to ondisk inode translation for the Sun proms.
665 */
666 int
667 write_disklabel (void)
668 {
669
670 #ifdef DISKLABEL_CMD
671 /* disklabel the disk */
672 return run_program(RUN_DISPLAY, "%s -f /tmp/disktab %s '%s'",
673 DISKLABEL_CMD, pm->diskdev, pm->bsddiskname);
674 #else
675 return 0;
676 #endif
677 }
678
679
680 static int
681 ptn_sort(const void *a, const void *b)
682 {
683 return strcmp(pm->bsdlabel[*(const int *)a].pi_mount,
684 pm->bsdlabel[*(const int *)b].pi_mount);
685 }
686
687 int
688 make_filesystems(void)
689 {
690 unsigned int i;
691 int ptn;
692 int ptn_order[nelem(pm->bsdlabel)];
693 int error = 0;
694 unsigned int maxpart = getmaxpartitions();
695 char *newfs = NULL, *dev = NULL, *devdev = NULL;
696 partinfo *lbl;
697
698 if (maxpart > nelem(pm->bsdlabel))
699 maxpart = nelem(pm->bsdlabel);
700
701 /* Making new file systems and mounting them */
702
703 /* sort to ensure /usr/local is mounted after /usr (etc) */
704 for (i = 0; i < maxpart; i++)
705 ptn_order[i] = i;
706 qsort(ptn_order, maxpart, sizeof ptn_order[0], ptn_sort);
707
708 for (i = 0; i < maxpart; i++) {
709 /*
710 * newfs and mount. For now, process only BSD filesystems.
711 * but if this is the mounted-on root, has no mount
712 * point defined, or is marked preserve, don't touch it!
713 */
714 ptn = ptn_order[i];
715 lbl = pm->bsdlabel + ptn;
716
717 if (is_active_rootpart(pm->diskdev, ptn))
718 continue;
719
720 if (*lbl->pi_mount == 0)
721 /* No mount point */
722 continue;
723
724 if (pm->isspecial) {
725 asprintf(&dev, "%s", pm->diskdev);
726 ptn = 0 - 'a';
727 } else {
728 asprintf(&dev, "%s%c", pm->diskdev, 'a' + ptn);
729 }
730 if (dev == NULL)
731 return (ENOMEM);
732 asprintf(&devdev, "/dev/%s", dev);
733 if (devdev == NULL)
734 return (ENOMEM);
735
736 newfs = NULL;
737 lbl->mnt_opts = NULL;
738 lbl->fsname = NULL;
739 switch (lbl->pi_fstype) {
740 case FS_APPLEUFS:
741 asprintf(&newfs, "/sbin/newfs %s%.0d",
742 lbl->pi_isize != 0 ? "-i" : "", lbl->pi_isize);
743 lbl->mnt_opts = "-tffs -o async";
744 lbl->fsname = "ffs";
745 break;
746 case FS_BSDFFS:
747 asprintf(&newfs,
748 "/sbin/newfs -V2 -O %d -b %d -f %d%s%.0d",
749 lbl->pi_flags & PIF_FFSv2 ? 2 : 1,
750 lbl->pi_fsize * lbl->pi_frag, lbl->pi_fsize,
751 lbl->pi_isize != 0 ? " -i " : "", lbl->pi_isize);
752 if (lbl->pi_flags & PIF_LOG)
753 lbl->mnt_opts = "-tffs -o log";
754 else
755 lbl->mnt_opts = "-tffs -o async";
756 lbl->fsname = "ffs";
757 break;
758 case FS_BSDLFS:
759 asprintf(&newfs, "/sbin/newfs_lfs -b %d",
760 lbl->pi_fsize * lbl->pi_frag);
761 lbl->mnt_opts = "-tlfs";
762 lbl->fsname = "lfs";
763 break;
764 case FS_MSDOS:
765 #ifdef USE_NEWFS_MSDOS
766 asprintf(&newfs, "/sbin/newfs_msdos");
767 #endif
768 lbl->mnt_opts = "-tmsdos";
769 lbl->fsname = "msdos";
770 break;
771 #ifdef USE_SYSVBFS
772 case FS_SYSVBFS:
773 asprintf(&newfs, "/sbin/newfs_sysvbfs");
774 lbl->mnt_opts = "-tsysvbfs";
775 lbl->fsname = "sysvbfs";
776 break;
777 #endif
778 #ifdef USE_EXT2FS
779 case FS_EX2FS:
780 asprintf(&newfs, "/sbin/newfs_ext2fs");
781 lbl->mnt_opts = "-text2fs";
782 lbl->fsname = "ext2fs";
783 break;
784 #endif
785 }
786 if (lbl->pi_flags & PIF_NEWFS && newfs != NULL) {
787 #ifdef USE_NEWFS_MSDOS
788 if (lbl->pi_fstype == FS_MSDOS) {
789 /* newfs only if mount fails */
790 if (run_program(RUN_SILENT | RUN_ERROR_OK,
791 "mount -rt msdos /dev/%s /mnt2", dev) != 0)
792 error = run_program(
793 RUN_DISPLAY | RUN_PROGRESS,
794 "%s /dev/r%s",
795 newfs, dev);
796 else {
797 run_program(RUN_SILENT | RUN_ERROR_OK,
798 "umount /mnt2");
799 error = 0;
800 }
801 } else
802 #endif
803 error = run_program(RUN_DISPLAY | RUN_PROGRESS,
804 "%s /dev/r%s", newfs, dev);
805 } else {
806 /* We'd better check it isn't dirty */
807 error = fsck_preen(pm->diskdev, ptn, lbl->fsname);
808 }
809 free(newfs);
810 if (error != 0)
811 return error;
812
813 lbl->pi_flags ^= PIF_NEWFS;
814 md_pre_mount();
815
816 if (partman_go == 0 && lbl->pi_flags & PIF_MOUNT &&
817 lbl->mnt_opts != NULL) {
818 make_target_dir(lbl->pi_mount);
819 error = target_mount_do(lbl->mnt_opts, devdev, lbl->pi_mount);
820 if (error) {
821 msg_display(MSG_mountfail, dev, ' ', lbl->pi_mount);
822 process_menu(MENU_ok, NULL);
823 return error;
824 }
825 }
826 free(devdev);
827 free(dev);
828 }
829 return 0;
830 }
831
832 int
833 make_fstab(void)
834 {
835 FILE *f;
836 int i, swap_dev = -1;
837 const char *dump_dev;
838 char *dev = NULL;
839 pm_devs_t *pm_i;
840 #ifndef HAVE_TMPFS
841 pm_devs_t *pm_with_swap = NULL;
842 #endif
843
844 /* Create the fstab. */
845 make_target_dir("/etc");
846 f = target_fopen("/etc/fstab", "w");
847 scripting_fprintf(NULL, "cat <<EOF >%s/etc/fstab\n", target_prefix());
848
849 if (logfp)
850 (void)fprintf(logfp,
851 "Making %s/etc/fstab (%s).\n", target_prefix(), pm->diskdev);
852
853 if (f == NULL) {
854 msg_display(MSG_createfstab);
855 if (logfp)
856 (void)fprintf(logfp, "Failed to make /etc/fstab!\n");
857 process_menu(MENU_ok, NULL);
858 #ifndef DEBUG
859 return 1;
860 #else
861 f = stdout;
862 #endif
863 }
864
865 scripting_fprintf(f, "# NetBSD %s/etc/fstab\n# See /usr/share/examples/"
866 "fstab/ for more examples.\n", target_prefix());
867 if (! partman_go) {
868 /* We want to process only one disk... */
869 pm_i = pm;
870 goto onlyonediskinfstab;
871 }
872 SLIST_FOREACH(pm_i, &pm_head, l) {
873 onlyonediskinfstab:
874 for (i = 0; i < getmaxpartitions(); i++) {
875 const char *s = "";
876 const char *mp = pm_i->bsdlabel[i].pi_mount;
877 const char *fstype = "ffs";
878 int fsck_pass = 0, dump_freq = 0;
879
880 if (dev != NULL)
881 free(dev);
882 if (pm_i->isspecial)
883 asprintf(&dev, "%s", pm_i->diskdev);
884 else
885 asprintf(&dev, "%s%c", pm_i->diskdev, 'a' + i);
886 if (dev == NULL)
887 return (ENOMEM);
888
889 if (!*mp) {
890 /*
891 * No mount point specified, comment out line and
892 * use /mnt as a placeholder for the mount point.
893 */
894 s = "# ";
895 mp = "/mnt";
896 }
897
898 switch (pm_i->bsdlabel[i].pi_fstype) {
899 case FS_UNUSED:
900 continue;
901 case FS_BSDLFS:
902 /* If there is no LFS, just comment it out. */
903 if (!check_lfs_progs())
904 s = "# ";
905 fstype = "lfs";
906 /* FALLTHROUGH */
907 case FS_BSDFFS:
908 fsck_pass = (strcmp(mp, "/") == 0) ? 1 : 2;
909 dump_freq = 1;
910 break;
911 case FS_MSDOS:
912 fstype = "msdos";
913 break;
914 case FS_SWAP:
915 if (pm_i->isspecial)
916 continue;
917 if (swap_dev == -1) {
918 swap_dev = i;
919 dump_dev = ",dp";
920 #ifndef HAVE_TMPFS
921 pm_with_swap = pm_i;
922 #endif
923 } else {
924 dump_dev = "";
925 }
926 scripting_fprintf(f, "/dev/%s\t\tnone\tswap\tsw%s\t\t 0 0\n",
927 dev, dump_dev);
928 continue;
929 #ifdef USE_SYSVBFS
930 case FS_SYSVBFS:
931 fstype = "sysvbfs";
932 make_target_dir("/stand");
933 break;
934 #endif
935 default:
936 fstype = "???";
937 s = "# ";
938 break;
939 }
940 /* The code that remounts root rw doesn't check the partition */
941 if (strcmp(mp, "/") == 0 && !(pm_i->bsdlabel[i].pi_flags & PIF_MOUNT))
942 s = "# ";
943
944 scripting_fprintf(f,
945 "%s/dev/%s\t\t%s\t%s\trw%s%s%s%s%s%s%s%s\t\t %d %d\n",
946 s, dev, mp, fstype,
947 pm_i->bsdlabel[i].pi_flags & PIF_LOG ? ",log" : "",
948 pm_i->bsdlabel[i].pi_flags & PIF_MOUNT ? "" : ",noauto",
949 pm_i->bsdlabel[i].pi_flags & PIF_ASYNC ? ",async" : "",
950 pm_i->bsdlabel[i].pi_flags & PIF_NOATIME ? ",noatime" : "",
951 pm_i->bsdlabel[i].pi_flags & PIF_NODEV ? ",nodev" : "",
952 pm_i->bsdlabel[i].pi_flags & PIF_NODEVMTIME ? ",nodevmtime" : "",
953 pm_i->bsdlabel[i].pi_flags & PIF_NOEXEC ? ",noexec" : "",
954 pm_i->bsdlabel[i].pi_flags & PIF_NOSUID ? ",nosuid" : "",
955 dump_freq, fsck_pass);
956 if (pm_i->isspecial)
957 /* Special device (such as dk*) have only one partition */
958 break;
959 }
960 /* Simple install, only one disk */
961 if (!partman_go)
962 break;
963 }
964 if (tmp_ramdisk_size != 0) {
965 #ifdef HAVE_TMPFS
966 scripting_fprintf(f, "tmpfs\t\t/tmp\ttmpfs\trw,-m=1777,-s=%"
967 PRIi64 "\n",
968 tmp_ramdisk_size * 512);
969 #else
970 if (swap_dev != -1 && pm_with_swap != NULL)
971 scripting_fprintf(f, "/dev/%s%c\t\t/tmp\tmfs\trw,-s=%"
972 PRIi64 "\n",
973 pm_with_swap->diskdev, 'a' + swap_dev, tmp_ramdisk_size);
974 else
975 scripting_fprintf(f, "swap\t\t/tmp\tmfs\trw,-s=%"
976 PRIi64 "\n",
977 tmp_ramdisk_size);
978 #endif
979 }
980
981 /* Add /kern, /proc and /dev/pts to fstab and make mountpoint. */
982 scripting_fprintf(f, "kernfs\t\t/kern\tkernfs\trw\n");
983 scripting_fprintf(f, "ptyfs\t\t/dev/pts\tptyfs\trw\n");
984 scripting_fprintf(f, "procfs\t\t/proc\tprocfs\trw\n");
985 scripting_fprintf(f, "/dev/%s\t\t/cdrom\tcd9660\tro,noauto\n",
986 get_default_cdrom());
987 scripting_fprintf(f, "%stmpfs\t\t/var/shm\ttmpfs\trw,-m1777,-sram%%25\n",
988 tmpfs_on_var_shm() ? "" : "#");
989 make_target_dir("/kern");
990 make_target_dir("/proc");
991 make_target_dir("/dev/pts");
992 make_target_dir("/cdrom");
993 make_target_dir("/var/shm");
994
995 scripting_fprintf(NULL, "EOF\n");
996
997 if (dev != NULL)
998 free(dev);
999 fclose(f);
1000 fflush(NULL);
1001 return 0;
1002 }
1003
1004
1005
1006 static int
1007 /*ARGSUSED*/
1008 foundffs(struct data *list, size_t num)
1009 {
1010 int error;
1011
1012 if (num < 2 || strcmp(list[1].u.s_val, "/") == 0 ||
1013 strstr(list[2].u.s_val, "noauto") != NULL)
1014 return 0;
1015
1016 error = fsck_preen(list[0].u.s_val, ' '-'a', "ffs");
1017 if (error != 0)
1018 return error;
1019
1020 error = target_mount("", list[0].u.s_val, ' '-'a', list[1].u.s_val);
1021 if (error != 0) {
1022 msg_display(MSG_mount_failed, list[0].u.s_val);
1023 process_menu(MENU_noyes, NULL);
1024 if (!yesno)
1025 return error;
1026 }
1027 return 0;
1028 }
1029
1030 #ifdef USE_SYSVBFS
1031 static int
1032 /*ARGSUSED*/
1033 foundsysvbfs(struct data *list, size_t num)
1034 {
1035 int error;
1036
1037 if (num < 2 || strcmp(list[1].u.s_val, "/") == 0 ||
1038 strstr(list[2].u.s_val, "noauto") != NULL)
1039 return 0;
1040
1041 error = target_mount("", list[0].u.s_val, ' '-'a', list[1].u.s_val);
1042 if (error != 0)
1043 return error;
1044 return 0;
1045 }
1046 #endif
1047
1048 /*
1049 * Do an fsck. On failure, inform the user by showing a warning
1050 * message and doing menu_ok() before proceeding.
1051 * Returns 0 on success, or nonzero return code from fsck() on failure.
1052 */
1053 static int
1054 fsck_preen(const char *disk, int ptn, const char *fsname)
1055 {
1056 char *prog;
1057 int error;
1058
1059 ptn = (ptn < 0)? 0 : 'a' + ptn;
1060 if (fsname == NULL)
1061 return 0;
1062 /* first, check if fsck program exists, if not, assume ok */
1063 asprintf(&prog, "/sbin/fsck_%s", fsname);
1064 if (prog == NULL)
1065 return 0;
1066 if (access(prog, X_OK) != 0)
1067 return 0;
1068 if (!strcmp(fsname,"ffs"))
1069 fixsb(prog, disk, ptn);
1070 error = run_program(0, "%s -p -q /dev/r%s%c", prog, disk, ptn);
1071 free(prog);
1072 if (error != 0) {
1073 msg_display(MSG_badfs, disk, ptn, error);
1074 process_menu(MENU_noyes, NULL);
1075 if (yesno)
1076 error = 0;
1077 /* XXX at this point maybe we should run a full fsck? */
1078 }
1079 return error;
1080 }
1081
1082 /* This performs the same function as the etc/rc.d/fixsb script
1083 * which attempts to correct problems with ffs1 filesystems
1084 * which may have been introduced by booting a netbsd-current kernel
1085 * from between April of 2003 and January 2004. For more information
1086 * This script was developed as a response to NetBSD pr install/25138
1087 * Additional prs regarding the original issue include:
1088 * bin/17910 kern/21283 kern/21404 port-macppc/23925 port-macppc/23926
1089 */
1090 static void
1091 fixsb(const char *prog, const char *disk, char ptn)
1092 {
1093 int fd;
1094 int rval;
1095 union {
1096 struct fs fs;
1097 char buf[SBLOCKSIZE];
1098 } sblk;
1099 struct fs *fs = &sblk.fs;
1100
1101 snprintf(sblk.buf, sizeof(sblk.buf), "/dev/r%s%c",
1102 disk, ptn == ' ' ? 0 : ptn);
1103 fd = open(sblk.buf, O_RDONLY);
1104 if (fd == -1)
1105 return;
1106
1107 /* Read ffsv1 main superblock */
1108 rval = pread(fd, sblk.buf, sizeof sblk.buf, SBLOCK_UFS1);
1109 close(fd);
1110 if (rval != sizeof sblk.buf)
1111 return;
1112
1113 if (fs->fs_magic != FS_UFS1_MAGIC &&
1114 fs->fs_magic != FS_UFS1_MAGIC_SWAPPED)
1115 /* Not FFSv1 */
1116 return;
1117 if (fs->fs_old_flags & FS_FLAGS_UPDATED)
1118 /* properly updated fslevel 4 */
1119 return;
1120 if (fs->fs_bsize != fs->fs_maxbsize)
1121 /* not messed up */
1122 return;
1123
1124 /*
1125 * OK we have a munged fs, first 'upgrade' to fslevel 4,
1126 * We specify -b16 in order to stop fsck bleating that the
1127 * sb doesn't match the first alternate.
1128 */
1129 run_program(RUN_DISPLAY | RUN_PROGRESS,
1130 "%s -p -b 16 -c 4 /dev/r%s%c", prog, disk, ptn);
1131 /* Then downgrade to fslevel 3 */
1132 run_program(RUN_DISPLAY | RUN_PROGRESS,
1133 "%s -p -c 3 /dev/r%s%c", prog, disk, ptn);
1134 }
1135
1136 /*
1137 * fsck and mount the root partition.
1138 */
1139 static int
1140 mount_root(void)
1141 {
1142 int error;
1143 int ptn = (pm->isspecial)? 0 - 'a' : pm->rootpart;
1144
1145 error = fsck_preen(pm->diskdev, ptn, "ffs");
1146 if (error != 0)
1147 return error;
1148
1149 md_pre_mount();
1150
1151 /* Mount /dev/<diskdev>a on target's "".
1152 * If we pass "" as mount-on, Prefixing will DTRT.
1153 * for now, use no options.
1154 * XXX consider -o remount in case target root is
1155 * current root, still readonly from single-user?
1156 */
1157 return target_mount("", pm->diskdev, ptn, "");
1158 }
1159
1160 /* Get information on the file systems mounted from the root filesystem.
1161 * Offer to convert them into 4.4BSD inodes if they are not 4.4BSD
1162 * inodes. Fsck them. Mount them.
1163 */
1164
1165 int
1166 mount_disks(void)
1167 {
1168 char *fstab;
1169 int fstabsize;
1170 int error;
1171
1172 static struct lookfor fstabbuf[] = {
1173 {"/dev/", "/dev/%s %s ffs %s", "c", NULL, 0, 0, foundffs},
1174 {"/dev/", "/dev/%s %s ufs %s", "c", NULL, 0, 0, foundffs},
1175 #ifdef USE_SYSVBFS
1176 {"/dev/", "/dev/%s %s sysvbfs %s", "c", NULL, 0, 0,
1177 foundsysvbfs},
1178 #endif
1179 };
1180 static size_t numfstabbuf = sizeof(fstabbuf) / sizeof(struct lookfor);
1181
1182 /* First the root device. */
1183 if (target_already_root())
1184 /* avoid needing to call target_already_root() again */
1185 targetroot_mnt[0] = 0;
1186 else {
1187 error = mount_root();
1188 if (error != 0 && error != EBUSY)
1189 return -1;
1190 }
1191
1192 /* Check the target /etc/fstab exists before trying to parse it. */
1193 if (target_dir_exists_p("/etc") == 0 ||
1194 target_file_exists_p("/etc/fstab") == 0) {
1195 msg_display(MSG_noetcfstab, pm->diskdev);
1196 process_menu(MENU_ok, NULL);
1197 return -1;
1198 }
1199
1200
1201 /* Get fstab entries from the target-root /etc/fstab. */
1202 fstabsize = target_collect_file(T_FILE, &fstab, "/etc/fstab");
1203 if (fstabsize < 0) {
1204 /* error ! */
1205 msg_display(MSG_badetcfstab, pm->diskdev);
1206 process_menu(MENU_ok, NULL);
1207 return -2;
1208 }
1209 error = walk(fstab, (size_t)fstabsize, fstabbuf, numfstabbuf);
1210 free(fstab);
1211
1212 return error;
1213 }
1214
1215 int
1216 set_swap_if_low_ram(const char *disk, partinfo *pp) {
1217 if (get_ramsize() <= 32)
1218 return set_swap(disk, pp);
1219 return 0;
1220 }
1221
1222 int
1223 set_swap(const char *disk, partinfo *pp)
1224 {
1225 int i;
1226 char *cp;
1227 int rval;
1228
1229 if (pp == NULL)
1230 pp = pm->oldlabel;
1231
1232 for (i = 0; i < MAXPARTITIONS; i++) {
1233 if (pp[i].pi_fstype != FS_SWAP)
1234 continue;
1235 asprintf(&cp, "/dev/%s%c", disk, 'a' + i);
1236 rval = swapctl(SWAP_ON, cp, 0);
1237 free(cp);
1238 if (rval != 0)
1239 return -1;
1240 }
1241
1242 return 0;
1243 }
1244
1245 int
1246 check_swap(const char *disk, int remove_swap)
1247 {
1248 struct swapent *swap;
1249 char *cp;
1250 int nswap;
1251 int l;
1252 int rval = 0;
1253
1254 nswap = swapctl(SWAP_NSWAP, 0, 0);
1255 if (nswap <= 0)
1256 return 0;
1257
1258 swap = malloc(nswap * sizeof *swap);
1259 if (swap == NULL)
1260 return -1;
1261
1262 nswap = swapctl(SWAP_STATS, swap, nswap);
1263 if (nswap < 0)
1264 goto bad_swap;
1265
1266 l = strlen(disk);
1267 while (--nswap >= 0) {
1268 /* Should we check the se_dev or se_path? */
1269 cp = swap[nswap].se_path;
1270 if (memcmp(cp, "/dev/", 5) != 0)
1271 continue;
1272 if (memcmp(cp + 5, disk, l) != 0)
1273 continue;
1274 if (!isalpha(*(unsigned char *)(cp + 5 + l)))
1275 continue;
1276 if (cp[5 + l + 1] != 0)
1277 continue;
1278 /* ok path looks like it is for this device */
1279 if (!remove_swap) {
1280 /* count active swap areas */
1281 rval++;
1282 continue;
1283 }
1284 if (swapctl(SWAP_OFF, cp, 0) == -1)
1285 rval = -1;
1286 }
1287
1288 done:
1289 free(swap);
1290 return rval;
1291
1292 bad_swap:
1293 rval = -1;
1294 goto done;
1295 }
1296
1297 #ifdef HAVE_BOOTXX_xFS
1298 char *
1299 bootxx_name(void)
1300 {
1301 int fstype;
1302 const char *bootxxname;
1303 char *bootxx;
1304
1305 /* check we have boot code for the root partition type */
1306 fstype = pm->bsdlabel[pm->rootpart].pi_fstype;
1307 switch (fstype) {
1308 #if defined(BOOTXX_FFSV1) || defined(BOOTXX_FFSV2)
1309 case FS_BSDFFS:
1310 if (pm->bsdlabel[pm->rootpart].pi_flags & PIF_FFSv2) {
1311 #ifdef BOOTXX_FFSV2
1312 bootxxname = BOOTXX_FFSV2;
1313 #else
1314 bootxxname = NULL;
1315 #endif
1316 } else {
1317 #ifdef BOOTXX_FFSV1
1318 bootxxname = BOOTXX_FFSV1;
1319 #else
1320 bootxxname = NULL;
1321 #endif
1322 }
1323 break;
1324 #endif
1325 #ifdef BOOTXX_LFS
1326 case FS_BSDLFS:
1327 bootxxname = BOOTXX_LFS;
1328 break;
1329 #endif
1330 default:
1331 bootxxname = NULL;
1332 break;
1333 }
1334
1335 if (bootxxname == NULL)
1336 return NULL;
1337
1338 asprintf(&bootxx, "%s/%s", BOOTXXDIR, bootxxname);
1339 return bootxx;
1340 }
1341 #endif
1342
1343 static int
1344 get_fsid_by_gptuuid(const char *str)
1345 {
1346 int i;
1347 uuid_t uuid;
1348 uint32_t status;
1349
1350 uuid_from_string(str, &uuid, &status);
1351 if (status == uuid_s_ok) {
1352 for (i = 0; gpt_filesystems[i].id > 0; i++)
1353 if (uuid_equal(&uuid, &(gpt_filesystems[i].uuid), NULL))
1354 return gpt_filesystems[i].id;
1355 }
1356 return FS_OTHER;
1357 }
1358
1359 const char *
1360 get_gptfs_by_id(int filesystem)
1361 {
1362 int i;
1363 for (i = 0; gpt_filesystems[i].id > 0; i++)
1364 if (filesystem == gpt_filesystems[i].id)
1365 return gpt_filesystems[i].name;
1366 return NULL;
1367 }
1368
1369 /* from dkctl.c */
1370 static int
1371 get_dkwedges_sort(const void *a, const void *b)
1372 {
1373 const struct dkwedge_info *dkwa = a, *dkwb = b;
1374 const daddr_t oa = dkwa->dkw_offset, ob = dkwb->dkw_offset;
1375 return (oa < ob) ? -1 : (oa > ob) ? 1 : 0;
1376 }
1377
1378 int
1379 get_dkwedges(struct dkwedge_info **dkw, const char *diskdev)
1380 {
1381 int fd;
1382 char buf[STRSIZE];
1383 size_t bufsize;
1384 struct dkwedge_list dkwl;
1385
1386 *dkw = NULL;
1387 dkwl.dkwl_buf = *dkw;
1388 dkwl.dkwl_bufsize = 0;
1389 fd = opendisk(diskdev, O_RDONLY, buf, STRSIZE, 0);
1390 if (fd < 0)
1391 return -1;
1392
1393 for (;;) {
1394 if (ioctl(fd, DIOCLWEDGES, &dkwl) == -1)
1395 return -2;
1396 if (dkwl.dkwl_nwedges == dkwl.dkwl_ncopied)
1397 break;
1398 bufsize = dkwl.dkwl_nwedges * sizeof(**dkw);
1399 if (dkwl.dkwl_bufsize < bufsize) {
1400 *dkw = realloc(dkwl.dkwl_buf, bufsize);
1401 if (*dkw == NULL)
1402 return -3;
1403 dkwl.dkwl_buf = *dkw;
1404 dkwl.dkwl_bufsize = bufsize;
1405 }
1406 }
1407
1408 if (dkwl.dkwl_nwedges > 0 && *dkw != NULL)
1409 qsort(*dkw, dkwl.dkwl_nwedges, sizeof(**dkw), get_dkwedges_sort);
1410
1411 close(fd);
1412 return dkwl.dkwl_nwedges;
1413 }
1414
1415 /* XXX: rewrite */
1416 static int
1417 incoregpt(pm_devs_t *pm_cur, partinfo *lp)
1418 {
1419 int i, num;
1420 unsigned int p_num;
1421 uint64_t p_start, p_size;
1422 char *textbuf, *t, *tt, p_type[STRSIZE];
1423 struct dkwedge_info *dkw;
1424
1425 num = get_dkwedges(&dkw, pm_cur->diskdev);
1426 if (dkw != NULL) {
1427 for (i = 0; i < num && i < MAX_WEDGES; i++)
1428 run_program(RUN_SILENT, "dkctl %s delwedge %s",
1429 pm_cur->diskdev, dkw[i].dkw_devname);
1430 free (dkw);
1431 }
1432
1433 if (collect(T_OUTPUT, &textbuf, "gpt show -u %s 2>/dev/null", pm_cur->diskdev) < 1)
1434 return -1;
1435
1436 (void)strtok(textbuf, "\n"); /* ignore first line */
1437 while ((t = strtok(NULL, "\n")) != NULL) {
1438 i = 0; p_start = 0; p_size = 0; p_num = 0; strcpy(p_type, ""); /* init */
1439 while ((tt = strsep(&t, " \t")) != NULL) {
1440 if (strlen(tt) == 0)
1441 continue;
1442 if (i == 0)
1443 p_start = strtouq(tt, NULL, 10);
1444 if (i == 1)
1445 p_size = strtouq(tt, NULL, 10);
1446 if (i == 2)
1447 p_num = strtouq(tt, NULL, 10);
1448 if (i > 2 || (i == 2 && p_num == 0))
1449 if (
1450 strcmp(tt, "GPT") &&
1451 strcmp(tt, "part") &&
1452 strcmp(tt, "-")
1453 )
1454 strncat(p_type, tt, STRSIZE);
1455 i++;
1456 }
1457 if (p_start == 0 || p_size == 0)
1458 continue;
1459 else if (! strcmp(p_type, "Pritable"))
1460 pm_cur->ptstart = p_start + p_size;
1461 else if (! strcmp(p_type, "Sectable"))
1462 pm_cur->ptsize = p_start - pm_cur->ptstart - 1;
1463 else if (p_num == 0 && strlen(p_type) > 0)
1464 /* Utilitary entry (PMBR, etc) */
1465 continue;
1466 else if (p_num == 0) {
1467 /* Free space */
1468 continue;
1469 } else {
1470 /* Usual partition */
1471 lp[p_num].pi_size = p_size;
1472 lp[p_num].pi_offset = p_start;
1473 lp[p_num].pi_fstype = get_fsid_by_gptuuid(p_type);
1474 }
1475 }
1476 free(textbuf);
1477
1478 return 0;
1479 }
1480
1481 static bool
1482 is_gpt(const char *dev)
1483 {
1484 if (!have_gpt_binary())
1485 return false;
1486
1487 return !run_program(RUN_SILENT | RUN_ERROR_OK,
1488 "sh -c 'gpt show %s |grep -e Pri\\ GPT\\ table'", dev);
1489 }
1490