test.sh

##!/bin/bash

TEST_TARGET=$1

try() {
    expected="$1"
    input="$2"

    echo "$TEST_TARGET --str '$input'"
    $TEST_TARGET --str "$input" >tmp.s
    if [[ "$?" != "0" ]]; then
        echo 'Compilation error'
        exit 1
    fi

    gcc -no-pie -g -o tmp tmp.s ./test/lib.o
    ./tmp
    actual="$?"

    if [ "$actual" = "$expected" ]; then
        echo " -> $actual"
    else
        echo "$expected expected, but got $actual"
        exit 1
    fi
}

try 0   'int main() { 0; }'
try 42  'int main() { 42; }'
try 21  'int main() { 5+20-4; }'
try 41  'int main() {  12 + 34 - 5 ; }'
try 47  'int main() { 5+6*7; }'
try 15  'int main() { 5*(9-6); }'
try 4   'int main() { (3+5)/2; }'
try 4   'int main() { +4; }'
try 0   'int main() { 4 + -4; }'
try 1   'int main() { 10 == 10; }'
try 0   'int main() { 10 != 10; }'
try 1   'int main() { 10 <= 10; }'
try 1   'int main() { 10 >= 10; }'
try 0   'int main() { 10 > 10; }'
try 0   'int main() { 10 < 10; }'
try 1   'int main() { 1 + 2 == 3; }'
try 0   'int main() { 1 + 1 == 3; }'
try 1   'int main() { 3 == 1 + 2; }'
try 0   'int main() { 3 == 1 + 1; }'
try 1   'int main() { 1 + 2 <= 3; }'
try 0   'int main() { 1 + 2 < 3; }'
try 1   'int main() { 3 >= 1 + 2; }'
try 0   'int main() { 3 > 1 + 2; }'
try 1   'int main() { int a; a = 1; a; }'
try 3   'int main() { int a; int b; a = 1; b = 2; a+b; }'
try 1   'int main() { int a; int b; a = 1; b = 2; (a<b); }'
try 1   'int main() { return 1; }'
try 5   'int main() { return 5; return 8; }'
try 6   'int main() { int foo; int bar; foo = 1; bar = 2 + 3; return foo + bar; }'
try 30  'int main() { int x_d; int y_d; x_d = 10; y_d = 20; x_d+y_d; }'
try 3   'int main() { if (1) return 3; }'
try 0   'int main() { int foo; int bar; foo = 1; bar = 2; if (1) return 0; }'
try 1   'int main() { int foo; int bar; foo = 1; bar = 2; if (foo==1) return 1; }'
try 2   'int main() { int foo; int bar; foo = 1; bar = 2; if (foo==bar) return 1; else return 2; }'
try 0   'int main() { while (0) return 1; 0; }'
try 1   'int main() { while (1) return 1; 0; }'
try 3   'int main() { int i; i=0; while (i!=3) i=i+1; i; }'
try 1   'int main() { for (;;) return 1; }'
try 10  'int main() { int i; for (i=0; i<10; i=i+1) 1; return i; }'
try 1   'int main() { int i; for (i=0; i<10; i=i+1) return 1+i; return i; }'
try 45  'int main() { int i; int c; c=0; for (i=0; i<10; i=i+1) c = c+i; return c; }'
try 10  'int main() { int i; int c;c=0; for (i=0; i<10; i=i+1) c = c+1; return c; }'
try 1   'int main() { int c; if (1) {c=1; return c;} }'
try 9   'int main() { int i; for (i = 0; i<10; i=i+1) { if (i == 9) {return i;}} }'
try 9   'int main() { int i; for (i = 0; i<10; i=i+1) { if (i == 9) {return i;}} }'
try 0   'int main() { foo(); }'
try 1   'int main() { int i; for (i = 0; i<3; i=i+1) { foo(); } return 1; }'
try 3   'int main() { int a; int b; a = 1; b = 2; add(a, b); }'
try 0   'int main() { return 0; }'
try 2   'int main() { return bar(1, 1); } int bar(int x, int y) { return x + y;}'
try 0   'int main() { return fib(0); } int fib(int n) { if (n <= 1) { return n; } return fib(n - 2) + fib(n - 1); }'
try 1   'int main() { return fib(1); } int fib(int n) { if (n <= 1) { return n; } return fib(n - 2) + fib(n - 1); }'
try 1   'int main() { return fib(2); } int fib(int n) { if (n <= 1) { return n; } return fib(n - 2) + fib(n - 1); }'
try 2   'int main() { return fib(3); } int fib(int n) { if (n <= 1) { return n; } return fib(n - 2) + fib(n - 1); }'
try 3   'int main() { return fib(4); } int fib(int n) { if (n <= 1) { return n; } return fib(n - 2) + fib(n - 1); }'
try 5   'int main() { return fib(5); } int fib(int n) { if (n <= 1) { return n; } return fib(n - 2) + fib(n - 1); }'
try 3   'int main() { return f(0); } int f(int n) { if (n == 3) { return n; } return f(n+1); }'
try 3   'int main() { int x; x=3; int* y; y=&x; return *y; }'
try 1   'int main() { int* p; alloc4(&p, 1, 2, 4, 8); return *p; }'
try 4   'int main() { int* p; alloc4(&p, 1, 2, 4, 8); int* q; q = p + 2; return *q; }'
try 8   'int main() { int* p; alloc4(&p, 1, 2, 4, 8); int* q; q = p + 2; return *(1+q); }'
try 8   'int main() { int* p; alloc4(&p, 1, 2, 4, 8); int* q; q = p + 2; return *(1+q); }'
try 8   'int main() { int* p; alloc4(&p, 1, 2, 4, 8); int* q; int d; d = 2; q = p + d; return *(1+q); }'
try 0   'int main() { int* p; return (p+1)-(1+p)+(p-1)-(p-1); }'
try 0   'int main() { int* p; int** q; q = &p; return q-&p;}'
try 8   'int main() { int* p; int** q; q = &p; return (q+1)-&p;}'
try 0   'int main() { int* p; int** q; q = &p; return (q+1)-(&p+1);}'
try 4   'int main() { return sizeof(1);}'
try 4   'int main() { return sizeof(sizeof(4));}'
try 4   'int main() { int x; return sizeof(x);}'
try 4   'int main() { int x; return sizeof(x + 3);}'
try 8   'int main() { int* x; return sizeof(x);}'
try 8   'int main() { int* x; return sizeof(x + 1);}'
try 4   'int main() { int* x; return sizeof(*x);}'
try 0   'int main() { int a[10]; return 0;}'
try 1   'int main() { int a[10]; *a=1; return *a;}'
try 1   'int main() { int a[10]; *(a+0)=1; return *a;}'
try 1   'int main() { int a[10]; *a=1; int* p; p=a; return *p;}'
try 2   'int main() { int a[10]; *a=1; *(a+1)=2; int* p; p=a; return *(p+1);}'
try 1   'int main() { int a[10]; a[0]=1; return *a;}'
try 5   'int main() { int a[10]; a[0]=1; a[1]=3; a[2]=1; return a[0]+a[1]+a[2];}'
try 40  'int main() { int a[10]; return sizeof(a);}'
try 4   'int main() { int a[1]; return sizeof(a);}'
try 80  'int main() { int* a[10]; return sizeof(a);}'
try 8   'int main() { int* a[1]; return sizeof(a);}'
try 0   'int main() { int* a[4]; return a-&a[0];}'
try 2   'int x; int main() { x = 1; return x+1;}'
try 4   'int* foo[10]; int main() { foo[0] = 1; foo[9] = 3; return foo[0]+foo[9];}'
try 1   'int main() { char x; x = 1; return x; }'
try 3   'int main() { char x[3]; x[0] = -1; x[1] = 2; int y; y = 4; return x[0] + y; }'
try 97  'int main() { char* x; x = "abc"; return *x; }'
try 97  'int main() { char* x; x = "abc"; return x[0]; }'
try 0   'int main() { printf("hello, world\n"); return 0; }'
try 5   'int main() { int x; set(&x); return x;} int set(int* x) { *x=5; return 0; }'
try 8   'int main() { int x; int y; set(&x, &y); return x+y;} int set(int* x, int* y) { *x=4; *y=4; return 0;}'
try 1   'int main() { int x; x = 0; return ++x; }'
try 2   'int main() { int x; x = 0; if (++x==1) ++x; return x; }'
try 0   'int main() { int x; x = 0; if (++x==1) --x; return x; }'
try 1   'int main() { char x[4]; x[0] = 0; ++x[0]; return x[0]; }'
try 3   'int main() { int x[4]; x[0] = 0; x[1] = 1; ++x[0]; ++x[1]; return x[0]+x[1]; }'
try 2   'int main() { int x[4]; x[0] = 1; x[1] = 1; --x[0]; ++x[1]; return x[0]+x[1]; }'
try 1   'int main() { int x; x = 0; x++; return x; }'
try 2   'int main() { int x; x = 0; if (x++==0) ++x; return x; }'
try 1   'int main() { int x; x = 1; x--; ++x; return x; }'
try 2   'int main() { int x[4]; x[0] = 1; x[1] = 1; x[0]--; x[1]++; return x[0]+x[1]; }'
try 2   'int main() { char x[4]; x[1] = 1; x[1]++; return x[1]; }'
try 2   'int main() { char x[4]; x[0] = 1; x[1] = 1; return x[0] + x[1]; }'
try 2   'int main() { char x[4]; x[0] = 1; x[1] = 1; x[0]--; x[1]++; return x[0]+x[1]; }'
try 1   'int main() { return 1&&1; }'
try 0   'int main() { return 0&&1; }'
try 0   'int main() { return 1&&0; }'
try 0   'int main() { return 0&&0; }'
try 1   'int main() { int a; a = 1; int b; b = 0; if (a==1&&b==0) return 1; else return 0; }'
try 0   'int main() { int a; a = 1; int b; b = 0; if (a==1&&b==1) return 1; else return 0; }'
try 1   'int main() { int a; a = 1; int b; b = 0; if (a==1&&b==0&&1) return 1; else return 0; }'
try 0   'int main() { int a; a = 1; int b; b = 0; if (a==1&&b==0&&1&&0) return 1; else return 0; }'
try 1   'int main() { return 1||1; }'
try 1   'int main() { return 0||1; }'
try 1   'int main() { return 1||0; }'
try 0   'int main() { return 0||0; }'
try 0   'int main() { int const a; a=0; return a; }'
try 0   'int main() { int const* a; a=0; return a; }'
try 0   'int main() { int const* const* a; a=0; return a; }'
try 0   'int main() { static int const a; a=0; return a; }'
try 0   'int main() { static int const* const* a; a=0; return a; }'
try 3   'struct hoge { int x; int y; }; int main() { struct hoge obj; obj.x = 3; return obj.x; }'
try 8   'struct hoge { int x; int y; }; int main() { struct hoge obj; obj.x = 3; return obj.x+5; }'
try 5   'struct hoge { int x; int y; }; int main() { struct hoge obj; obj.x = 3; obj.y=2; return obj.x+obj.y; }'
try 3   'struct hoge { int x; int y; }; int main() { struct hoge obj; obj.x = 3; int* a; a = &obj.x; *a = 3; return obj.x; }'
try 5   'struct hoge { int x; int y; }; int main() { struct hoge obj; obj.x = 3; int* a; a = &obj.x; *a = 3; obj.x = 5; return *a; }'
try 5   'struct hoge { int x; int y; }; int main() { struct hoge obj; struct hoge* ptr; ptr = &obj; ptr->x = 5; return obj.x; }'
try 0   'struct hoge { int x; int y; }; int main() { return foo(); } struct hoge* foo() { return 0; }'
try 8   'struct hoge { int x; int y; }; int main() { return sizeof(struct hoge); }'
try 128 'int main() { return sizeof(void*) * 16; }'
try 0   'int main() { return; }'
try 3   'int main() { int x = 3; return x; }'
try 1   'int main() { for (int i=0; i<10; i=i+1) return 1+i; return 0; }'
try 0   'struct hoge { int x; int y; }; int main() { struct hoge obj; struct hoge* p = &obj; p->x = 10; p->y = 8; if (p->x==p->y) {return 1;} return 0; }'
try 4   'struct hoge { int x; int y; }; int main() { struct hoge obj; obj.x = 3; return ++obj.x; }'
try 4   'struct hoge { int x; int y; }; int main() { struct hoge obj; obj.x = 3; obj.x++; return obj.x; }'
try 6   'struct hoge { int x; int y; }; int main() { struct hoge obj; struct hoge* ptr; ptr = &obj; obj.x = 5; ptr->x++; return obj.x; }'
try 6   'struct hoge { int x; int y; }; int main() { struct hoge obj; struct hoge* ptr; ptr = &obj; obj.x = 5; ++ptr->x; return obj.x; }'
try 5   'struct abc { int* x; }; int main() { struct abc obj; int z = 1; obj.x = &z; obj.x[0] = 5; return z;}'
try 5   'struct abc { int* x; }; int main() { struct abc obj; int z = 1; struct abc* p = &obj.x; obj.x = &z; obj.x[0] = 5; return p->x[0];}'
try 5   'struct abc { int* x; }; int main() { struct abc obj; int y = 5; int z = 0; obj.x = &y; return obj.x[z--];}'
try 8   'struct abc { int x; }; int main() { struct abc abc; abc.x = 8; return abc.x;}'
try 4   'int main() { int x=1; x+=3; return x;}'
try 3   'int main() { int x=6; x-=3; return x;}'
try 18  'int main() { int x=6; x*=3; return x;}'
try 2   'int main() { int x=6; x/=3; return x;}'
try 1   'struct abc { int x; }; int main() { struct abc abc; abc.x = 0; abc.x+=1; return abc.x;}'
try 2   'struct abc { int x; }; int main() { struct abc abc; abc.x = 3; abc.x-=1; return abc.x;}'
try 3   'struct abc { int x; }; int main() { struct abc abc; abc.x = 3; abc.x*=1; return abc.x;}'
try 3   'struct abc { int x; }; int main() { struct abc abc; abc.x = 3; abc.x/=1; return abc.x;}'
try 2   'int main() { int x=6; x/=3; return (char)x;}'
try 2   'int main() { int x=6; x/=3; return (int)x;}'
try 2   'int main() { int x=6; x/=3; return (void*)x;}'
try 5   'struct abc { int* x; }; int main() { struct abc obj; char y = 5; obj.x = (int*)&y; return y;}'
try 5   'struct hoge { int x; }; struct piyo { int y; struct hoge h; }; int main() { struct piyo a; a.h.x = 5; return a.h.x; }'
try 5   'struct hoge { int x; }; struct piyo { int y; struct hoge* h; }; int main() { struct hoge hoge; struct piyo a; a.h = &hoge; a.h->x = 5; return hoge.x; }'
try 1   'int main() { while (1) { break; return 10; } return 1; }'
try 10  'int main() { int i=0; while (1) { if (i==10) break; ++i; } return i; }'
try 9   'int main() { int i; for (i = 0; i<10; i=i+1) { if (i == 9) {break;}} return i; }'
try 1   'struct hoge { struct hoge* x; }; int main() {return 1;}'
try 5   'struct hoge { struct hoge* x; int y; }; int main() { struct hoge a; struct hoge b; b.y = 5; a.x = &b; return a.x->y;}'
try 0   'int main() {return !1;}'
try 1   'int main() {return !0;}'
try 0   'int main() {return !32;}'
try 0   'enum {HOGE, FUGA=5, PIYO}; int main() {return HOGE;}'
try 5   'enum {HOGE, FUGA=5, PIYO}; int main() {return FUGA;}'
try 6   'enum {HOGE, FUGA=5, PIYO}; int main() {return PIYO;}'
try 3   'struct hoge { int x; int y; }; typedef struct hoge Hoge; int main() { Hoge obj; obj.x = 3; return obj.x; }'
try 5   'struct hoge { int x; }; typedef struct hoge Hoge; struct piyo { int y; Hoge* h; }; int main() { Hoge hoge; struct piyo a; a.h = &hoge; a.h->x = 5; return hoge.x; }'
try 0   'int main() { size_t i = 0; return i;}'
try 0   'void main(void) { size_t i = 0; return i;}'
try 65  "int main(void) { return 'A';}"
try 0   "int main(void) { return '\0';}"
try 48  "int main(void) { return '0';}"
try 0   "struct hoge; int main(void) { struct hoge* h; return 0;}"
try 0   "int foo(void); int main(void) { return 0;}"
try 1   "extern int outer; int main(void) { outer = 1; return outer;}"
try 0   "int main(void) { while (1) { int i = 0; break; } return 0;}"
try 1   "int main(void) { int i = 1 != 0 && 1 < 10;  return i;}"
try 1   "int main(void* a) { int i = 1 != 0 && 1 < 10;  return i;}"
try 5   'struct hoge { int x; }; int main() { struct hoge** ar; struct hoge* ptr; struct hoge h; h.x = 5; ptr = &h; ar = &ptr; return ar[0]->x; }'
try 0   "void* foo(void) { return 1229801703532086340; } int main(void) { return foo() - 1229801703532086340;}"
try 0   "void* foo(void) { return 1229801703532086340; } int main(void) { size_t p; p = foo(); return p - 1229801703532086340;}"
try 0   "void* foo(size_t n) { return n - 1229801703532086340; } int main(void) { return foo(1229801703532086340);}"
try 0   "int main(void) { void** data = 0; return data + 0;}"
try 8   "int main(void) { void** data = 0; size_t i = 1; return data + i++;}"
try 1   "int main(void) { int i = 1 != 0 &&  1 < 10; return i; }"
try 10  "int main(void) { char* q = 10576963; if (!q) {return 0;} return 10; }"