source: src/tests/sched-int-block.c @ b1d4d60

#include <fstream>
#include <kernel>
#include <monitor>
#include <stdlib>
#include <thread>

static const unsigned N = 100_000;

enum state_t { WAITED, SIGNAL, BARGE };

monitor global_data_t {
        thread_desc * last_thread;
        thread_desc * last_signaller;
};

void ?{} ( global_data_t * this ) {
        this->last_thread = NULL;
        this->last_signaller = NULL;
}

void ^?{} ( global_data_t * this ) {}

global_data_t globalA, globalB;

condition cond;

volatile bool done;

//------------------------------------------------------------------------------
void wait_op( global_data_t * mutex a, global_data_t * mutex b, unsigned i ) {
        wait( &cond, (uintptr_t)this_thread() );

        yield( ((unsigned)rand48()) % 10 );

        if(a->last_thread != a->last_signaller || b->last_thread != b->last_signaller ) {
                sout | "ERROR Barging detected, expected" | a->last_signaller | b->last_signaller | "got" | a->last_thread | b->last_thread | endl;
                abort();
        }

        a->last_thread = b->last_thread = this_thread();

        yield( ((unsigned)rand48()) % 10 );
}

thread Waiter {};
void main( Waiter* this ) {
        for( int i = 0; i < N; i++ ) {
                wait_op( &globalA, &globalB, i );
        }
}

//------------------------------------------------------------------------------
void signal_op( global_data_t * mutex a, global_data_t * mutex b ) {
        yield( ((unsigned)rand48()) % 10 );

        a->last_thread = b->last_thread = a->last_signaller = b->last_signaller = this_thread();

        if( !is_empty( &cond ) ) {

                thread_desc * next = front( &cond );

                if( ! signal_block( &cond ) ) {
                        sout | "ERROR expected to be able to signal" | endl;
                        abort();
                }

                yield( ((unsigned)rand48()) % 10 );

                if(a->last_thread != next || b->last_thread != next) {
                        sout | "ERROR Barging detected, expected" | next | "got" | a->last_thread | b->last_thread | endl;
                        abort();
                }
        }

}

thread Signaller {};
void main( Signaller* this ) {
        while( !done ) {
                signal_op( &globalA, &globalB );
        }
}

//------------------------------------------------------------------------------
void barge_op( global_data_t * mutex a ) {
        a->last_thread = this_thread();
}

thread Barger {};
void main( Barger* this ) {
        for( unsigned i = 0; !done; i++ ) {
                //Choose some monitor to barge into with some irregular pattern
                bool choose_a = (i % 13) > (i % 17);
                barge_op( choose_a ? &globalA : &globalB );
        }
}

//------------------------------------------------------------------------------

int main(int argc, char* argv[]) {
        rand48seed(0);
        done = false;
        processor p;
        {
                Signaller s[4];
                Barger b[13];
                sout | "Starting waiters" | endl;
                {
                        Waiter w[3];
                }
                sout | "Waiters done" | endl;
                done = true;
        }
}