glua-function-reconstructor/function_dump.lua at master · moderndx/glua-function-reconstructor · GitHub

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
local function dump_function_params(func_call)
  local param_location = 2
  local func_param = debug.getlocal( func_call, 1 )
  local func_params = {}
  while func_param ~= nil do
    table.insert(func_params, func_param)
    func_param = debug.getlocal( func_call, param_location )
    param_location = param_location + 1
  end
  return func_params
end

local function dump_function(func_call)
  local final_output = "function %s( %s )"
  local func_name = ""
  local func_param_string = ""
  for k, v in pairs(_G) do
    if type(v) == "table" then
      for x, y in pairs(v) do
        if (y == func_call) then
          func_name = k.."."..x
        end
      end
    end
    if (v == func_call) then
      func_name = k
    end
  end

  if (!func_name) then error("Couldn't locate function") end

  if (debug.getinfo(func_call).short_src == "[C]") then error("C Function, this cannot be reconstructed") end

  local dumped_params = dump_function_params(func_call)

  func_param_string = table.concat( dumped_params, ", " )
  final_output = string.format(final_output, func_name, func_param_string)

  local variable_location = -1
  local func_variable = jit.util.funck(func_call, variable_location);
  local last_saved_table = nil
  local open_pairs = false
  while func_variable != nil do
    variable_location = variable_location - 1;
    local next_variable = jit.util.funck(func_call, variable_location);
    if _G[func_variable] && isfunction(_G[func_variable]) then
      if (func_variable == "pairs") then
        final_output = string.format("%s\n for k, v in %s( cant locate ) do ", final_output, func_variable)
        open_pairs = true
      else
        if (open_pairs) then
          final_output = string.format("%s\n%s", final_output, "end")
        end
        final_output = string.format("%s\n function %s( %s )", final_output, func_variable, table.concat( dump_function_params(_G[func_variable]), ", " ))
      end
    else
      if (istable(_G[func_variable])) then
        if (next_variable != nil) then
          if (_G[func_variable][next_variable] && isfunction(_G[func_variable][next_variable])) then
            final_output = string.format("%s\n   %s( %s )", final_output, func_variable.."."..next_variable, table.concat( dump_function_params(_G[func_variable][next_variable]), ", " ))
            last_saved_table = func_variable
          end
        end
      else
        if (_G[func_variable] == nil && last_saved_table && _G[last_saved_table][func_variable]) then
          final_output = string.format("%s\n   %s( %s )", final_output, last_saved_table.."."..func_variable, table.concat( dump_function_params(_G[last_saved_table][func_variable]), ", " ))
        else
          final_output = string.format("%s\n     %s", final_output, func_variable)
        end
      end
    end
    func_variable = next_variable;
  end

  if (open_pairs) then
    final_output = string.format("%s\n %s", final_output, "end")
  end

  final_output = string.format("%s\n%s\n", final_output, "end")
  print(final_output)
end