/*============================================================================== * FUNCTION: TypeTest::testDataInterval * OVERVIEW: Test the DataIntervalMap class *============================================================================*/ void TypeTest::testDataInterval() { DataIntervalMap dim; Prog* prog = new Prog; UserProc* proc = (UserProc*) prog->newProc("test", 0x123); std::string name("test"); proc->setSignature(Signature::instantiate(PLAT_PENTIUM, CONV_C, name.c_str())); dim.setProc(proc); dim.addItem(0x1000, "first", new IntegerType(32, 1)); dim.addItem(0x1004, "second", new FloatType(64)); std::string actual(dim.prints()); std::string expected("0x1000 first int\n" "0x1004 second double\n"); CPPUNIT_ASSERT_EQUAL(expected, actual); DataIntervalEntry* pdie = dim.find(0x1000); expected = "first"; CPPUNIT_ASSERT(pdie); actual = pdie->second.name; CPPUNIT_ASSERT_EQUAL(expected, actual); pdie = dim.find(0x1003); CPPUNIT_ASSERT(pdie); actual = pdie->second.name; CPPUNIT_ASSERT_EQUAL(expected, actual); pdie = dim.find(0x1004); CPPUNIT_ASSERT(pdie); expected = "second"; actual = pdie->second.name; CPPUNIT_ASSERT_EQUAL(expected, actual); pdie = dim.find(0x1007); CPPUNIT_ASSERT(pdie); actual = pdie->second.name; CPPUNIT_ASSERT_EQUAL(expected, actual); CompoundType ct; ct.addType(new IntegerType(16, 1), "short1"); ct.addType(new IntegerType(16, 1), "short2"); ct.addType(new IntegerType(32, 1), "int1"); ct.addType(new FloatType(32), "float1"); dim.addItem(0x1010, "struct1", &ct); ComplexTypeCompList& ctcl = ct.compForAddress(0x1012, dim); unsigned ua = ctcl.size(); unsigned ue = 1; CPPUNIT_ASSERT_EQUAL(ue, ua); ComplexTypeComp& ctc = ctcl.front(); ue = 0; ua = ctc.isArray; CPPUNIT_ASSERT_EQUAL(ue, ua); expected = "short2"; actual = ctc.u.memberName; CPPUNIT_ASSERT_EQUAL(expected, actual); // An array of 10 struct1's ArrayType at(&ct, 10); dim.addItem(0x1020, "array1", &at); ComplexTypeCompList& ctcl2 = at.compForAddress(0x1020+0x3C+8, dim); // Should be 2 components: [5] and .float1 ue = 2; ua = ctcl2.size(); CPPUNIT_ASSERT_EQUAL(ue, ua); ComplexTypeComp& ctc0 = ctcl2.front(); ComplexTypeComp& ctc1 = ctcl2.back(); ue = 1; ua = ctc0.isArray; CPPUNIT_ASSERT_EQUAL(ue, ua); ue = 5; ua = ctc0.u.index; CPPUNIT_ASSERT_EQUAL(ue, ua); ue = 0; ua = ctc1.isArray; CPPUNIT_ASSERT_EQUAL(ue, ua); expected = "float1"; actual = ctc1.u.memberName; CPPUNIT_ASSERT_EQUAL(expected, actual); }
/*============================================================================== * FUNCTION: TypeTest::testDataIntervalOverlaps * OVERVIEW: Test the DataIntervalMap class with overlapping addItems *============================================================================*/ void TypeTest::testDataIntervalOverlaps() { DataIntervalMap dim; Prog* prog = new Prog; UserProc* proc = (UserProc*) prog->newProc("test", 0x123); std::string name("test"); proc->setSignature(Signature::instantiate(PLAT_PENTIUM, CONV_C, name.c_str())); dim.setProc(proc); dim.addItem(0x1000, "firstInt", new IntegerType(32, 1)); dim.addItem(0x1004, "firstFloat", new FloatType(32)); dim.addItem(0x1008, "secondInt", new IntegerType(32, 1)); dim.addItem(0x100C, "secondFloat", new FloatType(32)); CompoundType ct; ct.addType(new IntegerType(32, 1), "int3"); ct.addType(new FloatType(32), "float3"); dim.addItem(0x1010, "existingStruct", &ct); // First insert a new struct over the top of the existing middle pair CompoundType ctu; ctu.addType(new IntegerType(32, 0), "newInt"); // This int has UNKNOWN sign ctu.addType(new FloatType(32), "newFloat"); dim.addItem(0x1008, "replacementStruct", &ctu); DataIntervalEntry* pdie = dim.find(0x1008); std::string expected = "struct { int newInt; float newFloat; }"; std::string actual = pdie->second.type->getCtype(); CPPUNIT_ASSERT_EQUAL(expected, actual); // Attempt a weave; should fail CompoundType ct3; ct3.addType(new FloatType(32), "newFloat3"); ct3.addType(new IntegerType(32, 0), "newInt3"); dim.addItem(0x1004, "weaveStruct1", &ct3); pdie = dim.find(0x1004); expected = "firstFloat"; actual = pdie->second.name; CPPUNIT_ASSERT_EQUAL(expected, actual); // Totally unaligned dim.addItem(0x1001, "weaveStruct2", &ct3); pdie = dim.find(0x1001); expected = "firstInt"; actual = pdie->second.name; CPPUNIT_ASSERT_EQUAL(expected, actual); dim.addItem(0x1004, "firstInt", new IntegerType(32, 1)); // Should fail pdie = dim.find(0x1004); expected = "firstFloat"; actual = pdie->second.name; CPPUNIT_ASSERT_EQUAL(expected, actual); // Set up three ints dim.deleteItem(0x1004); dim.addItem(0x1004, "firstInt", new IntegerType(32, 1)); // Definately signed dim.deleteItem(0x1008); dim.addItem(0x1008, "firstInt", new IntegerType(32, 0)); // Unknown signedess // then, add an array over the three integers ArrayType at(new IntegerType(32, 0), 3); dim.addItem(0x1000, "newArray", &at); pdie = dim.find(0x1005); // Check middle element expected = "newArray"; actual = pdie->second.name; CPPUNIT_ASSERT_EQUAL(expected, actual); pdie = dim.find(0x1000); // Check first actual = pdie->second.name; CPPUNIT_ASSERT_EQUAL(expected, actual); pdie = dim.find(0x100B); // Check last actual = pdie->second.name; CPPUNIT_ASSERT_EQUAL(expected, actual); // Already have an array of 3 ints at 0x1000. Put a new array completely before, then with only one word overlap dim.addItem(0xF00, "newArray2", &at); pdie = dim.find(0x1000); // Shouyld still be newArray at 0x1000 actual = pdie->second.name; CPPUNIT_ASSERT_EQUAL(expected, actual); pdie = dim.find(0xF00); expected = "newArray2"; actual = pdie->second.name; CPPUNIT_ASSERT_EQUAL(expected, actual); dim.addItem(0xFF8, "newArray3", &at); // Should fail pdie = dim.find(0xFF8); unsigned ue = 0; // Expect NULL unsigned ua = (unsigned)pdie; CPPUNIT_ASSERT_EQUAL(ue, ua); }
std::vector<ADDRESS> FrontEnd::getEntryPoints() { std::vector<ADDRESS> entrypoints; bool gotMain = false; ADDRESS a = getMainEntryPoint(gotMain); if (a != NO_ADDRESS) entrypoints.push_back(a); else { // try some other tricks const char *fname = pBF->getFilename(); // X11 Module if (!strcmp(fname + strlen(fname) - 6, "_drv.o")) { const char *p = fname + strlen(fname) - 6; while (*p != '/' && *p != '\\' && p != fname) p--; if (p != fname) { p++; char *name = (char*)malloc(strlen(p) + 30); strcpy(name, p); name[strlen(name)-6] = 0; strcat(name, "ModuleData"); ADDRESS a = pBF->GetAddressByName(name, true); if (a != NO_ADDRESS) { ADDRESS vers, setup, teardown; vers = pBF->readNative4(a); setup = pBF->readNative4(a+4); teardown = pBF->readNative4(a+8); if (setup) { Type *ty = NamedType::getNamedType("ModuleSetupProc"); assert(ty->isFunc()); UserProc *proc = (UserProc*)prog->setNewProc(setup); assert(proc); Signature *sig = ty->asFunc()->getSignature()->clone(); const char *sym = pBF->SymbolByAddress(setup); if (sym) sig->setName(sym); sig->setForced(true); proc->setSignature(sig); entrypoints.push_back(setup); } if (teardown) { Type *ty = NamedType::getNamedType("ModuleTearDownProc"); assert(ty->isFunc()); UserProc *proc = (UserProc*)prog->setNewProc(teardown); assert(proc); Signature *sig = ty->asFunc()->getSignature()->clone(); const char *sym = pBF->SymbolByAddress(teardown); if (sym) sig->setName(sym); sig->setForced(true); proc->setSignature(sig); entrypoints.push_back(teardown); } } } } // Linux kernel module if (!strcmp(fname + strlen(fname) - 3, ".ko")) { a = pBF->GetAddressByName("init_module"); if (a != NO_ADDRESS) entrypoints.push_back(a); a = pBF->GetAddressByName("cleanup_module"); if (a != NO_ADDRESS) entrypoints.push_back(a); } } return entrypoints; }