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__FBSDID("$FreeBSD: src/sbin/gpt/destroy.c,v 1.6.10.1 2010/02/10 00:26:20 kensmith Exp $");
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
42static int recoverable;
43
44static void
45usage_destroy(void)
46{
47
48	fprintf(stderr,
49	    "usage: %s [-r] device ...\n", getprogname());
50	exit(1);
51}
52
53static void
54destroy(int fd)
55{
56	map_t *pri_hdr, *sec_hdr;
57	map_t *pmbr;
58
59	pri_hdr = map_find(MAP_TYPE_PRI_GPT_HDR);
60	sec_hdr = map_find(MAP_TYPE_SEC_GPT_HDR);
61	pmbr = map_find(MAP_TYPE_PMBR);
62
63	if (pri_hdr == NULL && sec_hdr == NULL) {
64		warnx("%s: error: device doesn't contain a GPT", device_name);
65		return;
66	}
67
68	if (recoverable && sec_hdr == NULL) {
69		warnx("%s: error: recoverability not possible", device_name);
70		return;
71	}
72
73	if (pri_hdr != NULL) {
74		bzero(pri_hdr->map_data, secsz);
75		gpt_write(fd, pri_hdr);
76	}
77
78	if (!recoverable && sec_hdr != NULL) {
79		bzero(sec_hdr->map_data, secsz);
80		gpt_write(fd, sec_hdr);
81	}
82
83	if (!recoverable && pmbr != NULL) {
84		bzero(pmbr->map_data, secsz);
85		gpt_write(fd, pmbr);
86	}
87}
88
89int
90cmd_destroy(int argc, char *argv[])
91{
92	int ch, fd;
93
94	while ((ch = getopt(argc, argv, "r")) != -1) {
95		switch(ch) {
96		case 'r':
97			recoverable = 1;
98			break;
99		default:
100			usage_destroy();
101		}
102	}
103
104	if (argc == optind)
105		usage_destroy();
106
107	while (optind < argc) {
108		fd = gpt_open(argv[optind++]);
109		if (fd == -1) {
110			warn("unable to open device '%s'", device_name);
111			return (1);
112		}
113
114		destroy(fd);
115
116		gpt_close(fd);
117	}
118
119	return (0);
120}
121