2 * Copyright (c) 2002 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>
42 static int primary_only;
49 "usage: %s [-p] device ...\n", getprogname());
66 last = mediasz / secsz - 1LL;
68 if (map_find(MAP_TYPE_PRI_GPT_HDR) != NULL ||
69 map_find(MAP_TYPE_SEC_GPT_HDR) != NULL) {
70 warnx("%s: error: device already contains a GPT", device_name);
73 if (map_find(MAP_TYPE_MBR) != NULL) {
74 warnx("%s: error: device contains a MBR", device_name);
81 if (map_find(MAP_TYPE_PMBR) == NULL) {
82 if (map_free(0LL, 1LL) == 0) {
83 warnx("%s: error: no room for the PMBR", device_name);
86 mbr = gpt_read(fd, 0LL, 1);
87 bzero(mbr, sizeof(*mbr));
88 mbr->mbr_sig = htole16(MBR_SIG);
89 mbr->mbr_part[0].part_shd = 0xff;
90 mbr->mbr_part[0].part_ssect = 0xff;
91 mbr->mbr_part[0].part_scyl = 0xff;
92 mbr->mbr_part[0].part_typ = 0xee;
93 mbr->mbr_part[0].part_ehd = 0xff;
94 mbr->mbr_part[0].part_esect = 0xff;
95 mbr->mbr_part[0].part_ecyl = 0xff;
96 mbr->mbr_part[0].part_start_lo = htole16(1);
97 if (last > 0xffffffff) {
98 mbr->mbr_part[0].part_size_lo = htole16(0xffff);
99 mbr->mbr_part[0].part_size_hi = htole16(0xffff);
101 mbr->mbr_part[0].part_size_lo = htole16(last);
102 mbr->mbr_part[0].part_size_hi = htole16(last >> 16);
104 map = map_add(0LL, 1LL, MAP_TYPE_PMBR, mbr);
108 /* Get the amount of free space after the MBR */
109 blocks = map_free(1LL, 0LL);
111 warnx("%s: error: no room for the GPT header", device_name);
115 /* Don't create more than parts entries. */
116 if ((uint64_t)(blocks - 1) * secsz > parts * sizeof(struct gpt_ent)) {
117 blocks = (parts * sizeof(struct gpt_ent)) / secsz;
118 if ((parts * sizeof(struct gpt_ent)) % secsz)
120 blocks++; /* Don't forget the header itself */
123 /* Never cross the median of the device. */
124 if ((blocks + 1LL) > ((last + 1LL) >> 1))
125 blocks = ((last + 1LL) >> 1) - 1LL;
128 * Get the amount of free space at the end of the device and
129 * calculate the size for the GPT structures.
132 if (map->map_type != MAP_TYPE_UNUSED) {
133 warnx("%s: error: no room for the backup header", device_name);
137 if (map->map_size < blocks)
138 blocks = map->map_size;
140 warnx("%s: error: no room for the GPT table", device_name);
144 blocks--; /* Number of blocks in the GPT table. */
145 gpt = map_add(1LL, 1LL, MAP_TYPE_PRI_GPT_HDR, calloc(1, secsz));
146 tbl = map_add(2LL, blocks, MAP_TYPE_PRI_GPT_TBL,
147 calloc(blocks, secsz));
148 if (gpt == NULL || tbl == NULL)
152 memcpy(hdr->hdr_sig, GPT_HDR_SIG, sizeof(hdr->hdr_sig));
153 hdr->hdr_revision = htole32(GPT_HDR_REVISION);
155 * XXX struct gpt_hdr is not a multiple of 8 bytes in size and thus
156 * contains padding we must not include in the size.
158 hdr->hdr_size = htole32(offsetof(struct gpt_hdr, padding));
159 hdr->hdr_lba_self = htole64(gpt->map_start);
160 hdr->hdr_lba_alt = htole64(last);
161 hdr->hdr_lba_start = htole64(tbl->map_start + blocks);
162 hdr->hdr_lba_end = htole64(last - blocks - 1LL);
163 uuid_create(&uuid, NULL);
164 le_uuid_enc(&hdr->hdr_uuid, &uuid);
165 hdr->hdr_lba_table = htole64(tbl->map_start);
166 hdr->hdr_entries = htole32((blocks * secsz) / sizeof(struct gpt_ent));
167 if (le32toh(hdr->hdr_entries) > parts)
168 hdr->hdr_entries = htole32(parts);
169 hdr->hdr_entsz = htole32(sizeof(struct gpt_ent));
172 for (i = 0; i < le32toh(hdr->hdr_entries); i++) {
173 uuid_create(&uuid, NULL);
174 le_uuid_enc(&ent[i].ent_uuid, &uuid);
177 hdr->hdr_crc_table = htole32(crc32(ent, le32toh(hdr->hdr_entries) *
178 le32toh(hdr->hdr_entsz)));
179 hdr->hdr_crc_self = htole32(crc32(hdr, le32toh(hdr->hdr_size)));
185 * Create backup GPT if the user didn't suppress it.
188 tpg = map_add(last, 1LL, MAP_TYPE_SEC_GPT_HDR,
190 lbt = map_add(last - blocks, blocks, MAP_TYPE_SEC_GPT_TBL,
192 memcpy(tpg->map_data, gpt->map_data, secsz);
194 hdr->hdr_lba_self = htole64(tpg->map_start);
195 hdr->hdr_lba_alt = htole64(gpt->map_start);
196 hdr->hdr_lba_table = htole64(lbt->map_start);
197 hdr->hdr_crc_self = 0; /* Don't ever forget this! */
198 hdr->hdr_crc_self = htole32(crc32(hdr, le32toh(hdr->hdr_size)));
205 cmd_create(int argc, char *argv[])
209 while ((ch = getopt(argc, argv, "p")) != -1) {
222 while (optind < argc) {
223 fd = gpt_open(argv[optind++]);
225 warn("unable to open device '%s'", device_name);