-
Notifications
You must be signed in to change notification settings - Fork 1
/
toy-jit.lua
93 lines (85 loc) · 1.88 KB
/
toy-jit.lua
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
local string = require('string')
local JUMP_IF_A = 1
local MOV_A_R = 2
local MOV_R_A = 3
local ADD_R_TO_A = 4
local DECR_A = 5
local RETURN_A = 6
local function jit_interpret(bytecode, a)
f_str = [[
function _jit(a)
local reg = {
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
}
]]
local pc = 1
while pc < #bytecode + 1 do
local opcode = bytecode[pc]
f_str = f_str .. string.format('::op_%d::\n', pc)
pc = pc + 1
if opcode == JUMP_IF_A then
local target = bytecode[pc]
pc = pc + 1
f_str = f_str .. string.format([[
if a == 0 then
goto op_%d
end
goto op_%d
]], pc, target)
elseif opcode == MOV_R_A then
local n = bytecode[pc]
pc = pc + 1
f_str = f_str .. string.format([[
a = reg[%d]
]], n)
elseif opcode == MOV_A_R then
local n = bytecode[pc]
pc = pc + 1
f_str = f_str .. string.format([[
reg[%d] = a
]], n)
elseif opcode == ADD_R_TO_A then
local n = bytecode[pc]
pc = pc + 1
f_str = f_str .. string.format([[
a = a + reg[%d]
]], n)
elseif opcode == DECR_A then
f_str = f_str .. string.format([[
a = a - 1
]])
elseif opcode == RETURN_A then
f_str = f_str .. string.format([[
return a
]])
end
end
f_str = f_str .. 'end\n'
loadstring(f_str)()
return _jit(a)
end
-- __________ Entry point __________
local function entry_point(argv)
-- the program we want to interpret
-- it computes the square of its argument
bytecode = {
MOV_A_R, 1, -- i = a
MOV_A_R, 2, -- copy of 'a'
-- 5:
MOV_R_A, 1, -- i--
DECR_A,
MOV_A_R, 1,
MOV_R_A, 3, -- res += a
ADD_R_TO_A, 2,
MOV_A_R, 3,
MOV_R_A, 1, -- if i!=0: goto 5
JUMP_IF_A, 5,
MOV_R_A, 3,
RETURN_A
}
-- result = interpret(bytecode, tonumber(argv[1]))
result = jit_interpret(bytecode, tonumber(argv[1]))
print(result)
end
-- main function
entry_point(arg)