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
117
|
#include <stdio.h>
#include <string.h>
#include "parser/parser.h"
#include "codegen/nasm.h"
#include "tests.h"
static void test_generate_consts_and_data(struct TestContext* context)
{
struct Lexer lexer = create_lexer("const N = 5\ndata msg = \"hi\"\n");
struct Program program;
check(context, parse_program(&lexer, &program));
FILE* out = tmpfile();
generate_nasm(&program, out);
fflush(out);
rewind(out);
char buffer[1024];
size_t read = fread(buffer, 1, sizeof(buffer) - 1, out);
buffer[read] = '\0';
fclose(out);
check(context, strstr(buffer, "N equ 5") != NULL);
check(context, strstr(buffer, "section .data") != NULL);
check(context, strstr(buffer, "msg: db `hi`") != NULL);
check(context, strstr(buffer, ".len equ $ - msg") != NULL);
check(context, strstr(buffer, "section .text") != NULL);
check(context, strstr(buffer, "global _start") != NULL);
free_program(&program);
}
static void test_generate_text(struct TestContext* context)
{
struct Lexer lexer = create_lexer(
"proc main\n{\nrax = SYS_WRITE\nrsi = message\nrdx = message.len\n^rdi = rax\nloop:\nsyscall\ngoto loop\n}\n");
struct Program program;
check(context, parse_program(&lexer, &program));
FILE* out = tmpfile();
generate_nasm(&program, out);
fflush(out);
rewind(out);
char buffer[1024];
size_t read = fread(buffer, 1, sizeof(buffer) - 1, out);
buffer[read] = '\0';
fclose(out);
check(context, strstr(buffer, "_start:") != NULL);
check(context, strstr(buffer, "mov rax, SYS_WRITE") != NULL);
check(context, strstr(buffer, "mov rsi, message") != NULL);
check(context, strstr(buffer, "mov rdx, message.len") != NULL);
check(context, strstr(buffer, "mov [rdi], rax") != NULL);
check(context, strstr(buffer, "loop:") != NULL);
check(context, strstr(buffer, "syscall") != NULL);
check(context, strstr(buffer, "jmp loop") != NULL);
free_program(&program);
}
static void test_generate_if(struct TestContext* context)
{
struct Lexer lexer = create_lexer("proc main\n{\nif rcx != 0\ngoto loop\n}\n");
struct Program program;
check(context, parse_program(&lexer, &program));
FILE* out = tmpfile();
generate_nasm(&program, out);
fflush(out);
rewind(out);
char buffer[1024];
size_t read = fread(buffer, 1, sizeof(buffer) - 1, out);
buffer[read] = '\0';
fclose(out);
check(context, strstr(buffer, "cmp rcx, 0") != NULL);
check(context, strstr(buffer, "je .if_end_0") != NULL);
check(context, strstr(buffer, "jmp loop") != NULL);
check(context, strstr(buffer, ".if_end_0:") != NULL);
free_program(&program);
}
static void test_generate_call(struct TestContext* context)
{
struct Lexer lexer = create_lexer(
"proc print_number(value: rdi)\n{\nsyscall\n}\nproc main\n{\nprint_number(r12)\n}\n");
struct Program program;
check(context, parse_program(&lexer, &program));
FILE* out = tmpfile();
generate_nasm(&program, out);
fflush(out);
rewind(out);
char buffer[1024];
size_t read = fread(buffer, 1, sizeof(buffer) - 1, out);
buffer[read] = '\0';
fclose(out);
check(context, strstr(buffer, "mov rdi, r12") != NULL);
check(context, strstr(buffer, "call print_number") != NULL);
check(context, strstr(buffer, "\tret\n") != NULL);
free_program(&program);
}
void run_codegen_tests(struct TestContext* context)
{
test_generate_consts_and_data(context);
test_generate_text(context);
test_generate_if(context);
test_generate_call(context);
}
|
