Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Small fixes #127

Merged
merged 1 commit into from
Oct 3, 2023
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
20 changes: 0 additions & 20 deletions lib/Core/ExecutionState.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -363,26 +363,6 @@ void ExecutionState::addUniquePointerResolution(ref<Expr> address,
}
}

bool ExecutionState::resolveOnSymbolics(const ref<ConstantExpr> &addr,
IDType &result) const {
uint64_t address = addr->getZExtValue();

for (const auto &res : symbolics) {
const auto &mo = res.memoryObject;
// Check if the provided address is between start and end of the object
// [mo->address, mo->address + mo->size) or the object is a 0-sized object.
ref<ConstantExpr> size = cast<ConstantExpr>(
constraints.cs().concretization().evaluate(mo->getSizeExpr()));
if ((size->getZExtValue() == 0 && address == mo->address) ||
(address - mo->address < size->getZExtValue())) {
result = mo->id;
return true;
}
}

return false;
}

/**/

llvm::raw_ostream &klee::operator<<(llvm::raw_ostream &os,
Expand Down
1 change: 0 additions & 1 deletion lib/Core/ExecutionState.h
Original file line number Diff line number Diff line change
Expand Up @@ -428,7 +428,6 @@ class ExecutionState {
unsigned size = 0);
void addUniquePointerResolution(ref<Expr> address, const MemoryObject *mo,
unsigned size = 0);
bool resolveOnSymbolics(const ref<ConstantExpr> &addr, IDType &result) const;

void addConstraint(ref<Expr> e, const Assignment &c);
void addCexPreference(const ref<Expr> &cond);
Expand Down
68 changes: 53 additions & 15 deletions lib/Core/Executor.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -6949,19 +6949,43 @@ void Executor::logState(const ExecutionState &state, int id,
}
}

void Executor::setInitializationGraph(const ExecutionState &state,
const Assignment &model, KTest &ktest) {
bool resolveOnSymbolics(const std::vector<klee::Symbolic> &symbolics,
const Assignment &assn,
const ref<klee::ConstantExpr> &addr, IDType &result) {
uint64_t address = addr->getZExtValue();

for (const auto &res : symbolics) {
const auto &mo = res.memoryObject;
// Check if the provided address is between start and end of the object
// [mo->address, mo->address + mo->size) or the object is a 0-sized object.
ref<klee::ConstantExpr> size =
cast<klee::ConstantExpr>(assn.evaluate(mo->getSizeExpr()));
if ((size->getZExtValue() == 0 && address == mo->address) ||
(address - mo->address < size->getZExtValue())) {
result = mo->id;
return true;
}
}

return false;
}

void Executor::setInitializationGraph(
const ExecutionState &state, const std::vector<klee::Symbolic> &symbolics,
const Assignment &model, KTest &ktest) {
std::map<size_t, std::vector<Pointer>> pointers;
std::map<size_t, std::map<unsigned, std::pair<unsigned, unsigned>>> s;
ExprHashMap<std::pair<IDType, ref<Expr>>> resolvedPointers;

std::unordered_map<IDType, ref<const MemoryObject>> idToSymbolics;
for (const auto &symbolic : state.symbolics) {
for (const auto &symbolic : symbolics) {
ref<const MemoryObject> mo = symbolic.memoryObject;
idToSymbolics[mo->id] = mo;
}

for (const auto &symbolic : state.symbolics) {
const klee::Assignment &assn = state.constraints.cs().concretization();

for (const auto &symbolic : symbolics) {
KType *symbolicType = symbolic.type;
if (!symbolicType->getRawType()) {
continue;
Expand Down Expand Up @@ -6989,7 +7013,7 @@ void Executor::setInitializationGraph(const ExecutionState &state,
ref<ConstantExpr> constantAddress = cast<ConstantExpr>(addressInModel);
IDType idResult;

if (state.resolveOnSymbolics(constantAddress, idResult)) {
if (resolveOnSymbolics(symbolics, assn, constantAddress, idResult)) {
ref<const MemoryObject> mo = idToSymbolics[idResult];
resolvedPointers[address] =
std::make_pair(idResult, mo->getOffsetExpr(address));
Expand Down Expand Up @@ -7024,12 +7048,12 @@ void Executor::setInitializationGraph(const ExecutionState &state,
// The objects have to be symbolic
bool pointerFound = false, pointeeFound = false;
size_t pointerIndex = 0, pointeeIndex = 0;
for (size_t i = 0; i < state.symbolics.size(); i++) {
if (state.symbolics[i].memoryObject == pointerResolution.first) {
for (size_t i = 0; i < symbolics.size(); i++) {
if (symbolics[i].memoryObject == pointerResolution.first) {
pointerIndex = i;
pointerFound = true;
}
if (state.symbolics[i].memoryObject->id == pointer.second.first) {
if (symbolics[i].memoryObject->id == pointer.second.first) {
pointeeIndex = i;
pointeeFound = true;
}
Expand Down Expand Up @@ -7081,6 +7105,16 @@ Assignment Executor::computeConcretization(const ConstraintSet &constraints,
return concretization;
}

bool isReproducible(const klee::Symbolic &symb) {
auto arr = symb.array;
bool bad = IrreproducibleSource::classof(arr->source.get());
if (bad)
klee_warning_once(arr->source.get(),
"A irreproducible symbolic %s reaches a test",
arr->getIdentifier().c_str());
return !bad;
}

bool Executor::getSymbolicSolution(const ExecutionState &state, KTest &res) {
solver->setTimeout(coreSolverTimeout);

Expand Down Expand Up @@ -7117,10 +7151,14 @@ bool Executor::getSymbolicSolution(const ExecutionState &state, KTest &res) {
}
}

std::vector<klee::Symbolic> symbolics;
std::copy_if(state.symbolics.begin(), state.symbolics.end(),
std::back_inserter(symbolics), isReproducible);

std::vector<SparseStorage<unsigned char>> values;
std::vector<const Array *> objects;
for (unsigned i = 0; i != state.symbolics.size(); ++i)
objects.push_back(state.symbolics[i].array);
for (unsigned i = 0; i != symbolics.size(); ++i)
objects.push_back(symbolics[i].array);
bool success = solver->getInitialValues(extendedConstraints.cs(), objects,
values, state.queryMetaData);
solver->setTimeout(time::Span());
Expand All @@ -7131,11 +7169,11 @@ bool Executor::getSymbolicSolution(const ExecutionState &state, KTest &res) {
return false;
}

res.objects = new KTestObject[state.symbolics.size()];
res.numObjects = state.symbolics.size();
res.objects = new KTestObject[symbolics.size()];
res.numObjects = symbolics.size();

for (unsigned i = 0; i != state.symbolics.size(); ++i) {
auto mo = state.symbolics[i].memoryObject;
for (unsigned i = 0; i != symbolics.size(); ++i) {
auto mo = symbolics[i].memoryObject;
KTestObject *o = &res.objects[i];
o->name = const_cast<char *>(mo->name.c_str());
o->address = mo->address;
Expand All @@ -7151,7 +7189,7 @@ bool Executor::getSymbolicSolution(const ExecutionState &state, KTest &res) {
model.bindings.insert(binding);
}

setInitializationGraph(state, model, res);
setInitializationGraph(state, symbolics, model, res);

return true;
}
Expand Down
1 change: 1 addition & 0 deletions lib/Core/Executor.h
Original file line number Diff line number Diff line change
Expand Up @@ -793,6 +793,7 @@ class Executor : public Interpreter {
Interpreter::LogType logFormat = Interpreter::STP) override;

void setInitializationGraph(const ExecutionState &state,
const std::vector<klee::Symbolic> &symbolics,
const Assignment &model, KTest &tc);

void logState(const ExecutionState &state, int id,
Expand Down
16 changes: 16 additions & 0 deletions test/regression/2023-10-02-test-from-mocked-global.c
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
// Darwin does not support section attribute: `argument to 'section' attribute is not valid for this target: mach-o section specifier requires a segment whose length is between 1 and 16 characters`
// REQUIRES: not-darwin
// RUN: %clang %s -g -emit-llvm %O0opt -c -o %t1.bc
// RUN: rm -rf %t.klee-out
// RUN: %klee --mock-all-externals --write-xml-tests --output-dir=%t.klee-out %t1.bc
// RUN: FileCheck --input-file %t.klee-out/test000001.xml %s

extern void *__crc_mc44s803_attach __attribute__((__weak__));
static unsigned long const __kcrctab_mc44s803_attach __attribute__((__used__, __unused__,
__section__("___kcrctab+mc44s803_attach"))) = (unsigned long const)((unsigned long)(&__crc_mc44s803_attach));

int main() {
return 0;
}

// CHECK-NOT: <input