#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
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;
+};
+
+struct user_summary_info {
+ struct user_info *ui;
+ int ret;
+ int osl_errno;
+};
+
+
enum user_stats_flags {
USF_PRINT_DIRNAME = 1,
USF_PRINT_BYTES = 2,
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_user_files(struct osl_row *row, struct user_info *ui,
+ uint64_t *num_files)
+{
+ struct osl_object obj;
+ int ret = osl(osl_get_object(ui->table, row, UT_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 get_num_user_bytes(struct osl_row *row, struct user_info *ui,
+ uint64_t *num_bytes)
+{
+ struct osl_object obj;
+ int ret = osl(osl_get_object(ui->table, row, UT_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();
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)
return ret;
}
-static int print_user_summary_line(struct user_info *ui, __a_unused void *data)
+static int user_summary_loop_function(struct osl_row *row, void *data)
+{
+ struct user_summary_info *usi = data;
+ uint64_t num;
+ int ret;
+
+ ret = get_num_user_files(row, usi->ui, &num);
+ if (ret < 0)
+ goto err;
+ usi->ui->files += num;
+ ret = get_num_user_bytes(row, usi->ui, &num);
+ if (ret < 0)
+ goto err;
+ usi->ui->bytes += num;
+ usi->ui->dirs++;
+ return 1;
+err:
+ usi->ret = ret;
+ usi->osl_errno = (ret == -E_OSL)? osl_errno : 0;
+ return -1;
+}
+
+static int compute_user_summary(struct user_info *ui, __a_unused void *data)
+{
+ struct user_summary_info usi = {.ui = ui};
+
+ return adu_loop_reverse(ui->table, UT_BYTES, &usi, user_summary_loop_function,
+ &usi.ret, &usi.osl_errno);
+}
+
+static int print_user_summary_line(struct user_info *ui, void *data)
{
struct format_info *fi = data;
union atom_value values[] = {
ui->pw_name : "?"},
[usa_uid] = {.num_value = (long long unsigned)ui->uid},
[usa_dirs] = {.num_value = (long long unsigned)ui->dirs},
- [usa_files] = {.num_value = (long long unsigned)ui->files},
+ [usa_files] = {.num_value = (long long unsigned)ui->files},
[usa_size] = {.num_value = (long long unsigned)ui->bytes}
};
- char *buf = format_items(fi, values);
- int ret = output("%s", buf);
+ char *buf;
+ int ret;
+ buf = format_items(fi, values);
+ ret = output("%s", buf);
free(buf);
return ret;
}
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)
return ret;
}
+ int ret = for_each_admissible_user(compute_user_summary, fi);
+ if (ret < 0)
+ return ret;
sort_hash_table(summary_comparators[select_conf.user_summary_sort_arg]);
return for_each_admissible_user(print_user_summary_line, fi);
}
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;
if (ret < 0)
return ret;
}
- if (select_conf.sort_arg == sort_arg_files)
+ if (select_conf.list_sort_arg == list_sort_arg_file_count)
sort_column = DT_FILES;
return adu_loop_reverse(dir_table, sort_column, &gli,
global_list_loop_function, &gli.ret, &gli.osl_errno);
projects / adu.git / blobdiff