2 * Copyright (C) 2008 Andre Noll <maan@systemlinux.org>
4 * Licensed under the GPL v2. For licencing details see COPYING.
7 /** \file create.c The create mode of adu. */
9 #include <dirent.h> /* readdir() */
11 #include "gcc-compat.h"
16 #include "portable_io.h"
18 /* Id of the device containing the base dir. */
19 static dev_t device_id;
21 static int write_uid(struct user_info *ui, void *data)
25 write_u32(*p, ui->uid);
26 *p += sizeof(uint32_t);
30 static int write_uid_list(void)
32 char *buf, *p, *filename;
35 size_t size = num_uids * sizeof(uint32_t);
40 buf = p = adu_malloc(size);
41 ret = for_each_admissible_user(write_uid, &p);
44 filename = get_uid_list_name();
45 ret = adu_write_file(filename, buf, size);
52 static int add_directory(char *dirname, uint64_t *dir_num, uint64_t *parent_dir_num,
53 uint64_t *dir_size, uint64_t *dir_files)
55 struct osl_object dir_objects[NUM_DT_COLUMNS];
57 INFO_LOG("adding #%llu: %s\n", (long long unsigned)*dir_num, dirname);
58 dir_objects[DT_NAME].data = dirname;
59 dir_objects[DT_NAME].size = strlen(dirname) + 1;
60 dir_objects[DT_NUM].data = dir_num;
61 dir_objects[DT_NUM].size = sizeof(*dir_num);
62 dir_objects[DT_PARENT_NUM].data = parent_dir_num;
63 dir_objects[DT_PARENT_NUM].size = sizeof(*parent_dir_num);
64 dir_objects[DT_BYTES].data = dir_size;
65 dir_objects[DT_BYTES].size = sizeof(*dir_size);
66 dir_objects[DT_FILES].data = dir_files;
67 dir_objects[DT_FILES].size = sizeof(*dir_files);
68 return osl(osl_add_row(dir_table, dir_objects));
71 static int update_user_row(struct osl_table *t, uint64_t dir_num,
75 struct osl_object obj = {.data = &dir_num, .size = sizeof(dir_num)};
77 int ret = osl(osl_get_row(t, UT_DIR_NUM, &obj, &row));
79 if (ret == -E_OSL && osl_errno != E_OSL_RB_KEY_NOT_FOUND)
81 if (ret < 0) { /* this is the first file we add */
82 struct osl_object objects[NUM_UT_COLUMNS];
83 uint64_t num_files = 1;
85 objects[UT_DIR_NUM].data = &dir_num;
86 objects[UT_DIR_NUM].size = sizeof(dir_num);
87 objects[UT_BYTES].data = add;
88 objects[UT_BYTES].size = sizeof(*add);
89 objects[UT_FILES].data = &num_files;
90 objects[UT_FILES].size = sizeof(num_files);
91 INFO_LOG("######################### ret: %d\n", ret);
92 ret = osl(osl_add_row(t, objects));
93 INFO_LOG("######################### ret: %d\n", ret);
95 } else { /* add size and increment file count */
97 struct osl_object obj1, obj2 = {.data = &num, .size = sizeof(num)};
99 ret = osl(osl_get_object(t, row, UT_BYTES, &obj1));
102 num = *(uint64_t *)obj1.data + *add;
103 ret = osl(osl_update_object(t, row, UT_BYTES, &obj2));
106 ret = osl(osl_get_object(t, row, UT_FILES, &obj1));
109 num = *(uint64_t *)obj1.data + 1;
110 return osl(osl_update_object(t, row, UT_FILES, &obj2));
114 static int scan_dir(char *dirname, uint64_t *parent_dir_num)
117 struct dirent *entry;
118 int ret, cwd_fd, ret2;
119 uint64_t dir_size = 0, dir_files = 0;
121 static uint64_t current_dir_num;
123 uint64_t this_dir_num = ++current_dir_num;
126 DEBUG_LOG("----------------- %llu: %s\n", (long long unsigned)current_dir_num, dirname);
127 ret = adu_opendir(dirname, &dir, &cwd_fd);
129 if (ret != -ERRNO_TO_ERROR(EACCES))
131 WARNING_LOG("permission denied for %s\n", dirname);
134 while ((entry = readdir(dir))) {
139 struct user_info *ui;
141 if (!strcmp(entry->d_name, "."))
143 if (!strcmp(entry->d_name, ".."))
145 if (lstat64(entry->d_name, &s) == -1) {
146 WARNING_LOG("lstat64 error for %s/%s (%s)\n",
147 dirname, entry->d_name, strerror(errno));
151 if (!S_ISREG(m) && !S_ISDIR(m))
154 if (conf.one_file_system_given && s.st_dev != device_id)
156 ret = scan_dir(entry->d_name, &this_dir_num);
166 ret = search_uid(uid, CREATE_USER_TABLE | OPEN_USER_TABLE, &ui);
171 ret = update_user_row(ui->table, this_dir_num, &size);
175 ret = add_directory(dirname, &this_dir_num, parent_dir_num,
176 &dir_size, &dir_files);
179 ret2 = adu_fchdir(cwd_fd);
180 if (ret2 < 0 && ret >= 0)
188 uint64_t zero = 0ULL;
192 if (lstat(conf.base_dir_arg, &statbuf) == -1)
193 return -ERRNO_TO_ERROR(errno);
194 if (!S_ISDIR(statbuf.st_mode))
195 return -ERRNO_TO_ERROR(ENOTDIR);
196 device_id = statbuf.st_dev;
197 create_hash_table(conf.hash_table_bits_arg);
198 ret = open_dir_table(1);
202 ret = scan_dir(conf.base_dir_arg, &zero);
205 ret = write_uid_list();