Root/
Source at commit HEAD created 5 years 6 days ago. By ifabio, Few update to kernelPatcher (Credits to CrazyBirdy) | |
---|---|
1 | /*␊ |
2 | * Copyright (c) 2003 Apple Computer, Inc. All rights reserved.␊ |
3 | *␊ |
4 | * @APPLE_LICENSE_HEADER_START@␊ |
5 | * ␊ |
6 | * The contents of this file constitute Original Code as defined in and␊ |
7 | * are subject to the Apple Public Source License Version 2.0 (the␊ |
8 | * "License"). You may not use this file except in compliance with the␊ |
9 | * License. Please obtain a copy of the License at␊ |
10 | * http://www.apple.com/publicsource and read it before using this file.␊ |
11 | * ␊ |
12 | * This Original Code and all software distributed under the License are␊ |
13 | * distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY KIND, EITHER␊ |
14 | * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,␊ |
15 | * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,␊ |
16 | * FITNESS FOR A PARTICULAR PURPOSE OR NON-INFRINGEMENT. Please see the␊ |
17 | * License for the specific language governing rights and limitations␊ |
18 | * under the License.␊ |
19 | * ␊ |
20 | * @APPLE_LICENSE_HEADER_END@␊ |
21 | *␊ |
22 | * load.c - Functions for decoding a Mach-o Kernel.␊ |
23 | *␊ |
24 | * Copyright (c) 1998-2003 Apple Computer, Inc.␊ |
25 | *␊ |
26 | */␊ |
27 | ␊ |
28 | #include <mach-o/fat.h>␊ |
29 | #include <mach-o/loader.h>␊ |
30 | #include <mach/machine/thread_status.h>␊ |
31 | ␊ |
32 | #include <sl.h>␊ |
33 | ␊ |
34 | #if DEBUG␊ |
35 | ␉#define DBG(x...)␉printf(x)␊ |
36 | #else␊ |
37 | ␉#define DBG(x...)␉msglog(x)␊ |
38 | #endif␊ |
39 | ␊ |
40 | // Private functions.␊ |
41 | static long DecodeSegment(long cmdBase, unsigned int*load_addr, unsigned int *load_size);␊ |
42 | static long DecodeUnixThread(long cmdBase, unsigned int *entry);␊ |
43 | static long DecodeSymbolTable(long cmdBase);␊ |
44 | ␊ |
45 | ␊ |
46 | static unsigned long gBinaryAddress;␊ |
47 | bool gHaveKernelCache;␉␉␉/* XXX aserebln: uninitialized? and only set to true, never to false */␊ |
48 | cpu_type_t archCpuType = CPU_TYPE_I386;␊ |
49 | ␊ |
50 | ␊ |
51 | //==============================================================================␊ |
52 | // Public function.␊ |
53 | ␊ |
54 | long ThinFatFile(void **binary, unsigned long *length)␊ |
55 | {␊ |
56 | ␉unsigned long nfat, swapped, size = 0;␊ |
57 | ␉struct fat_header *fhp = (struct fat_header *)*binary;␊ |
58 | ␉struct fat_arch *fap = (struct fat_arch *)((unsigned long)*binary + sizeof(struct fat_header));␊ |
59 | ␉cpu_type_t fapcputype;␊ |
60 | ␉uint32_t fapoffset;␊ |
61 | ␉uint32_t fapsize;␊ |
62 | ␊ |
63 | ␉if (fhp->magic == FAT_MAGIC)/* 0xcafebabe */␊ |
64 | ␉{␊ |
65 | ␉␉nfat = fhp->nfat_arch;␊ |
66 | ␉␉swapped = 0;␊ |
67 | ␉}␊ |
68 | ␉else if (fhp->magic == FAT_CIGAM)/* 0xbebafeca */␊ |
69 | ␉{␊ |
70 | ␉␉nfat = OSSwapInt32(fhp->nfat_arch);␊ |
71 | ␉␉swapped = 1;␊ |
72 | ␉}␊ |
73 | ␉else␊ |
74 | ␉{␊ |
75 | ␉␉return -1;␊ |
76 | ␉}␊ |
77 | ␊ |
78 | ␉for (; nfat > 0; nfat--, fap++)␊ |
79 | ␉{␊ |
80 | ␉␉if (swapped)␊ |
81 | ␉␉{␊ |
82 | ␉␉␉fapcputype = OSSwapInt32(fap->cputype);␊ |
83 | ␉␉␉fapoffset = OSSwapInt32(fap->offset);␊ |
84 | ␉␉␉fapsize = OSSwapInt32(fap->size);␊ |
85 | ␉␉}␊ |
86 | ␉␉else␊ |
87 | ␉␉{␊ |
88 | ␉␉␉fapcputype = fap->cputype;␊ |
89 | ␉␉␉fapoffset = fap->offset;␊ |
90 | ␉␉␉fapsize = fap->size;␊ |
91 | ␉␉}␊ |
92 | ␊ |
93 | ␉␉if (fapcputype == archCpuType)␊ |
94 | ␉␉{␊ |
95 | ␉␉␉*binary = (void *) ((unsigned long)*binary + fapoffset);␊ |
96 | ␉␉␉size = fapsize;␊ |
97 | ␉␉␉break;␊ |
98 | ␉␉}␊ |
99 | ␉}␊ |
100 | ␊ |
101 | ␉if (length != 0)␊ |
102 | ␉{␊ |
103 | ␉␉*length = size;␊ |
104 | ␉}␊ |
105 | ␊ |
106 | ␉return 0;␊ |
107 | }␊ |
108 | ␊ |
109 | ␊ |
110 | //==============================================================================␊ |
111 | ␊ |
112 | long DecodeMachO(void *binary, entry_t *rentry, char **raddr, int *rsize)␊ |
113 | {␊ |
114 | ␉struct mach_header *mH;␊ |
115 | ␉unsigned long ncmds, cmdBase, cmd, cmdsize, cmdstart;␊ |
116 | ␉// long headerBase, headerAddr, headerSize;␊ |
117 | ␉unsigned int vmaddr = ~0;␊ |
118 | ␉unsigned int vmend = 0;␊ |
119 | ␉unsigned long cnt;␊ |
120 | ␉long ret = -1;␊ |
121 | ␉unsigned int entry = 0;␊ |
122 | ␊ |
123 | ␉gBinaryAddress = (unsigned long)binary;␊ |
124 | ␊ |
125 | ␉mH = (struct mach_header *)(gBinaryAddress);␊ |
126 | ␊ |
127 | /*#if DEBUG␊ |
128 | ␉DBG("magic: 0x%x\n", (unsigned)mH->magic);␊ |
129 | ␉DBG("cputype: 0x%x\n", (unsigned)mH->cputype);␊ |
130 | ␉DBG("cpusubtype: 0x%x\n", (unsigned)mH->cpusubtype);␊ |
131 | ␉DBG("filetype: 0x%x\n", (unsigned)mH->filetype);␊ |
132 | ␉DBG("ncmds: 0x%x\n", (unsigned)mH->ncmds);␊ |
133 | ␉DBG("sizeofcmds: 0x%x\n", (unsigned)mH->sizeofcmds);␊ |
134 | ␉DBG("flags: 0x%x\n", (unsigned)mH->flags);␊ |
135 | ␉DBG("archCpuType: 0x%x\n", archCpuType);␊ |
136 | ␉//getchar();␊ |
137 | #endif*/␊ |
138 | ␊ |
139 | ␉switch (archCpuType)␊ |
140 | ␉{␊ |
141 | ␉␉case CPU_TYPE_I386:␊ |
142 | ␊ |
143 | ␉␉␉if (mH->magic != MH_MAGIC)␊ |
144 | ␉␉␉{␊ |
145 | ␉␉␉␉error("Mach-O (i386) file has bad magic number\n");␊ |
146 | ␉␉␉␉return -1;␊ |
147 | ␉␉␉}␊ |
148 | ␊ |
149 | ␉␉␉cmdstart = (unsigned long)gBinaryAddress + sizeof(struct mach_header);␊ |
150 | ␉␉␉break;␊ |
151 | ␊ |
152 | ␉␉case CPU_TYPE_X86_64:␊ |
153 | /*␊ |
154 | ␉␉␉if (mH->magic != MH_MAGIC_64 && mH->magic == MH_MAGIC)␊ |
155 | ␉␉␉{␊ |
156 | ␉␉␉␉return -1;␊ |
157 | ␉␉␉}␊ |
158 | */␊ |
159 | ␉␉␉if (mH->magic != MH_MAGIC_64)␊ |
160 | ␉␉␉{␊ |
161 | ␉␉␉␉error("Mach-O file (x86_64) has bad magic number\n");␊ |
162 | ␉␉␉␉return -1;␊ |
163 | ␉␉␉}␊ |
164 | ␊ |
165 | ␉␉␉cmdstart = (unsigned long)gBinaryAddress + sizeof(struct mach_header_64);␊ |
166 | ␉␉␉break;␊ |
167 | ␊ |
168 | ␉␉default:␊ |
169 | ␊ |
170 | ␉␉␉error("Unknown CPU type\n");␊ |
171 | ␉␉␉return -1;␊ |
172 | ␉}␊ |
173 | ␊ |
174 | ␉cmdBase = cmdstart;␊ |
175 | ␉ncmds = mH->ncmds;␊ |
176 | ␊ |
177 | ␉for (cnt = 0; cnt < ncmds; cnt++)␊ |
178 | ␉{␊ |
179 | ␉␉cmd = ((long *)cmdBase)[0];␊ |
180 | ␉␉cmdsize = ((long *)cmdBase)[1];␊ |
181 | ␉␉unsigned int load_addr;␊ |
182 | ␉␉unsigned int load_size;␊ |
183 | ␊ |
184 | ␉␉switch (cmd)␊ |
185 | ␉␉{␊ |
186 | ␉␉␉case LC_SEGMENT:␊ |
187 | ␉␉␉case LC_SEGMENT_64:␊ |
188 | ␉␉␉ret = DecodeSegment(cmdBase, &load_addr, &load_size);␊ |
189 | ␊ |
190 | ␉␉␉if (ret == 0 && load_size != 0 && load_addr >= KERNEL_ADDR)␊ |
191 | ␉␉␉{␊ |
192 | ␉␉␉␉vmaddr = MIN(vmaddr, load_addr);␊ |
193 | ␉␉␉␉vmend = MAX(vmend, load_addr + load_size);␊ |
194 | ␉␉␉}␊ |
195 | ␉␉␉break;␊ |
196 | ␊ |
197 | ␉␉␉case LC_MAIN:␉/* Mountain Lion's replacement for LC_UNIXTHREAD */␊ |
198 | ␉␉␉case LC_UNIXTHREAD:␊ |
199 | ␉␉␉␉ret = DecodeUnixThread(cmdBase, &entry);␊ |
200 | ␉␉␉break;␊ |
201 | ␊ |
202 | ␉␉␉case LC_SYMTAB:␊ |
203 | ␉␉␉break;␊ |
204 | ␊ |
205 | ␉␉␉default:␊ |
206 | #if NOTDEF␊ |
207 | ␉␉␉printf("Ignoring cmd type %d.\n", (unsigned)cmd);␊ |
208 | #endif␊ |
209 | ␉␉␉break;␊ |
210 | ␉␉}␊ |
211 | ␊ |
212 | ␊ |
213 | ␉if (ret != 0)␊ |
214 | ␉{␊ |
215 | ␉␉return -1;␊ |
216 | ␉}␊ |
217 | ␊ |
218 | ␉cmdBase += cmdsize;␊ |
219 | ␉}␊ |
220 | ␊ |
221 | ␉*rentry = (entry_t)( (unsigned long) entry & 0x3fffffff );␊ |
222 | ␉*rsize = vmend - vmaddr;␊ |
223 | ␉*raddr = (char *)vmaddr;␊ |
224 | ␊ |
225 | ␉cmdBase = cmdstart;␊ |
226 | ␊ |
227 | ␉for (cnt = 0; cnt < ncmds; cnt++)␊ |
228 | ␉{␊ |
229 | ␉␉cmd = ((long *)cmdBase)[0];␊ |
230 | ␉␉cmdsize = ((long *)cmdBase)[1];␊ |
231 | ␉␉␊ |
232 | ␉␉if (cmd == LC_SYMTAB)␊ |
233 | ␉␉{␊ |
234 | ␉␉␉if (DecodeSymbolTable(cmdBase) != 0)␊ |
235 | ␉␉␉{␊ |
236 | ␉␉␉␉return -1;␊ |
237 | ␉␉␉}␊ |
238 | ␉␉}␊ |
239 | ␊ |
240 | ␉␉cmdBase += cmdsize;␊ |
241 | ␉}␊ |
242 | ␊ |
243 | ␉return ret;␊ |
244 | }␊ |
245 | ␊ |
246 | ␊ |
247 | //==============================================================================␊ |
248 | // Private function.␊ |
249 | ␊ |
250 | ␊ |
251 | static long DecodeSegment(long cmdBase, unsigned int *load_addr, unsigned int *load_size)␊ |
252 | {␊ |
253 | ␉char *segname;␊ |
254 | ␉long vmsize, filesize;␊ |
255 | ␉unsigned long vmaddr, fileaddr;␊ |
256 | ␊ |
257 | ␉if (((long *)cmdBase)[0] == LC_SEGMENT_64)␊ |
258 | ␉{␊ |
259 | ␉␉struct segment_command_64 *segCmd;␊ |
260 | ␉␉segCmd = (struct segment_command_64 *)cmdBase;␊ |
261 | ␉␉vmaddr = (segCmd->vmaddr & 0x3fffffff);␊ |
262 | ␉␉vmsize = segCmd->vmsize;␊ |
263 | ␉␉fileaddr = (gBinaryAddress + segCmd->fileoff);␊ |
264 | ␉␉filesize = segCmd->filesize;␊ |
265 | ␉␉segname = segCmd->segname;␊ |
266 | ␊ |
267 | #ifdef DEBUG␊ |
268 | ␉printf("segname: %s, vmaddr: %x, vmsize: %x, fileoff: %x, filesize: %x, nsects: %d, flags: %x.\n",␊ |
269 | ␉␉segCmd->segname, (unsigned)vmaddr, (unsigned)vmsize, (unsigned)fileaddr, (unsigned)filesize,␊ |
270 | ␉␉(unsigned) segCmd->nsects, (unsigned)segCmd->flags);␊ |
271 | ␉getchar();␊ |
272 | #endif␊ |
273 | ␉}␊ |
274 | ␉else␊ |
275 | ␉{␊ |
276 | ␉␉struct segment_command *segCmd;␊ |
277 | ␊ |
278 | ␉␉segCmd = (struct segment_command *)cmdBase;␊ |
279 | ␊ |
280 | ␉␉vmaddr = (segCmd->vmaddr & 0x3fffffff);␊ |
281 | ␉␉vmsize = segCmd->vmsize;␊ |
282 | ␉␉fileaddr = (gBinaryAddress + segCmd->fileoff);␊ |
283 | ␉␉filesize = segCmd->filesize;␊ |
284 | ␉␉segname = segCmd->segname;␊ |
285 | ␊ |
286 | #ifdef DEBUG␊ |
287 | ␉printf("segname: %s, vmaddr: %x, vmsize: %x, fileoff: %x, filesize: %x, nsects: %d, flags: %x.\n",␊ |
288 | ␉␉segCmd->segname, (unsigned)vmaddr, (unsigned)vmsize, (unsigned)fileaddr, (unsigned)filesize, (unsigned) segCmd->nsects, (unsigned)segCmd->flags);␊ |
289 | ␉getchar();␊ |
290 | #endif␊ |
291 | ␉}␊ |
292 | ␊ |
293 | //===================================================␊ |
294 | ␊ |
295 | ␉if (vmsize == 0 || filesize == 0)␊ |
296 | ␉{␊ |
297 | ␉␉*load_addr = ~0;␊ |
298 | ␉␉*load_size = 0;␊ |
299 | ␉␉return 0;␊ |
300 | ␉}␊ |
301 | ␊ |
302 | ␉if (! ((vmaddr >= KERNEL_ADDR && (vmaddr + vmsize) <= (KERNEL_ADDR + KERNEL_LEN)) ||␊ |
303 | ␉␉ (vmaddr >= HIB_ADDR && (vmaddr + vmsize) <= (HIB_ADDR + HIB_LEN))))␊ |
304 | ␉{␊ |
305 | ␉␉stop("Kernel overflows available space");␊ |
306 | ␉}␊ |
307 | ␊ |
308 | ␉if (vmsize && ((strncmp(segname, "__PRELINK_INFO", sizeof("__PRELINK_INFO")) == 0) || (strncmp(segname, "__PRELINK", sizeof("__PRELINK")) == 0)))␊ |
309 | ␉{␊ |
310 | ␉␉gHaveKernelCache = true;␊ |
311 | ␉}␊ |
312 | ␊ |
313 | ␉// Copy from file load area.␊ |
314 | ␉if (vmsize>0 && filesize > 0)␊ |
315 | ␉{␊ |
316 | ␉␉bcopy((char *)fileaddr, (char *)vmaddr, vmsize > filesize ? filesize : vmsize);␊ |
317 | ␉}␊ |
318 | ␊ |
319 | ␉// Zero space at the end of the segment.␊ |
320 | ␉if (vmsize > filesize)␊ |
321 | ␉{␊ |
322 | ␉␉bzero((char *)(vmaddr + filesize), vmsize - filesize);␊ |
323 | ␉}␊ |
324 | ␊ |
325 | ␉*load_addr = vmaddr;␊ |
326 | ␉*load_size = vmsize;␊ |
327 | ␊ |
328 | ␉return 0;␊ |
329 | }␊ |
330 | ␊ |
331 | //==============================================================================␊ |
332 | ␊ |
333 | static long DecodeUnixThread(long cmdBase, unsigned int *entry)␊ |
334 | {␊ |
335 | ␉switch (archCpuType)␊ |
336 | ␉{␊ |
337 | ␉␉case CPU_TYPE_I386:␊ |
338 | ␉␉{␊ |
339 | ␉␉␉i386_thread_state_t *i386ThreadState;␊ |
340 | ␉␉␉i386ThreadState = (i386_thread_state_t *) (cmdBase + sizeof(struct thread_command) + 8);␊ |
341 | ␊ |
342 | ␉␉␉*entry = i386ThreadState->eip;␊ |
343 | ␉␉␉return 0;␊ |
344 | ␉␉}␊ |
345 | ␉␉␉␊ |
346 | ␉␉case CPU_TYPE_X86_64:␊ |
347 | ␉␉{␊ |
348 | ␉␉␉x86_thread_state64_t *x86_64ThreadState;␊ |
349 | ␉␉␉x86_64ThreadState = (x86_thread_state64_t *) (cmdBase + sizeof(struct thread_command) + 8);␊ |
350 | ␉␉␉*entry = x86_64ThreadState->rip;␊ |
351 | ␉␉␉return 0;␊ |
352 | ␉␉}␊ |
353 | ␉␉␉␊ |
354 | ␉␉default:␊ |
355 | ␉␉␉error("Unknown CPU type\n");␊ |
356 | ␉␉␉return -1;␊ |
357 | ␉}␊ |
358 | }␊ |
359 | ␊ |
360 | ␊ |
361 | //==============================================================================␊ |
362 | ␊ |
363 | static long DecodeSymbolTable(long cmdBase)␊ |
364 | {␊ |
365 | ␉long tmpAddr, symsSize, totalSize;␊ |
366 | ␉long gSymbolTableAddr;␊ |
367 | ␉long gSymbolTableSize;␊ |
368 | ␊ |
369 | ␉struct symtab_command *symTab, *symTableSave;␊ |
370 | ␊ |
371 | ␉symTab = (struct symtab_command *)cmdBase;␊ |
372 | ␊ |
373 | #if DEBUG␊ |
374 | ␉printf("symoff: %x, nsyms: %x, stroff: %x, strsize: %x\n",␊ |
375 | ␉␉symTab->symoff, symTab->nsyms, symTab->stroff, symTab->strsize);␊ |
376 | ␉getchar();␊ |
377 | #endif␊ |
378 | ␊ |
379 | ␉symsSize = symTab->stroff - symTab->symoff;␊ |
380 | ␉totalSize = symsSize + symTab->strsize;␊ |
381 | ␊ |
382 | ␉gSymbolTableSize = totalSize + sizeof(struct symtab_command);␊ |
383 | ␉gSymbolTableAddr = AllocateKernelMemory(gSymbolTableSize);␊ |
384 | ␉// Add the SymTab to the memory-map.␊ |
385 | ␉AllocateMemoryRange("Kernel-__SYMTAB", gSymbolTableAddr, gSymbolTableSize, -1);␊ |
386 | ␊ |
387 | ␉symTableSave = (struct symtab_command *)gSymbolTableAddr;␊ |
388 | ␉tmpAddr = gSymbolTableAddr + sizeof(struct symtab_command);␊ |
389 | ␊ |
390 | ␉symTableSave->symoff = tmpAddr;␊ |
391 | ␉symTableSave->nsyms = symTab->nsyms;␊ |
392 | ␉symTableSave->stroff = tmpAddr + symsSize;␊ |
393 | ␉symTableSave->strsize = symTab->strsize;␊ |
394 | ␉␊ |
395 | ␉bcopy((char *)(gBinaryAddress + symTab->symoff), (char *)tmpAddr, totalSize);␊ |
396 | ␊ |
397 | ␉return 0;␊ |
398 | }␊ |
399 |