//--------------------------------------------------------------------------
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);
}
tid_t type_builder_t::get_structure(const qstring name)
{
tid_t struct_type_id = add_struc(BADADDR, name.c_str());
if (struct_type_id != 0 || struct_type_id != -1)
{
struc_t * struc = get_struc(struct_type_id);
if(struc != NULL)
{
opinfo_t opinfo;
opinfo.tid = struct_type_id;
int j = 0;
for(std::map<int, struct_filed>::iterator i = structure.begin(); i != structure.end() ; i ++)
{
VTBL_info_t vtbl;
flags_t member_flgs = 0;
if(i->second.size == 1)
member_flgs = byteflag();
else if (i->second.size == 2)
member_flgs = wordflag();
else if (i->second.size == 4)
member_flgs = dwrdflag();
else if (i->second.size == 8)
member_flgs = qwrdflag();
char field_name[258];
memset(field_name, 0x00, sizeof(field_name));
if((i->second.vftbl != BADADDR) && get_vbtbl_by_ea(i->second.vftbl, vtbl))
{
qstring vftbl_name = name;
vftbl_name.cat_sprnt("_VTABLE_%X_%p", i->second.offset, i->second.vftbl);
tid_t vtbl_str_id = create_vtbl_struct(vtbl.ea_begin, vtbl.ea_end, (char *)vftbl_name.c_str(), 0);
if (vtbl_str_id != BADADDR) {
sprintf_s(field_name, sizeof(field_name), "vftbl_%d_%p", j, i->second.vftbl);
int iRet = add_struc_member(struc, field_name, i->second.offset, member_flgs, NULL, i->second.size);
member_t * membr = get_member_by_name(struc, field_name);
if (membr != NULL) {
tinfo_t new_type = create_typedef((char *)vftbl_name.c_str());
if(new_type.is_correct()) {
smt_code_t dd = set_member_tinfo2(struc, membr, 0, make_pointer(new_type), SET_MEMTI_COMPATIBLE);
}
}
}
}
else
{
sprintf_s(field_name, sizeof(field_name), "field_%X", i->second.offset);
int iRet = add_struc_member(struc, field_name, i->second.offset, member_flgs, NULL, i->second.size);
}
j ++;
}
}
}
return struct_type_id;
}
//----------------------------------------------------------------------
//
// defines, names and comments an item
//
static void define_item( ushort address, asize_t size, char *shortdesc, char *comment )
{
do_unknown( address, true );
do_data_ex( address, (size == IOREG_16 ? wordflag() : byteflag() ), size, BADNODE );
set_name( address, shortdesc );
set_cmt( address, comment, true );
}
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);
}
tid_t type_builder_t::get_structure(char * name)
{
tid_t struct_type_id = add_struc(BADADDR, name);
if (struct_type_id != 0 || struct_type_id != -1)
{
struc_t * struc = get_struc(struct_type_id);
if(struc != NULL)
{
sort_fields(structure);
int offs = 0;
opinfo_t opinfo;
opinfo.tid = struct_type_id;
for(unsigned int i = 0 ; i < structure.size() ; i ++)
{
if(structure[i].offset > offs)
{
offs = structure[i].offset;
}
flags_t member_flgs = 0;
if(structure[i].size == 1)
member_flgs = byteflag();
else if (structure[i].size == 2)
member_flgs = wordflag();
else if (structure[i].size == 4)
member_flgs = dwrdflag();
else if (structure[i].size == 8)
member_flgs = qwrdflag();
char field_name[258];
memset(field_name, 0x00, sizeof(field_name));
sprintf_s(field_name, sizeof(field_name), "field_%d", i);
int iRet = add_struc_member(struc, field_name, structure[i].offset, member_flgs, NULL, structure[i].size);
offs += structure[i].size;
}
}
}
return struct_type_id;
}
