--require("COM");
COM={};

function COM.normStr(s)
	if string.len(s)<8 then
		s=string.format("%s%s",string.rep("0",8-string.len(s)),s);
	end;
	s=string.upper(s);
	s=string.gsub(s,"%s+","");
	return s;
end

function COM.InstallMenu(mnis,main,need,cb)
	local mi,s,mpos;
	mpos=menuItem_getCount(mnis);
	for i=0,menuItem_getCount(mnis)-1 do
		mi=menuItem_getItem(mnis, i);
		s=menuItem_getCaption(mi);
		s=string.gsub(s,"&","");
		if string.upper(s)==string.upper(need) then
			return mi;
		end
		if string.upper(s)==string.upper(main) then
			mpos=i;
		end
	end
	-- not found, so we create it
	mi=createMenuItem(mnis);
	menuItem_setCaption(mi,need);
	if cb~=nil then
		menuItem_onClick(mi,cb);
	end
	menuItem_insert(mnis, mpos, mi);
	return mi;
end
--if (VTBL==nil) then
  VTBL = {
	title="Object Constructor Finder",
	width=800,
	height=300,
	form=nil,
	addr=nil,
    list=nil,
	addrlist=nil,
	backsize=50,
    foundlist={},foundnum=nil,lastindex=0,
	btnJump=nil,
    mem=nil,
    dis=nil
    }
--end

function VTBL.Show()
  local te,s,i;
  --te=addresslist_getSelectedRecord(VTBL.addrlist);
  --s=memoryrecord_getValue(te);
  s=string.format("%x",VTBL.getSelectedAddressValue());
  s=VTBL.normStr(s);
  if s==VTBL.normStr(control_getCaption(VTBL.addr)) then
    VTBL.ShowFoundList();
  else
    lastindex=-1;
    listview_clear(VTBL.list);
	VTBL.ClearFoundList();
    control_setCaption(VTBL.foundnum,0);

  end
  control_setCaption(VTBL.addr,s);
  VTBL.ChangeJumpToState();
  form_show(VTBL.form);
end

function VTBL.getSelectedAddressValue()
	local a,b,c;

	--a=form_getMenu(getMainForm());
	a=addresslist_getSelectedRecord(getAddressList());
	b=memoryrecord_getAddress(a);
	c=readInteger(b);
	return c;
	--a=control_getPopupMenu();
	--print(string.format("%x",c));
end

function VTBL.normStr(s)
	if string.len(s)<8 then
		s=string.format("%s%s",string.rep("0",8-string.len(s)),s);
	end;
	s=string.upper(s);
	s=string.gsub(s,"%s+","");
	return s;
end
function VTBL.ClearFoundList()
	local i;
	for i in pairs(VTBL.foundlist) do
		VTBL.foundlist[i]=nil;
	end
end
function VTBL.GetFoundCount()
  local num=0;
  local a;
  if (VTBL.foundlist~=nil) then
    for a in pairs(VTBL.foundlist) do num = num + 1 end
  end
  return num;
end
function VTBL.Close()
  form_hide(VTBL.form);
end
function VTBL.AOBScan()
	local i,j,t,u,v;
	local s=control_getCaption(VTBL.addr);
	s=VTBL.normStr(s);
	control_setCaption(VTBL.addr,s);
	if s=="00000000" then
		showMessage("Please choose another address");
	else
		listview_clear(VTBL.list);
		t="";

		for i=0,3 do
			u=string.sub(s,i*2+1,i*2+2);
			t=string.format("%s %s",u,t);
		end
		t=string.sub(t,1,11);
		VTBL.FilterAOBList(t);
		control_setCaption(VTBL.foundnum,VTBL.GetFoundCount());
		VTBL.ShowFoundList();
	end
end
function VTBL.GetAddr(num)
  local i=num,c,m,j,found;
  local s,address,rs;
  rs={};
  address=VTBL.normStr(control_getCaption(VTBL.addr));
  address=string.format(",%s",address);
  --address=string.upper(address);
  m=0;
  while m<VTBL.backsize do
    c=getInstructionSize(num);
    s=disassemble(num);
    if string.find(s, address)~=nil then
		found=0;
		for j in pairs(rs) do
			if rs[j]==s then
				found=1;
				break;
			end
		end
		if found==0 then
			table.insert(rs,s);
		end
    end
    num=num+c;
    m=m+c;

  end
  return rs;
end

function VTBL.FilterAOBList(sb)
	local rs;
	local it,i,num,k,t,v,rm,j,found;
	rs=AOBScan(sb,"+X");
	found=0;
	if (rs~=nil) then
	  VTBL.ClearFoundList();

	  for i=0,strings_getCount(rs)-1 do
		it=stringlist_getString( rs, i );
		num = tonumber(it,16)-VTBL.backsize;
		t=VTBL.GetAddr(num);
		for j in pairs(t) do
			rm=0;
			for k in pairs(VTBL.foundlist) do
				if t[j]==VTBL.foundlist[k] then
				  rm=1;
				  break;
				end
			end
			if rm==0 then
				table.insert(VTBL.foundlist,t[j]);
				found=1;
			end
			t[j]=nil;
		end
	  end
	  object_destroy(rs);
		
	end
	if found==0 then
		showMessage("Sorry, nothing found.");
	end
end

function VTBL.ShowDisAssambler()
  local lits,li,cap,i,loc;
  lits=listview_getItems(VTBL.list);
  li=listitems_getItem(lits, lastindex);
  cap=listitem_getCaption(li);
  for i in pairs(VTBL.foundlist) do
    if VTBL.foundlist[i]==cap then
      loc=string.sub(cap,1,8);

      disassemblerview_setSelectedAddress(VTBL.dis,loc);
	  --form_show(VTBL.dis);
	  --control_setVisible(VTBL.dis, true);
      break;
    end
  end
end

function VTBL.onListViewSelected()
  lastindex=listview_getItemIndex(VTBL.list);
  VTBL.ChangeJumpToState();
end
function VTBL.GetModuleName(address)
	local t,u,v;
	t=getNameFromAddress(address);
	if t==nil then return "";end
	u=string.find(t,"%.");
	
	v=string.sub(t,1,u-1);
	return string.upper(v);
end
function VTBL.ShowFoundList()
  local t,lits,li,a,mdl,li1;

  listview_clear(VTBL.list);
  lits=listview_getItems(VTBL.list);

  for a in pairs(VTBL.foundlist) do
    li=listitems_add(lits);
    listitem_setCaption(li,VTBL.foundlist[a]);
	mdl=VTBL.GetModuleName(string.sub(VTBL.foundlist[a],1,8));
	li1=listitem_getSubItems(li);
	
	strings_add(li1,mdl);
	
	--listitem_setCaption(li1,mdl);
  end
end
function VTBL.ChangeJumpToState()
  if lastindex==-1 then
	control_setEnabled(VTBL.btnJump, false);
  else
	control_setEnabled(VTBL.btnJump, true);
  end
end

function VTBL.init()
  local mf,mi,mis,co,it,mi1;
  local lbl,btn,lvc,lvci;
  VTBL.addrlist=getAddressList();
  VTBL.mem=getMemoryViewForm();
  VTBL.dis=memoryview_getDisassemblerView(VTBL.mem);
  --  VTBL.form=nil;
  --  VTBL.Show();
  if (VTBL.form==nil) then
    VTBL.form=createForm(false);
    control_setSize(VTBL.form, VTBL.width,VTBL.height);
    control_setCaption(VTBL.form,VTBL.title);
    form_centerScreen(VTBL.form);

    --label for editbox
    lbl=createLabel(VTBL.form);
    control_setCaption(lbl,"Virtual Table Address:");
    control_setPosition(lbl,10,10);

    --editbox
    VTBL.addr=createEdit(VTBL.form);
    control_setPosition(VTBL.addr,10,30);
    control_setSize(VTBL.addr, VTBL.width-120,0);


    --scan button
    btn=createButton(VTBL.form);
    control_setSize(btn, 90,24);
    control_setPosition(btn,VTBL.width-100,30);
    control_setCaption(btn,"Scan");
    control_onClick(btn,VTBL.AOBScan);

    --label for list view
    lbl=createLabel(VTBL.form);
    control_setCaption(lbl,"Constructor(s) Found :");
    control_setPosition(lbl,10,60);

    VTBL.foundnum=createLabel(VTBL.form);
    control_setCaption(VTBL.foundnum,VTBL.GetFoundCount());
    control_setPosition(VTBL.foundnum,150,60);


    --list view
    VTBL.list=createListView(VTBL.form);
    control_setPosition(VTBL.list,10,80);
    control_setSize(VTBL.list, VTBL.width-20,VTBL.height-120);
    control_onClick(VTBL.list,VTBL.onListViewSelected);
	setMethodProperty(VTBL.list,"ondblclick",VTBL.ShowDisAssambler);
    -- columns
    lvc=listview_getColumns(VTBL.list);
    lvci=listcolumns_add(lvc);
    listcolumn_setCaption(lvci,"Addresss");
    listcolumn_setWidth(lvci,VTBL.width-200);
	
	lvci=listcolumns_add(lvc);
	listcolumn_setCaption(lvci,"Module");
    listcolumn_setWidth(lvci,150);


    --close btn
    btn=createButton(VTBL.form);
    control_setPosition(btn,VTBL.width-90,VTBL.height-33);
    control_setCaption(btn,"Close");
    control_onClick(btn,VTBL.Close);

    --jump to btn
    VTBL.btnJump=createButton(VTBL.form);
    control_setPosition(VTBL.btnJump,VTBL.width-180,VTBL.height-33);
    control_setCaption(VTBL.btnJump,"Jump to");
    control_onClick(VTBL.btnJump,VTBL.ShowDisAssambler);
  end
	
	mi=COM.InstallMenu(menu_getItems(form_getMenu(getMainForm())),"help","LMHS",nil);
	COM.InstallMenu(mi,"",VTBL.title,VTBL.Show);
	--[[
  mf=form_getMenu(getMainForm());
  mi=createMenuItem(mf);
  menuItem_setCaption(mi,"LMHS");
  mis=menu_getItems(mf);
  co=menuItem_getCount(mis)-1;
  menuItem_insert(mis,co-1,mi);

  mi1=createMenuItem(mi);
  menuItem_add(mi,mi1);
  menuItem_setCaption(mi1,VTBL.title);
  menuItem_onClick(mi1, VTBL.Show);
]]--
end

VTBL.init();