1 /* $NetBSD: lvmdiskscan.c,v 1.1.1.1 2008/12/22 00:19:05 haad Exp $ */
2
3 /*
4 * Copyright (C) 2002-2004 Sistina Software, Inc. All rights reserved.
5 * Copyright (C) 2004-2007 Red Hat, Inc. All rights reserved.
6 *
7 * This file is part of LVM2.
8 *
9 * This copyrighted material is made available to anyone wishing to use,
10 * modify, copy, or redistribute it subject to the terms and conditions
11 * of the GNU Lesser General Public License v.2.1.
12 *
13 * You should have received a copy of the GNU Lesser General Public License
14 * along with this program; if not, write to the Free Software Foundation,
15 * Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
16 */
17
18 /*
19 * Changelog
20 *
21 * 05/02/2002 - First drop [HM]
22 */
23
24 #include "tools.h"
25
26 int disks_found;
27 int parts_found;
28 int pv_disks_found;
29 int pv_parts_found;
30 int max_len;
31
_get_max_dev_name_len(struct dev_filter * filter)32 static int _get_max_dev_name_len(struct dev_filter *filter)
33 {
34 int len = 0;
35 int maxlen = 0;
36 struct dev_iter *iter;
37 struct device *dev;
38
39 if (!(iter = dev_iter_create(filter, 1))) {
40 log_error("dev_iter_create failed");
41 return 0;
42 }
43
44 /* Do scan */
45 for (dev = dev_iter_get(iter); dev; dev = dev_iter_get(iter)) {
46 len = strlen(dev_name(dev));
47 if (len > maxlen)
48 maxlen = len;
49 }
50 dev_iter_destroy(iter);
51
52 return maxlen;
53 }
54
_count(struct device * dev,int * disks,int * parts)55 static void _count(struct device *dev, int *disks, int *parts)
56 {
57 int c = dev_name(dev)[strlen(dev_name(dev)) - 1];
58
59 if (!isdigit(c))
60 (*disks)++;
61 else
62 (*parts)++;
63 }
64
_print(struct cmd_context * cmd,const struct device * dev,uint64_t size,const char * what)65 static void _print(struct cmd_context *cmd, const struct device *dev,
66 uint64_t size, const char *what)
67 {
68 log_print("%-*s [%15s] %s", max_len, dev_name(dev),
69 display_size(cmd, size), what ? : "");
70 }
71
_check_device(struct cmd_context * cmd,struct device * dev)72 static int _check_device(struct cmd_context *cmd, struct device *dev)
73 {
74 char buffer;
75 uint64_t size;
76
77 if (!dev_open(dev)) {
78 return 0;
79 }
80 if (!dev_read(dev, UINT64_C(0), (size_t) 1, &buffer)) {
81 dev_close(dev);
82 return 0;
83 }
84 if (!dev_get_size(dev, &size)) {
85 log_error("Couldn't get size of \"%s\"", dev_name(dev));
86 }
87 _print(cmd, dev, size, NULL);
88 _count(dev, &disks_found, &parts_found);
89 if (!dev_close(dev)) {
90 log_error("dev_close on \"%s\" failed", dev_name(dev));
91 return 0;
92 }
93 return 1;
94 }
95
lvmdiskscan(struct cmd_context * cmd,int argc __attribute ((unused)),char ** argv __attribute ((unused)))96 int lvmdiskscan(struct cmd_context *cmd, int argc __attribute((unused)),
97 char **argv __attribute((unused)))
98 {
99 uint64_t size;
100 struct dev_iter *iter;
101 struct device *dev;
102 struct label *label;
103
104 /* initialise these here to avoid problems with the lvm shell */
105 disks_found = 0;
106 parts_found = 0;
107 pv_disks_found = 0;
108 pv_parts_found = 0;
109
110 if (arg_count(cmd, lvmpartition_ARG))
111 log_warn("WARNING: only considering LVM devices");
112
113 max_len = _get_max_dev_name_len(cmd->filter);
114
115 if (!(iter = dev_iter_create(cmd->filter, 0))) {
116 log_error("dev_iter_create failed");
117 return ECMD_FAILED;
118 }
119
120 /* Do scan */
121 for (dev = dev_iter_get(iter); dev; dev = dev_iter_get(iter)) {
122 /* Try if it is a PV first */
123 if ((label_read(dev, &label, UINT64_C(0)))) {
124 if (!dev_get_size(dev, &size)) {
125 log_error("Couldn't get size of \"%s\"",
126 dev_name(dev));
127 continue;
128 }
129 _print(cmd, dev, size, "LVM physical volume");
130 _count(dev, &pv_disks_found, &pv_parts_found);
131 continue;
132 }
133 /* If user just wants PVs we are done */
134 if (arg_count(cmd, lvmpartition_ARG))
135 continue;
136
137 /* What other device is it? */
138 if (!_check_device(cmd, dev))
139 continue;
140 }
141 dev_iter_destroy(iter);
142
143 /* Display totals */
144 if (!arg_count(cmd, lvmpartition_ARG)) {
145 log_print("%d disk%s",
146 disks_found, disks_found == 1 ? "" : "s");
147 log_print("%d partition%s",
148 parts_found, parts_found == 1 ? "" : "s");
149 }
150 log_print("%d LVM physical volume whole disk%s",
151 pv_disks_found, pv_disks_found == 1 ? "" : "s");
152 log_print("%d LVM physical volume%s",
153 pv_parts_found, pv_parts_found == 1 ? "" : "s");
154
155 return ECMD_PROCESSED;
156 }
157