dyld-45.1.tar.gz
[apple/dyld.git] / unit-tests / test-cases / bundle-reload / 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
27 #include "test.h" // PASS(), FAIL()
28
29 ///
30 /// The point of this test is to load and unload a bundle and
31 /// verify that the address->image cache is properly invalidated. rdar://problem/4212667
32 ///
33
34 typedef void (*fooProc)();
35
36 // test.bundle
37 void doit()
38 {
39 NSObjectFileImage ofi;
40 if ( NSCreateObjectFileImageFromFile("test.bundle", &ofi) != NSObjectFileImageSuccess ) {
41 FAIL("NSCreateObjectFileImageFromFile failed");
42 exit(0);
43 }
44
45 NSModule mod = NSLinkModule(ofi,"test.bundle", NSLINKMODULE_OPTION_PRIVATE | NSLINKMODULE_OPTION_BINDNOW | NSLINKMODULE_OPTION_RETURN_ON_ERROR);
46 if ( mod == NULL ) {
47 FAIL("NSLinkModule failed");
48 exit(0);
49 }
50
51 NSSymbol sym = NSLookupSymbolInModule(mod, "_foo");
52 if ( sym == NULL ) {
53 FAIL("NSLookupSymbolInModule failed");
54 exit(0);
55 }
56
57 fooProc func = (fooProc)NSAddressOfSymbol(sym);
58 (*func)();
59
60 if ( !NSUnLinkModule(mod, NSUNLINKMODULE_OPTION_NONE) ) {
61 FAIL("NSUnLinkModule failed");
62 exit(0);
63 }
64
65 if ( !NSDestroyObjectFileImage(ofi) ) {
66 FAIL("NSDestroyObjectFileImage failed");
67 exit(0);
68 }
69 }
70
71
72 static void myRemoveImage(const struct mach_header *mh, intptr_t vmaddr_slide)
73 {
74 // calling _dyld_get_image_header_containing_address() during the remove image hook
75 // could cause dyld to not flush the address->image cache
76 _dyld_get_image_header_containing_address(mh);
77 }
78
79
80 int main()
81 {
82 _dyld_register_func_for_remove_image(&myRemoveImage);
83
84 doit();
85 doit();
86 doit();
87
88 PASS("bundle-reload");
89 return 0;
90 }
91