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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
|
//===-- SMTParser.cpp -----------------------------------------------------===//
//
// The KLEE Symbolic Virtual Machine
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#include "SMTParser.h"
#include "expr/Parser.h"
#include <iostream>
#include <fstream>
#include <string>
#include <sstream>
#include <cassert>
#include <stack>
using namespace std;
using namespace klee;
using namespace klee::expr;
extern int smtlibparse();
extern void *smtlib_createBuffer(int);
extern void smtlib_deleteBuffer(void *);
extern void smtlib_switchToBuffer(void *);
extern int smtlib_bufSize(void);
extern void smtlib_setInteractive(bool);
SMTParser* SMTParser::parserTemp = NULL;
SMTParser::SMTParser(const std::string filename) : fileName(filename),
lineNum(1),
done(false),
query(NULL),
bvSize(0),
queryParsed(false) {
is = new ifstream(filename.c_str());
// Initial empty environments
varEnvs.push(VarEnv());
fvarEnvs.push(FVarEnv());
}
void SMTParser::Init() {
SMTParser::parserTemp = this;
void *buf = smtlib_createBuffer(smtlib_bufSize());
smtlib_switchToBuffer(buf);
smtlib_setInteractive(false);
smtlibparse();
cout << "Parsed successfully.\n";
}
Decl* SMTParser::ParseTopLevelDecl() {
assert(done);
return new QueryCommand(assumptions, query,
std::vector<ExprHandle>(),
std::vector<const Array*>());
}
// XXX: give more info
int SMTParser::Error(const string& s) {
std::cerr << "error: " << s << "\n";
exit(1);
return 0;
}
int SMTParser::StringToInt(const std::string& s) {
std::stringstream str(s);
int x;
str >> x;
assert(str);
return x;
}
void SMTParser::DeclareExpr(std::string name, Expr::Width w) {
// for now, only allow variables which are multiples of 8
if (w % 8 != 0) {
cout << "BitVec not multiple of 8 (" << w << "). Need to update code.\n";
exit(1);
}
std::cout << "Declaring " << name << " of width " << w << "\n";
Array *arr = new Array(name, w / 8);
ref<Expr> *kids = new ref<Expr>[w/8];
for (unsigned i=0; i < w/8; i++)
kids[i] = ReadExpr::create(UpdateList(arr, NULL),
ConstantExpr::create(i, 32));
ref<Expr> var = ConcatExpr::createN(w/8, kids);
delete [] kids;
AddVar(name, var);
}
ExprHandle SMTParser::GetConstExpr(std::string val, uint8_t base, klee::Expr::Width w) {
cerr << "In GetConstExpr(): val=" << val << ", base=" << (unsigned)base << ", width=" << w << "\n";
assert(base == 2 || base == 10 || base == 16);
llvm::APInt ap(w, val.c_str(), val.length(), base);
return klee::ConstantExpr::alloc(ap);
}
void SMTParser::PushVarEnv() {
cout << "Pushing new var env\n";
varEnvs.push(VarEnv(varEnvs.top()));
}
void SMTParser::PopVarEnv() {
cout << "Popping var env\n";
varEnvs.pop();
}
void SMTParser::AddVar(std::string name, ExprHandle val) {
cout << "Adding (" << name << ", " << val << ") to current var env.\n";
varEnvs.top()[name] = val;
}
ExprHandle SMTParser::GetVar(std::string name) {
VarEnv top = varEnvs.top();
if (top.find(name) == top.end()) {
std::cerr << "Cannot find variable ?" << name << "\n";
exit(1);
}
return top[name];
}
void SMTParser::PushFVarEnv() {
fvarEnvs.push(FVarEnv(fvarEnvs.top()));
}
void SMTParser::PopFVarEnv(void) {
cout << "Popping fvar env\n";
fvarEnvs.pop();
}
void SMTParser::AddFVar(std::string name, ExprHandle val) {
cout << "Adding (" << name << ", " << val << ") to current fvar env.\n";
fvarEnvs.top()[name] = val;
}
ExprHandle SMTParser::GetFVar(std::string name) {
FVarEnv top = fvarEnvs.top();
if (top.find(name) == top.end()) {
std::cerr << "Cannot find fvar $" << name << "\n";
exit(1);
}
return top[name];
}
|
