dyld-43.tar.gz
[apple/dyld.git] / unit-tests / test-cases / bundle-memory-load / 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 <stdbool.h>
25 #include <mach-o/dyld.h>
26 #include <sys/types.h>
27 #include <sys/stat.h>
28 #include <sys/mman.h>
29 #include <unistd.h>
30 #include <fcntl.h>
31
32 #include "test.h" // PASS(), FAIL()
33
34 typedef bool (*CheckFunc)();
35
36 int main()
37 {
38 int fd = open("test.bundle", O_RDONLY, 0);
39 if ( fd == -1 ) {
40 FAIL("open() failed");
41 return 1;
42 }
43
44 struct stat stat_buf;
45 if ( fstat(fd, &stat_buf) == -1) {
46 FAIL("fstat() failed");
47 return 1;
48 }
49
50 void* loadAddress = mmap(NULL, stat_buf.st_size, PROT_READ, MAP_FILE | MAP_PRIVATE, fd, 0);
51 if ( loadAddress == ((void*)(-1)) ) {
52 FAIL("mmap() failed");
53 return 1;
54 }
55
56 close(fd);
57
58 NSObjectFileImage ofi;
59 if ( NSCreateObjectFileImageFromMemory(loadAddress, stat_buf.st_size, &ofi) != NSObjectFileImageSuccess ) {
60 FAIL("NSCreateObjectFileImageFromMemory failed");
61 return 1;
62 }
63
64 NSModule mod = NSLinkModule(ofi, "he_he", NSLINKMODULE_OPTION_NONE);
65 if ( mod == NULL ) {
66 FAIL("NSLinkModule failed");
67 return 1;
68 }
69
70 NSSymbol sym = NSLookupSymbolInModule(mod, "_checkdata");
71 if ( sym == NULL ) {
72 FAIL("NSLookupSymbolInModule failed");
73 return 1;
74 }
75
76 CheckFunc func = NSAddressOfSymbol(sym);
77 if ( !func() ) {
78 FAIL("NSAddressOfSymbol failed");
79 return 1;
80 }
81
82 if ( !NSUnLinkModule(mod, NSUNLINKMODULE_OPTION_NONE) ) {
83 FAIL("NSUnLinkModule failed");
84 return 1;
85 }
86
87 if ( !NSDestroyObjectFileImage(ofi) ) {
88 FAIL("NSDestroyObjectFileImage failed");
89 return 1;
90 }
91
92 // Should check that loadAddress is unmmaped now (by call to NSDestroyObjectFileImage)
93
94 PASS("bundle-memory-load");
95 return 0;
96 }