1 files changed, 69 insertions, 0 deletions
diff --git a/lib/Solver/KQueryLoggingSolver.cpp b/lib/Solver/KQueryLoggingSolver.cpp
new file mode 100644
index 00000000..a0ac950c
--- /dev/null
+++ b/lib/Solver/KQueryLoggingSolver.cpp
@@ -0,0 +1,69 @@
+//===-- KQueryLoggingSolver.cpp -----------------------------------------------===//
+//
+//                     The KLEE Symbolic Virtual Machine
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+
+#include "QueryLoggingSolver.h"
+
+#include "klee/Expr.h"
+#include "klee/util/ExprPPrinter.h"
+#include "klee/Internal/Support/QueryLog.h"
+
+using namespace klee;
+
+///
+
+class KQueryLoggingSolver : public QueryLoggingSolver {
+
+private :
+    ExprPPrinter *printer;
+
+    virtual void printQuery(const Query& query,
+                            const Query* falseQuery = 0,
+                            const std::vector<const Array*>* objects = 0) {
+
+        const ref<Expr>* evalExprsBegin = 0;
+        const ref<Expr>* evalExprsEnd = 0;
+
+        if (0 != falseQuery) {
+            evalExprsBegin = &query.expr;
+            evalExprsEnd = &query.expr + 1;
+        }
+
+        const Array* const *evalArraysBegin = 0;
+        const Array* const *evalArraysEnd = 0;
+
+        if ((0 != objects) && (false == objects->empty())) {
+            evalArraysBegin = &((*objects)[0]);
+            evalArraysEnd = &((*objects)[0]) + objects->size();
+        }
+
+        const Query* q = (0 == falseQuery) ? &query : falseQuery;
+
+        printer->printQuery(logBuffer, q->constraints, q->expr,
+                            evalExprsBegin, evalExprsEnd,
+                            evalArraysBegin, evalArraysEnd);
+    }
+
+public:
+    KQueryLoggingSolver(Solver *_solver, std::string path, int queryTimeToLog)
+    : QueryLoggingSolver(_solver, path, "#", queryTimeToLog),
+    printer(ExprPPrinter::create(logBuffer)) {
+    }
+
+    virtual ~KQueryLoggingSolver() {
+        delete printer;
+    }
+};
+
+///
+
+Solver *klee::createKQueryLoggingSolver(Solver *_solver, std::string path,
+                                    int minQueryTimeToLog) {
+  return new Solver(new KQueryLoggingSolver(_solver, path, minQueryTimeToLog));
+}
+