1/*
2 * Copyright (c) 2013 Apple 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>  // fprintf(), NULL
24#include <stdlib.h> // exit(), EXIT_SUCCESS
25#include <dlfcn.h>
26
27#include "test.h" // PASS(), FAIL(), XPASS(), XFAIL()
28
29
30
31
32int main()
33{
34	Dl_info info;
35
36	// load foo
37	void* handleFoo = dlopen("libfoo.dylib", RTLD_LAZY);
38	if ( handleFoo == NULL ) {
39		FAIL("dlclose-dylib-dynamic-ref: dlopen(\"libfoo.dylib\", RTLD_LAZY) failed with dlerror()=%s", dlerror());
40		exit(0);
41	}
42
43	// load bar
44	void* handleBar = dlopen("libbar.dylib", RTLD_LAZY);
45	if ( handleBar == NULL ) {
46		FAIL("dlclose-dylib-dynamic-ref: dlopen(\"libbar.dylib\", RTLD_LAZY) failed with dlerror()=%s", dlerror());
47		exit(0);
48	}
49
50	// load baz
51	void* handleBaz = dlopen("libbaz.dylib", RTLD_LAZY);
52	if ( handleBaz == NULL ) {
53		FAIL("dlclose-dylib-dynamic-ref: dlopen(\"libbaz.dylib\", RTLD_LAZY) failed with dlerror()=%s", dlerror());
54		exit(0);
55	}
56
57
58	void* sym_foo = dlsym(handleFoo, "foo");
59	if ( sym_foo == NULL ) {
60		FAIL("dlclose-dylib-dynamic-ref: dlsym(handleFoo, \"foo\") failed");
61		exit(0);
62	}
63
64	void* sym_bar = dlsym(handleBar, "bar");
65	if ( sym_bar == NULL ) {
66		FAIL("dlclose-dylib-dynamic-ref: dlsym(handleBar, \"bar\") failed");
67		exit(0);
68	}
69
70	void* sym_baz = dlsym(handleBaz, "baz");
71	if ( sym_baz == NULL ) {
72		FAIL("dlclose-dylib-dynamic-ref: dlsym(handleBaz, \"baz\") failed");
73		exit(0);
74	}
75
76	// since foo loaded first, bar and baz should have a dynamic reference to foo because all
77	// the weak mydata symbols were coaleseced to the one in foo.
78
79
80	if ( dlclose(handleFoo) != 0 ) {
81		FAIL("dlclose-dylib-dynamic-ref: dlclose(handleBar) != 0, dlerrr()=%s", dlerror());
82		exit(0);
83	}
84
85	// sym_foo should still be accessible via dladdr() because libbar and libbaz
86	// have dynamic references to libfoo.
87	if ( dladdr(sym_foo, &info) == 0 ) {
88		FAIL("dlclose-dylib-dynamic-ref: dladdr(sym_base) == 0, but should have succeeded");
89		exit(0);
90	}
91
92
93	if ( dlclose(handleBar) != 0 ) {
94		FAIL("dlclose-dylib-dynamic-ref: dlclose(handleBar) != 0, dlerrr()=%s", dlerror());
95		exit(0);
96	}
97
98
99	// sym_bar should not be accessible via dladdr() because libbar was dlclose'ed
100	if ( dladdr(sym_bar, &info) != 0 ) {
101		FAIL("dlclose-dylib-dynamic-ref: dladdr(sym_bar) != 0, but should have failed");
102		exit(0);
103	}
104
105	// sym_foo should still be accessible via dladdr() because of external libbaz
106	// has a dynamic references to libfoo.
107	if ( dladdr(sym_foo, &info) == 0 ) {
108		FAIL("dlclose-dylib-dynamic-ref: dladdr(sym_base) == 0, but should have succeeded");
109		exit(0);
110	}
111
112
113	if ( dlclose(handleBaz) != 0 ) {
114		FAIL("dlclose-dylib-dynamic-ref: dlclose(handleBar) != 0, dlerrr()=%s", dlerror());
115		exit(0);
116	}
117
118	// sym_baz should not be accessible via dladdr() because libbar was dlclose'ed
119	if ( dladdr(sym_baz, &info) != 0 ) {
120		FAIL("dlclose-dylib-dynamic-ref: dladdr(sym_baz) != 0, but should have failed");
121		exit(0);
122	}
123
124
125	// sym_foo should finally be inaccessible via dladdr() because all dynamic references to libfoo are gone
126	if ( dladdr(sym_foo, &info) != 0 ) {
127		FAIL("dlclose-dylib-dynamic-ref: dladdr(sym_foo) == 0, but should have succeeded");
128		exit(0);
129	}
130
131
132	PASS("dlclose-dylib-dynamic-ref");
133	return EXIT_SUCCESS;
134}
135