blob: 91a6d883c9005a696c290f2eed8b206c4cb5b88d (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
//===-- PTree.cpp ---------------------------------------------------------===//
//
// The KLEE Symbolic Virtual Machine
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#include "PTree.h"
#include "klee/ExecutionState.h"
#include "klee/Expr/Expr.h"
#include "klee/Expr/ExprPPrinter.h"
#include <vector>
using namespace klee;
PTree::PTree(ExecutionState *initialState) {
root = std::make_unique<PTreeNode>(nullptr, initialState);
}
void PTree::attach(PTreeNode *node, ExecutionState *leftState, ExecutionState *rightState) {
assert(node && !node->left && !node->right);
node->state = nullptr;
node->left = std::make_unique<PTreeNode>(node, leftState);
node->right = std::make_unique<PTreeNode>(node, rightState);
}
void PTree::remove(PTreeNode *n) {
assert(!n->left && !n->right);
do {
PTreeNode *p = n->parent;
if (p) {
if (n == p->left.get()) {
p->left = nullptr;
} else {
assert(n == p->right.get());
p->right = nullptr;
}
}
n = p;
} while (n && !n->left && !n->right);
}
void PTree::dump(llvm::raw_ostream &os) {
ExprPPrinter *pp = ExprPPrinter::create(os);
pp->setNewline("\\l");
os << "digraph G {\n";
os << "\tsize=\"10,7.5\";\n";
os << "\tratio=fill;\n";
os << "\trotate=90;\n";
os << "\tcenter = \"true\";\n";
os << "\tnode [style=\"filled\",width=.1,height=.1,fontname=\"Terminus\"]\n";
os << "\tedge [arrowsize=.3]\n";
std::vector<const PTreeNode*> stack;
stack.push_back(root.get());
while (!stack.empty()) {
const PTreeNode *n = stack.back();
stack.pop_back();
os << "\tn" << n << " [shape=diamond";
if (n->state)
os << ",fillcolor=green";
os << "];\n";
if (n->left) {
os << "\tn" << n << " -> n" << n->left.get() << ";\n";
stack.push_back(n->left.get());
}
if (n->right) {
os << "\tn" << n << " -> n" << n->right.get() << ";\n";
stack.push_back(n->right.get());
}
}
os << "}\n";
delete pp;
}
PTreeNode::PTreeNode(PTreeNode *parent, ExecutionState *state) : parent{parent}, state{state} {
state->ptreeNode = this;
}
|
