Adding upstream version 2.36.1.upstream/2.36.1upstream

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

1 files changed, 506 insertions, 0 deletions

diff --git a/disk-utils/blockdev.c b/disk-utils/blockdev.c
new file mode 100644
index 0000000..f425d15
--- /dev/null
+++ b/disk-utils/blockdev.c
@@ -0,0 +1,506 @@
+/*
+ * blockdev.c --- Do various simple block device ioctls from the command line
+ * aeb, 991028
+ */
+
+#include <stdio.h>
+#include <fcntl.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <sys/ioctl.h>
+#include <errno.h>
+
+#include "c.h"
+#include "nls.h"
+#include "blkdev.h"
+#include "pathnames.h"
+#include "closestream.h"
+#include "sysfs.h"
+
+struct bdc {
+	long		ioc;		/* ioctl code */
+	const char	*iocname;	/* ioctl name (e.g. BLKROSET) */
+	long		argval;		/* default argument */
+
+	const char	*name;		/* --setfoo */
+	const char	*argname;	/* argument name or NULL */
+
+	const char	*help;
+
+	int		argtype;
+	int		flags;
+};
+
+/* command flags */
+enum {
+	FL_NOPTR	= (1 << 1),	/* does not assume pointer (ARG_INT only)*/
+	FL_NORESULT	= (1 << 2)	/* does not return any data */
+};
+
+/* ioctl argument types */
+enum {
+	ARG_NONE,
+	ARG_USHRT,
+	ARG_INT,
+	ARG_UINT,
+	ARG_LONG,
+	ARG_ULONG,
+	ARG_LLONG,
+	ARG_ULLONG
+};
+
+#define IOCTL_ENTRY( io )	.ioc = io, .iocname = # io
+
+static const struct bdc bdcms[] =
+{
+	{
+		IOCTL_ENTRY(BLKROSET),
+		.name = "--setro",
+		.argtype = ARG_INT,
+		.argval = 1,
+		.flags = FL_NORESULT,
+		.help = N_("set read-only")
+	},{
+		IOCTL_ENTRY(BLKROSET),
+		.name = "--setrw",
+		.argtype = ARG_INT,
+		.argval = 0,
+		.flags = FL_NORESULT,
+		.help = N_("set read-write")
+	},{
+		IOCTL_ENTRY(BLKROGET),
+		.name = "--getro",
+		.argtype = ARG_INT,
+		.argval = -1,
+		.help = N_("get read-only")
+	},{
+		IOCTL_ENTRY(BLKDISCARDZEROES),
+		.name = "--getdiscardzeroes",
+		.argtype = ARG_UINT,
+		.argval = -1,
+		.help = N_("get discard zeroes support status")
+	},{
+		IOCTL_ENTRY(BLKSSZGET),
+		.name = "--getss",
+		.argtype = ARG_INT,
+		.argval = -1,
+		.help = N_("get logical block (sector) size")
+	},{
+		IOCTL_ENTRY(BLKPBSZGET),
+		.name = "--getpbsz",
+		.argtype = ARG_UINT,
+		.argval = -1,
+		.help = N_("get physical block (sector) size")
+	},{
+		IOCTL_ENTRY(BLKIOMIN),
+		.name = "--getiomin",
+		.argtype = ARG_UINT,
+		.argval = -1,
+		.help = N_("get minimum I/O size")
+	},{
+		IOCTL_ENTRY(BLKIOOPT),
+		.name = "--getioopt",
+		.argtype = ARG_UINT,
+		.argval = -1,
+		.help = N_("get optimal I/O size")
+	},{
+		IOCTL_ENTRY(BLKALIGNOFF),
+		.name = "--getalignoff",
+		.argtype = ARG_INT,
+		.argval = -1,
+		.help = N_("get alignment offset in bytes")
+	},{
+		IOCTL_ENTRY(BLKSECTGET),
+		.name = "--getmaxsect",
+		.argtype = ARG_USHRT,
+		.argval = -1,
+		.help = N_("get max sectors per request")
+	},{
+		IOCTL_ENTRY(BLKBSZGET),
+		.name = "--getbsz",
+		.argtype = ARG_INT,
+		.argval = -1,
+		.help = N_("get blocksize")
+	},{
+		IOCTL_ENTRY(BLKBSZSET),
+		.name = "--setbsz",
+		.argname = "<bytes>",
+		.argtype = ARG_INT,
+		.flags = FL_NORESULT,
+	        .help = N_("set blocksize on file descriptor opening the block device")
+	},{
+		IOCTL_ENTRY(BLKGETSIZE),
+		.name = "--getsize",
+		.argtype = ARG_ULONG,
+		.argval = -1,
+		.help = N_("get 32-bit sector count (deprecated, use --getsz)")
+	},{
+		IOCTL_ENTRY(BLKGETSIZE64),
+		.name = "--getsize64",
+		.argtype = ARG_ULLONG,
+		.argval = -1,
+		.help = N_("get size in bytes")
+	},{
+		IOCTL_ENTRY(BLKRASET),
+		.name = "--setra",
+		.argname = "<sectors>",
+		.argtype = ARG_INT,
+		.flags = FL_NOPTR | FL_NORESULT,
+		.help = N_("set readahead")
+	},{
+		IOCTL_ENTRY(BLKRAGET),
+		.name = "--getra",
+		.argtype = ARG_LONG,
+		.argval = -1,
+		.help = N_("get readahead")
+	},{
+		IOCTL_ENTRY(BLKFRASET),
+		.name = "--setfra",
+		.argname = "<sectors>",
+		.argtype = ARG_INT,
+		.flags = FL_NOPTR | FL_NORESULT,
+		.help = N_("set filesystem readahead")
+	},{
+		IOCTL_ENTRY(BLKFRAGET),
+		.name = "--getfra",
+		.argtype = ARG_LONG,
+		.argval = -1,
+		.help = N_("get filesystem readahead")
+	},{
+		IOCTL_ENTRY(BLKFLSBUF),
+		.name = "--flushbufs",
+		.help = N_("flush buffers")
+	},{
+		IOCTL_ENTRY(BLKRRPART),
+		.name = "--rereadpt",
+		.help = N_("reread partition table")
+	}
+};
+
+static void __attribute__((__noreturn__)) usage(void)
+{
+	size_t i;
+
+	fputs(USAGE_HEADER, stdout);
+	printf(_(
+	         " %1$s [-v|-q] commands devices\n"
+	         " %1$s --report [devices]\n"
+	         " %1$s -h|-V\n"
+		), program_invocation_short_name);
+
+	fputs(USAGE_SEPARATOR, stdout);
+	puts(  _("Call block device ioctls from the command line."));
+
+	fputs(USAGE_OPTIONS, stdout);
+	puts(  _(" -q             quiet mode"));
+	puts(  _(" -v             verbose mode"));
+	puts(  _("     --report   print report for specified (or all) devices"));
+	fputs(USAGE_SEPARATOR, stdout);
+	printf(USAGE_HELP_OPTIONS(16));
+
+	fputs(USAGE_SEPARATOR, stdout);
+	puts(  _("Available commands:"));
+	printf(_(" %-25s get size in 512-byte sectors\n"), "--getsz");
+	for (i = 0; i < ARRAY_SIZE(bdcms); i++) {
+		if (bdcms[i].argname)
+			printf(" %s %-*s %s\n", bdcms[i].name,
+				(int)(24 - strlen(bdcms[i].name)),
+				bdcms[i].argname, _(bdcms[i].help));
+		else
+			printf(" %-25s %s\n", bdcms[i].name,
+				_(bdcms[i].help));
+	}
+
+	printf(USAGE_MAN_TAIL("blockdev(8)"));
+	exit(EXIT_SUCCESS);
+}
+
+static int find_cmd(char *s)
+{
+	size_t j;
+
+	for (j = 0; j < ARRAY_SIZE(bdcms); j++)
+		if (!strcmp(s, bdcms[j].name))
+			return j;
+	return -1;
+}
+
+static void do_commands(int fd, char **argv, int d);
+static void report_header(void);
+static void report_device(char *device, int quiet);
+static void report_all_devices(void);
+
+int main(int argc, char **argv)
+{
+	int fd, d, j, k;
+
+	setlocale(LC_ALL, "");
+	bindtextdomain(PACKAGE, LOCALEDIR);
+	textdomain(PACKAGE);
+	close_stdout_atexit();
+
+	if (argc < 2) {
+		warnx(_("not enough arguments"));
+		errtryhelp(EXIT_FAILURE);
+	}
+
+	/* -V not together with commands */
+	if (!strcmp(argv[1], "-V") || !strcmp(argv[1], "--version"))
+		print_version(EXIT_SUCCESS);
+	if (!strcmp(argv[1], "-h") || !strcmp(argv[1], "--help"))
+		usage();
+
+	/* --report not together with other commands */
+	if (!strcmp(argv[1], "--report")) {
+		report_header();
+		if (argc > 2) {
+			for (d = 2; d < argc; d++)
+				report_device(argv[d], 0);
+		} else {
+			report_all_devices();
+		}
+		return EXIT_SUCCESS;
+	}
+
+	/* do each of the commands on each of the devices */
+	/* devices start after last command */
+	for (d = 1; d < argc; d++) {
+		j = find_cmd(argv[d]);
+		if (j >= 0) {
+			if (bdcms[j].argname)
+				d++;
+			continue;
+		}
+		if (!strcmp(argv[d], "--getsz"))
+			continue;
+		if (!strcmp(argv[d], "--")) {
+			d++;
+			break;
+		}
+		if (argv[d][0] != '-')
+			break;
+	}
+
+	if (d >= argc) {
+		warnx(_("no device specified"));
+		errtryhelp(EXIT_FAILURE);
+	}
+
+	for (k = d; k < argc; k++) {
+		fd = open(argv[k], O_RDONLY, 0);
+		if (fd < 0)
+			err(EXIT_FAILURE, _("cannot open %s"), argv[k]);
+		do_commands(fd, argv, d);
+		close(fd);
+	}
+	return EXIT_SUCCESS;
+}
+
+static void do_commands(int fd, char **argv, int d)
+{
+	int res, i, j;
+	int iarg = 0;
+	unsigned int uarg = 0;
+	unsigned short huarg = 0;
+	long larg = 0;
+	long long llarg = 0;
+	unsigned long lu = 0;
+	unsigned long long llu = 0;
+	int verbose = 0;
+
+	for (i = 1; i < d; i++) {
+		if (!strcmp(argv[i], "-v")) {
+			verbose = 1;
+			continue;
+		}
+		if (!strcmp(argv[i], "-q")) {
+			verbose = 0;
+			continue;
+		}
+
+		if (!strcmp(argv[i], "--getsz")) {
+			res = blkdev_get_sectors(fd, &llu);
+			if (res == 0)
+				printf("%lld\n", llu);
+			else
+				errx(EXIT_FAILURE,
+				     _("could not get device size"));
+			continue;
+		}
+
+		j = find_cmd(argv[i]);
+		if (j == -1) {
+			warnx(_("Unknown command: %s"), argv[i]);
+			errtryhelp(EXIT_FAILURE);
+		}
+
+		switch (bdcms[j].argtype) {
+		default:
+		case ARG_NONE:
+			res = ioctl(fd, bdcms[j].ioc, 0);
+			break;
+		case ARG_USHRT:
+			huarg = bdcms[j].argval;
+			res = ioctl(fd, bdcms[j].ioc, &huarg);
+			break;
+		case ARG_INT:
+			if (bdcms[j].argname) {
+				if (i == d - 1) {
+					warnx(_("%s requires an argument"),
+					      bdcms[j].name);
+					errtryhelp(EXIT_FAILURE);
+				}
+				iarg = atoi(argv[++i]);
+			} else
+				iarg = bdcms[j].argval;
+
+			res = bdcms[j].flags & FL_NOPTR ?
+			    ioctl(fd, bdcms[j].ioc, iarg) :
+			    ioctl(fd, bdcms[j].ioc, &iarg);
+			break;
+		case ARG_UINT:
+			uarg = bdcms[j].argval;
+			res = ioctl(fd, bdcms[j].ioc, &uarg);
+			break;
+		case ARG_LONG:
+			larg = bdcms[j].argval;
+			res = ioctl(fd, bdcms[j].ioc, &larg);
+			break;
+		case ARG_LLONG:
+			llarg = bdcms[j].argval;
+			res = ioctl(fd, bdcms[j].ioc, &llarg);
+			break;
+		case ARG_ULONG:
+			lu = bdcms[j].argval;
+			res = ioctl(fd, bdcms[j].ioc, &lu);
+			break;
+		case ARG_ULLONG:
+			llu = bdcms[j].argval;
+			res = ioctl(fd, bdcms[j].ioc, &llu);
+			break;
+		}
+
+		if (res == -1) {
+			warn(_("ioctl error on %s"), bdcms[j].iocname);
+			if (verbose)
+				printf(_("%s failed.\n"), _(bdcms[j].help));
+			exit(EXIT_FAILURE);
+		}
+
+		if (bdcms[j].argtype == ARG_NONE ||
+		    (bdcms[j].flags & FL_NORESULT)) {
+			if (verbose)
+				printf(_("%s succeeded.\n"), _(bdcms[j].help));
+			continue;
+		}
+
+		if (verbose)
+			printf("%s: ", _(bdcms[j].help));
+
+		switch (bdcms[j].argtype) {
+		case ARG_USHRT:
+			printf("%hu\n", huarg);
+			break;
+		case ARG_INT:
+			printf("%d\n", iarg);
+			break;
+		case ARG_UINT:
+			printf("%u\n", uarg);
+			break;
+		case ARG_LONG:
+			printf("%ld\n", larg);
+			break;
+		case ARG_LLONG:
+			printf("%lld\n", llarg);
+			break;
+		case ARG_ULONG:
+			printf("%lu\n", lu);
+			break;
+		case ARG_ULLONG:
+			printf("%llu\n", llu);
+			break;
+		}
+	}
+}
+
+static void report_all_devices(void)
+{
+	FILE *procpt;
+	char line[200];
+	char ptname[200 + 1];
+	char device[210];
+	int ma, mi, sz;
+
+	procpt = fopen(_PATH_PROC_PARTITIONS, "r");
+	if (!procpt)
+		err(EXIT_FAILURE, _("cannot open %s"), _PATH_PROC_PARTITIONS);
+
+	while (fgets(line, sizeof(line), procpt)) {
+		if (sscanf(line, " %d %d %d %200[^\n ]",
+			   &ma, &mi, &sz, ptname) != 4)
+			continue;
+
+		sprintf(device, "/dev/%s", ptname);
+		report_device(device, 1);
+	}
+
+	fclose(procpt);
+}
+
+static void report_device(char *device, int quiet)
+{
+	int fd;
+	int ro, ssz, bsz;
+	long ra;
+	unsigned long long bytes;
+	uint64_t start = 0;
+	char start_str[11] = { "\0" };
+	struct stat st;
+
+	fd = open(device, O_RDONLY | O_NONBLOCK);
+	if (fd < 0) {
+		if (!quiet)
+			warn(_("cannot open %s"), device);
+		return;
+	}
+
+	ro = ssz = bsz = 0;
+	ra = 0;
+	if (fstat(fd, &st) == 0) {
+		dev_t disk;
+		struct path_cxt *pc;
+
+		pc = ul_new_sysfs_path(st.st_rdev, NULL, NULL);
+		if (pc &&
+		    sysfs_blkdev_get_wholedisk(pc, NULL, 0, &disk) == 0 &&
+		    disk != st.st_rdev) {
+
+			if (ul_path_read_u64(pc, &start, "start") != 0)
+				/* TRANSLATORS: Start sector not available. Max. 10 letters. */
+				sprintf(start_str, "%10s", _("N/A"));
+		}
+		ul_unref_path(pc);
+	}
+	if (!*start_str)
+		sprintf(start_str, "%10ju", start);
+
+	if (ioctl(fd, BLKROGET, &ro) == 0 &&
+	    ioctl(fd, BLKRAGET, &ra) == 0 &&
+	    ioctl(fd, BLKSSZGET, &ssz) == 0 &&
+	    ioctl(fd, BLKBSZGET, &bsz) == 0 &&
+	    blkdev_get_size(fd, &bytes) == 0) {
+		printf("%s %5ld %5d %5d %s %15lld   %s\n",
+			ro ? "ro" : "rw", ra, ssz, bsz, start_str, bytes, device);
+	} else {
+		if (!quiet)
+			warnx(_("ioctl error on %s"), device);
+	}
+
+	close(fd);
+}
+
+static void report_header(void)
+{
+	printf(_("RO    RA   SSZ   BSZ   StartSec            Size   Device\n"));
+}