squash: enforce true

emqx · May 17, 2024 · 68a8d9d · 68a8d9d
1 parent 58e099d
commit 68a8d9d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/c_src/quicer_nif.c b/c_src/quicer_nif.c
@@ -1365,8 +1365,9 @@ controlling_process(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[])
   ERL_NIF_TERM new_owner = argv[1];
   ERL_NIF_TERM res = ATOM_OK;
   BOOLEAN is_locked = FALSE;
-  if (argc == 3)
+  if (argc == 3 || IS_SAME_TERM(argv[2], ATOM_TRUE))
     {
+      // give hint to this call that stream mutex is locked.
       is_locked = TRUE;
     }
 

diff --git a/src/quicer.erl b/src/quicer.erl
@@ -1154,7 +1154,7 @@ handoff_stream(Stream, NewOwner, HandoffData) ->
             Res =
                 case forward_stream_msgs(Stream, NewOwner) of
                     ok ->
-                        case quicer_nif:controlling_process(Stream, NewOwner, unlock) of
+                        case quicer_nif:controlling_process(Stream, NewOwner, true) of
                             {error, _} ->
                                 quicer_nif:unlock_stream(Stream);
                             ok ->