void idaapi run(int arg)
{
char *st_name = "struc_2";
char buf[MAXSTR];
static char st_name_buf[100];
if (arg == 0)
{
strcpy(buf, st_name);
CWin32InputBox::InputBox("Struct name", "Enter struct name:", st_name_buf, sizeof(st_name_buf));
st_name = st_name_buf;
}
size_t st_qty = get_struc_qty();
msg("we have: %d structs\n", st_qty);
// find the structure by name
tid_t id = get_struc_id(st_name);
// get the size of the structure
asize_t st_sz = get_struc_size(id);
msg("sizeof(%s) = %d\n", st_name, st_sz);
struc_t *st = get_struc(id);
enum_members2(st);
return;
}
//--------------------------------------------------------------------------
static void declare_class(ea_t ea, const char *entryname)
{
static const char class_name[] = "ClassStruct";
struc_t *sptr = get_struc(get_struc_id(class_name));
if ( sptr == NULL )
{
sptr = get_struc(add_struc(BADADDR, class_name));
if ( sptr == NULL )
return;
opinfo_t mt;
mt.ri.flags = REF_OFF32;
mt.ri.target = BADADDR;
mt.ri.base = 0;
mt.ri.tdelta = 0;
add_struc_member(sptr, "superClass", BADADDR, offflag()|dwrdflag(), &mt, 4);
add_struc_member(sptr, "masterOffset", BADADDR, wordflag(), NULL, 2);
add_struc_member(sptr, "methodCount", BADADDR, decflag()|wordflag(), NULL, 2);
add_struc_member(sptr, "instanceSize", BADADDR, decflag()|wordflag(), NULL, 2);
add_struc_member(sptr, "vdRelocTable", BADADDR, wordflag(), NULL, 2);
add_struc_member(sptr, "relocTable", BADADDR, wordflag(), NULL, 2);
mt.ec.tid = get_class_struct_flags_enum();
mt.ec.serial = 0;
add_struc_member(sptr, "flags", BADADDR, enumflag()|byteflag(), &mt, 1);
add_struc_member(sptr, "masterMethods",BADADDR, byteflag(), NULL, 1);
}
asize_t size = get_struc_size(sptr);
doStruct(ea, size, sptr->id);
int count = get_word(ea+6);
// bool c_handlers = get_byte(ea+14) & (1<<6);
ea += size;
ea_t messages = ea;
doWord(ea, count*2);
op_dec(ea, 0);
ea += 2*count;
doDwrd(ea, count*4);
set_offset(ea, 0, 0);
for ( int i=0; i < count; i++ )
{
ea_t idx = ea + 4*i;
ea_t pea = toEA(get_word(idx+2), get_word(idx));
auto_make_proc(pea);
char name[MAXSTR];
qsnprintf(name, sizeof(name), "%s_%u", entryname, get_word(messages+2*i));
add_entry(pea, pea, name, true);
}
// commented out because it doesn't work properly
// see geoplan.geo, entry number 1 for example
// if ( c_handlers )
// declare_parameter_types(ea+count*4, count);
}
static void declare_parameter_types(ea_t ea, int count)
{
static const char class_name[] = "CMethodDef";
struc_t *sptr = get_struc(get_struc_id(class_name));
if ( sptr == NULL )
{
sptr = get_struc(add_struc(-1, class_name));
if ( sptr == NULL ) return;
add_struc_member(sptr, "methodParameterDef", -1, wordflag(), NULL, 2);
add_struc_member(sptr, "handlerTypeDef", -1, byteflag(), NULL, 1);
}
size_t size = get_struc_size(sptr);
doStruct(ea, size*count, sptr->id);
}
std::vector<IDAStructure> GetStructsFromDb()
{
std::vector<IDAStructure> structures;
constexpr size_t bufferSize = 256;
std::array<char, bufferSize> buffer;
for (auto i = get_first_struc_idx(); i != -1; i = get_next_struc_idx(i))
{
IDAStructure newStruct;
const struc_t* idaStruct = get_struc(get_struc_by_idx(i));
get_struc_name(idaStruct->id, buffer.data(), bufferSize);
newStruct.m_name = std::string(buffer.data());
get_struc_cmt(idaStruct->id, true, buffer.data(), bufferSize);
newStruct.m_comment = std::string(buffer.data());
newStruct.m_size = get_struc_size(idaStruct->id);
msg("Struct %d = %s (%s) [%d bytes]\n", i, newStruct.m_name.c_str(), newStruct.m_comment.c_str(), newStruct.m_size);
size_t offset = 0;
member_t* idaStructMember = get_member(idaStruct, offset);
while (idaStructMember != nullptr)
{
IDAStructure::Member newMember;
get_member_fullname(idaStructMember->id, buffer.data(), bufferSize);
newMember.m_name = std::string(buffer.data());
{
tinfo_t typeInfo;
get_member_tinfo2(idaStructMember, &typeInfo);
qstring typeName;
if (typeInfo.get_type_name(&typeName))
{
newMember.m_type = std::string(typeName.c_str());
}
else
{
newMember.m_type = "undefined";
}
}
get_member_cmt(idaStructMember->id, true, buffer.data(), bufferSize);
newMember.m_comment = std::string(buffer.data());
newMember.m_size = get_member_size(idaStructMember);
offset += newMember.m_size;
msg(" %s {%s} (%s) [%d bytes]\n", newMember.m_name.c_str(), newMember.m_type.c_str(), newMember.m_comment.c_str(), newMember.m_size);
newStruct.m_members.push_back(std::move(newMember));
idaStructMember = get_member(idaStruct, offset);
}
structures.push_back(std::move(newStruct));
}
return std::move(structures);
}
