diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/aup.h | 19 | ||||
| -rw-r--r-- | src/compile.c | 599 | ||||
| -rw-r--r-- | src/main.c | 38 |
3 files changed, 656 insertions, 0 deletions
diff --git a/src/aup.h b/src/aup.h new file mode 100644 index 0000000..aa48daf --- /dev/null +++ b/src/aup.h @@ -0,0 +1,19 @@ +#ifndef AUP_H +#define AUP_H +#include <stdbool.h> +#include <stddef.h> + +/* aup v0a: source text to a Pure module v1 (Pure docs/design/008). */ +enum { AUP_SOURCE_LIMIT = 32768, AUP_MESSAGE = 160 }; + +typedef struct { + int line, column; + char message[AUP_MESSAGE]; +} AupDiagnostic; + +/* Compile source into a newly allocated module (free with free()). The same + * source always produces the same bytes. On failure *module is NULL and the + * diagnostic names the first error. */ +bool aup_compile(const char *source, size_t length, unsigned char **module, size_t *size, + AupDiagnostic *error); +#endif diff --git a/src/compile.c b/src/compile.c new file mode 100644 index 0000000..24b5c71 --- /dev/null +++ b/src/compile.c @@ -0,0 +1,599 @@ +#include "aup.h" +#include <math.h> +#include <setjmp.h> +#include <stdarg.h> +#include <stdint.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> + +/* Limits and opcodes of Pure module format v1. */ +enum { STATES = 32, INPUTS = 8, VIEWS = 24, HANDLERS = 64, CODE = 512, STACK = 64, NAME = 32, DEPTH = 64 }; +enum { CONST, LOAD, DT, ADD, SUB, MUL, DIV, NEG, LT, LE, GT, GE, EQ, NE, AND, OR, NOT, + MIN, MAX, CLAMP, STORE, JZ, JMP }; + +typedef enum { T_EOF, T_NL, T_NUMBER, T_NAME, T_PUNCT } TokenKind; +typedef struct { + TokenKind kind; + int line, column; + double number; + char text[NAME + 1]; /* name, or punctuation of one or two characters */ +} Token; + +typedef enum { E_NUMBER, E_STATE, E_DT, E_UNARY, E_BINARY, E_CALL } ExprKind; +typedef struct Expr { + ExprKind kind; + int line, column, op, state; + double number; + char name[NAME]; + struct Expr *args[3]; + unsigned count; +} Expr; + +typedef struct Stmt { + bool branch; + int line, column, state; + char name[NAME]; + Expr *value; /* assignment value or branch condition */ + struct Stmt *then, *otherwise, *next; +} Stmt; + +typedef struct { + const char *source; + size_t length, at; + int line, column; + Token token, ahead; + bool has_ahead; + unsigned depth; + jmp_buf fail; + AupDiagnostic *error; + void **blocks; + size_t block_count, block_capacity; + /* Declarations in source order. */ + char state_names[STATES][NAME]; + double initial[STATES]; + unsigned states; + char input_names[INPUTS][NAME], input_keys[INPUTS][NAME]; + unsigned inputs; + char view_names[VIEWS][NAME]; + Expr *view_value[VIEWS]; + unsigned views; + struct { char input[NAME]; int edge, line, column; Stmt *body; } handlers[HANDLERS]; + unsigned handler_count; + /* Code generation. */ + struct { uint8_t op; uint32_t arg; double value; } code[CODE]; + unsigned code_count; + uint32_t view_code[VIEWS][2], handler_code[HANDLERS][2]; +} Compiler; + +static const char *const reserved[] = {"state", "input", "view", "on", "if", "else", "key", + "frame", "dt", "min", "max", "clamp"}; + +static _Noreturn void fail_at(Compiler *c, int line, int column, const char *format, ...) { + va_list args; + va_start(args, format); + c->error->line = line; + c->error->column = column; + vsnprintf(c->error->message, sizeof(c->error->message), format, args); + va_end(args); + longjmp(c->fail, 1); +} + +static void *allocate(Compiler *c, size_t size) { + if (c->block_count == c->block_capacity) { + size_t capacity = c->block_capacity ? 2 * c->block_capacity : 64; + void **blocks = realloc(c->blocks, capacity * sizeof(*blocks)); + if (!blocks) fail_at(c, c->line, c->column, "out of memory"); + c->blocks = blocks; + c->block_capacity = capacity; + } + void *block = calloc(1, size); + if (!block) fail_at(c, c->line, c->column, "out of memory"); + return c->blocks[c->block_count++] = block; +} + +/* ---- Lexer ---------------------------------------------------------- */ + +static char peek_char(const Compiler *c, size_t offset) { + return c->at + offset < c->length ? c->source[c->at + offset] : 0; +} + +static void advance(Compiler *c) { + if (c->source[c->at++] == '\n') { ++c->line; c->column = 1; } else ++c->column; +} + +static bool digit(char ch) { return ch >= '0' && ch <= '9'; } +static bool letter(char ch) { return (ch >= 'a' && ch <= 'z') || (ch >= 'A' && ch <= 'Z') || ch == '_'; } + +static Token scan(Compiler *c) { + for (;;) { + char ch = peek_char(c, 0); + if (ch == ' ' || ch == '\t' || ch == '\r') advance(c); + else if (ch == '#') while (c->at < c->length && peek_char(c, 0) != '\n') advance(c); + else break; + } + Token token = {.line = c->line, .column = c->column}; + if (c->at >= c->length) { token.kind = T_EOF; return token; } + char ch = peek_char(c, 0); + if (ch == '\n') { advance(c); token.kind = T_NL; return token; } + if (digit(ch) || (ch == '.' && digit(peek_char(c, 1)))) { + /* \d+(\.\d*)?|\.\d+ with an optional exponent, as the workbench reads it. */ + char text[400]; + size_t n = 0; + while (digit(peek_char(c, 0)) || peek_char(c, 0) == '.') { + if (peek_char(c, 0) == '.' && memchr(text, '.', n)) break; + if (n + 1 >= sizeof(text)) fail_at(c, token.line, token.column, "number is too long"); + text[n++] = peek_char(c, 0); + advance(c); + } + char e = peek_char(c, 0); + size_t sign = peek_char(c, 1) == '+' || peek_char(c, 1) == '-'; + if ((e == 'e' || e == 'E') && digit(peek_char(c, 1 + sign))) { + for (size_t i = 0; i < 1 + sign; ++i) { text[n++] = peek_char(c, 0); advance(c); } + while (digit(peek_char(c, 0))) { + if (n + 1 >= sizeof(text)) fail_at(c, token.line, token.column, "number is too long"); + text[n++] = peek_char(c, 0); + advance(c); + } + } + text[n] = 0; + token.kind = T_NUMBER; + token.number = strtod(text, NULL); + if (!isfinite(token.number)) fail_at(c, token.line, token.column, "number must be finite"); + return token; + } + if (letter(ch)) { + size_t n = 0; + while (letter(peek_char(c, 0)) || digit(peek_char(c, 0))) { + if (n + 1 >= NAME) fail_at(c, token.line, token.column, "name exceeds 31 bytes"); + token.text[n++] = peek_char(c, 0); + advance(c); + } + token.kind = T_NAME; + return token; + } + static const char *const pairs[] = {"<=", ">=", "==", "!=", "&&", "||"}; + for (size_t i = 0; i < sizeof(pairs) / sizeof(*pairs); ++i) + if (ch == pairs[i][0] && peek_char(c, 1) == pairs[i][1]) { + memcpy(token.text, pairs[i], 3); + advance(c); + advance(c); + token.kind = T_PUNCT; + return token; + } + if (strchr("=+-*/!<>(){},.;", ch)) { + token.text[0] = ch; + advance(c); + token.kind = T_PUNCT; + return token; + } + if ((unsigned char)ch >= 32 && (unsigned char)ch < 127) + fail_at(c, token.line, token.column, "unexpected character '%c'", ch); + fail_at(c, token.line, token.column, "unexpected byte 0x%02x", (unsigned char)ch); +} + +static Token take(Compiler *c) { + Token token = c->token; + if (c->has_ahead) { c->token = c->ahead; c->has_ahead = false; } + else c->token = scan(c); + return token; +} + +static const Token *lookahead(Compiler *c) { + if (!c->has_ahead) { c->ahead = scan(c); c->has_ahead = true; } + return &c->ahead; +} + +static bool is(const Compiler *c, const char *punct) { + return c->token.kind == T_PUNCT && !strcmp(c->token.text, punct); +} + +static bool word(const Compiler *c, const char *name) { + return c->token.kind == T_NAME && !strcmp(c->token.text, name); +} + +static Token expect(Compiler *c, const char *punct, const char *what) { + if (!is(c, punct)) fail_at(c, c->token.line, c->token.column, "expected %s", what); + return take(c); +} + +static Token expect_name(Compiler *c, const char *what) { + if (c->token.kind != T_NAME) fail_at(c, c->token.line, c->token.column, "expected %s", what); + return take(c); +} + +static void newlines(Compiler *c) { while (c->token.kind == T_NL) take(c); } +static void separators(Compiler *c) { while (c->token.kind == T_NL || is(c, ";")) take(c); } + +static void end(Compiler *c) { + if (c->token.kind != T_NL && c->token.kind != T_EOF && !is(c, ";") && !is(c, "}")) + fail_at(c, c->token.line, c->token.column, "expected newline or semicolon"); + separators(c); +} + +static bool reserved_name(const char *name) { + for (size_t i = 0; i < sizeof(reserved) / sizeof(*reserved); ++i) + if (!strcmp(name, reserved[i])) return true; + return false; +} + +/* ---- Parser --------------------------------------------------------- */ + +static const struct { const char *text; int precedence, op; } binary[] = { + {"||", 1, OR}, {"&&", 2, AND}, {"==", 3, EQ}, {"!=", 3, NE}, {"<", 4, LT}, {"<=", 4, LE}, + {">", 4, GT}, {">=", 4, GE}, {"+", 5, ADD}, {"-", 5, SUB}, {"*", 6, MUL}, {"/", 6, DIV}, +}; + +static Expr *node(Compiler *c, ExprKind kind, const Token *at) { + Expr *e = allocate(c, sizeof(*e)); + e->kind = kind; + e->line = at->line; + e->column = at->column; + return e; +} + +static Expr *expression(Compiler *c, int minimum, bool multiline) { + if (++c->depth > DEPTH) fail_at(c, c->token.line, c->token.column, "expression nesting exceeds 64"); + if (multiline) newlines(c); + Token token = take(c); + Expr *left; + if (token.kind == T_NUMBER) { + left = node(c, E_NUMBER, &token); + left->number = token.number; + } else if (token.kind == T_PUNCT && (!strcmp(token.text, "-") || !strcmp(token.text, "!"))) { + left = node(c, E_UNARY, &token); + left->op = token.text[0] == '-' ? NEG : NOT; + left->args[0] = expression(c, 7, multiline); + left->count = 1; + } else if (token.kind == T_PUNCT && !strcmp(token.text, "(")) { + left = expression(c, 0, true); + newlines(c); + expect(c, ")", "')'"); + } else if (token.kind == T_NAME) { + static const struct { const char *name; unsigned count; int op; } calls[] = + {{"min", 2, MIN}, {"max", 2, MAX}, {"clamp", 3, CLAMP}}; + int call = -1; + for (int i = 0; i < 3; ++i) if (!strcmp(token.text, calls[i].name)) call = i; + if (call >= 0 && is(c, "(")) { + take(c); + left = node(c, E_CALL, &token); + left->op = calls[call].op; + newlines(c); + while (!is(c, ")")) { + if (left->count == 3) fail_at(c, token.line, token.column, "%s expects %u arguments", token.text, calls[call].count); + left->args[left->count++] = expression(c, 0, true); + newlines(c); + if (!is(c, ",")) break; + take(c); + newlines(c); + } + expect(c, ")", "')'"); + if (left->count != calls[call].count) + fail_at(c, token.line, token.column, "%s expects %u arguments", token.text, calls[call].count); + } else if (!strcmp(token.text, "dt")) { + left = node(c, E_DT, &token); + } else { + if (reserved_name(token.text)) fail_at(c, token.line, token.column, "expected a number, state, or expression"); + left = node(c, E_STATE, &token); + memcpy(left->name, token.text, NAME); + } + } else fail_at(c, token.line, token.column, "expected a number, state, or expression"); + for (;;) { + if (multiline) newlines(c); + int found = -1; + if (c->token.kind == T_PUNCT) + for (int i = 0; i < 12; ++i) if (!strcmp(c->token.text, binary[i].text)) found = i; + if (found < 0 || binary[found].precedence < minimum) break; + Token op = take(c); + Expr *e = node(c, E_BINARY, &op); + e->op = binary[found].op; + e->args[0] = left; + e->args[1] = expression(c, binary[found].precedence + 1, multiline); + e->count = 2; + left = e; + } + --c->depth; + return left; +} + +static Stmt *block(Compiler *c, unsigned depth); + +static Stmt *statement(Compiler *c, unsigned depth) { + Token token = expect_name(c, "assignment, if, or '}'"); + Stmt *s = allocate(c, sizeof(*s)); + s->line = token.line; + s->column = token.column; + if (!strcmp(token.text, "if")) { + s->branch = true; + s->value = expression(c, 0, false); + s->then = block(c, depth + 1); + /* else may follow on the same line or after newlines */ + bool has_else = word(c, "else"); + if (!has_else && c->token.kind == T_NL) { + const Token *next = lookahead(c); + if (next->kind == T_NAME && !strcmp(next->text, "else")) { take(c); has_else = true; } + } + if (has_else) { + take(c); + if (word(c, "if")) s->otherwise = statement(c, depth + 1); + else s->otherwise = block(c, depth + 1); + } + return s; + } + if (reserved_name(token.text)) fail_at(c, token.line, token.column, "expected assignment, if, or '}'"); + memcpy(s->name, token.text, NAME); + expect(c, "=", "'='"); + s->value = expression(c, 0, false); + return s; +} + +static Stmt *block(Compiler *c, unsigned depth) { + if (depth >= DEPTH) fail_at(c, c->token.line, c->token.column, "block nesting exceeds 64"); + newlines(c); + expect(c, "{", "'{'"); + separators(c); + Stmt *first = NULL, **tail = &first; + while (!is(c, "}")) { + if (c->token.kind == T_EOF) fail_at(c, c->token.line, c->token.column, "expected '}'"); + *tail = statement(c, depth); + tail = &(*tail)->next; + end(c); + } + take(c); + return first; +} + +static const char *declared_kind(const Compiler *c, const char *name) { + for (unsigned i = 0; i < c->states; ++i) if (!strcmp(c->state_names[i], name)) return "state"; + for (unsigned i = 0; i < c->inputs; ++i) if (!strcmp(c->input_names[i], name)) return "input"; + for (unsigned i = 0; i < c->views; ++i) if (!strcmp(c->view_names[i], name)) return "view"; + return NULL; +} + +static void declare(Compiler *c, const Token *name) { + if (reserved_name(name->text)) fail_at(c, name->line, name->column, "'%s' is reserved", name->text); + const char *kind = declared_kind(c, name->text); + if (kind) fail_at(c, name->line, name->column, "'%s' is already declared as a %s", name->text, kind); +} + +static bool key_name(const char *key) { + static const char *const named[] = {"Space", "Enter", "ArrowLeft", "ArrowRight", "ArrowUp", "ArrowDown"}; + for (size_t i = 0; i < sizeof(named) / sizeof(*named); ++i) if (!strcmp(key, named[i])) return true; + if (!strncmp(key, "Key", 3)) return key[3] >= 'A' && key[3] <= 'Z' && !key[4]; + if (!strncmp(key, "Digit", 5)) return key[5] >= '0' && key[5] <= '9' && !key[6]; + return false; +} + +static void declaration(Compiler *c) { + Token kind = expect_name(c, "state, input, view, or on"); + if (!strcmp(kind.text, "state")) { + Token name = expect_name(c, "state name"); + declare(c, &name); + if (c->states == STATES) fail_at(c, name.line, name.column, "at most 32 states are allowed"); + expect(c, "=", "'='"); + double sign = 1; + if (is(c, "-") || is(c, "+")) sign = take(c).text[0] == '-' ? -1 : 1; + if (c->token.kind != T_NUMBER) fail_at(c, c->token.line, c->token.column, "expected a numeric state initializer"); + memcpy(c->state_names[c->states], name.text, NAME); + c->initial[c->states++] = sign * take(c).number; + } else if (!strcmp(kind.text, "input")) { + Token name = expect_name(c, "input name"); + declare(c, &name); + if (c->inputs == INPUTS) fail_at(c, name.line, name.column, "at most 8 inputs are allowed"); + expect(c, "=", "'='"); + if (!word(c, "key")) fail_at(c, c->token.line, c->token.column, "expected 'key'"); + take(c); + Token key = expect_name(c, "key name"); + if (!key_name(key.text)) fail_at(c, key.line, key.column, "unknown key '%s'", key.text); + for (unsigned i = 0; i < c->inputs; ++i) + if (!strcmp(c->input_keys[i], key.text)) + fail_at(c, key.line, key.column, "key %s is already used by input '%s'", key.text, c->input_names[i]); + memcpy(c->input_names[c->inputs], name.text, NAME); + memcpy(c->input_keys[c->inputs++], key.text, NAME); + } else if (!strcmp(kind.text, "view")) { + Token name = expect_name(c, "view name"); + declare(c, &name); + if (c->views == VIEWS) fail_at(c, name.line, name.column, "at most 24 views are allowed"); + expect(c, "=", "'='"); + memcpy(c->view_names[c->views], name.text, NAME); + c->view_value[c->views++] = expression(c, 0, false); + } else if (!strcmp(kind.text, "on")) { + if (c->handler_count == HANDLERS) fail_at(c, kind.line, kind.column, "at most 64 handlers are allowed"); + Token event = expect_name(c, "frame or an input event"); + unsigned index = c->handler_count++; + c->handlers[index].line = event.line; + c->handlers[index].column = event.column; + if (strcmp(event.text, "frame")) { + /* The input may be declared later; it is resolved at generation. */ + expect(c, ".", "'.' and down or up"); + Token phase = expect_name(c, "down or up"); + if (strcmp(phase.text, "down") && strcmp(phase.text, "up")) + fail_at(c, phase.line, phase.column, "expected down or up"); + memcpy(c->handlers[index].input, event.text, NAME); + c->handlers[index].edge = phase.text[0] == 'u'; + } + c->handlers[index].body = block(c, 0); + } else fail_at(c, kind.line, kind.column, "unknown declaration '%s'", kind.text); + end(c); +} + +/* ---- Checks and code generation ------------------------------------- */ + +static void emit(Compiler *c, int line, int column, uint8_t op, uint32_t arg, double value) { + if (c->code_count == CODE) fail_at(c, line, column, "program exceeds 512 instructions"); + c->code[c->code_count].op = op; + c->code[c->code_count].arg = arg; + c->code[c->code_count++].value = value; +} + +static int state_index(const Compiler *c, const char *name) { + for (unsigned i = 0; i < c->states; ++i) if (!strcmp(c->state_names[i], name)) return (int)i; + return -1; +} + +/* Emit an expression; height is the stack height before it. */ +static void value(Compiler *c, const Expr *e, bool handler, int height) { + if (height + 1 > STACK) fail_at(c, e->line, e->column, "expression needs more than 64 stack values"); + switch (e->kind) { + case E_NUMBER: emit(c, e->line, e->column, CONST, 0, e->number); return; + case E_DT: + if (!handler) fail_at(c, e->line, e->column, "dt is only available in handlers"); + emit(c, e->line, e->column, DT, 0, 0); + return; + case E_STATE: { + int index = state_index(c, e->name); + if (index < 0) { + const char *kind = declared_kind(c, e->name); + if (kind) fail_at(c, e->line, e->column, "'%s' is a %s, not a state", e->name, kind); + fail_at(c, e->line, e->column, "unknown state '%s'", e->name); + } + emit(c, e->line, e->column, LOAD, (uint32_t)index, 0); + return; + } + default: + for (unsigned i = 0; i < e->count; ++i) value(c, e->args[i], handler, height + (int)i); + emit(c, e->line, e->column, (uint8_t)e->op, 0, 0); + } +} + +static void statements(Compiler *c, const Stmt *s) { + for (; s; s = s->next) { + if (!s->branch) { + int index = state_index(c, s->name); + if (index < 0) { + const char *kind = declared_kind(c, s->name); + if (kind) fail_at(c, s->line, s->column, "cannot assign to %s '%s'", kind, s->name); + fail_at(c, s->line, s->column, "unknown state '%s'", s->name); + } + value(c, s->value, true, 0); + emit(c, s->line, s->column, STORE, (uint32_t)index, 0); + continue; + } + value(c, s->value, true, 0); + unsigned jump = c->code_count; + emit(c, s->line, s->column, JZ, 0, 0); + statements(c, s->then); + if (s->otherwise) { + unsigned skip = c->code_count; + emit(c, s->line, s->column, JMP, 0, 0); + c->code[jump].arg = c->code_count; + statements(c, s->otherwise); + c->code[skip].arg = c->code_count; + } else c->code[jump].arg = c->code_count; + } +} + +static unsigned char *put32(unsigned char *p, uint32_t v) { + for (unsigned i = 0; i < 4; ++i) p[i] = (unsigned char)(v >> (8 * i)); + return p + 4; +} + +static unsigned char *put_double(unsigned char *p, double value) { + uint64_t bits; + memcpy(&bits, &value, sizeof(bits)); + put32(p, (uint32_t)bits); + return put32(p + 4, (uint32_t)(bits >> 32)); +} + +/* Deduplicated [u8 length][bytes] strings; returns the offset. */ +static uint32_t intern(unsigned char *strings, size_t *used, const char *text) { + size_t length = strlen(text); + for (size_t at = 0; at < *used; at += 1u + strings[at]) + if (strings[at] == length && !memcmp(strings + at + 1, text, length)) return (uint32_t)at; + uint32_t offset = (uint32_t)*used; + strings[(*used)++] = (unsigned char)length; + memcpy(strings + *used, text, length); + *used += length; + return offset; +} + +static void generate(Compiler *c, unsigned char **module, size_t *size) { + uint32_t events[HANDLERS]; + for (unsigned i = 0; i < c->handler_count; ++i) { + const char *name = c->handlers[i].input; + events[i] = 0; /* frame */ + if (!*name) continue; + int input = -1; + for (unsigned k = 0; k < c->inputs; ++k) if (!strcmp(c->input_names[k], name)) input = (int)k; + if (input < 0) { + const char *kind = declared_kind(c, name); + if (kind) fail_at(c, c->handlers[i].line, c->handlers[i].column, "'%s' is a %s, not an input", name, kind); + fail_at(c, c->handlers[i].line, c->handlers[i].column, "unknown input '%s'", name); + } + events[i] = 1u + 2u * (uint32_t)input + (uint32_t)c->handlers[i].edge; + } + /* Views, then handlers, each as one contiguous block. */ + for (unsigned i = 0; i < c->views; ++i) { + c->view_code[i][0] = c->code_count; + value(c, c->view_value[i], false, 0); + c->view_code[i][1] = c->code_count - c->view_code[i][0]; + } + for (unsigned i = 0; i < c->handler_count; ++i) { + c->handler_code[i][0] = c->code_count; + statements(c, c->handlers[i].body); + c->handler_code[i][1] = c->code_count - c->handler_code[i][0]; + } + static unsigned char strings[STATES * NAME + 2 * INPUTS * NAME + VIEWS * NAME]; + size_t used = 0; + uint32_t state_name[STATES], input_name[INPUTS], input_key[INPUTS], view_name[VIEWS]; + for (unsigned i = 0; i < c->states; ++i) state_name[i] = intern(strings, &used, c->state_names[i]); + for (unsigned i = 0; i < c->inputs; ++i) { + input_name[i] = intern(strings, &used, c->input_names[i]); + input_key[i] = intern(strings, &used, c->input_keys[i]); + } + for (unsigned i = 0; i < c->views; ++i) view_name[i] = intern(strings, &used, c->view_names[i]); + *size = 32 + 16u * c->states + 12u * (c->inputs + c->views + c->handler_count) + 16u * c->code_count + used; + unsigned char *p = *module = allocate(c, *size); + memcpy(p, "PUREMOD1", 8); + p[8] = (unsigned char)c->states; p[10] = (unsigned char)c->inputs; + p[12] = (unsigned char)c->views; p[14] = (unsigned char)c->handler_count; + put32(p + 16, c->code_count); + put32(p + 20, (uint32_t)used); + p += 32; + for (unsigned i = 0; i < c->states; ++i) { p = put32(p, state_name[i]); p = put32(p, 0); p = put_double(p, c->initial[i]); } + for (unsigned i = 0; i < c->inputs; ++i) { p = put32(p, input_name[i]); p = put32(p, 1); p = put32(p, input_key[i]); } + for (unsigned i = 0; i < c->views; ++i) { p = put32(p, view_name[i]); p = put32(p, c->view_code[i][0]); p = put32(p, c->view_code[i][1]); } + for (unsigned i = 0; i < c->handler_count; ++i) { + p = put32(p, events[i]); + p = put32(p, c->handler_code[i][0]); + p = put32(p, c->handler_code[i][1]); + } + for (unsigned i = 0; i < c->code_count; ++i) { + *p++ = c->code[i].op; *p++ = 0; *p++ = 0; *p++ = 0; + p = put32(p, c->code[i].arg); + p = put_double(p, c->code[i].value); + } + memcpy(p, strings, used); +} + +bool aup_compile(const char *source, size_t length, unsigned char **module, size_t *size, + AupDiagnostic *error) { + *module = NULL; + *size = 0; + *error = (AupDiagnostic){0}; + Compiler *c = calloc(1, sizeof(*c)); + if (!c) { *error = (AupDiagnostic){1, 1, "out of memory"}; return false; } + c->source = source; + c->length = length; + c->line = c->column = 1; + c->error = error; + unsigned char *result = NULL; + if (!setjmp(c->fail)) { + if (length > AUP_SOURCE_LIMIT) fail_at(c, 1, 1, "source exceeds 32768 bytes"); + c->token = scan(c); + separators(c); + while (c->token.kind != T_EOF) declaration(c); + generate(c, &result, size); + *module = malloc(*size); + if (!*module) fail_at(c, 1, 1, "out of memory"); + memcpy(*module, result, *size); + } else { + free(*module); + *module = NULL; + *size = 0; + } + for (size_t i = 0; i < c->block_count; ++i) free(c->blocks[i]); + free(c->blocks); + free(c); + return *module != NULL; +} diff --git a/src/main.c b/src/main.c new file mode 100644 index 0000000..3731eec --- /dev/null +++ b/src/main.c @@ -0,0 +1,38 @@ +#define _POSIX_C_SOURCE 200809L +#include "aup.h" +#include <stdio.h> +#include <stdlib.h> +#include <string.h> + +static int usage(void) { + fputs("aup check FILE\naup build FILE -o OUT.pmod\n", stderr); + return 2; +} + +int main(int argc, char **argv) { + bool build = argc == 5 && !strcmp(argv[1], "build") && !strcmp(argv[3], "-o"); + if (!build && !(argc == 3 && !strcmp(argv[1], "check"))) return usage(); + const char *path = argv[2]; + FILE *file = fopen(path, "rb"); + if (!file) { perror(path); return 2; } + static char source[AUP_SOURCE_LIMIT + 1]; + size_t length = fread(source, 1, sizeof(source), file); + bool unreadable = ferror(file); + fclose(file); + if (unreadable) { fprintf(stderr, "%s: cannot read\n", path); return 2; } + unsigned char *module; + size_t size; + AupDiagnostic error; + if (!aup_compile(source, length, &module, &size, &error)) { + fprintf(stderr, "%s:%d:%d: %s\n", path, error.line, error.column, error.message); + return 1; + } + int status = 0; + if (build) { + FILE *out = fopen(argv[4], "wb"); + if (!out || fwrite(module, 1, size, out) != size) { perror(argv[4]); status = 2; } + if (out && fclose(out)) { perror(argv[4]); status = 2; } + } + free(module); + return status; +} |
