From a9393215d1ef70f52450f866c2ca764bee34ef11 Mon Sep 17 00:00:00 2001 From: Andre Noll Date: Sat, 1 Nov 2008 15:26:57 +0100 Subject: [PATCH] Fix the global summary. This broke by the previous patch. --- select.c | 104 ++++++++++++++++++++++++++++++++++++------------------- 1 file changed, 68 insertions(+), 36 deletions(-) diff --git a/select.c b/select.c index 7b33383..577ff72 100644 --- a/select.c +++ b/select.c @@ -16,14 +16,6 @@ #include "error.h" #include "portable_io.h" -/** Global dir count. */ -static uint64_t num_dirs; -/** Global files count. */ -static uint64_t num_files; -/** Global bytes count. */ -static uint64_t num_bytes; - - /** The decimal representation of an uint64_t never exceeds that size. */ #define FORMATED_VALUE_SIZE 25 @@ -86,6 +78,17 @@ struct global_list_info { struct format_info *fi; }; +struct global_summary_info { + /** Global dir count. */ + uint64_t num_dirs; + /** Global files count. */ + uint64_t num_files; + /** Global bytes count. */ + uint64_t num_bytes; + int ret; + int osl_errno; +}; + enum user_stats_flags { USF_PRINT_DIRNAME = 1, USF_PRINT_BYTES = 2, @@ -218,11 +221,32 @@ static int get_dir_name_of_row(struct osl_row *dir_table_row, char **name) return get_dir_name_by_number((uint64_t *)obj.data, name); } +static int get_num_files_of_row(struct osl_row *row, uint64_t *num_files) +{ + struct osl_object obj; + int ret = osl(osl_get_object(dir_table, row, DT_FILES, &obj)); + if (ret < 0) + return ret; + *num_files = *(uint64_t *)obj.data; + return 1; +} + +static int get_num_bytes_of_row(struct osl_row *row, uint64_t *num_bytes) +{ + struct osl_object obj; + int ret = osl(osl_get_object(dir_table, row, DT_BYTES, &obj)); + if (ret < 0) + return ret; + *num_bytes = *(uint64_t *)obj.data; + return 1; +} + + static int user_stats_loop_function(struct osl_row *row, void *data) { struct user_stats_info *usi = data; struct osl_object obj; - int ret, summary = usi->flags & GSF_COMPUTE_SUMMARY; + int ret, summary = usi->flags & USF_COMPUTE_SUMMARY; char formated_value[FORMATED_VALUE_SIZE]; check_signals(); @@ -322,18 +346,48 @@ static int adu_loop_reverse(struct osl_table *t, unsigned col_num, void *private return check_loop_return(ret, *loop_ret, *loop_osl_errno); } +static int global_summary_loop_function(struct osl_row *row, void *data) +{ + struct global_summary_info *gsi = data; + int ret; + uint64_t num; + + ret = get_num_files_of_row(row, &num); + if (ret < 0) + goto err; + gsi->num_files += num; + + ret = get_num_bytes_of_row(row, &num); + if (ret < 0) + goto err; + gsi->num_bytes += num; + gsi->num_dirs++; + return 1; +err: + gsi->ret = ret; + gsi->osl_errno = (ret == -E_OSL)? osl_errno : 0; + return -1; +} + static int print_global_summary(struct format_info *fi) { int ret; char *buf; + struct global_summary_info gsi = {.num_dirs = 0}; + union atom_value values[] = { - [gsa_dirs] = {.num_value = (long long unsigned)num_dirs}, - [gsa_files] = {.num_value = (long long unsigned)num_files}, - [gsa_size] = {.num_value = (long long unsigned)num_bytes} + [gsa_dirs] = {.num_value = 0ULL}, + [gsa_files] = {.num_value = 0ULL}, + [gsa_size] = {.num_value = 0ULL} }; - if (select_conf.no_global_summary_given) - return 1; + ret = adu_loop_reverse(dir_table, DT_BYTES, &gsi, + global_summary_loop_function, &gsi.ret, &gsi.osl_errno); + if (ret < 0) + return ret; + values[gsa_dirs].num_value = (long long unsigned)gsi.num_dirs; + values[gsa_files].num_value = (long long unsigned)gsi.num_files; + values[gsa_size].num_value = (long long unsigned)gsi.num_bytes; if (!select_conf.no_headers_given) { ret = output("Global summary\n"); if (ret < 0) @@ -415,8 +469,6 @@ static int (*summary_comparators[])(const void *, const void *) = { static int print_user_summary(struct format_info *fi) { - if (select_conf.no_user_summary_given) - return 1; if (!select_conf.no_headers_given) { int ret = output("User summary\n"); if (ret < 0) @@ -489,26 +541,6 @@ static int print_user_lists(void) return for_each_admissible_user(print_user_list, NULL); } -static int get_num_files_of_row(struct osl_row *row, uint64_t *num_files) -{ - struct osl_object obj; - int ret = osl(osl_get_object(dir_table, row, DT_FILES, &obj)); - if (ret < 0) - return ret; - *num_files = *(uint64_t *)obj.data; - return 1; -} - -static int get_num_bytes_of_row(struct osl_row *row, uint64_t *num_bytes) -{ - struct osl_object obj; - int ret = osl(osl_get_object(dir_table, row, DT_BYTES, &obj)); - if (ret < 0) - return ret; - *num_bytes = *(uint64_t *)obj.data; - return 1; -} - static int global_list_loop_function(struct osl_row *row, void *data) { struct global_list_info *gli = data; -- 2.39.2