openkylin
/
e2fsprogs
mirror of https://gitee.com/openkylin/e2fsprogs.git
9
0
Fork 0
Code Issues Projects Releases Wiki Activity
e2fsprogs/misc/blkid.c

434 lines
9.3 KiB
C
Raw Permalink Blame History

/*
* blkid.c - User command-line interface for libblkid
*
* Copyright (C) 2001 Andreas Dilger
*
* %Begin-Header%
* This file may be redistributed under the terms of the
* GNU Lesser General Public License.
* %End-Header%
*/
#include "config.h"
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#ifdef HAVE_TERMIOS_H
#include <termios.h>
#endif
#ifdef HAVE_TERMIO_H
#include <termio.h>
#endif
#ifdef HAVE_SYS_IOCTL_H
#include <sys/ioctl.h>
#endif
#ifdef HAVE_GETOPT_H
#include <getopt.h>
#else
extern int getopt(int argc, char * const argv[], const char *optstring);
extern char *optarg;
extern int optind;
#endif
#define OUTPUT_VALUE_ONLY 0x0001
#define OUTPUT_DEVICE_ONLY 0x0002
#define OUTPUT_PRETTY_LIST 0x0004
#include "ext2fs/ext2fs.h"
#include "blkid/blkid.h"
static const char *progname = "blkid";
static void print_version(FILE *out)
{
fprintf(out, "%s %s (%s)\n", progname, BLKID_VERSION, BLKID_DATE);
}
static void usage(int error)
{
FILE *out = error ? stderr : stdout;
print_version(out);
fprintf(out,
"usage:\t%s [-c <file>] [-ghlLv] [-o format] "
"[-s <tag>] [-t <token>]\n [-w <file>] [dev ...]\n"
"\t-c\tcache file (default: /etc/blkid.tab, /dev/null = none)\n"
"\t-h\tprint this usage message and exit\n"
"\t-g\tgarbage collect the blkid cache\n"
"\t-s\tshow specified tag(s) (default show all tags)\n"
"\t-t\tfind device with a specific token (NAME=value pair)\n"
"\t-l\tlookup the the first device with arguments specified by -t\n"
"\t-v\tprint version and exit\n"
"\t-w\twrite cache to different file (/dev/null = no write)\n"
"\tdev\tspecify device(s) to probe (default: all devices)\n",
progname);
exit(error);
}
/*
* This function does "safe" printing. It will convert non-printable
* ASCII characters using '^' and M- notation.
*/
static void safe_print(const char *cp, int len)
{
unsigned char ch;
if (len < 0)
len = strlen(cp);
while (len--) {
ch = *cp++;
if (ch > 128) {
fputs("M-", stdout);
ch -= 128;
}
if ((ch < 32) || (ch == 0x7f)) {
fputc('^', stdout);
ch ^= 0x40; /* ^@, ^A, ^B; ^? for DEL */
}
if (ch != '"') {
fputc(ch, stdout);
}
}
}
static int get_terminal_width(void)
{
#ifdef TIOCGSIZE
struct ttysize t_win;
#endif
#ifdef TIOCGWINSZ
struct winsize w_win;
#endif
const char *cp;
int width = 80;
#ifdef TIOCGSIZE
if (ioctl (0, TIOCGSIZE, &t_win) == 0) {
width = t_win.ts_cols;
goto got_it;
}
#endif
#ifdef TIOCGWINSZ
if (ioctl (0, TIOCGWINSZ, &w_win) == 0) {
width = w_win.ws_col;
goto got_it;
}
#endif
cp = getenv("COLUMNS");
if (cp)
width = atoi(cp);
got_it:
if (width > 4096)
return 4096; /* sanity check */
return width;
}
static int pretty_print_word(const char *str, int max_len,
int left_len, int overflow_nl)
{
int len = strlen(str) + left_len;
int ret = 0;
fputs(str, stdout);
if (overflow_nl && len > max_len) {
fputc('\n', stdout);
len = 0;
} else if (len > max_len)
ret = len - max_len;
do {
fputc(' ', stdout);
} while (len++ < max_len);
return ret;
}
static void pretty_print_line(const char *device, const char *fs_type,
const char *label, const char *mtpt,
const char *uuid)
{
static int device_len = 10, fs_type_len = 7;
static int label_len = 8, mtpt_len = 14;
static int term_width = -1;
int len, w;
if (term_width < 0) {
term_width = get_terminal_width();
if (term_width > 80) {
term_width -= 80;
w = term_width / 10;
if (w > 8)
w = 8;
term_width -= 2*w;
label_len += w;
fs_type_len += w;
w = term_width/2;
device_len += w;
mtpt_len +=w;
}
}
len = pretty_print_word(device, device_len, 0, 1);
len = pretty_print_word(fs_type, fs_type_len, len, 0);
len = pretty_print_word(label, label_len, len, 0);
len = pretty_print_word(mtpt, mtpt_len, len, 0);
fputs(uuid, stdout);
fputc('\n', stdout);
}
static void pretty_print_dev(blkid_dev dev)
{
blkid_tag_iterate iter;
const char *type, *value, *devname;
const char *uuid = "", *fs_type = "", *label = "";
int len, mount_flags;
char mtpt[80];
errcode_t retval;
if (dev == NULL) {
pretty_print_line("device", "fs_type", "label",
"mount point", "UUID");
for (len=get_terminal_width()-1; len > 0; len--)
fputc('-', stdout);
fputc('\n', stdout);
return;
}
devname = blkid_dev_devname(dev);
if (access(devname, F_OK))
return;
/* Get the uuid, label, type */
iter = blkid_tag_iterate_begin(dev);
while (blkid_tag_next(iter, &type, &value) == 0) {
if (!strcmp(type, "UUID"))
uuid = value;
if (!strcmp(type, "TYPE"))
fs_type = value;
if (!strcmp(type, "LABEL"))
label = value;
}
blkid_tag_iterate_end(iter);
/* Get the mount point */
mtpt[0] = 0;
retval = ext2fs_check_mount_point(devname, &mount_flags,
mtpt, sizeof(mtpt));
if (retval == 0) {
if (mount_flags & EXT2_MF_MOUNTED) {
if (!mtpt[0])
strcpy(mtpt, "(mounted, mtpt unknown)");
} else if (mount_flags & EXT2_MF_BUSY)
strcpy(mtpt, "(in use)");
else
strcpy(mtpt, "(not mounted)");
}
pretty_print_line(devname, fs_type, label, mtpt, uuid);
}
static void print_tags(blkid_dev dev, char *show[], int numtag, int output)
{
blkid_tag_iterate iter;
const char *type, *value;
int i, first = 1;
if (!dev)
return;
if (output & OUTPUT_PRETTY_LIST) {
pretty_print_dev(dev);
return;
}
if (output & OUTPUT_DEVICE_ONLY) {
printf("%s\n", blkid_dev_devname(dev));
return;
}
iter = blkid_tag_iterate_begin(dev);
while (blkid_tag_next(iter, &type, &value) == 0) {
if (numtag && show) {
for (i=0; i < numtag; i++)
if (!strcmp(type, show[i]))
break;
if (i >= numtag)
continue;
}
if (output & OUTPUT_VALUE_ONLY) {
fputs(value, stdout);
fputc('\n', stdout);
} else {
if (first) {
printf("%s: ", blkid_dev_devname(dev));
first = 0;
}
fputs(type, stdout);
fputs("=\"", stdout);
safe_print(value, -1);
fputs("\" ", stdout);
}
}
blkid_tag_iterate_end(iter);
if (!first && !(output & OUTPUT_VALUE_ONLY))
printf("\n");
}
int main(int argc, char **argv)
{
blkid_cache cache = NULL;
char *devices[128] = { NULL, };
char *show[128] = { NULL, };
char *search_type = NULL, *search_value = NULL;
char *read = NULL;
char *write = NULL;
unsigned int numdev = 0, numtag = 0;
int version = 0;
int err = 4;
unsigned int i;
int output_format = 0;
int lookup = 0, gc = 0;
int c;
while ((c = getopt (argc, argv, "c:f:ghlLo:s:t:w:v")) != EOF)
switch (c) {
case 'c':
read = optarg;
if (!write)
write = read;
break;
case 'l':
lookup++;
break;
case 'L':
output_format = OUTPUT_PRETTY_LIST;
break;
case 'g':
gc = 1;
break;
case 'o':
if (!strcmp(optarg, "value"))
output_format = OUTPUT_VALUE_ONLY;
else if (!strcmp(optarg, "device"))
output_format = OUTPUT_DEVICE_ONLY;
else if (!strcmp(optarg, "list"))
output_format = OUTPUT_PRETTY_LIST;
else if (!strcmp(optarg, "full"))
output_format = 0;
else {
fprintf(stderr, "Invalid output format %s. "
"Choose from value,\n\t"
"device, list, or full\n", optarg);
exit(1);
}
break;
case 's':
if (numtag >= sizeof(show) / sizeof(*show)) {
fprintf(stderr, "Too many tags specified\n");
usage(err);
}
show[numtag++] = optarg;
break;
case 't':
if (search_type) {
fprintf(stderr, "Can only search for "
"one NAME=value pair\n");
usage(err);
}
if (blkid_parse_tag_string(optarg,
&search_type,
&search_value)) {
fprintf(stderr, "-t needs NAME=value pair\n");
usage(err);
}
break;
case 'v':
version = 1;
break;
case 'w':
write = optarg;
break;
case 'h':
err = 0;
/* fallthrough */
default:
usage(err);
}
while (optind < argc)
devices[numdev++] = argv[optind++];
if (version) {
print_version(stdout);
goto exit;
}
if (blkid_get_cache(&cache, read) < 0)
goto exit;
err = 2;
if (gc) {
blkid_gc_cache(cache);
goto exit;
}
if (output_format & OUTPUT_PRETTY_LIST)
pretty_print_dev(NULL);
if (lookup) {
blkid_dev dev;
if (!search_type) {
fprintf(stderr, "The lookup option requires a "
"search type specified using -t\n");
exit(1);
}
/* Load any additional devices not in the cache */
for (i = 0; i < numdev; i++)
blkid_get_dev(cache, devices[i], BLKID_DEV_NORMAL);
if ((dev = blkid_find_dev_with_tag(cache, search_type,
search_value))) {
print_tags(dev, show, numtag, output_format);
err = 0;
}
/* If we didn't specify a single device, show all available devices */
} else if (!numdev) {
blkid_dev_iterate iter;
blkid_dev dev;
blkid_probe_all(cache);
iter = blkid_dev_iterate_begin(cache);
blkid_dev_set_search(iter, search_type, search_value);
while (blkid_dev_next(iter, &dev) == 0) {
dev = blkid_verify(cache, dev);
if (!dev)
continue;
print_tags(dev, show, numtag, output_format);
err = 0;
}
blkid_dev_iterate_end(iter);
/* Add all specified devices to cache (optionally display tags) */
} else for (i = 0; i < numdev; i++) {
blkid_dev dev = blkid_get_dev(cache, devices[i],
BLKID_DEV_NORMAL);
if (dev) {
if (search_type &&
!blkid_dev_has_tag(dev, search_type,
search_value))
continue;
print_tags(dev, show, numtag, output_format);
err = 0;
}
}
exit:
free(search_type);
free(search_value);
blkid_put_cache(cache);
return err;
}
Reference in New Issue View Git Blame Copy Permalink