2 #include <ccan/talloc/talloc.h>
3 #include <ccan/str/str.h>
4 #include <ccan/str_talloc/str_talloc.h>
5 #include <ccan/grab_file/grab_file.h>
6 #include <ccan/noerr/noerr.h>
21 const char *get_ccan_file_contents(struct ccan_file *f)
24 f->contents = grab_file(f, f->fullname, &f->contents_size);
26 err(1, "Reading file %s", f->fullname);
31 char **get_ccan_file_lines(struct ccan_file *f)
34 f->lines = strsplit(f, get_ccan_file_contents(f),
40 struct list_head *get_ccan_file_docs(struct ccan_file *f)
42 if (!f->doc_sections) {
43 get_ccan_file_lines(f);
44 f->doc_sections = extract_doc_sections(f->lines, f->num_lines);
46 return f->doc_sections;
49 struct ccan_file *new_ccan_file(const void *ctx, const char *dir, char *name)
53 assert(dir[0] == '/');
55 f = talloc(ctx, struct ccan_file);
58 f->doc_sections = NULL;
60 f->name = talloc_steal(f, name);
61 f->fullname = talloc_asprintf(f, "%s/%s", dir, f->name);
63 f->cov_compiled = NULL;
67 static void add_files(struct manifest *m, const char *dir)
77 err(1, "Opening directory %s", dir[0] ? dir : ".");
79 while ((ent = readdir(d)) != NULL) {
82 struct list_head *dest;
85 if (ent->d_name[0] == '.')
88 f = new_ccan_file(m, m->dir,
89 talloc_asprintf(m, "%s%s",
91 if (lstat(f->name, &st) != 0)
92 err(1, "lstat %s", f->name);
94 if (S_ISDIR(st.st_mode)) {
95 f->name = talloc_append_string(f->name, "/");
96 add_files(m, f->name);
99 if (!S_ISREG(st.st_mode)) {
104 if (streq(f->name, "_info")) {
109 is_c_src = strends(f->name, ".c");
110 if (!is_c_src && !strends(f->name, ".h")) {
111 dest = &m->other_files;
115 if (!strchr(f->name, '/')) {
120 } else if (strstarts(f->name, "test/")) {
122 if (strstarts(f->name, "test/api"))
123 dest = &m->api_tests;
124 else if (strstarts(f->name, "test/run"))
125 dest = &m->run_tests;
126 else if (strstarts(f->name, "test/compile_ok"))
127 dest = &m->compile_ok_tests;
128 else if (strstarts(f->name, "test/compile_fail"))
129 dest = &m->compile_fail_tests;
131 dest = &m->other_test_c_files;
133 dest = &m->other_test_files;
135 dest = &m->other_files;
137 list_add(dest, &f->list);
142 char *report_on_lines(struct list_head *files,
143 char *(*report)(const char *),
148 list_for_each(files, f, list) {
150 char **lines = get_ccan_file_lines(f);
152 for (i = 0; i < f->num_lines; i++) {
153 char *r = report(lines[i]);
157 sofar = talloc_asprintf_append(sofar,
166 struct manifest *get_manifest(const void *ctx, const char *dir)
168 struct manifest *m = talloc(ctx, struct manifest);
173 list_head_init(&m->c_files);
174 list_head_init(&m->h_files);
175 list_head_init(&m->api_tests);
176 list_head_init(&m->run_tests);
177 list_head_init(&m->compile_ok_tests);
178 list_head_init(&m->compile_fail_tests);
179 list_head_init(&m->other_test_c_files);
180 list_head_init(&m->other_test_files);
181 list_head_init(&m->other_files);
182 list_head_init(&m->examples);
183 list_head_init(&m->mangled_examples);
184 list_head_init(&m->dep_dirs);
186 olddir = talloc_getcwd(NULL);
188 err(1, "Getting current directory");
191 err(1, "Failed to chdir to %s", dir);
193 m->dir = talloc_getcwd(m);
195 err(1, "Getting current directory");
197 len = strlen(m->dir);
198 while (len && m->dir[len-1] == '/')
199 m->dir[--len] = '\0';
201 m->basename = strrchr(m->dir, '/');
203 errx(1, "I don't expect to be run from the root directory");
206 /* We expect the ccan dir to be two levels above module dir. */
209 ccan_dir = talloc_strdup(NULL, m->dir);
210 p = strrchr(ccan_dir, '/');
212 p = strrchr(ccan_dir, '/');
218 if (chdir(olddir) != 0)
219 err(1, "Returning to original directory '%s'", olddir);
227 * remove_comments - strip comments from a line, return copy.
228 * @line: line to copy
229 * @in_comment: are we already within a comment (from prev line).
230 * @unterminated: are we still in a comment for next line.
232 static char *remove_comments(const char *line, bool in_comment,
235 char *p, *ret = talloc_array(line, char, strlen(line) + 1);
240 /* Find first comment. */
241 const char *old_comment = strstr(line, "/*");
242 const char *new_comment = strstr(line, "//");
245 if (new_comment && old_comment)
246 comment = new_comment < old_comment
247 ? new_comment : old_comment;
248 else if (old_comment)
249 comment = old_comment;
250 else if (new_comment)
251 comment = new_comment;
255 *unterminated = false;
259 /* Copy up to comment. */
260 memcpy(p, line, comment - line);
262 line += comment - line + 2;
264 if (comment == new_comment) {
265 /* We're done: goes to EOL. */
267 *unterminated = false;
274 const char *end = strstr(line, "*/");
276 *unterminated = true;
287 static bool is_empty(const char *line)
289 return strspn(line, " \t") == strlen(line);
292 static bool continues(const char *line)
294 /* Technically, any odd number of these. But who cares? */
295 return strends(line, "\\");
298 /* Get token if it's equal to token. */
299 bool get_token(const char **line, const char *token)
303 *line += strspn(*line, " \t");
304 if (isalnum(token[0]) || token[0] == '_')
305 toklen = strspn(*line, IDENT_CHARS);
307 /* FIXME: real tokenizer handles ++ and other multi-chars. */
308 toklen = strlen(token);
311 if (toklen == strlen(token) && !strncmp(*line, token, toklen)) {
318 char *get_symbol_token(void *ctx, const char **line)
323 *line += strspn(*line, " \t");
324 toklen = strspn(*line, IDENT_CHARS);
327 ret = talloc_strndup(ctx, *line, toklen);
332 static bool parse_hash_if(struct pp_conditions *cond, const char **line)
334 bool brackets, defined;
336 cond->inverse = get_token(line, "!");
337 defined = get_token(line, "defined");
338 brackets = get_token(line, "(");
339 cond->symbol = get_symbol_token(cond, line);
342 if (brackets && !get_token(line, ")"))
345 cond->type = PP_COND_IF;
349 /* FIXME: Get serious! */
350 static struct pp_conditions *analyze_directive(struct ccan_file *f,
352 struct pp_conditions *parent)
354 struct pp_conditions *cond = talloc(f, struct pp_conditions);
357 line = remove_comments(line, false, &unused);
359 cond->parent = parent;
360 cond->type = PP_COND_IFDEF;
362 if (!get_token(&line, "#"))
365 if (get_token(&line, "if")) {
366 if (!parse_hash_if(cond, &line))
368 } else if (get_token(&line, "elif")) {
372 cond->parent = parent->parent;
373 /* FIXME: Not quite true. This implies !parent, but we don't
374 * do multiple conditionals yet. */
375 if (!parse_hash_if(cond, &line))
377 } else if (get_token(&line, "ifdef")) {
379 cond->inverse = false;
380 brackets = get_token(&line, "(");
381 cond->symbol = get_symbol_token(cond, &line);
384 if (brackets && !get_token(&line, ")"))
386 } else if (get_token(&line, "ifndef")) {
388 cond->inverse = true;
389 brackets = get_token(&line, "(");
390 cond->symbol = get_symbol_token(cond, &line);
393 if (brackets && !get_token(&line, ")"))
395 } else if (get_token(&line, "else")) {
401 cond->inverse = !cond->inverse;
403 } else if (get_token(&line, "endif")) {
409 return parent->parent;
411 /* Not a conditional. */
421 cond->type = PP_COND_UNKNOWN;
425 /* This parser is rough, but OK if code is reasonably neat. */
426 struct line_info *get_ccan_line_info(struct ccan_file *f)
428 bool continued = false, in_comment = false;
429 struct pp_conditions *cond = NULL;
435 get_ccan_file_lines(f);
436 f->line_info = talloc_array(f->lines, struct line_info, f->num_lines);
438 for (i = 0; i < f->num_lines; continued = continues(f->lines[i++])) {
442 /* Current conditions apply to this line. */
443 f->line_info[i].cond = cond;
444 f->line_info[i].continued = continued;
447 /* Same as last line. */
448 f->line_info[i].type = f->line_info[i-1].type;
449 /* Update in_comment. */
450 remove_comments(f->lines[i], in_comment, &in_comment);
454 /* Preprocessor directive? */
456 && f->lines[i][strspn(f->lines[i], " \t")] == '#') {
457 f->line_info[i].type = PREPROC_LINE;
458 cond = analyze_directive(f, f->lines[i], cond);
462 still_doc_line = (in_comment
463 && f->line_info[i-1].type == DOC_LINE);
465 p = remove_comments(f->lines[i], in_comment, &in_comment);
467 if (strstarts(f->lines[i], "/**") || still_doc_line)
468 f->line_info[i].type = DOC_LINE;
470 f->line_info[i].type = COMMENT_LINE;
472 f->line_info[i].type = CODE_LINE;
479 struct list_node list;
481 const unsigned int *value;
484 static struct symbol *find_symbol(struct list_head *syms, const char *sym)
488 list_for_each(syms, i, list)
489 if (streq(sym, i->name))
494 static enum line_compiled get_pp(struct pp_conditions *cond,
495 struct list_head *syms)
499 enum line_compiled parent, ret;
501 /* No conditions? Easy. */
505 /* Check we get here at all. */
506 parent = get_pp(cond->parent, syms);
507 if (parent == NOT_COMPILED)
510 if (cond->type == PP_COND_UNKNOWN)
511 return MAYBE_COMPILED;
513 sym = find_symbol(syms, cond->symbol);
515 return MAYBE_COMPILED;
517 switch (cond->type) {
519 /* Undefined is 0. */
520 val = sym->value ? *sym->value : 0;
521 if (!val == cond->inverse)
528 if (cond->inverse == !sym->value)
538 /* If parent didn't know, NO == NO, but YES == MAYBE. */
539 if (parent == MAYBE_COMPILED && ret == COMPILED)
540 ret = MAYBE_COMPILED;
544 static void add_symbol(struct list_head *head,
545 const char *symbol, const unsigned int *value)
547 struct symbol *sym = talloc(head, struct symbol);
550 list_add(head, &sym->list);
553 enum line_compiled get_ccan_line_pp(struct pp_conditions *cond,
555 const unsigned int *value,
558 enum line_compiled ret;
559 struct list_head *head;
562 head = talloc(NULL, struct list_head);
563 list_head_init(head);
566 add_symbol(head, symbol, value);
568 while ((symbol = va_arg(ap, const char *)) != NULL) {
569 value = va_arg(ap, const unsigned int *);
570 add_symbol(head, symbol, value);
572 ret = get_pp(cond, head);