git.ozlabs.org Git - ccan/blob - tools/ccanlint/tests/idempotent.c

   1 #include <tools/ccanlint/ccanlint.h>
   2 #include <tools/tools.h>
   3 #include <ccan/talloc/talloc.h>
   4 #include <ccan/str/str.h>
   5 #include <sys/types.h>
   6 #include <sys/stat.h>
   7 #include <fcntl.h>
   8 #include <unistd.h>
   9 #include <limits.h>
  10 #include <errno.h>
  11 #include <stdlib.h>
  12 #include <stdio.h>
  13 #include <err.h>
  14 #include <string.h>
  15 #include <ctype.h>
  16
  17 static const char explain[]
  18 = "Headers usually start with the C preprocessor lines to prevent multiple\n"
  19   "inclusions.  These look like the following:\n"
  20   "#ifndef CCAN_<MODNAME>_H\n"
  21   "#define CCAN_<MODNAME>_H\n"
  22   "...\n"
  23   "#endif /* CCAN_<MODNAME>_H */\n";
  24
  25 static void fix_name(char *name)
  26 {
  27         unsigned int i, j;
  28
  29         for (i = j = 0; name[i]; i++) {
  30                 if (isalnum(name[i]) || name[i] == '_')
  31                         name[j++] = toupper(name[i]);
  32         }
  33         name[j] = '\0';
  34 }
  35
  36 static void handle_idem(struct manifest *m, struct score *score)
  37 {
  38         struct file_error *e;
  39
  40         list_for_each(&score->per_file_errors, e, list) {
  41                 char *name, *q, *tmpname;
  42                 FILE *out;
  43                 unsigned int i;
  44
  45                 /* Main header gets CCAN_FOO_H, others CCAN_FOO_XXX_H */
  46                 if (strstarts(e->file->name, m->basename)
  47                     || strlen(e->file->name) != strlen(m->basename) + 2)
  48                         name = talloc_asprintf(score, "CCAN_%s_H", m->basename);
  49                 else
  50                         name = talloc_asprintf(score, "CCAN_%s_%s_H",
  51                                                m->basename, e->file->name);
  52                 fix_name(name);
  53
  54                 q = talloc_asprintf(score,
  55                             "Should I wrap %s in #ifndef/#define %s for you?",
  56                             e->file->name, name);
  57                 if (!ask(q))
  58                         continue;
  59
  60                 tmpname = maybe_temp_file(score, ".h", false, e->file->name);
  61                 out = fopen(tmpname, "w");
  62                 if (!out)
  63                         err(1, "Opening %s", tmpname);
  64                 if (fprintf(out, "#ifndef %s\n#define %s\n", name, name) < 0)
  65                         err(1, "Writing %s", tmpname);
  66
  67                 for (i = 0; i < e->file->num_lines; i++)
  68                         if (fprintf(out, "%s\n", e->file->lines[i]) < 0)
  69                                 err(1, "Writing %s", tmpname);
  70
  71                 if (fprintf(out, "#endif /* %s */\n", name) < 0)
  72                         err(1, "Writing %s", tmpname);
  73
  74                 if (fclose(out) != 0)
  75                         err(1, "Closing %s", tmpname);
  76
  77                 if (!move_file(tmpname, e->file->fullname))
  78                         err(1, "Moving %s to %s", tmpname, e->file->fullname);
  79         }
  80 }
  81
  82 static bool check_idem(struct ccan_file *f, struct score *score)
  83 {
  84         struct line_info *line_info;
  85         unsigned int i, first_preproc_line;
  86         const char *line, *sym;
  87
  88         line_info = get_ccan_line_info(f);
  89         if (f->num_lines < 3)
  90                 /* FIXME: We assume small headers probably uninteresting. */
  91                 return true;
  92
  93         score->error = "Headers are not idempotent";
  94         for (i = 0; i < f->num_lines; i++) {
  95                 if (line_info[i].type == DOC_LINE
  96                     || line_info[i].type == COMMENT_LINE)
  97                         continue;
  98                 if (line_info[i].type == CODE_LINE) {
  99                         score_file_error(score, f, i+1,
 100                                          "Expect first non-comment line to be"
 101                                          " #ifndef.");
 102                         return false;
 103                 } else if (line_info[i].type == PREPROC_LINE)
 104                         break;
 105         }
 106
 107         /* No code at all?  Don't complain. */
 108         if (i == f->num_lines)
 109                 return true;
 110
 111         first_preproc_line = i;
 112         for (i = first_preproc_line+1; i < f->num_lines; i++) {
 113                 if (line_info[i].type == DOC_LINE
 114                     || line_info[i].type == COMMENT_LINE)
 115                         continue;
 116                 if (line_info[i].type == CODE_LINE) {
 117                         score_file_error(score, f, i+1,
 118                                          "Expect second non-comment line to be"
 119                                          " #define.");
 120                         return false;
 121                 } else if (line_info[i].type == PREPROC_LINE)
 122                         break;
 123         }
 124
 125         /* No code at all?  Weird. */
 126         if (i == f->num_lines)
 127                 return true;
 128
 129         /* We expect a condition on this line. */
 130         if (!line_info[i].cond) {
 131                 score_file_error(score, f, i+1, "Expected #ifndef");
 132                 return false;
 133         }
 134
 135         line = f->lines[i];
 136
 137         /* We expect the condition to be ! IFDEF <symbol>. */
 138         if (line_info[i].cond->type != PP_COND_IFDEF
 139             || !line_info[i].cond->inverse) {
 140                 score_file_error(score, f, i+1, "Expected #ifndef");
 141                 return false;
 142         }
 143
 144         /* And this to be #define <symbol> */
 145         if (!get_token(&line, "#"))
 146                 abort();
 147         if (!get_token(&line, "define")) {
 148                 char *str = talloc_asprintf(score,
 149                                             "expected '#define %s'",
 150                                             line_info[i].cond->symbol);
 151                 score_file_error(score, f, i+1, str);
 152                 return false;
 153         }
 154         sym = get_symbol_token(f, &line);
 155         if (!sym || !streq(sym, line_info[i].cond->symbol)) {
 156                 char *str = talloc_asprintf(score,
 157                                             "expected '#define %s'",
 158                                             line_info[i].cond->symbol);
 159                 score_file_error(score, f, i+1, str);
 160                 return false;
 161         }
 162
 163         /* Rest of code should all be covered by that conditional. */
 164         for (i++; i < f->num_lines; i++) {
 165                 unsigned int val = 0;
 166                 if (line_info[i].type == DOC_LINE
 167                     || line_info[i].type == COMMENT_LINE)
 168                         continue;
 169                 if (get_ccan_line_pp(line_info[i].cond, sym, &val, NULL)
 170                     != NOT_COMPILED) {
 171                         score_file_error(score, f, i+1, "code outside"
 172                                          " idempotent region");
 173                         return false;
 174                 }
 175         }
 176
 177         return true;
 178 }
 179
 180 static void check_idempotent(struct manifest *m,
 181                              bool keep,
 182                              unsigned int *timeleft, struct score *score)
 183 {
 184         struct ccan_file *f;
 185
 186         list_for_each(&m->h_files, f, list) {
 187                 if (!check_idem(f, score))
 188                         return;
 189         }
 190         score->pass = true;
 191         score->score = score->total;
 192 }
 193
 194 struct ccanlint idempotent = {
 195         .key = "idempotent",
 196         .name = "Module headers are #ifndef/#define wrapped",
 197         .check = check_idempotent,
 198         .handle = handle_idem,
 199 };
 200
 201 REGISTER_TEST(idempotent, NULL);