xref: /netbsd-src/sbin/dkscan_bsdlabel/dkscan_bsdlabel.c (revision 171fa94db7c608ce8d92e208fa9ddbacab27f725)
1 /* $NetBSD: dkscan_bsdlabel.c,v 1.5 2020/04/11 17:21:16 jdolecek Exp $ */
2 
3 /*-
4  * Copyright (c) 2007 The NetBSD Foundation, Inc.
5  * All rights reserved.
6  *
7  * This code is derived from software contributed to The NetBSD Foundation
8  * by Martin Husemann <martin@NetBSD.org>.
9  *
10  * Redistribution and use in source and binary forms, with or without
11  * modification, are permitted provided that the following conditions
12  * are met:
13  * 1. Redistributions of source code must retain the above copyright
14  *    notice, this list of conditions and the following disclaimer.
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in the
17  *    documentation and/or other materials provided with the distribution.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29  * POSSIBILITY OF SUCH DAMAGE.
30  */
31 
32 #include <stdio.h>
33 #include <stdlib.h>
34 #include <stddef.h>
35 #include <string.h>
36 #include <unistd.h>
37 #include <fcntl.h>
38 #include <err.h>
39 #include <util.h>
40 #include <sys/disk.h>
41 #include <sys/buf.h>
42 
43 #include "dkscan_util.h"
44 
45 struct disk {
46 	const char	*dk_name;	/* disk name */
47 	int		dk_blkshift;	/* shift to convert DEV_BSIZE to blks */
48 };
49 
50 static struct buf *
geteblk(int size)51 geteblk(int size)
52 {
53 	struct buf *bp = malloc(sizeof(*bp) + size);
54 
55 	bp->b_data = (void *)&bp[1];
56 
57 	return bp;
58 }
59 
60 static void
brelse(struct buf * bp,int set)61 brelse(struct buf *bp, int set)
62 {
63 	free(bp);
64 }
65 
66 #include "dkwedge_bsdlabel.c"
67 
68 __dead static void usage(void);
69 
70 int
main(int argc,char ** argv)71 main(int argc, char **argv)
72 {
73 	struct disk d;
74 	int ch;
75 	char buf[PATH_MAX];
76 	const char *devpart;
77 
78 	if (argc < 2)
79 		usage();
80 
81 	while ((ch = getopt(argc, argv, "nv")) != -1) {
82 		switch (ch) {
83 		case 'n':
84 			no_action = 1;
85 			break;
86 		case 'v':
87 			verbose++;
88 			break;
89 		default:
90 			usage();
91 		}
92 	}
93 	if (optind >= argc)
94 		usage();
95 
96 	disk_fd = opendisk(argv[optind], O_RDWR, buf, PATH_MAX, 0);
97 	if (disk_fd == -1)
98 		err(1, "%s", argv[optind]);
99 
100 	devpart = strrchr(argv[optind], '/');
101 	if (devpart == NULL)
102 		devpart = argv[optind];
103 
104 	memset(&d, 0, sizeof(d));
105 	d.dk_name  = devpart;
106 	dkwedge_discover_bsdlabel(&d, NULL);
107 
108 	close(disk_fd);
109 	return 0;
110 }
111 
112 void
usage(void)113 usage(void)
114 {
115 	fprintf(stderr, "usage: %s [-vn] <diskname>\n"
116 	    "  where\n"
117 	    "    -n don't change anything, just print info\n"
118 	    "    -v be more verbose\n"
119 	    "    <diskname> device to scan\n",
120 	    getprogname());
121 	exit(1);
122 }
123