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