marko10_000
/
btrfs-progs
forked from Mirrors/btrfs-progs
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

btrfs-progs: extend pretty printers with unit mode 

The functionality of pretty unit printing was duplicated by
df_pretty_sizes, merge it with pretty_size and enhance the interface
with more suffix mode. Raw, binary or decimal.

Signed-off-by: David Sterba <dsterba@suse.cz>
master
David Sterba 2014-04-28 18:04:48 +02:00
parent db5d44d132
commit 6f23efdf27
2 changed files with 67 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

77
utils.c
View File

@ -1294,35 +1294,62 @@ out:
return ret;
}
static char *size_strs[] = { "B", "KiB", "MiB", "GiB", "TiB", "PiB", "EiB"};
int pretty_size_snprintf(u64 size, char *str, size_t str_bytes)
{
int num_divs = 0;
float fraction;
static const char const *unit_suffix_binary[] =
{ "B", "KiB", "MiB", "GiB", "TiB", "PiB", "EiB"};
static const char const *unit_suffix_decimal[] =
{ "B", "KB", "MB", "GB", "TB", "PB", "EB"};
if (str_bytes == 0)
int pretty_size_snprintf(u64 size, char *str, size_t str_size, int unit_mode)
{
int num_divs;
float fraction;
int base = 0;
const char const **suffix = NULL;
u64 last_size;
if (str_size == 0)
return 0;
if( size < 1024 ){
fraction = size;
num_divs = 0;
} else {
u64 last_size = size;
num_divs = 0;
while(size >= 1024){
last_size = size;
size /= 1024;
num_divs ++;
}
if (num_divs >= ARRAY_SIZE(size_strs)) {
str[0] = '\0';
return -1;
}
fraction = (float)last_size / 1024;
if (unit_mode == UNITS_RAW) {
snprintf(str, str_size, "%llu", size);
return 0;
}
return snprintf(str, str_bytes, "%.2f%s", fraction,
size_strs[num_divs]);
if (unit_mode == UNITS_BINARY) {
base = 1024;
suffix = unit_suffix_binary;
} else if (unit_mode == UNITS_DECIMAL) {
base = 1000;
suffix = unit_suffix_decimal;
}
/* Unknown mode */
if (!base) {
fprintf(stderr, "INTERNAL ERROR: unknown unit base, mode %d",
unit_mode);
assert(0);
return -1;
}
num_divs = 0;
last_size = size;
while (size >= base) {
last_size = size;
size /= base;
num_divs++;
}
if (num_divs >= ARRAY_SIZE(unit_suffix_binary)) {
str[0] = '\0';
printf("INTERNAL ERROR: unsupported unit suffix, index %d\n",
num_divs);
assert(0);
return -1;
}
fraction = (float)last_size / base;
return snprintf(str, str_size, "%.2f%s", fraction, suffix[num_divs]);
}
/*

21
utils.h
View File

@ -47,6 +47,14 @@ int check_argc_max(int nargs, int expected);
void fixup_argv0(char **argv, const char *token);
void set_argv0(char **argv);
/*
* Output mode of byte units
*/
#define UNITS_RAW (1)
#define UNITS_BINARY (2)
#define UNITS_DECIMAL (3)
#define UNITS_HUMAN UNITS_BINARY
int make_btrfs(int fd, const char *device, const char *label,
char *fs_uuid, u64 blocks[6], u64 num_bytes, u32 nodesize,
u32 leafsize, u32 sectorsize, u32 stripesize, u64 features);
@ -68,12 +76,13 @@ int check_mounted_where(int fd, const char *file, char *where, int size,
int btrfs_device_already_in_root(struct btrfs_root *root, int fd,
int super_offset);
int pretty_size_snprintf(u64 size, char *str, size_t str_bytes);
#define pretty_size(size) \
({ \
static __thread char _str[24]; \
(void)pretty_size_snprintf((size), _str, sizeof(_str)); \
_str; \
int pretty_size_snprintf(u64 size, char *str, size_t str_bytes, int unit_mode);
#define pretty_size(size) pretty_size_mode(size, UNITS_BINARY)
#define pretty_size_mode(size, mode) \
({ \
static __thread char _str[32]; \
(void)pretty_size_snprintf((size), _str, sizeof(_str), mode); \
_str; \
})
int get_mountpt(char *dev, char *mntpt, size_t size);