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
118
119
120
121
122
123
124
|
/* obc.c
* Created: 2024-12-03
* Author: Aryadev Chavali
* Description: Entrypoint of compiler
*/
#include <stdio.h>
#include <string.h>
#include "./assembler.h"
#include "./lib.h"
#include "./parser.h"
void usage(const char *name, FILE *fp)
{
fprintf(
fp,
"Usage: %s [-s ASM-NAME | -c OBJ-NAME | -o EXEC-NAME] FILE\n"
"\tGiven BrainFuck source code FILE, compile it.\n"
"\tOptions:\n"
"\t\t-s ASM-NAME: Transpile to assembly, store in ASM-NAME\n"
"\t\t-c OBJ-NAME: Transpile and assemble, store in OBJ-NAME\n"
"\t\t-o EXEC-NAME: Transpile, assemble and link, store in EXEC-NAME\n",
name);
}
struct Config
{
char *prog_name, *asm_name, *obj_name, *exec_name;
};
int parse_config(struct Config *config, int argc, char *argv[])
{
for (int i = 1; i < argc; ++i)
{
size_t size = strlen(argv[i]);
if (argv[i][0] == '-')
{
if (size > 2 || !strspn(argv[i] + 1, "cso") || i == argc - 1)
return 0;
switch (argv[i][1])
{
case 's':
config->asm_name = argv[i + 1];
++i;
break;
case 'c':
config->obj_name = argv[i + 1];
++i;
break;
case 'o':
config->exec_name = argv[i + 1];
++i;
break;
}
}
else
config->prog_name = argv[i];
}
if (!config->prog_name)
return 0;
return 1;
}
int main(int argc, char *argv[])
{
if (argc == 1)
{
usage(argv[0], stderr);
return 1;
}
char *file_data = NULL;
buffer_t *buffer = NULL;
vec_t asm_buffer = {0};
struct PResult res = {0};
struct Config config = {0};
int ret = parse_config(&config, argc, argv);
if (!ret)
{
usage(argv[0], stderr);
goto end;
}
if (!config.asm_name)
config.asm_name = "a.asm";
if (!config.obj_name)
config.obj_name = "a.o";
if (!config.exec_name)
config.exec_name = "a.out";
ret = 0;
FILE *handle = fopen(config.prog_name, "r");
if (!handle)
{
fprintf(stderr, "ERROR: Could not open \"%s\"\n", config.prog_name);
ret = 1;
goto end;
}
file_data = fread_all(handle);
fclose(handle);
buffer = buffer_init_str(config.prog_name, file_data, strlen(file_data));
res = parse_buffer(buffer);
if (res.nodes == NULL)
{
fputs("[WARNING]: Empty source file.", stderr);
ret = 1;
goto end;
}
asm_translate_nodes(&asm_buffer, res, config.prog_name);
asm_compile(&asm_buffer, config.asm_name, config.obj_name, config.exec_name);
end:
vec_free(&asm_buffer);
if (res.nodes)
free(res.nodes);
if (buffer)
free(buffer);
if (file_data)
free(file_data);
return ret;
}
|
