htable: clean up interface, document htable_type better.
[ccan] / tools / ccanlint / tests / reduce_features.c
1 #include <tools/ccanlint/ccanlint.h>
2 #include <tools/tools.h>
3 #include <ccan/htable/htable_type.h>
4 #include <ccan/foreach/foreach.h>
5 #include <ccan/talloc/talloc.h>
6 #include <ccan/grab_file/grab_file.h>
7 #include <ccan/str/str.h>
8 #include <ccan/str_talloc/str_talloc.h>
9 #include <ccan/hash/hash.h>
10 #include <ccan/read_write_all/read_write_all.h>
11 #include <errno.h>
12 #include <err.h>
13 #include <sys/types.h>
14 #include <sys/stat.h>
15 #include <fcntl.h>
16 #include <unistd.h>
17 #include "reduce_features.h"
18
19 bool features_were_reduced;
20
21 static const char *can_run(struct manifest *m)
22 {
23         if (!config_header)
24                 return talloc_strdup(m, "Could not read config.h");
25         return NULL;
26 }
27
28 static size_t option_hash(const char *name)
29 {
30         return hash(name, strlen(name), 0);
31 }
32
33 static const char *option_name(const char *name)
34 {
35         return name;
36 }
37
38 static bool option_cmp(const char *name1, const char *name2)
39 {
40         return streq(name1, name2);
41 }
42
43 HTABLE_DEFINE_TYPE(char, option_name, option_hash, option_cmp, htable_option);
44
45 static struct htable_option *htable_option_new(void)
46 {
47         struct htable_option *opts = malloc(sizeof(*opts));
48         htable_option_init(opts);
49         return opts;
50 }
51
52 static void htable_option_free(struct htable_option *opts)
53 {
54         htable_option_clear(opts);
55         free(opts);
56 }
57
58 static unsigned int add_options(struct htable_option *opts,
59                                 struct pp_conditions *cond)
60 {
61         unsigned int num = 0;
62         if (cond->parent)
63                 num += add_options(opts, cond->parent);
64         if (cond->type == PP_COND_IF || cond->type == PP_COND_IFDEF) {
65                 if (strstarts(cond->symbol, "HAVE_")) {
66                         if (!htable_option_get(opts, cond->symbol)) {
67                                 htable_option_add(opts, cond->symbol);
68                                 num++;
69                         }
70                 }
71         }
72         return num;
73 }
74
75 static struct htable_option *get_used_options(struct manifest *m)
76 {
77         struct list_head *list;
78         struct ccan_file *f;
79         unsigned int i, num;
80         struct htable_option *opts = htable_option_new();
81         struct line_info *info;
82
83         num = 0;
84         foreach_ptr(list, &m->c_files, &m->h_files) {
85                 list_for_each(list, f, list) {
86                         info = get_ccan_line_info(f);
87                         struct pp_conditions *prev = NULL;
88
89                         for (i = 0; i < f->num_lines; i++) {
90                                 if (info[i].cond && info[i].cond != prev) {
91                                         num += add_options(opts, info[i].cond);
92                                         prev = info[i].cond;
93                                 }
94                         }
95                 }
96         }
97
98         if (!num) {
99                 htable_option_free(opts);
100                 opts = NULL;
101         }
102         return opts;
103 }
104
105 static struct htable_option *get_config_options(struct manifest *m)
106 {
107         const char **lines = (const char **)strsplit(m, config_header, "\n");
108         unsigned int i;
109         struct htable_option *opts = htable_option_new();
110
111         for (i = 0; i < talloc_array_length(lines) - 1; i++) {
112                 char *sym;
113
114                 if (!get_token(&lines[i], "#"))
115                         continue;
116                 if (!get_token(&lines[i], "define"))
117                         continue;
118                 sym = get_symbol_token(lines, &lines[i]);
119                 if (!strstarts(sym, "HAVE_"))
120                         continue;
121                 /* Don't override endian... */
122                 if (strends(sym, "_ENDIAN"))
123                         continue;
124                 if (!get_token(&lines[i], "1"))
125                         continue;
126                 htable_option_add(opts, sym);
127         }
128         return opts;
129 }
130
131 static void do_reduce_features(struct manifest *m,
132                                bool keep,
133                                unsigned int *timeleft, struct score *score)
134 {
135         struct htable_option *options_used, *options_avail, *options;
136         struct htable_option_iter i;
137         int fd;
138         const char *sym;
139         char *hdr;
140
141         /* This isn't really a test, as such. */
142         score->total = 0;
143         score->pass = true;
144
145         options_used = get_used_options(m);
146         if (!options_used) {
147                 return;
148         }
149         options_avail = get_config_options(m);
150
151         options = NULL;
152         for (sym = htable_option_first(options_used, &i);
153              sym;
154              sym = htable_option_next(options_used, &i)) {
155                 if (htable_option_get(options_avail, sym)) {
156                         if (!options)
157                                 options = htable_option_new();
158                         htable_option_add(options, sym);
159                 }
160         }
161         htable_option_free(options_avail);
162         htable_option_free(options_used);
163
164         if (!options)
165                 return;
166
167         /* Now make our own config.h variant, with our own options. */
168         hdr = talloc_strdup(m, "/* Modified by reduce_features */\n");
169         hdr = talloc_append_string(hdr, config_header);
170         for (sym = htable_option_first(options, &i);
171              sym;
172              sym = htable_option_next(options, &i)) {
173                 hdr = talloc_asprintf_append
174                         (hdr, "#undef %s\n#define %s 0\n", sym, sym);
175         }
176         if (mkdir("reduced-features", 0700) != 0)
177                 err(1, "Creating reduced-features directory");
178
179         fd = open("reduced-features/config.h", O_EXCL|O_CREAT|O_RDWR, 0600);
180         if (fd < 0)
181                 err(1, "Creating reduced-features/config.h");
182         if (!write_all(fd, hdr, strlen(hdr)))
183                 err(1, "Writing reduced-features/config.h");
184         close(fd);
185         features_were_reduced = true;
186 }
187
188 struct ccanlint reduce_features = {
189         .key = "reduce_features",
190         .name = "Produce config.h with reduced features",
191         .can_run = can_run,
192         .check = do_reduce_features,
193         /* We only want to compile up versions with reduced featuress once
194          * objects for normal tests are built. */
195         .needs = "tests_compile"
196 };
197 REGISTER_TEST(reduce_features);