From dbf5a6da6a4295ce26edd1ce34fde567d19afa02 Mon Sep 17 00:00:00 2001 From: Ari Sundholm Date: Sat, 2 Jan 2016 01:18:32 +0100 Subject: blkdiscard: new applet function old new delta blkdiscard_main - 264 +264 Signed-off-by: Ari Sundholm Signed-off-by: Denys Vlasenko --- util-linux/blkdiscard.c | 83 +++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 83 insertions(+) create mode 100644 util-linux/blkdiscard.c diff --git a/util-linux/blkdiscard.c b/util-linux/blkdiscard.c new file mode 100644 index 0000000..ace88a1 --- /dev/null +++ b/util-linux/blkdiscard.c @@ -0,0 +1,99 @@ +/* + * Mini blkdiscard implementation for busybox + * + * Copyright (C) 2015 by Ari Sundholm and Tuxera Inc. + * + * Licensed under GPLv2 or later, see file LICENSE in this source tree. + */ + +//config:config BLKDISCARD +//config: bool "blkdiscard" +//config: default y +//config: help +//config: blkdiscard discards sectors on a given device. + +//kbuild:lib-$(CONFIG_BLKDISCARD) += blkdiscard.o +//applet:IF_BLKDISCARD(APPLET(blkdiscard, BB_DIR_USR_BIN, BB_SUID_DROP)) + +//usage:#define blkdiscard_trivial_usage +//usage: "[-o OFS] [-l LEN] [-s] DEVICE" +//usage:#define blkdiscard_full_usage "\n\n" +//usage: "Discard sectors on DEVICE\n" +//usage: "\n -o OFS Byte offset into device" +//usage: "\n -l LEN Number of bytes to discard" +//usage: "\n -s Perform a secure discard" +//usage: +//usage:#define blkdiscard_example_usage +//usage: "$ blkdiscard -o 0 -l 1G /dev/sdb" + +#include "libbb.h" +#include + +static const struct suffix_mult kMG_suffixes[] = { + { "kB", 1000 }, + { "kD", 1000 }, + { "k", 1024 }, + { "K", 1024 }, /* compat with coreutils dd (it also accepts KB and KD, TODO?) */ + { "MB", 1000000 }, + { "MD", 1000000 }, + { "M", 1024*1024 }, + { "GB", 1000000000 }, + { "GD", 1000000000 }, + { "G", 1024*1024*1024 }, + /* "D" suffix for decimal is not in coreutils manpage, looks like it's deprecated */ + /* coreutils also understands TPEZY suffixes for tera- and so on, with B suffix for decimal */ + { "", 0 } +}; + +int blkdiscard_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; +int blkdiscard_main(int argc UNUSED_PARAM, char **argv) +{ + unsigned opts; + const char *offset_str = "0"; + const char *length_str; + uint64_t offset; /* Leaving these two variables out does not */ + uint64_t length; /* shrink code size and hampers readability. */ + uint64_t range[2]; +// struct stat st; + int fd; + + enum { + OPT_OFFSET = (1 << 0), + OPT_LENGTH = (1 << 1), + OPT_SECURE = (1 << 2), + }; + + opt_complementary = "=1"; + opts = getopt32(argv, "o:l:s", &offset_str, &length_str); + argv += optind; + + fd = xopen(argv[0], O_RDWR|O_EXCL); +//Why bother, BLK[SEC]DISCARD will fail on non-blockdevs anyway? +// xfstat(fd, &st); +// if (!S_ISBLK(st.st_mode)) +// bb_error_msg_and_die("%s: not a block device", argv[0]); + + offset = xatoull_sfx(offset_str, kMG_suffixes); + + if (opts & OPT_LENGTH) + length = xatoull_sfx(length_str, kMG_suffixes); + else { + xioctl(fd, BLKGETSIZE64, &length); + length -= offset; + } + + range[0] = offset; + range[1] = length; + ioctl_or_perror_and_die(fd, + (opts & OPT_SECURE) ? BLKSECDISCARD : BLKDISCARD, + &range, + "%s: %s failed", + argv[0], + (opts & OPT_SECURE) ? "BLKSECDISCARD" : "BLKDISCARD" + ); + + if (ENABLE_FEATURE_CLEAN_UP) + close(fd); + + return EXIT_SUCCESS; +} -- cgit v0.12