expression.c

#include "expression.h"
#include "fixed-point.h"

#include <linux/ctype.h> /* for isspace */
#include <linux/kernel.h>
#include <linux/limits.h>
#include <linux/module.h>
#include <linux/string.h>

#define NAN_INT ((1 << 4) | ((1 << 4) - 1))
#define INF_INT ((1 << 5) | ((1 << 4) - 1))
#define MASK(n) (((n) > 0) << 4)
/*
 * LSB 4 bits for precision, 2^3, one for sign
 * MSB 28 bits for integer
 * LSB all 1 is NaN
 */

/*
 * Expression data types
 */

static uint64_t GET_NUM(uint64_t n)
{
    fixedp num = {.data = n};
    num.frac = 0;
    return num.data;
}

static uint64_t GET_FRAC(uint64_t n)
{
    fixedp num = {.data = n};
    num.inte = 0;
    return num.data;
}

static int FP2INT(int n, int d)
{
    while (n && n % 10 == 0) {
        ++d;
        n /= 10;
    }
    if (d == -1) {
        n *= 10;
        --d;
    }

    return ((n << 4) | (d & 15));
}

static int isNan(int x)
{
    return GET_FRAC(x) == GET_FRAC(NAN_INT);
}

enum compare_type { LOWER = 1, EQUAL = 2, GREATER = 4 };

enum expr_type {
    OP_UNKNOWN,
    OP_UNARY_MINUS,
    OP_UNARY_LOGICAL_NOT,
    OP_UNARY_BITWISE_NOT,

    OP_POWER,
    OP_DIVIDE,
    OP_MULTIPLY,
    OP_REMAINDER,

    OP_PLUS,
    OP_MINUS,

    OP_SHL,
    OP_SHR,

    OP_LT,
    OP_LE,
    OP_GT,
    OP_GE,
    OP_EQ,
    OP_NE,

    OP_BITWISE_AND,
    OP_BITWISE_OR,
    OP_BITWISE_XOR,

    OP_LOGICAL_AND,
    OP_LOGICAL_OR,

    OP_ASSIGN,
    OP_COMMA,

    OP_CONST,
    OP_VAR,
    OP_FUNC,
};

static int prec[] = {0, 1, 1, 1, 2, 2, 2, 2, 3,  3,  4,  4, 5, 5,
                     5, 5, 5, 5, 6, 7, 8, 9, 10, 11, 12, 0, 0, 0};

/* clang-format off */
#define expr_init()         \
    {                       \
        (enum expr_type) 0, \
        { { 0 } }           \
    }
/* clang-format on */

static int expr_is_unary(enum expr_type op)
{
    return op == OP_UNARY_MINUS || op == OP_UNARY_LOGICAL_NOT ||
           op == OP_UNARY_BITWISE_NOT;
}

static int expr_is_binary(enum expr_type op)
{
    return !expr_is_unary(op) && op != OP_CONST && op != OP_VAR &&
           op != OP_FUNC && op != OP_UNKNOWN;
}

static int expr_prec(enum expr_type a, enum expr_type b)
{
    int left =
        expr_is_binary(a) && a != OP_ASSIGN && a != OP_POWER && a != OP_COMMA;
    return (left && prec[a] >= prec[b]) || (prec[a] > prec[b]);
}

#define isfirstvarchr(c) \
    (((unsigned char) c >= '@' && c != '^' && c != '|') || c == '$')
#define isvarchr(c)                                                    \
    (((unsigned char) c >= '@' && c != '^' && c != '|') || c == '$' || \
     c == '#' || (c >= '0' && c <= '9'))

static struct {
    const char *s;
    const enum expr_type op;
} OPS[] = {
    {"-u", OP_UNARY_MINUS},
    {"!u", OP_UNARY_LOGICAL_NOT},
    {"^u", OP_UNARY_BITWISE_NOT},
    {"**", OP_POWER},
    {"*", OP_MULTIPLY},
    {"/", OP_DIVIDE},
    {"%", OP_REMAINDER},
    {"+", OP_PLUS},
    {"-", OP_MINUS},
    {"<<", OP_SHL},
    {">>", OP_SHR},
    {"<", OP_LT},
    {"<=", OP_LE},
    {">", OP_GT},
    {">=", OP_GE},
    {"==", OP_EQ},
    {"!=", OP_NE},
    {"&", OP_BITWISE_AND},
    {"|", OP_BITWISE_OR},
    {"^", OP_BITWISE_XOR},
    {"&&", OP_LOGICAL_AND},
    {"||", OP_LOGICAL_OR},
    {"=", OP_ASSIGN},
    {",", OP_COMMA},

    /* These are used by lexer and must be ignored by parser, so we put
       them at the end */
    {"-", OP_UNARY_MINUS},
    {"!", OP_UNARY_LOGICAL_NOT},
    {"^", OP_UNARY_BITWISE_NOT},
};

static enum expr_type expr_op(const char *s, size_t len, int unary)
{
    for (unsigned int i = 0; i < sizeof(OPS) / sizeof(OPS[0]); i++) {
        if (strlen(OPS[i].s) == len && strncmp(OPS[i].s, s, len) == 0 &&
            (unary == -1 || expr_is_unary(OPS[i].op) == unary)) {
            return OPS[i].op;
        }
    }
    return OP_UNKNOWN;
}

static uint64_t expr_parse_number(const char *s, size_t len)
{
    fixedp num = {0};
    int frac = 0;
    int dot = 0; /* FIXME: not good enough */
    unsigned int digits = 0;
    for (unsigned int i = 0; i < len; i++) {
        if (s[i] == '.' && dot == 0) {
            dot = i + 1;
            continue;
        }
        if (isdigit(s[i])) {
            if (dot)
                frac++;
            else
                digits++;
            // num = num * 10 + (s[i] - '0');
        } else
            return NAN_INT;
    }

    const int pow10[] = {
        1,      10,      100,      1000,      10000,
        100000, 1000000, 10000000, 100000000, 1000000000,
    };

    if (dot) {
        uint32_t ipt = 0, tmp;
        uint32_t mask = pow10[frac];
        int i = 31;
        sscanf(s, "%u.%u", &tmp, &ipt);
        while (ipt && i) {
            ipt <<= 1;
            if (ipt >= mask) {
                num.frac |= 1 << i;
                ipt %= mask;
            }
            i--;
        }
    }

    if (digits)
        sscanf(s, "%u", &num.inte);

    return (digits > 0 ? num.data : NAN_INT);
}

/*
 * Functions
 */
struct expr_func *expr_func(struct expr_func *funcs, const char *s, size_t len)
{
    for (struct expr_func *f = funcs; f->name; f++) {
        if (strlen(f->name) == len && strncmp(f->name, s, len) == 0)
            return f;
    }
    return NULL;
}

/*
 * Variables
 */

struct expr_var *expr_var(struct expr_var_list *vars, const char *s, size_t len)
{
    struct expr_var *v = NULL;
    if (len == 0 || !isfirstvarchr(*s))
        return NULL;
    for (v = vars->head; v; v = v->next) {
        if (strlen(v->name) == len && strncmp(v->name, s, len) == 0)
            return v;
    }
    v = (struct expr_var *) kcalloc(1, sizeof(struct expr_var) + len + 1,
                                    GFP_KERNEL);
    if (!v)
        return NULL; /* allocation failed */
    v->next = vars->head;
    v->value = 0;
    strncpy(v->name, s, len);
    v->name[len] = '\0';
    vars->head = v;
    return v;
}

static uint64_t mult(uint64_t a, uint64_t b)
{
    fixedp fa = {.data = a}, fb = {.data = b};
    /* (a + b) * (c + d) = ac + ad + bc + bd */
    fixedp result = {.inte = fa.inte * fb.inte};
    uint64_t ad = (GET_NUM(a) >> 32) * GET_FRAC(b);
    uint64_t bc = GET_FRAC(a) * (GET_NUM(b) >> 32);

    return result.data + ad + bc;
}

static int divid(uint64_t a, uint64_t b)
{
    return (a << 32) / (b << 32);
}

static int remain(int a, int b)
{
    int frac1 = GET_FRAC(a);
    int frac2 = GET_FRAC(b);
    int n1 = GET_NUM(a);
    int n2 = GET_NUM(b);
    if (n2 == 0)
        return NAN_INT;

    /* to int */
    while (frac1 > 0) {
        n1 /= 10;
        --frac1;
    }
    while (frac2 > 0) {
        n2 /= 10;
        --frac2;
    }

    n1 %= n2;

    return FP2INT(n1, frac1);
}

static uint64_t right_shift(uint64_t a, int b)
{
    /* FIXME: should use 2-base? */
    return a >> b;
}

static int power(int a, int b)
{
    int frac1 = GET_FRAC(a);
    int frac2 = GET_FRAC(b);
    int n1 = GET_NUM(a);
    int n2 = GET_NUM(b);

    while (frac2 != 0) {
        if (frac2 < 0) {
            n2 /= 10;
            ++frac2;
        } else {
            n2 *= 10;
            --frac2;
        }
    }

    int on1 = n1;
    int of1 = frac1;
    n1 = 1;
    for (int i = 0; i < n2; ++i) {
        frac1 += of1;
        n1 *= on1;
    }
    return FP2INT(n1, frac1);
}

static uint64_t left_shift(uint64_t a, int b)
{
    /* FIXME: should use 2-base? */
    return a << b;
}

static uint64_t plus(uint64_t a, uint64_t b)
{
    return a + b;
}

static uint64_t minus(uint64_t a, uint64_t b)
{
    return a - b;
}

static int compare(uint64_t a, uint64_t b)
{
    int flags = 0;
    if (a < b)
        flags |= LOWER;
    if (a > b)
        flags |= GREATER;
    if (a == b)
        flags |= EQUAL;

    return flags;
}

static uint64_t not(uint64_t a)
{
    return !a;
}

/* TODO: change logic */
uint64_t expr_eval(struct expr *e)
{
    int n;
    switch (e->type) {
    case OP_UNARY_MINUS: /* OK */
        return minus(0, expr_eval(&e->param.op.args.buf[0]));
    case OP_UNARY_LOGICAL_NOT: /* OK */
        return not(expr_eval(&e->param.op.args.buf[0]));
    case OP_UNARY_BITWISE_NOT: /* OK */
        return ~expr_eval(&e->param.op.args.buf[0]);
    case OP_POWER:
        return power(expr_eval(&e->param.op.args.buf[0]),
                     expr_eval(&e->param.op.args.buf[1]));
    case OP_MULTIPLY: /* OK */
        return mult(expr_eval(&e->param.op.args.buf[0]),
                    expr_eval(&e->param.op.args.buf[1]));
    case OP_DIVIDE: /* OK, precision */
        return divid(expr_eval(&e->param.op.args.buf[0]),
                     expr_eval(&e->param.op.args.buf[1]));
    case OP_REMAINDER: /* OK, no floating point */
        return remain(expr_eval(&e->param.op.args.buf[0]),
                      expr_eval(&e->param.op.args.buf[1]));
    case OP_PLUS: /* OK */
        return plus(expr_eval(&e->param.op.args.buf[0]),
                    expr_eval(&e->param.op.args.buf[1]));
    case OP_MINUS: /* OK */
        return minus(expr_eval(&e->param.op.args.buf[0]),
                     expr_eval(&e->param.op.args.buf[1]));
    case OP_SHL: /* oK */
        return left_shift(expr_eval(&e->param.op.args.buf[0]),
                          expr_eval(&e->param.op.args.buf[1]));
    case OP_SHR: /* OK */
        return right_shift(expr_eval(&e->param.op.args.buf[0]),
                           expr_eval(&e->param.op.args.buf[1]));
    case OP_LT: /* OK */
        return MASK(compare(expr_eval(&e->param.op.args.buf[0]),
                            expr_eval(&e->param.op.args.buf[1])) &
                    (LOWER));
    case OP_LE: /* OK */
        return MASK(compare(expr_eval(&e->param.op.args.buf[0]),
                            expr_eval(&e->param.op.args.buf[1])) &
                    (LOWER | EQUAL));
    case OP_GT: /* OK */
        return MASK(compare(expr_eval(&e->param.op.args.buf[0]),
                            expr_eval(&e->param.op.args.buf[1])) &
                    (GREATER));
    case OP_GE: /* OK */
        return MASK(compare(expr_eval(&e->param.op.args.buf[0]),
                            expr_eval(&e->param.op.args.buf[1])) &
                    (GREATER | EQUAL));
    case OP_EQ: /* OK */
        return MASK(compare(expr_eval(&e->param.op.args.buf[0]),
                            expr_eval(&e->param.op.args.buf[1])) &
                    (EQUAL));
    case OP_NE: /* OK */
        return MASK(!(compare(expr_eval(&e->param.op.args.buf[0]),
                              expr_eval(&e->param.op.args.buf[1])) &
                      (EQUAL)));
    case OP_BITWISE_AND: /* OK */
        return expr_eval(&e->param.op.args.buf[0]) &
               expr_eval(&e->param.op.args.buf[1]);
    case OP_BITWISE_OR: /* OK */
        return expr_eval(&e->param.op.args.buf[0]) |
               expr_eval(&e->param.op.args.buf[1]);
    case OP_BITWISE_XOR: /* OK */
        return expr_eval(&e->param.op.args.buf[0]) ^
               expr_eval(&e->param.op.args.buf[1]);
    case OP_LOGICAL_AND: /* OK */
        n = expr_eval(&e->param.op.args.buf[0]);
        if (n) {
            n = expr_eval(&e->param.op.args.buf[1]);
            if (n)
                return n;
        }
        return 0;
    case OP_LOGICAL_OR: /* OK */
        n = expr_eval(&e->param.op.args.buf[0]);
        if (n && !isNan(n)) {
            return n;
        } else {
            n = expr_eval(&e->param.op.args.buf[1]);
            if (n)
                return n;
        }
        return 0;
    case OP_ASSIGN: /* OK */
        n = expr_eval(&e->param.op.args.buf[1]);
        if (vec_nth(&e->param.op.args, 0).type == OP_VAR)
            *e->param.op.args.buf[0].param.var.value = n;
        return n;
    case OP_COMMA: /* OK */
        expr_eval(&e->param.op.args.buf[0]);
        return expr_eval(&e->param.op.args.buf[1]);
    case OP_CONST:
        return e->param.num.value;
    case OP_VAR:
        return *e->param.var.value;
    case OP_FUNC:
        return e->param.func.f->f(e->param.func.f, e->param.func.args,
                                  e->param.func.context);
    default:
        return NAN_INT;
    }
}

int expr_next_token(const char *s, size_t len, int *flags)
{
    unsigned int i = 0;
    if (len == 0) {
        return 0;
    }
    char c = s[0];
    if (c == '#') {
        for (; i < len && s[i] != '\n'; i++)
            ;
        return i;
    } else if (c == '\n') {
        for (; i < len && isspace(s[i]); i++)
            ;
        if (*flags & EXPR_TOP) {
            if (i == len || s[i] == ')')
                *flags = *flags & (~EXPR_COMMA);
            else
                *flags = EXPR_TNUMBER | EXPR_TWORD | EXPR_TOPEN | EXPR_COMMA;
        }
        return i;
    } else if (isspace(c)) {
        while (i < len && isspace(s[i]) && s[i] != '\n') {
            i++;
        }
        return i;
    } else if (isdigit(c)) {
        if ((*flags & EXPR_TNUMBER) == 0)
            return -1; /* unexpected number */
        *flags = EXPR_TOP | EXPR_TCLOSE;
        while ((c == '.' || isdigit(c)) && i < len) {
            i++;
            c = s[i];
        }
        return i;
    } else if (isfirstvarchr(c)) {
        if ((*flags & EXPR_TWORD) == 0)
            return -2; /* unexpected word */
        *flags = EXPR_TOP | EXPR_TOPEN | EXPR_TCLOSE;
        while ((isvarchr(c)) && i < len) {
            i++;
            c = s[i];
        }
        return i;
    } else if (c == '(' || c == ')') {
        if (c == '(' && (*flags & EXPR_TOPEN) != 0)
            *flags = EXPR_TNUMBER | EXPR_TWORD | EXPR_TOPEN | EXPR_TCLOSE;
        else if (c == ')' && (*flags & EXPR_TCLOSE) != 0)
            *flags = EXPR_TOP | EXPR_TCLOSE;
        else
            return -3; /* unexpected parenthesis */
        return 1;
    } else {
        if ((*flags & EXPR_TOP) == 0) {
            if (expr_op(&c, 1, 1) == OP_UNKNOWN)
                return -4; /* missing expected operand */
            *flags = EXPR_TNUMBER | EXPR_TWORD | EXPR_TOPEN | EXPR_UNARY;
            return 1;
        } else {
            int found = 0;
            while (!isvarchr(c) && !isspace(c) && c != '(' && c != ')' &&
                   i < len) {
                if (expr_op(s, i + 1, 0) != OP_UNKNOWN)
                    found = 1;
                else if (found)
                    break;
                i++;
                c = s[i];
            }
            if (!found)
                return -5; /* unknown operator */
            *flags = EXPR_TNUMBER | EXPR_TWORD | EXPR_TOPEN;
            return i;
        }
    }
}

#define EXPR_PAREN_ALLOWED 0
#define EXPR_PAREN_EXPECTED 1
#define EXPR_PAREN_FORBIDDEN 2

static int expr_bind(const char *s, size_t len, vec_expr_t *es)
{
    enum expr_type op = expr_op(s, len, -1);
    if (op == OP_UNKNOWN)
        return -1;

    if (expr_is_unary(op)) {
        if (vec_len(es) < 1)
            return -1;
        struct expr arg = vec_pop(es);
        struct expr unary = expr_init();
        unary.type = op;
        vec_push(&unary.param.op.args, arg);
        vec_push(es, unary);
    } else {
        if (vec_len(es) < 2)
            return -1;
        struct expr b = vec_pop(es);
        struct expr a = vec_pop(es);
        struct expr binary = expr_init();
        binary.type = op;
        if (op == OP_ASSIGN && a.type != OP_VAR)
            return -1; /* Bad assignment */
        vec_push(&binary.param.op.args, a);
        vec_push(&binary.param.op.args, b);
        vec_push(es, binary);
    }
    return 0;
}

static struct expr expr_const(uint64_t value)
{
    struct expr e = expr_init();
    e.type = OP_CONST;
    e.param.num.value = value;
    return e;
}

static struct expr expr_varref(struct expr_var *v)
{
    struct expr e = expr_init();
    e.type = OP_VAR;
    e.param.var.value = &v->value;
    return e;
}

static struct expr expr_binary(enum expr_type type,
                               struct expr a,
                               struct expr b)
{
    struct expr e = expr_init();
    e.type = type;
    vec_push(&e.param.op.args, a);
    vec_push(&e.param.op.args, b);
    return e;
}

static inline void expr_copy(struct expr *dst, struct expr *src)
{
    int i;
    struct expr arg;
    dst->type = src->type;
    if (src->type == OP_FUNC) {
        dst->param.func.f = src->param.func.f;
        vec_foreach (&src->param.func.args, arg, i) {
            struct expr tmp = expr_init();
            expr_copy(&tmp, &arg);
            vec_push(&dst->param.func.args, tmp);
        }
        if (src->param.func.f->ctxsz > 0) {
            dst->param.func.context =
                kcalloc(1, src->param.func.f->ctxsz, GFP_KERNEL);
        }
    } else if (src->type == OP_CONST) {
        dst->param.num.value = src->param.num.value;
    } else if (src->type == OP_VAR) {
        dst->param.var.value = src->param.var.value;
    } else {
        vec_foreach (&src->param.op.args, arg, i) {
            struct expr tmp = expr_init();
            expr_copy(&tmp, &arg);
            vec_push(&dst->param.op.args, tmp);
        }
    }
}

static void expr_destroy_args(struct expr *e);

struct expr *expr_create(const char *s,
                         size_t len,
                         struct expr_var_list *vars,
                         struct expr_func *funcs)
{
    uint64_t num;
    struct expr_var *v;
    const char *id = NULL;
    size_t idn = 0;

    struct expr *result = NULL;

    vec_expr_t es = vec_init();
    vec_str_t os = vec_init();
    vec_arg_t as = vec_init();

    struct macro {
        char *name;
        vec_expr_t body;
    };
    vec(struct macro) macros = vec_init();

    int flags = EXPR_TDEFAULT;
    int paren = EXPR_PAREN_ALLOWED;
    for (;;) {
        int n = expr_next_token(s, len, &flags);
        if (n == 0)
            break;
        else if (n < 0)
            goto cleanup;
        const char *tok = s;
        s = s + n;
        len = len - n;
        if (*tok == '#')
            continue;
        if (flags & EXPR_UNARY) {
            if (n == 1) {
                switch (*tok) {
                case '-':
                    tok = "-u";
                    break;
                case '^':
                    tok = "^u";
                    break;
                case '!':
                    tok = "!u";
                    break;
                default:
                    goto cleanup;
                }
                n = 2;
            }
        }
        if (*tok == '\n' && (flags & EXPR_COMMA)) {
            flags = flags & (~EXPR_COMMA);
            n = 1;
            tok = ",";
        }
        if (isspace(*tok))
            continue;
        int paren_next = EXPR_PAREN_ALLOWED;

        if (idn > 0) {
            if (n == 1 && *tok == '(') {
                int i;
                int has_macro = 0;
                struct macro m;
                vec_foreach (&macros, m, i) {
                    if (strlen(m.name) == idn &&
                        strncmp(m.name, id, idn) == 0) {
                        has_macro = 1;
                        break;
                    }
                }
                if ((idn == 1 && id[0] == '$') || has_macro ||
                    expr_func(funcs, id, idn)) {
                    struct expr_string str = {id, (int) idn};
                    vec_push(&os, str);
                    paren = EXPR_PAREN_EXPECTED;
                } else
                    goto cleanup; /* invalid function name */
            } else if ((v = expr_var(vars, id, idn))) {
                vec_push(&es, expr_varref(v));
                paren = EXPR_PAREN_FORBIDDEN;
            }
            id = NULL;
            idn = 0;
        }

        if (n == 1 && *tok == '(') {
            if (paren == EXPR_PAREN_EXPECTED) {
                struct expr_string str = {"{", 1};
                vec_push(&os, str);
                struct expr_arg arg = {vec_len(&os), vec_len(&es), vec_init()};
                vec_push(&as, arg);
            } else if (paren == EXPR_PAREN_ALLOWED) {
                struct expr_string str = {"(", 1};
                vec_push(&os, str);
            } else
                goto cleanup; /* Bad call */
        } else if (paren == EXPR_PAREN_EXPECTED) {
            goto cleanup; /* Bad call */
        } else if (n == 1 && *tok == ')') {
            int minlen = (vec_len(&as) > 0 ? vec_peek(&as).oslen : 0);
            while (vec_len(&os) > minlen && *vec_peek(&os).s != '(' &&
                   *vec_peek(&os).s != '{') {
                struct expr_string str = vec_pop(&os);
                if (expr_bind(str.s, str.n, &es) == -1)
                    goto cleanup;
            }
            if (vec_len(&os) == 0)
                goto cleanup; /* Bad parens */
            struct expr_string str = vec_pop(&os);
            if (str.n == 1 && *str.s == '{') {
                str = vec_pop(&os);
                struct expr_arg arg = vec_pop(&as);
                if (vec_len(&es) > arg.eslen)
                    vec_push(&arg.args, vec_pop(&es));
                if (str.n == 1 && str.s[0] == '$') {
                    if (vec_len(&arg.args) < 1) {
                        vec_free(&arg.args);
                        goto cleanup; /* too few arguments for $() function */
                    }
                    struct expr *u = &vec_nth(&arg.args, 0);
                    if (u->type != OP_VAR) {
                        vec_free(&arg.args);
                        goto cleanup; /* first argument is not a variable */
                    }
                    for (v = vars->head; v; v = v->next) {
                        if (&v->value == u->param.var.value) {
                            struct macro m = {v->name, arg.args};
                            vec_push(&macros, m);
                            break;
                        }
                    }
                    vec_push(&es, expr_const(0));
                } else {
                    int i = 0;
                    int found = -1;
                    struct macro m;
                    vec_foreach (&macros, m, i) {
                        if (strlen(m.name) == (size_t) str.n &&
                            strncmp(m.name, str.s, str.n) == 0)
                            found = i;
                    }
                    if (found != -1) {
                        m = vec_nth(&macros, found);
                        struct expr root = expr_const(0);
                        struct expr *p = &root;
                        /* Assign macro parameters */
                        for (int j = 0; j < vec_len(&arg.args); j++) {
                            char varname[4];
                            snprintf(varname, sizeof(varname) - 1, "$%d",
                                     (j + 1));
                            v = expr_var(vars, varname, strlen(varname));
                            struct expr ev = expr_varref(v);
                            struct expr assign = expr_binary(
                                OP_ASSIGN, ev, vec_nth(&arg.args, j));
                            *p = expr_binary(OP_COMMA, assign, expr_const(0));
                            p = &vec_nth(&p->param.op.args, 1);
                        }
                        /* Expand macro body */
                        for (int j = 1; j < vec_len(&m.body); j++) {
                            if (j < vec_len(&m.body) - 1) {
                                *p = expr_binary(OP_COMMA, expr_const(0),
                                                 expr_const(0));
                                expr_copy(&vec_nth(&p->param.op.args, 0),
                                          &vec_nth(&m.body, j));
                            } else
                                expr_copy(p, &vec_nth(&m.body, j));
                            p = &vec_nth(&p->param.op.args, 1);
                        }
                        vec_push(&es, root);
                        vec_free(&arg.args);
                    } else {
                        struct expr_func *f = expr_func(funcs, str.s, str.n);
                        struct expr bound_func = expr_init();
                        bound_func.type = OP_FUNC;
                        bound_func.param.func.f = f;
                        bound_func.param.func.args = arg.args;
                        if (f->ctxsz > 0) {
                            void *p = kcalloc(1, f->ctxsz, GFP_KERNEL);
                            if (!p)
                                goto cleanup; /* allocation failed */
                            bound_func.param.func.context = p;
                        }
                        vec_push(&es, bound_func);
                    }
                }
            }
            paren_next = EXPR_PAREN_FORBIDDEN;
        } else if (!isNan(num = expr_parse_number(tok, n))) {
            vec_push(&es, expr_const(num));
            paren_next = EXPR_PAREN_FORBIDDEN;
        } else if (expr_op(tok, n, -1) != OP_UNKNOWN) {
            enum expr_type op = expr_op(tok, n, -1);
            struct expr_string o2 = {NULL, 0};
            if (vec_len(&os) > 0)
                o2 = vec_peek(&os);
            for (;;) {
                if (n == 1 && *tok == ',' && vec_len(&os) > 0) {
                    struct expr_string str = vec_peek(&os);
                    if (str.n == 1 && *str.s == '{') {
                        struct expr e = vec_pop(&es);
                        vec_push(&vec_peek(&as).args, e);
                        break;
                    }
                }
                enum expr_type type2 = expr_op(o2.s, o2.n, -1);
                if (!(type2 != OP_UNKNOWN && expr_prec(op, type2))) {
                    struct expr_string str = {tok, n};
                    vec_push(&os, str);
                    break;
                }

                if (expr_bind(o2.s, o2.n, &es) == -1)
                    goto cleanup;
                (void) vec_pop(&os);
                if (vec_len(&os) > 0)
                    o2 = vec_peek(&os);
                else
                    o2.n = 0;
            }
        } else {
            if (n > 0 && !isdigit(*tok)) {
                /* Valid identifier, a variable or a function */
                id = tok;
                idn = n;
            } else
                goto cleanup; /* Bad variable name, e.g. '2.3.4' or '4ever' */
        }
        paren = paren_next;
    }

    if (idn > 0)
        vec_push(&es, expr_varref(expr_var(vars, id, idn)));

    while (vec_len(&os) > 0) {
        struct expr_string rest = vec_pop(&os);
        if (rest.n == 1 && (*rest.s == '(' || *rest.s == ')'))
            goto cleanup; /* Bad paren */
        if (expr_bind(rest.s, rest.n, &es) == -1)
            goto cleanup;
    }

    result = (struct expr *) kcalloc(1, sizeof(struct expr), GFP_KERNEL);
    if (result) {
        if (vec_len(&es) == 0)
            result->type = OP_CONST;
        else
            *result = vec_pop(&es);
    }

    int i, j;
    struct macro m;
    struct expr e;
    struct expr_arg a;
cleanup:
    vec_foreach (&macros, m, i) {
        vec_foreach (&m.body, e, j) {
            expr_destroy_args(&e);
        }
        vec_free(&m.body);
    }
    vec_free(&macros);

    vec_foreach (&es, e, i) {
        expr_destroy_args(&e);
    }
    vec_free(&es);

    vec_foreach (&as, a, i) {
        vec_foreach (&a.args, e, j) {
            expr_destroy_args(&e);
        }
        vec_free(&a.args);
    }
    vec_free(&as);

    /*vec_foreach(&os, o, i) {vec_free(&m.body);}*/
    vec_free(&os);
    return result;
}

static void expr_destroy_args(struct expr *e)
{
    int i;
    struct expr arg;
    if (e->type == OP_FUNC) {
        vec_foreach (&e->param.func.args, arg, i) {
            expr_destroy_args(&arg);
        }
        vec_free(&e->param.func.args);
        if (e->param.func.context) {
            if (e->param.func.f->cleanup) {
                e->param.func.f->cleanup(e->param.func.f,
                                         e->param.func.context);
            }
            kfree(e->param.func.context);
        }
    } else if (e->type != OP_CONST && e->type != OP_VAR) {
        vec_foreach (&e->param.op.args, arg, i) {
            expr_destroy_args(&arg);
        }
        vec_free(&e->param.op.args);
    }
}

void expr_destroy(struct expr *e, struct expr_var_list *vars)
{
    if (e) {
        expr_destroy_args(e);
        kfree(e);
    }
    if (vars) {
        for (struct expr_var *v = vars->head; v;) {
            struct expr_var *next = v->next;
            kfree(v);
            v = next;
        }
    }
}