tests/bionic/libc/common/test_pthread_cleanup_push.c

   1 /*
   2  * Copyright (C) 2008 The Android Open Source Project
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  *  * Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  *  * Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in
  12  *    the documentation and/or other materials provided with the
  13  *    distribution.
  14  *
  15  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  16  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  17  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  18  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
  19  * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
  20  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
  21  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
  22  * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
  23  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  24  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
  25  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  26  * SUCH DAMAGE.
  27  */
  28 #include <pthread.h>
  29 #include <stdio.h>
  30 #include <stdlib.h>
  31
  32
  33 #define MAGIC1 0xcafebabeU
  34 #define MAGIC2 0x8badf00dU
  35 #define MAGIC3 0x12345667U
  36
  37 static int g_ok1 = 0;
  38 static int g_ok2 = 0;
  39 static int g_ok3 = 0;
  40
  41 static void
  42 cleanup1( void* arg )
  43 {
  44     if ((unsigned)arg != MAGIC1)
  45         g_ok1 = -1;
  46     else
  47         g_ok1 = +1;
  48 }
  49
  50 static void
  51 cleanup2( void* arg )
  52 {
  53     if ((unsigned)arg != MAGIC2) {
  54         g_ok2 = -1;
  55     } else
  56         g_ok2 = +1;
  57 }
  58
  59 static void
  60 cleanup3( void* arg )
  61 {
  62     if ((unsigned)arg != MAGIC3)
  63         g_ok3 = -1;
  64     else
  65         g_ok3 = +1;
  66 }
  67
  68
  69 static void*
  70 thread1_func( void* arg )
  71 {
  72     pthread_cleanup_push( cleanup1, (void*)MAGIC1 );
  73     pthread_cleanup_push( cleanup2, (void*)MAGIC2 );
  74     pthread_cleanup_push( cleanup3, (void*)MAGIC3 );
  75
  76     if (arg != NULL)
  77         pthread_exit(0);
  78
  79     pthread_cleanup_pop(0);
  80     pthread_cleanup_pop(1);
  81     pthread_cleanup_pop(1);
  82
  83     return NULL;
  84 }
  85
  86 static int test( int do_exit )
  87 {
  88     pthread_t t;
  89
  90     pthread_create( &t, NULL, thread1_func, (void*)do_exit );
  91     pthread_join( t, NULL );
  92
  93     if (g_ok1 != +1) {
  94         if (g_ok1 == 0) {
  95             fprintf(stderr, "cleanup1 not called !!\n");
  96         } else {
  97             fprintf(stderr, "cleanup1 called with wrong argument\n" );
  98         }
  99         exit(1);
 100     }
 101     else if (g_ok2 != +1) {
 102         if (g_ok2 == 0)
 103             fprintf(stderr, "cleanup2 not called !!\n");
 104         else
 105             fprintf(stderr, "cleanup2 called with wrong argument\n");
 106         exit(2);
 107     }
 108     else if (do_exit && g_ok3 != +1) {
 109         if (g_ok3 == 0) {
 110             fprintf(stderr, "cleanup3 not called !!\n");
 111         } else {
 112             fprintf(stderr, "cleanup3 called with bad argument !!\n");
 113         }
 114         exit(3);
 115     }
 116     else if (!do_exit && g_ok3 != 0) {
 117         if (g_ok3 == 1) {
 118             fprintf(stderr, "cleanup3 wrongly called !!\n");
 119         } else {
 120             fprintf(stderr, "cleanup3 wrongly called with bad argument !!\n");
 121         }
 122         exit(3);
 123     }
 124
 125     return 0;
 126 }
 127
 128 int main( void )
 129 {
 130     test(0);
 131     test(1);
 132     printf("OK\n");
 133     return 0;
 134 }