summaryrefslogtreecommitdiff
path: root/sys/geom/part/g_part_gpt.c
diff options
context:
space:
mode:
authorMarcel Moolenaar <marcel@FreeBSD.org>2007-02-07 18:55:31 +0000
committerMarcel Moolenaar <marcel@FreeBSD.org>2007-02-07 18:55:31 +0000
commit1d3aed33e8fe3530c44df6fb4a6d196684566bf9 (patch)
treee3bae4a31a9a612e220a9627e78c8b5fef9685ab /sys/geom/part/g_part_gpt.c
parente8865caffb1a0ea61715b420fa398672dab8907a (diff)
downloadsrc-test2-1d3aed33e8fe3530c44df6fb4a6d196684566bf9.tar.gz
src-test2-1d3aed33e8fe3530c44df6fb4a6d196684566bf9.zip
Notes
Diffstat (limited to 'sys/geom/part/g_part_gpt.c')
-rw-r--r--sys/geom/part/g_part_gpt.c736
1 files changed, 736 insertions, 0 deletions
diff --git a/sys/geom/part/g_part_gpt.c b/sys/geom/part/g_part_gpt.c
new file mode 100644
index 000000000000..f8370a530f75
--- /dev/null
+++ b/sys/geom/part/g_part_gpt.c
@@ -0,0 +1,736 @@
+/*-
+ * Copyright (c) 2002, 2005, 2006, 2007 Marcel Moolenaar
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
+ * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
+ * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
+ * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
+ * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+ * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
+ * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include <sys/cdefs.h>
+__FBSDID("$FreeBSD$");
+
+#include <sys/param.h>
+#include <sys/bio.h>
+#include <sys/diskmbr.h>
+#include <sys/endian.h>
+#include <sys/gpt.h>
+#include <sys/kernel.h>
+#include <sys/kobj.h>
+#include <sys/limits.h>
+#include <sys/lock.h>
+#include <sys/malloc.h>
+#include <sys/mutex.h>
+#include <sys/queue.h>
+#include <sys/sbuf.h>
+#include <sys/systm.h>
+#include <sys/uuid.h>
+#include <geom/geom.h>
+#include <geom/part/g_part.h>
+
+#include "g_part_if.h"
+
+CTASSERT(offsetof(struct gpt_hdr, padding) == 92);
+CTASSERT(sizeof(struct gpt_ent) == 128);
+
+#define EQUUID(a,b) (memcmp(a, b, sizeof(struct uuid)) == 0)
+
+enum gpt_elt {
+ GPT_ELT_PRIHDR,
+ GPT_ELT_PRITBL,
+ GPT_ELT_SECHDR,
+ GPT_ELT_SECTBL,
+ GPT_ELT_COUNT
+};
+
+enum gpt_state {
+ GPT_STATE_UNKNOWN, /* Not determined. */
+ GPT_STATE_MISSING, /* No signature found. */
+ GPT_STATE_CORRUPT, /* Checksum mismatch. */
+ GPT_STATE_INVALID, /* Nonconformant/invalid. */
+ GPT_STATE_OK /* Perfectly fine. */
+};
+
+struct g_part_gpt_table {
+ struct g_part_table base;
+ struct gpt_hdr hdr;
+ quad_t lba[GPT_ELT_COUNT];
+ enum gpt_state state[GPT_ELT_COUNT];
+};
+
+struct g_part_gpt_entry {
+ struct g_part_entry base;
+ struct gpt_ent ent;
+};
+
+static int g_part_gpt_add(struct g_part_table *, struct g_part_entry *,
+ struct g_part_parms *);
+static int g_part_gpt_create(struct g_part_table *, struct g_part_parms *);
+static int g_part_gpt_destroy(struct g_part_table *, struct g_part_parms *);
+static int g_part_gpt_dumpto(struct g_part_table *, struct g_part_entry *);
+static int g_part_gpt_modify(struct g_part_table *, struct g_part_entry *,
+ struct g_part_parms *);
+static char *g_part_gpt_name(struct g_part_table *, struct g_part_entry *,
+ char *, size_t);
+static int g_part_gpt_probe(struct g_part_table *, struct g_consumer *);
+static int g_part_gpt_read(struct g_part_table *, struct g_consumer *);
+static const char *g_part_gpt_type(struct g_part_table *, struct g_part_entry *,
+ char *, size_t);
+static int g_part_gpt_write(struct g_part_table *, struct g_consumer *);
+
+static kobj_method_t g_part_gpt_methods[] = {
+ KOBJMETHOD(g_part_add, g_part_gpt_add),
+ KOBJMETHOD(g_part_create, g_part_gpt_create),
+ KOBJMETHOD(g_part_destroy, g_part_gpt_destroy),
+ KOBJMETHOD(g_part_dumpto, g_part_gpt_dumpto),
+ KOBJMETHOD(g_part_modify, g_part_gpt_modify),
+ KOBJMETHOD(g_part_name, g_part_gpt_name),
+ KOBJMETHOD(g_part_probe, g_part_gpt_probe),
+ KOBJMETHOD(g_part_read, g_part_gpt_read),
+ KOBJMETHOD(g_part_type, g_part_gpt_type),
+ KOBJMETHOD(g_part_write, g_part_gpt_write),
+ { 0, 0 }
+};
+
+static struct g_part_scheme g_part_gpt_scheme = {
+ "GPT",
+ g_part_gpt_methods,
+ sizeof(struct g_part_gpt_table),
+ .gps_entrysz = sizeof(struct g_part_gpt_entry),
+ .gps_minent = 128,
+ .gps_maxent = INT_MAX,
+};
+G_PART_SCHEME_DECLARE(g_part_gpt_scheme);
+
+static struct uuid gpt_uuid_efi = GPT_ENT_TYPE_EFI;
+static struct uuid gpt_uuid_freebsd = GPT_ENT_TYPE_FREEBSD;
+static struct uuid gpt_uuid_freebsd_swap = GPT_ENT_TYPE_FREEBSD_SWAP;
+static struct uuid gpt_uuid_freebsd_ufs = GPT_ENT_TYPE_FREEBSD_UFS;
+static struct uuid gpt_uuid_freebsd_vinum = GPT_ENT_TYPE_FREEBSD_VINUM;
+static struct uuid gpt_uuid_linux_swap = GPT_ENT_TYPE_LINUX_SWAP;
+static struct uuid gpt_uuid_mbr = GPT_ENT_TYPE_MBR;
+static struct uuid gpt_uuid_unused = GPT_ENT_TYPE_UNUSED;
+
+static void
+gpt_read_hdr(struct g_part_gpt_table *table, struct g_consumer *cp,
+ enum gpt_elt elt, struct gpt_hdr *hdr)
+{
+ struct uuid uuid;
+ struct g_provider *pp;
+ char *buf;
+ quad_t lba, last;
+ int error;
+ uint32_t crc, sz;
+
+ pp = cp->provider;
+ last = (pp->mediasize / pp->sectorsize) - 1;
+ table->lba[elt] = (elt == GPT_ELT_PRIHDR) ? 1 : last;
+ table->state[elt] = GPT_STATE_MISSING;
+ buf = g_read_data(cp, table->lba[elt] * pp->sectorsize, pp->sectorsize,
+ &error);
+ if (buf == NULL)
+ return;
+ bcopy(buf, hdr, sizeof(*hdr));
+ if (memcmp(hdr->hdr_sig, GPT_HDR_SIG, sizeof(hdr->hdr_sig)) != 0)
+ return;
+
+ table->state[elt] = GPT_STATE_CORRUPT;
+ sz = le32toh(hdr->hdr_size);
+ if (sz < 92 || sz > pp->sectorsize)
+ return;
+ crc = le32toh(hdr->hdr_crc_self);
+ hdr->hdr_crc_self = 0;
+ if (crc32(hdr, sz) != crc)
+ return;
+ hdr->hdr_size = sz;
+ hdr->hdr_crc_self = crc;
+
+ table->state[elt] = GPT_STATE_INVALID;
+ hdr->hdr_revision = le32toh(hdr->hdr_revision);
+ if (hdr->hdr_revision < 0x00010000)
+ return;
+ hdr->hdr_lba_self = le64toh(hdr->hdr_lba_self);
+ if (hdr->hdr_lba_self != table->lba[elt])
+ return;
+ hdr->hdr_lba_alt = le64toh(hdr->hdr_lba_alt);
+
+ /* Check the managed area. */
+ hdr->hdr_lba_start = le64toh(hdr->hdr_lba_start);
+ if (hdr->hdr_lba_start < 2 || hdr->hdr_lba_start >= last)
+ return;
+ hdr->hdr_lba_end = le64toh(hdr->hdr_lba_end);
+ if (hdr->hdr_lba_end < hdr->hdr_lba_start || hdr->hdr_lba_end >= last)
+ return;
+
+ /* Check the table location and size of the table. */
+ hdr->hdr_entries = le32toh(hdr->hdr_entries);
+ hdr->hdr_entsz = le32toh(hdr->hdr_entsz);
+ if (hdr->hdr_entries == 0 || hdr->hdr_entsz < 128 ||
+ (hdr->hdr_entsz & 7) != 0)
+ return;
+ hdr->hdr_lba_table = le64toh(hdr->hdr_lba_table);
+ if (hdr->hdr_lba_table < 2 || hdr->hdr_lba_table >= last)
+ return;
+ if (hdr->hdr_lba_table >= hdr->hdr_lba_start &&
+ hdr->hdr_lba_table <= hdr->hdr_lba_end)
+ return;
+ lba = hdr->hdr_lba_table +
+ (hdr->hdr_entries * hdr->hdr_entsz + pp->sectorsize - 1) /
+ pp->sectorsize - 1;
+ if (lba >= last)
+ return;
+ if (lba >= hdr->hdr_lba_start && lba <= hdr->hdr_lba_end)
+ return;
+
+ table->state[elt] = GPT_STATE_OK;
+ le_uuid_dec(&hdr->hdr_uuid, &uuid);
+ hdr->hdr_uuid = uuid;
+ hdr->hdr_crc_table = le32toh(hdr->hdr_crc_table);
+}
+
+static struct gpt_ent *
+gpt_read_tbl(struct g_part_gpt_table *table, struct g_consumer *cp,
+ enum gpt_elt elt, struct gpt_hdr *hdr)
+{
+ struct g_provider *pp;
+ struct gpt_ent *ent, *tbl;
+ char *buf, *p;
+ unsigned int idx, sectors, tblsz;
+ int error;
+ uint16_t ch;
+
+ pp = cp->provider;
+ table->lba[elt] = hdr->hdr_lba_table;
+
+ table->state[elt] = GPT_STATE_MISSING;
+ tblsz = hdr->hdr_entries * hdr->hdr_entsz;
+ sectors = (tblsz + pp->sectorsize - 1) / pp->sectorsize;
+ buf = g_read_data(cp, table->lba[elt] * pp->sectorsize,
+ sectors * pp->sectorsize, &error);
+ if (buf == NULL)
+ return (NULL);
+
+ table->state[elt] = GPT_STATE_CORRUPT;
+ if (crc32(buf, tblsz) != hdr->hdr_crc_table) {
+ g_free(buf);
+ return (NULL);
+ }
+
+ table->state[elt] = GPT_STATE_OK;
+ tbl = g_malloc(hdr->hdr_entries * sizeof(struct gpt_ent),
+ M_WAITOK | M_ZERO);
+
+ for (idx = 0, ent = tbl, p = buf;
+ idx < hdr->hdr_entries;
+ idx++, ent++, p += hdr->hdr_entsz) {
+ le_uuid_dec(p, &ent->ent_type);
+ le_uuid_dec(p + 16, &ent->ent_uuid);
+ ent->ent_lba_start = le64dec(p + 32);
+ ent->ent_lba_end = le64dec(p + 40);
+ ent->ent_attr = le64dec(p + 48);
+ for (ch = 0; ch < sizeof(ent->ent_name)/2; ch++)
+ ent->ent_name[ch] = le16dec(p + 56 + ch * 2);
+ }
+
+ g_free(buf);
+ return (tbl);
+}
+
+static int
+gpt_matched_hdrs(struct gpt_hdr *pri, struct gpt_hdr *sec)
+{
+
+ if (!EQUUID(&pri->hdr_uuid, &sec->hdr_uuid))
+ return (0);
+ return ((pri->hdr_revision == sec->hdr_revision &&
+ pri->hdr_size == sec->hdr_size &&
+ pri->hdr_lba_start == sec->hdr_lba_start &&
+ pri->hdr_lba_end == sec->hdr_lba_end &&
+ pri->hdr_entries == sec->hdr_entries &&
+ pri->hdr_entsz == sec->hdr_entsz &&
+ pri->hdr_crc_table == sec->hdr_crc_table) ? 1 : 0);
+}
+
+static int
+gpt_parse_type(const char *type, struct uuid *uuid)
+{
+ struct uuid tmp;
+ int error;
+
+ if (type[0] != '@') {
+ error = parse_uuid(type, &tmp);
+ if (error)
+ return (error);
+ if (EQUUID(&tmp, &gpt_uuid_unused))
+ return (EINVAL);
+ *uuid = tmp;
+ return (0);
+ }
+ if (!strcmp(type, g_part_alias_name(G_PART_ALIAS_EFI)))
+ *uuid = gpt_uuid_efi;
+ else if (!strcmp(type, g_part_alias_name(G_PART_ALIAS_FREEBSD)))
+ *uuid = gpt_uuid_freebsd;
+ else if (!strcmp(type, g_part_alias_name(G_PART_ALIAS_FREEBSD_SWAP)))
+ *uuid = gpt_uuid_freebsd_swap;
+ else if (!strcmp(type, g_part_alias_name(G_PART_ALIAS_FREEBSD_UFS)))
+ *uuid = gpt_uuid_freebsd_ufs;
+ else if (!strcmp(type, g_part_alias_name(G_PART_ALIAS_FREEBSD_VINUM)))
+ *uuid = gpt_uuid_freebsd_vinum;
+ else if (!strcmp(type, g_part_alias_name(G_PART_ALIAS_MBR)))
+ *uuid = gpt_uuid_mbr;
+ else
+ return (EINVAL);
+ return (0);
+}
+
+static int
+g_part_gpt_add(struct g_part_table *basetable, struct g_part_entry *baseentry,
+ struct g_part_parms *gpp)
+{
+ struct g_part_gpt_entry *entry;
+ int error;
+
+ entry = (struct g_part_gpt_entry *)baseentry;
+ error = gpt_parse_type(gpp->gpp_type, &entry->ent.ent_type);
+ if (error)
+ return (error);
+ kern_uuidgen(&entry->ent.ent_uuid, 1);
+ entry->ent.ent_lba_start = baseentry->gpe_start;
+ entry->ent.ent_lba_end = baseentry->gpe_end;
+ if (baseentry->gpe_deleted) {
+ entry->ent.ent_attr = 0;
+ bzero(entry->ent.ent_name, sizeof(entry->ent.ent_name));
+ }
+ /* XXX label */
+ return (0);
+}
+
+static int
+g_part_gpt_create(struct g_part_table *basetable, struct g_part_parms *gpp)
+{
+ struct g_provider *pp;
+ struct g_part_gpt_table *table;
+ quad_t last;
+ size_t tblsz;
+
+ table = (struct g_part_gpt_table *)basetable;
+ pp = gpp->gpp_provider;
+ tblsz = (basetable->gpt_entries * sizeof(struct gpt_ent) +
+ pp->sectorsize - 1) / pp->sectorsize;
+ if (pp->sectorsize < 512 ||
+ pp->mediasize < (3 + 2 * tblsz + basetable->gpt_entries) *
+ pp->sectorsize)
+ return (ENOSPC);
+
+ last = (pp->mediasize / pp->sectorsize) - 1;
+
+ table->lba[GPT_ELT_PRIHDR] = 1;
+ table->lba[GPT_ELT_PRITBL] = 2;
+ table->lba[GPT_ELT_SECHDR] = last;
+ table->lba[GPT_ELT_SECTBL] = last - tblsz;
+
+ bcopy(GPT_HDR_SIG, table->hdr.hdr_sig, sizeof(table->hdr.hdr_sig));
+ table->hdr.hdr_revision = GPT_HDR_REVISION;
+ table->hdr.hdr_size = offsetof(struct gpt_hdr, padding);
+ table->hdr.hdr_lba_start = 2 + tblsz;
+ table->hdr.hdr_lba_end = last - tblsz - 1;
+ kern_uuidgen(&table->hdr.hdr_uuid, 1);
+ table->hdr.hdr_entries = basetable->gpt_entries;
+ table->hdr.hdr_entsz = sizeof(struct gpt_ent);
+
+ basetable->gpt_first = table->hdr.hdr_lba_start;
+ basetable->gpt_last = table->hdr.hdr_lba_end;
+ return (0);
+}
+
+static int
+g_part_gpt_destroy(struct g_part_table *basetable, struct g_part_parms *gpp)
+{
+
+ /*
+ * Wipe the first 2 sectors as well as the last to clear the
+ * partitioning.
+ */
+ basetable->gpt_smhead |= 3;
+ basetable->gpt_smtail |= 1;
+ return (0);
+}
+
+static int
+g_part_gpt_dumpto(struct g_part_table *table, struct g_part_entry *baseentry)
+{
+ struct g_part_gpt_entry *entry;
+
+ entry = (struct g_part_gpt_entry *)baseentry;
+ return ((EQUUID(&entry->ent.ent_type, &gpt_uuid_freebsd_swap) ||
+ EQUUID(&entry->ent.ent_type, &gpt_uuid_linux_swap)) ? 1 : 0);
+}
+
+static int
+g_part_gpt_modify(struct g_part_table *basetable,
+ struct g_part_entry *baseentry, struct g_part_parms *gpp)
+{
+ struct g_part_gpt_entry *entry;
+ int error;
+
+ entry = (struct g_part_gpt_entry *)baseentry;
+ error = gpt_parse_type(gpp->gpp_type, &entry->ent.ent_type);
+ if (error)
+ return (error);
+ /* XXX label */
+ return (0);
+}
+
+static char *
+g_part_gpt_name(struct g_part_table *table, struct g_part_entry *baseentry,
+ char *buf, size_t bufsz)
+{
+ struct g_part_gpt_entry *entry;
+ char c;
+
+ entry = (struct g_part_gpt_entry *)baseentry;
+ c = (EQUUID(&entry->ent.ent_type, &gpt_uuid_freebsd)) ? 's' : 'p';
+ snprintf(buf, bufsz, "%c%d", c, baseentry->gpe_index);
+ return (buf);
+}
+
+static int
+g_part_gpt_probe(struct g_part_table *table, struct g_consumer *cp)
+{
+ struct g_provider *pp;
+ char *buf;
+ int error, res;
+
+ /* We don't nest, which means that our depth should be 0. */
+ if (table->gpt_depth != 0)
+ return (ENXIO);
+
+ pp = cp->provider;
+
+ /*
+ * Sanity-check the provider. Since the first sector on the provider
+ * must be a PMBR and a PMBR is 512 bytes large, the sector size
+ * must be at least 512 bytes. Also, since the theoretical minimum
+ * number of sectors needed by GPT is 6, any medium that has less
+ * than 6 sectors is never going to be able to hold a GPT. The
+ * number 6 comes from:
+ * 1 sector for the PMBR
+ * 2 sectors for the GPT headers (each 1 sector)
+ * 2 sectors for the GPT tables (each 1 sector)
+ * 1 sector for an actual partition
+ * It's better to catch this pathological case early than behaving
+ * pathologically later on...
+ */
+ if (pp->sectorsize < 512 || pp->mediasize < 6 * pp->sectorsize)
+ return (ENOSPC);
+
+ /* Check that there's a MBR. */
+ buf = g_read_data(cp, 0L, pp->sectorsize, &error);
+ if (buf == NULL)
+ return (error);
+ res = le16dec(buf + DOSMAGICOFFSET);
+ g_free(buf);
+ if (res != DOSMAGIC)
+ return (ENXIO);
+
+ /* Check that there's a primary header. */
+ buf = g_read_data(cp, pp->sectorsize, pp->sectorsize, &error);
+ if (buf == NULL)
+ return (error);
+ res = memcmp(buf, GPT_HDR_SIG, 8);
+ g_free(buf);
+ if (res == 0)
+ return (G_PART_PROBE_PRI_HIGH);
+
+ /* No primary? Check that there's a secondary. */
+ buf = g_read_data(cp, pp->mediasize - pp->sectorsize, pp->sectorsize,
+ &error);
+ if (buf == NULL)
+ return (error);
+ res = memcmp(buf, GPT_HDR_SIG, 8);
+ g_free(buf);
+ return ((res == 0) ? G_PART_PROBE_PRI_HIGH : ENXIO);
+}
+
+static int
+g_part_gpt_read(struct g_part_table *basetable, struct g_consumer *cp)
+{
+ struct gpt_hdr prihdr, sechdr;
+ struct gpt_ent *tbl, *pritbl, *sectbl;
+ struct g_provider *pp;
+ struct g_part_gpt_table *table;
+ struct g_part_gpt_entry *entry;
+ int index;
+
+ table = (struct g_part_gpt_table *)basetable;
+ pp = cp->provider;
+
+ /* Read the primary header and table. */
+ gpt_read_hdr(table, cp, GPT_ELT_PRIHDR, &prihdr);
+ if (table->state[GPT_ELT_PRIHDR] == GPT_STATE_OK) {
+ pritbl = gpt_read_tbl(table, cp, GPT_ELT_PRITBL, &prihdr);
+ } else {
+ table->state[GPT_ELT_PRITBL] = GPT_STATE_MISSING;
+ pritbl = NULL;
+ }
+
+ /* Read the secondary header and table. */
+ gpt_read_hdr(table, cp, GPT_ELT_SECHDR, &sechdr);
+ if (table->state[GPT_ELT_SECHDR] == GPT_STATE_OK) {
+ sectbl = gpt_read_tbl(table, cp, GPT_ELT_SECTBL, &sechdr);
+ } else {
+ table->state[GPT_ELT_SECTBL] = GPT_STATE_MISSING;
+ sectbl = NULL;
+ }
+
+ /* Fail if we haven't got any good tables at all. */
+ if (table->state[GPT_ELT_PRITBL] != GPT_STATE_OK &&
+ table->state[GPT_ELT_SECTBL] != GPT_STATE_OK) {
+ printf("GEOM: %s: corrupt or invalid GPT detected.\n",
+ pp->name);
+ printf("GEOM: %s: GPT rejected -- may not be recoverable.\n",
+ pp->name);
+ return (EINVAL);
+ }
+
+ /*
+ * If both headers are good but they disagree with each other,
+ * then invalidate one. We prefer to keep the primary header,
+ * unless the primary table is corrupt.
+ */
+ if (table->state[GPT_ELT_PRIHDR] == GPT_STATE_OK &&
+ table->state[GPT_ELT_SECHDR] == GPT_STATE_OK &&
+ !gpt_matched_hdrs(&prihdr, &sechdr)) {
+ if (table->state[GPT_ELT_PRITBL] == GPT_STATE_OK)
+ table->state[GPT_ELT_SECHDR] = GPT_STATE_INVALID;
+ else
+ table->state[GPT_ELT_PRIHDR] = GPT_STATE_INVALID;
+ }
+
+ if (table->state[GPT_ELT_PRIHDR] != GPT_STATE_OK) {
+ printf("GEOM: %s: the primary GPT table is corrupt or "
+ "invalid.\n", pp->name);
+ printf("GEOM: %s: using the secondary instead -- recovery "
+ "strongly advised.\n", pp->name);
+ table->hdr = sechdr;
+ tbl = sectbl;
+ if (pritbl != NULL)
+ g_free(pritbl);
+ } else {
+ if (table->state[GPT_ELT_SECHDR] != GPT_STATE_OK) {
+ printf("GEOM: %s: the secondary GPT table is corrupt "
+ "or invalid.\n", pp->name);
+ printf("GEOM: %s: using the primary only -- recovery "
+ "suggested.\n", pp->name);
+ }
+ table->hdr = prihdr;
+ tbl = pritbl;
+ if (sectbl != NULL)
+ g_free(sectbl);
+ }
+
+ basetable->gpt_first = table->hdr.hdr_lba_start;
+ basetable->gpt_last = table->hdr.hdr_lba_end;
+ basetable->gpt_entries = table->hdr.hdr_entries;
+
+ for (index = basetable->gpt_entries - 1; index >= 0; index--) {
+ if (EQUUID(&tbl[index].ent_type, &gpt_uuid_unused))
+ continue;
+ entry = (struct g_part_gpt_entry *)g_part_new_entry(basetable,
+ index+1, tbl[index].ent_lba_start, tbl[index].ent_lba_end);
+ entry->ent = tbl[index];
+ }
+
+ g_free(tbl);
+ return (0);
+}
+
+static const char *
+g_part_gpt_type(struct g_part_table *basetable, struct g_part_entry *baseentry,
+ char *buf, size_t bufsz)
+{
+ struct g_part_gpt_entry *entry;
+ struct uuid *type;
+
+ entry = (struct g_part_gpt_entry *)baseentry;
+ type = &entry->ent.ent_type;
+ if (EQUUID(type, &gpt_uuid_efi))
+ return (g_part_alias_name(G_PART_ALIAS_EFI));
+ if (EQUUID(type, &gpt_uuid_freebsd))
+ return (g_part_alias_name(G_PART_ALIAS_FREEBSD));
+ if (EQUUID(type, &gpt_uuid_freebsd_swap))
+ return (g_part_alias_name(G_PART_ALIAS_FREEBSD_SWAP));
+ if (EQUUID(type, &gpt_uuid_freebsd_ufs))
+ return (g_part_alias_name(G_PART_ALIAS_FREEBSD_UFS));
+ if (EQUUID(type, &gpt_uuid_freebsd_vinum))
+ return (g_part_alias_name(G_PART_ALIAS_FREEBSD_VINUM));
+ if (EQUUID(type, &gpt_uuid_mbr))
+ return (g_part_alias_name(G_PART_ALIAS_MBR));
+ snprintf_uuid(buf, bufsz, type);
+ return (buf);
+}
+
+static int
+g_part_gpt_write(struct g_part_table *basetable, struct g_consumer *cp)
+{
+ unsigned char *buf, *bp;
+ struct g_provider *pp;
+ struct g_part_entry *baseentry;
+ struct g_part_gpt_entry *entry;
+ struct g_part_gpt_table *table;
+ size_t tlbsz;
+ uint32_t crc;
+ int error, index;
+
+ pp = cp->provider;
+ table = (struct g_part_gpt_table *)basetable;
+ tlbsz = (table->hdr.hdr_entries * table->hdr.hdr_entsz +
+ pp->sectorsize - 1) / pp->sectorsize;
+
+ if (basetable->gpt_created) {
+ buf = g_malloc(pp->sectorsize, M_WAITOK | M_ZERO);
+ le16enc(buf + DOSMAGICOFFSET, DOSMAGIC);
+ buf[DOSPARTOFF + 1] = 0xff; /* shd */
+ buf[DOSPARTOFF + 2] = 0xff; /* ssect */
+ buf[DOSPARTOFF + 3] = 0xff; /* scyl */
+ buf[DOSPARTOFF + 4] = 0xee; /* typ */
+ buf[DOSPARTOFF + 5] = 0xff; /* ehd */
+ buf[DOSPARTOFF + 6] = 0xff; /* esect */
+ buf[DOSPARTOFF + 7] = 0xff; /* ecyl */
+ le32enc(buf + DOSPARTOFF + 8, 1); /* start */
+ le32enc(buf + DOSPARTOFF + 12,
+ MIN(pp->mediasize / pp->sectorsize - 1, 0xffffffffLL));
+ error = g_write_data(cp, 0, buf, pp->sectorsize);
+ g_free(buf);
+ if (error)
+ return (error);
+ }
+
+ /* Allocate space for the header and entries. */
+ buf = g_malloc((tlbsz + 1) * pp->sectorsize, M_WAITOK | M_ZERO);
+
+ memcpy(buf, table->hdr.hdr_sig, sizeof(table->hdr.hdr_sig));
+ le32enc(buf + 8, table->hdr.hdr_revision);
+ le32enc(buf + 12, table->hdr.hdr_size);
+ le64enc(buf + 40, table->hdr.hdr_lba_start);
+ le64enc(buf + 48, table->hdr.hdr_lba_end);
+ le_uuid_enc(buf + 56, &table->hdr.hdr_uuid);
+ le32enc(buf + 80, table->hdr.hdr_entries);
+ le32enc(buf + 84, table->hdr.hdr_entsz);
+
+ LIST_FOREACH(baseentry, &basetable->gpt_entry, gpe_entry) {
+ entry = (struct g_part_gpt_entry *)baseentry;
+ index = baseentry->gpe_index - 1;
+ bp = buf + pp->sectorsize + table->hdr.hdr_entsz * index;
+ le_uuid_enc(bp, &entry->ent.ent_type);
+ le_uuid_enc(bp + 16, &entry->ent.ent_uuid);
+ le64enc(bp + 32, entry->ent.ent_lba_start);
+ le64enc(bp + 40, entry->ent.ent_lba_end);
+ le64enc(bp + 48, entry->ent.ent_attr);
+ memcpy(bp + 56, entry->ent.ent_name,
+ sizeof(entry->ent.ent_name));
+ }
+
+ crc = crc32(buf + pp->sectorsize,
+ table->hdr.hdr_entries * table->hdr.hdr_entsz);
+ le32enc(buf + 88, crc);
+
+ /* Write primary meta-data. */
+ le32enc(buf + 16, 0); /* hdr_crc_self. */
+ le64enc(buf + 24, table->lba[GPT_ELT_PRIHDR]); /* hdr_lba_self. */
+ le64enc(buf + 32, table->lba[GPT_ELT_SECHDR]); /* hdr_lba_alt. */
+ le64enc(buf + 72, table->lba[GPT_ELT_PRITBL]); /* hdr_lba_table. */
+ crc = crc32(buf, table->hdr.hdr_size);
+ le32enc(buf + 16, crc);
+
+ error = g_write_data(cp, table->lba[GPT_ELT_PRITBL] * pp->sectorsize,
+ buf + pp->sectorsize, tlbsz * pp->sectorsize);
+ if (error)
+ goto out;
+ error = g_write_data(cp, table->lba[GPT_ELT_PRIHDR] * pp->sectorsize,
+ buf, pp->sectorsize);
+ if (error)
+ goto out;
+
+ /* Write secondary meta-data. */
+ le32enc(buf + 16, 0); /* hdr_crc_self. */
+ le64enc(buf + 24, table->lba[GPT_ELT_SECHDR]); /* hdr_lba_self. */
+ le64enc(buf + 32, table->lba[GPT_ELT_PRIHDR]); /* hdr_lba_alt. */
+ le64enc(buf + 72, table->lba[GPT_ELT_SECTBL]); /* hdr_lba_table. */
+ crc = crc32(buf, table->hdr.hdr_size);
+ le32enc(buf + 16, crc);
+
+ error = g_write_data(cp, table->lba[GPT_ELT_SECTBL] * pp->sectorsize,
+ buf + pp->sectorsize, tlbsz * pp->sectorsize);
+ if (error)
+ goto out;
+ error = g_write_data(cp, table->lba[GPT_ELT_SECHDR] * pp->sectorsize,
+ buf, pp->sectorsize);
+
+ out:
+ g_free(buf);
+ return (error);
+}
+
+#if 0
+static void
+g_gpt_to_utf8(struct sbuf *sb, uint16_t *str, size_t len)
+{
+ u_int bo;
+ uint32_t ch;
+ uint16_t c;
+
+ bo = BYTE_ORDER;
+ while (len > 0 && *str != 0) {
+ ch = (bo == BIG_ENDIAN) ? be16toh(*str) : le16toh(*str);
+ str++, len--;
+ if ((ch & 0xf800) == 0xd800) {
+ if (len > 0) {
+ c = (bo == BIG_ENDIAN) ? be16toh(*str)
+ : le16toh(*str);
+ str++, len--;
+ } else
+ c = 0xfffd;
+ if ((ch & 0x400) == 0 && (c & 0xfc00) == 0xdc00) {
+ ch = ((ch & 0x3ff) << 10) + (c & 0x3ff);
+ ch += 0x10000;
+ } else
+ ch = 0xfffd;
+ } else if (ch == 0xfffe) { /* BOM (U+FEFF) swapped. */
+ bo = (bo == BIG_ENDIAN) ? LITTLE_ENDIAN : BIG_ENDIAN;
+ continue;
+ } else if (ch == 0xfeff) /* BOM (U+FEFF) unswapped. */
+ continue;
+
+ if (ch < 0x80)
+ sbuf_printf(sb, "%c", ch);
+ else if (ch < 0x800)
+ sbuf_printf(sb, "%c%c", 0xc0 | (ch >> 6),
+ 0x80 | (ch & 0x3f));
+ else if (ch < 0x10000)
+ sbuf_printf(sb, "%c%c%c", 0xe0 | (ch >> 12),
+ 0x80 | ((ch >> 6) & 0x3f), 0x80 | (ch & 0x3f));
+ else if (ch < 0x200000)
+ sbuf_printf(sb, "%c%c%c%c", 0xf0 | (ch >> 18),
+ 0x80 | ((ch >> 12) & 0x3f),
+ 0x80 | ((ch >> 6) & 0x3f), 0x80 | (ch & 0x3f));
+ }
+}
+#endif