-enum user_stats_flags {
- USF_PRINT_DIRNAME = 1,
- USF_PRINT_BYTES = 2,
- USF_PRINT_FILES = 4,
- USF_COMPUTE_SUMMARY = 8,
-};
-
-struct user_stats_info {
- uint32_t count;
- enum user_stats_flags flags;
- struct user_info *ui;
-};
-
-static int user_stats_loop_function(struct osl_row *row, void *data)
-{
- struct user_stats_info *usi = data;
- struct osl_row *dir_row;
- struct osl_object obj;
- int ret, summary = usi->flags & GSF_COMPUTE_SUMMARY;
-
- if (!usi->count && !summary)
- return -E_LOOP_COMPLETE;
- if (usi->count && (usi->flags & USF_PRINT_DIRNAME)) {
- char *dirname;
- ret = osl_get_object(usi->ui->table, row, UT_DIR_NUM, &obj);
- if (ret < 0)
- return ret;
- ret = osl_get_row(dir_table, DT_NUM, &obj, &dir_row);
- if (ret < 0)
- return ret;
- ret = osl_get_object(dir_table, dir_row, DT_NAME, &obj);
- if (ret < 0)
- return ret;
- dirname = obj.data;
- printf("%s%s",
- dirname,
- (usi->flags & (USF_PRINT_FILES | USF_PRINT_BYTES))?
- "\t" : "\n"
- );
- }
- if (summary || (usi->count && (usi->flags & USF_PRINT_FILES))) {
- uint64_t files;
- ret = osl_get_object(usi->ui->table, row, UT_FILES, &obj);
- if (ret < 0)
- return ret;
- files = *(uint64_t *)obj.data;
- if (usi->count && (usi->flags & USF_PRINT_FILES))
- printf("%llu%s",
- (long long unsigned)files,
- (usi->flags & USF_PRINT_BYTES)? "\t" : "\n"
- );
- if (summary)
- usi->ui->files += files;
- }
- if (summary || (usi->count && (usi->flags & USF_PRINT_BYTES))) {
- uint64_t bytes;
- ret = osl_get_object(usi->ui->table, row, UT_BYTES, &obj);
- if (ret < 0)
- return ret;
- bytes = *(uint64_t *)obj.data;
- if (usi->count && (usi->flags & USF_PRINT_BYTES))
- printf("%llu\n", (long long unsigned)bytes);
- if (summary) {
- usi->ui->bytes += bytes;
- usi->ui->dirs++;
- }
-
- }
- if (usi->count)
- usi->count--;
- return 1;
-}
-
-static void print_user_stats(void)
-{
- struct user_info *ui;
-
- FOR_EACH_USER(ui) {
- struct user_stats_info usi = {
- .count = 10,
- .ui = ui
- };
- if (!ui->table)
- continue;
- usi.flags = USF_PRINT_DIRNAME | USF_PRINT_BYTES | USF_COMPUTE_SUMMARY;
- printf("************************************************ uid %u\n",
- (unsigned) ui->uid);
- if (!ui->table)
- continue;
- printf("----------------- Largest dirs -------------------\n");
- osl_rbtree_loop_reverse(ui->table, UT_BYTES, &usi,
- user_stats_loop_function);
- printf("---------- dirs containing most files ------------\n");
- usi.count = 10;
- usi.flags = USF_PRINT_DIRNAME | USF_PRINT_FILES;
- osl_rbtree_loop_reverse(ui->table, UT_FILES, &usi,
- user_stats_loop_function);
- }
-}
-
-static int print_statistics(void)
-{
- int ret;
- struct global_stats_info gsi = {
- .count = 10,
- .flags = GSF_PRINT_DIRNAME | GSF_PRINT_BYTES | GSF_COMPUTE_SUMMARY
- };
-
- printf("----------------- Largest dirs -------------------\n");
- ret = osl_rbtree_loop_reverse(dir_table, DT_BYTES, &gsi,
- global_stats_loop_function);
- if (ret < 0 && ret != -E_LOOP_COMPLETE)
- return ret;
- gsi.count = 10;
-
- gsi.flags = GSF_PRINT_DIRNAME | GSF_PRINT_FILES;
- printf("---------- dirs containing most files ------------\n");
- ret = osl_rbtree_loop_reverse(dir_table, DT_FILES, &gsi,
- global_stats_loop_function);
- if (ret < 0 && ret != -E_LOOP_COMPLETE)
- return ret;
-
- printf("------------------ Global summary (dirs/files/bytes)\n"
- "%llu\t%llu\t%llu\n",
- (long long unsigned)num_dirs, (long long unsigned)num_files,
- (long long unsigned)num_bytes);
- print_user_stats();
- print_id_stats();
- return 1;
-}
-
-static int write_uid_list(void)
-{
- char *buf;
- uint32_t count = 0;
- struct user_info *ui;
- size_t size = num_uids * sizeof(uint32_t);
- int ret;
-
- if (!num_uids)
- return 0;
- buf = para_malloc(size);
- FOR_EACH_USER(ui) {
- if (!ui->table)
- continue;
- write_u32(buf + count++ * sizeof(uint32_t), ui->uid);
- }
- ret = para_write_file(UID_LIST, buf, size);
- free(buf);
- return ret;
-}