4 #include <ccan/talloc/talloc.h>
5 #include <ccan/str/str.h>
6 #include <ccan/str_talloc/str_talloc.h>
7 #include <ccan/talloc_link/talloc_link.h>
8 #include <ccan/hash/hash.h>
9 #include <ccan/htable/htable_type.h>
10 #include <ccan/grab_file/grab_file.h>
11 #include <ccan/noerr/noerr.h>
12 #include <ccan/foreach/foreach.h>
13 #include <ccan/asort/asort.h>
14 #include <ccan/array_size/array_size.h>
16 #include <sys/types.h>
26 static size_t dir_hash(const char *name)
28 return hash(name, strlen(name), 0);
31 static const char *manifest_name(const struct manifest *m)
36 static bool dir_cmp(const struct manifest *m, const char *dir)
38 return strcmp(m->dir, dir) == 0;
41 HTABLE_DEFINE_TYPE(struct manifest, manifest_name, dir_hash, dir_cmp,
43 static struct htable_manifest *manifests;
45 const char *get_ccan_file_contents(struct ccan_file *f)
48 f->contents = grab_file(f, f->fullname, &f->contents_size);
50 err(1, "Reading file %s", f->fullname);
55 char **get_ccan_file_lines(struct ccan_file *f)
58 f->lines = strsplit(f, get_ccan_file_contents(f), "\n");
60 /* FIXME: is f->num_lines necessary? */
61 f->num_lines = talloc_array_length(f->lines) - 1;
65 struct ccan_file *new_ccan_file(const void *ctx, const char *dir, char *name)
70 assert(dir[0] == '/');
72 f = talloc(ctx, struct ccan_file);
75 f->doc_sections = NULL;
76 for (i = 0; i < ARRAY_SIZE(f->compiled); i++)
77 f->compiled[i] = NULL;
78 f->name = talloc_steal(f, name);
79 f->fullname = talloc_asprintf(f, "%s/%s", dir, f->name);
85 static void add_files(struct manifest *m, const char *dir)
96 err(1, "Opening directory %s", dir[0] ? dir : ".");
98 while ((ent = readdir(d)) != NULL) {
101 struct list_head *dest;
104 if (ent->d_name[0] == '.')
107 f = new_ccan_file(m, m->dir,
108 talloc_asprintf(m, "%s%s",
110 if (lstat(f->name, &st) != 0)
111 err(1, "lstat %s", f->name);
113 if (S_ISDIR(st.st_mode)) {
114 size_t len = talloc_array_length(subs);
115 subs = talloc_realloc(m, subs, char *, len+1);
116 subs[len] = talloc_append_string(f->name, "/");
119 if (!S_ISREG(st.st_mode)) {
124 if (streq(f->name, "_info")) {
129 is_c_src = strends(f->name, ".c");
130 if (!is_c_src && !strends(f->name, ".h")) {
131 dest = &m->other_files;
132 } else if (!strchr(f->name, '/')) {
137 } else if (strstarts(f->name, "test/")) {
139 if (strstarts(f->name, "test/api"))
140 dest = &m->api_tests;
141 else if (strstarts(f->name, "test/run"))
142 dest = &m->run_tests;
143 else if (strstarts(f->name, "test/compile_ok"))
144 dest = &m->compile_ok_tests;
145 else if (strstarts(f->name, "test/compile_fail"))
146 dest = &m->compile_fail_tests;
148 dest = &m->other_test_c_files;
150 dest = &m->other_test_files;
152 dest = &m->other_files;
154 list_add(dest, &f->list);
158 /* Before we recurse, sanity check this is a ccan module. */
163 && list_empty(&m->c_files)
164 && list_empty(&m->h_files))
165 errx(1, "No _info, C or H files found here!");
167 for (i = 0; i < talloc_array_length(subs); i++)
168 add_files(m, subs[i]);
173 static int cmp_names(struct ccan_file *const *a, struct ccan_file *const *b,
176 return strcmp((*a)->name, (*b)->name);
179 static void sort_files(struct list_head *list)
181 struct ccan_file **files = NULL, *f;
185 while ((f = list_top(list, struct ccan_file, list)) != NULL) {
186 files = talloc_realloc(NULL, files, struct ccan_file *, num+1);
190 asort(files, num, cmp_names, NULL);
192 for (i = 0; i < num; i++)
193 list_add_tail(list, &files[i]->list);
197 struct manifest *get_manifest(const void *ctx, const char *dir)
200 char *olddir, *canon_dir;
202 struct list_head *list;
205 manifests = talloc(NULL, struct htable_manifest);
206 htable_manifest_init(manifests);
209 olddir = talloc_getcwd(NULL);
211 err(1, "Getting current directory");
214 err(1, "Failed to chdir to %s", dir);
216 canon_dir = talloc_getcwd(olddir);
218 err(1, "Getting current directory");
220 m = htable_manifest_get(manifests, canon_dir);
224 m = talloc_linked(ctx, talloc(NULL, struct manifest));
226 m->compiled[COMPILE_NORMAL] = m->compiled[COMPILE_NOFEAT] = NULL;
227 m->dir = talloc_steal(m, canon_dir);
228 list_head_init(&m->c_files);
229 list_head_init(&m->h_files);
230 list_head_init(&m->api_tests);
231 list_head_init(&m->run_tests);
232 list_head_init(&m->compile_ok_tests);
233 list_head_init(&m->compile_fail_tests);
234 list_head_init(&m->other_test_c_files);
235 list_head_init(&m->other_test_files);
236 list_head_init(&m->other_files);
237 list_head_init(&m->examples);
238 list_head_init(&m->mangled_examples);
239 list_head_init(&m->deps);
241 len = strlen(m->dir);
242 while (len && m->dir[len-1] == '/')
243 m->dir[--len] = '\0';
245 m->basename = strrchr(m->dir, '/');
247 errx(1, "I don't expect to be run from the root directory");
252 /* Nicer to run tests in a predictable order. */
253 foreach_ptr(list, &m->api_tests, &m->run_tests, &m->compile_ok_tests,
254 &m->compile_fail_tests)
257 htable_manifest_add(manifests, m);
260 if (chdir(olddir) != 0)
261 err(1, "Returning to original directory '%s'", olddir);