update

1 files changed, 76 insertions, 10 deletions

diff --git a/bruiser/lua-scripts/demo1.lua b/bruiser/lua-scripts/demo1.lua
index cb912b8..de83431 100644
--- a/bruiser/lua-scripts/demo1.lua
+++ b/bruiser/lua-scripts/demo1.lua
@@ -1,17 +1,17 @@
 --
--- get the .so object names
+-- get the .so object names in a table
 -- objload("elf_get_obj_names", "../bfd/test/test.so", "symbol_list")
 --
--- get the .so object sizes
+-- get the .so object sizes in a table
 -- objload("elf_get_obj_sizes", "../bfd/test/test.so", "symbol_list")
 --
--- get the .so function names
+-- get the .so function names in a table
 -- objload("elf_get_func_names", "../bfd/test/test.so", "symbol_list")
 --
--- get the .so function code
+-- get the .so function code in a table of tables
 -- objload("elf_get_func_code", "../bfd/test/test.so", "code_list")
 --
-
+--------------------------------------------------------------------------------------------------------------
 function printObjNames()
   local c = objload("elf_get_obj_names", "../bfd/test/test.so", "symbol_list")
   for k,v in ipairs(c) do
@@ -39,13 +39,79 @@ function printFuncCode()
     print(k,v)
     if #v ~= 0 then
       for k1,v1 in ipairs(v) do
-        print(k1, v1)
+        io.write(string.format('%02x', v1), " ")
+      end
+      io.write("\n")
+    end
+  end
+end
+
+function findMain()
+  local c = objload("elf_get_func_names", "../bfd/test/test.so", "symbol_list")
+  for k,v in ipairs(c) do
+    if v == "'main'" then 
+      io.write("main index is".." "..k.."\n")
+      return k
+    end
+  end
+end
+
+function codeTables()
+  local return_table = {}
+  local func_name_table = objload("elf_get_func_names", "../bfd/test/test.so", "symbol_list")
+  local code_table = objload("elf_get_func_code", "../bfd/test/test.so", "code_list")
+  for i=1,#func_name_table,1 do
+    return_table[func_name_table[i]] = code_table[i]
+  end
+  return return_table
+end
+
+function codeTableByName(name)
+  local return_table = {}
+  local func_name_table = objload("elf_get_func_names", "../bfd/test/test.so", "symbol_list")
+  local code_table = objload("elf_get_func_code", "../bfd/test/test.so", "code_list")
+  for k,v in ipairs(func_name_table) do
+    if v == name then
+      for k1, v1 in ipairs(code_table[k]) do
+        table.insert(return_table, string.format('%02x', v1))
       end
+      return return_table
     end
   end
+  return nil
+end
+
+-- start of @placeholder
+function XobjRegister(code_table, registration_name)
+  -- lightuserdata
+  local Xobjpointer
+  return Xobjpointer
+end
+-- should be varargs
+-- registration_name()
+-- end of @placeholder
+
+function main()
+  printObjNames()
+  printObjSizes()
+  printFuncNames()
+  printFuncCode()
+  findMain()
+
+  local code_table = codeTables()
+  print(code_table["'main'"])
+  for k,v in ipairs(code_table["'main'"]) do
+    io.write(string.format('%02x', v), " ")
+  end
+  io.write("\n")
+
+  local C_main_code = codeTableByName("'main'")
+  for k, v in ipairs(C_main_code) do
+    io.write(v, " ")
+  end
+  io.write("\n")
 end
 
-printObjNames()
-printObjSizes()
-printFuncNames()
-printFuncCode()
+main()
+--------------------------------------------------------------------------------------------------------------
+