Adding upstream version 2.39.3.upstream/2.39.3

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 465 insertions, 0 deletions

diff --git a/misc-utils/lsfd-file.c b/misc-utils/lsfd-file.c
new file mode 100644
index 0000000..bdaac3f
--- /dev/null
+++ b/misc-utils/lsfd-file.c
@@ -0,0 +1,465 @@
+/*
+ * lsfd(1) - list file descriptors
+ *
+ * Copyright (C) 2021 Red Hat, Inc. All rights reserved.
+ * Written by Masatake YAMATO <yamato@redhat.com>
+ *
+ * Very generally based on lsof(8) by Victor A. Abell <abe@purdue.edu>
+ * It supports multiple OSes. lsfd specializes to Linux.
+ *
+ * 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 <unistd.h>
+
+#ifdef HAVE_LINUX_NSFS_H
+# include <linux/nsfs.h>
+# if defined(NS_GET_NSTYPE)
+#  define USE_NS_GET_API	1
+#  include <sys/ioctl.h>
+# endif
+#endif
+#include <linux/sched.h>
+
+#include "xalloc.h"
+#include "nls.h"
+#include "buffer.h"
+#include "idcache.h"
+#include "strutils.h"
+
+#include "libsmartcols.h"
+
+#include "lsfd.h"
+
+static struct idcache *username_cache;
+
+static const char *assocstr[N_ASSOCS] = {
+	[ASSOC_CWD]       = "cwd",
+	[ASSOC_EXE]       = "exe",
+	/* "root" appears as user names, too.
+	 * So we use "rtd" here instead of "root". */
+	[ASSOC_ROOT]      = "rtd",
+	[ASSOC_NS_CGROUP] = "cgroup",
+	[ASSOC_NS_IPC]    = "ipc",
+	[ASSOC_NS_MNT]    = "mnt",
+	[ASSOC_NS_NET]    = "net",
+	[ASSOC_NS_PID]    = "pid",
+	[ASSOC_NS_PID4C]  = "pid4c",
+	[ASSOC_NS_TIME]   = "time",
+	[ASSOC_NS_TIME4C] = "time4c",
+	[ASSOC_NS_USER]   = "user",
+	[ASSOC_NS_UTS]    = "uts",
+	[ASSOC_MEM]       = "mem",
+	[ASSOC_SHM]       = "shm",
+};
+
+static const char *strftype(mode_t ftype)
+{
+	switch (ftype) {
+	case S_IFBLK:
+		return "BLK";
+	case S_IFCHR:
+		return "CHR";
+	case S_IFDIR:
+		return "DIR";
+	case S_IFIFO:
+		return "FIFO";
+	case S_IFLNK:
+		return "LINK";
+	case S_IFREG:
+		return "REG";
+	case S_IFSOCK:
+		return "SOCK";
+	default:
+		return "UNKN";
+	}
+}
+
+extern void lsfd_decode_file_flags(struct ul_buffer *buf, int flags);
+static void file_fill_flags_buf(struct ul_buffer *buf, int flags)
+{
+	lsfd_decode_file_flags(buf, flags);
+}
+
+#define does_file_has_fdinfo_alike(file)	\
+	((file)->association >= 0		\
+	 || (file)->association == -ASSOC_SHM	\
+	 || (file)->association == -ASSOC_MEM)
+
+static uint64_t get_map_length(struct file *file)
+{
+	uint64_t res = 0;
+
+	if (is_association(file, SHM) || is_association(file, MEM)) {
+		static size_t pagesize = 0;
+
+		if (!pagesize)
+			pagesize = getpagesize();
+
+		res = (file->map_end - file->map_start) / pagesize;
+	}
+
+	return res;
+}
+
+static bool file_fill_column(struct proc *proc,
+			     struct file *file,
+			     struct libscols_line *ln,
+			     int column_id,
+			     size_t column_index)
+{
+	char *str = NULL;
+	mode_t ftype;
+	const char *partition;
+
+	switch(column_id) {
+	case COL_COMMAND:
+		if (proc->command
+		    && scols_line_set_data(ln, column_index, proc->command))
+			err(EXIT_FAILURE, _("failed to add output data"));
+		return true;
+	case COL_KNAME:
+	case COL_NAME:
+		if (file->name
+		    && scols_line_set_data(ln, column_index, file->name))
+			err(EXIT_FAILURE, _("failed to add output data"));
+		return true;
+	case COL_STTYPE:
+	case COL_TYPE:
+		ftype = file->stat.st_mode & S_IFMT;
+		if (scols_line_set_data(ln, column_index, strftype(ftype)))
+			err(EXIT_FAILURE, _("failed to add output data"));
+		return true;
+	case COL_USER:
+		add_uid(username_cache, (int)proc->uid);
+		if (scols_line_set_data(ln, column_index,
+					get_id(username_cache,
+					       (int)proc->uid)->name))
+			err(EXIT_FAILURE, _("failed to add output data"));
+		return true;
+	case COL_OWNER:
+		add_uid(username_cache, (int)file->stat.st_uid);
+		if (scols_line_set_data(ln, column_index,
+					get_id(username_cache,
+					       (int)file->stat.st_uid)->name))
+			err(EXIT_FAILURE, _("failed to add output data"));
+		return true;
+	case COL_DEVTYPE:
+		if (scols_line_set_data(ln, column_index,
+					"nodev"))
+			err(EXIT_FAILURE, _("failed to add output data"));
+		return true;
+	case COL_FD:
+		if (!is_opened_file(file))
+			return false;
+		/* FALL THROUGH */
+	case COL_ASSOC:
+		if (is_opened_file(file))
+			xasprintf(&str, "%d", file->association);
+		else {
+			int assoc = file->association * -1;
+			if (assoc >= N_ASSOCS)
+				return false; /* INTERNAL ERROR */
+			xasprintf(&str, "%s", assocstr[assoc]);
+		}
+		break;
+	case COL_INODE:
+		xasprintf(&str, "%llu", (unsigned long long)file->stat.st_ino);
+		break;
+	case COL_SOURCE:
+		if (major(file->stat.st_dev) == 0) {
+			const char *filesystem = get_nodev_filesystem(minor(file->stat.st_dev));
+			if (filesystem) {
+				xasprintf(&str, "%s", filesystem);
+				break;
+			}
+		}
+		/* FALL THROUGH */
+	case COL_PARTITION:
+		partition = get_partition(file->stat.st_dev);
+		if (partition) {
+			str = xstrdup(partition);
+			break;
+		}
+		/* FALL THROUGH */
+	case COL_DEV:
+	case COL_MAJMIN:
+		xasprintf(&str, "%u:%u",
+			  major(file->stat.st_dev),
+			  minor(file->stat.st_dev));
+		break;
+	case COL_RDEV:
+		xasprintf(&str, "%u:%u",
+			  major(file->stat.st_rdev),
+			  minor(file->stat.st_rdev));
+		break;
+	case COL_PID:
+		xasprintf(&str, "%d", (int)proc->leader->pid);
+		break;
+	case COL_TID:
+		xasprintf(&str, "%d", (int)proc->pid);
+		break;
+	case COL_UID:
+		xasprintf(&str, "%d", (int)proc->uid);
+		break;
+	case COL_FUID:
+		xasprintf(&str, "%d", (int)file->stat.st_uid);
+		break;
+	case COL_SIZE:
+		xasprintf(&str, "%jd", (intmax_t)file->stat.st_size);
+		break;
+	case COL_NLINK:
+		xasprintf(&str, "%ju", (uintmax_t)file->stat.st_nlink);
+		break;
+	case COL_DELETED:
+		xasprintf(&str, "%d", file->stat.st_nlink == 0);
+		break;
+	case COL_KTHREAD:
+		xasprintf(&str, "%u", proc->kthread);
+		break;
+	case COL_MNT_ID:
+		xasprintf(&str, "%d", is_opened_file(file)? file->mnt_id: 0);
+		break;
+	case COL_MODE:
+		if (does_file_has_fdinfo_alike(file))
+			xasprintf(&str, "%c%c%c",
+				  file->mode & S_IRUSR? 'r': '-',
+				  file->mode & S_IWUSR? 'w': '-',
+				  (is_mapped_file(file)
+				   && file->mode & S_IXUSR)? 'x': '-');
+		else
+			xasprintf(&str, "---");
+		break;
+	case COL_POS:
+		xasprintf(&str, "%" PRIu64,
+			  (does_file_has_fdinfo_alike(file))? file->pos: 0);
+		break;
+	case COL_FLAGS: {
+		struct ul_buffer buf = UL_INIT_BUFFER;
+
+		if (!is_opened_file(file))
+			return true;
+
+		if (file->sys_flags == 0)
+			return true;
+
+		file_fill_flags_buf(&buf, file->sys_flags);
+		if (ul_buffer_is_empty(&buf))
+			return true;
+		str = ul_buffer_get_data(&buf, NULL, NULL);
+		break;
+	}
+	case COL_MAPLEN:
+		if (!is_mapped_file(file))
+			return true;
+		xasprintf(&str, "%ju", (uintmax_t)get_map_length(file));
+		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 int file_handle_fdinfo(struct file *file, const char *key, const char* value)
+{
+	int rc;
+
+	if (strcmp(key, "pos") == 0) {
+		rc = ul_strtou64(value, &file->pos, 10);
+
+	} else if (strcmp(key, "flags") == 0) {
+		rc = ul_strtou32(value, &file->sys_flags, 8);
+
+	} else if (strcmp(key, "mnt_id") == 0) {
+		rc = ul_strtou32(value, &file->mnt_id, 10);
+
+	} else
+		return 0;	/* ignore -- unknown item */
+
+	if (rc < 0)
+		return 0;	/* ignore -- parse failed */
+
+	return 1;		/* success */
+}
+
+static void file_free_content(struct file *file)
+{
+	free(file->name);
+}
+
+static void file_class_initialize(void)
+{
+	username_cache = new_idcache();
+	if (!username_cache)
+		err(EXIT_FAILURE, _("failed to allocate UID cache"));
+}
+
+static void file_class_finalize(void)
+{
+	free_idcache(username_cache);
+}
+
+const struct file_class file_class = {
+	.super = NULL,
+	.size = sizeof(struct file),
+	.initialize_class = file_class_initialize,
+	.finalize_class = file_class_finalize,
+	.fill_column = file_fill_column,
+	.handle_fdinfo = file_handle_fdinfo,
+	.free_content = file_free_content,
+};
+
+/*
+ * Regular files on NSFS
+ */
+
+struct nsfs_file {
+	struct file file;
+	int clone_type;
+};
+
+static const char *get_ns_type_name(int clone_type)
+{
+	switch (clone_type) {
+#ifdef USE_NS_GET_API
+	case CLONE_NEWNS:
+		return "mnt";
+	case CLONE_NEWCGROUP:
+		return "cgroup";
+	case CLONE_NEWUTS:
+		return "uts";
+	case CLONE_NEWIPC:
+		return "ipc";
+	case CLONE_NEWUSER:
+		return "user";
+	case CLONE_NEWPID:
+		return "pid";
+	case CLONE_NEWNET:
+		return "net";
+#ifdef CLONE_NEWTIME
+	case CLONE_NEWTIME:
+		return "time";
+#endif	/* CLONE_NEWTIME */
+#endif	/* USE_NS_GET_API */
+	default:
+		return "unknown";
+	}
+}
+
+static void init_nsfs_file_content(struct file *file)
+{
+	struct nsfs_file *nsfs_file = (struct nsfs_file *)file;
+	nsfs_file->clone_type = -1;
+
+#ifdef USE_NS_GET_API
+	char *proc_fname = NULL;
+	int ns_fd;
+	int ns_type;
+
+	if (is_association (file, NS_CGROUP))
+		nsfs_file->clone_type = CLONE_NEWCGROUP;
+	else if (is_association (file, NS_IPC))
+		nsfs_file->clone_type = CLONE_NEWIPC;
+	else if (is_association (file, NS_MNT))
+		nsfs_file->clone_type = CLONE_NEWNS;
+	else if (is_association (file, NS_NET))
+		nsfs_file->clone_type = CLONE_NEWNET;
+	else if (is_association (file, NS_PID)
+		 || is_association (file, NS_PID4C))
+		nsfs_file->clone_type = CLONE_NEWPID;
+#ifdef CLONE_NEWTIME
+	else if (is_association (file, NS_TIME)
+		 || is_association (file, NS_TIME4C))
+		nsfs_file->clone_type = CLONE_NEWTIME;
+#endif
+	else if (is_association (file, NS_USER))
+		nsfs_file->clone_type = CLONE_NEWUSER;
+	else if (is_association (file, NS_UTS))
+		nsfs_file->clone_type = CLONE_NEWUTS;
+
+	if (nsfs_file->clone_type != -1)
+		return;
+
+	if (!is_opened_file(file))
+		return;
+
+	if (!file->name)
+		return;
+
+	xasprintf(&proc_fname, "/proc/%d/fd/%d",
+		  file->proc->pid, file->association);
+	ns_fd = open(proc_fname, O_RDONLY);
+	free(proc_fname);
+	if (ns_fd < 0)
+		return;
+
+	ns_type = ioctl(ns_fd, NS_GET_NSTYPE);
+	close(ns_fd);
+	if (ns_type < 0)
+		return;
+
+	nsfs_file->clone_type = ns_type;
+#endif	/* USE_NS_GET_API */
+}
+
+
+static bool nsfs_file_fill_column(struct proc *proc __attribute__((__unused__)),
+				  struct file *file,
+				  struct libscols_line *ln,
+				  int column_id,
+				  size_t column_index)
+{
+	struct nsfs_file *nsfs_file = (struct nsfs_file *)file;
+	char *name = NULL;
+
+	if (nsfs_file->clone_type == -1)
+		return false;
+
+	switch (column_id) {
+	case COL_NS_NAME:
+		xasprintf(&name, "%s:[%llu]",
+			  get_ns_type_name(nsfs_file->clone_type),
+			  (unsigned long long)file->stat.st_ino);
+		break;
+	case COL_NS_TYPE:
+		if (scols_line_set_data(ln, column_index,
+					get_ns_type_name(nsfs_file->clone_type)))
+			err(EXIT_FAILURE, _("failed to add output data"));
+		return true;
+	default:
+		return false;
+	}
+
+	if (name && scols_line_refer_data(ln, column_index, name))
+		err(EXIT_FAILURE, _("failed to add output data"));
+
+	return true;
+}
+
+const struct file_class nsfs_file_class = {
+	.super = &file_class,
+	.size = sizeof(struct nsfs_file),
+	.initialize_class = NULL,
+	.finalize_class = NULL,
+	.initialize_content = init_nsfs_file_content,
+	.free_content = NULL,
+	.fill_column = nsfs_file_fill_column,
+	.handle_fdinfo = NULL,
+};