xref: /openbmc/u-boot/scripts/kconfig/expr.c (revision ddf56bc7) 
1 /*
2  * Copyright (C) 2002 Roman Zippel <zippel@linux-m68k.org>
3  * Released under the terms of the GNU GPL v2.0.
4  */
5 
6 #include <stdio.h>
7 #include <stdlib.h>
8 #include <string.h>
9 
10 #include "lkc.h"
11 
12 #define DEBUG_EXPR	0
13 
14 static int expr_eq(struct expr *e1, struct expr *e2);
15 static struct expr *expr_eliminate_yn(struct expr *e);
16 static struct expr *expr_extract_eq_and(struct expr **ep1, struct expr **ep2);
17 static struct expr *expr_extract_eq_or(struct expr **ep1, struct expr **ep2);
18 static void expr_extract_eq(enum expr_type type, struct expr **ep, struct expr **ep1, struct expr **ep2);
19 
20 struct expr *expr_alloc_symbol(struct symbol *sym)
21 {
22 	struct expr *e = xcalloc(1, sizeof(*e));
23 	e->type = E_SYMBOL;
24 	e->left.sym = sym;
25 	return e;
26 }
27 
28 struct expr *expr_alloc_one(enum expr_type type, struct expr *ce)
29 {
30 	struct expr *e = xcalloc(1, sizeof(*e));
31 	e->type = type;
32 	e->left.expr = ce;
33 	return e;
34 }
35 
36 struct expr *expr_alloc_two(enum expr_type type, struct expr *e1, struct expr *e2)
37 {
38 	struct expr *e = xcalloc(1, sizeof(*e));
39 	e->type = type;
40 	e->left.expr = e1;
41 	e->right.expr = e2;
42 	return e;
43 }
44 
45 struct expr *expr_alloc_comp(enum expr_type type, struct symbol *s1, struct symbol *s2)
46 {
47 	struct expr *e = xcalloc(1, sizeof(*e));
48 	e->type = type;
49 	e->left.sym = s1;
50 	e->right.sym = s2;
51 	return e;
52 }
53 
54 struct expr *expr_alloc_and(struct expr *e1, struct expr *e2)
55 {
56 	if (!e1)
57 		return e2;
58 	return e2 ? expr_alloc_two(E_AND, e1, e2) : e1;
59 }
60 
61 struct expr *expr_alloc_or(struct expr *e1, struct expr *e2)
62 {
63 	if (!e1)
64 		return e2;
65 	return e2 ? expr_alloc_two(E_OR, e1, e2) : e1;
66 }
67 
68 struct expr *expr_copy(const struct expr *org)
69 {
70 	struct expr *e;
71 
72 	if (!org)
73 		return NULL;
74 
75 	e = xmalloc(sizeof(*org));
76 	memcpy(e, org, sizeof(*org));
77 	switch (org->type) {
78 	case E_SYMBOL:
79 		e->left = org->left;
80 		break;
81 	case E_NOT:
82 		e->left.expr = expr_copy(org->left.expr);
83 		break;
84 	case E_EQUAL:
85 	case E_UNEQUAL:
86 		e->left.sym = org->left.sym;
87 		e->right.sym = org->right.sym;
88 		break;
89 	case E_AND:
90 	case E_OR:
91 	case E_LIST:
92 		e->left.expr = expr_copy(org->left.expr);
93 		e->right.expr = expr_copy(org->right.expr);
94 		break;
95 	default:
96 		printf("can't copy type %d\n", e->type);
97 		free(e);
98 		e = NULL;
99 		break;
100 	}
101 
102 	return e;
103 }
104 
105 void expr_free(struct expr *e)
106 {
107 	if (!e)
108 		return;
109 
110 	switch (e->type) {
111 	case E_SYMBOL:
112 		break;
113 	case E_NOT:
114 		expr_free(e->left.expr);
115 		return;
116 	case E_EQUAL:
117 	case E_UNEQUAL:
118 		break;
119 	case E_OR:
120 	case E_AND:
121 		expr_free(e->left.expr);
122 		expr_free(e->right.expr);
123 		break;
124 	default:
125 		printf("how to free type %d?\n", e->type);
126 		break;
127 	}
128 	free(e);
129 }
130 
131 static int trans_count;
132 
133 #define e1 (*ep1)
134 #define e2 (*ep2)
135 
136 static void __expr_eliminate_eq(enum expr_type type, struct expr **ep1, struct expr **ep2)
137 {
138 	if (e1->type == type) {
139 		__expr_eliminate_eq(type, &e1->left.expr, &e2);
140 		__expr_eliminate_eq(type, &e1->right.expr, &e2);
141 		return;
142 	}
143 	if (e2->type == type) {
144 		__expr_eliminate_eq(type, &e1, &e2->left.expr);
145 		__expr_eliminate_eq(type, &e1, &e2->right.expr);
146 		return;
147 	}
148 	if (e1->type == E_SYMBOL && e2->type == E_SYMBOL &&
149 	    e1->left.sym == e2->left.sym &&
150 	    (e1->left.sym == &symbol_yes || e1->left.sym == &symbol_no))
151 		return;
152 	if (!expr_eq(e1, e2))
153 		return;
154 	trans_count++;
155 	expr_free(e1); expr_free(e2);
156 	switch (type) {
157 	case E_OR:
158 		e1 = expr_alloc_symbol(&symbol_no);
159 		e2 = expr_alloc_symbol(&symbol_no);
160 		break;
161 	case E_AND:
162 		e1 = expr_alloc_symbol(&symbol_yes);
163 		e2 = expr_alloc_symbol(&symbol_yes);
164 		break;
165 	default:
166 		;
167 	}
168 }
169 
170 void expr_eliminate_eq(struct expr **ep1, struct expr **ep2)
171 {
172 	if (!e1 || !e2)
173 		return;
174 	switch (e1->type) {
175 	case E_OR:
176 	case E_AND:
177 		__expr_eliminate_eq(e1->type, ep1, ep2);
178 	default:
179 		;
180 	}
181 	if (e1->type != e2->type) switch (e2->type) {
182 	case E_OR:
183 	case E_AND:
184 		__expr_eliminate_eq(e2->type, ep1, ep2);
185 	default:
186 		;
187 	}
188 	e1 = expr_eliminate_yn(e1);
189 	e2 = expr_eliminate_yn(e2);
190 }
191 
192 #undef e1
193 #undef e2
194 
195 static int expr_eq(struct expr *e1, struct expr *e2)
196 {
197 	int res, old_count;
198 
199 	if (e1->type != e2->type)
200 		return 0;
201 	switch (e1->type) {
202 	case E_EQUAL:
203 	case E_UNEQUAL:
204 		return e1->left.sym == e2->left.sym && e1->right.sym == e2->right.sym;
205 	case E_SYMBOL:
206 		return e1->left.sym == e2->left.sym;
207 	case E_NOT:
208 		return expr_eq(e1->left.expr, e2->left.expr);
209 	case E_AND:
210 	case E_OR:
211 		e1 = expr_copy(e1);
212 		e2 = expr_copy(e2);
213 		old_count = trans_count;
214 		expr_eliminate_eq(&e1, &e2);
215 		res = (e1->type == E_SYMBOL && e2->type == E_SYMBOL &&
216 		       e1->left.sym == e2->left.sym);
217 		expr_free(e1);
218 		expr_free(e2);
219 		trans_count = old_count;
220 		return res;
221 	case E_LIST:
222 	case E_RANGE:
223 	case E_NONE:
224 		/* panic */;
225 	}
226 
227 	if (DEBUG_EXPR) {
228 		expr_fprint(e1, stdout);
229 		printf(" = ");
230 		expr_fprint(e2, stdout);
231 		printf(" ?\n");
232 	}
233 
234 	return 0;
235 }
236 
237 static struct expr *expr_eliminate_yn(struct expr *e)
238 {
239 	struct expr *tmp;
240 
241 	if (e) switch (e->type) {
242 	case E_AND:
243 		e->left.expr = expr_eliminate_yn(e->left.expr);
244 		e->right.expr = expr_eliminate_yn(e->right.expr);
245 		if (e->left.expr->type == E_SYMBOL) {
246 			if (e->left.expr->left.sym == &symbol_no) {
247 				expr_free(e->left.expr);
248 				expr_free(e->right.expr);
249 				e->type = E_SYMBOL;
250 				e->left.sym = &symbol_no;
251 				e->right.expr = NULL;
252 				return e;
253 			} else if (e->left.expr->left.sym == &symbol_yes) {
254 				free(e->left.expr);
255 				tmp = e->right.expr;
256 				*e = *(e->right.expr);
257 				free(tmp);
258 				return e;
259 			}
260 		}
261 		if (e->right.expr->type == E_SYMBOL) {
262 			if (e->right.expr->left.sym == &symbol_no) {
263 				expr_free(e->left.expr);
264 				expr_free(e->right.expr);
265 				e->type = E_SYMBOL;
266 				e->left.sym = &symbol_no;
267 				e->right.expr = NULL;
268 				return e;
269 			} else if (e->right.expr->left.sym == &symbol_yes) {
270 				free(e->right.expr);
271 				tmp = e->left.expr;
272 				*e = *(e->left.expr);
273 				free(tmp);
274 				return e;
275 			}
276 		}
277 		break;
278 	case E_OR:
279 		e->left.expr = expr_eliminate_yn(e->left.expr);
280 		e->right.expr = expr_eliminate_yn(e->right.expr);
281 		if (e->left.expr->type == E_SYMBOL) {
282 			if (e->left.expr->left.sym == &symbol_no) {
283 				free(e->left.expr);
284 				tmp = e->right.expr;
285 				*e = *(e->right.expr);
286 				free(tmp);
287 				return e;
288 			} else if (e->left.expr->left.sym == &symbol_yes) {
289 				expr_free(e->left.expr);
290 				expr_free(e->right.expr);
291 				e->type = E_SYMBOL;
292 				e->left.sym = &symbol_yes;
293 				e->right.expr = NULL;
294 				return e;
295 			}
296 		}
297 		if (e->right.expr->type == E_SYMBOL) {
298 			if (e->right.expr->left.sym == &symbol_no) {
299 				free(e->right.expr);
300 				tmp = e->left.expr;
301 				*e = *(e->left.expr);
302 				free(tmp);
303 				return e;
304 			} else if (e->right.expr->left.sym == &symbol_yes) {
305 				expr_free(e->left.expr);
306 				expr_free(e->right.expr);
307 				e->type = E_SYMBOL;
308 				e->left.sym = &symbol_yes;
309 				e->right.expr = NULL;
310 				return e;
311 			}
312 		}
313 		break;
314 	default:
315 		;
316 	}
317 	return e;
318 }
319 
320 /*
321  * bool FOO!=n => FOO
322  */
323 struct expr *expr_trans_bool(struct expr *e)
324 {
325 	if (!e)
326 		return NULL;
327 	switch (e->type) {
328 	case E_AND:
329 	case E_OR:
330 	case E_NOT:
331 		e->left.expr = expr_trans_bool(e->left.expr);
332 		e->right.expr = expr_trans_bool(e->right.expr);
333 		break;
334 	case E_UNEQUAL:
335 		// FOO!=n -> FOO
336 		if (e->left.sym->type == S_TRISTATE) {
337 			if (e->right.sym == &symbol_no) {
338 				e->type = E_SYMBOL;
339 				e->right.sym = NULL;
340 			}
341 		}
342 		break;
343 	default:
344 		;
345 	}
346 	return e;
347 }
348 
349 /*
350  * e1 || e2 -> ?
351  */
352 static struct expr *expr_join_or(struct expr *e1, struct expr *e2)
353 {
354 	struct expr *tmp;
355 	struct symbol *sym1, *sym2;
356 
357 	if (expr_eq(e1, e2))
358 		return expr_copy(e1);
359 	if (e1->type != E_EQUAL && e1->type != E_UNEQUAL && e1->type != E_SYMBOL && e1->type != E_NOT)
360 		return NULL;
361 	if (e2->type != E_EQUAL && e2->type != E_UNEQUAL && e2->type != E_SYMBOL && e2->type != E_NOT)
362 		return NULL;
363 	if (e1->type == E_NOT) {
364 		tmp = e1->left.expr;
365 		if (tmp->type != E_EQUAL && tmp->type != E_UNEQUAL && tmp->type != E_SYMBOL)
366 			return NULL;
367 		sym1 = tmp->left.sym;
368 	} else
369 		sym1 = e1->left.sym;
370 	if (e2->type == E_NOT) {
371 		if (e2->left.expr->type != E_SYMBOL)
372 			return NULL;
373 		sym2 = e2->left.expr->left.sym;
374 	} else
375 		sym2 = e2->left.sym;
376 	if (sym1 != sym2)
377 		return NULL;
378 	if (sym1->type != S_BOOLEAN && sym1->type != S_TRISTATE)
379 		return NULL;
380 	if (sym1->type == S_TRISTATE) {
381 		if (e1->type == E_EQUAL && e2->type == E_EQUAL &&
382 		    ((e1->right.sym == &symbol_yes && e2->right.sym == &symbol_mod) ||
383 		     (e1->right.sym == &symbol_mod && e2->right.sym == &symbol_yes))) {
384 			// (a='y') || (a='m') -> (a!='n')
385 			return expr_alloc_comp(E_UNEQUAL, sym1, &symbol_no);
386 		}
387 		if (e1->type == E_EQUAL && e2->type == E_EQUAL &&
388 		    ((e1->right.sym == &symbol_yes && e2->right.sym == &symbol_no) ||
389 		     (e1->right.sym == &symbol_no && e2->right.sym == &symbol_yes))) {
390 			// (a='y') || (a='n') -> (a!='m')
391 			return expr_alloc_comp(E_UNEQUAL, sym1, &symbol_mod);
392 		}
393 		if (e1->type == E_EQUAL && e2->type == E_EQUAL &&
394 		    ((e1->right.sym == &symbol_mod && e2->right.sym == &symbol_no) ||
395 		     (e1->right.sym == &symbol_no && e2->right.sym == &symbol_mod))) {
396 			// (a='m') || (a='n') -> (a!='y')
397 			return expr_alloc_comp(E_UNEQUAL, sym1, &symbol_yes);
398 		}
399 	}
400 	if (sym1->type == S_BOOLEAN && sym1 == sym2) {
401 		if ((e1->type == E_NOT && e1->left.expr->type == E_SYMBOL && e2->type == E_SYMBOL) ||
402 		    (e2->type == E_NOT && e2->left.expr->type == E_SYMBOL && e1->type == E_SYMBOL))
403 			return expr_alloc_symbol(&symbol_yes);
404 	}
405 
406 	if (DEBUG_EXPR) {
407 		printf("optimize (");
408 		expr_fprint(e1, stdout);
409 		printf(") || (");
410 		expr_fprint(e2, stdout);
411 		printf(")?\n");
412 	}
413 	return NULL;
414 }
415 
416 static struct expr *expr_join_and(struct expr *e1, struct expr *e2)
417 {
418 	struct expr *tmp;
419 	struct symbol *sym1, *sym2;
420 
421 	if (expr_eq(e1, e2))
422 		return expr_copy(e1);
423 	if (e1->type != E_EQUAL && e1->type != E_UNEQUAL && e1->type != E_SYMBOL && e1->type != E_NOT)
424 		return NULL;
425 	if (e2->type != E_EQUAL && e2->type != E_UNEQUAL && e2->type != E_SYMBOL && e2->type != E_NOT)
426 		return NULL;
427 	if (e1->type == E_NOT) {
428 		tmp = e1->left.expr;
429 		if (tmp->type != E_EQUAL && tmp->type != E_UNEQUAL && tmp->type != E_SYMBOL)
430 			return NULL;
431 		sym1 = tmp->left.sym;
432 	} else
433 		sym1 = e1->left.sym;
434 	if (e2->type == E_NOT) {
435 		if (e2->left.expr->type != E_SYMBOL)
436 			return NULL;
437 		sym2 = e2->left.expr->left.sym;
438 	} else
439 		sym2 = e2->left.sym;
440 	if (sym1 != sym2)
441 		return NULL;
442 	if (sym1->type != S_BOOLEAN && sym1->type != S_TRISTATE)
443 		return NULL;
444 
445 	if ((e1->type == E_SYMBOL && e2->type == E_EQUAL && e2->right.sym == &symbol_yes) ||
446 	    (e2->type == E_SYMBOL && e1->type == E_EQUAL && e1->right.sym == &symbol_yes))
447 		// (a) && (a='y') -> (a='y')
448 		return expr_alloc_comp(E_EQUAL, sym1, &symbol_yes);
449 
450 	if ((e1->type == E_SYMBOL && e2->type == E_UNEQUAL && e2->right.sym == &symbol_no) ||
451 	    (e2->type == E_SYMBOL && e1->type == E_UNEQUAL && e1->right.sym == &symbol_no))
452 		// (a) && (a!='n') -> (a)
453 		return expr_alloc_symbol(sym1);
454 
455 	if ((e1->type == E_SYMBOL && e2->type == E_UNEQUAL && e2->right.sym == &symbol_mod) ||
456 	    (e2->type == E_SYMBOL && e1->type == E_UNEQUAL && e1->right.sym == &symbol_mod))
457 		// (a) && (a!='m') -> (a='y')
458 		return expr_alloc_comp(E_EQUAL, sym1, &symbol_yes);
459 
460 	if (sym1->type == S_TRISTATE) {
461 		if (e1->type == E_EQUAL && e2->type == E_UNEQUAL) {
462 			// (a='b') && (a!='c') -> 'b'='c' ? 'n' : a='b'
463 			sym2 = e1->right.sym;
464 			if ((e2->right.sym->flags & SYMBOL_CONST) && (sym2->flags & SYMBOL_CONST))
465 				return sym2 != e2->right.sym ? expr_alloc_comp(E_EQUAL, sym1, sym2)
466 							     : expr_alloc_symbol(&symbol_no);
467 		}
468 		if (e1->type == E_UNEQUAL && e2->type == E_EQUAL) {
469 			// (a='b') && (a!='c') -> 'b'='c' ? 'n' : a='b'
470 			sym2 = e2->right.sym;
471 			if ((e1->right.sym->flags & SYMBOL_CONST) && (sym2->flags & SYMBOL_CONST))
472 				return sym2 != e1->right.sym ? expr_alloc_comp(E_EQUAL, sym1, sym2)
473 							     : expr_alloc_symbol(&symbol_no);
474 		}
475 		if (e1->type == E_UNEQUAL && e2->type == E_UNEQUAL &&
476 			   ((e1->right.sym == &symbol_yes && e2->right.sym == &symbol_no) ||
477 			    (e1->right.sym == &symbol_no && e2->right.sym == &symbol_yes)))
478 			// (a!='y') && (a!='n') -> (a='m')
479 			return expr_alloc_comp(E_EQUAL, sym1, &symbol_mod);
480 
481 		if (e1->type == E_UNEQUAL && e2->type == E_UNEQUAL &&
482 			   ((e1->right.sym == &symbol_yes && e2->right.sym == &symbol_mod) ||
483 			    (e1->right.sym == &symbol_mod && e2->right.sym == &symbol_yes)))
484 			// (a!='y') && (a!='m') -> (a='n')
485 			return expr_alloc_comp(E_EQUAL, sym1, &symbol_no);
486 
487 		if (e1->type == E_UNEQUAL && e2->type == E_UNEQUAL &&
488 			   ((e1->right.sym == &symbol_mod && e2->right.sym == &symbol_no) ||
489 			    (e1->right.sym == &symbol_no && e2->right.sym == &symbol_mod)))
490 			// (a!='m') && (a!='n') -> (a='m')
491 			return expr_alloc_comp(E_EQUAL, sym1, &symbol_yes);
492 
493 		if ((e1->type == E_SYMBOL && e2->type == E_EQUAL && e2->right.sym == &symbol_mod) ||
494 		    (e2->type == E_SYMBOL && e1->type == E_EQUAL && e1->right.sym == &symbol_mod) ||
495 		    (e1->type == E_SYMBOL && e2->type == E_UNEQUAL && e2->right.sym == &symbol_yes) ||
496 		    (e2->type == E_SYMBOL && e1->type == E_UNEQUAL && e1->right.sym == &symbol_yes))
497 			return NULL;
498 	}
499 
500 	if (DEBUG_EXPR) {
501 		printf("optimize (");
502 		expr_fprint(e1, stdout);
503 		printf(") && (");
504 		expr_fprint(e2, stdout);
505 		printf(")?\n");
506 	}
507 	return NULL;
508 }
509 
510 static void expr_eliminate_dups1(enum expr_type type, struct expr **ep1, struct expr **ep2)
511 {
512 #define e1 (*ep1)
513 #define e2 (*ep2)
514 	struct expr *tmp;
515 
516 	if (e1->type == type) {
517 		expr_eliminate_dups1(type, &e1->left.expr, &e2);
518 		expr_eliminate_dups1(type, &e1->right.expr, &e2);
519 		return;
520 	}
521 	if (e2->type == type) {
522 		expr_eliminate_dups1(type, &e1, &e2->left.expr);
523 		expr_eliminate_dups1(type, &e1, &e2->right.expr);
524 		return;
525 	}
526 	if (e1 == e2)
527 		return;
528 
529 	switch (e1->type) {
530 	case E_OR: case E_AND:
531 		expr_eliminate_dups1(e1->type, &e1, &e1);
532 	default:
533 		;
534 	}
535 
536 	switch (type) {
537 	case E_OR:
538 		tmp = expr_join_or(e1, e2);
539 		if (tmp) {
540 			expr_free(e1); expr_free(e2);
541 			e1 = expr_alloc_symbol(&symbol_no);
542 			e2 = tmp;
543 			trans_count++;
544 		}
545 		break;
546 	case E_AND:
547 		tmp = expr_join_and(e1, e2);
548 		if (tmp) {
549 			expr_free(e1); expr_free(e2);
550 			e1 = expr_alloc_symbol(&symbol_yes);
551 			e2 = tmp;
552 			trans_count++;
553 		}
554 		break;
555 	default:
556 		;
557 	}
558 #undef e1
559 #undef e2
560 }
561 
562 static void expr_eliminate_dups2(enum expr_type type, struct expr **ep1, struct expr **ep2)
563 {
564 #define e1 (*ep1)
565 #define e2 (*ep2)
566 	struct expr *tmp, *tmp1, *tmp2;
567 
568 	if (e1->type == type) {
569 		expr_eliminate_dups2(type, &e1->left.expr, &e2);
570 		expr_eliminate_dups2(type, &e1->right.expr, &e2);
571 		return;
572 	}
573 	if (e2->type == type) {
574 		expr_eliminate_dups2(type, &e1, &e2->left.expr);
575 		expr_eliminate_dups2(type, &e1, &e2->right.expr);
576 	}
577 	if (e1 == e2)
578 		return;
579 
580 	switch (e1->type) {
581 	case E_OR:
582 		expr_eliminate_dups2(e1->type, &e1, &e1);
583 		// (FOO || BAR) && (!FOO && !BAR) -> n
584 		tmp1 = expr_transform(expr_alloc_one(E_NOT, expr_copy(e1)));
585 		tmp2 = expr_copy(e2);
586 		tmp = expr_extract_eq_and(&tmp1, &tmp2);
587 		if (expr_is_yes(tmp1)) {
588 			expr_free(e1);
589 			e1 = expr_alloc_symbol(&symbol_no);
590 			trans_count++;
591 		}
592 		expr_free(tmp2);
593 		expr_free(tmp1);
594 		expr_free(tmp);
595 		break;
596 	case E_AND:
597 		expr_eliminate_dups2(e1->type, &e1, &e1);
598 		// (FOO && BAR) || (!FOO || !BAR) -> y
599 		tmp1 = expr_transform(expr_alloc_one(E_NOT, expr_copy(e1)));
600 		tmp2 = expr_copy(e2);
601 		tmp = expr_extract_eq_or(&tmp1, &tmp2);
602 		if (expr_is_no(tmp1)) {
603 			expr_free(e1);
604 			e1 = expr_alloc_symbol(&symbol_yes);
605 			trans_count++;
606 		}
607 		expr_free(tmp2);
608 		expr_free(tmp1);
609 		expr_free(tmp);
610 		break;
611 	default:
612 		;
613 	}
614 #undef e1
615 #undef e2
616 }
617 
618 struct expr *expr_eliminate_dups(struct expr *e)
619 {
620 	int oldcount;
621 	if (!e)
622 		return e;
623 
624 	oldcount = trans_count;
625 	while (1) {
626 		trans_count = 0;
627 		switch (e->type) {
628 		case E_OR: case E_AND:
629 			expr_eliminate_dups1(e->type, &e, &e);
630 			expr_eliminate_dups2(e->type, &e, &e);
631 		default:
632 			;
633 		}
634 		if (!trans_count)
635 			break;
636 		e = expr_eliminate_yn(e);
637 	}
638 	trans_count = oldcount;
639 	return e;
640 }
641 
642 struct expr *expr_transform(struct expr *e)
643 {
644 	struct expr *tmp;
645 
646 	if (!e)
647 		return NULL;
648 	switch (e->type) {
649 	case E_EQUAL:
650 	case E_UNEQUAL:
651 	case E_SYMBOL:
652 	case E_LIST:
653 		break;
654 	default:
655 		e->left.expr = expr_transform(e->left.expr);
656 		e->right.expr = expr_transform(e->right.expr);
657 	}
658 
659 	switch (e->type) {
660 	case E_EQUAL:
661 		if (e->left.sym->type != S_BOOLEAN)
662 			break;
663 		if (e->right.sym == &symbol_no) {
664 			e->type = E_NOT;
665 			e->left.expr = expr_alloc_symbol(e->left.sym);
666 			e->right.sym = NULL;
667 			break;
668 		}
669 		if (e->right.sym == &symbol_mod) {
670 			printf("boolean symbol %s tested for 'm'? test forced to 'n'\n", e->left.sym->name);
671 			e->type = E_SYMBOL;
672 			e->left.sym = &symbol_no;
673 			e->right.sym = NULL;
674 			break;
675 		}
676 		if (e->right.sym == &symbol_yes) {
677 			e->type = E_SYMBOL;
678 			e->right.sym = NULL;
679 			break;
680 		}
681 		break;
682 	case E_UNEQUAL:
683 		if (e->left.sym->type != S_BOOLEAN)
684 			break;
685 		if (e->right.sym == &symbol_no) {
686 			e->type = E_SYMBOL;
687 			e->right.sym = NULL;
688 			break;
689 		}
690 		if (e->right.sym == &symbol_mod) {
691 			printf("boolean symbol %s tested for 'm'? test forced to 'y'\n", e->left.sym->name);
692 			e->type = E_SYMBOL;
693 			e->left.sym = &symbol_yes;
694 			e->right.sym = NULL;
695 			break;
696 		}
697 		if (e->right.sym == &symbol_yes) {
698 			e->type = E_NOT;
699 			e->left.expr = expr_alloc_symbol(e->left.sym);
700 			e->right.sym = NULL;
701 			break;
702 		}
703 		break;
704 	case E_NOT:
705 		switch (e->left.expr->type) {
706 		case E_NOT:
707 			// !!a -> a
708 			tmp = e->left.expr->left.expr;
709 			free(e->left.expr);
710 			free(e);
711 			e = tmp;
712 			e = expr_transform(e);
713 			break;
714 		case E_EQUAL:
715 		case E_UNEQUAL:
716 			// !a='x' -> a!='x'
717 			tmp = e->left.expr;
718 			free(e);
719 			e = tmp;
720 			e->type = e->type == E_EQUAL ? E_UNEQUAL : E_EQUAL;
721 			break;
722 		case E_OR:
723 			// !(a || b) -> !a && !b
724 			tmp = e->left.expr;
725 			e->type = E_AND;
726 			e->right.expr = expr_alloc_one(E_NOT, tmp->right.expr);
727 			tmp->type = E_NOT;
728 			tmp->right.expr = NULL;
729 			e = expr_transform(e);
730 			break;
731 		case E_AND:
732 			// !(a && b) -> !a || !b
733 			tmp = e->left.expr;
734 			e->type = E_OR;
735 			e->right.expr = expr_alloc_one(E_NOT, tmp->right.expr);
736 			tmp->type = E_NOT;
737 			tmp->right.expr = NULL;
738 			e = expr_transform(e);
739 			break;
740 		case E_SYMBOL:
741 			if (e->left.expr->left.sym == &symbol_yes) {
742 				// !'y' -> 'n'
743 				tmp = e->left.expr;
744 				free(e);
745 				e = tmp;
746 				e->type = E_SYMBOL;
747 				e->left.sym = &symbol_no;
748 				break;
749 			}
750 			if (e->left.expr->left.sym == &symbol_mod) {
751 				// !'m' -> 'm'
752 				tmp = e->left.expr;
753 				free(e);
754 				e = tmp;
755 				e->type = E_SYMBOL;
756 				e->left.sym = &symbol_mod;
757 				break;
758 			}
759 			if (e->left.expr->left.sym == &symbol_no) {
760 				// !'n' -> 'y'
761 				tmp = e->left.expr;
762 				free(e);
763 				e = tmp;
764 				e->type = E_SYMBOL;
765 				e->left.sym = &symbol_yes;
766 				break;
767 			}
768 			break;
769 		default:
770 			;
771 		}
772 		break;
773 	default:
774 		;
775 	}
776 	return e;
777 }
778 
779 int expr_contains_symbol(struct expr *dep, struct symbol *sym)
780 {
781 	if (!dep)
782 		return 0;
783 
784 	switch (dep->type) {
785 	case E_AND:
786 	case E_OR:
787 		return expr_contains_symbol(dep->left.expr, sym) ||
788 		       expr_contains_symbol(dep->right.expr, sym);
789 	case E_SYMBOL:
790 		return dep->left.sym == sym;
791 	case E_EQUAL:
792 	case E_UNEQUAL:
793 		return dep->left.sym == sym ||
794 		       dep->right.sym == sym;
795 	case E_NOT:
796 		return expr_contains_symbol(dep->left.expr, sym);
797 	default:
798 		;
799 	}
800 	return 0;
801 }
802 
803 bool expr_depends_symbol(struct expr *dep, struct symbol *sym)
804 {
805 	if (!dep)
806 		return false;
807 
808 	switch (dep->type) {
809 	case E_AND:
810 		return expr_depends_symbol(dep->left.expr, sym) ||
811 		       expr_depends_symbol(dep->right.expr, sym);
812 	case E_SYMBOL:
813 		return dep->left.sym == sym;
814 	case E_EQUAL:
815 		if (dep->left.sym == sym) {
816 			if (dep->right.sym == &symbol_yes || dep->right.sym == &symbol_mod)
817 				return true;
818 		}
819 		break;
820 	case E_UNEQUAL:
821 		if (dep->left.sym == sym) {
822 			if (dep->right.sym == &symbol_no)
823 				return true;
824 		}
825 		break;
826 	default:
827 		;
828 	}
829  	return false;
830 }
831 
832 static struct expr *expr_extract_eq_and(struct expr **ep1, struct expr **ep2)
833 {
834 	struct expr *tmp = NULL;
835 	expr_extract_eq(E_AND, &tmp, ep1, ep2);
836 	if (tmp) {
837 		*ep1 = expr_eliminate_yn(*ep1);
838 		*ep2 = expr_eliminate_yn(*ep2);
839 	}
840 	return tmp;
841 }
842 
843 static struct expr *expr_extract_eq_or(struct expr **ep1, struct expr **ep2)
844 {
845 	struct expr *tmp = NULL;
846 	expr_extract_eq(E_OR, &tmp, ep1, ep2);
847 	if (tmp) {
848 		*ep1 = expr_eliminate_yn(*ep1);
849 		*ep2 = expr_eliminate_yn(*ep2);
850 	}
851 	return tmp;
852 }
853 
854 static void expr_extract_eq(enum expr_type type, struct expr **ep, struct expr **ep1, struct expr **ep2)
855 {
856 #define e1 (*ep1)
857 #define e2 (*ep2)
858 	if (e1->type == type) {
859 		expr_extract_eq(type, ep, &e1->left.expr, &e2);
860 		expr_extract_eq(type, ep, &e1->right.expr, &e2);
861 		return;
862 	}
863 	if (e2->type == type) {
864 		expr_extract_eq(type, ep, ep1, &e2->left.expr);
865 		expr_extract_eq(type, ep, ep1, &e2->right.expr);
866 		return;
867 	}
868 	if (expr_eq(e1, e2)) {
869 		*ep = *ep ? expr_alloc_two(type, *ep, e1) : e1;
870 		expr_free(e2);
871 		if (type == E_AND) {
872 			e1 = expr_alloc_symbol(&symbol_yes);
873 			e2 = expr_alloc_symbol(&symbol_yes);
874 		} else if (type == E_OR) {
875 			e1 = expr_alloc_symbol(&symbol_no);
876 			e2 = expr_alloc_symbol(&symbol_no);
877 		}
878 	}
879 #undef e1
880 #undef e2
881 }
882 
883 struct expr *expr_trans_compare(struct expr *e, enum expr_type type, struct symbol *sym)
884 {
885 	struct expr *e1, *e2;
886 
887 	if (!e) {
888 		e = expr_alloc_symbol(sym);
889 		if (type == E_UNEQUAL)
890 			e = expr_alloc_one(E_NOT, e);
891 		return e;
892 	}
893 	switch (e->type) {
894 	case E_AND:
895 		e1 = expr_trans_compare(e->left.expr, E_EQUAL, sym);
896 		e2 = expr_trans_compare(e->right.expr, E_EQUAL, sym);
897 		if (sym == &symbol_yes)
898 			e = expr_alloc_two(E_AND, e1, e2);
899 		if (sym == &symbol_no)
900 			e = expr_alloc_two(E_OR, e1, e2);
901 		if (type == E_UNEQUAL)
902 			e = expr_alloc_one(E_NOT, e);
903 		return e;
904 	case E_OR:
905 		e1 = expr_trans_compare(e->left.expr, E_EQUAL, sym);
906 		e2 = expr_trans_compare(e->right.expr, E_EQUAL, sym);
907 		if (sym == &symbol_yes)
908 			e = expr_alloc_two(E_OR, e1, e2);
909 		if (sym == &symbol_no)
910 			e = expr_alloc_two(E_AND, e1, e2);
911 		if (type == E_UNEQUAL)
912 			e = expr_alloc_one(E_NOT, e);
913 		return e;
914 	case E_NOT:
915 		return expr_trans_compare(e->left.expr, type == E_EQUAL ? E_UNEQUAL : E_EQUAL, sym);
916 	case E_UNEQUAL:
917 	case E_EQUAL:
918 		if (type == E_EQUAL) {
919 			if (sym == &symbol_yes)
920 				return expr_copy(e);
921 			if (sym == &symbol_mod)
922 				return expr_alloc_symbol(&symbol_no);
923 			if (sym == &symbol_no)
924 				return expr_alloc_one(E_NOT, expr_copy(e));
925 		} else {
926 			if (sym == &symbol_yes)
927 				return expr_alloc_one(E_NOT, expr_copy(e));
928 			if (sym == &symbol_mod)
929 				return expr_alloc_symbol(&symbol_yes);
930 			if (sym == &symbol_no)
931 				return expr_copy(e);
932 		}
933 		break;
934 	case E_SYMBOL:
935 		return expr_alloc_comp(type, e->left.sym, sym);
936 	case E_LIST:
937 	case E_RANGE:
938 	case E_NONE:
939 		/* panic */;
940 	}
941 	return NULL;
942 }
943 
944 tristate expr_calc_value(struct expr *e)
945 {
946 	tristate val1, val2;
947 	const char *str1, *str2;
948 
949 	if (!e)
950 		return yes;
951 
952 	switch (e->type) {
953 	case E_SYMBOL:
954 		sym_calc_value(e->left.sym);
955 		return e->left.sym->curr.tri;
956 	case E_AND:
957 		val1 = expr_calc_value(e->left.expr);
958 		val2 = expr_calc_value(e->right.expr);
959 		return EXPR_AND(val1, val2);
960 	case E_OR:
961 		val1 = expr_calc_value(e->left.expr);
962 		val2 = expr_calc_value(e->right.expr);
963 		return EXPR_OR(val1, val2);
964 	case E_NOT:
965 		val1 = expr_calc_value(e->left.expr);
966 		return EXPR_NOT(val1);
967 	case E_EQUAL:
968 		sym_calc_value(e->left.sym);
969 		sym_calc_value(e->right.sym);
970 		str1 = sym_get_string_value(e->left.sym);
971 		str2 = sym_get_string_value(e->right.sym);
972 		return !strcmp(str1, str2) ? yes : no;
973 	case E_UNEQUAL:
974 		sym_calc_value(e->left.sym);
975 		sym_calc_value(e->right.sym);
976 		str1 = sym_get_string_value(e->left.sym);
977 		str2 = sym_get_string_value(e->right.sym);
978 		return !strcmp(str1, str2) ? no : yes;
979 	default:
980 		printf("expr_calc_value: %d?\n", e->type);
981 		return no;
982 	}
983 }
984 
985 static int expr_compare_type(enum expr_type t1, enum expr_type t2)
986 {
987 	if (t1 == t2)
988 		return 0;
989 	switch (t1) {
990 	case E_EQUAL:
991 	case E_UNEQUAL:
992 		if (t2 == E_NOT)
993 			return 1;
994 	case E_NOT:
995 		if (t2 == E_AND)
996 			return 1;
997 	case E_AND:
998 		if (t2 == E_OR)
999 			return 1;
1000 	case E_OR:
1001 		if (t2 == E_LIST)
1002 			return 1;
1003 	case E_LIST:
1004 		if (t2 == 0)
1005 			return 1;
1006 	default:
1007 		return -1;
1008 	}
1009 	printf("[%dgt%d?]", t1, t2);
1010 	return 0;
1011 }
1012 
1013 static inline struct expr *
1014 expr_get_leftmost_symbol(const struct expr *e)
1015 {
1016 
1017 	if (e == NULL)
1018 		return NULL;
1019 
1020 	while (e->type != E_SYMBOL)
1021 		e = e->left.expr;
1022 
1023 	return expr_copy(e);
1024 }
1025 
1026 /*
1027  * Given expression `e1' and `e2', returns the leaf of the longest
1028  * sub-expression of `e1' not containing 'e2.
1029  */
1030 struct expr *expr_simplify_unmet_dep(struct expr *e1, struct expr *e2)
1031 {
1032 	struct expr *ret;
1033 
1034 	switch (e1->type) {
1035 	case E_OR:
1036 		return expr_alloc_and(
1037 		    expr_simplify_unmet_dep(e1->left.expr, e2),
1038 		    expr_simplify_unmet_dep(e1->right.expr, e2));
1039 	case E_AND: {
1040 		struct expr *e;
1041 		e = expr_alloc_and(expr_copy(e1), expr_copy(e2));
1042 		e = expr_eliminate_dups(e);
1043 		ret = (!expr_eq(e, e1)) ? e1 : NULL;
1044 		expr_free(e);
1045 		break;
1046 		}
1047 	default:
1048 		ret = e1;
1049 		break;
1050 	}
1051 
1052 	return expr_get_leftmost_symbol(ret);
1053 }
1054 
1055 void expr_print(struct expr *e, void (*fn)(void *, struct symbol *, const char *), void *data, int prevtoken)
1056 {
1057 	if (!e) {
1058 		fn(data, NULL, "y");
1059 		return;
1060 	}
1061 
1062 	if (expr_compare_type(prevtoken, e->type) > 0)
1063 		fn(data, NULL, "(");
1064 	switch (e->type) {
1065 	case E_SYMBOL:
1066 		if (e->left.sym->name)
1067 			fn(data, e->left.sym, e->left.sym->name);
1068 		else
1069 			fn(data, NULL, "<choice>");
1070 		break;
1071 	case E_NOT:
1072 		fn(data, NULL, "!");
1073 		expr_print(e->left.expr, fn, data, E_NOT);
1074 		break;
1075 	case E_EQUAL:
1076 		if (e->left.sym->name)
1077 			fn(data, e->left.sym, e->left.sym->name);
1078 		else
1079 			fn(data, NULL, "<choice>");
1080 		fn(data, NULL, "=");
1081 		fn(data, e->right.sym, e->right.sym->name);
1082 		break;
1083 	case E_UNEQUAL:
1084 		if (e->left.sym->name)
1085 			fn(data, e->left.sym, e->left.sym->name);
1086 		else
1087 			fn(data, NULL, "<choice>");
1088 		fn(data, NULL, "!=");
1089 		fn(data, e->right.sym, e->right.sym->name);
1090 		break;
1091 	case E_OR:
1092 		expr_print(e->left.expr, fn, data, E_OR);
1093 		fn(data, NULL, " || ");
1094 		expr_print(e->right.expr, fn, data, E_OR);
1095 		break;
1096 	case E_AND:
1097 		expr_print(e->left.expr, fn, data, E_AND);
1098 		fn(data, NULL, " && ");
1099 		expr_print(e->right.expr, fn, data, E_AND);
1100 		break;
1101 	case E_LIST:
1102 		fn(data, e->right.sym, e->right.sym->name);
1103 		if (e->left.expr) {
1104 			fn(data, NULL, " ^ ");
1105 			expr_print(e->left.expr, fn, data, E_LIST);
1106 		}
1107 		break;
1108 	case E_RANGE:
1109 		fn(data, NULL, "[");
1110 		fn(data, e->left.sym, e->left.sym->name);
1111 		fn(data, NULL, " ");
1112 		fn(data, e->right.sym, e->right.sym->name);
1113 		fn(data, NULL, "]");
1114 		break;
1115 	default:
1116 	  {
1117 		char buf[32];
1118 		sprintf(buf, "<unknown type %d>", e->type);
1119 		fn(data, NULL, buf);
1120 		break;
1121 	  }
1122 	}
1123 	if (expr_compare_type(prevtoken, e->type) > 0)
1124 		fn(data, NULL, ")");
1125 }
1126 
1127 static void expr_print_file_helper(void *data, struct symbol *sym, const char *str)
1128 {
1129 	xfwrite(str, strlen(str), 1, data);
1130 }
1131 
1132 void expr_fprint(struct expr *e, FILE *out)
1133 {
1134 	expr_print(e, expr_print_file_helper, out, E_NONE);
1135 }
1136 
1137 static void expr_print_gstr_helper(void *data, struct symbol *sym, const char *str)
1138 {
1139 	struct gstr *gs = (struct gstr*)data;
1140 	const char *sym_str = NULL;
1141 
1142 	if (sym)
1143 		sym_str = sym_get_string_value(sym);
1144 
1145 	if (gs->max_width) {
1146 		unsigned extra_length = strlen(str);
1147 		const char *last_cr = strrchr(gs->s, '\n');
1148 		unsigned last_line_length;
1149 
1150 		if (sym_str)
1151 			extra_length += 4 + strlen(sym_str);
1152 
1153 		if (!last_cr)
1154 			last_cr = gs->s;
1155 
1156 		last_line_length = strlen(gs->s) - (last_cr - gs->s);
1157 
1158 		if ((last_line_length + extra_length) > gs->max_width)
1159 			str_append(gs, "\\\n");
1160 	}
1161 
1162 	str_append(gs, str);
1163 	if (sym && sym->type != S_UNKNOWN)
1164 		str_printf(gs, " [=%s]", sym_str);
1165 }
1166 
1167 void expr_gstr_print(struct expr *e, struct gstr *gs)
1168 {
1169 	expr_print(e, expr_print_gstr_helper, gs, E_NONE);
1170 }
1171