larpcc/resolve.c

#include "resolve.h"
#include "ast.h"
#include "collection.h"
#include <assert.h>
#include <stdio.h>
#include <stdlib.h>

struct OldSyms {
    ARRAY_FIELD(Hashmap, maps);
};

OldSyms* oldsyms_new()
{
    OldSyms* s = malloc(sizeof(OldSyms));
    *s = (OldSyms) {
        ARRAY_FIELD_INIT(maps),
    };

    ARRAY_FIELD_PUSH(s, maps, &(Hashmap) { });
    hashmap_init(&s->maps[0], sizeof(OldSym*));

    return s;
}

void oldsyms_free(OldSyms* s)
{
    for (size_t i = 0; i < s->maps_size; ++i)
        hashmap_deinit(&s->maps[i]);
    free(s->maps);
    free(s);
}

void oldsyms_push_scope(OldSyms* s)
{
    ARRAY_FIELD_PUSH(s, maps, &(Hashmap) { });
    hashmap_init(&s->maps[s->maps_size - 1], sizeof(OldSym*));
}

void oldsyms_pop_scope(OldSyms* s)
{
    assert(s->maps_size > 0);
    hashmap_deinit(&s->maps[s->maps_size - 1]);
    s->maps_size -= 1;
}

void oldsyms_define(OldSyms* s, const char* ident, OldSym* sym)
{
    Hashmap* m = &s->maps[s->maps_size - 1];
    hashmap_set(m, ident, &sym);
}

OldSym* oldsyms_get(OldSyms* s, const char* ident)
{
    OldSym** res;
    for (size_t i = 0; i < s->maps_size; ++i)
        if ((res = hashmap_get(&s->maps[s->maps_size - i - 1], ident)))
            return *res;
    return nullptr;
}

typedef enum : unsigned int {
    ResolverFlags_InTypedef,
} ResolverFlags;

struct Resolver {
    OldSyms* ord_syms;
    OldSyms* tag_syms;
    OldSyms* label_syms;

    Arena* arena;

    Keymap resol_map;

    AstStmt* current_decl_stmt;
    ResolverFlags flags;
};

Resolver* oldresolver_new(Arena* arena)
{
    Resolver* r = malloc(sizeof(Resolver));
    *r = (Resolver) {
        .ord_syms = oldsyms_new(),
        .tag_syms = oldsyms_new(),
        .label_syms = oldsyms_new(),

        .arena = arena,

        .resol_map = { },

        .current_decl_stmt = nullptr,
        .flags = 0,
    };

    keymap_init(&r->resol_map, sizeof(OldSym*));

    return r;
}

void oldresolver_free(Resolver* r)
{
    keymap_deinit(&r->resol_map);

    oldsyms_free(r->ord_syms);
    oldsyms_free(r->tag_syms);
    oldsyms_free(r->label_syms);
    free(r);
}

static AstVisitorResult visit_stmt_block(AstVisitor* v, AstStmtBlock* node)
{
    Resolver* r = v->op;
    oldsyms_push_scope(r->ord_syms);

    ast_stmt_block_visit_body(v, node);

    oldsyms_pop_scope(r->ord_syms);
    return AstVisitor_Break;
}

static AstVisitorResult visit_stmt(AstVisitor* v, AstStmt* node)
{
    Resolver* r = v->op;

    if (node->ty == AstStmt_Decl) {
        r->current_decl_stmt = node;

        ast_stmt_visit_body(v, node);

        return AstVisitor_Break;
    }

    return AstVisitor_Continue;
}

static AstVisitorResult visit_decl(AstVisitor* v, AstDecl* node)
{
    Resolver* r = v->op;

    if (node->spec & AstSpec_Typedef) {
        r->flags |= ResolverFlags_InTypedef;

        ast_decl_visit_body(v, node);

        r->flags &= ~ResolverFlags_InTypedef;
        return AstVisitor_Break;
    }

    if (node->function_body) {

        assert(node->declars_size == 1
            && node->declars[0]->ty == AstDeclar_Function);

        AstDeclar* d = node->declars[0];

        ast_declar_visit(v, d->function.declar);

        oldsyms_push_scope(r->ord_syms);

        for (size_t i = 0; i < d->function.params_size; ++i)
            ast_decl_visit(v, d->function.params[i]);

        ast_stmt_block_visit(v, node->function_body);

        oldsyms_pop_scope(r->ord_syms);
        return AstVisitor_Break;
    }

    return AstVisitor_Continue;
}

static AstVisitorResult visit_declar(AstVisitor* v, AstDeclar* node)
{
    Resolver* r = v->op;

    if (node->ty == AstDeclar_Function) {
        ast_declar_visit(v, node->function.declar);

        oldsyms_push_scope(r->ord_syms);

        for (size_t i = 0; i < node->function.params_size; ++i)
            ast_decl_visit(v, node->function.params[i]);

        oldsyms_pop_scope(r->ord_syms);
        return AstVisitor_Break;
    }

    if (node->ty == AstDeclar_Ident) {
        assert(r->current_decl_stmt);

        OldSym* sym = arena_alloc(r->arena, sizeof(OldSym));
        *sym = (OldSym) {
            .ty = OldSym_Declarator,
            .declarator = {
                .stmt = r->current_decl_stmt,
                .declar = node,
            },
        };
        oldsyms_define(r->ord_syms, node->ident, sym);
        return AstVisitor_Break;
    }

    return AstVisitor_Continue;
}

static AstVisitorResult visit_expr(AstVisitor* v, AstExpr* node)
{
    Resolver* r = v->op;

    if (node->ty == AstExpr_Ident) {

        OldSym* sym = oldsyms_get(r->ord_syms, node->ident);
        if (!sym) {
            fprintf(stderr,
                "error: use of undeclared '%s' on line %d\n",
                node->ident,
                node->loc.line);
            exit(1);
        }
    }

    return AstVisitor_Continue;
}

void oldresolver_resolve_top_stmt(Resolver* r, AstStmt* stmt)
{
    if (stmt->ty != AstStmt_Decl) {
        fprintf(
            stderr, "error: expected declaration on line %d\n", stmt->loc.line);
        exit(1);
    }

    AstVisitor v = {
        .visit_stmt_block = visit_stmt_block,
        .visit_stmt = visit_stmt,
        .visit_decl = visit_decl,
        .visit_declar = visit_declar,
        .visit_expr = visit_expr,
        .op = r,
    };

    ast_stmt_visit(&v, stmt);
}