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.
29 #include <sys/types.h>
50 "usage: %s [-p] device ...\n", getprogname());
66 if (map_find(MAP_TYPE_PRI_GPT_HDR) != NULL ||
67 map_find(MAP_TYPE_SEC_GPT_HDR) != NULL) {
68 warnx("%s: error: device already contains a GPT", device_name);
71 if (map_find(MAP_TYPE_MBR) != NULL) {
72 warnx("%s: error: device contains a MBR", device_name);
79 if (map_find(MAP_TYPE_PMBR) == NULL) {
80 if (map_free(0LL, 1LL) == 0) {
81 warnx("%s: error: no room for the PMBR", device_name);
84 mbr = gpt_read(fd, 0LL, 1);
85 bzero(mbr, sizeof(*mbr));
86 mbr->mbr_sig = MBR_SIG;
87 mbr->mbr_part[0].part_shd = 0xff;
88 mbr->mbr_part[0].part_ssect = 0xff;
89 mbr->mbr_part[0].part_scyl = 0xff;
90 mbr->mbr_part[0].part_typ = 0xee;
91 mbr->mbr_part[0].part_ehd = 0xff;
92 mbr->mbr_part[0].part_esect = 0xff;
93 mbr->mbr_part[0].part_ecyl = 0xff;
94 mbr->mbr_part[0].part_start_lo = 1;
95 if (mediasz > 0xffffffff) {
96 mbr->mbr_part[0].part_size_lo = 0xffff;
97 mbr->mbr_part[0].part_size_hi = 0xffff;
99 mbr->mbr_part[0].part_size_lo = mediasz & 0xffff;
100 mbr->mbr_part[0].part_size_hi = mediasz >> 16;
102 map = map_add(0LL, 1LL, MAP_TYPE_PMBR, mbr);
106 /* Get the amount of free space after the MBR */
107 blocks = map_free(1LL, 0LL);
109 warnx("%s: error: no room for the GPT header", device_name);
113 /* Don't create more than parts entries. */
114 if ((uint64_t)(blocks - 1) * secsz > parts * sizeof(struct gpt_ent)) {
115 blocks = (parts * sizeof(struct gpt_ent)) / secsz;
116 if ((parts * sizeof(struct gpt_ent)) % secsz)
118 blocks++; /* Don't forget the header itself */
121 last = mediasz / secsz - 1LL;
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 = 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 = offsetof(struct gpt_hdr, padding);
159 hdr->hdr_lba_self = gpt->map_start;
160 hdr->hdr_lba_alt = last;
161 hdr->hdr_lba_start = tbl->map_start + blocks;
162 hdr->hdr_lba_end = last - blocks - 1LL;
163 uuid_create(&hdr->hdr_uuid, NULL);
164 hdr->hdr_lba_table = tbl->map_start;
165 hdr->hdr_entries = (blocks * secsz) / sizeof(struct gpt_ent);
166 if (hdr->hdr_entries > parts)
167 hdr->hdr_entries = parts;
168 hdr->hdr_entsz = sizeof(struct gpt_ent);
171 for (i = 0; i < hdr->hdr_entries; i++)
172 uuid_create(&ent[i].ent_uuid, NULL);
174 hdr->hdr_crc_table = crc32(ent, hdr->hdr_entries * hdr->hdr_entsz);
175 hdr->hdr_crc_self = crc32(hdr, hdr->hdr_size);
181 * Create backup GPT if the user didn't suppress it.
184 tpg = map_add(last, 1LL, MAP_TYPE_SEC_GPT_HDR,
186 lbt = map_add(last - blocks, blocks, MAP_TYPE_SEC_GPT_TBL,
188 memcpy(tpg->map_data, gpt->map_data, secsz);
190 hdr->hdr_lba_self = tpg->map_start;
191 hdr->hdr_lba_alt = gpt->map_start;
192 hdr->hdr_lba_table = lbt->map_start;
193 hdr->hdr_crc_self = 0; /* Don't ever forget this! */
194 hdr->hdr_crc_self = crc32(hdr, hdr->hdr_size);
201 cmd_create(int argc, char *argv[])
205 while ((ch = getopt(argc, argv, "p")) != -1) {
218 while (optind < argc) {
219 fd = gpt_open(argv[optind++]);
221 warn("unable to open device '%s'", device_name);