diff options
Diffstat (limited to 'misc-utils/lsfd-bdev.c')
| -rw-r--r-- | misc-utils/lsfd-bdev.c | 163 |
1 files changed, 163 insertions, 0 deletions
diff --git a/misc-utils/lsfd-bdev.c b/misc-utils/lsfd-bdev.c new file mode 100644 index 0000000..7be99db --- /dev/null +++ b/misc-utils/lsfd-bdev.c @@ -0,0 +1,163 @@ +/* + * lsfd-bdev.c - handle associations opening block devices + * + * Copyright (C) 2021 Red Hat, Inc. All rights reserved. + * Written by Masatake YAMATO <yamato@redhat.com> + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it would be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software Foundation, + * Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ + +#include "xalloc.h" +#include "nls.h" +#include "libsmartcols.h" + +#include "lsfd.h" + +static struct list_head partitions; + +struct partition { + struct list_head partitions; + dev_t dev; + char *name; +}; + +static bool bdev_fill_column(struct proc *proc __attribute__((__unused__)), + struct file *file __attribute__((__unused__)), + struct libscols_line *ln, + int column_id, + size_t column_index) +{ + char *str = NULL; + const char *partition, *devdrv; + + switch(column_id) { + case COL_TYPE: + if (scols_line_set_data(ln, column_index, "BLK")) + err(EXIT_FAILURE, _("failed to add output data")); + return true; + case COL_BLKDRV: + devdrv = get_blkdrv(major(file->stat.st_rdev)); + if (devdrv) + str = xstrdup(devdrv); + else + xasprintf(&str, "%u", + major(file->stat.st_rdev)); + break; + case COL_DEVTYPE: + if (scols_line_set_data(ln, column_index, + "blk")) + err(EXIT_FAILURE, _("failed to add output data")); + return true; + case COL_SOURCE: + case COL_PARTITION: + partition = get_partition(file->stat.st_rdev); + if (partition) { + str = xstrdup(partition); + break; + } + devdrv = get_blkdrv(major(file->stat.st_rdev)); + if (devdrv) { + xasprintf(&str, "%s:%u", devdrv, + minor(file->stat.st_rdev)); + break; + } + /* FALL THROUGH */ + case COL_MAJMIN: + xasprintf(&str, "%u:%u", + major(file->stat.st_rdev), + minor(file->stat.st_rdev)); + break; + default: + return false; + } + + if (!str) + err(EXIT_FAILURE, _("failed to add output data")); + if (scols_line_refer_data(ln, column_index, str)) + err(EXIT_FAILURE, _("failed to add output data")); + return true; +} + +static struct partition *new_partition(dev_t dev, const char *name) +{ + struct partition *partition = xcalloc(1, sizeof(*partition)); + + INIT_LIST_HEAD(&partition->partitions); + + partition->dev = dev; + partition->name = xstrdup(name); + + return partition; +} + +static void free_partition(struct partition *partition) +{ + free(partition->name); + free(partition); +} + +static void read_partitions(struct list_head *partitions_list, FILE *part_fp) +{ + unsigned int major, minor; + char line[256]; + char name[sizeof(line)]; + + while (fgets(line, sizeof(line), part_fp)) { + struct partition *partition; + + if (sscanf(line, "%u %u %*u %s", &major, &minor, name) != 3) + continue; + partition = new_partition(makedev(major, minor), name); + list_add_tail(&partition->partitions, partitions_list); + } +} + +static void bdev_class_initialize(void) +{ + FILE *part_fp; + + INIT_LIST_HEAD(&partitions); + + part_fp = fopen("/proc/partitions", "r"); + if (part_fp) { + read_partitions(&partitions, part_fp); + fclose(part_fp); + } +} + +static void bdev_class_finalize(void) +{ + list_free(&partitions, struct partition, partitions, free_partition); +} + +const char *get_partition(dev_t dev) +{ + struct list_head *p; + list_for_each(p, &partitions) { + struct partition *partition = list_entry(p, struct partition, partitions); + if (partition->dev == dev) + return partition->name; + } + return NULL; +} + +const struct file_class bdev_class = { + .super = &file_class, + .size = sizeof(struct file), + .initialize_class = bdev_class_initialize, + .finalize_class = bdev_class_finalize, + .fill_column = bdev_fill_column, + .free_content = NULL, +}; |