blockdev: new applet

function                                             old     new   delta
blockdev_main                                          -     287    +287
bdcommands                                             -     160    +160
packed_usage                                       27203   27298     +95
applet_names                                        2297    2306      +9
applet_main                                         1352    1356      +4
applet_nameofs                                       676     678      +2
applet_install_loc                                   169     170      +1
------------------------------------------------------------------------------
(add/remove: 3/0 grow/shrink: 5/0 up/down: 558/0)             Total: 558 bytes

Signed-off-by: Sergey Naumov <sknaumov@gmail.com>
Signed-off-by: Denys Vlasenko <dvlasenk@redhat.com>

1 files changed, 180 insertions, 0 deletions

diff --git a/util-linux/blockdev.c b/util-linux/blockdev.c
new file mode 100644
index 000000000..eb5914b58
--- /dev/null
+++ b/util-linux/blockdev.c
@@ -0,0 +1,180 @@
+/*
+ * blockdev implementation for busybox
+ *
+ * Copyright (C) 2010 Sergey Naumov <sknaumov@gmail.com>
+ *
+ * Licensed under GPLv2, see file License in this tarball for details.
+ */
+
+//applet:IF_BLOCKDEV(APPLET(blockdev, _BB_DIR_SBIN, _BB_SUID_DROP))
+
+//kbuild:lib-$(CONFIG_BLOCKDEV) += blockdev.o
+
+//config:config BLOCKDEV
+//config:	bool "blockdev"
+//config:	default y
+//config:	help
+//config:	  Performs some ioctls with block devices.
+
+//usage:#define blockdev_trivial_usage
+//usage:	"OPTION [OPTARG] DEVICE"
+//usage:#define blockdev_full_usage "\n\n"
+//usage:       "Options:"
+//usage:     "\n	--setro		Set ro"
+//usage:     "\n	--setrw		Set rw"
+//usage:     "\n	--getro		Get ro"
+//usage:     "\n	--getss		Get sector size"
+//usage:     "\n	--getbsz	Get block size"
+//usage:     "\n	--setbsz BYTES	Set block size"
+//usage:     "\n	--getsize	Get device size in 512-byte sectors"
+//usage:     "\n	--getsize64	Get device size in bytes"
+//usage:     "\n	--flushbufs	Flush buffers"
+//usage:     "\n	--rereadpt	Reread partition table"
+
+
+#include "libbb.h"
+#include <linux/fs.h>
+
+enum {
+	ARG_NONE   = 0,
+	ARG_INT    = 1,
+	ARG_ULONG  = 2,
+	ARG_ULLONG = 3,
+	ARG_MASK   = 3,
+
+	FL_USRARG   = 4, /* argument is provided by user */
+	FL_NORESULT = 8,
+};
+
+struct bdc {
+	uint32_t   ioc;                       /* ioctl code */
+	const char name[sizeof("flushbufs")]; /* "--setfoo" wothout "--" */
+	uint8_t    flags;
+	int8_t     argval;                    /* default argument value */
+};
+
+static const struct bdc bdcommands[] = {
+	{
+		.ioc = BLKROSET,
+		.name = "setro",
+		.flags = ARG_INT + FL_NORESULT,
+		.argval = 1,
+	},{
+		.ioc = BLKROSET,
+		.name = "setrw",
+		.flags = ARG_INT + FL_NORESULT,
+		.argval = 0,
+	},{
+		.ioc = BLKROGET,
+		.name = "getro",
+		.flags = ARG_INT,
+		.argval = -1,
+	},{
+		.ioc = BLKSSZGET,
+		.name = "getss",
+		.flags = ARG_INT,
+		.argval = -1,
+	},{
+		.ioc = BLKBSZGET,
+		.name = "getbsz",
+		.flags = ARG_INT,
+		.argval = -1,
+	},{
+		.ioc = BLKBSZSET,
+		.name = "setbsz",
+		.flags = ARG_INT + FL_NORESULT + FL_USRARG,
+		.argval = 0,
+	},{
+		.ioc = BLKGETSIZE,
+		.name = "getsize",
+		.flags = ARG_ULONG,
+		.argval = -1,
+	},{
+		.ioc = BLKGETSIZE64,
+		.name = "getsize64",
+		.flags = ARG_ULLONG,
+		.argval = -1,
+	},{
+		.ioc = BLKFLSBUF,
+		.name = "flushbufs",
+		.flags = ARG_NONE + FL_NORESULT,
+		.argval = 0,
+	},{
+		.ioc = BLKRRPART,
+		.name = "rereadpt",
+		.flags = ARG_NONE + FL_NORESULT,
+		.argval = 0,
+	}
+};
+
+static const struct bdc *find_cmd(const char *s)
+{
+	int j;
+	if (*s++ == '-')
+		if (*s++ == '-')
+			for (j = 0; j < ARRAY_SIZE(bdcommands); j++)
+				if (strcmp(s, bdcommands[j].name) == 0)
+					return &bdcommands[j];
+	bb_show_usage();
+}
+
+int blockdev_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
+int blockdev_main(int argc, char **argv)
+{
+	const struct bdc *bdcmd;
+	void *ioctl_ptr;
+	int fd;
+	int iarg;
+	unsigned long lu;
+	unsigned long long llu;
+
+	if ((unsigned)(argc - 3) > 1) /* must have 2 or 3 args */
+		bb_show_usage();
+
+	bdcmd = find_cmd(*++argv);
+
+	llu = (int)bdcmd->argval;
+	if (bdcmd->flags & FL_USRARG)
+		llu = xatoi_positive(*++argv);
+	lu = llu;
+	iarg = llu;
+
+	if (!*++argv)
+		bb_show_usage();
+	fd = xopen(*argv, O_RDONLY);
+
+	ioctl_ptr = NULL;
+	switch (bdcmd->flags & ARG_MASK) {
+	case ARG_INT:
+		ioctl_ptr =  &iarg;
+		break;
+	case ARG_ULONG:
+		ioctl_ptr = &lu;
+		break;
+	case ARG_ULLONG:
+		ioctl_ptr = &llu;
+		break;
+	}
+
+	if (ioctl(fd, bdcmd->ioc, ioctl_ptr) == -1)
+		bb_simple_perror_msg_and_die(*argv);
+
+	switch (bdcmd->flags & (ARG_MASK+FL_NORESULT)) {
+	case ARG_INT:
+		/* Smaller code when we use long long
+		 * (gcc tail-merges printf call)
+		 */
+		printf("%lld\n", (long long)iarg);
+		break;
+	case ARG_ULONG:
+		llu = lu;
+		/* FALLTHROUGH */
+	case ARG_ULLONG:
+		printf("%llu\n", llu);
+		break;
+	}
+
+	if (ENABLE_FEATURE_CLEAN_UP)
+		close(fd);
+	return EXIT_SUCCESS;
+}