1/*- 2 * Copyright (c) 2002 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/recover.c,v 1.8 2005/08/31 01:47:19 marcel Exp $"); 30#endif 31#ifdef __RCSID 32__RCSID("$NetBSD: recover.c,v 1.3 2007/12/18 21:46:47 riz 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 47static int recoverable; 48 49const char recovermsg[] = "recover device ..."; 50 51__dead static void 52usage_recover(void) 53{ 54 55 fprintf(stderr, 56 "usage: %s %s\n", getprogname(), recovermsg); 57 exit(1); 58} 59 60static void 61recover(int fd) 62{ 63 off_t last; 64 map_t *gpt, *tpg; 65 map_t *tbl, *lbt; 66 struct gpt_hdr *hdr; 67 68 if (map_find(MAP_TYPE_MBR) != NULL) { 69 warnx("%s: error: device contains a MBR", device_name); 70 return; 71 } 72 73 gpt = map_find(MAP_TYPE_PRI_GPT_HDR); 74 tpg = map_find(MAP_TYPE_SEC_GPT_HDR); 75 tbl = map_find(MAP_TYPE_PRI_GPT_TBL); 76 lbt = map_find(MAP_TYPE_SEC_GPT_TBL); 77 78 if (gpt == NULL && tpg == NULL) { 79 warnx("%s: no primary or secondary GPT headers, can't recover", 80 device_name); 81 return; 82 } 83 if (tbl == NULL && lbt == NULL) { 84 warnx("%s: no primary or secondary GPT tables, can't recover", 85 device_name); 86 return; 87 } 88 89 last = mediasz / secsz - 1LL; 90 91 if (tbl != NULL && lbt == NULL) { 92 lbt = map_add(last - tbl->map_size, tbl->map_size, 93 MAP_TYPE_SEC_GPT_TBL, tbl->map_data); 94 if (lbt == NULL) { 95 warnx("%s: adding secondary GPT table failed", 96 device_name); 97 return; 98 } 99 gpt_write(fd, lbt); 100 warnx("%s: recovered secondary GPT table from primary", 101 device_name); 102 } else if (tbl == NULL && lbt != NULL) { 103 tbl = map_add(2LL, lbt->map_size, MAP_TYPE_PRI_GPT_TBL, 104 lbt->map_data); 105 if (tbl == NULL) { 106 warnx("%s: adding primary GPT table failed", 107 device_name); 108 return; 109 } 110 gpt_write(fd, tbl); 111 warnx("%s: recovered primary GPT table from secondary", 112 device_name); 113 } 114 115 if (gpt != NULL && tpg == NULL) { 116 tpg = map_add(last, 1LL, MAP_TYPE_SEC_GPT_HDR, 117 calloc(1, secsz)); 118 if (tpg == NULL) { 119 warnx("%s: adding secondary GPT header failed", 120 device_name); 121 return; 122 } 123 memcpy(tpg->map_data, gpt->map_data, secsz); 124 hdr = tpg->map_data; 125 hdr->hdr_lba_self = htole64(tpg->map_start); 126 hdr->hdr_lba_alt = htole64(gpt->map_start); 127 hdr->hdr_lba_table = htole64(lbt->map_start); 128 hdr->hdr_crc_self = 0; 129 hdr->hdr_crc_self = htole32(crc32(hdr, le32toh(hdr->hdr_size))); 130 gpt_write(fd, tpg); 131 warnx("%s: recovered secondary GPT header from primary", 132 device_name); 133 } else if (gpt == NULL && tpg != NULL) { 134 gpt = map_add(1LL, 1LL, MAP_TYPE_PRI_GPT_HDR, 135 calloc(1, secsz)); 136 if (gpt == NULL) { 137 warnx("%s: adding primary GPT header failed", 138 device_name); 139 return; 140 } 141 memcpy(gpt->map_data, tpg->map_data, secsz); 142 hdr = gpt->map_data; 143 hdr->hdr_lba_self = htole64(gpt->map_start); 144 hdr->hdr_lba_alt = htole64(tpg->map_start); 145 hdr->hdr_lba_table = htole64(tbl->map_start); 146 hdr->hdr_crc_self = 0; 147 hdr->hdr_crc_self = htole32(crc32(hdr, le32toh(hdr->hdr_size))); 148 gpt_write(fd, gpt); 149 warnx("%s: recovered primary GPT header from secondary", 150 device_name); 151 } 152} 153 154int 155cmd_recover(int argc, char *argv[]) 156{ 157 int ch, fd; 158 159 while ((ch = getopt(argc, argv, "r")) != -1) { 160 switch(ch) { 161 case 'r': 162 recoverable = 1; 163 break; 164 default: 165 usage_recover(); 166 } 167 } 168 169 if (argc == optind) 170 usage_recover(); 171 172 while (optind < argc) { 173 fd = gpt_open(argv[optind++]); 174 if (fd == -1) { 175 warn("unable to open device '%s'", device_name); 176 continue; 177 } 178 179 recover(fd); 180 181 gpt_close(fd); 182 } 183 184 return (0); 185} 186