static int recover(gpt_t gpt, int recoverable) { off_t last = gpt_last(gpt); map_t map; struct mbr *mbr; if (map_find(gpt, MAP_TYPE_MBR) != NULL) { gpt_warnx(gpt, "Device contains an MBR"); return -1; } gpt->gpt = map_find(gpt, MAP_TYPE_PRI_GPT_HDR); gpt->tpg = map_find(gpt, MAP_TYPE_SEC_GPT_HDR); gpt->tbl = map_find(gpt, MAP_TYPE_PRI_GPT_TBL); gpt->lbt = map_find(gpt, MAP_TYPE_SEC_GPT_TBL); if (gpt->gpt == NULL && gpt->tpg == NULL) { gpt_warnx(gpt, "No primary or secondary GPT headers, " "can't recover"); return -1; } if (gpt->tbl == NULL && gpt->lbt == NULL) { gpt_warnx(gpt, "No primary or secondary GPT tables, " "can't recover"); return -1; } if (gpt->gpt != NULL && ((struct gpt_hdr *)(gpt->gpt->map_data))->hdr_lba_alt != (uint64_t)last) { gpt_warnx(gpt, "Media size has changed, please use " "'%s resizedisk'", getprogname()); return -1; } if (gpt->tbl != NULL && gpt->lbt == NULL) { if (recover_gpt_tbl(gpt, MAP_TYPE_SEC_GPT_TBL, last - gpt->tbl->map_size) == -1) return -1; } else if (gpt->tbl == NULL && gpt->lbt != NULL) { if (recover_gpt_tbl(gpt, MAP_TYPE_PRI_GPT_TBL, 2LL) == -1) return -1; } if (gpt->gpt != NULL && gpt->tpg == NULL) { if (recover_gpt_hdr(gpt, MAP_TYPE_SEC_GPT_HDR, last) == -1) return -1; } else if (gpt->gpt == NULL && gpt->tpg != NULL) { if (recover_gpt_hdr(gpt, MAP_TYPE_PRI_GPT_HDR, 1LL) == -1) return -1; } /* * Create PMBR if it doesn't already exist. */ if (map_find(gpt, MAP_TYPE_PMBR) == NULL) { if (map_free(gpt, 0LL, 1LL) == 0) { gpt_warnx(gpt, "No room for the PMBR"); return -1; } mbr = gpt_read(gpt, 0LL, 1); if (mbr == NULL) { gpt_warnx(gpt, "Error reading MBR"); return -1; } memset(mbr, 0, sizeof(*mbr)); mbr->mbr_sig = htole16(MBR_SIG); gpt_create_pmbr_part(mbr->mbr_part, last, 0); map = map_add(gpt, 0LL, 1LL, MAP_TYPE_PMBR, mbr, 1); if (gpt_write(gpt, map) == -1) { gpt_warn(gpt, "Can't write PMBR"); return -1; } gpt_msg(gpt, "Recreated PMBR (you may need to rerun 'gpt biosboot'"); } return 0; }
static struct gpt_ent * migrate_disklabel(gpt_t gpt, off_t start, struct gpt_ent *ent, gpt_type_t (*convert)(gpt_t, u_int, u_int)) { char *buf; struct disklabel *dl; off_t ofs, rawofs; unsigned int i; gpt_type_t type; buf = gpt_read(gpt, start + LABELSECTOR, 1); if (buf == NULL) { gpt_warn(gpt, "Error reading label"); return NULL; } dl = (void*)(buf + LABELOFFSET); if (le32toh(dl->d_magic) != DISKMAGIC || le32toh(dl->d_magic2) != DISKMAGIC) { gpt_warnx(gpt, "MBR partition without disklabel"); free(buf); return ent; } rawofs = le32toh(dl->d_partitions[RAW_PART].p_offset) * le32toh(dl->d_secsize); for (i = 0; i < le16toh(dl->d_npartitions); i++) { if (dl->d_partitions[i].p_fstype == FS_UNUSED) continue; ofs = le32toh(dl->d_partitions[i].p_offset) * le32toh(dl->d_secsize); if (ofs < rawofs) rawofs = 0; } if (gpt->verbose > 1) gpt_msg(gpt, "rawofs=%ju", (uintmax_t)rawofs); rawofs /= gpt->secsz; for (i = 0; i < le16toh(dl->d_npartitions); i++) { if (gpt->verbose > 1) gpt_msg(gpt, "Disklabel partition %u type %s", i, fstypename(dl->d_partitions[i].p_fstype)); type = (*convert)(gpt, i, dl->d_partitions[i].p_fstype); if (type == GPT_TYPE_INVALID) continue; gpt_uuid_create(type, ent->ent_type, ent->ent_name, sizeof(ent->ent_name)); ofs = (le32toh(dl->d_partitions[i].p_offset) * le32toh(dl->d_secsize)) / gpt->secsz; ofs = (ofs > 0) ? ofs - rawofs : 0; ent->ent_lba_start = htole64((uint64_t)ofs); ent->ent_lba_end = htole64((uint64_t)(ofs + (off_t)le32toh((uint64_t)dl->d_partitions[i].p_size) - 1LL)); ent++; } free(buf); return ent; }
/* * clocksource: the GPT device is a decrementing counters, so we negate * the value being read. */ static cycle_t bcm_read_timer(struct clocksource *cs) { return GPT_MAX_COUNT - gpt_read(config.cs_index); }