1 files changed, 338 insertions, 0 deletions

diff --git a/programs/src/kconfig_parser/cnfexpr.c b/programs/src/kconfig_parser/cnfexpr.c
new file mode 100644
index 0000000..50f0929
--- /dev/null
+++ b/programs/src/kconfig_parser/cnfexpr.c
@@ -0,0 +1,338 @@
+#include "cnfexpr.h"
+
+struct cnfexpr *cnf_sym(struct symlist *sl, bool not, struct symbol *sym);
+struct cnfexpr *cnf_eql(struct symlist *sl, bool not, struct symbol *sym1,
+                        struct symbol *sym2);
+struct cnfexpr *cnf_or(struct cnfexpr *e1, struct cnfexpr *e2);
+struct cnfexpr *cnf_and(struct cnfexpr *e1, struct cnfexpr *e2);
+void free_cnf(struct cnfexpr *e);
+
+struct cnfexpr *kconfig_cnfexpr(struct symlist *sl, struct expr *expr) {
+    struct stck {
+        struct expr *expr;
+        struct cnfexpr *cnf;
+    };
+    struct expr **back;
+    int back_size = 2, back_pos = -1;
+    back = malloc((unsigned) back_size * sizeof(struct expr *));
+    struct stck *stack;
+    int stack_size = 2, stack_pos = -1;
+    stack = malloc((unsigned) stack_size * sizeof(struct stck));
+    struct cnfexpr *rtrn = NULL;
+    bool nt = false;
+
+    while (expr != NULL) {
+        if ((back_pos >= 0 && back[back_pos] != expr) || back_pos < 0) {
+            if (++back_pos == back_size) {
+                back_size *= 2;
+                back =
+                    realloc(back,
+                            (unsigned) back_size * sizeof(struct expr *));
+            }
+            back[back_pos] = expr;
+        }
+        switch (expr->type) {
+        case E_SYMBOL:
+            rtrn = cnf_sym(sl, nt, expr->left.sym);
+            goto go_up;
+        case E_NOT:
+            nt = !nt;
+            if (rtrn == NULL)
+                expr = expr->left.expr;
+            else
+                goto go_up;
+            break;
+        case E_OR:
+        case E_AND:
+            if (stack_pos < 0 || stack[stack_pos].expr != expr) {
+                if (rtrn == NULL) {
+                    expr = expr->left.expr;
+                } else {
+                    if (stack_size == ++stack_pos) {
+                        stack_size *= 2;
+                        stack =
+                            realloc(stack,
+                                    (unsigned) stack_size *
+                                    sizeof(struct stck));
+                    }
+                    stack[stack_pos].expr = expr;
+                    stack[stack_pos].cnf = rtrn;
+                    expr = expr->right.expr;
+                    rtrn = NULL;
+                }
+            } else {
+                if ((!nt && expr->type == E_OR)
+                    || (nt && expr->type == E_AND))
+                    rtrn = cnf_or(stack[stack_pos].cnf, rtrn);
+                else
+                    rtrn = cnf_and(stack[stack_pos].cnf, rtrn);
+                stack_pos--;
+                goto go_up;
+            }
+            break;
+        case E_EQUAL:
+            rtrn = cnf_eql(sl, nt, expr->left.sym, expr->right.sym);
+            goto go_up;
+        case E_UNEQUAL:
+            rtrn = cnf_eql(sl, !nt, expr->left.sym, expr->right.sym);
+            goto go_up;
+        default:
+            fprintf(stderr, "ERROR: Unknown expression type.\n");
+        }
+        continue;
+
+      go_up:
+        if (--back_pos >= 0)
+            expr = back[back_pos];
+        else
+            expr = NULL;
+    }
+
+    free(back);
+    free(stack);
+    return rtrn;
+}
+
+void cnf_printf(struct cnfexpr *wcnf) {
+    if (wcnf == NULL) {
+        printf("hey NULL\n");
+        return;
+    }
+    switch (wcnf->type) {
+    case CT_TRUE:
+        printf("True\n");
+        return;
+    case CT_FALSE:
+        printf("False\n");
+        return;
+    case CT_EXPR:
+        break;
+    }
+    unsigned i, r;
+    for (i = 0; i < wcnf->size; i++) {
+        for (r = 0; r < wcnf->sizes[i]; r++) {
+            printf("%d ", wcnf->exprs[i][r]);
+        }
+        if (i < wcnf->size - 1)
+            printf("x ");
+    }
+    printf("\n");
+}
+
+struct cnfexpr *cnf_sym(struct symlist *sl, bool not, struct symbol *sym) {
+    struct cnfexpr *w = malloc(sizeof(struct cnfexpr));
+    struct symlist_el *se = symlist_find(sl, sym->name);
+    if (se == NULL) {
+        if (!not)
+            w->type = CT_FALSE;
+        else
+            w->type = CT_TRUE;
+    } else {
+        w->type = CT_EXPR;
+        w->size = 1;
+        w->sizes = malloc(sizeof(int));
+        w->sizes[0] = 1;
+        w->exprs = malloc(sizeof(int *));
+        w->exprs[0] = malloc(sizeof(int));
+        w->exprs[0][0] = (int) se->id;
+        if (not)
+            w->exprs[0][0] *= -1;
+    }
+    return w;
+}
+
+struct cnfexpr *cnf_eql(struct symlist *sl, bool not, struct symbol *sym1,
+                        struct symbol *sym2) {
+    if (!strcmp(sym2->name, "m")) {
+        struct cnfexpr *fls = malloc(sizeof(struct cnfexpr));
+        if (!not)
+            fls->type = CT_FALSE;
+        else
+            fls->type = CT_TRUE;
+        return fls;
+    }
+    if (!strcmp(sym2->name, "n")) {
+        struct cnfexpr *w = cnf_sym(sl, not, sym1);
+        w->exprs[0][0] *= -1;
+        return w;
+    }
+    if (!strcmp(sym2->name, "y")) {
+        return cnf_sym(sl, not, sym1);
+    }
+
+    struct cnfexpr *w1 = cnf_sym(sl, not, sym1);
+    struct cnfexpr *w2 = cnf_sym(sl, not, sym2);
+    struct cnfexpr *w3 = cnf_sym(sl, !not, sym1);
+    struct cnfexpr *w4 = cnf_sym(sl, !not, sym2);
+    struct cnfexpr *wa = cnf_or(w1, w2);
+    struct cnfexpr *wb = cnf_or(w3, w4);
+    struct cnfexpr *w = cnf_and(wa, wb);
+    return w;
+}
+
+struct cnfexpr *cnf_or(struct cnfexpr *e1, struct cnfexpr *e2) {
+    switch (e1->type) {
+    case CT_TRUE:
+        free_cnf(e2);
+        return e1;
+    case CT_FALSE:
+        free_cnf(e1);
+        return e2;
+    case CT_EXPR:
+        switch (e2->type) {
+        case CT_TRUE:
+            free_cnf(e1);
+            return e2;
+        case CT_FALSE:
+            free_cnf(e2);
+            return e1;
+        case CT_EXPR:
+            break;
+        }
+        break;
+    }
+
+    unsigned oldsize = e1->size;
+    e1->size *= e2->size;
+    e1->sizes = realloc(e1->sizes, e1->size * sizeof(int));
+    e1->exprs = realloc(e1->exprs, e1->size * sizeof(int *));
+    unsigned i1, i2;
+    // Duplicate e2->size times e1
+    for (i2 = 1; i2 < e2->size; i2++) {
+        memcpy(e1->sizes + (i2 * oldsize), e1->sizes,
+               oldsize * sizeof(int));
+        for (i1 = 0; i1 < oldsize; i1++) {
+            e1->exprs[(i2 * oldsize) + i1] =
+                malloc(e1->sizes[i1] * sizeof(int));
+            memcpy(e1->exprs[(i2 * oldsize) + i1], e1->exprs[i1],
+                   e1->sizes[i1] * sizeof(int));
+        }
+    }
+    unsigned oldsizes;
+    // Append e2->exprs to e1->exprs
+    for (i2 = 0; i2 < e2->size; i2++) {
+        for (i1 = 0; i1 < oldsize; i1++) {
+            oldsizes = e1->sizes[(i2 * oldsize) + i1];
+            e1->sizes[(i2 * oldsize) + i1] += e2->sizes[i2];
+            e1->exprs[(i2 * oldsize) + i1] =
+                realloc(e1->exprs[(i2 * oldsize) + i1],
+                        e1->sizes[(i2 * oldsize) + i1] * sizeof(int));
+            memcpy(e1->exprs[(i2 * oldsize) + i1] + oldsizes,
+                   e2->exprs[i2], e2->sizes[i2] * sizeof(int));
+        }
+    }
+    free_cnf(e2);
+    return e1;
+}
+
+struct cnfexpr *cnf_and(struct cnfexpr *e1, struct cnfexpr *e2) {
+    switch (e1->type) {
+    case CT_FALSE:
+        free_cnf(e2);
+        return e1;
+    case CT_TRUE:
+        free_cnf(e1);
+        return e2;
+    case CT_EXPR:
+        switch (e2->type) {
+        case CT_FALSE:
+            free_cnf(e1);
+            return e2;
+        case CT_TRUE:
+            free_cnf(e2);
+            return e1;
+        case CT_EXPR:
+            break;
+        }
+        break;
+    }
+
+    unsigned oldsize = e1->size;
+    e1->size += e2->size;
+    e1->sizes = realloc(e1->sizes, e1->size * sizeof(int));
+    e1->exprs = realloc(e1->exprs, e1->size * sizeof(int *));
+    memcpy(e1->sizes + oldsize, e2->sizes, e2->size * sizeof(int));
+    unsigned i;
+    for (i = 0; i < e2->size; i++) {
+        e1->exprs[oldsize + i] = malloc(e2->sizes[i] * sizeof(int));
+        memcpy(e1->exprs[oldsize + i], e2->exprs[i],
+               e2->sizes[i] * sizeof(int));
+    }
+    free_cnf(e2);
+    return e1;
+}
+
+void free_cnf(struct cnfexpr *e) {
+    if (e->type != CT_EXPR) {
+        free(e);
+        return;
+    }
+    unsigned i;
+    for (i = 0; i < e->size; i++) {
+        free(e->exprs[i]);
+    }
+    free(e->exprs);
+    free(e->sizes);
+    free(e);
+}
+
+
+struct boolexp *printf_original(struct symlist *sl, struct expr *expr) {
+    switch (expr->type) {
+    case E_OR:
+        printf("  OR\n");
+        printf_original(sl, expr->left.expr);
+        printf_original(sl, expr->right.expr);
+        break;
+    case E_AND:
+        printf("  AND\n");
+        printf_original(sl, expr->left.expr);
+        printf_original(sl, expr->right.expr);
+        break;
+    case E_NOT:
+        printf("  NOT\n");
+        printf_original(sl, expr->left.expr);
+        break;
+    case E_EQUAL:
+        printf("  = ");
+        printf("%s ", expr->left.sym->name);
+        if (!strcmp("y", expr->right.sym->name))
+            printf("YES\n");
+        else if (!strcmp("n", expr->right.sym->name))
+            printf("NO\n");
+        else if (!strcmp("m", expr->right.sym->name))
+            printf("MODULE\n");
+        else
+            printf("%s\n", expr->left.sym->name);
+        break;
+    case E_UNEQUAL:
+        printf("  != ");
+        printf("%s ", expr->left.sym->name);
+        if (!strcmp("y", expr->right.sym->name))
+            printf("YES\n");
+        else if (!strcmp("n", expr->right.sym->name))
+            printf("NO\n");
+        else
+            printf("OTHER %s\n", expr->right.sym->name);
+        break;
+    case E_LIST:
+        printf("  list\n");
+        break;
+    case E_SYMBOL:
+        printf("  symbol");
+        if (expr->left.sym->name != NULL)
+                printf(": %s", expr->left.sym->name);
+        printf("\n");
+        break;
+    case E_RANGE:
+        printf("  range\n");
+        break;
+    case E_NONE:
+        printf("  none\n");
+        break;
+    default:
+        printf("  ERROR\n");
+        break;
+    }
+
+}