]> git.ipfire.org Git - thirdparty/util-linux.git/blame - misc-utils/lsblk.c
lsblk: check ul_path_scanf() return value [coverity scan]
[thirdparty/util-linux.git] / misc-utils / lsblk.c
CommitLineData
2a4c734b 1/*
f77fa578 2 * lsblk(8) - list block devices
2a4c734b 3 *
b93585e8 4 * Copyright (C) 2010-2018 Red Hat, Inc. All rights reserved.
2a4c734b
MB
5 * Written by Milan Broz <mbroz@redhat.com>
6 * Karel Zak <kzak@redhat.com>
7 *
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 2 of the License, or
11 * (at your option) any later version.
12 *
13 * This program is distributed in the hope that it would be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
17 *
7cebf0bb
SK
18 * You should have received a copy of the GNU General Public License along
19 * with this program; if not, write to the Free Software Foundation, Inc.,
20 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
2a4c734b 21 */
2a4c734b
MB
22#include <stdio.h>
23#include <errno.h>
24#include <getopt.h>
2a4c734b
MB
25#include <stdlib.h>
26#include <unistd.h>
2a4c734b
MB
27#include <sys/types.h>
28#include <sys/stat.h>
29#include <dirent.h>
30#include <fcntl.h>
31#include <string.h>
2a4c734b 32#include <sys/ioctl.h>
2a4c734b
MB
33#include <stdarg.h>
34#include <locale.h>
35#include <pwd.h>
36#include <grp.h>
18eac5ba 37#include <ctype.h>
ccafadb7 38#include <assert.h>
2a4c734b
MB
39
40#include <blkid.h>
2a4c734b 41
540d506f 42#include "c.h"
2a4c734b
MB
43#include "pathnames.h"
44#include "blkdev.h"
45#include "canonicalize.h"
2a4c734b 46#include "nls.h"
2a4c734b
MB
47#include "xalloc.h"
48#include "strutils.h"
766ab9b0 49#include "sysfs.h"
c05a80ca 50#include "closestream.h"
05187653 51#include "optutils.h"
7761bd3b 52#include "fileutils.h"
2a4c734b 53
14560b7f 54#include "lsblk.h"
7e786eca 55
14560b7f 56UL_DEBUG_DEFINE_MASK(lsblk);
7e786eca
KZ
57UL_DEBUG_DEFINE_MASKNAMES(lsblk) = UL_DEBUG_EMPTY_MASKNAMES;
58
1539750f
KZ
59#define LSBLK_EXIT_SOMEOK 64
60#define LSBLK_EXIT_ALLFAILED 32
61
be685b98
KZ
62static int column_id_to_number(int id);
63
2a4c734b
MB
64/* column IDs */
65enum {
66 COL_NAME = 0,
67 COL_KNAME,
66638b0b 68 COL_PATH,
2a4c734b 69 COL_MAJMIN,
bc405d64
KZ
70 COL_FSAVAIL,
71 COL_FSSIZE,
2a4c734b 72 COL_FSTYPE,
bc405d64
KZ
73 COL_FSUSED,
74 COL_FSUSEPERC,
2a4c734b
MB
75 COL_TARGET,
76 COL_LABEL,
77 COL_UUID,
1b06b33d
MB
78 COL_PTUUID,
79 COL_PTTYPE,
4686ba17 80 COL_PARTTYPE,
aa0903e0
KZ
81 COL_PARTLABEL,
82 COL_PARTUUID,
5a2fd932 83 COL_PARTFLAGS,
150db7a9 84 COL_RA,
2a4c734b 85 COL_RO,
627970a7 86 COL_RM,
483987c2 87 COL_HOTPLUG,
2a4c734b 88 COL_MODEL,
460c7afb 89 COL_SERIAL,
2a4c734b 90 COL_SIZE,
01e487c0 91 COL_STATE,
2a4c734b
MB
92 COL_OWNER,
93 COL_GROUP,
94 COL_MODE,
95 COL_ALIOFF,
96 COL_MINIO,
97 COL_OPTIO,
98 COL_PHYSEC,
99 COL_LOGSEC,
100 COL_ROTA,
101 COL_SCHED,
64c9e22a 102 COL_RQ_SIZE,
18eac5ba 103 COL_TYPE,
2d232246
MP
104 COL_DALIGN,
105 COL_DGRAN,
106 COL_DMAX,
107 COL_DZERO,
2adb1a44 108 COL_WSAME,
88ca32b3 109 COL_WWN,
12b06c3d 110 COL_RAND,
310c0603 111 COL_PKNAME,
699e5c4f 112 COL_HCTL,
a5fb4d23 113 COL_TRANSPORT,
7f14ee1b 114 COL_SUBSYS,
d6681cee 115 COL_REV,
f2df4365
DLM
116 COL_VENDOR,
117 COL_ZONED,
2a4c734b
MB
118};
119
9bd4e5c0
OO
120/* basic table settings */
121enum {
122 LSBLK_ASCII = (1 << 0),
123 LSBLK_RAW = (1 << 1),
124 LSBLK_NOHEADINGS = (1 << 2),
bb6e822a
KZ
125 LSBLK_EXPORT = (1 << 3),
126 LSBLK_TREE = (1 << 4),
4a102a48 127 LSBLK_JSON = (1 << 5),
9bd4e5c0
OO
128};
129
407e0ea6 130/* Types used for qsort() and JSON */
642048e4 131enum {
407e0ea6
KZ
132 COLTYPE_STR = 0, /* default */
133 COLTYPE_NUM = 1, /* always u64 number */
134 COLTYPE_SORTNUM = 2, /* string on output, u64 for qsort() */
135 COLTYPE_SIZE = 3, /* srring by default, number when --bytes */
136 COLTYPE_BOOL = 4 /* 0 or 1 */
642048e4
KZ
137};
138
2a4c734b
MB
139/* column names */
140struct colinfo {
141 const char *name; /* header */
142 double whint; /* width hint (N < 1 is in percent of termwidth) */
9bd4e5c0 143 int flags; /* SCOLS_FL_* */
2a4c734b 144 const char *help;
407e0ea6 145 int type; /* COLTYPE_* */
2a4c734b
MB
146};
147
148/* columns descriptions */
507d39c2 149static struct colinfo infos[] = {
b9c088f2 150 [COL_NAME] = { "NAME", 0.25, SCOLS_FL_NOEXTREMES, N_("device name") },
e22d8b95 151 [COL_KNAME] = { "KNAME", 0.3, 0, N_("internal kernel device name") },
bc405d64 152 [COL_PKNAME] = { "PKNAME", 0.3, 0, N_("internal parent kernel device name") },
66638b0b 153 [COL_PATH] = { "PATH", 0.3, 0, N_("path to the device node") },
407e0ea6 154 [COL_MAJMIN] = { "MAJ:MIN", 6, 0, N_("major:minor device number"), COLTYPE_SORTNUM },
bc405d64
KZ
155
156 [COL_FSAVAIL] = { "FSAVAIL", 5, SCOLS_FL_RIGHT, N_("filesystem size available") },
157 [COL_FSSIZE] = { "FSSIZE", 5, SCOLS_FL_RIGHT, N_("filesystem size") },
158 [COL_FSTYPE] = { "FSTYPE", 0.1, SCOLS_FL_TRUNC, N_("filesystem type") },
159 [COL_FSUSED] = { "FSUSED", 5, SCOLS_FL_RIGHT, N_("filesystem size used") },
160 [COL_FSUSEPERC] = { "FSUSE%", 3, SCOLS_FL_RIGHT, N_("filesystem use percentage") },
161
9bd4e5c0 162 [COL_TARGET] = { "MOUNTPOINT", 0.10, SCOLS_FL_TRUNC, N_("where the device is mounted") },
2a4c734b
MB
163 [COL_LABEL] = { "LABEL", 0.1, 0, N_("filesystem LABEL") },
164 [COL_UUID] = { "UUID", 36, 0, N_("filesystem UUID") },
aa0903e0 165
1b06b33d
MB
166 [COL_PTUUID] = { "PTUUID", 36, 0, N_("partition table identifier (usually UUID)") },
167 [COL_PTTYPE] = { "PTTYPE", 0.1, 0, N_("partition table type") },
168
4686ba17 169 [COL_PARTTYPE] = { "PARTTYPE", 36, 0, N_("partition type UUID") },
aa0903e0
KZ
170 [COL_PARTLABEL] = { "PARTLABEL", 0.1, 0, N_("partition LABEL") },
171 [COL_PARTUUID] = { "PARTUUID", 36, 0, N_("partition UUID") },
5a2fd932 172 [COL_PARTFLAGS] = { "PARTFLAGS", 36, 0, N_("partition flags") },
aa0903e0 173
407e0ea6
KZ
174 [COL_RA] = { "RA", 3, SCOLS_FL_RIGHT, N_("read-ahead of the device"), COLTYPE_NUM },
175 [COL_RO] = { "RO", 1, SCOLS_FL_RIGHT, N_("read-only device"), COLTYPE_BOOL },
176 [COL_RM] = { "RM", 1, SCOLS_FL_RIGHT, N_("removable device"), COLTYPE_BOOL },
177 [COL_HOTPLUG]= { "HOTPLUG", 1, SCOLS_FL_RIGHT, N_("removable or hotplug device (usb, pcmcia, ...)"), COLTYPE_BOOL },
178 [COL_ROTA] = { "ROTA", 1, SCOLS_FL_RIGHT, N_("rotational device"), COLTYPE_BOOL },
179 [COL_RAND] = { "RAND", 1, SCOLS_FL_RIGHT, N_("adds randomness"), COLTYPE_BOOL },
9bd4e5c0
OO
180 [COL_MODEL] = { "MODEL", 0.1, SCOLS_FL_TRUNC, N_("device identifier") },
181 [COL_SERIAL] = { "SERIAL", 0.1, SCOLS_FL_TRUNC, N_("disk serial number") },
407e0ea6 182 [COL_SIZE] = { "SIZE", 5, SCOLS_FL_RIGHT, N_("size of the device"), COLTYPE_SIZE },
9bd4e5c0
OO
183 [COL_STATE] = { "STATE", 7, SCOLS_FL_TRUNC, N_("state of the device") },
184 [COL_OWNER] = { "OWNER", 0.1, SCOLS_FL_TRUNC, N_("user name"), },
185 [COL_GROUP] = { "GROUP", 0.1, SCOLS_FL_TRUNC, N_("group name") },
2a4c734b 186 [COL_MODE] = { "MODE", 10, 0, N_("device node permissions") },
407e0ea6
KZ
187 [COL_ALIOFF] = { "ALIGNMENT", 6, SCOLS_FL_RIGHT, N_("alignment offset"), COLTYPE_NUM },
188 [COL_MINIO] = { "MIN-IO", 6, SCOLS_FL_RIGHT, N_("minimum I/O size"), COLTYPE_NUM },
189 [COL_OPTIO] = { "OPT-IO", 6, SCOLS_FL_RIGHT, N_("optimal I/O size"), COLTYPE_NUM },
190 [COL_PHYSEC] = { "PHY-SEC", 7, SCOLS_FL_RIGHT, N_("physical sector size"), COLTYPE_NUM },
191 [COL_LOGSEC] = { "LOG-SEC", 7, SCOLS_FL_RIGHT, N_("logical sector size"), COLTYPE_NUM },
18eac5ba 192 [COL_SCHED] = { "SCHED", 0.1, 0, N_("I/O scheduler name") },
407e0ea6 193 [COL_RQ_SIZE]= { "RQ-SIZE", 5, SCOLS_FL_RIGHT, N_("request queue size"), COLTYPE_NUM },
2d232246 194 [COL_TYPE] = { "TYPE", 4, 0, N_("device type") },
407e0ea6
KZ
195 [COL_DALIGN] = { "DISC-ALN", 6, SCOLS_FL_RIGHT, N_("discard alignment offset"), COLTYPE_NUM },
196 [COL_DGRAN] = { "DISC-GRAN", 6, SCOLS_FL_RIGHT, N_("discard granularity"), COLTYPE_SIZE },
197 [COL_DMAX] = { "DISC-MAX", 6, SCOLS_FL_RIGHT, N_("discard max bytes"), COLTYPE_SIZE },
198 [COL_DZERO] = { "DISC-ZERO", 1, SCOLS_FL_RIGHT, N_("discard zeroes data"), COLTYPE_BOOL },
199 [COL_WSAME] = { "WSAME", 6, SCOLS_FL_RIGHT, N_("write same max bytes"), COLTYPE_SIZE },
88ca32b3 200 [COL_WWN] = { "WWN", 18, 0, N_("unique storage identifier") },
699e5c4f 201 [COL_HCTL] = { "HCTL", 10, 0, N_("Host:Channel:Target:Lun for SCSI") },
a5fb4d23 202 [COL_TRANSPORT] = { "TRAN", 6, 0, N_("device transport type") },
7f14ee1b 203 [COL_SUBSYS] = { "SUBSYSTEMS", 0.1, SCOLS_FL_NOEXTREMES, N_("de-duplicated chain of subsystems") },
9bd4e5c0
OO
204 [COL_REV] = { "REV", 4, SCOLS_FL_RIGHT, N_("device revision") },
205 [COL_VENDOR] = { "VENDOR", 0.1, SCOLS_FL_TRUNC, N_("device vendor") },
f2df4365 206 [COL_ZONED] = { "ZONED", 0.3, 0, N_("zone model") },
2a4c734b
MB
207};
208
14560b7f 209struct lsblk *lsblk; /* global handler */
507d39c2 210
b93585e8
KZ
211/*
212 * columns[] array specifies all currently wanted output column. The columns
b6327c6f
KZ
213 * are defined by infos[] array and you can specify (on command line) each
214 * column twice. That's enough, dynamically allocated array of the columns is
b93585e8
KZ
215 * unnecessary overkill and over-engineering in this case
216 */
b6327c6f 217static int columns[ARRAY_SIZE(infos) * 2];
40b17508 218static size_t ncolumns;
b6327c6f 219
be685b98 220static inline void add_column(int id)
b6327c6f 221{
be685b98 222 if (ncolumns >= ARRAY_SIZE(columns))
b6327c6f 223 errx(EXIT_FAILURE, _("too many columns specified, "
1d231190 224 "the limit is %zu columns"),
be685b98
KZ
225 ARRAY_SIZE(columns) - 1);
226 columns[ ncolumns++ ] = id;
b6327c6f
KZ
227}
228
be685b98
KZ
229static inline void add_uniq_column(int id)
230{
231 if (column_id_to_number(id) < 0)
232 add_column(id);
233}
2a4c734b 234
b93585e8
KZ
235static void lsblk_init_debug(void)
236{
237 __UL_INIT_DEBUG_FROM_ENV(lsblk, LSBLK_DEBUG_, 0, LSBLK_DEBUG);
238}
239
240/*
241 * exclude/include devices filter based on major device numbers
242 */
f6da50d4
BV
243static int excludes[256];
244static size_t nexcludes;
2a4c734b 245
2ef4e8ba
KZ
246static int includes[256];
247static size_t nincludes;
248
2a4c734b
MB
249static int is_maj_excluded(int maj)
250{
6aace32f 251 size_t i;
2a4c734b
MB
252
253 assert(ARRAY_SIZE(excludes) > nexcludes);
254
2ef4e8ba 255 if (!nexcludes)
9e930041 256 return 0; /* filter not enabled, device not excluded */
2ef4e8ba 257
7e786eca
KZ
258 for (i = 0; i < nexcludes; i++) {
259 if (excludes[i] == maj) {
260 DBG(FILTER, ul_debug("exclude: maj=%d", maj));
2a4c734b 261 return 1;
7e786eca
KZ
262 }
263 }
2a4c734b
MB
264 return 0;
265}
266
2ef4e8ba
KZ
267static int is_maj_included(int maj)
268{
269 size_t i;
270
271 assert(ARRAY_SIZE(includes) > nincludes);
272
273 if (!nincludes)
274 return 1; /* filter not enabled, device is included */
275
7e786eca
KZ
276 for (i = 0; i < nincludes; i++) {
277 if (includes[i] == maj) {
278 DBG(FILTER, ul_debug("include: maj=%d", maj));
2ef4e8ba 279 return 1;
7e786eca
KZ
280 }
281 }
2ef4e8ba
KZ
282 return 0;
283}
284
b93585e8 285/* Converts column sequential number to column ID (COL_*) */
2a4c734b
MB
286static int get_column_id(int num)
287{
40b17508
KZ
288 assert(num >= 0);
289 assert((size_t) num < ncolumns);
b6327c6f 290 assert(columns[num] < (int) ARRAY_SIZE(infos));
2a4c734b
MB
291 return columns[num];
292}
293
b93585e8 294/* Returns column description for the column sequential number */
2a4c734b
MB
295static struct colinfo *get_column_info(int num)
296{
297 return &infos[ get_column_id(num) ];
298}
299
b93585e8 300/* Converts column name (as defined in the infos[] to the column ID */
2a4c734b
MB
301static int column_name_to_id(const char *name, size_t namesz)
302{
507d39c2 303 size_t i;
2a4c734b 304
03df0d11 305 for (i = 0; i < ARRAY_SIZE(infos); i++) {
2a4c734b
MB
306 const char *cn = infos[i].name;
307
308 if (!strncasecmp(name, cn, namesz) && !*(cn + namesz))
309 return i;
310 }
311 warnx(_("unknown column: %s"), name);
312 return -1;
313}
314
b93585e8 315/* Converts column ID (COL_*) to column sequential number */
642048e4
KZ
316static int column_id_to_number(int id)
317{
318 size_t i;
319
40b17508 320 for (i = 0; i < ncolumns; i++)
642048e4
KZ
321 if (columns[i] == id)
322 return i;
323 return -1;
324}
325
b93585e8 326/* Checks for DM prefix in the device name */
2a4c734b
MB
327static int is_dm(const char *name)
328{
329 return strncmp(name, "dm-", 3) ? 0 : 1;
330}
331
b158bd29 332/* Returns full pat to the device node (TODO: what about sysfs_blkdev_get_path()) */
fed34a1e 333static char *get_device_path(struct lsblk_device *dev)
2a4c734b
MB
334{
335 char path[PATH_MAX];
336
fed34a1e
KZ
337 assert(dev);
338 assert(dev->name);
2a4c734b 339
fed34a1e
KZ
340 if (is_dm(dev->name))
341 return __canonicalize_dm_name(lsblk->sysroot, dev->name);
2a4c734b 342
fed34a1e 343 snprintf(path, sizeof(path), "/dev/%s", dev->name);
92441d5c 344 sysfs_devname_sys_to_dev(path);
2a4c734b
MB
345 return xstrdup(path);
346}
347
fed34a1e 348static int is_readonly_device(struct lsblk_device *dev)
2a4c734b
MB
349{
350 int fd, ro = 0;
351
fed34a1e 352 if (ul_path_scanf(dev->sysfs, "ro", "%d", &ro) == 1)
2a4c734b
MB
353 return ro;
354
355 /* fallback if "ro" attribute does not exist */
fed34a1e 356 fd = open(dev->filename, O_RDONLY);
2a4c734b 357 if (fd != -1) {
9c53a49c
KZ
358 if (ioctl(fd, BLKROGET, &ro) != 0)
359 ro = 0;
2a4c734b
MB
360 close(fd);
361 }
362 return ro;
363}
364
fed34a1e 365static char *get_scheduler(struct lsblk_device *dev)
2a4c734b 366{
f153614c 367 char buf[128];
2a4c734b
MB
368 char *p, *res = NULL;
369
fed34a1e 370 if (ul_path_read_buffer(dev->sysfs, buf, sizeof(buf), "queue/scheduler") == 0)
2a4c734b 371 return NULL;
f153614c 372 p = strchr(buf, '[');
2a4c734b
MB
373 if (p) {
374 res = p + 1;
375 p = strchr(res, ']');
376 if (p) {
377 *p = '\0';
378 res = xstrdup(res);
379 } else
380 res = NULL;
381 }
2a4c734b
MB
382 return res;
383}
384
fed34a1e 385static char *get_type(struct lsblk_device *dev)
18eac5ba
MB
386{
387 char *res = NULL, *p;
388
0bd05f5e 389 if (device_is_partition(dev))
ef2ce68b
KZ
390 return xstrdup("part");
391
fed34a1e 392 if (is_dm(dev->name)) {
f153614c 393 char *dm_uuid = NULL;
18eac5ba
MB
394
395 /* The DM_UUID prefix should be set to subsystem owning
396 * the device - LVM, CRYPT, DMRAID, MPATH, PART */
fed34a1e 397 if (ul_path_read_string(dev->sysfs, &dm_uuid, "dm/uuid") > 0
f153614c 398 && dm_uuid) {
18eac5ba
MB
399 char *tmp = dm_uuid;
400 char *dm_uuid_prefix = strsep(&tmp, "-");
401
402 if (dm_uuid_prefix) {
403 /* kpartx hack to remove partition number */
404 if (strncasecmp(dm_uuid_prefix, "part", 4) == 0)
405 dm_uuid_prefix[4] = '\0';
406
407 res = xstrdup(dm_uuid_prefix);
408 }
409 }
410
411 free(dm_uuid);
412 if (!res)
413 /* No UUID or no prefix - just mark it as DM device */
414 res = xstrdup("dm");
415
fed34a1e 416 } else if (!strncmp(dev->name, "loop", 4)) {
18eac5ba
MB
417 res = xstrdup("loop");
418
fed34a1e 419 } else if (!strncmp(dev->name, "md", 2)) {
f153614c
KZ
420 char *md_level = NULL;
421
fed34a1e 422 ul_path_read_string(dev->sysfs, &md_level, "md/level");
18eac5ba
MB
423 res = md_level ? md_level : xstrdup("md");
424
425 } else {
8d687180 426 const char *type = NULL;
90e9fcda 427 int x = 0;
9d738ff8 428
fed34a1e 429 if (ul_path_read_s32(dev->sysfs, &x, "device/type") == 0)
8d687180 430 type = blkdev_scsi_type_to_name(x);
9d738ff8 431 if (!type)
ef2ce68b 432 type = "disk";
9d738ff8 433 res = xstrdup(type);
18eac5ba
MB
434 }
435
436 for (p = res; p && *p; p++)
437 *p = tolower((unsigned char) *p);
438 return res;
439}
440
a5fb4d23 441/* Thanks to lsscsi code for idea of detection logic used here */
fed34a1e 442static char *get_transport(struct lsblk_device *dev)
a5fb4d23 443{
fed34a1e 444 struct path_cxt *sysfs = dev->sysfs;
2f2f7e80
KZ
445 char *attr = NULL;
446 const char *trans = NULL;
a5fb4d23 447
f153614c 448
28ffc2b7 449 /* SCSI - Serial Peripheral Interface */
f153614c 450 if (sysfs_blkdev_scsi_host_is(sysfs, "spi"))
2f2f7e80 451 trans = "spi";
a5fb4d23
MB
452
453 /* FC/FCoE - Fibre Channel / Fibre Channel over Ethernet */
f153614c
KZ
454 else if (sysfs_blkdev_scsi_host_is(sysfs, "fc")) {
455 attr = sysfs_blkdev_scsi_host_strdup_attribute(sysfs, "fc", "symbolic_name");
23a11c74 456 if (!attr)
a5fb4d23 457 return NULL;
2f2f7e80
KZ
458 trans = strstr(attr, " over ") ? "fcoe" : "fc";
459 free(attr);
a5fb4d23
MB
460 }
461
462 /* SAS - Serial Attached SCSI */
f153614c
KZ
463 else if (sysfs_blkdev_scsi_host_is(sysfs, "sas") ||
464 sysfs_blkdev_scsi_has_attribute(sysfs, "sas_device"))
2f2f7e80 465 trans = "sas";
a5fb4d23 466
a5fb4d23
MB
467
468 /* SBP - Serial Bus Protocol (FireWire) */
f153614c 469 else if (sysfs_blkdev_scsi_has_attribute(sysfs, "ieee1394_id"))
2f2f7e80 470 trans = "sbp";
a5fb4d23
MB
471
472 /* iSCSI */
f153614c 473 else if (sysfs_blkdev_scsi_host_is(sysfs, "iscsi"))
2f2f7e80 474 trans ="iscsi";
a5fb4d23
MB
475
476 /* USB - Universal Serial Bus */
f153614c 477 else if (sysfs_blkdev_scsi_path_contains(sysfs, "usb"))
2f2f7e80 478 trans = "usb";
a5fb4d23
MB
479
480 /* ATA, SATA */
f153614c
KZ
481 else if (sysfs_blkdev_scsi_host_is(sysfs, "scsi")) {
482 attr = sysfs_blkdev_scsi_host_strdup_attribute(sysfs, "scsi", "proc_name");
23a11c74 483 if (!attr)
a5fb4d23 484 return NULL;
2f2f7e80
KZ
485 if (!strncmp(attr, "ahci", 4) || !strncmp(attr, "sata", 4))
486 trans = "sata";
487 else if (strstr(attr, "ata"))
488 trans = "ata";
489 free(attr);
3d305170 490
fed34a1e 491 } else if (strncmp(dev->name, "nvme", 4) == 0)
3d305170 492 trans = "nvme";
a5fb4d23 493
2f2f7e80 494 return trans ? xstrdup(trans) : NULL;
a5fb4d23
MB
495}
496
fed34a1e 497static char *get_subsystems(struct lsblk_device *dev)
7f14ee1b
KZ
498{
499 char path[PATH_MAX];
500 char *sub, *chain, *res = NULL;
501 size_t len = 0, last = 0;
502
fed34a1e 503 chain = sysfs_blkdev_get_devchain(dev->sysfs, path, sizeof(path));
7f14ee1b
KZ
504 if (!chain)
505 return NULL;
506
fed34a1e 507 while (sysfs_blkdev_next_subsystem(dev->sysfs, chain, &sub) == 0) {
7f14ee1b
KZ
508 size_t sz;
509
510 /* don't create "block:scsi:scsi", but "block:scsi" */
56e78cb5
AH
511 if (len && strcmp(res + last, sub) == 0) {
512 free(sub);
7f14ee1b 513 continue;
56e78cb5 514 }
7f14ee1b
KZ
515
516 sz = strlen(sub);
517 res = xrealloc(res, len + sz + 2);
518 if (len)
519 res[len++] = ':';
520
521 memcpy(res + len, sub, sz + 1);
522 last = len;
523 len += sz;
524 free(sub);
525 }
526
527 return res;
528}
529
530
9bd4e5c0 531#define is_parsable(_l) (scols_table_is_raw((_l)->table) || \
4a102a48
KZ
532 scols_table_is_export((_l)->table) || \
533 scols_table_is_json((_l)->table))
7c9c872c 534
c7e76cd1
KZ
535static char *mk_name(const char *name)
536{
537 char *p;
538 if (!name)
539 return NULL;
540 if (lsblk->paths)
541 xasprintf(&p, "/dev/%s", name);
542 else
543 p = xstrdup(name);
92441d5c
KZ
544 if (p)
545 sysfs_devname_sys_to_dev(p);
c7e76cd1
KZ
546 return p;
547}
548
549static char *mk_dm_name(const char *name)
550{
551 char *p;
552 if (!name)
553 return NULL;
554 if (lsblk->paths)
555 xasprintf(&p, "/dev/mapper/%s", name);
556 else
557 p = xstrdup(name);
558 return p;
559}
560
642048e4
KZ
561/* stores data to scols cell userdata (invisible and independent on output)
562 * to make the original values accessible for sort functions
563 */
564static void set_sortdata_u64(struct libscols_line *ln, int col, uint64_t x)
565{
566 struct libscols_cell *ce = scols_line_get_cell(ln, col);
567 uint64_t *data;
568
569 if (!ce)
570 return;
571 data = xmalloc(sizeof(uint64_t));
572 *data = x;
573 scols_cell_set_userdata(ce, data);
574}
575
d81e32bd
KZ
576/* do not modify *data on any error */
577static void str2u64(const char *str, uint64_t *data)
642048e4 578{
d81e32bd
KZ
579 uintmax_t num;
580 char *end = NULL;
642048e4 581
d81e32bd
KZ
582 errno = 0;
583 if (str == NULL || *str == '\0')
642048e4 584 return;
d81e32bd 585 num = strtoumax(str, &end, 10);
642048e4 586
d81e32bd
KZ
587 if (errno || str == end || (end && *end))
588 return;
589 *data = num;
642048e4
KZ
590}
591
592static void unref_sortdata(struct libscols_table *tb)
593{
594 struct libscols_iter *itr;
595 struct libscols_line *ln;
596
597 if (!tb || !lsblk->sort_col)
598 return;
599 itr = scols_new_iter(SCOLS_ITER_FORWARD);
600 if (!itr)
601 return;
602 while (scols_table_next_line(tb, itr, &ln) == 0) {
603 struct libscols_cell *ce = scols_line_get_column_cell(ln,
604 lsblk->sort_col);
605 void *data = scols_cell_get_userdata(ce);
606 free(data);
607 }
608
609 scols_free_iter(itr);
610}
611
fed34a1e 612static char *get_vfs_attribute(struct lsblk_device *dev, int id)
bc405d64
KZ
613{
614 char *sizestr;
615 uint64_t vfs_attr = 0;
616 char *mnt;
617
fed34a1e
KZ
618 if (!dev->fsstat.f_blocks) {
619 mnt = lsblk_device_get_mountpoint(dev);
620 if (!mnt || dev->is_swap)
bc405d64 621 return NULL;
fed34a1e 622 if (statvfs(mnt, &dev->fsstat) != 0)
bc405d64
KZ
623 return NULL;
624 }
625
626 switch(id) {
627 case COL_FSSIZE:
fed34a1e 628 vfs_attr = dev->fsstat.f_frsize * dev->fsstat.f_blocks;
bc405d64
KZ
629 break;
630 case COL_FSAVAIL:
fed34a1e 631 vfs_attr = dev->fsstat.f_frsize * dev->fsstat.f_bavail;
bc405d64
KZ
632 break;
633 case COL_FSUSED:
fed34a1e 634 vfs_attr = dev->fsstat.f_frsize * (dev->fsstat.f_blocks - dev->fsstat.f_bfree);
bc405d64
KZ
635 break;
636 case COL_FSUSEPERC:
fed34a1e 637 if (dev->fsstat.f_blocks == 0)
bc405d64
KZ
638 return xstrdup("-");
639
640 xasprintf(&sizestr, "%.0f%%",
fed34a1e
KZ
641 (double)(dev->fsstat.f_blocks - dev->fsstat.f_bfree) /
642 dev->fsstat.f_blocks * 100);
bc405d64
KZ
643 return sizestr;
644 }
645
646 if (!vfs_attr)
647 sizestr = xstrdup("0");
648 else if (lsblk->bytes)
649 xasprintf(&sizestr, "%ju", vfs_attr);
650 else
651 sizestr = size_to_human_string(SIZE_SUFFIX_1LETTER, vfs_attr);
652
653 return sizestr;
654}
655
fed34a1e 656static struct stat *device_get_stat(struct lsblk_device *dev)
652cb1cd 657{
404eef65
KZ
658 if (!dev->st.st_rdev
659 && stat(dev->filename, &dev->st) != 0)
660 return NULL;
652cb1cd 661
fed34a1e 662 return &dev->st;
652cb1cd 663}
bc405d64 664
81a8936c
KZ
665static int is_removable_device(struct lsblk_device *dev, struct lsblk_device *parent)
666{
667 struct path_cxt *pc;
668
669 if (dev->removable != -1)
670 goto done;
671 if (ul_path_scanf(dev->sysfs, "removable", "%d", &dev->removable) == 1)
672 goto done;
673
674 if (parent) {
675 pc = sysfs_blkdev_get_parent(dev->sysfs);
676 if (!pc)
677 goto done;
678
74152bf9 679 /* dev is partition and parent is whole-disk */
81a8936c 680 if (pc == parent->sysfs)
81a8936c 681 dev->removable = is_removable_device(parent, NULL);
74152bf9
KZ
682
683 /* parent is something else, use sysfs parent */
684 else if (ul_path_scanf(pc, "removable", "%d", &dev->removable) != 1)
685 dev->removable = 0;
81a8936c 686 }
81a8936c
KZ
687done:
688 if (dev->removable == -1)
689 dev->removable = 0;
690 return dev->removable;
691}
692
8a2f175c
KZ
693static uint64_t device_get_discard_granularity(struct lsblk_device *dev)
694{
695 if (dev->discard_granularity == (uint64_t) -1
696 && ul_path_read_u64(dev->sysfs, &dev->discard_granularity,
697 "queue/discard_granularity") != 0)
698 dev->discard_granularity = 0;
699
700 return dev->discard_granularity;
701}
702
b93585e8 703/*
d81e32bd
KZ
704 * Generates data (string) for column specified by column ID for specified device. If sortdata
705 * is not NULL then returns number usable to sort the column if the data are available for the
706 * column.
b93585e8 707 */
d81e32bd
KZ
708static char *device_get_data(
709 struct lsblk_device *dev, /* device */
710 struct lsblk_device *parent, /* device parent as defined in the tree */
711 int id, /* column ID (COL_*) */
712 uint64_t *sortdata) /* returns sort data as number */
2a4c734b 713{
baad6dcc 714 struct lsblk_devprop *prop;
bb6e822a 715 char *str = NULL;
2a4c734b 716
2a4c734b
MB
717 switch(id) {
718 case COL_NAME:
fed34a1e 719 str = dev->dm_name ? mk_dm_name(dev->dm_name) : mk_name(dev->name);
c7e76cd1 720 break;
2a4c734b 721 case COL_KNAME:
fed34a1e 722 str = mk_name(dev->name);
2a4c734b 723 break;
310c0603 724 case COL_PKNAME:
ae4c2c37
KZ
725 if (parent)
726 str = mk_name(parent->name);
310c0603 727 break;
66638b0b 728 case COL_PATH:
fed34a1e
KZ
729 if (dev->filename)
730 str = xstrdup(dev->filename);
66638b0b 731 break;
2a4c734b
MB
732 case COL_OWNER:
733 {
fed34a1e 734 struct stat *st = device_get_stat(dev);
e361253e 735 struct passwd *pw = st ? getpwuid(st->st_uid) : NULL;
2a4c734b 736 if (pw)
bb6e822a 737 str = xstrdup(pw->pw_name);
2a4c734b
MB
738 break;
739 }
740 case COL_GROUP:
741 {
fed34a1e 742 struct stat *st = device_get_stat(dev);
e361253e 743 struct group *gr = st ? getgrgid(st->st_gid) : NULL;
2a4c734b 744 if (gr)
bb6e822a 745 str = xstrdup(gr->gr_name);
2a4c734b
MB
746 break;
747 }
748 case COL_MODE:
749 {
fed34a1e 750 struct stat *st = device_get_stat(dev);
652cb1cd 751 char md[11] = { '\0' };
1dc42eb6 752
652cb1cd
KZ
753 if (st)
754 str = xstrdup(xstrmode(st->st_mode, md));
2a4c734b
MB
755 break;
756 }
757 case COL_MAJMIN:
7c9c872c 758 if (is_parsable(lsblk))
fed34a1e 759 xasprintf(&str, "%u:%u", dev->maj, dev->min);
2a4c734b 760 else
fed34a1e 761 xasprintf(&str, "%3u:%-3u", dev->maj, dev->min);
d81e32bd
KZ
762 if (sortdata)
763 *sortdata = makedev(dev->maj, dev->min);
2a4c734b
MB
764 break;
765 case COL_FSTYPE:
fed34a1e 766 prop = lsblk_device_get_properties(dev);
baad6dcc
KZ
767 if (prop && prop->fstype)
768 str = xstrdup(prop->fstype);
2a4c734b 769 break;
bc405d64
KZ
770 case COL_FSSIZE:
771 case COL_FSAVAIL:
772 case COL_FSUSED:
773 case COL_FSUSEPERC:
fed34a1e 774 str = get_vfs_attribute(dev, id);
bc405d64 775 break;
2a4c734b 776 case COL_TARGET:
fed34a1e 777 str = xstrdup(lsblk_device_get_mountpoint(dev));
2a4c734b
MB
778 break;
779 case COL_LABEL:
fed34a1e 780 prop = lsblk_device_get_properties(dev);
baad6dcc
KZ
781 if (prop && prop->label)
782 str = xstrdup(prop->label);
2a4c734b
MB
783 break;
784 case COL_UUID:
fed34a1e 785 prop = lsblk_device_get_properties(dev);
baad6dcc
KZ
786 if (prop && prop->uuid)
787 str = xstrdup(prop->uuid);
2a4c734b 788 break;
1b06b33d 789 case COL_PTUUID:
fed34a1e 790 prop = lsblk_device_get_properties(dev);
baad6dcc
KZ
791 if (prop && prop->ptuuid)
792 str = xstrdup(prop->ptuuid);
1b06b33d
MB
793 break;
794 case COL_PTTYPE:
fed34a1e 795 prop = lsblk_device_get_properties(dev);
baad6dcc
KZ
796 if (prop && prop->pttype)
797 str = xstrdup(prop->pttype);
1b06b33d 798 break;
4686ba17 799 case COL_PARTTYPE:
fed34a1e 800 prop = lsblk_device_get_properties(dev);
baad6dcc
KZ
801 if (prop && prop->parttype)
802 str = xstrdup(prop->parttype);
4686ba17 803 break;
aa0903e0 804 case COL_PARTLABEL:
fed34a1e 805 prop = lsblk_device_get_properties(dev);
baad6dcc
KZ
806 if (prop && prop->partlabel)
807 str = xstrdup(prop->partlabel);
aa0903e0
KZ
808 break;
809 case COL_PARTUUID:
fed34a1e 810 prop = lsblk_device_get_properties(dev);
baad6dcc
KZ
811 if (prop && prop->partuuid)
812 str = xstrdup(prop->partuuid);
aa0903e0 813 break;
5a2fd932 814 case COL_PARTFLAGS:
fed34a1e 815 prop = lsblk_device_get_properties(dev);
baad6dcc
KZ
816 if (prop && prop->partflags)
817 str = xstrdup(prop->partflags);
5a2fd932 818 break;
88ca32b3 819 case COL_WWN:
fed34a1e 820 prop = lsblk_device_get_properties(dev);
baad6dcc
KZ
821 if (prop && prop->wwn)
822 str = xstrdup(prop->wwn);
88ca32b3 823 break;
150db7a9 824 case COL_RA:
fed34a1e 825 ul_path_read_string(dev->sysfs, &str, "queue/read_ahead_kb");
d81e32bd
KZ
826 if (sortdata)
827 str2u64(str, sortdata);
150db7a9 828 break;
2a4c734b 829 case COL_RO:
fed34a1e 830 str = xstrdup(is_readonly_device(dev) ? "1" : "0");
2a4c734b 831 break;
627970a7 832 case COL_RM:
81a8936c 833 str = xstrdup(is_removable_device(dev, parent) ? "1" : "0");
2a4c734b 834 break;
483987c2 835 case COL_HOTPLUG:
fed34a1e 836 str = sysfs_blkdev_is_hotpluggable(dev->sysfs) ? xstrdup("1") : xstrdup("0");
483987c2 837 break;
2a4c734b 838 case COL_ROTA:
fed34a1e 839 ul_path_read_string(dev->sysfs, &str, "queue/rotational");
2a4c734b 840 break;
12b06c3d 841 case COL_RAND:
fed34a1e 842 ul_path_read_string(dev->sysfs, &str, "queue/add_random");
12b06c3d 843 break;
2a4c734b 844 case COL_MODEL:
10501add 845 if (!device_is_partition(dev) && dev->nslaves == 0) {
fed34a1e 846 prop = lsblk_device_get_properties(dev);
baad6dcc
KZ
847 if (prop && prop->model)
848 str = xstrdup(prop->model);
13cbc6f2 849 else
fed34a1e 850 ul_path_read_string(dev->sysfs, &str, "device/model");
13cbc6f2 851 }
d6681cee 852 break;
460c7afb 853 case COL_SERIAL:
10501add 854 if (!device_is_partition(dev) && dev->nslaves == 0) {
fed34a1e 855 prop = lsblk_device_get_properties(dev);
baad6dcc
KZ
856 if (prop && prop->serial)
857 str = xstrdup(prop->serial);
3d305170 858 else
fed34a1e 859 ul_path_read_string(dev->sysfs, &str, "device/serial");
460c7afb
KZ
860 }
861 break;
d6681cee 862 case COL_REV:
10501add 863 if (!device_is_partition(dev) && dev->nslaves == 0)
fed34a1e 864 ul_path_read_string(dev->sysfs, &str, "device/rev");
99f43b72
MB
865 break;
866 case COL_VENDOR:
10501add 867 if (!device_is_partition(dev) && dev->nslaves == 0)
fed34a1e 868 ul_path_read_string(dev->sysfs, &str, "device/vendor");
2a4c734b
MB
869 break;
870 case COL_SIZE:
fed34a1e 871 if (!dev->size)
bb6e822a
KZ
872 break;
873 if (lsblk->bytes)
fed34a1e 874 xasprintf(&str, "%ju", dev->size);
bb6e822a 875 else
fed34a1e 876 str = size_to_human_string(SIZE_SUFFIX_1LETTER, dev->size);
d81e32bd
KZ
877 if (sortdata)
878 *sortdata = dev->size;
2a4c734b 879 break;
01e487c0 880 case COL_STATE:
10501add 881 if (!device_is_partition(dev) && !dev->dm_name)
fed34a1e
KZ
882 ul_path_read_string(dev->sysfs, &str, "device/state");
883 else if (dev->dm_name) {
01e487c0 884 int x = 0;
fed34a1e 885 if (ul_path_read_s32(dev->sysfs, &x, "dm/suspended") == 0)
bb6e822a 886 str = xstrdup(x ? "suspended" : "running");
01e487c0 887 }
01e487c0 888 break;
2a4c734b 889 case COL_ALIOFF:
fed34a1e 890 ul_path_read_string(dev->sysfs, &str, "alignment_offset");
d81e32bd
KZ
891 if (sortdata)
892 str2u64(str, sortdata);
2a4c734b
MB
893 break;
894 case COL_MINIO:
fed34a1e 895 ul_path_read_string(dev->sysfs, &str, "queue/minimum_io_size");
d81e32bd
KZ
896 if (sortdata)
897 str2u64(str, sortdata);
2a4c734b
MB
898 break;
899 case COL_OPTIO:
fed34a1e 900 ul_path_read_string(dev->sysfs, &str, "queue/optimal_io_size");
d81e32bd
KZ
901 if (sortdata)
902 str2u64(str, sortdata);
2a4c734b
MB
903 break;
904 case COL_PHYSEC:
fed34a1e 905 ul_path_read_string(dev->sysfs, &str, "queue/physical_block_size");
d81e32bd
KZ
906 if (sortdata)
907 str2u64(str, sortdata);
2a4c734b
MB
908 break;
909 case COL_LOGSEC:
fed34a1e 910 ul_path_read_string(dev->sysfs, &str, "queue/logical_block_size");
d81e32bd
KZ
911 if (sortdata)
912 str2u64(str, sortdata);
2a4c734b
MB
913 break;
914 case COL_SCHED:
fed34a1e 915 str = get_scheduler(dev);
2a4c734b 916 break;
64c9e22a 917 case COL_RQ_SIZE:
fed34a1e 918 ul_path_read_string(dev->sysfs, &str, "queue/nr_requests");
d81e32bd
KZ
919 if (sortdata)
920 str2u64(str, sortdata);
64c9e22a 921 break;
18eac5ba 922 case COL_TYPE:
fed34a1e 923 str = get_type(dev);
18eac5ba 924 break;
699e5c4f
MB
925 case COL_HCTL:
926 {
927 int h, c, t, l;
fed34a1e 928 if (sysfs_blkdev_scsi_get_hctl(dev->sysfs, &h, &c, &t, &l) == 0)
bb6e822a 929 xasprintf(&str, "%d:%d:%d:%d", h, c, t, l);
699e5c4f
MB
930 break;
931 }
a5fb4d23 932 case COL_TRANSPORT:
fed34a1e 933 str = get_transport(dev);
a5fb4d23 934 break;
7f14ee1b 935 case COL_SUBSYS:
fed34a1e 936 str = get_subsystems(dev);
7f14ee1b 937 break;
2d232246 938 case COL_DALIGN:
8a2f175c 939 if (device_get_discard_granularity(dev) > 0)
fed34a1e 940 ul_path_read_string(dev->sysfs, &str, "discard_alignment");
bb6e822a
KZ
941 if (!str)
942 str = xstrdup("0");
d81e32bd
KZ
943 if (sortdata)
944 str2u64(str, sortdata);
2d232246
MP
945 break;
946 case COL_DGRAN:
642048e4 947 if (lsblk->bytes) {
fed34a1e 948 ul_path_read_string(dev->sysfs, &str, "queue/discard_granularity");
d81e32bd
KZ
949 if (sortdata)
950 str2u64(str, sortdata);
642048e4 951 } else {
8a2f175c
KZ
952 uint64_t x = device_get_discard_granularity(dev);
953 str = size_to_human_string(SIZE_SUFFIX_1LETTER, x);
d81e32bd
KZ
954 if (sortdata)
955 *sortdata = x;
9cfe1b9c 956 }
2d232246
MP
957 break;
958 case COL_DMAX:
642048e4 959 if (lsblk->bytes) {
fed34a1e 960 ul_path_read_string(dev->sysfs, &str, "queue/discard_max_bytes");
d81e32bd
KZ
961 if (sortdata)
962 str2u64(str, sortdata);
642048e4 963 } else {
9cfe1b9c 964 uint64_t x;
fed34a1e 965 if (ul_path_read_u64(dev->sysfs, &x, "queue/discard_max_bytes") == 0) {
bb6e822a 966 str = size_to_human_string(SIZE_SUFFIX_1LETTER, x);
d81e32bd
KZ
967 if (sortdata)
968 *sortdata = x;
642048e4 969 }
9cfe1b9c 970 }
2d232246
MP
971 break;
972 case COL_DZERO:
8a2f175c 973 if (device_get_discard_granularity(dev) > 0)
fed34a1e 974 ul_path_read_string(dev->sysfs, &str, "queue/discard_zeroes_data");
bb6e822a
KZ
975 if (!str)
976 str = xstrdup("0");
2d232246 977 break;
2adb1a44 978 case COL_WSAME:
642048e4 979 if (lsblk->bytes) {
fed34a1e 980 ul_path_read_string(dev->sysfs, &str, "queue/write_same_max_bytes");
d81e32bd
KZ
981 if (sortdata)
982 str2u64(str, sortdata);
642048e4 983 } else {
2adb1a44
MB
984 uint64_t x;
985
fed34a1e 986 if (ul_path_read_u64(dev->sysfs, &x, "queue/write_same_max_bytes") == 0) {
bb6e822a 987 str = size_to_human_string(SIZE_SUFFIX_1LETTER, x);
d81e32bd
KZ
988 if (sortdata)
989 *sortdata = x;
642048e4 990 }
2adb1a44 991 }
bb6e822a
KZ
992 if (!str)
993 str = xstrdup("0");
2adb1a44 994 break;
f2df4365 995 case COL_ZONED:
fed34a1e 996 ul_path_read_string(dev->sysfs, &str, "queue/zoned");
f2df4365 997 break;
2a4c734b 998 };
bb6e822a 999
d81e32bd 1000 return str;
2a4c734b
MB
1001}
1002
b93585e8
KZ
1003/*
1004 * Adds data for all wanted columns about the device to the smartcols table
1005 */
d81e32bd
KZ
1006static void device_to_scols(
1007 struct lsblk_device *dev,
1008 struct lsblk_device *parent,
1009 struct libscols_table *tab,
1010 struct libscols_line *parent_line)
2a4c734b 1011{
40b17508 1012 size_t i;
d81e32bd 1013 struct libscols_line *ln;
f43b8297
KZ
1014 struct lsblk_iter itr;
1015 struct lsblk_device *child = NULL;
0bd05f5e 1016 int link_group = 0;
2a4c734b 1017
fe22543d
KZ
1018
1019 DBG(DEV, ul_debugobj(dev, "add '%s' to scols", dev->name));
1020 ON_DBG(DEV, if (ul_path_isopen_dirfd(dev->sysfs)) ul_debugobj(dev, " %s ---> is open!", dev->name));
ff7992d6 1021
0bd05f5e
KZ
1022 /* Do not print device more than one in --list mode */
1023 if (!(lsblk->flags & LSBLK_TREE) && dev->is_printed)
1024 return;
1025
1026 if (lsblk->merge && list_count_entries(&dev->parents) > 1) {
1027 if (!lsblk_device_is_last_parent(dev, parent))
1028 return;
1029 link_group = 1;
1030 }
1031
1032 ln = scols_table_new_line(tab, link_group ? NULL : parent_line);
d81e32bd 1033 if (!ln)
780ce22c 1034 err(EXIT_FAILURE, _("failed to allocate output line"));
2a4c734b 1035
0bd05f5e
KZ
1036 dev->is_printed = 1;
1037
1038 if (link_group) {
1039 struct lsblk_device *p;
1040 struct libscols_line *gr = parent_line;
1041
1042 /* Merge all my parents to the one group */
fe22543d 1043 DBG(DEV, ul_debugobj(dev, " grouping parents [--merge]"));
0bd05f5e
KZ
1044 lsblk_reset_iter(&itr, LSBLK_ITER_FORWARD);
1045 while (lsblk_device_next_parent(dev, &itr, &p) == 0) {
fe22543d
KZ
1046 if (!p->scols_line) {
1047 DBG(DEV, ul_debugobj(dev, " *** ignore '%s' no scols line yet", p->name));
0bd05f5e 1048 continue;
fe22543d
KZ
1049 }
1050 DBG(DEV, ul_debugobj(dev, " group '%s'", p->name));
1051 scols_table_group_lines(tab, p->scols_line, gr, 0);
0bd05f5e
KZ
1052 }
1053
1054 /* Link the group -- this makes group->child connection */
fe22543d 1055 DBG(DEV, ul_debugobj(dev, " linking the group [--merge]"));
0bd05f5e
KZ
1056 scols_line_link_group(ln, gr, 0);
1057 }
1058
d81e32bd
KZ
1059 /* read column specific data and set it to smartcols table line */
1060 for (i = 0; i < ncolumns; i++) {
1061 char *data;
1062 int id = get_column_id(i);
1063
1064 if (lsblk->sort_id != id)
1065 data = device_get_data(dev, parent, id, NULL);
1066 else {
1067 uint64_t sortdata = (uint64_t) -1;
1068
1069 data = device_get_data(dev, parent, id, &sortdata);
1070 if (data && sortdata != (uint64_t) -1)
1071 set_sortdata_u64(ln, i, sortdata);
1072 }
fe22543d 1073 DBG(DEV, ul_debugobj(dev, " refer data[%zu]=\"%s\"", i, data));
d81e32bd
KZ
1074 if (data && scols_line_refer_data(ln, i, data))
1075 err(EXIT_FAILURE, _("failed to add output data"));
1076 }
f43b8297 1077
0bd05f5e
KZ
1078 dev->scols_line = ln;
1079
ff7992d6
KZ
1080 if (dev->npartitions == 0)
1081 /* For partitions we often read from parental whole-disk sysfs,
1082 * otherwise we can close */
1083 ul_path_close_dirfd(dev->sysfs);
1084
0bd05f5e 1085 lsblk_reset_iter(&itr, LSBLK_ITER_FORWARD);
fe22543d
KZ
1086 while (lsblk_device_next_child(dev, &itr, &child) == 0) {
1087 DBG(DEV, ul_debugobj(dev, "%s -> continue to child", dev->name));
d81e32bd 1088 device_to_scols(child, dev, tab, ln);
fe22543d
KZ
1089 DBG(DEV, ul_debugobj(dev, "%s <- child done", dev->name));
1090 }
ff7992d6
KZ
1091
1092 /* Let's be careful with number of open files */
1093 ul_path_close_dirfd(dev->sysfs);
f43b8297
KZ
1094}
1095
b93585e8
KZ
1096/*
1097 * Walks on tree and adds one line for each device to the smartcols table
1098 */
f43b8297
KZ
1099static void devtree_to_scols(struct lsblk_devtree *tr, struct libscols_table *tab)
1100{
1101 struct lsblk_iter itr;
1102 struct lsblk_device *dev = NULL;
1103
1104 lsblk_reset_iter(&itr, LSBLK_ITER_FORWARD);
1105
1106 while (lsblk_devtree_next_root(tr, &itr, &dev) == 0)
d81e32bd 1107 device_to_scols(dev, NULL, tab, NULL);
2a4c734b
MB
1108}
1109
b93585e8
KZ
1110/*
1111 * Reads very basic information about the device from sysfs into the device struct
1112 */
b158bd29 1113static int initialize_device(struct lsblk_device *dev,
fed34a1e 1114 struct lsblk_device *wholedisk,
09a71aa1 1115 const char *name)
2a4c734b 1116{
766ab9b0 1117 dev_t devno;
2a4c734b 1118
b158bd29
KZ
1119 DBG(DEV, ul_debugobj(dev, "initialize %s [wholedisk=%p %s]",
1120 name, wholedisk, wholedisk ? wholedisk->name : ""));
7e786eca 1121
fed34a1e 1122 dev->name = xstrdup(name);
10501add
KZ
1123
1124 if (wholedisk) {
1125 dev->wholedisk = wholedisk;
1126 lsblk_ref_device(wholedisk);
1127 }
2a4c734b 1128
fed34a1e
KZ
1129 dev->filename = get_device_path(dev);
1130 if (!dev->filename) {
3bf2a36c 1131 DBG(DEV, ul_debugobj(dev, "%s: failed to get device path", dev->name));
270e66ec
MB
1132 return -1;
1133 }
3bf2a36c 1134 DBG(DEV, ul_debugobj(dev, "%s: filename=%s", dev->name, dev->filename));
2a4c734b 1135
fed34a1e 1136 devno = __sysfs_devname_to_devno(lsblk->sysroot, dev->name, wholedisk ? wholedisk->name : NULL);
270e66ec 1137 if (!devno) {
3bf2a36c 1138 DBG(DEV, ul_debugobj(dev, "%s: unknown device name", dev->name));
270e66ec
MB
1139 return -1;
1140 }
2a4c734b 1141
0e086f82
KZ
1142 dev->sysfs = ul_new_sysfs_path(devno, wholedisk ? wholedisk->sysfs : NULL, lsblk->sysroot);
1143 if (!dev->sysfs) {
1144 DBG(DEV, ul_debugobj(dev, "%s: failed to initialize sysfs handler", dev->name));
1145 return -1;
270e66ec 1146 }
766ab9b0 1147
fed34a1e
KZ
1148 dev->maj = major(devno);
1149 dev->min = minor(devno);
1150 dev->size = 0;
2a4c734b 1151
fed34a1e
KZ
1152 if (ul_path_read_u64(dev->sysfs, &dev->size, "size") == 0) /* in sectors */
1153 dev->size <<= 9; /* in bytes */
90e9fcda 1154
2a4c734b 1155 /* Ignore devices of zero size */
fed34a1e 1156 if (!lsblk->all_devices && dev->size == 0) {
3bf2a36c 1157 DBG(DEV, ul_debugobj(dev, "zero size device -- ignore"));
2a4c734b 1158 return -1;
7e786eca 1159 }
fed34a1e
KZ
1160 if (is_dm(dev->name)) {
1161 ul_path_read_string(dev->sysfs, &dev->dm_name, "dm/name");
1162 if (!dev->dm_name) {
3bf2a36c 1163 DBG(DEV, ul_debugobj(dev, "%s: failed to get dm name", dev->name));
270e66ec
MB
1164 return -1;
1165 }
766ab9b0 1166 }
2a4c734b 1167
fed34a1e
KZ
1168 dev->npartitions = sysfs_blkdev_count_partitions(dev->sysfs, dev->name);
1169 dev->nholders = ul_path_count_dirents(dev->sysfs, "holders");
1170 dev->nslaves = ul_path_count_dirents(dev->sysfs, "slaves");
2a4c734b 1171
3bf2a36c 1172 DBG(DEV, ul_debugobj(dev, "%s: npartitions=%d, nholders=%d, nslaves=%d",
fed34a1e 1173 dev->name, dev->npartitions, dev->nholders, dev->nslaves));
7e786eca 1174
28ffc2b7 1175 /* ignore non-SCSI devices */
fed34a1e 1176 if (lsblk->scsi && sysfs_blkdev_scsi_get_hctl(dev->sysfs, NULL, NULL, NULL, NULL)) {
3bf2a36c 1177 DBG(DEV, ul_debugobj(dev, "non-scsi device -- ignore"));
28ffc2b7 1178 return -1;
7e786eca 1179 }
28ffc2b7 1180
3bf2a36c 1181 DBG(DEV, ul_debugobj(dev, "%s: context successfully initialized", dev->name));
2a4c734b
MB
1182 return 0;
1183}
1184
4ca3c472 1185static struct lsblk_device *devtree_get_device_or_new(struct lsblk_devtree *tr,
583b6489
KZ
1186 struct lsblk_device *disk,
1187 const char *name)
1188{
1189 struct lsblk_device *dev = lsblk_devtree_get_device(tr, name);
1190
1191 if (!dev) {
10501add 1192 dev = lsblk_new_device();
583b6489
KZ
1193 if (!dev)
1194 err(EXIT_FAILURE, _("failed to allocate device"));
1195
b158bd29 1196 if (initialize_device(dev, disk, name) != 0) {
583b6489
KZ
1197 lsblk_unref_device(dev);
1198 return NULL;
1199 }
1200 lsblk_devtree_add_device(tr, dev);
1201 lsblk_unref_device(dev); /* keep it referenced by devtree only */
1202 } else
1203 DBG(DEV, ul_debugobj(dev, "%s: already processed", name));
1204
1205 return dev;
1206}
a0a76f42 1207
4ca3c472
KZ
1208static int process_dependencies(
1209 struct lsblk_devtree *tr,
1210 struct lsblk_device *dev,
1211 int do_partitions);
09a71aa1 1212
2a4c734b 1213/*
4ca3c472 1214 * Read devices from whole-disk device into tree
2a4c734b 1215 */
cf3c1bc7 1216static int process_partitions(struct lsblk_devtree *tr, struct lsblk_device *disk)
2a4c734b
MB
1217{
1218 DIR *dir;
1219 struct dirent *d;
2a4c734b 1220
cf3c1bc7 1221 assert(disk);
f31505fe 1222
09a71aa1
PR
1223 /*
1224 * Do not process further if there are no partitions for
1225 * this device or the device itself is a partition.
1226 */
10501add 1227 if (!disk->npartitions || device_is_partition(disk))
cf3c1bc7 1228 return -EINVAL;
2a4c734b 1229
cf3c1bc7 1230 DBG(DEV, ul_debugobj(disk, "%s: probe whole-disk for partitions", disk->name));
7e786eca 1231
cf3c1bc7 1232 dir = ul_path_opendir(disk->sysfs, NULL);
2a4c734b
MB
1233 if (!dir)
1234 err(EXIT_FAILURE, _("failed to open device directory in sysfs"));
1235
1236 while ((d = xreaddir(dir))) {
4ca3c472 1237 struct lsblk_device *part;
2a4c734b 1238
cf3c1bc7 1239 if (!(sysfs_blkdev_is_partition_dirent(dir, d, disk->name)))
270e66ec 1240 continue;
09a71aa1 1241
cf3c1bc7 1242 DBG(DEV, ul_debugobj(disk, " checking %s", d->d_name));
7e786eca 1243
cf3c1bc7 1244 part = devtree_get_device_or_new(tr, disk, d->d_name);
4ca3c472
KZ
1245 if (!part)
1246 continue;
1247
cf3c1bc7 1248 if (lsblk_device_new_dependence(disk, part) == 0)
4ca3c472 1249 process_dependencies(tr, part, 0);
ff7992d6
KZ
1250
1251 ul_path_close_dirfd(part->sysfs);
2a4c734b 1252 }
09a71aa1 1253
ff7992d6
KZ
1254 /* For partitions we need parental (whole-disk) sysfs directory pretty
1255 * often, so close it now when all is done */
1256 ul_path_close_dirfd(disk->sysfs);
1257
cf3c1bc7 1258 DBG(DEV, ul_debugobj(disk, "probe whole-disk for partitions -- done"));
2a4c734b 1259 closedir(dir);
4ca3c472 1260 return 0;
09a71aa1
PR
1261}
1262
4ca3c472 1263static char *get_wholedisk_from_partition_dirent(DIR *dir, struct dirent *d, char *buf, size_t bufsz)
09a71aa1 1264{
09a71aa1
PR
1265 char *p;
1266 int len;
1267
4ca3c472 1268 if ((len = readlinkat(dirfd(dir), d->d_name, buf, bufsz - 1)) < 0)
09a71aa1 1269 return 0;
540d506f 1270
4ca3c472 1271 buf[len] = '\0';
09a71aa1
PR
1272
1273 /* The path ends with ".../<device>/<partition>" */
4ca3c472 1274 p = strrchr(buf, '/');
540d506f 1275 if (!p)
4ca3c472 1276 return NULL;
540d506f
KZ
1277 *p = '\0';
1278
4ca3c472 1279 p = strrchr(buf, '/');
540d506f 1280 if (!p)
4ca3c472 1281 return NULL;
540d506f 1282 p++;
2a4c734b 1283
4ca3c472 1284 return p;
09a71aa1
PR
1285}
1286
1287/*
b93585e8 1288 * Reads slaves/holders and partitions for specified device into device tree
09a71aa1 1289 */
4ca3c472
KZ
1290static int process_dependencies(
1291 struct lsblk_devtree *tr,
1292 struct lsblk_device *dev,
1293 int do_partitions)
09a71aa1
PR
1294{
1295 DIR *dir;
1296 struct dirent *d;
540d506f 1297 const char *depname;
09a71aa1 1298
fed34a1e 1299 assert(dev);
09a71aa1 1300
57a06133
KZ
1301 if (lsblk->nodeps)
1302 return 0;
1303
4ca3c472
KZ
1304 /* read all or specified partition */
1305 if (do_partitions && dev->npartitions)
1306 process_partitions(tr, dev);
1307
4ca3c472 1308 DBG(DEV, ul_debugobj(dev, "%s: reading dependencies", dev->name));
7e786eca 1309
4ca3c472
KZ
1310 if (!(lsblk->inverse ? dev->nslaves : dev->nholders)) {
1311 DBG(DEV, ul_debugobj(dev, " ignore (no slaves/holders)"));
09a71aa1 1312 return 0;
4ca3c472 1313 }
09a71aa1 1314
540d506f 1315 depname = lsblk->inverse ? "slaves" : "holders";
fed34a1e 1316 dir = ul_path_opendir(dev->sysfs, depname);
4ca3c472
KZ
1317 if (!dir) {
1318 DBG(DEV, ul_debugobj(dev, " ignore (no slaves/holders directory)"));
2a4c734b 1319 return 0;
4ca3c472 1320 }
ff7992d6 1321 ul_path_close_dirfd(dev->sysfs);
2a4c734b 1322
4ca3c472 1323 DBG(DEV, ul_debugobj(dev, " %s: checking for '%s' dependence", dev->name, depname));
7e786eca 1324
2a4c734b 1325 while ((d = xreaddir(dir))) {
ff7992d6
KZ
1326 struct lsblk_device *dep = NULL;
1327 struct lsblk_device *disk = NULL;
4ca3c472 1328
09a71aa1 1329 /* Is the dependency a partition? */
f153614c 1330 if (sysfs_blkdev_is_partition_dirent(dir, d, NULL)) {
4ca3c472
KZ
1331
1332 char buf[PATH_MAX];
1333 char *diskname;
4ca3c472
KZ
1334
1335 DBG(DEV, ul_debugobj(dev, " %s: dependence is partition", d->d_name));
1336
1337 diskname = get_wholedisk_from_partition_dirent(dir, d, buf, sizeof(buf));
1338 if (diskname)
0e086f82 1339 disk = devtree_get_device_or_new(tr, NULL, diskname);
4ca3c472
KZ
1340 if (!disk) {
1341 DBG(DEV, ul_debugobj(dev, " ignore no wholedisk ???"));
ff7992d6 1342 goto next;
7e786eca 1343 }
4ca3c472 1344
0e086f82 1345 dep = devtree_get_device_or_new(tr, disk, d->d_name);
4ca3c472 1346 if (!dep)
ff7992d6 1347 goto next;
4ca3c472
KZ
1348
1349 if (lsblk_device_new_dependence(dev, dep) == 0)
1350 process_dependencies(tr, dep, 1);
1351
1352 if (lsblk->inverse
1353 && lsblk_device_new_dependence(dep, disk) == 0)
1354 process_dependencies(tr, disk, 0);
270e66ec 1355 }
09a71aa1 1356 /* The dependency is a whole device. */
4ca3c472
KZ
1357 else {
1358 DBG(DEV, ul_debugobj(dev, " %s: %s: dependence is whole-disk",
fed34a1e 1359 dev->name, d->d_name));
4ca3c472 1360
0e086f82 1361 dep = devtree_get_device_or_new(tr, NULL, d->d_name);
4ca3c472 1362 if (!dep)
ff7992d6 1363 goto next;
4ca3c472
KZ
1364
1365 if (lsblk_device_new_dependence(dev, dep) == 0)
1366 /* For inverse tree we don't want to show partitions
1367 * if the dependence is on whole-disk */
1368 process_dependencies(tr, dep, lsblk->inverse ? 0 : 1);
7e786eca 1369 }
ff7992d6
KZ
1370next:
1371 if (dep && dep->sysfs)
1372 ul_path_close_dirfd(dep->sysfs);
1373 if (disk && disk->sysfs)
1374 ul_path_close_dirfd(disk->sysfs);
2a4c734b
MB
1375 }
1376 closedir(dir);
1377
3bf2a36c 1378 DBG(DEV, ul_debugobj(dev, "%s: checking for '%s' -- done", dev->name, depname));
2a4c734b
MB
1379 return 0;
1380}
1381
b93585e8
KZ
1382/*
1383 * Defines the device as root node in the device tree and walks on all dependencies of the device.
1384 */
e7b5353e 1385static int __process_one_device(struct lsblk_devtree *tr, char *devname, dev_t devno)
2a4c734b 1386{
291606fa 1387 struct lsblk_device *dev = NULL;
ff7992d6 1388 struct lsblk_device *disk = NULL;
1539750f 1389 char buf[PATH_MAX + 1], *name = NULL, *diskname = NULL;
1539750f 1390 int real_part = 0, rc = -EINVAL;
2a4c734b 1391
e7b5353e
KZ
1392 if (devno == 0) {
1393 struct stat st;
1394
1395 DBG(DEV, ul_debug("%s: reading alone device", devname));
1396
1397 if (stat(devname, &st) || !S_ISBLK(st.st_mode)) {
1398 warnx(_("%s: not a block device"), devname);
1399 goto leave;
1400 }
1401 devno = st.st_rdev;
1402 } else
1403 DBG(DEV, ul_debug("%d:%d: reading alone device", major(devno), minor(devno)));
da30cb2a 1404
f153614c
KZ
1405 /* TODO: sysfs_devno_to_devname() internally initializes path_cxt, it
1406 * would be better to use ul_new_sysfs_path() + sysfs_blkdev_get_name()
b158bd29 1407 * and reuse path_cxt for initialize_device()
f153614c 1408 */
e7b5353e 1409 name = sysfs_devno_to_devname(devno, buf, sizeof(buf));
f153614c 1410 if (!name) {
e7b5353e
KZ
1411 if (devname)
1412 warn(_("%s: failed to get sysfs name"), devname);
1539750f 1413 goto leave;
2a4c734b 1414 }
3273ce07 1415 name = xstrdup(name);
da30cb2a
PR
1416
1417 if (!strncmp(name, "dm-", 3)) {
1418 /* dm mapping is never a real partition! */
1419 real_part = 0;
1420 } else {
ff7992d6
KZ
1421 dev_t diskno = 0;
1422
1423 if (blkid_devno_to_wholedisk(devno, buf, sizeof(buf), &diskno)) {
e7b5353e 1424 warn(_("%s: failed to get whole-disk device number"), name);
1539750f 1425 goto leave;
da30cb2a
PR
1426 }
1427 diskname = buf;
ff7992d6 1428 real_part = devno != diskno;
da30cb2a
PR
1429 }
1430
1431 if (!real_part) {
2a4c734b 1432 /*
09a71aa1 1433 * Device is not a partition.
2a4c734b 1434 */
4ca3c472
KZ
1435 DBG(DEV, ul_debug(" non-partition"));
1436
0e086f82 1437 dev = devtree_get_device_or_new(tr, NULL, name);
291606fa 1438 if (!dev)
270e66ec 1439 goto leave;
291606fa 1440
4ca3c472
KZ
1441 lsblk_devtree_add_root(tr, dev);
1442 process_dependencies(tr, dev, !lsblk->inverse);
270e66ec 1443 } else {
2a4c734b 1444 /*
291606fa 1445 * Partition, read sysfs name of the disk device
2a4c734b 1446 */
4ca3c472
KZ
1447 DBG(DEV, ul_debug(" partition"));
1448
0e086f82
KZ
1449 disk = devtree_get_device_or_new(tr, NULL, diskname);
1450 if (!disk)
270e66ec 1451 goto leave;
291606fa 1452
0e086f82 1453 dev = devtree_get_device_or_new(tr, disk, name);
291606fa 1454 if (!dev)
270e66ec 1455 goto leave;
09a71aa1 1456
4ca3c472
KZ
1457 lsblk_devtree_add_root(tr, dev);
1458 process_dependencies(tr, dev, 1);
291606fa 1459
4ca3c472 1460 if (lsblk->inverse
0e086f82
KZ
1461 && lsblk_device_new_dependence(dev, disk) == 0)
1462 process_dependencies(tr, disk, 0);
ff7992d6
KZ
1463 else
1464 ul_path_close_dirfd(disk->sysfs);
2a4c734b
MB
1465 }
1466
1539750f 1467 rc = 0;
270e66ec 1468leave:
ff7992d6
KZ
1469 if (dev && dev->sysfs)
1470 ul_path_close_dirfd(dev->sysfs);
1471 if (disk && disk->sysfs)
1472 ul_path_close_dirfd(disk->sysfs);
da30cb2a 1473 free(name);
1539750f 1474 return rc;
2a4c734b
MB
1475}
1476
e7b5353e
KZ
1477static int process_one_device(struct lsblk_devtree *tr, char *devname)
1478{
1479 return __process_one_device(tr, devname, 0);
1480}
1481
1482/*
1483 * The /sys/block contains only root devices, and no partitions. It seems more
1484 * simple to scan /sys/dev/block where are all devices without exceptions to get
1485 * top-level devices for the reverse tree.
1486 */
1487static int process_all_devices_inverse(struct lsblk_devtree *tr)
e15c9e3d
KZ
1488{
1489 DIR *dir;
1490 struct dirent *d;
e7b5353e
KZ
1491 struct path_cxt *pc = ul_new_path(_PATH_SYS_DEVBLOCK);
1492
1493 assert(lsblk->inverse);
e15c9e3d
KZ
1494
1495 if (!pc)
1496 err(EXIT_FAILURE, _("failed to allocate /sys handler"));
1497
1498 ul_path_set_prefix(pc, lsblk->sysroot);
e7b5353e
KZ
1499 dir = ul_path_opendir(pc, NULL);
1500 if (!dir)
1501 goto done;
1502
1503 DBG(DEV, ul_debug("iterate on " _PATH_SYS_DEVBLOCK));
1504
1505 while ((d = xreaddir(dir))) {
1506 dev_t devno;
1507 int maj, min;
e15c9e3d 1508
e7b5353e
KZ
1509 DBG(DEV, ul_debug(" %s dentry", d->d_name));
1510
1511 if (sscanf(d->d_name, "%d:%d", &maj, &min) != 2)
1512 continue;
1513 devno = makedev(maj, min);
1514
1515 if (is_maj_excluded(maj) || !is_maj_included(maj))
1516 continue;
1517 if (ul_path_countf_dirents(pc, "%s/holders", d->d_name) != 0)
1518 continue;
1519 if (sysfs_devno_count_partitions(devno) != 0)
1520 continue;
1521 __process_one_device(tr, NULL, devno);
1522 }
1523
1524 closedir(dir);
1525done:
1526 ul_unref_path(pc);
1527 DBG(DEV, ul_debug("iterate on " _PATH_SYS_DEVBLOCK " -- done"));
1528 return 0;
1529}
1530
b93585e8
KZ
1531/*
1532 * Reads root nodes (devices) from /sys/block into devices tree
1533 */
e7b5353e
KZ
1534static int process_all_devices(struct lsblk_devtree *tr)
1535{
1536 DIR *dir;
1537 struct dirent *d;
1538 struct path_cxt *pc;
1539
1540 assert(lsblk->inverse == 0);
1541
1542 pc = ul_new_path(_PATH_SYS_BLOCK);
1543 if (!pc)
1544 err(EXIT_FAILURE, _("failed to allocate /sys handler"));
1545
1546 ul_path_set_prefix(pc, lsblk->sysroot);
e15c9e3d
KZ
1547 dir = ul_path_opendir(pc, NULL);
1548 if (!dir)
1549 goto done;
1550
e7b5353e 1551 DBG(DEV, ul_debug("iterate on " _PATH_SYS_BLOCK));
e15c9e3d
KZ
1552
1553 while ((d = xreaddir(dir))) {
ff7992d6 1554 struct lsblk_device *dev = NULL;
e15c9e3d
KZ
1555
1556 DBG(DEV, ul_debug(" %s dentry", d->d_name));
e15c9e3d
KZ
1557 dev = devtree_get_device_or_new(tr, NULL, d->d_name);
1558 if (!dev)
ff7992d6 1559 goto next;
e15c9e3d
KZ
1560
1561 /* remove unwanted devices */
1562 if (is_maj_excluded(dev->maj) || !is_maj_included(dev->maj)) {
1563 DBG(DEV, ul_debug(" %s: ignore (by filter)", d->d_name));
1564 lsblk_devtree_remove_device(tr, dev);
f6f8a671 1565 dev = NULL;
ff7992d6 1566 goto next;
e15c9e3d
KZ
1567 }
1568
e7b5353e
KZ
1569 if (dev->nslaves) {
1570 DBG(DEV, ul_debug(" %s: ignore (in-middle)", d->d_name));
ff7992d6 1571 goto next;
e15c9e3d 1572 }
0bd05f5e 1573
e7b5353e
KZ
1574 lsblk_devtree_add_root(tr, dev);
1575 process_dependencies(tr, dev, 1);
ff7992d6
KZ
1576next:
1577 /* Let's be careful with number of open files */
1578 if (dev && dev->sysfs)
1579 ul_path_close_dirfd(dev->sysfs);
e15c9e3d
KZ
1580 }
1581
1582 closedir(dir);
1583done:
1584 ul_unref_path(pc);
1585 DBG(DEV, ul_debug("iterate on " _PATH_SYS_BLOCK " -- done"));
1586 return 0;
1587}
1588
b93585e8
KZ
1589/*
1590 * Parses major numbers as specified on lsblk command line
1591 */
2dc03af7 1592static void parse_excludes(const char *str0)
2a4c734b 1593{
2dc03af7
KZ
1594 const char *str = str0;
1595
2a4c734b
MB
1596 while (str && *str) {
1597 char *end = NULL;
ed34643c 1598 unsigned long n;
2a4c734b
MB
1599
1600 errno = 0;
1601 n = strtoul(str, &end, 10);
1602
2dc03af7
KZ
1603 if (end == str || (end && *end && *end != ','))
1604 errx(EXIT_FAILURE, _("failed to parse list '%s'"), str0);
1605 if (errno != 0 && (n == ULONG_MAX || n == 0))
1606 err(EXIT_FAILURE, _("failed to parse list '%s'"), str0);
2a4c734b
MB
1607 excludes[nexcludes++] = n;
1608
1609 if (nexcludes == ARRAY_SIZE(excludes))
f14f02f5 1610 /* TRANSLATORS: The standard value for %d is 256. */
2a4c734b
MB
1611 errx(EXIT_FAILURE, _("the list of excluded devices is "
1612 "too large (limit is %d devices)"),
1613 (int)ARRAY_SIZE(excludes));
2dc03af7 1614
2a4c734b
MB
1615 str = end && *end ? end + 1 : NULL;
1616 }
1617}
1618
b93585e8
KZ
1619/*
1620 * Parses major numbers as specified on lsblk command line
1621 * (TODO: what about refactor and merge parse_excludes() and parse_includes().)
1622 */
2dc03af7 1623static void parse_includes(const char *str0)
2ef4e8ba 1624{
2dc03af7
KZ
1625 const char *str = str0;
1626
2ef4e8ba
KZ
1627 while (str && *str) {
1628 char *end = NULL;
1629 unsigned long n;
1630
1631 errno = 0;
1632 n = strtoul(str, &end, 10);
1633
2dc03af7
KZ
1634 if (end == str || (end && *end && *end != ','))
1635 errx(EXIT_FAILURE, _("failed to parse list '%s'"), str0);
1636 if (errno != 0 && (n == ULONG_MAX || n == 0))
1637 err(EXIT_FAILURE, _("failed to parse list '%s'"), str0);
2ef4e8ba
KZ
1638 includes[nincludes++] = n;
1639
1640 if (nincludes == ARRAY_SIZE(includes))
1641 /* TRANSLATORS: The standard value for %d is 256. */
1642 errx(EXIT_FAILURE, _("the list of included devices is "
1643 "too large (limit is %d devices)"),
1644 (int)ARRAY_SIZE(includes));
1645 str = end && *end ? end + 1 : NULL;
1646 }
1647}
1648
642048e4
KZ
1649/*
1650 * see set_sortdata_u64() and columns initialization in main()
1651 */
1652static int cmp_u64_cells(struct libscols_cell *a,
1653 struct libscols_cell *b,
1654 __attribute__((__unused__)) void *data)
1655{
1656 uint64_t *adata = (uint64_t *) scols_cell_get_userdata(a),
1657 *bdata = (uint64_t *) scols_cell_get_userdata(b);
1658
1659 if (adata == NULL && bdata == NULL)
1660 return 0;
1661 if (adata == NULL)
1662 return -1;
1663 if (bdata == NULL)
1664 return 1;
1665 return *adata == *bdata ? 0 : *adata >= *bdata ? 1 : -1;
1666}
1667
dc4662f0
KZ
1668static void device_set_dedupkey(
1669 struct lsblk_device *dev,
1670 struct lsblk_device *parent,
1671 int id)
1672{
1673 struct lsblk_iter itr;
1674 struct lsblk_device *child = NULL;
1675
1676 dev->dedupkey = device_get_data(dev, parent, id, NULL);
1677 if (dev->dedupkey)
1678 DBG(DEV, ul_debugobj(dev, "%s: de-duplication key: %s", dev->name, dev->dedupkey));
1679
1680 if (dev->npartitions == 0)
1681 /* For partitions we often read from parental whole-disk sysfs,
1682 * otherwise we can close */
1683 ul_path_close_dirfd(dev->sysfs);
1684
1685 lsblk_reset_iter(&itr, LSBLK_ITER_FORWARD);
1686
1687 while (lsblk_device_next_child(dev, &itr, &child) == 0)
1688 device_set_dedupkey(child, dev, id);
1689
1690 /* Let's be careful with number of open files */
1691 ul_path_close_dirfd(dev->sysfs);
1692}
1693
1694static void devtree_set_dedupkeys(struct lsblk_devtree *tr, int id)
1695{
1696 struct lsblk_iter itr;
1697 struct lsblk_device *dev = NULL;
1698
1699 lsblk_reset_iter(&itr, LSBLK_ITER_FORWARD);
1700
1701 while (lsblk_devtree_next_root(tr, &itr, &dev) == 0)
1702 device_set_dedupkey(dev, NULL, id);
1703}
1704
86be6a32 1705static void __attribute__((__noreturn__)) usage(void)
2a4c734b 1706{
86be6a32 1707 FILE *out = stdout;
507d39c2 1708 size_t i;
2a4c734b 1709
39b232c1
SK
1710 fputs(USAGE_HEADER, out);
1711 fprintf(out, _(" %s [options] [<device> ...]\n"), program_invocation_short_name);
451dbcfa
BS
1712
1713 fputs(USAGE_SEPARATOR, out);
1714 fputs(_("List information about block devices.\n"), out);
1715
39b232c1 1716 fputs(USAGE_OPTIONS, out);
0bd05f5e
KZ
1717 fputs(_(" -D, --discard print discard capabilities\n"), out);
1718 fputs(_(" -E, --dedup <column> de-duplicate output by <column>\n"), out);
1719 fputs(_(" -I, --include <list> show only devices with specified major numbers\n"), out);
1720 fputs(_(" -J, --json use JSON output format\n"), out);
1721 fputs(_(" -O, --output-all output all columns\n"), out);
1722 fputs(_(" -P, --pairs use key=\"value\" output format\n"), out);
1723 fputs(_(" -S, --scsi output info about SCSI devices\n"), out);
b9c088f2 1724 fputs(_(" -T, --tree[=<column>] use tree format output\n"), out);
39b232c1
SK
1725 fputs(_(" -a, --all print all devices\n"), out);
1726 fputs(_(" -b, --bytes print SIZE in bytes rather than in human readable format\n"), out);
1727 fputs(_(" -d, --nodeps don't print slaves or holders\n"), out);
39b232c1 1728 fputs(_(" -e, --exclude <list> exclude devices by major number (default: RAM disks)\n"), out);
39b232c1 1729 fputs(_(" -f, --fs output info about filesystems\n"), out);
39b232c1 1730 fputs(_(" -i, --ascii use ascii characters only\n"), out);
39b232c1 1731 fputs(_(" -l, --list use list format output\n"), out);
0bd05f5e 1732 fputs(_(" -M, --merge group parents of sub-trees (usable for RAIDs, Multi-path)\n"), out);
3e59c481 1733 fputs(_(" -m, --perms output info about permissions\n"), out);
39b232c1
SK
1734 fputs(_(" -n, --noheadings don't print headings\n"), out);
1735 fputs(_(" -o, --output <list> output columns\n"), out);
ef75bc88 1736 fputs(_(" -p, --paths print complete device path\n"), out);
39b232c1
SK
1737 fputs(_(" -r, --raw use raw output format\n"), out);
1738 fputs(_(" -s, --inverse inverse dependencies\n"), out);
3e59c481 1739 fputs(_(" -t, --topology output info about topology\n"), out);
0bd05f5e 1740 fputs(_(" -z, --zoned print zone model\n"), out);
b5af3ee8 1741 fputs(_(" -x, --sort <column> sort output by <column>\n"), out);
498d1486 1742 fputs(_(" --sysroot <dir> use specified directory as system root\n"), out);
39b232c1 1743 fputs(USAGE_SEPARATOR, out);
f45f3ec3 1744 printf(USAGE_HELP_OPTIONS(22));
2a4c734b 1745
c3a4cfc5 1746 fprintf(out, USAGE_COLUMNS);
2a4c734b 1747
03df0d11 1748 for (i = 0; i < ARRAY_SIZE(infos); i++)
57912387 1749 fprintf(out, " %11s %s\n", infos[i].name, _(infos[i].help));
2a4c734b 1750
f45f3ec3 1751 printf(USAGE_MAN_TAIL("lsblk(8)"));
2a4c734b 1752
86be6a32 1753 exit(EXIT_SUCCESS);
2a4c734b
MB
1754}
1755
f65b3bb6
SK
1756static void check_sysdevblock(void)
1757{
1758 if (access(_PATH_SYS_DEVBLOCK, R_OK) != 0)
1759 err(EXIT_FAILURE, _("failed to access sysfs directory: %s"),
1760 _PATH_SYS_DEVBLOCK);
1761}
1762
2a4c734b
MB
1763int main(int argc, char *argv[])
1764{
dc4662f0
KZ
1765 struct lsblk _ls = {
1766 .sort_id = -1,
1767 .dedup_id = -1,
b9c088f2
KZ
1768 .flags = LSBLK_TREE,
1769 .tree_id = COL_NAME
dc4662f0 1770 };
57502dd3 1771 struct lsblk_devtree *tr = NULL;
40b17508 1772 int c, status = EXIT_FAILURE;
43bca827 1773 char *outarg = NULL;
40b17508 1774 size_t i;
b9c088f2 1775 int force_tree = 0, has_tree_col = 0;
2a4c734b 1776
ffc27623
KZ
1777 enum {
1778 OPT_SYSROOT = CHAR_MAX + 1
1779 };
1780
6c7d5ae9 1781 static const struct option longopts[] = {
87918040
SK
1782 { "all", no_argument, NULL, 'a' },
1783 { "bytes", no_argument, NULL, 'b' },
1784 { "nodeps", no_argument, NULL, 'd' },
1785 { "discard", no_argument, NULL, 'D' },
0bd05f5e 1786 { "dedup", required_argument, NULL, 'E' },
f2df4365 1787 { "zoned", no_argument, NULL, 'z' },
87918040
SK
1788 { "help", no_argument, NULL, 'h' },
1789 { "json", no_argument, NULL, 'J' },
1790 { "output", required_argument, NULL, 'o' },
1791 { "output-all", no_argument, NULL, 'O' },
0bd05f5e 1792 { "merge", no_argument, NULL, 'M' },
87918040
SK
1793 { "perms", no_argument, NULL, 'm' },
1794 { "noheadings", no_argument, NULL, 'n' },
1795 { "list", no_argument, NULL, 'l' },
1796 { "ascii", no_argument, NULL, 'i' },
1797 { "raw", no_argument, NULL, 'r' },
1798 { "inverse", no_argument, NULL, 's' },
1799 { "fs", no_argument, NULL, 'f' },
1800 { "exclude", required_argument, NULL, 'e' },
1801 { "include", required_argument, NULL, 'I' },
1802 { "topology", no_argument, NULL, 't' },
1803 { "paths", no_argument, NULL, 'p' },
1804 { "pairs", no_argument, NULL, 'P' },
1805 { "scsi", no_argument, NULL, 'S' },
1806 { "sort", required_argument, NULL, 'x' },
ffc27623 1807 { "sysroot", required_argument, NULL, OPT_SYSROOT },
b9c088f2 1808 { "tree", optional_argument, NULL, 'T' },
87918040
SK
1809 { "version", no_argument, NULL, 'V' },
1810 { NULL, 0, NULL, 0 },
2a4c734b
MB
1811 };
1812
a7349ee3 1813 static const ul_excl_t excl[] = { /* rows and cols in ASCII order */
1b4d2a4a 1814 { 'D','O' },
98f2dc7a 1815 { 'I','e' },
4a102a48 1816 { 'J', 'P', 'r' },
1b4d2a4a
MY
1817 { 'O','S' },
1818 { 'O','f' },
1819 { 'O','m' },
a44cd891 1820 { 'O','o' },
1b4d2a4a 1821 { 'O','t' },
1d9e35cc 1822 { 'P','T', 'l','r' },
98f2dc7a
KZ
1823 { 0 }
1824 };
1825 int excl_st[ARRAY_SIZE(excl)] = UL_EXCL_STATUS_INIT;
1826
2a4c734b
MB
1827 setlocale(LC_ALL, "");
1828 bindtextdomain(PACKAGE, LOCALEDIR);
1829 textdomain(PACKAGE);
2c308875 1830 close_stdout_atexit();
2a4c734b
MB
1831
1832 lsblk = &_ls;
2a4c734b 1833
7e786eca
KZ
1834 lsblk_init_debug();
1835
98f2dc7a 1836 while((c = getopt_long(argc, argv,
b9c088f2 1837 "abdDzE:e:fhJlnMmo:OpPiI:rstVST:x:", longopts, NULL)) != -1) {
98f2dc7a
KZ
1838
1839 err_exclusive_options(c, longopts, excl, excl_st);
1840
2a4c734b
MB
1841 switch(c) {
1842 case 'a':
1843 lsblk->all_devices = 1;
1844 break;
1845 case 'b':
1846 lsblk->bytes = 1;
1847 break;
f31505fe
KZ
1848 case 'd':
1849 lsblk->nodeps = 1;
1850 break;
2d232246 1851 case 'D':
be685b98
KZ
1852 add_uniq_column(COL_NAME);
1853 add_uniq_column(COL_DALIGN);
1854 add_uniq_column(COL_DGRAN);
1855 add_uniq_column(COL_DMAX);
1856 add_uniq_column(COL_DZERO);
2d232246 1857 break;
f2df4365 1858 case 'z':
be685b98
KZ
1859 add_uniq_column(COL_NAME);
1860 add_uniq_column(COL_ZONED);
f2df4365 1861 break;
2a4c734b
MB
1862 case 'e':
1863 parse_excludes(optarg);
1864 break;
4a102a48 1865 case 'J':
341c4ae2 1866 lsblk->flags |= LSBLK_JSON;
4a102a48 1867 break;
2a4c734b 1868 case 'l':
341c4ae2 1869 lsblk->flags &= ~LSBLK_TREE; /* disable the default */
2a4c734b 1870 break;
0bd05f5e
KZ
1871 case 'M':
1872 lsblk->merge = 1;
1873 break;
2a4c734b 1874 case 'n':
341c4ae2 1875 lsblk->flags |= LSBLK_NOHEADINGS;
2a4c734b
MB
1876 break;
1877 case 'o':
43bca827 1878 outarg = optarg;
2a4c734b 1879 break;
1b4d2a4a 1880 case 'O':
40b17508 1881 for (ncolumns = 0 ; ncolumns < ARRAY_SIZE(infos); ncolumns++)
1b4d2a4a
MY
1882 columns[ncolumns] = ncolumns;
1883 break;
c7e76cd1
KZ
1884 case 'p':
1885 lsblk->paths = 1;
1886 break;
6ea1bdd3 1887 case 'P':
341c4ae2
KZ
1888 lsblk->flags |= LSBLK_EXPORT;
1889 lsblk->flags &= ~LSBLK_TREE; /* disable the default */
6ea1bdd3 1890 break;
2a4c734b 1891 case 'i':
341c4ae2 1892 lsblk->flags |= LSBLK_ASCII;
2a4c734b 1893 break;
2ef4e8ba 1894 case 'I':
2ef4e8ba
KZ
1895 parse_includes(optarg);
1896 break;
2a4c734b 1897 case 'r':
341c4ae2
KZ
1898 lsblk->flags &= ~LSBLK_TREE; /* disable the default */
1899 lsblk->flags |= LSBLK_RAW; /* enable raw */
2a4c734b 1900 break;
09a71aa1
PR
1901 case 's':
1902 lsblk->inverse = 1;
1903 break;
2a4c734b 1904 case 'f':
be685b98
KZ
1905 add_uniq_column(COL_NAME);
1906 add_uniq_column(COL_FSTYPE);
1907 add_uniq_column(COL_LABEL);
1908 add_uniq_column(COL_UUID);
bc405d64
KZ
1909 add_uniq_column(COL_FSAVAIL);
1910 add_uniq_column(COL_FSUSEPERC);
be685b98 1911 add_uniq_column(COL_TARGET);
2a4c734b
MB
1912 break;
1913 case 'm':
be685b98
KZ
1914 add_uniq_column(COL_NAME);
1915 add_uniq_column(COL_SIZE);
1916 add_uniq_column(COL_OWNER);
1917 add_uniq_column(COL_GROUP);
1918 add_uniq_column(COL_MODE);
2a4c734b
MB
1919 break;
1920 case 't':
be685b98
KZ
1921 add_uniq_column(COL_NAME);
1922 add_uniq_column(COL_ALIOFF);
1923 add_uniq_column(COL_MINIO);
1924 add_uniq_column(COL_OPTIO);
1925 add_uniq_column(COL_PHYSEC);
1926 add_uniq_column(COL_LOGSEC);
1927 add_uniq_column(COL_ROTA);
1928 add_uniq_column(COL_SCHED);
1929 add_uniq_column(COL_RQ_SIZE);
1930 add_uniq_column(COL_RA);
1931 add_uniq_column(COL_WSAME);
2a4c734b 1932 break;
28ffc2b7
MB
1933 case 'S':
1934 lsblk->nodeps = 1;
1935 lsblk->scsi = 1;
be685b98
KZ
1936 add_uniq_column(COL_NAME);
1937 add_uniq_column(COL_HCTL);
1938 add_uniq_column(COL_TYPE);
1939 add_uniq_column(COL_VENDOR);
1940 add_uniq_column(COL_MODEL);
1941 add_uniq_column(COL_REV);
1942 add_uniq_column(COL_TRANSPORT);
28ffc2b7 1943 break;
1d9e35cc
KZ
1944 case 'T':
1945 force_tree = 1;
b9c088f2
KZ
1946 if (optarg)
1947 lsblk->tree_id = column_name_to_id(optarg, strlen(optarg));
1d9e35cc 1948 break;
ffc27623
KZ
1949 case OPT_SYSROOT:
1950 lsblk->sysroot = optarg;
1951 break;
0bd05f5e 1952 case 'E':
dc4662f0
KZ
1953 lsblk->dedup_id = column_name_to_id(optarg, strlen(optarg));
1954 if (lsblk->dedup_id >= 0)
1955 break;
1956 errtryhelp(EXIT_FAILURE);
1957 break;
642048e4 1958 case 'x':
341c4ae2 1959 lsblk->flags &= ~LSBLK_TREE; /* disable the default */
642048e4
KZ
1960 lsblk->sort_id = column_name_to_id(optarg, strlen(optarg));
1961 if (lsblk->sort_id >= 0)
1962 break;
2c308875
KZ
1963 errtryhelp(EXIT_FAILURE);
1964 break;
1965
1966 case 'h':
1967 usage();
1968 case 'V':
1969 print_version(EXIT_SUCCESS);
2a4c734b 1970 default:
193b3239 1971 errtryhelp(EXIT_FAILURE);
2a4c734b
MB
1972 }
1973 }
1974
1d9e35cc
KZ
1975 if (force_tree)
1976 lsblk->flags |= LSBLK_TREE;
1977
f65b3bb6
SK
1978 check_sysdevblock();
1979
2a4c734b 1980 if (!ncolumns) {
be685b98
KZ
1981 add_column(COL_NAME);
1982 add_column(COL_MAJMIN);
1983 add_column(COL_RM);
1984 add_column(COL_SIZE);
1985 add_column(COL_RO);
1986 add_column(COL_TYPE);
1987 add_column(COL_TARGET);
2a4c734b
MB
1988 }
1989
43bca827
MB
1990 if (outarg && string_add_to_idarray(outarg, columns, ARRAY_SIZE(columns),
1991 &ncolumns, column_name_to_id) < 0)
1992 return EXIT_FAILURE;
1993
b9dd5721 1994 if (lsblk->all_devices == 0 && nexcludes == 0 && nincludes == 0)
2a4c734b 1995 excludes[nexcludes++] = 1; /* default: ignore RAM disks */
9554f7ab 1996
5e2305ca
KZ
1997 if (lsblk->sort_id < 0)
1998 /* Since Linux 4.8 we have sort devices by default, because
1999 * /sys is no more sorted */
2000 lsblk->sort_id = COL_MAJMIN;
2001
091683a8
KZ
2002 /* For --inverse --list we still follow parent->child relation */
2003 if (lsblk->inverse && !(lsblk->flags & LSBLK_TREE))
2004 lsblk->force_tree_order = 1;
2005
a6dc8dcd
KZ
2006 if (lsblk->sort_id >= 0 && column_id_to_number(lsblk->sort_id) < 0) {
2007 /* the sort column is not between output columns -- add as hidden */
be685b98 2008 add_column(lsblk->sort_id);
a6dc8dcd
KZ
2009 lsblk->sort_hidden = 1;
2010 }
642048e4 2011
dc4662f0
KZ
2012 if (lsblk->dedup_id >= 0 && column_id_to_number(lsblk->dedup_id) < 0) {
2013 /* the deduplication column is not between output columns -- add as hidden */
2014 add_column(lsblk->dedup_id);
2015 lsblk->dedup_hidden = 1;
2016 }
2017
e0c016f1 2018 lsblk_mnt_init();
710ed55d 2019 scols_init_debug(0);
5a89aa99 2020 ul_path_init_debug();
9554f7ab 2021
2a4c734b
MB
2022 /*
2023 * initialize output columns
2024 */
0925a9dd 2025 if (!(lsblk->table = scols_new_table()))
780ce22c 2026 errx(EXIT_FAILURE, _("failed to allocate output table"));
341c4ae2
KZ
2027 scols_table_enable_raw(lsblk->table, !!(lsblk->flags & LSBLK_RAW));
2028 scols_table_enable_export(lsblk->table, !!(lsblk->flags & LSBLK_EXPORT));
2029 scols_table_enable_ascii(lsblk->table, !!(lsblk->flags & LSBLK_ASCII));
2030 scols_table_enable_json(lsblk->table, !!(lsblk->flags & LSBLK_JSON));
2031 scols_table_enable_noheadings(lsblk->table, !!(lsblk->flags & LSBLK_NOHEADINGS));
2a4c734b 2032
341c4ae2 2033 if (lsblk->flags & LSBLK_JSON)
4a102a48
KZ
2034 scols_table_set_name(lsblk->table, "blockdevices");
2035
2a4c734b
MB
2036 for (i = 0; i < ncolumns; i++) {
2037 struct colinfo *ci = get_column_info(i);
642048e4
KZ
2038 struct libscols_column *cl;
2039 int id = get_column_id(i), fl = ci->flags;
2a4c734b 2040
b9c088f2
KZ
2041 if ((lsblk->flags & LSBLK_TREE)
2042 && has_tree_col == 0
2043 && id == lsblk->tree_id) {
2044 fl |= SCOLS_FL_TREE;
2045 fl &= ~SCOLS_FL_RIGHT;
2046 has_tree_col = 1;
2047 }
2048
a6dc8dcd
KZ
2049 if (lsblk->sort_hidden && lsblk->sort_id == id)
2050 fl |= SCOLS_FL_HIDDEN;
dc4662f0
KZ
2051 if (lsblk->dedup_hidden && lsblk->dedup_id == id)
2052 fl |= SCOLS_FL_HIDDEN;
2a4c734b 2053
fb376180
KZ
2054 if (force_tree
2055 && lsblk->flags & LSBLK_JSON
2056 && has_tree_col == 0
2057 && i + 1 == ncolumns)
2058 /* The "--tree --json" specified, but no column with
2059 * SCOLS_FL_TREE yet; force it for the last column
2060 */
2061 fl |= SCOLS_FL_TREE;
2062
642048e4
KZ
2063 cl = scols_table_new_column(lsblk->table, ci->name, ci->whint, fl);
2064 if (!cl) {
780ce22c 2065 warn(_("failed to allocate output column"));
2a4c734b
MB
2066 goto leave;
2067 }
642048e4
KZ
2068 if (!lsblk->sort_col && lsblk->sort_id == id) {
2069 lsblk->sort_col = cl;
2070 scols_column_set_cmpfunc(cl,
407e0ea6
KZ
2071 ci->type == COLTYPE_NUM ? cmp_u64_cells :
2072 ci->type == COLTYPE_SIZE ? cmp_u64_cells :
2073 ci->type == COLTYPE_SORTNUM ? cmp_u64_cells : scols_cmpstr_cells,
2074 NULL);
2075 }
2076 if (lsblk->flags & LSBLK_JSON) {
2077 switch (ci->type) {
2078 case COLTYPE_SIZE:
2079 if (!lsblk->bytes)
2080 break;
2081 /* fallthrough */
2082 case COLTYPE_NUM:
2083 scols_column_set_json_type(cl, SCOLS_JSON_NUMBER);
2084 break;
2085 case COLTYPE_BOOL:
2086 scols_column_set_json_type(cl, SCOLS_JSON_BOOLEAN);
2087 break;
2088 default:
2089 scols_column_set_json_type(cl, SCOLS_JSON_STRING);
2090 break;
2091 }
642048e4 2092 }
2a4c734b
MB
2093 }
2094
f43b8297
KZ
2095 tr = lsblk_new_devtree();
2096 if (!tr)
2097 err(EXIT_FAILURE, _("failed to allocate device tree"));
2098
e7b5353e
KZ
2099 if (optind == argc) {
2100 int rc = lsblk->inverse ?
2101 process_all_devices_inverse(tr) :
2102 process_all_devices(tr);
2103
2104 status = rc == 0 ? EXIT_SUCCESS : EXIT_FAILURE;
2105 } else {
1539750f
KZ
2106 int cnt = 0, cnt_err = 0;
2107
2108 while (optind < argc) {
291606fa 2109 if (process_one_device(tr, argv[optind++]) != 0)
1539750f
KZ
2110 cnt_err++;
2111 cnt++;
2112 }
2113 status = cnt == 0 ? EXIT_FAILURE : /* nothing */
2114 cnt == cnt_err ? LSBLK_EXIT_ALLFAILED :/* all failed */
2115 cnt_err ? LSBLK_EXIT_SOMEOK : /* some ok */
2116 EXIT_SUCCESS; /* all success */
2117 }
2a4c734b 2118
dc4662f0
KZ
2119 if (lsblk->dedup_id > -1) {
2120 devtree_set_dedupkeys(tr, lsblk->dedup_id);
2121 lsblk_devtree_deduplicate_devices(tr);
2122 }
2123
f43b8297
KZ
2124 devtree_to_scols(tr, lsblk->table);
2125
642048e4
KZ
2126 if (lsblk->sort_col)
2127 scols_sort_table(lsblk->table, lsblk->sort_col);
091683a8
KZ
2128 if (lsblk->force_tree_order)
2129 scols_sort_table_by_tree(lsblk->table);
642048e4 2130
9bd4e5c0 2131 scols_print_table(lsblk->table);
2a4c734b
MB
2132
2133leave:
642048e4
KZ
2134 if (lsblk->sort_col)
2135 unref_sortdata(lsblk->table);
2136
9bd4e5c0 2137 scols_unref_table(lsblk->table);
50fccba1 2138
e0c016f1 2139 lsblk_mnt_deinit();
ccafadb7 2140 lsblk_properties_deinit();
57502dd3 2141 lsblk_unref_devtree(tr);
ccafadb7 2142
2a4c734b
MB
2143 return status;
2144}