- DIR *dir;
- struct dirent *entry;
- int ret, cwd_fd, ret2;
- uint64_t dir_size = 0, dir_files = 0;
- uint64_t this_dir_num = ++num_dirs;
-
- DEBUG_LOG("----------------- %llu: %s\n", (long long unsigned)num_dirs, dirname);
- ret = para_opendir(dirname, &dir, &cwd_fd);
- if (ret < 0) {
- if (ret != -ERRNO_TO_ERROR(EACCES))
- return ret;
- WARNING_LOG("permission denied for %s\n", dirname);
- return 1;
- }
- while ((entry = readdir(dir))) {
- mode_t m;
- struct stat s;
- uint32_t uid;
- uint64_t size;
- struct user_info *ui;
-
- if (!strcmp(entry->d_name, "."))
- continue;
- if (!strcmp(entry->d_name, ".."))
- continue;
- if (lstat(entry->d_name, &s) == -1) {
- WARNING_LOG("lstat error for %s/%s\n", dirname,
- entry->d_name);
- continue;
- }
- m = s.st_mode;
- if (!S_ISREG(m) && !S_ISDIR(m))
- continue;
- if (S_ISDIR(m)) {
- ret = scan_dir(entry->d_name, &this_dir_num);
- if (ret < 0)
- goto out;
- continue;
- }
- /* regular file */
- size = s.st_size;
- dir_size += size;
- num_bytes += size;
- dir_files++;
- num_files++;
- uid = s.st_uid;
- ret = search_uid(uid, CREATE_USER_TABLE | OPEN_USER_TABLE, &ui);
- if (ret < 0)
- goto out;
- ui->bytes += size;
- ui->files++;
- ret = update_user_row(ui->table, this_dir_num, &size);
- if (ret < 0)
- goto out;
- }
- ret = add_directory(dirname, &this_dir_num, parent_dir_num,
- &dir_size, &dir_files);
-out:
- closedir(dir);
- ret2 = para_fchdir(cwd_fd);
- if (ret2 < 0 && ret >= 0)
- ret = ret2;
- close(cwd_fd);
- return ret;
-}
-
-static int get_dir_name_by_number(uint64_t *dirnum, char **name)
-{
- char *result = NULL, *tmp;
- struct osl_row *row;
- uint64_t val = *dirnum;
- struct osl_object obj = {.data = &val, .size = sizeof(val)};
- int ret;
-
-again:
- ret = osl_get_row(dir_table, DT_NUM, &obj, &row);
- if (ret < 0)
- goto out;
- ret = osl_get_object(dir_table, row, DT_NAME, &obj);
- if (ret < 0)
- goto out;
- if (result) {
- tmp = make_message("%s/%s", (char *)obj.data, result);
- free(result);
- result = tmp;
- } else
- result = para_strdup((char *)obj.data);
- ret = osl_get_object(dir_table, row, DT_PARENT_NUM, &obj);
- if (ret < 0)
- goto out;
- val = *(uint64_t *)obj.data;
- if (val)
- goto again;
-out:
- if (ret < 0) {
- free(result);
- *name = NULL;
- } else
- *name = result;
- return ret;
-}
-
-static int get_dir_name_of_row(struct osl_row *dir_table_row, char **name)
-{
- struct osl_object obj;
- int ret;
- char *this_dir, *prefix = NULL;
-
- *name = NULL;
- ret = osl_get_object(dir_table, dir_table_row, DT_NAME, &obj);
- if (ret < 0)
- return ret;
- this_dir = para_strdup((char *)obj.data);
- ret = osl_get_object(dir_table, dir_table_row, DT_PARENT_NUM, &obj);
- if (ret < 0)
- goto out;
- if (!*(uint64_t *)obj.data) {
- *name = this_dir;
- return 1;
- }
- ret = get_dir_name_by_number((uint64_t *)obj.data, &prefix);
- if (ret < 0)
- goto out;
- *name = make_message("%s/%s", prefix, this_dir);
- free(prefix);
- ret = 1;
-out:
- free(this_dir);
- return ret;
-}
-
-const uint64_t size_unit_divisors[] = {
- [size_unit_arg_b] = 1ULL,
- [size_unit_arg_k] = 1024ULL,
- [size_unit_arg_m] = 1024ULL * 1024ULL,
- [size_unit_arg_g] = 1024ULL * 1024ULL * 1024ULL,
- [size_unit_arg_t] = 1024ULL * 1024ULL * 1024ULL * 1024ULL,
-};
-
-const uint64_t count_unit_divisors[] = {
-
- [count_unit_arg_n] = 1ULL,
- [count_unit_arg_k] = 1000ULL,
- [count_unit_arg_m] = 1000ULL * 1000ULL,
- [count_unit_arg_g] = 1000ULL * 1000ULL * 1000ULL,
- [count_unit_arg_t] = 1000ULL * 1000ULL * 1000ULL * 1000ULL,
-};
-
-const char size_unit_abbrevs[] = " BKMGT";
-const char count_unit_abbrevs[] = " KMGT";
-
-static void format_size_value(enum enum_size_unit unit, uint64_t value, char *result)
-{
- if (unit == size_unit_arg_h) /* human readable */
- for (unit = size_unit_arg_b; unit < size_unit_arg_t && value > size_unit_divisors[unit + 1]; unit++)
- ; /* nothing */
- sprintf(result, "%llu%c", (long long unsigned)value / size_unit_divisors[unit], size_unit_abbrevs[unit]);