void
BPatch_basicBlock::getAllPostDominates(BPatch_Set<BPatch_basicBlock*>& buffer){
std::set<BPatch_basicBlock *> tmp;
getAllPostDominates(tmp);
std::copy(tmp.begin(), tmp.end(), std::inserter(buffer.int_set, buffer.begin()));
}
bool instrumentMemoryWrites(dynHandle *dh, BPatch_function *func)
{
BPatch_Set<BPatch_basicBlock*> allBlocks;
BPatch_snippet incSnippet;
BPatch_Set<BPatch_opCode> ops;
BPatch_Set<BPatch_basicBlock*>::iterator iter;
int bb_warn_cnt = 0, bb_pass_cnt = 0;
sendMsg(config.outfd, ID_INST_MEM_WRITE, VERB2);
sendMsg(config.outfd, ID_GET_CFG, VERB3);
BPatch_flowGraph *appCFG = func->getCFG();
if (!appCFG) {
sendMsg(config.outfd, ID_GET_CFG, VERB3, ID_FAIL,
"Failure in BPatch_function::getCFG()");
goto fail;
} else {
sendMsg(config.outfd, ID_GET_CFG, VERB3, ID_PASS);
}
sendMsg(config.outfd, ID_INST_GET_BB, VERB3);
if (!appCFG->getAllBasicBlocks(allBlocks)) {
sendMsg(config.outfd, ID_INST_GET_BB, VERB3, ID_FAIL,
"Failure in BPatch_flowGraph::getAllBasicBlocks()");
goto fail;
} else if (allBlocks.size() == 0) {
sendMsg(config.outfd, ID_INST_GET_BB, VERB3, ID_WARN,
"No basic blocks found in function");
goto fail;
} else {
sendMsg(config.outfd, ID_INST_GET_BB, VERB3, ID_PASS);
}
if (! generateInstrumentation (dh, func, &incSnippet))
goto fail;
ops.insert(BPatch_opStore);
sendMsg(config.outfd, ID_INST_BB_LIST, VERB3);
for (iter = allBlocks.begin(); iter != allBlocks.end(); iter++) {
if (!shouldInsert())
continue;
sendMsg(config.outfd, ID_INST_GET_BB_POINTS, VERB4);
BPatch_Vector<BPatch_point*> *points = (*iter)->findPoint(ops);
if (!points) {
sendMsg(config.outfd, ID_INST_GET_BB_POINTS, VERB4, ID_WARN,
"Failure in BPatch_basicBlock::findPoint()");
++bb_warn_cnt;
continue;
} else if (points->size() == 0) {
sendMsg(config.outfd, ID_INST_GET_BB_POINTS, VERB4, ID_WARN,
"No instrumentation points found in basic block");
++bb_warn_cnt;
continue;
} else {
sendMsg(config.outfd, ID_INST_GET_BB_POINTS, VERB4, ID_PASS);
}
sendMsg(config.outfd, ID_INST_INSERT_CODE, VERB4);
BPatchSnippetHandle *handle = dh->addSpace->insertSnippet(incSnippet, *points);
if (!handle) {
sendMsg(config.outfd, ID_INST_INSERT_CODE, VERB4, ID_FAIL,
"Failure in BPatch_process::insertSnippet()");
++bb_warn_cnt;
continue;
} else {
sendMsg(config.outfd, ID_INST_INSERT_CODE, VERB4, ID_PASS);
++bb_pass_cnt;
}
}
if (bb_warn_cnt)
sendMsg(config.outfd, ID_INST_BB_LIST, VERB3, ID_WARN,
sprintf_static("%d warning(s), %d passed.", bb_warn_cnt, bb_pass_cnt));
else
sendMsg(config.outfd, ID_INST_BB_LIST, VERB3, ID_PASS);
sendMsg(config.outfd, ID_INST_MEM_WRITE, VERB2, ID_PASS);
return true;
fail:
sendMsg(config.outfd, ID_INST_MEM_WRITE, VERB2, ID_WARN,
"Failure while instrumenting memory writes.");
return false;
}
