2 files changed, 48 insertions, 0 deletions
diff --git a/lib/Support/Time.cpp b/lib/Support/Time.cpp
index be5eaf18..f508e6fa 100644
--- a/lib/Support/Time.cpp
+++ b/lib/Support/Time.cpp
@@ -10,12 +10,43 @@
 #include "klee/Config/Version.h"
 #include "klee/Internal/System/Time.h"
 
+#if LLVM_VERSION_CODE >= LLVM_VERSION(4, 0)
+#include <chrono>
+
+#include <llvm/Support/Chrono.h>
+#else
 #include "llvm/Support/TimeValue.h"
+#endif
+
 #include "llvm/Support/Process.h"
 
 using namespace llvm;
 using namespace klee;
 
+#if LLVM_VERSION_CODE >= LLVM_VERSION(4, 0)
+double util::durationToDouble(std::chrono::nanoseconds dur)
+{
+  return dur.count() / 1e9;
+}
+
+double util::getUserTime() {
+  sys::TimePoint<> now;
+  std::chrono::nanoseconds user, sys;
+
+  sys::Process::GetTimeUsage(now, user, sys);
+
+  return durationToDouble(user);
+}
+
+double util::getWallTime() {
+  return durationToDouble(getWallTimeVal().time_since_epoch());
+}
+
+sys::TimePoint<> util::getWallTimeVal() {
+  return std::chrono::system_clock::now();
+}
+
+#else
 double util::getUserTime() {
   sys::TimeValue now(0,0),user(0,0),sys(0,0);
   sys::Process::GetTimeUsage(now,user,sys);
@@ -30,3 +61,4 @@ double util::getWallTime() {
 sys::TimeValue util::getWallTimeVal() {
   return sys::TimeValue::now();
 }
+#endif
diff --git a/lib/Support/Timer.cpp b/lib/Support/Timer.cpp
index da969810..0e727bb4 100644
--- a/lib/Support/Timer.cpp
+++ b/lib/Support/Timer.cpp
@@ -15,6 +15,20 @@
 using namespace klee;
 using namespace llvm;
 
+#if LLVM_VERSION_CODE >= LLVM_VERSION(4, 0)
+
+WallTimer::WallTimer() {
+  start = util::getWallTimeVal();
+}
+
+uint64_t WallTimer::check() {
+  auto now = util::getWallTimeVal();
+  return std::chrono::duration_cast<std::chrono::microseconds>(now -
+    start).count();
+}
+
+#else
+
 WallTimer::WallTimer() {
   startMicroseconds = util::getWallTimeVal().usec();
 }
@@ -22,3 +36,5 @@ WallTimer::WallTimer() {
 uint64_t WallTimer::check() {
   return util::getWallTimeVal().usec() - startMicroseconds;
 }
+
+#endif