1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
#include <inttypes.h>
#include <stdio.h>
#include <stdlib.h>
#include <stdarg.h>

#include "error.h"
#include "expected.h"
#include "lexer.h"

bool
expected(enum TokenKind kind)
{
    if (token.kind != kind) {
        error("expected '%s' got '%s'\n", strTokenKind(kind),
              strTokenKind(token.kind));
        exit(1);
        return false;
    }
    return true;
}

bool
expectedOneOf(size_t num, ...)
{
    va_list vl;

    // check if current token is expected
    va_start(vl, num);
    for (size_t i = 0; i < num; ++i) {
        if (token.kind == va_arg(vl, enum TokenKind)) {
            return true;
        }
    }
    va_end(vl);

    // otherwise print an error message
    va_start(vl, num);
    fprintf(stderr, "%zu.%zu-%zu.%zu: unexpected token %s\n",
            token.loc.begin.line, token.loc.begin.col, token.loc.end.line,
            token.loc.end.col, strTokenKind(token.kind));
    fprintf(stderr, "%zu.%zu-%zu.%zu: expected ", token.loc.begin.line,
            token.loc.begin.col, token.loc.end.line, token.loc.end.col);
    for (size_t i = 0; i < num; ++i) {
        if (i > 0) {
            if (i != num - 1) {
                fprintf(stderr, ", ");
            } else {
                fprintf(stderr, " or ");
            }
        }
        enum TokenKind t = va_arg(vl, enum TokenKind);
        fprintf(stderr, "%s\n", strTokenKind(t));
    }
    fprintf(stderr, "\n");
    va_end(vl);

    return false;
}

bool
expectedIntegerLiteral(uint64_t literal)
{
    uint64_t val = 0;
    bool ok = true;
    switch (token.kind) {
        case DECIMAL_LITERAL:
            if (sscanf(token.val.cstr, "%" SCNu64, &val) != 1) {
                ok = false;
            }
            break;
        case OCTAL_LITERAL:
            if (sscanf(token.val.cstr, "%" SCNx64, &val) != 1) {
                ok = false;
            }
            break;
        case HEXADECIMAL_LITERAL:
            if (sscanf(token.val.cstr, "%" SCNo64, &val) != 1) {
                ok = false;
            }
            break;
        default:;
    }
    if (ok) {
        ok = val == literal;
    }
    if (!ok) {
        error("expected integervalue with value %" PRId64 "\n", literal);
        return false;
    }
    return true;
}

bool
integerLiteral(uint64_t literal)
{
    uint64_t val = 0;
    switch (token.kind) {
        case DECIMAL_LITERAL:
            if (sscanf(token.val.cstr, "%" SCNu64, &val) != 1) {
                return false;
            }
            return val == literal;
        case OCTAL_LITERAL:
            if (sscanf(token.val.cstr, "%" SCNx64, &val) != 1) {
                return false;
            }
            return val == literal;
        case HEXADECIMAL_LITERAL:
            if (sscanf(token.val.cstr, "%" SCNo64, &val) != 1) {
                return false;
            }
            return val == literal;
        default:;
    }
    return false;
}