2 * Copyright (c) 2003 Marcel Moolenaar
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
16 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
17 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
18 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
19 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
20 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
21 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
22 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
24 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27 #include <sys/cdefs.h>
28 __FBSDID("$FreeBSD$");
30 #include <sys/types.h>
31 #include <sys/disklabel.h>
32 #include <sys/diskmbr.h>
47 parse_disk(char *conftxt, const char *name)
51 struct dos_partition *part;
56 disk = (struct disk *)calloc(sizeof *disk, 1);
60 disk->name = strdup(name);
61 p = strsep(&conftxt, " "); /* media size */
62 disk->media_size = strtoimax(p, &q, 0);
66 p = strsep(&conftxt, " "); /* sector size */
67 disk->sector_size = strtoul(p, &q, 0);
71 if (disk->sector_size == 0)
72 disk->sector_size = 512;
74 if (disk->media_size % disk->sector_size)
78 * We need to read the disk to get GPT specific information.
81 snprintf(devname, sizeof(devname), "%s%s", _PATH_DEV, name);
82 fd = open(devname, O_RDONLY);
85 buffer = malloc(2 * disk->sector_size);
90 if (read(fd, buffer, 2 * disk->sector_size) == -1) {
97 gpt = (struct gpt_hdr *)(buffer + disk->sector_size);
98 if (memcmp(gpt->hdr_sig, GPT_HDR_SIG, sizeof(gpt->hdr_sig))) {
100 * No GPT present. Check if the MBR is empty (if present)
101 * or is a PMBR before declaring this disk as empty. If
102 * the MBR isn't empty, bail out. Let's not risk nuking a
105 if (*(u_short *)(buffer + DOSMAGICOFFSET) == DOSMAGIC) {
106 for (i = 0; i < 4; i++) {
107 part = (struct dos_partition *)
108 (buffer + DOSPARTOFF + i * DOSPARTSIZE);
109 if (part->dp_typ != 0 &&
110 part->dp_typ != DOSPTYP_PMBR)
118 disk->gpt_size = 128;
119 disk->lba_start = (disk->gpt_size * sizeof(struct gpt_ent)) /
120 disk->sector_size + 2;
121 disk->lba_end = (disk->media_size / disk->sector_size) -
124 disk->lba_start = gpt->hdr_lba_start;
125 disk->lba_end = gpt->hdr_lba_end;
126 disk->gpt_size = gpt->hdr_entries;
129 Add_Chunk(disk, disk->lba_start, disk->lba_end - disk->lba_start + 1,
130 name, whole, 0, 0, "-");
140 Int_Open_Disk(const char *name, char *conftxt)
144 char *p, *q, *r, *s, *sd;
148 while (p != NULL && *p != 0) {
150 if (strcmp(q, "0") == 0) {
152 if (strcmp(q, "DISK") == 0) {
154 if (strcmp(q, name) == 0)
159 if (p != NULL && *p == '\n')
163 if (p == NULL || *p == 0)
166 conftxt = strchr(p, '\n');
170 disk = parse_disk(p, name);
174 while (conftxt != NULL && *conftxt != 0) {
176 conftxt = strchr(p, '\n');
181 * 1 PART da0p4 34359738368 512
182 * i 4 o 52063912960 ty freebsd-ufs
183 * xs GPT xt 516e7cb6-6ecf-11d6-8ff8-00022d09712b
185 sd = strsep(&p, " "); /* depth */
186 if (strcmp(sd, "0") == 0)
189 q = strsep(&p, " "); /* type */
190 if (strcmp(q, "PART") != 0)
193 chunk.name = strsep(&p, " "); /* name */
195 q = strsep(&p, " "); /* length */
196 i = strtoimax(q, &r, 0);
199 chunk.end = i / disk->sector_size;
201 q = strsep(&p, " "); /* sector size */
208 i = strtoimax(r, &s, 0);
209 if (strcmp(q, "ty") == 0 && *s != '\0') {
210 if (!strcmp(r, "efi"))
212 else if (!strcmp(r, "freebsd")) {
213 chunk.type = freebsd;
214 chunk.subtype = 0xa5;
215 } else if (!strcmp(r, "freebsd-swap")) {
217 chunk.subtype = FS_SWAP;
218 } else if (!strcmp(r, "freebsd-ufs")) {
220 chunk.subtype = FS_BSDFFS;
223 chunk.subtype = FS_OTHER;
227 chunk.offset = i / disk->sector_size;
228 else if (!strcmp(q, "i"))
229 chunk.flags = CHUNK_ITOF(i) |
234 Add_Chunk(disk, chunk.offset, chunk.end, chunk.name,
235 chunk.type, chunk.subtype, chunk.flags, 0);