Annotation of src/sbin/gpt/type.c, Revision 1.1
1.1 ! jnemeth 1: /*-
! 2: * Copyright (c) 2004 Marcel Moolenaar
! 3: * All rights reserved.
! 4: *
! 5: * Redistribution and use in source and binary forms, with or without
! 6: * modification, are permitted provided that the following conditions
! 7: * are met:
! 8: *
! 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.
! 14: *
! 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.
! 25: */
! 26:
! 27: #include <sys/cdefs.h>
! 28: #ifdef __FBSDID
! 29: __FBSDID("$FreeBSD: src/sbin/gpt/remove.c,v 1.10 2006/10/04 18:20:25 marcel Exp $");
! 30: #endif
! 31: #ifdef __RCSID
! 32: __RCSID("$NetBSD: remove.c,v 1.12 2013/11/28 01:37:14 jnemeth Exp $");
! 33: #endif
! 34:
! 35: #include <sys/types.h>
! 36:
! 37: #include <err.h>
! 38: #include <stddef.h>
! 39: #include <stdio.h>
! 40: #include <stdlib.h>
! 41: #include <string.h>
! 42: #include <unistd.h>
! 43:
! 44: #include "map.h"
! 45: #include "gpt.h"
! 46:
! 47: static int all;
! 48: static uuid_t type, newtype;
! 49: static off_t block, size;
! 50: static unsigned int entry;
! 51:
! 52: const char typemsg1[] = "type -a -T newtype device ...";
! 53: const char typemsg2[] = "type [-b blocknr] [-i index] [-s sectors] [-t type]";
! 54: const char typemsg3[] = " -T newtype device ...";
! 55:
! 56: __dead static void
! 57: usage_type(void)
! 58: {
! 59:
! 60: fprintf(stderr,
! 61: "usage: %s %s\n"
! 62: " %s %s\n"
! 63: " %*s %s\n", getprogname(), typemsg1,
! 64: getprogname(), typemsg2, (int)strlen(getprogname()), "", typemsg3);
! 65: exit(1);
! 66: }
! 67:
! 68: static void
! 69: chtype(int fd)
! 70: {
! 71: uuid_t uuid;
! 72: map_t *gpt, *tpg;
! 73: map_t *tbl, *lbt;
! 74: map_t *m;
! 75: struct gpt_hdr *hdr;
! 76: struct gpt_ent *ent;
! 77: unsigned int i;
! 78:
! 79: gpt = map_find(MAP_TYPE_PRI_GPT_HDR);
! 80: if (gpt == NULL) {
! 81: warnx("%s: error: no primary GPT header; run create or recover",
! 82: device_name);
! 83: return;
! 84: }
! 85:
! 86: tpg = map_find(MAP_TYPE_SEC_GPT_HDR);
! 87: if (tpg == NULL) {
! 88: warnx("%s: error: no secondary GPT header; run recover",
! 89: device_name);
! 90: return;
! 91: }
! 92:
! 93: tbl = map_find(MAP_TYPE_PRI_GPT_TBL);
! 94: lbt = map_find(MAP_TYPE_SEC_GPT_TBL);
! 95: if (tbl == NULL || lbt == NULL) {
! 96: warnx("%s: error: run recover -- trust me", device_name);
! 97: return;
! 98: }
! 99:
! 100: /* Change type of all matching entries in the map. */
! 101: for (m = map_first(); m != NULL; m = m->map_next) {
! 102: if (m->map_type != MAP_TYPE_GPT_PART || m->map_index < 1)
! 103: continue;
! 104: if (entry > 0 && entry != m->map_index)
! 105: continue;
! 106: if (block > 0 && block != m->map_start)
! 107: continue;
! 108: if (size > 0 && size != m->map_size)
! 109: continue;
! 110:
! 111: i = m->map_index - 1;
! 112:
! 113: hdr = gpt->map_data;
! 114: ent = (void*)((char*)tbl->map_data + i *
! 115: le32toh(hdr->hdr_entsz));
! 116: le_uuid_dec(ent->ent_type, &uuid);
! 117: if (!uuid_is_nil(&type, NULL) &&
! 118: !uuid_equal(&type, &uuid, NULL))
! 119: continue;
! 120:
! 121: /* Change the primary entry. */
! 122: le_uuid_enc(ent->ent_type, &newtype);
! 123:
! 124: hdr->hdr_crc_table = htole32(crc32(tbl->map_data,
! 125: le32toh(hdr->hdr_entries) * le32toh(hdr->hdr_entsz)));
! 126: hdr->hdr_crc_self = 0;
! 127: hdr->hdr_crc_self = htole32(crc32(hdr, le32toh(hdr->hdr_size)));
! 128:
! 129: gpt_write(fd, gpt);
! 130: gpt_write(fd, tbl);
! 131:
! 132: hdr = tpg->map_data;
! 133: ent = (void*)((char*)lbt->map_data + i *
! 134: le32toh(hdr->hdr_entsz));
! 135:
! 136: /* Change the secondary entry. */
! 137: le_uuid_enc(ent->ent_type, &newtype);
! 138:
! 139: hdr->hdr_crc_table = htole32(crc32(lbt->map_data,
! 140: le32toh(hdr->hdr_entries) * le32toh(hdr->hdr_entsz)));
! 141: hdr->hdr_crc_self = 0;
! 142: hdr->hdr_crc_self = htole32(crc32(hdr, le32toh(hdr->hdr_size)));
! 143:
! 144: gpt_write(fd, lbt);
! 145: gpt_write(fd, tpg);
! 146: printf("partition %d type changed\n", m->map_index);
! 147: }
! 148: }
! 149:
! 150: int
! 151: cmd_type(int argc, char *argv[])
! 152: {
! 153: char *p;
! 154: int ch, fd;
! 155: int64_t human_num;
! 156:
! 157: /* Get the type options */
! 158: while ((ch = getopt(argc, argv, "ab:i:s:t:T:")) != -1) {
! 159: switch(ch) {
! 160: case 'a':
! 161: if (all > 0)
! 162: usage_type();
! 163: all = 1;
! 164: break;
! 165: case 'b':
! 166: if (block > 0)
! 167: usage_type();
! 168: if (dehumanize_number(optarg, &human_num) < 0)
! 169: usage_type();
! 170: block = human_num;
! 171: if (block < 1)
! 172: usage_type();
! 173: break;
! 174: case 'i':
! 175: if (entry > 0)
! 176: usage_type();
! 177: entry = strtoul(optarg, &p, 10);
! 178: if (*p != 0 || entry < 1)
! 179: usage_type();
! 180: break;
! 181: case 's':
! 182: if (size > 0)
! 183: usage_type();
! 184: size = strtoll(optarg, &p, 10);
! 185: if (*p != 0 || size < 1)
! 186: usage_type();
! 187: break;
! 188: case 't':
! 189: if (!uuid_is_nil(&type, NULL))
! 190: usage_type();
! 191: if (parse_uuid(optarg, &type) != 0)
! 192: usage_type();
! 193: break;
! 194: case 'T':
! 195: if (!uuid_is_nil(&newtype, NULL))
! 196: usage_type();
! 197: if (parse_uuid(optarg, &newtype) != 0)
! 198: usage_type();
! 199: break;
! 200: default:
! 201: usage_type();
! 202: }
! 203: }
! 204:
! 205: if (!all ^
! 206: (block > 0 || entry > 0 || size > 0 || !uuid_is_nil(&type, NULL)))
! 207: usage_type();
! 208: if (uuid_is_nil(&newtype, NULL))
! 209: usage_type();
! 210:
! 211: if (argc == optind)
! 212: usage_type();
! 213:
! 214: while (optind < argc) {
! 215: fd = gpt_open(argv[optind++]);
! 216: if (fd == -1) {
! 217: warn("unable to open device '%s'", device_name);
! 218: continue;
! 219: }
! 220:
! 221: chtype(fd);
! 222:
! 223: gpt_close(fd);
! 224: }
! 225:
! 226: return (0);
! 227: }
CVSweb <webmaster@jp.NetBSD.org>
![[BACK]](/icons/back.gif){kind=icon}
Return to type.c CVS log ![[TXT]](/icons/text.gif){kind=icon}
![[DIR]](/icons/dir.gif){kind=icon}
Up to [cvs.NetBSD.org] / src / sbin / gpt