Fixes calls to timer:sleep(infinity)

Signed-off-by: Fred Dushin <[email protected]>

Author: fadushin

libs/estdlib/src/timer.erl

@@ -22,18 +22,9 @@
 
 -export([sleep/1]).
 
--define(MAX_INT, 4294967295).
-
 -spec sleep(non_neg_integer() | infinity) -> ok.
-sleep(infinity) ->
-    receive
-        '$atomvm_timer_interrupt' ->
-            {error, unexpected_interrupt}
-    after ?MAX_INT -> sleep(infinity)
-    end;
-sleep(MSecs) ->
+sleep(Timeout) ->
     receive
-        '$atomvm_timer_interrupt' ->
-            {error, unexpected_interrupt}
-    after MSecs -> ok
+    after Timeout ->
+        ok
     end.

src/libAtomVM/opcodesswitch.h

@@ -2120,7 +2120,12 @@ static bool maybe_call_native(Context *ctx, AtomString module_name, AtomString f
                 DECODE_COMPACT_TERM(timeout, code, i, next_off)
 
                 #ifdef IMPL_EXECUTE_LOOP
-                    if (!term_is_integer(timeout) && UNLIKELY(timeout != INFINITY_ATOM)) {
+                    if (term_is_any_integer(timeout)) {
+                        avm_int64_t t = term_maybe_unbox_int64(timeout);
+                        if (t < 0 || t > 4294967295L) {
+                            RAISE_ERROR(TIMEOUT_VALUE_ATOM);
+                        }
+                    } else if (UNLIKELY(timeout != INFINITY_ATOM)) {
                         RAISE_ERROR(TIMEOUT_VALUE_ATOM);
                     }
                     TRACE("wait_timeout/2, label: %i, timeout: %li\n", label, (long int) term_to_int32(timeout));
@@ -2137,7 +2142,7 @@ static bool maybe_call_native(Context *ctx, AtomString module_name, AtomString f
                             scheduler_set_timeout(ctx, term_to_int32(timeout));
                         }
                         needs_to_wait = 1;
-                    } else if ((ctx->flags & WaitingTimeout) == 0) {
+                    } else if ((ctx->flags & WaitingTimeout) != 0) {
                         needs_to_wait = 1;
                     } else if (!list_is_empty(&ctx->save_queue)) {
                         needs_to_wait = 1;

tests/libs/estdlib/test_timer.erl

@@ -22,65 +22,62 @@
 
 -export([test/0]).
 
+-include("etest.hrl").
+
 test() ->
     ok = test_timer(),
-    ok = test_timer_interrupt(),
     ok = test_timer_loop(),
+    ok = test_timer_badargs(),
+    ok = test_infinity(),
     ok.
 
--include("etest.hrl").
-
 test_timer() ->
-    T0 = erlang:timestamp(),
+    T0 = erlang:system_time(millisecond),
     ok = timer:sleep(101),
-    T1 = erlang:timestamp(),
-    ok = etest:assert_true((to_ms(T1) - to_ms(T0)) >= 101),
+    T1 = erlang:system_time(millisecond),
+    ok = etest:assert_true((T1 - T0) >= 101),
     ok.
 
-test_timer_interrupt() ->
-    Self = self(),
-    Pid = spawn(fun() -> do_test_interrupt(Self) end),
-    receive
-        ready -> ok
-    end,
-
-    %% this message should not interrupt the timer
-    Pid ! try_to_interrupt,
-
-    Pid ! '$atomvm_timer_interrupt',
-    ?ASSERT_MATCH(pop_mailbox(), ok),
-    ok.
-
-pop_mailbox() ->
-    receive
-        X -> X
-    end.
-
-do_test_interrupt(Pid) ->
-    Pid ! ready,
-    case timer:sleep(infinity) of
-        {error, unexpected_interrupt} ->
-            Pid ! ok;
-        _ ->
-            Pid ! error
-    end.
-
 test_timer_loop() ->
     Self = self(),
     spawn(fun() ->
-        timer:sleep(220),
-        Self ! ping
+        Self ! ready,
+        timer:sleep(50),
+        Self ! noise
     end),
+    receive
+        ready ->
+            ok
+    end,
     ok = timer_loop(5).
 
 timer_loop(0) ->
-    ok;
+    receive
+        noise ->
+            ok;
+        SomethingElse ->
+            {error, SomethingElse}
+    end;
 timer_loop(I) ->
-    T0 = erlang:timestamp(),
+    T0 = erlang:system_time(millisecond),
     ok = timer:sleep(101),
-    T1 = erlang:timestamp(),
-    ok = etest:assert_true((to_ms(T1) - to_ms(T0)) >= 101),
+    T1 = erlang:system_time(millisecond),
+    ok = etest:assert_true((T1 - T0) >= 101),
     timer_loop(I - 1).
 
-to_ms({MegaSecs, Secs, MicroSecs}) ->
-    ((MegaSecs * 1000000 + Secs) * 1000 + MicroSecs div 1000).
+test_timer_badargs() ->
+    {'EXIT', {timeout_value, _}} = (catch timer:sleep(-1)),
+    {'EXIT', {timeout_value, _}} = (catch timer:sleep(4294967295 + 1)),
+    {'EXIT', {timeout_value, _}} = (catch timer:sleep(not_infinity)),
+    ok.
+
+test_infinity() ->
+    Self = self(),
+    Pid = spawn(fun() ->
+        Self ! ok,
+        timer:sleep(infinity)
+    end),
+    receive
+        ok ->
+            ok = etest:assert_true(erlang:is_process_alive(Pid))
+    end.

tests/libs/estdlib/tests.erl

@@ -32,6 +32,6 @@ start() ->
         test_io_lib,
         test_maps,
         test_proplists,
-        % , test_timer TODO: enable it again, once we try a way to make it less flaky
+        test_timer,
         test_supervisor
     ]).