test.bash

#!/usr/bin/env bash
BUILD_TYPE=${BUILD_TYPE:-debug}

if [ ! -f "$MOCC" ]; then
    echo "type 'make BUILD_TYPE=\"$BUILD_TYPE\"' before run test"
    exit 1
fi

dir="$(cd "$(dirname "$0")" || exit 1; pwd)"

try() {
    local test_name=$1
    local input=$2
    local expected=$3

    local c="$dir/tmp/$test_name.c"
    local asm="$dir/tmp/$test_name.s"
    local bin="$dir/tmp/$test_name"

    local exit_code

    echo "$MOCC $test_name.c"

    echo -n "$input" > "$c"

    "$MOCC" "$c" "$asm"
    exit_code="$?"
    if [ "$exit_code" -ne 0 ]; then
        echo "$test_name: compilation failed with exit code $exit_code"
        exit 1
    fi

    gcc "$asm" -o "$bin"
    exit_code="$?"
    if [ "$exit_code" -ne 0 ]; then
        echo "$test_name: assemble failed"
        exit 1
    fi

    "$bin"
    exit_code="$?"
    if [ "$exit_code" -ne "$expected" ]; then
        echo "$test_name: expected $expected, actual $exit_code"
        exit 1
    fi

    rm "$c" "$asm" "$bin"
}

try "c$LINENO" 'int main(void) { return 0; }' 0
try "c$LINENO" 'int main(void) { return 42; }' 42
try "c$LINENO" "int main(void) { return 'A'; }" 65
try "c$LINENO" "int main(void) { return '\0'; }" 0
try "c$LINENO" "int main(void) { return '\n'; }" 10

try "c$LINENO" '
    int main(void) {
        int a;
        a = 2;
        return a;
    }' 2

try "c$LINENO" '
    int main(void) {
        int a = 3;
        int b = a;
        return b;
    }' 3

try "c$LINENO" '
    int main(void) {
        int a = 5 + 2;
        return a + a - 6 - 3;
    }' 5

try "c$LINENO" 'int main(void) { return 17 / 5; }' 3
try "c$LINENO" 'int main(void) { return 17 / 12; }' 1
try "c$LINENO" 'int main(void) { return 17 % 5; }' 2
try "c$LINENO" 'int main(void) { return 17 % 12; }' 5
try "c$LINENO" 'int main(void) { return 12 & 5; }' 4
try "c$LINENO" 'int main(void) { return 15 & 5; }' 5

try "c$LINENO" '
    int main(void) {
        int a;
        *&a = 5;
        return *&a;
    }' 5

try "c$LINENO" '
    int main(void) {
        int a = 5;
        int *p = &a;
        return *p;
    }' 5

try "c$LINENO" '
    int main(void) {
        int a;
        int *p = &a;
        *p = 10;
        return a;
    }' 10

try "c$LINENO" '
    int f(void) { return 42; }
    int main(void) { return f(); }
    ' 42

try "c$LINENO" '
    int f(void) { return 10; }
    int g(void) { return f() + 2; }
    int main(void) { return g(); }
    ' 12

try "c$LINENO" '
    int f(int x) { return x + x; }
    int main(void) { return f(4); }
    ' 8

try "c$LINENO" '
    int f(int x, int y) { return x + y; }
    int g(int x, int y, int z) { return x + y + z; }
    int main(void) { return f(4, 6) - g(2, 3, 5); }
    ' 0

try "c$LINENO" '
    int f(int x) {
        if (x) return x;
        else return 10;
    }
    int main(void) { return f(0) + f(8); }
    ' 18

try "c$LINENO" '
    int f(int x) {
        if (x - 0)
            if (x - 1) return f(x - 1) + f(x - 2);
            else return 1;
        return 0;
    }
    int main(void) { return f(9); }
    ' 34

try "c$LINENO" '
    int main(void) {
        int f = 1, i = 4;
        while (i) { f = f * i; i = i - 1; }
        return f;
    }
    ' 24

try "c$LINENO" '
    int main(void) {
        int sum = 0, i;
        for (i=0; i<10; i=i+1) sum=sum+i;
        return sum;
    }
    ' 45

try "c$LINENO" '
    int main(void) {
        int sum = 0;
        for (int i=0; i<=10; i=i+1) sum=sum+i;
        return sum;
    }
    ' 55

try "c$LINENO" '
    int main(void) {
        int a = 5, *p = &a;
        return p[0];
    }' 5

try "c$LINENO" '
    int main(void) {
        int a, *p = &a;
        p[0] = 4;
        return a;
    }' 4

try "c$LINENO" '
    int main(void) {
        int a[1];
        a[0] = 4;
        return a[0];
    }' 4

try "c$LINENO" '
    int main(void) {
        int a[3];
        a[0] = 4;
        a[1] = 7;
        a[2] = 11;
        return a[2] + a[0];
    }' 15

try "c$LINENO" '
    int main(void) {
        int a[2][3];
        a[0][0] = 3; a[0][1] = 4; a[0][2] = 5;
        a[1][0] = 6; a[1][1] = 7; a[1][2] = 8;
        return *a[0] + *a[1];
    }' 9

try "c$LINENO" '
    int main(void) {
        char a = 257;
        return a;
    }' 1

try "c$LINENO" '
    char f(char c) { return c; }
    int main(void) {
        char a = f(258);
        return a;
    }' 2

try "c$LINENO" '
    int f(char c) { return c + 1; }
    int main(void) {
        char a = f(258);
        return a;
    }' 3

try "c$LINENO" 'int main(void) { return sizeof 0; }' 4
try "c$LINENO" 'int main(void) { return sizeof(char); }' 1
try "c$LINENO" 'int main(void) { return sizeof(int); }' 4
try "c$LINENO" 'int main(void) { return sizeof(void *); }' 8

try "c$LINENO" 'int main(void) { (void)0; return 0; }' 0
try "c$LINENO" 'int main(void) { return (char)257; }' 1
try "c$LINENO" 'int main(void) { (void *)0; return 0; }' 0
try "c$LINENO" 'int main(void) { (const void *)0; return 0; }' 0
try "c$LINENO" 'int main(void) { return !0 == 1 && !1 == 0 && !2 == 0; }' 1

try "c$LINENO" '
    int;
    int main(void) { return 0; }
    ' 0

try "c$LINENO" '
    int a, b;
    int main(void) { a = 1; b = 2; return a + b; }
    ' 3

try "c$LINENO" 'int main(void) { return 1 && 1; }' 1
try "c$LINENO" 'int main(void) { return 0 && 1; }' 0
try "c$LINENO" 'int main(void) { return 1 && 0; }' 0
try "c$LINENO" 'int main(void) { return 0 && 0; }' 0

try "c$LINENO" 'int main(void) { return 1 || 1; }' 1
try "c$LINENO" 'int main(void) { return 0 || 1; }' 1
try "c$LINENO" 'int main(void) { return 1 || 0; }' 1
try "c$LINENO" 'int main(void) { return 0 || 0; }' 0

try "c$LINENO" '
    int f(void) { return 1; }
    int main(void) {
        return 1 && f();
    }
    ' 1

try "c$LINENO" '
    void abort(void);
    int g(void) { abort(); }
    int main(void) {
        return 0 && g();
    }
    ' 0

try "c$LINENO" '
    int f(void) { return 1; }
    int main(void) {
        return 1 || f();
    }
    ' 1

try "c$LINENO" '
    void abort(void);
    int g(void) { abort(); }
    int main(void) {
        return 1 || g();
    }
    ' 1

try "c$LINENO" 'int main(void) { return 2 == 2; }' 1
try "c$LINENO" 'int main(void) { return 2 == 1; }' 0
try "c$LINENO" 'int main(void) { return 2 != 2; }' 0
try "c$LINENO" 'int main(void) { return 2 != 1; }' 1
try "c$LINENO" 'int main(void) { return +2 == 2; }' 1
try "c$LINENO" 'int main(void) { return -2 == 0-2; }' 1
try "c$LINENO" 'int main(void) { return - -2 == 2; }' 1

try "c$LINENO" '
    int f(int x, int y);
    int main(void) { return f(10, 20); }
    int f(int a, int b) { return a + b; }
    ' 30

try "c$LINENO" '
    int f(int, int);
    int main(void) { return f(10, 20); }
    int f(int a, int b) { return a + b; }
    ' 30

try "c$LINENO" '
    typedef int size_t;
    size_t strlen(const char *s);
    int main(void) { return strlen("hello\n"); }
    ' 6

try "c$LINENO" '
    typedef int size_t;
    size_t strlen(const char *s);
    int main(void) { return strlen(&"hello\n"[3]); }
    ' 3

try "c$LINENO" '
    int strcmp(const char *a, const char *b);
    int main(void) { return strcmp("world", &"hello, world"[7]); }
    ' 0

try "c$LINENO" '
    struct S { int x; int y, z; };
    int main(void) { struct S s; s.x=2; s.y=4; s.z=5; return s.x + s.y + s.z; }
    ' 11

try "c$LINENO" '
    int main(void) {
        struct S { int x, y; } s, *p = &s;
        s.x=2; p->y=7;
        return s.y - p->x;
    }
    ' 5

try "c$LINENO" '
    typedef int T;
    T main(void) { T x = 12; return x; }
    ' 12

try "c$LINENO" '
    typedef struct S S;
    struct S { int x; };
    struct S2 { int x; };
    int main(void) { S s; s.x = 13; return s.x; }
    ' 13

try "c$LINENO" '
    typedef enum E { A, B, C = 5, D, } E;
    int main(void) {
        enum E e1 = D;
        E e2 = e1;
        return A == 0 && B == 1 && C == 5 && D == 6 &&
            e1 == D && e2 == D;
    }
    ' 1

try "c$LINENO" '
    int strcmp(const char *a, const char *b);
    int snprintf(char *buffer, int n, const char *format, ...);
    int main(void) {
        char buffer[256];
        snprintf(buffer, 256, "%s, %s! %d", "Hello", "world", 42);
        return strcmp(buffer, "Hello, world! 42");
    }
    ' 0