Subversion Repositories openrisc

//==========================================================================

//

//        pthread3.cxx

//

//        POSIX pthread test 2 - cancellation

//

//==========================================================================

// ####ECOSGPLCOPYRIGHTBEGIN####                                            

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

// This file is part of eCos, the Embedded Configurable Operating System.   

// Copyright (C) 1998, 1999, 2000, 2001, 2002 Free Software Foundation, Inc.

//

// eCos is free software; you can redistribute it and/or modify it under    

// the terms of the GNU General Public License as published by the Free     

// Software Foundation; either version 2 or (at your option) any later      

// version.                                                                 

//

// eCos is distributed in the hope that it will be useful, but WITHOUT      

// ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or    

// FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License    

// for more details.                                                        

//

// You should have received a copy of the GNU General Public License        

// along with eCos; if not, write to the Free Software Foundation, Inc.,    

// 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.            

//

// As a special exception, if other files instantiate templates or use      

// macros or inline functions from this file, or you compile this file      

// and link it with other works to produce a work based on this file,       

// this file does not by itself cause the resulting work to be covered by   

// the GNU General Public License. However the source code for this file    

// must still be made available in accordance with section (3) of the GNU   

// General Public License v2.                                               

//

// This exception does not invalidate any other reasons why a work based    

// on this file might be covered by the GNU General Public License.         

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

// ####ECOSGPLCOPYRIGHTEND####                                              

//==========================================================================

//#####DESCRIPTIONBEGIN####

//

// Author(s):     nickg

// Contributors:  nickg, jlarmour

// Date:          2000-04-10

// Description:   Tests POSIX cancellation.

//

//####DESCRIPTIONEND####

//==========================================================================

#include <cyg/infra/testcase.h>

#include <pkgconf/posix.h>

#ifndef CYGPKG_POSIX_PTHREAD

#define NA_MSG "POSIX threads not enabled"

#endif

#ifdef NA_MSG

void

cyg_start(void)

{

    CYG_TEST_INIT();

    CYG_TEST_NA(NA_MSG);

}

#else

#include <sys/types.h>

#include <pthread.h>

#include <unistd.h> // sleep()

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

// Thread info

#define NTHREADS 3

char thread_stack[NTHREADS][PTHREAD_STACK_MIN*2];

pthread_t thread[NTHREADS];

void *pthread_entry1( void *arg);

void *pthread_entry2( void *arg);

void *pthread_entry3( void *arg);

void *(*pthread_entry[NTHREADS])(void *) =

{

    pthread_entry1,

    pthread_entry2,

    pthread_entry3

};

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

volatile cyg_bool cancel_handler1_called = false;

volatile cyg_bool cancel_handler2_called = false;

volatile cyg_bool cancel_handler3_called = false;

volatile cyg_bool thread_ready[NTHREADS];

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

void cancel_handler1( void * arg )

{

    CYG_TEST_INFO( "cancel_handler1 called" );

    CYG_TEST_CHECK( (long)arg == 0x12340000, "cancel_handler1: bad arg value");

    cancel_handler1_called = true;

}

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

void cancel_handler2( void * arg )

{

    CYG_TEST_INFO( "cancel_handler2 called" );

    CYG_TEST_CHECK( (long)arg == 0xFFFF1111, "cancel_handler2: bad arg value");

    cancel_handler2_called = true;

}

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

void cancel_handler3( void * arg )

{

    CYG_TEST_INFO( "cancel_handler3 called" );

    CYG_TEST_CHECK( (long)arg == 0x12340001, "cancel_handler3: bad arg value");

    cancel_handler3_called = true;

}

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

void function1(void)

{

    pthread_cleanup_push( cancel_handler2, (void *)0xFFFF1111 );

    for(;;)

    {

        sched_yield();

        pthread_testcancel();

    }

    pthread_cleanup_pop( 0 );

}

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

void *pthread_entry1( void *arg)

{

    int retval = 1;

    CYG_TEST_INFO( "pthread_entry1 entered");

    pthread_setcanceltype( PTHREAD_CANCEL_DEFERRED, NULL );

    pthread_cleanup_push( cancel_handler1, arg );

    thread_ready[0] = true;

    function1();

    pthread_cleanup_pop( 0 );

    pthread_exit( (void *)retval );

}

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

void *pthread_entry2( void *arg)

{

    int retval = 1;

    CYG_TEST_INFO( "pthread_entry2 entered");

    pthread_setcanceltype( PTHREAD_CANCEL_ASYNCHRONOUS, NULL );

    pthread_cleanup_push( cancel_handler3, arg );

    thread_ready[1] = true;

    for(;;) sched_yield();

    pthread_cleanup_pop( 0 );

    pthread_exit( (void *)retval );

}

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

void *pthread_entry3( void *arg)

{

    int retval = 1;

    CYG_TEST_INFO( "pthread_entry3 entered");

    pthread_setcanceltype( PTHREAD_CANCEL_DEFERRED, NULL );

    thread_ready[2] = true;

    // stop in a cancellation point

    sleep( 99999 );

    pthread_exit( (void *)retval );

}

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

int main(int argc, char **argv)

{

    int i, j;

    int ret;

    void *retval[NTHREADS];

    CYG_TEST_INIT();

    // Create test threads

    for( i = 0; i < NTHREADS; i++ )

    {

        pthread_attr_t attr;

        pthread_attr_init( &attr );

        pthread_attr_setstackaddr( &attr, (void *)&thread_stack[i][sizeof(thread_stack[i])] );

        pthread_attr_setstacksize( &attr, sizeof(thread_stack[i]) );

        ret = pthread_create( &thread[i],

                              &attr,

                              pthread_entry[i],

                              (void *)(0x12340000+i));

        CYG_TEST_CHECK( ret == 0, "pthread_create() returned error");

    }

    // Let the threads get going    

    for ( i = 0; i < NTHREADS ; i++ ) {

        while ( thread_ready[i] == false )

            sched_yield();

    }

    // Now wait a bit to be sure that the other threads have reached

    // their cancellation points.

    for ( j = 0; j < 20 ; j++ )

        sched_yield();

    // Now cancel them

    for( i = 0; i < NTHREADS; i++ )

        pthread_cancel( thread[i] );

    // Now join with threads

    for( i = 0; i < NTHREADS; i++ )

        pthread_join( thread[i], &retval[i] );

    // check retvals

    for( i = 0; i < NTHREADS; i++ )

        CYG_TEST_CHECK( retval[i] == PTHREAD_CANCELED,

                        "thread didn't exit with PTHREAD_CANCELED" );

    CYG_TEST_CHECK( cancel_handler1_called, "cancel_handler1 not called" );

    CYG_TEST_CHECK( cancel_handler2_called, "cancel_handler2 not called" );

    CYG_TEST_CHECK( cancel_handler3_called, "cancel_handler3 not called" );

    CYG_TEST_PASS_FINISH( "pthread3" );

}

#endif

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

// end of pthread3.c