dyld-43.tar.gz
[apple/dyld.git] / unit-tests / test-cases / deadlock / main.c
1 /*
2 * Copyright (c) 2005 Apple Computer, Inc. All rights reserved.
3 *
4 * @APPLE_LICENSE_HEADER_START@
5 *
6 * This file contains Original Code and/or Modifications of Original Code
7 * as defined in and that are subject to the Apple Public Source License
8 * Version 2.0 (the 'License'). You may not use this file except in
9 * compliance with the License. Please obtain a copy of the License at
10 * http://www.opensource.apple.com/apsl/ and read it before using this
11 * file.
12 *
13 * The Original Code and all software distributed under the License are
14 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
15 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
16 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
17 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
18 * Please see the License for the specific language governing rights and
19 * limitations under the License.
20 *
21 * @APPLE_LICENSE_HEADER_END@
22 */
23 #include <stdio.h>
24 #include <stdlib.h>
25 #include <mach-o/dyld.h>
26 #include <pthread.h>
27
28 #include "test.h"
29
30 /// rdar://problem/3811777
31
32 // barrier thread 1 thread 2
33 // add image
34 // 1
35 // acquire sMyLock
36 // 2
37 // in callback acquire sMyLock call lazy pointer
38 // release sMyLock release sMyLock
39
40 extern void foo();
41
42
43 static volatile int sBarrier = 0;
44 static pthread_mutex_t sBarrierMutex;
45 static pthread_cond_t sBarrierFree;
46
47 static void blockUntilBarrier(int n)
48 {
49 pthread_mutex_lock(&sBarrierMutex);
50 while ( sBarrier < n )
51 pthread_cond_wait(&sBarrierFree, &sBarrierMutex);
52 pthread_mutex_unlock(&sBarrierMutex);
53 }
54
55 static void advanceToBarrier(int n)
56 {
57 pthread_mutex_lock(&sBarrierMutex);
58 sBarrier = n;
59 pthread_cond_broadcast(&sBarrierFree);
60 pthread_mutex_unlock(&sBarrierMutex);
61 }
62
63
64
65
66
67 static pthread_mutex_t sMyLock;
68
69 static void* thread2(void* arg)
70 {
71 // thread 2
72 blockUntilBarrier(1);
73 pthread_mutex_lock(&sMyLock);
74 advanceToBarrier(2);
75 foo();
76 pthread_mutex_unlock(&sMyLock);
77 return NULL;
78 }
79
80
81
82 static void myImageHandler(const struct mach_header *mh, intptr_t vmaddr_slide)
83 {
84 // thread 1
85 if ( NSLookupSymbolInImage(mh, "_bar", 0) != NULL ) {
86 advanceToBarrier(1);
87 blockUntilBarrier(2);
88 pthread_mutex_lock(&sMyLock);
89 pthread_mutex_unlock(&sMyLock);
90 }
91 }
92
93 int main()
94 {
95 pthread_mutex_init(&sBarrierMutex, NULL);
96 pthread_cond_init(&sBarrierFree, NULL);
97 pthread_mutex_init(&sMyLock, NULL);
98
99 // self-terminate this process if it locks up for two seconds
100 alarm(2);
101
102 advanceToBarrier(0);
103
104 pthread_t pthread2;
105 if ( pthread_create(&pthread2, NULL, thread2, NULL) != 0 ) {
106 FAIL("pthread_create failed");
107 exit(0);
108 }
109
110 // thread 1
111 _dyld_register_func_for_add_image(&myImageHandler);
112 NSAddImage("bar.dylib", 0);
113
114 PASS("deadlock");
115 }