Changeset beeff61e for libcfa/src/concurrency/locks.cfa

Timestamp:

May 1, 2023, 4:00:06 PM (13 months ago)

Author:

caparsons <caparson@…>

Branches:

ADT, ast-experimental, master

Children:

73bf7ddc

Parents:

bb7422a

Message:

some cleanup and a bunch of changes to support waituntil statement

File:

• libcfa/src/concurrency/locks.cfa (modified) (16 diffs)

libcfa/src/concurrency/locks.cfa

@@ -79 +79 @@
         // lock is held by some other thread
         if ( owner != 0p && owner != thrd ) {
-                insert_last( blocked_threads, *thrd );
+        select_node node;
+                insert_last( blocked_threads, node );
                 wait_count++;
                 unlock( lock );
                 park( );
-        }
-        // multi acquisition lock is held by current thread
-        else if ( owner == thrd && multi_acquisition ) {
+        return;
+        } else if ( owner == thrd && multi_acquisition ) { // multi acquisition lock is held by current thread
                 recursion_count++;
-                unlock( lock );
-        }
-        // lock isn't held
-        else {
+        } else {  // lock isn't held
                 owner = thrd;
                 recursion_count = 1;
-                unlock( lock );
-        }
+        }
+    unlock( lock );
 }
 
@@ -117 +114 @@
 }
 
-static void pop_and_set_new_owner( blocking_lock & this ) with( this ) {
-        thread$ * t = &try_pop_front( blocked_threads );
-        owner = t;
-        recursion_count = ( t ? 1 : 0 );
-        if ( t ) wait_count--;
-        unpark( t );
+// static void pop_and_set_new_owner( blocking_lock & this ) with( this ) {
+//      thread$ * t = &try_pop_front( blocked_threads );
+//      owner = t;
+//      recursion_count = ( t ? 1 : 0 );
+//      if ( t ) wait_count--;
+//      unpark( t );
+// }
+
+static inline void pop_node( blocking_lock & this ) with( this ) {
+    __handle_waituntil_OR( blocked_threads );
+    select_node * node = &try_pop_front( blocked_threads );
+    if ( node ) {
+        wait_count--;
+        owner = node->blocked_thread;
+        recursion_count = 1;
+        // if ( !node->clause_status || __make_select_node_available( *node ) ) unpark( node->blocked_thread );
+        wake_one( blocked_threads, *node );
+    } else {
+        owner = 0p;
+        recursion_count = 0;
+    }
 }
 
@@ -134 +146 @@
         recursion_count--;
         if ( recursion_count == 0 ) {
-                pop_and_set_new_owner( this );
+                pop_node( this );
         }
         unlock( lock );
@@ -147 +159 @@
         // lock held
         if ( owner != 0p ) {
-                insert_last( blocked_threads, *t );
+                insert_last( blocked_threads, *(select_node *)t->link_node );
                 wait_count++;
-                unlock( lock );
         }
         // lock not held
@@ -156 +167 @@
                 recursion_count = 1;
                 unpark( t );
-                unlock( lock );
-        }
+        }
+    unlock( lock );
 }
 
@@ -167 +178 @@
         size_t ret = recursion_count;
 
-        pop_and_set_new_owner( this );
+        pop_node( this );
+
+    select_node node;
+    active_thread()->link_node = (void *)&node;
         unlock( lock );
+
+    park();
+
         return ret;
 }
@@ -175 +192 @@
         recursion_count = recursion;
 }
+
+// waituntil() support
+bool register_select( blocking_lock & this, select_node & node ) with(this) {
+    lock( lock __cfaabi_dbg_ctx2 );
+        thread$ * thrd = active_thread();
+
+        // single acquisition lock is held by current thread
+        /* paranoid */ verifyf( owner != thrd || multi_acquisition, "Single acquisition lock holder (%p) attempted to reacquire the lock %p resulting in a deadlock.", owner, &this );
+
+    if ( !node.park_counter && ( (owner == thrd && multi_acquisition) || owner == 0p ) ) { // OR special case
+        if ( !__make_select_node_available( node ) ) { // we didn't win the race so give up on registering
+           unlock( lock );
+           return false;
+        }
+    }
+
+        // lock is held by some other thread
+        if ( owner != 0p && owner != thrd ) {
+                insert_last( blocked_threads, node );
+                wait_count++;
+                unlock( lock );
+        return false;
+        } else if ( owner == thrd && multi_acquisition ) { // multi acquisition lock is held by current thread
+                recursion_count++;
+        } else {  // lock isn't held
+                owner = thrd;
+                recursion_count = 1;
+        }
+
+    if ( node.park_counter ) __make_select_node_available( node );
+    unlock( lock );
+    return true;
+}
+
+bool unregister_select( blocking_lock & this, select_node & node ) with(this) {
+    lock( lock __cfaabi_dbg_ctx2 );
+    if ( node`isListed ) {
+        remove( node );
+        wait_count--;
+        unlock( lock );
+        return false;
+    }
+    
+    if ( owner == active_thread() ) {
+        /* paranoid */ verifyf( recursion_count == 1 || multi_acquisition, "Thread %p attempted to unlock owner lock %p in waituntil unregister, which is not recursive but has a recursive count of %zu", active_thread(), &this, recursion_count );
+        // if recursion count is zero release lock and set new owner if one is waiting
+        recursion_count--;
+        if ( recursion_count == 0 ) {
+            pop_node( this );
+        }
+    }
+        unlock( lock );
+    return false;
+}
+
+bool on_selected( blocking_lock & this, select_node & node ) { return true; }
 
 //-----------------------------------------------------------------------------
@@ -311 +384 @@
         int counter( condition_variable(L) & this ) with(this) { return count; }
 
-        static size_t queue_and_get_recursion( condition_variable(L) & this, info_thread(L) * i ) with(this) {
+        static void enqueue_thread( condition_variable(L) & this, info_thread(L) * i ) with(this) {
                 // add info_thread to waiting queue
                 insert_last( blocked_threads, *i );
                 count++;
-                size_t recursion_count = 0;
-                if (i->lock) {
+                // size_t recursion_count = 0;
+                // if (i->lock) {
+                //      // if lock was passed get recursion count to reset to after waking thread
+                //      recursion_count = on_wait( *i->lock );
+                // }
+                // return recursion_count;
+        }
+
+    static size_t block_and_get_recursion( info_thread(L) & i ) {
+        size_t recursion_count = 0;
+                if ( i.lock ) {
                         // if lock was passed get recursion count to reset to after waking thread
-                        recursion_count = on_wait( *i->lock );
-                }
-                return recursion_count;
-        }
+                        recursion_count = on_wait( *i.lock ); // this call blocks
+                } else park( );
+        return recursion_count;
+    }
 
         // helper for wait()'s' with no timeout
         static void queue_info_thread( condition_variable(L) & this, info_thread(L) & i ) with(this) {
                 lock( lock __cfaabi_dbg_ctx2 );
-                size_t recursion_count = queue_and_get_recursion(this, &i);
+        enqueue_thread( this, &i );
+                // size_t recursion_count = queue_and_get_recursion( this, &i );
                 unlock( lock );
 
                 // blocks here
-                park( );
+        size_t recursion_count = block_and_get_recursion( i );
+                // park( );
 
                 // resets recursion count here after waking
-                if (i.lock) on_wakeup(*i.lock, recursion_count);
+                if ( i.lock ) on_wakeup( *i.lock, recursion_count );
         }
 
@@ -343 +427 @@
         static void queue_info_thread_timeout( condition_variable(L) & this, info_thread(L) & info, Duration t, Alarm_Callback callback ) with(this) {
                 lock( lock __cfaabi_dbg_ctx2 );
-                size_t recursion_count = queue_and_get_recursion(this, &info);
+        enqueue_thread( this, &info );
+                // size_t recursion_count = queue_and_get_recursion( this, &info );
                 alarm_node_wrap(L) node_wrap = { t, 0`s, callback, &this, &info };
                 unlock( lock );
@@ -351 +436 @@
 
                 // blocks here
-                park();
+        size_t recursion_count = block_and_get_recursion( info );
+                // park();
 
                 // unregisters alarm so it doesn't go off if this happens first
@@ -357 +443 @@
 
                 // resets recursion count here after waking
-                if (info.lock) on_wakeup(*info.lock, recursion_count);
+                if ( info.lock ) on_wakeup( *info.lock, recursion_count );
         }
 
@@ -417 +503 @@
                 info_thread( L ) i = { active_thread(), info, &l };
                 insert_last( blocked_threads, i );
-                size_t recursion_count = on_wait( *i.lock );
-                park( );
+                size_t recursion_count = on_wait( *i.lock ); // blocks here
+                // park( );
                 on_wakeup(*i.lock, recursion_count);
         }
@@ -459 +545 @@
         bool empty ( pthread_cond_var(L) & this ) with(this) { return blocked_threads`isEmpty; }
 
-        static size_t queue_and_get_recursion( pthread_cond_var(L) & this, info_thread(L) * i ) with(this) {
-                // add info_thread to waiting queue
-                insert_last( blocked_threads, *i );
-                size_t recursion_count = 0;
-                recursion_count = on_wait( *i->lock );
-                return recursion_count;
-        }
+        // static size_t queue_and_get_recursion( pthread_cond_var(L) & this, info_thread(L) * i ) with(this) {
+        //      // add info_thread to waiting queue
+        //      insert_last( blocked_threads, *i );
+        //      size_t recursion_count = 0;
+        //      recursion_count = on_wait( *i->lock );
+        //      return recursion_count;
+        // }
+
         
         static void queue_info_thread_timeout( pthread_cond_var(L) & this, info_thread(L) & info, Duration t, Alarm_Callback callback ) with(this) {
                 lock( lock __cfaabi_dbg_ctx2 );
-                size_t recursion_count = queue_and_get_recursion(this, &info);
+                // size_t recursion_count = queue_and_get_recursion(this, &info);
+        insert_last( blocked_threads, info );
                 pthread_alarm_node_wrap(L) node_wrap = { t, 0`s, callback, &this, &info };
                 unlock( lock );
@@ -477 +565 @@
 
                 // blocks here
-                park();
+        size_t recursion_count = block_and_get_recursion( info );
+                // park();
 
                 // unregisters alarm so it doesn't go off if this happens first
@@ -483 +572 @@
 
                 // resets recursion count here after waking
-                if (info.lock) on_wakeup(*info.lock, recursion_count);
+                if ( info.lock ) on_wakeup( *info.lock, recursion_count );
         }
 
@@ -493 +582 @@
                 lock( lock __cfaabi_dbg_ctx2 );
                 info_thread( L ) i = { active_thread(), info, &l };
-                size_t recursion_count = queue_and_get_recursion(this, &i);
-                unlock( lock );
-                park( );
-                on_wakeup(*i.lock, recursion_count);
+        insert_last( blocked_threads, i );
+                // size_t recursion_count = queue_and_get_recursion( this, &i );
+                unlock( lock );
+
+        // blocks here
+                size_t recursion_count = block_and_get_recursion( i );
+                // park();
+                on_wakeup( *i.lock, recursion_count );
         }