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
|
#include <ducky.h>
#include <platform.h>
#define COMPILE 1
#define INTERP 2
#define EXECUTE 3
#define TRANSCOMPILE 4
#define EVERYTHING 5
char *progname;
void arg_error(void)
{
printf("Usage: %s [-aceit] FILE\n", progname);
printf(" -a: compile to machine code\n");
printf(" -c: compile to bytecode\n");
printf(" -e: execute bytecode (not needed)\n");
printf(" -i: interpret directly (default action)\n");
printf(" -t: transcompile to C\n");
printf("Default action is to interpret or execute bytecode.\n");
exit(EXIT_FAILURE);
}
int main(int argc, char *argv[])
{
int action = INTERP;
progname = argv[0];
char *file = NULL;
if(argc >= 2)
{
for(int i = 1; i < argc; ++i)
{
if(!strcmp(argv[i], "-a"))
action = EVERYTHING;
else if(!strcmp(argv[i], "-c"))
action = COMPILE;
else if(!strcmp(argv[i], "-i"))
action = INTERP;
else if(!strcmp(argv[i], "-e"))
action = EXECUTE;
else if(!strcmp(argv[i], "-t"))
action = TRANSCOMPILE;
else
file = argv[i];
}
if(file)
{
int fd = open(file, O_RDONLY), out_fd;
unsigned char header[4];
read(fd, &header, sizeof(header));
if(*((uint32_t*)header) == DUCKY_MAGIC)
{
if(action != TRANSCOMPILE)
action = EXECUTE;
}
lseek(fd, 0, SEEK_SET);
switch(action)
{
case INTERP:
if(ducky_interp(fd, false))
{
printf("Interpreter error.\n");
return 1;
}
break;
case COMPILE:
out_fd = open("a.out", O_WRONLY | O_CREAT | O_TRUNC, 0644);
if(ducky_compile(fd, false, out_fd))
{
printf("Compiler error.\n");
return 1;
}
break;
case EXECUTE:
if(ducky_vm(fd))
{
printf("Bytecode interpreter error.\n");
return 1;
}
break;
case TRANSCOMPILE:
out_fd = open("a.c", O_WRONLY | O_CREAT | O_TRUNC, 0644);
if(ducky_to_c(fd, out_fd))
{
printf("Transcompiler error.\n");
return 1;
}
break;
case EVERYTHING:
{
char bytecode[L_tmpnam];
tmpnam(bytecode);
out_fd = open(bytecode, O_WRONLY | O_CREAT | O_TRUNC, 0644);
if(ducky_compile(fd, false, out_fd))
{
printf("Compiler error.\n");
return 1;
}
close(fd);
close(out_fd);
char c_code[L_tmpnam];
tmpnam(c_code);
fd = open(bytecode, O_RDONLY);
out_fd = open(c_code, O_WRONLY | O_CREAT | O_TRUNC, 0644);
if(ducky_to_c(fd, out_fd))
{
printf("Transcompiler error.\n");
return 1;
}
close(fd);
close(out_fd);
char cmd[256];
snprintf(cmd, sizeof(cmd), "cc -O3 -lm -x c %s", c_code);
system(cmd);
}
default:
break;
}
}
}
else
{
arg_error();
}
return 0;
}
|
