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 * $FreeBSD: src/sbin/gpt/destroy.c,v 1.6 2005/08/31 01:47:19 marcel Exp $
27 * $DragonFly: src/sbin/gpt/destroy.c,v 1.1 2007/06/16 22:29:27 dillon Exp $
28 */
29
30 #include <sys/types.h>
31
32 #include <err.h>
33 #include <stddef.h>
34 #include <stdio.h>
35 #include <stdlib.h>
36 #include <string.h>
37 #include <unistd.h>
38
39 #include "map.h"
40 #include "gpt.h"
41
42 static int recoverable;
43 static int quiet;
44
45 static void
usage_destroy(void)46 usage_destroy(void)
47 {
48 fprintf(stderr,
49 "usage: %s [-r] device ...\n", getprogname());
50 exit(1);
51 }
52
53 static void
destroy(int fd)54 destroy(int fd)
55 {
56 map_t *pri_hdr, *sec_hdr;
57
58 pri_hdr = map_find(MAP_TYPE_PRI_GPT_HDR);
59 sec_hdr = map_find(MAP_TYPE_SEC_GPT_HDR);
60
61 if (pri_hdr == NULL && sec_hdr == NULL) {
62 if (quiet == 0) {
63 warnx("%s: error: device doesn't contain a GPT",
64 device_name);
65 }
66 return;
67 }
68
69 if (recoverable && sec_hdr == NULL) {
70 warnx("%s: error: recoverability not possible", device_name);
71 return;
72 }
73
74 if (pri_hdr != NULL) {
75 bzero(pri_hdr->map_data, secsz);
76 gpt_write(fd, pri_hdr);
77 }
78
79 if (!recoverable && sec_hdr != NULL) {
80 bzero(sec_hdr->map_data, secsz);
81 gpt_write(fd, sec_hdr);
82 }
83 }
84
85 int
cmd_destroy(int argc,char * argv[])86 cmd_destroy(int argc, char *argv[])
87 {
88 int ch, fd;
89
90 while ((ch = getopt(argc, argv, "r")) != -1) {
91 switch(ch) {
92 case 'r':
93 recoverable = 1;
94 break;
95 default:
96 usage_destroy();
97 }
98 }
99
100 if (argc == optind)
101 usage_destroy();
102
103 while (optind < argc) {
104 fd = gpt_open(argv[optind++]);
105 if (fd == -1) {
106 warn("unable to open device '%s'", device_name);
107 continue;
108 }
109
110 destroy(fd);
111
112 gpt_close(fd);
113 }
114
115 return (0);
116 }
117
118 void
do_destroy(int fd)119 do_destroy(int fd)
120 {
121 quiet = 1;
122 destroy(fd);
123 quiet = 0;
124 }
125