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; }
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; }
tid_t create_vtbl_struct(ea_t vtbl_addr, ea_t vtbl_addr_end, char* vtbl_name, uval_t idx, unsigned int* vtbl_len) { qstring struc_name = vtbl_name; //struc_name.append(qstring("_vtbl_struct")); tid_t id = add_struc(BADADDR, struc_name.c_str()); if (id == BADADDR) { struc_name.clear(); struc_name = askstr(HIST_IDENT, NULL, "Default name %s not correct. Enter other structure name: ", struc_name.c_str()); id = add_struc(BADADDR, struc_name.c_str()); set_struc_cmt(id, vtbl_name, true); } struc_t* new_struc = get_struc(id); if (!new_struc) return BADNODE; ea_t ea = vtbl_addr; ea_t offset = 0; while (ea < vtbl_addr_end) { offset = ea - vtbl_addr; qstring method_name; #ifndef __EA64__ ea_t method_ea = get_long(ea); // get function ea #else ea_t method_ea = get_64bit(ea); #endif if (method_ea == 0) break; if (!isEnabled(method_ea)) break; flags_t method_flags = getFlags(method_ea); char* struc_member_name = NULL; if (isFunc(method_flags)) { method_name = f_get_short_name(method_ea); // this line crash ida when compare qstring with null if (method_name.length() != 0) { struc_member_name = (char*)method_name.c_str(); } } #ifndef __EA64__ add_struc_member(new_struc, NULL, offset, dwrdflag(), NULL, 4); #else add_struc_member(new_struc, NULL, offset, qwrdflag(), NULL, sizeof(UINT64)); #endif if (struc_member_name) { if (!set_member_name(new_struc, offset, struc_member_name)) { //get_name(NULL, method_ea, method_name, sizeof(method_name)); f_get_ea_name(&method_name, method_ea); set_member_name(new_struc, offset, struc_member_name); } } #ifndef __EA64__ ea = ea + 4; #else ea = ea + sizeof(UINT64); #endif flags_t ea_flags = getFlags(ea); if (has_any_name(ea_flags)) break; } return id; }