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