Newer
Older
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 <stdlib.h>
#include <string.h>
#include <getopt.h>
#include <ctype.h>
#include "mcc/ast.h"
#include "mcc/parser.h"
#include "mcc/symbol_table.h"
#include "mcc/symbol_table_parse.h"
#include "mcc/symbol_table_print.h"
#include "mcc/printer.h"
#include "mcc/mcc_scope.h"
int main(int argc, char *argv[]) {
char* symbol_scope = "program";
FILE* out = stdout;
if (argc < 2) {
mcc_symbol_table_print_usage(argv[0]);
return EXIT_FAILURE;
}
int c;
static struct option options[] = {
{ "help", no_argument, 0, 'h' },
{ "output", required_argument, 0, 'o' },
{ "function", required_argument, 0, 'f' },
{ 0, 0, 0, 0 }
};
while ((c = getopt_long(argc, argv, "hf:o:", options, NULL)) != -1) {
switch(c) {
case 'h':
mcc_symbol_table_print_usage(argv[0]);
return EXIT_SUCCESS;
case 'o':
out = fopen(optarg, "rw");
break;
case 'f':
symbol_scope = optarg;
break;
case '?':
if(isprint(optopt))
fprintf(stderr, "Unknown option `-%c'.\n", optopt);
else
fprintf(stderr,
"Unknown option character `\\x%x'.\n",
optopt);
return 1;
default:
abort();
}
}
// get input src
char *input = NULL;
if (optind < argc) {
input = argv[optind];
} else {
fprintf(stderr, "%s: Missing input!\n", argv[0]);
mcc_symbol_table_print_usage(argv[0]);
return EXIT_FAILURE;
}
FILE *in;
if (strcmp("-", input) == 0) {
in = stdin;
} else {
in = fopen(input, "r");
if (!in) {
perror("fopen");
return EXIT_FAILURE;
}
}
struct mcc_ast_program *prog = NULL;
// parsing phase
{
struct mcc_parser_result result = mcc_parse_file(in);
if (result.status != MCC_PARSER_STATUS_OK) {
// print error message
printf("%s", result.parser_error -> error_msg);
return EXIT_FAILURE;
}
prog = result.program;
}
// create symbol table
struct mcc_symbol_table_error_collector *ec = mcc_symbol_table_new_error_collector();
struct mcc_symbol_table *symbol_table;
// check if scope should be limited
if (strcmp("program", symbol_scope) == 0) {
symbol_table = mcc_symbol_table_build(prog, ec);
} else {
struct mcc_ast_function *func = mcc_limit_scope_to_function(symbol_scope, prog);
symbol_table = mcc_symbol_table_build(func, ec);
}
if (symbol_table == NULL) {
mcc_symbol_table_print_error(ec, out);
return EXIT_FAILURE;
}
mcc_symbol_table_print(symbol_table,out);
mcc_symbol_table_delete_table(symbol_table);
mcc_symbol_table_delete_error_collector(ec);
mcc_ast_delete(prog);
return EXIT_SUCCESS;
}