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
|
--- contrib/restricted/wavm/Lib/LLVMJIT/EmitCore.cpp (9f792fa1777224478cd9d5b1bf90810582377ec2)
+++ contrib/restricted/wavm/Lib/LLVMJIT/EmitCore.cpp (index)
@@ -74,6 +74,10 @@ void EmitFunctionContext::loop(ControlStructureImm imm)
irBuilder.CreateBr(loopBodyBlock);
irBuilder.SetInsertPoint(loopBodyBlock);
+ emitRuntimeIntrinsic("throwIfCurrentTimeoutExpired",
+ FunctionType({}, {}, IR::CallingConvention::intrinsic),
+ {});
+
// Push a control context that ends at the end block/phi.
pushControlStack(ControlContext::Type::loop, blockType.results(), endBlock, endPHIs);
--- contrib/restricted/wavm/Lib/Runtime/Intrinsics.cpp (9f792fa1777224478cd9d5b1bf90810582377ec2)
+++ contrib/restricted/wavm/Lib/Runtime/Intrinsics.cpp (index)
@@ -255,3 +255,57 @@ HashMap<std::string, Intrinsics::Function*> Intrinsics::getUninstantiatedFunctio
return result;
}
+
+namespace WAVM { namespace Runtime {
+
+#ifdef __APPLE__
+static constexpr auto WAVM_CLOCK_TYPE = _CLOCK_MONOTONIC_RAW;
+#else
+static constexpr auto WAVM_CLOCK_TYPE = CLOCK_MONOTONIC_COARSE;
+#endif
+
+class Deadline {
+public:
+ bool isDeadlineReached() const {
+ if (!deadline.has_value()) {
+ return false;
+ }
+
+ struct timespec current = {};
+ if (clock_gettime(WAVM_CLOCK_TYPE, ¤t) != 0) {
+ // The simplest thing to do here is to assume that the deadline just hasn't arrived.
+ return false;
+ }
+
+ return current.tv_sec > deadline->tv_sec ||
+ (current.tv_sec == deadline->tv_sec && current.tv_nsec >= deadline->tv_nsec);
+ }
+
+ void setDeadline(std::optional<struct timespec> newDeadline) {
+ deadline = newDeadline;
+ }
+
+private:
+ std::optional<struct timespec> deadline;
+};
+
+static thread_local Deadline currentDeadline;
+
+__attribute__((__noinline__)) void setCurrentDeadline(std::optional<struct timespec> deadline)
+{
+ currentDeadline.setDeadline(deadline);
+}
+
+__attribute__((__noinline__)) bool isCurrentDeadlineReached()
+{
+ return currentDeadline.isDeadlineReached();
+}
+
+struct timespec getInstant()
+{
+ struct timespec current = {};
+ WAVM_ASSERT(clock_gettime(WAVM_CLOCK_TYPE, ¤t) == 0);
+ return current;
+}
+
+}}
--- contrib/restricted/wavm/Lib/Runtime/RuntimePrivate.h (9f792fa1777224478cd9d5b1bf90810582377ec2)
+++ contrib/restricted/wavm/Lib/Runtime/RuntimePrivate.h (index)
@@ -417,3 +417,9 @@ namespace WAVM { namespace Intrinsics {
HashMap<std::string, Function*> getUninstantiatedFunctions(
const std::initializer_list<const Intrinsics::Module*>& moduleRefs);
}}
+
+namespace WAVM { namespace Runtime {
+ void setCurrentDeadline(std::optional<struct timespec> deadline);
+ bool isCurrentDeadlineReached();
+ struct timespec getInstant();
+}}
--- contrib/restricted/wavm/Lib/Runtime/WAVMIntrinsics.cpp (9f792fa1777224478cd9d5b1bf90810582377ec2)
+++ contrib/restricted/wavm/Lib/Runtime/WAVMIntrinsics.cpp (index)
@@ -80,3 +80,19 @@ WAVM_DEFINE_INTRINSIC_FUNCTION(wavmIntrinsics, "debugBreak", void, debugBreak)
{
Log::printf(Log::debug, "================== wavmIntrinsics.debugBreak\n");
}
+
+WAVM_DEFINE_INTRINSIC_FUNCTION(wavmIntrinsics,
+ "throwIfCurrentTimeoutExpired",
+ void,
+ throwIfCurrentTimeoutExpired)
+{
+ static thread_local int counter = 0;
+ static constexpr int COUNTER_CHECK_PERIOD = 8192;
+
+ if (counter == COUNTER_CHECK_PERIOD) {
+ if(isCurrentDeadlineReached()) { throwException(ExceptionTypes::timeoutExpired); }
+ counter = 0;
+ } else {
+ ++counter;
+ }
+}
--- contrib/restricted/wavm/Include/WAVM/Runtime/Runtime.h (9f792fa1777224478cd9d5b1bf90810582377ec2)
+++ contrib/restricted/wavm/Include/WAVM/Runtime/Runtime.h (index)
@@ -182,7 +182,8 @@ namespace WAVM { namespace Runtime {
visit(outOfMemory); \
visit(misalignedAtomicMemoryAccess, WAVM::IR::ValueType::i64); \
visit(waitOnUnsharedMemory, WAVM::IR::ValueType::externref); \
- visit(invalidArgument);
+ visit(invalidArgument); \
+ visit(timeoutExpired);
// Information about a runtime exception.
namespace ExceptionTypes {
|
