forked from kspalaiologos/asm2ws
-
Notifications
You must be signed in to change notification settings - Fork 0
/
wsi.c
144 lines (127 loc) · 4.52 KB
/
wsi.c
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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
#include <assert.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdint.h>
#include "whitespace.h"
#include "common.h"
static void error(char * message) {
fprintf(stderr, "wsi: %s\n", message);
exit(1);
}
static void warn(char * message) {
fprintf(stderr, "wsi: warn: %s\n", message);
}
int main(int argc, char * argv[]) {
int i, dis = 0, count = 0, aot = 0, jit = 0, masm = 0, optimize = 0;
for(i = 1; i < argc; i++) {
char * arg = argv[i];
if(*arg == '-') {
if(!strcmp(arg, "--help") || arg[1] == 'h') {
fprintf(stderr,
"wsi: a whitespace omnitool.\n"
"copyright (c) by Kamila Szewczyk.\n"
"licensed under the terms of the GPL license.\n\n"
"usage:\n"
" wsi [...] program.ws"
" -h/--help: display this screen\n"
" -d/--disassemble: disassemble the whitespace program.\n"
" -c/--cycles: count operations made by the program.\n"
" -a/--aot: build a C source file.\n"
#ifdef JIT
" -j/--jit: enable the JIT compiler.\n"
#endif
" -m/--masm: run the macro assembler.\n"
" -Os: optimize for size.\n"
" -Of: optimize to produce fast code.\n"
"default operation: run whitespace code.\n"
);
return 1;
} else if(!strcmp(arg, "--disassemble") || arg[1] == 'd') {
dis = 1;
} else if(!strcmp(arg, "--cycles") || arg[1] == 'c') {
count = 1;
} else if(!strcmp(arg, "--masm") || arg[1] == 'm') {
masm = 1;
} else if(!strcmp(arg, "-Os")) {
optimize = 1;
} else if(!strcmp(arg, "-Of")) {
optimize = 2;
} else if(!strcmp(arg, "--aot") || arg[1] == 'a') {
aot = 1;
#ifdef JIT
} else if(!strcmp(arg, "--jit") || arg[1] == 'j') {
jit = 1;
#endif
} else {
fprintf(stderr, "wsi: unknown switch: '%s'\n", arg);
return 1;
}
} else {
if(dis && count) {
fprintf(stderr, "wsi: disassembling and cycle counting is mutually exclusive.\n");
return 1;
}
if(aot && (dis || count)) {
fprintf(stderr, "wsi: disassembling or cycle counting is mutually exclusive with ahead-of-time compilation.\n");
return 1;
}
#ifdef JIT
if(jit && (aot || dis || count)) {
fprintf(stderr, "wsi: --jit is exclusive with every other f-flag.");
return 1;
}
#endif
if(masm && (jit || aot || dis || count)) {
fprintf(stderr, "wsi: --masm is exclusive with every other f-flag.");
return 1;
}
FILE * input = fopen(arg, "rb");
if(!input) {
perror("wsi: fopen");
return 1;
}
if(masm) {
asm_file(input, optimize);
fclose(input);
return 0;
}
struct parse_result_t data = parse(input, error, warn);
fclose(input);
if(dis) {
disasm(stdout, data.program);
vector_free(data.program);
vector_free(data.labels);
return 0;
}
if(aot) {
char * buf = compile(data);
printf("%s", buf);
vector_free(data.program);
vector_free(data.labels);
free(buf);
return 0;
}
#ifdef JIT
if(jit) {
run_jit(data, error);
vector_free(data.program);
vector_free(data.labels);
return 0;
}
#endif
struct state s = { NULL, NULL, NULL };
int32_t cycles = run(data, &s, error);
if(count) {
fprintf(stderr, "finished, took %d cycles.\n", cycles);
}
vector_free(data.program);
vector_free(data.labels);
vector_free(s.callstack);
vector_free(s.heap);
vector_free(s.stack);
return 0;
}
}
error("invalid invocation. try -h.");
}