Subversion Repositories openrisc_me

//==========================================================================
//
//        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 Red Hat, 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.,
// 59 Temple Place, Suite 330, Boston, MA 02111-1307 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.
//
// This exception does not invalidate any other reasons why a work based on
// this file might be covered by the GNU General Public License.
//
// Alternative licenses for eCos may be arranged by contacting Red Hat, Inc.
// at http://sources.redhat.com/ecos/ecos-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