]> git.saurik.com Git - apple/xnu.git/blame - libsa/kld_patch.c
xnu-344.21.73.tar.gz
[apple/xnu.git] / libsa / kld_patch.c
CommitLineData
0b4e3aa0
A
1/*
2 * Copyright (c) 2001 Apple Computer, Inc. All rights reserved.
3 *
4 * @APPLE_LICENSE_HEADER_START@
5 *
d7e50217
A
6 * Copyright (c) 1999-2003 Apple Computer, Inc. All Rights Reserved.
7 *
8 * This file contains Original Code and/or Modifications of Original Code
9 * as defined in and that are subject to the Apple Public Source License
10 * Version 2.0 (the 'License'). You may not use this file except in
11 * compliance with the License. Please obtain a copy of the License at
12 * http://www.opensource.apple.com/apsl/ and read it before using this
13 * file.
0b4e3aa0
A
14 *
15 * The Original Code and all software distributed under the License are
16 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
17 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
18 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
d7e50217
A
19 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
20 * Please see the License for the specific language governing rights and
21 * limitations under the License.
0b4e3aa0
A
22 *
23 * @APPLE_LICENSE_HEADER_END@
24 */
25/*
26 * History:
27 * 2001-05-30 gvdl Initial implementation of the vtable patcher.
28 */
29// 45678901234567890123456789012345678901234567890123456789012345678901234567890
30
31#include <mach-o/fat.h>
32#include <mach-o/loader.h>
33#include <mach-o/nlist.h>
34#include <mach-o/reloc.h>
35
36#if KERNEL
37
38#include <stdarg.h>
39#include <string.h>
40
41#include <sys/systm.h>
42
43#include <libkern/OSTypes.h>
44
45#include <libsa/stdlib.h>
46#include <libsa/mach/mach.h>
47
48#include "mach_loader.h"
49
50#include <vm/vm_kern.h>
51
52enum { false = 0, true = 1 };
53
54#define vm_page_size page_size
55
56extern load_return_t fatfile_getarch(
57 void * vp, // normally a (struct vnode *)
58 vm_offset_t data_ptr,
59 struct fat_arch * archret);
60
9bccf70c
A
61__private_extern__ char *strstr(const char *in, const char *str);
62
0b4e3aa0 63#else /* !KERNEL */
9bccf70c 64
0b4e3aa0
A
65#include <unistd.h>
66
67#include <stdio.h>
68#include <stdlib.h>
69#include <string.h>
70
71#include <sys/errno.h>
72#include <sys/fcntl.h>
73#include <sys/stat.h>
74#include <sys/mman.h>
9bccf70c 75#include <sys/vm.h>
0b4e3aa0
A
76
77#include <mach/mach.h>
78#include <mach/mach_error.h>
79
80#include <mach-o/arch.h>
81
82#include <CoreFoundation/CoreFoundation.h>
9bccf70c
A
83
84#define PAGE_SIZE vm_page_size
85#define PAGE_MASK (PAGE_SIZE - 1)
86
0b4e3aa0
A
87#endif /* KERNEL */
88
89#include "kld_patch.h"
9bccf70c 90#include "c++rem3.h"
0b4e3aa0
A
91
92#if 0
9bccf70c
A
93#define DIE() do { for (;;) ; } while(0)
94
95#if KERNEL
96# define LOG_DELAY() /* IODelay(200000) */
97# define DEBUG_LOG(x) do { IOLog x; LOG_DELAY(); } while(0)
98#else
99# define LOG_DELAY()
100# define DEBUG_LOG(x) do { printf x; } while(0)
101#endif
0b4e3aa0 102
0b4e3aa0
A
103#else
104
105#define DIE()
106#define LOG_DELAY()
107#define DEBUG_LOG(x)
108
109#endif
110
111// OSObject symbol prefixes and suffixes
9bccf70c
A
112#define kCPPSymbolPrefix "_Z"
113#define kVTablePrefix "_" kCPPSymbolPrefix "TV"
114#define kOSObjPrefix "_" kCPPSymbolPrefix "N"
115#define kReservedNamePrefix "_RESERVED"
116#define k29SuperClassSuffix "superClass"
117#define k31SuperClassSuffix "10superClassE"
118#define kGMetaSuffix "10gMetaClassE"
0b4e3aa0
A
119#define kLinkEditSegName SEG_LINKEDIT
120
121// GCC 2.95 drops 2 leading constants in the vtable
122#define kVTablePreambleLen 2
123
124// Last address that I'm willing to try find vm in
125#define kTopAddr ((unsigned char *) (1024 * 1024 * 1024))
126
127// Size in bytes that Data Ref object's get increased in size
128// Must be a power of 2
129#define kDataCapacityIncrement 128
130
131// My usual set of helper macros. I personally find these macros
132// easier to read in the code rather than an explicit error condition
133// check. If I don't make it easy then I may get lazy ond not check
134// everything. I'm sorry if you find this code harder to read.
135
136// break_if will evaluate the expression and if it is true
137// then it will print the msg, which is enclosed in parens
138// and then break. Usually used in loops are do { } while (0)
139#define break_if(expr, msg) \
140 if (expr) { \
141 errprintf msg; \
142 break; \
143 }
144
145// return_if will evaluate expr and if true it will log the
146// msg, which is enclosed in parens, and then it will return
147// with the return code of ret.
148#define return_if(expr, ret, msg) do { \
149 if (expr) { \
150 errprintf msg; \
151 return ret; \
152 } \
153} while (0)
154
155#ifndef MIN
156#define MIN(a,b) (((a)<(b))?(a):(b))
157#endif /* MIN */
158#ifndef MAX
159#define MAX(a,b) (((a)>(b))?(a):(b))
160#endif /* MAX */
161
162typedef struct Data {
163 unsigned long fLength, fCapacity;
164 unsigned char *fData;
165} Data, *DataRef;
166
167struct sectionRecord {
168 const struct section *fSection;
169 DataRef fRelocCache;
170};
171
172enum patchState {
173 kSymbolIdentical,
174 kSymbolLocal,
175 kSymbolPadUpdate,
176 kSymbolSuperUpdate,
177 kSymbolMismatch
178};
179
180struct patchRecord {
181 struct nlist *fSymbol;
182 enum patchState fType;
183};
184
185struct relocRecord {
186 void *fValue;
187 const struct nlist *fSymbol;
188 struct relocation_info *fRInfo;
189 void *reserved;
190};
191
192struct metaClassRecord {
193 char *fSuperName;
194 struct fileRecord *fFile;
195 const struct nlist *fVTableSym;
196 struct patchRecord *fPatchedVTable;
197 char fClassName[1];
198};
199
200struct fileRecord {
201 size_t fMapSize, fMachOSize;
0b4e3aa0
A
202 unsigned char *fMap, *fMachO, *fPadEnd;
203 DataRef fClassList;
204 DataRef fSectData;
9bccf70c
A
205 DataRef fNewSymbols, fNewStringBlocks;
206 DataRef fSym2Strings;
0b4e3aa0
A
207 struct symtab_command *fSymtab;
208 struct sectionRecord *fSections;
9bccf70c
A
209 struct segment_command *fLinkEditSeg;
210 const char **fSymbToStringTable;
0b4e3aa0
A
211 char *fStringBase;
212 struct nlist *fSymbolBase;
213 const struct nlist *fLocalSyms;
214 unsigned int fNSects;
215 int fNLocal;
0b4e3aa0
A
216 Boolean fIsKernel, fNoKernelExecutable, fIsKmem;
217 Boolean fImageDirty, fSymbolsDirty;
9bccf70c
A
218 Boolean fRemangled, fFoundOSObject;
219 Boolean fIgnoreFile;
220 const char fPath[1];
0b4e3aa0
A
221};
222
223static DataRef sFilesTable;
224static struct fileRecord *sKernelFile;
225
226static DataRef sMergedFiles;
227static DataRef sMergeMetaClasses;
228static Boolean sMergedKernel;
229
230static void errprintf(const char *fmt, ...)
231{
232 extern void kld_error_vprintf(const char *format, va_list ap);
233
234 va_list ap;
235
236 va_start(ap, fmt);
237 kld_error_vprintf(fmt, ap);
238 va_end(ap);
239
240DIE();
241}
242
243static __inline__ unsigned long DataGetLength(DataRef data)
244{
245 return data->fLength;
246}
247
248static __inline__ unsigned char *DataGetPtr(DataRef data)
249{
250 return data->fData;
251}
252
9bccf70c
A
253static __inline__ unsigned char *DataGetEndPtr(DataRef data)
254{
255 return data->fData + data->fLength;
256}
257
258static __inline__ unsigned long DataRemaining(DataRef data)
259{
260 return data->fCapacity - data->fLength;
261}
0b4e3aa0
A
262
263static __inline__ Boolean DataContainsAddr(DataRef data, void *vAddr)
264{
9bccf70c
A
265 vm_offset_t offset = (vm_address_t) vAddr;
266
267 if (!data)
268 return false;
269
270 offset = (vm_address_t) vAddr - (vm_address_t) data->fData;
271 return (offset < data->fLength);
272}
273
274static Boolean DataEnsureCapacity(DataRef data, unsigned long capacity)
275{
276 // Don't bother to ever shrink a data object.
277 if (capacity > data->fCapacity) {
278 unsigned char *newData;
279
280 capacity += kDataCapacityIncrement - 1;
281 capacity &= ~(kDataCapacityIncrement - 1);
282 newData = (unsigned char *) realloc(data->fData, capacity);
283 if (!newData)
284 return false;
285
286 bzero(newData + data->fCapacity, capacity - data->fCapacity);
287 data->fData = newData;
288 data->fCapacity = capacity;
289 }
290
291 return true;
292}
0b4e3aa0 293
9bccf70c
A
294static __inline__ Boolean DataSetLength(DataRef data, unsigned long length)
295{
296 if (DataEnsureCapacity(data, length)) {
297 data->fLength = length;
298 return true;
299 }
300 else
301 return false;
0b4e3aa0
A
302}
303
304static __inline__ Boolean DataAddLength(DataRef data, unsigned long length)
305{
0b4e3aa0
A
306 return DataSetLength(data, data->fLength + length);
307}
308
309static __inline__ Boolean
310DataAppendBytes(DataRef data, const void *addr, unsigned int len)
311{
312 unsigned long size = DataGetLength(data);
313
314 if (!DataAddLength(data, len))
315 return false;
316
317 bcopy(addr, DataGetPtr(data) + size, len);
318 return true;
319}
320
321static __inline__ Boolean DataAppendData(DataRef dst, DataRef src)
322{
323 return DataAppendBytes(dst, DataGetPtr(src), DataGetLength(src));
324}
325
9bccf70c 326static DataRef DataCreate(unsigned long capacity)
0b4e3aa0
A
327{
328 DataRef data = (DataRef) malloc(sizeof(Data));
329
330 if (data) {
9bccf70c 331 if (!capacity)
0b4e3aa0
A
332 data->fCapacity = kDataCapacityIncrement;
333 else {
9bccf70c 334 data->fCapacity = capacity + kDataCapacityIncrement - 1;
0b4e3aa0
A
335 data->fCapacity &= ~(kDataCapacityIncrement - 1);
336 }
337
338 data->fData = (unsigned char *) malloc(data->fCapacity);
339 if (!data->fData) {
340 free(data);
341 return NULL;
342 }
343
344 bzero(data->fData, data->fCapacity);
9bccf70c 345 data->fLength = 0;
0b4e3aa0
A
346 }
347 return data;
348}
349
350static void DataRelease(DataRef data)
351{
352 if (data) {
353 if (data->fData)
354 free(data->fData);
355 data->fData = 0;
356 free(data);
357 }
358}
359
9bccf70c
A
360static __inline__ const char *
361symNameByIndex(const struct fileRecord *file, unsigned int symInd)
362{
363 return file->fSymbToStringTable[symInd];
364}
365
366static __inline__ const char *
0b4e3aa0
A
367symbolname(const struct fileRecord *file, const struct nlist *sym)
368{
9bccf70c 369 unsigned int index;
0b4e3aa0 370
9bccf70c
A
371 index = sym - file->fSymbolBase;
372 if (index < file->fSymtab->nsyms)
373 return symNameByIndex(file, index);
0b4e3aa0 374
9bccf70c
A
375 if (-1 == sym->n_un.n_strx)
376 return (const char *) sym->n_value;
0b4e3aa0 377
9bccf70c
A
378 // If the preceding tests fail then we have a getNewSymbol patch and
379 // the file it refers to has already been patched as the n_strx is set
380 // to -1 temporarily while we are still processing a file.
381 // Once we have finished with a file then we repair the 'strx' offset
382 // to be valid for the repaired file's string table.
383 return file->fStringBase + sym->n_un.n_strx;
0b4e3aa0
A
384}
385
9bccf70c
A
386static struct fileRecord *
387getFile(const char *path)
0b4e3aa0
A
388{
389 if (sFilesTable) {
390 int i, nfiles;
391 struct fileRecord **files;
392
393 // Check to see if we have already merged this file
394 nfiles = DataGetLength(sFilesTable) / sizeof(struct fileRecord *);
395 files = (struct fileRecord **) DataGetPtr(sFilesTable);
396 for (i = 0; i < nfiles; i++) {
397 if (!strcmp(path, files[i]->fPath))
398 return files[i];
399 }
400 }
401
402 return NULL;
403}
404
9bccf70c
A
405static struct fileRecord *
406addFile(struct fileRecord *file, const char *path)
0b4e3aa0
A
407{
408 struct fileRecord *newFile;
409
410 if (!sFilesTable) {
411 sFilesTable = DataCreate(0);
412 if (!sFilesTable)
413 return NULL;
414 }
415
9bccf70c
A
416 newFile = (struct fileRecord *)
417 malloc(sizeof(struct fileRecord) + strlen(path));
0b4e3aa0
A
418 if (!newFile)
419 return NULL;
420
421 if (!DataAppendBytes(sFilesTable, &newFile, sizeof(newFile))) {
422 free(newFile);
423 return NULL;
424 }
425
9bccf70c
A
426 bcopy(file, newFile, sizeof(struct fileRecord) - 1);
427 strcpy((char *) newFile->fPath, path);
428
0b4e3aa0
A
429 return newFile;
430}
431
432// @@@ gvdl: need to clean up the sMergeMetaClasses
433// @@@ gvdl: I had better fix the object file up again
9bccf70c 434static void unmapFile(struct fileRecord *file)
0b4e3aa0 435{
0b4e3aa0
A
436 if (file->fSectData) {
437 struct sectionRecord *section;
438 unsigned int i, nsect;
439
440 nsect = file->fNSects;
441 section = file->fSections;
442 for (i = 0; i < nsect; i++, section++) {
443 if (section->fRelocCache) {
444 DataRelease(section->fRelocCache);
445 section->fRelocCache = 0;
446 }
447 }
448
449 DataRelease(file->fSectData);
450 file->fSectData = 0;
451 file->fSections = 0;
452 file->fNSects = 0;
453 }
454
9bccf70c
A
455 if (file->fSym2Strings) {
456 DataRelease(file->fSym2Strings);
457 file->fSym2Strings = 0;
458 }
459
0b4e3aa0
A
460 if (file->fMap) {
461#if KERNEL
462 if (file->fIsKmem)
463 kmem_free(kernel_map, (vm_address_t) file->fMap, file->fMapSize);
464#else /* !KERNEL */
465 if (file->fPadEnd) {
466 vm_address_t padVM;
467 vm_size_t padSize;
468
469 padVM = round_page((vm_address_t) file->fMap + file->fMapSize);
470 padSize = (vm_size_t) ((vm_address_t) file->fPadEnd - padVM);
471 (void) vm_deallocate(mach_task_self(), padVM, padSize);
472 file->fPadEnd = 0;
473 }
474
475 (void) munmap((caddr_t) file->fMap, file->fMapSize);
476#endif /* !KERNEL */
477 file->fMap = 0;
478 }
9bccf70c
A
479}
480
481static void removeFile(struct fileRecord *file)
482{
483 if (file->fClassList) {
484 DataRelease(file->fClassList);
485 file->fClassList = 0;
486 }
0b4e3aa0 487
9bccf70c
A
488 unmapFile(file);
489
490 free(file);
0b4e3aa0
A
491}
492
493#if !KERNEL
494static Boolean
9bccf70c 495mapObjectFile(struct fileRecord *file, const char *pathName)
0b4e3aa0
A
496{
497 Boolean result = false;
9bccf70c 498 static unsigned char *sFileMapBaseAddr = 0;
0b4e3aa0
A
499
500 int fd = 0;
501
502 if (!sFileMapBaseAddr) {
503 kern_return_t ret;
504 vm_address_t probeAddr;
505
506 // If we don't already have a base addr find any random chunk
507 // of 32 meg of VM and to use the 16 meg boundrary as a base.
508 ret = vm_allocate(mach_task_self(), &probeAddr,
509 32 * 1024 * 1024, VM_FLAGS_ANYWHERE);
510 return_if(KERN_SUCCESS != ret, false,
511 ("Unable to allocate base memory %s\n", mach_error_string(ret)));
512 (void) vm_deallocate(mach_task_self(), probeAddr, 32 * 1024 * 1024);
513
514 // Now round to the next 16 Meg boundrary
515 probeAddr = (probeAddr + (16 * 1024 * 1024 - 1))
516 & ~(16 * 1024 * 1024 - 1);
517 sFileMapBaseAddr = (unsigned char *) probeAddr;
518 }
519
9bccf70c 520 fd = open(pathName, O_RDONLY, 0);
0b4e3aa0 521 return_if(fd == -1, false, ("Can't open %s for reading - %s\n",
9bccf70c 522 pathName, strerror(errno)));
0b4e3aa0
A
523
524 do {
525 kern_return_t ret;
526 struct stat sb;
527 int retaddr = -1;
528
529 break_if(fstat(fd, &sb) == -1,
530 ("Can't stat %s - %s\n", file->fPath, strerror(errno)));
531
532 file->fMapSize = sb.st_size;
533 file->fMap = sFileMapBaseAddr;
534 ret = KERN_SUCCESS;
535 while (file->fMap < kTopAddr) {
536 vm_address_t padVM;
537 vm_address_t padVMEnd;
538 vm_size_t padSize;
539
540 padVM = round_page((vm_address_t) file->fMap + file->fMapSize);
541 retaddr = (int) mmap(file->fMap, file->fMapSize,
542 PROT_READ|PROT_WRITE,
543 MAP_FIXED|MAP_FILE|MAP_PRIVATE,
544 fd, 0);
545 if (-1 == retaddr) {
546 break_if(ENOMEM != errno,
547 ("mmap failed %d - %s\n", errno, strerror(errno)));
548
549 file->fMap = (unsigned char *) padVM;
550 continue;
551 }
552
553
554 // Round up padVM to the next page after the file and assign at
555 // least another fMapSize more room rounded up to the next page
556 // boundary.
557 padVMEnd = round_page(padVM + file->fMapSize);
558 padSize = padVMEnd - padVM;
559 ret = vm_allocate(
560 mach_task_self(), &padVM, padSize, VM_FLAGS_FIXED);
561 if (KERN_SUCCESS == ret) {
562 file->fPadEnd = (unsigned char *) padVMEnd;
563 break;
564 }
565 else {
566 munmap(file->fMap, file->fMapSize);
567 break_if(KERN_INVALID_ADDRESS != ret,
568 ("Unable to allocate pad vm for %s - %s\n",
9bccf70c 569 pathName, mach_error_string(ret)));
0b4e3aa0
A
570
571 file->fMap = (unsigned char *) padVMEnd;
572 continue; // try again wherever the vm system wants
573 }
574 }
575
576 if (-1 == retaddr || KERN_SUCCESS != ret)
577 break;
578
579 break_if(file->fMap >= kTopAddr,
580 ("Unable to map memory %s\n", file->fPath));
581
582 sFileMapBaseAddr = file->fPadEnd;
583 result = true;
584 } while(0);
585
586 close(fd);
587 return result;
588}
589#endif /* !KERNEL */
590
9bccf70c 591static Boolean findBestArch(struct fileRecord *file, const char *pathName)
0b4e3aa0
A
592{
593 unsigned long magic;
594 struct fat_header *fat;
595
596
597 file->fMachOSize = file->fMapSize;
598 file->fMachO = file->fMap;
599 magic = ((const struct mach_header *) file->fMachO)->magic;
600 fat = (struct fat_header *) file->fMachO;
601
602 // Try to figure out what type of file this is
603 return_if(file->fMapSize < sizeof(unsigned long), false,
9bccf70c 604 ("%s isn't a valid object file - no magic\n", pathName));
0b4e3aa0
A
605
606#if KERNEL
607
608 // CIGAM is byte-swapped MAGIC
609 if (magic == FAT_MAGIC || magic == FAT_CIGAM) {
610
611 load_return_t load_return;
612 struct fat_arch fatinfo;
613
614 load_return = fatfile_getarch(NULL, (vm_address_t) fat, &fatinfo);
615 return_if(load_return != LOAD_SUCCESS, false,
9bccf70c 616 ("Extension \"%s\": has no code for this computer\n", pathName));
0b4e3aa0
A
617
618 file->fMachO = file->fMap + fatinfo.offset;
619 file->fMachOSize = fatinfo.size;
620 magic = ((const struct mach_header *) file->fMachO)->magic;
621 }
622
623#else /* !KERNEL */
624
625 // Do we need to in-place swap the endianness of the fat header?
626 if (magic == FAT_CIGAM) {
627 unsigned long i;
628 struct fat_arch *arch;
629
630 fat->nfat_arch = NXSwapBigLongToHost(fat->nfat_arch);
631 return_if(file->fMapSize < sizeof(struct fat_header)
632 + fat->nfat_arch * sizeof(struct fat_arch),
633 false, ("%s is too fat\n", file->fPath));
634
635 arch = (struct fat_arch *) &fat[1];
636 for (i = 0; i < fat->nfat_arch; i++) {
637 arch[i].cputype = NXSwapBigLongToHost(arch[i].cputype);
638 arch[i].cpusubtype = NXSwapBigLongToHost(arch[i].cpusubtype);
639 arch[i].offset = NXSwapBigLongToHost(arch[i].offset);
640 arch[i].size = NXSwapBigLongToHost(arch[i].size);
641 arch[i].align = NXSwapBigLongToHost(arch[i].align);
642 }
643
644 magic = NXSwapBigLongToHost(fat->magic);
645 }
646
647 // Now see if we can find any valid architectures
648 if (magic == FAT_MAGIC) {
649 const NXArchInfo *myArch;
650 unsigned long fatsize;
651 struct fat_arch *arch;
652
653 fatsize = sizeof(struct fat_header)
654 + fat->nfat_arch * sizeof(struct fat_arch);
655 return_if(file->fMapSize < fatsize,
9bccf70c 656 false, ("%s isn't a valid fat file\n", pathName));
0b4e3aa0
A
657
658 myArch = NXGetLocalArchInfo();
659 arch = NXFindBestFatArch(myArch->cputype, myArch->cpusubtype,
660 (struct fat_arch *) &fat[1], fat->nfat_arch);
661 return_if(!arch,
9bccf70c 662 false, ("%s hasn't got arch for %s\n", pathName, myArch->name));
0b4e3aa0 663 return_if(arch->offset + arch->size > file->fMapSize,
9bccf70c 664 false, ("%s's %s arch is incomplete\n", pathName, myArch->name));
0b4e3aa0
A
665 file->fMachO = file->fMap + arch->offset;
666 file->fMachOSize = arch->size;
667 magic = ((const struct mach_header *) file->fMachO)->magic;
668 }
669
670#endif /* KERNEL */
671
672 return_if(magic != MH_MAGIC,
9bccf70c 673 false, ("%s isn't a valid mach-o\n", pathName));
0b4e3aa0
A
674
675 return true;
676}
677
678static Boolean
679parseSegments(struct fileRecord *file, struct segment_command *seg)
680{
681 struct sectionRecord *sections;
682 int i, nsects = seg->nsects;
683 const struct segmentMap {
684 struct segment_command seg;
685 const struct section sect[1];
686 } *segMap;
687
0b4e3aa0
A
688 if (!file->fSectData) {
689 file->fSectData = DataCreate(0);
690 if (!file->fSectData)
691 return false;
692 }
693
694 // Increase length of section DataRef and cache data pointer
695 if (!DataAddLength(file->fSectData, nsects * sizeof(struct sectionRecord)))
696 return false;
697 file->fSections = (struct sectionRecord *) DataGetPtr(file->fSectData);
698
699 // Initialise the new sections
700 sections = &file->fSections[file->fNSects];
701 file->fNSects += nsects;
702 for (i = 0, segMap = (struct segmentMap *) seg; i < nsects; i++)
703 sections[i].fSection = &segMap->sect[i];
704
705 return true;
706}
707
9bccf70c
A
708static Boolean
709remangleExternSymbols(struct fileRecord *file, const char *pathName)
710{
711 const struct nlist *sym;
712 int i, nsyms, len;
713 DataRef strings = NULL;
714
715 DEBUG_LOG(("Remangling %s\n", pathName));
716
717 file->fNewStringBlocks = DataCreate(0);
718 return_if(!file->fNewStringBlocks, false,
719 ("Unable to allocate new string table for %s\n", pathName));
720
721 nsyms = file->fSymtab->nsyms;
722 for (i = 0, sym = file->fSymbolBase; i < nsyms; i++, sym++) {
723 Rem3Return ret;
724 const char *symname;
725 char *newname;
726 unsigned char n_type = sym->n_type;
727
728 // Not an external symbol or it is a stab in any case don't bother
729 if ((n_type ^ N_EXT) & (N_STAB | N_EXT))
730 continue;
731
732 symname = symNameByIndex(file, i);
733
734tryRemangleAgain:
735 if (!strings) {
736 strings = DataCreate(16 * 1024); // Arbitrary block size
737 return_if(!strings, false,
738 ("Unable to allocate new string block for %s\n", pathName));
739 }
740
741 len = DataRemaining(strings);
742 newname = DataGetEndPtr(strings);
743 ret = rem3_remangle_name(newname, &len, symname);
744 switch (ret) {
745 case kR3InternalNotRemangled:
746 errprintf("Remangler fails on %s in %s\n", symname, pathName);
747 /* No break */
748 case kR3NotRemangled:
749 break;
750
751 case kR3Remangled:
752 file->fSymbToStringTable[i] = newname;
753 file->fRemangled = file->fSymbolsDirty = true;
754 DataAddLength(strings, len + 1); // returns strlen
755 break;
756
757 case kR3BufferTooSmallRemangled:
758 return_if(!DataAppendBytes
759 (file->fNewStringBlocks, &strings, sizeof(strings)),
760 false, ("Unable to allocate string table for %s\n", pathName));
761 strings = NULL;
762 goto tryRemangleAgain;
763
764 case kR3BadArgument:
765 default:
766 return_if(true, false,
767 ("Internal error - remangle of %s\n", pathName));
768 }
769 }
770
771 if (strings) {
772 return_if(!DataAppendBytes
773 (file->fNewStringBlocks, &strings, sizeof(strings)),
774 false, ("Unable to allocate string table for %s\n", pathName));
775 }
776
777 return true;
778}
779
780static Boolean parseSymtab(struct fileRecord *file, const char *pathName)
781{
782 const struct nlist *sym;
783 unsigned int i, firstlocal, nsyms;
784 unsigned long strsize;
785 const char *strbase;
786 Boolean foundOSObject, found295CPP;
787
788 // we found a link edit segment so recompute the bases
789 if (file->fLinkEditSeg) {
790 struct segment_command *link = file->fLinkEditSeg;
791
792 file->fSymbolBase = (struct nlist *)
793 (link->vmaddr + (file->fSymtab->symoff - link->fileoff));
794 file->fStringBase = (char *)
795 (link->vmaddr + (file->fSymtab->stroff - link->fileoff));
796 return_if( ( (caddr_t) file->fStringBase + file->fSymtab->strsize
797 > (caddr_t) link->vmaddr + link->vmsize ), false,
798 ("%s isn't a valid mach-o le, bad symbols\n", pathName));
799 }
800 else {
801 file->fSymbolBase = (struct nlist *)
802 (file->fMachO + file->fSymtab->symoff);
803 file->fStringBase = (char *)
804 (file->fMachO + file->fSymtab->stroff);
805 return_if( ( file->fSymtab->stroff + file->fSymtab->strsize
806 > file->fMachOSize ), false,
807 ("%s isn't a valid mach-o, bad symbols\n", pathName));
808 }
809
810 nsyms = file->fSymtab->nsyms;
811
812 // If this file the kernel and do we have an executable image
813 file->fNoKernelExecutable = (vm_page_size == file->fSymtab->symoff)
814 && (file->fSections[0].fSection->size == 0);
815
816 // Generate a table of pointers to strings indexed by the symbol number
817
818 file->fSym2Strings = DataCreate(nsyms * sizeof(const char *));
819 DataSetLength(file->fSym2Strings, nsyms * sizeof(const char *));
820 return_if(!file->fSym2Strings, false,
821 ("Unable to allocate memory - symbol string trans\n", pathName));
822 file->fSymbToStringTable = (const char **) DataGetPtr(file->fSym2Strings);
823
824 // Search for the first non-stab symbol in table
825 strsize = file->fSymtab->strsize;
826 strbase = file->fStringBase;
827 firstlocal = 0;
828 found295CPP = foundOSObject = false;
829 for (i = 0, sym = file->fSymbolBase; i < nsyms; i++, sym++) {
830 long strx = sym->n_un.n_strx;
831 const char *symname = strbase + strx;
832 unsigned char n_type;
833
834 return_if(((unsigned long) strx > strsize), false,
835 ("%s has an illegal string offset in symbol %d\n", pathName, i));
836
837 // Load up lookup symbol look table with sym names
838 file->fSymbToStringTable[i] = symname;
839
840 n_type = sym->n_type & (N_TYPE | N_EXT);
841
842 // Find the first exported symbol
843 if ( !firstlocal && (n_type & N_EXT) ) {
844 firstlocal = i;
845 file->fLocalSyms = sym;
846 }
847
848 // Find the a OSObject based subclass by searching for symbols
849 // that have a suffix of '10superClassE'
850 symname++; // Skip leading '_'
851
852 if (!foundOSObject
853 && (n_type == (N_SECT | N_EXT) || n_type == (N_ABS | N_EXT))
854 && strx) {
855 const char *suffix, *endSym;
856
857 endSym = symname + strlen(symname);
858
859 // Find out if this symbol has the superclass suffix.
860 if (symname[0] == kCPPSymbolPrefix[0]
861 && symname[1] == kCPPSymbolPrefix[1]) {
862
863 suffix = endSym - sizeof(k31SuperClassSuffix) + 1;
864
865 // Check for a gcc3 OSObject subclass
866 if (suffix > symname
867 && !strcmp(suffix, k31SuperClassSuffix))
868 foundOSObject = true;
869 }
870 else {
871 suffix = endSym - sizeof(k29SuperClassSuffix);
872
873 // Check for a gcc295 OSObject subclass
874 if (suffix > symname
875 && ('.' == *suffix || '$' == *suffix)
876 && !strcmp(suffix+1, k29SuperClassSuffix)) {
877 found295CPP = foundOSObject = true;
878 }
879 else if (!found295CPP) {
880 // Finally just check if we need to remangle
881 symname++; // skip leading '__'
882 while (*symname) {
883 if ('_' == *symname++ && '_' == *symname++) {
884 found295CPP = true;
885 break;
886 }
887 }
888 }
889 }
890 }
891 else if (sym->n_type == (N_EXT | N_UNDF)) {
892 if ( !file->fNLocal) // Find the last local symbol
893 file->fNLocal = i - firstlocal;
894 if (!found295CPP) {
895 symname++; // Skip possible second '_' at start.
896 while (*symname) {
897 if ('_' == *symname++ && '_' == *symname++) {
898 found295CPP = true;
899 break;
900 }
901 }
902 }
903 }
904 // Note symname is trashed at this point
905 }
906 return_if(i < nsyms, false,
907 ("%s isn't a valid mach-o, bad symbol strings\n", pathName));
908
909 return_if(!file->fLocalSyms, false, ("%s has no symbols?\n", pathName));
910
911 // If we don't have any undefined symbols then all symbols
912 // must be local so just compute it now if necessary.
913 if ( !file->fNLocal )
914 file->fNLocal = i - firstlocal;
915
916 file->fFoundOSObject = foundOSObject;
917
918 if (found295CPP && !remangleExternSymbols(file, pathName))
919 return false;
920
921 return true;
922}
923
0b4e3aa0
A
924// @@@ gvdl: These functions need to be hashed they are
925// going to be way too slow for production code.
926static const struct nlist *
927findSymbolByAddress(const struct fileRecord *file, void *entry)
928{
929 // not quite so dumb linear search of all symbols
930 const struct nlist *sym;
931 int i, nsyms;
932
933 // First try to find the symbol in the most likely place which is the
934 // extern symbols
935 sym = file->fLocalSyms;
936 for (i = 0, nsyms = file->fNLocal; i < nsyms; i++, sym++) {
937 if (sym->n_value == (unsigned long) entry && !(sym->n_type & N_STAB) )
938 return sym;
939 }
940
941 // Didn't find it in the external symbols so try to local symbols before
942 // giving up.
943 sym = file->fSymbolBase;
944 for (i = 0, nsyms = file->fSymtab->nsyms; i < nsyms; i++, sym++) {
945 if ( (sym->n_type & N_EXT) )
946 return NULL;
947 if ( sym->n_value == (unsigned long) entry && !(sym->n_type & N_STAB) )
948 return sym;
949 }
950
951 return NULL;
952}
953
954struct searchContext {
955 const char *fSymname;
9bccf70c 956 const struct fileRecord *fFile;
0b4e3aa0
A
957};
958
959static int symbolSearch(const void *vKey, const void *vSym)
960{
961 const struct searchContext *key = (const struct searchContext *) vKey;
962 const struct nlist *sym = (const struct nlist *) vSym;
963
9bccf70c 964 return strcmp(key->fSymname + 1, symbolname(key->fFile, sym) + 1);
0b4e3aa0
A
965}
966
967static const struct nlist *
968findSymbolByName(struct fileRecord *file, const char *symname)
969{
9bccf70c
A
970 if (file->fRemangled) {
971 // @@@ gvdl: Performance problem
972 // Linear search as we don't sort after remangling
973 const struct nlist *sym;
974 int i = file->fLocalSyms - file->fSymbolBase;
975 int nLocal = file->fNLocal + i;
976
977 for (sym = file->fLocalSyms; i < nLocal; i++, sym++)
978 if (!strcmp(symNameByIndex(file, i) + 1, symname + 1))
979 return sym;
980 return NULL;
981 }
982 else {
983 struct searchContext context;
984
985 context.fSymname = symname;
986 context.fFile = file;
987 return (struct nlist *)
988 bsearch(&context,
989 file->fLocalSyms, file->fNLocal, sizeof(struct nlist),
990 symbolSearch);
991 }
0b4e3aa0
A
992}
993
994static Boolean
995relocateSection(const struct fileRecord *file, struct sectionRecord *sectionRec)
996{
997 const struct nlist *symbol;
998 const struct section *section;
999 struct relocRecord *rec;
1000 struct relocation_info *rinfo;
1001 unsigned long i;
1002 unsigned long r_address, r_symbolnum, r_length;
1003 enum reloc_type_generic r_type;
1004 UInt8 *sectionBase;
1005 void **entry;
1006
1007 sectionRec->fRelocCache = DataCreate(
1008 sectionRec->fSection->nreloc * sizeof(struct relocRecord));
1009 if (!sectionRec->fRelocCache)
1010 return false;
1011
1012 section = sectionRec->fSection;
1013 sectionBase = file->fMachO + section->offset;
1014
1015 rec = (struct relocRecord *) DataGetPtr(sectionRec->fRelocCache);
1016 rinfo = (struct relocation_info *) (file->fMachO + section->reloff);
1017 for (i = 0; i < section->nreloc; i++, rec++, rinfo++) {
1018
1019 // Totally uninterested in scattered relocation entries
1020 if ( (rinfo->r_address & R_SCATTERED) )
1021 continue;
1022
1023 r_address = rinfo->r_address;
1024 entry = (void **) (sectionBase + r_address);
1025
1026 /*
1027 * The r_address field is really an offset into the contents of the
1028 * section and must reference something inside the section (Note
1029 * that this is not the case for PPC_RELOC_PAIR entries but this
1030 * can't be one with the above checks).
1031 */
1032 return_if(r_address >= section->size, false,
1033 ("Invalid relocation entry in %s - not in section\n", file->fPath));
1034
1035 // If we don't have a VANILLA entry or the Vanilla entry isn't
1036 // a 'long' then ignore the entry and try the next.
1037 r_type = (enum reloc_type_generic) rinfo->r_type;
1038 r_length = rinfo->r_length;
1039 if (r_type != GENERIC_RELOC_VANILLA || r_length != 2)
1040 continue;
1041
1042 r_symbolnum = rinfo->r_symbolnum;
1043
1044 /*
1045 * If rinfo->r_extern is set this relocation entry is an external entry
1046 * else it is a local entry.
1047 */
1048 if (rinfo->r_extern) {
1049 /*
1050 * This is an external relocation entry.
1051 * r_symbolnum is an index into the input file's symbol table
1052 * of the symbol being refered to. The symbol must be
1053 * undefined to be used in an external relocation entry.
1054 */
1055 return_if(r_symbolnum >= file->fSymtab->nsyms, false,
1056 ("Invalid relocation entry in %s - no symbol\n", file->fPath));
1057
1058 /*
1059 * If this is an indirect symbol resolve indirection (all chains
1060 * of indirect symbols have been resolved so that they point at
1061 * a symbol that is not an indirect symbol).
1062 */
1063 symbol = file->fSymbolBase;
1064 if ((symbol[r_symbolnum].n_type & N_TYPE) == N_INDR)
1065 r_symbolnum = symbol[r_symbolnum].n_value;
1066 symbol = &symbol[r_symbolnum];
1067
1068 return_if(symbol->n_type != (N_EXT | N_UNDF), false,
1069 ("Invalid relocation entry in %s - extern\n", file->fPath));
1070 }
1071 else {
1072 /*
1073 * If the symbol is not in any section then it can't be a
1074 * pointer to a local segment and I don't care about it.
1075 */
1076 if (r_symbolnum == R_ABS)
1077 continue;
1078
1079 // Note segment references are offset by 1 from 0.
1080 return_if(r_symbolnum > file->fNSects, false,
1081 ("Invalid relocation entry in %s - local\n", file->fPath));
1082
1083 // Find the symbol, if any, that backs this entry
1084 symbol = findSymbolByAddress(file, *entry);
1085 }
1086
1087 rec->fValue = *entry; // Save the previous value
1088 rec->fRInfo = rinfo; // Save a pointer to the reloc
1089 rec->fSymbol = symbol; // Record the current symbol
1090
1091 *entry = (void *) rec; // Save pointer to record in object image
1092 }
1093
9bccf70c 1094 DataSetLength(sectionRec->fRelocCache, i * sizeof(struct relocRecord));
0b4e3aa0
A
1095 ((struct fileRecord *) file)->fImageDirty = true;
1096
1097 return true;
1098}
1099
1100static const struct nlist *
1101findSymbolRefAtLocation(const struct fileRecord *file,
1102 struct sectionRecord *sctn, void **loc)
1103{
1104 if (file->fIsKernel) {
1105 if (*loc)
1106 return findSymbolByAddress(file, *loc);
1107 }
1108 else if (sctn->fRelocCache || relocateSection(file, sctn)) {
1109 struct relocRecord *reloc = (struct relocRecord *) *loc;
1110
1111 if (DataContainsAddr(sctn->fRelocCache, reloc))
1112 return reloc->fSymbol;
1113 }
1114
1115 return NULL;
1116}
1117
1118static Boolean
1119addClass(struct fileRecord *file,
1120 struct metaClassRecord *inClass,
1121 const char *cname)
1122{
9bccf70c 1123 Boolean result = false;
0b4e3aa0
A
1124 struct metaClassRecord *newClass = NULL;
1125 struct metaClassRecord **fileClasses = NULL;
1126 int len;
1127
0b4e3aa0
A
1128 if (!file->fClassList) {
1129 file->fClassList = DataCreate(0);
1130 if (!file->fClassList)
1131 return false;
1132 }
1133
1134 do {
1135 // Attempt to allocate all necessary resource first
1136 len = strlen(cname) + 1
1137 + (int) (&((struct metaClassRecord *) 0)->fClassName);
1138 newClass = (struct metaClassRecord *) malloc(len);
1139 if (!newClass)
1140 break;
1141
1142 if (!DataAddLength(file->fClassList, sizeof(struct metaClassRecord *)))
1143 break;
1144 fileClasses = (struct metaClassRecord **)
1145 (DataGetPtr(file->fClassList) + DataGetLength(file->fClassList));
1146
9bccf70c
A
1147 // Copy the meta Class structure and string name into newClass and
1148 // insert object at end of the file->fClassList and sMergeMetaClasses
0b4e3aa0
A
1149 *newClass = *inClass;
1150 strcpy(newClass->fClassName, cname);
9bccf70c 1151 fileClasses[-1] = newClass;
0b4e3aa0
A
1152
1153 return true;
1154 } while (0);
1155
1156 if (fileClasses)
1157 DataAddLength(file->fClassList, -sizeof(struct metaClassRecord *));
1158
1159 if (newClass)
1160 free(newClass);
1161
9bccf70c 1162 return result;
0b4e3aa0
A
1163}
1164
1165static struct metaClassRecord *getClass(DataRef classList, const char *cname)
1166{
1167 if (classList) {
1168 int i, nclass;
1169 struct metaClassRecord **classes, *thisClass;
1170
1171 nclass = DataGetLength(classList) / sizeof(struct metaClassRecord *);
1172 classes = (struct metaClassRecord **) DataGetPtr(classList);
1173 for (i = 0; i < nclass; i++) {
1174 thisClass = classes[i];
1175 if (!strcmp(thisClass->fClassName, cname))
1176 return thisClass;
1177 }
1178 }
1179
1180 return NULL;
1181}
1182
1183// Add the class 'cname' to the list of known OSObject based classes
9bccf70c 1184// Note 'sym' is the <cname>10superClassE symbol.
0b4e3aa0
A
1185static Boolean
1186recordClass(struct fileRecord *file, const char *cname, const struct nlist *sym)
1187{
9bccf70c 1188 Boolean result = false;
0b4e3aa0
A
1189 char *supername = NULL;
1190 const char *classname = NULL;
1191 struct metaClassRecord newClass;
1192 char strbuffer[1024];
1193
9bccf70c 1194 // Only do the work to find the super class if we are
0b4e3aa0 1195 // not currently working on the kernel. The kernel is the end
9bccf70c 1196 // of all superclass chains by definition as the kernel must be binary
0b4e3aa0
A
1197 // compatible with itself.
1198 if (!file->fIsKernel) {
9bccf70c 1199 const char *suffix;
0b4e3aa0
A
1200 const struct nlist *supersym;
1201 const struct section *section;
1202 struct sectionRecord *sectionRec;
1203 unsigned char sectind = sym->n_sect;
1204 const char *superstr;
1205 void **location;
9bccf70c 1206 int snamelen;
0b4e3aa0
A
1207
1208 // We can't resolve anything that isn't in a real section
1209 // Note that the sectind is starts at one to make room for the
1210 // NO_SECT flag but the fNSects field isn't offset so we have a
1211 // '>' test. Which means this isn't an OSObject based class
9bccf70c
A
1212 if (sectind == NO_SECT || sectind > file->fNSects) {
1213 result = true;
1214 goto finish;
1215 }
0b4e3aa0
A
1216 sectionRec = file->fSections + sectind - 1;
1217 section = sectionRec->fSection;
1218 location = (void **) ( file->fMachO + section->offset
1219 + sym->n_value - section->addr );
1220
1221 supersym = findSymbolRefAtLocation(file, sectionRec, location);
9bccf70c
A
1222 if (!supersym) {
1223 result = true; // No superclass symbol then it isn't an OSObject.
1224 goto finish;
1225 }
0b4e3aa0 1226
9bccf70c 1227 // Find string in file and skip leading '_' and then find the suffix
0b4e3aa0 1228 superstr = symbolname(file, supersym) + 1;
9bccf70c
A
1229 suffix = superstr + strlen(superstr) - sizeof(kGMetaSuffix) + 1;
1230 if (suffix <= superstr || strcmp(suffix, kGMetaSuffix)) {
1231 result = true; // Not an OSObject superclass so ignore it..
1232 goto finish;
1233 }
0b4e3aa0 1234
9bccf70c
A
1235 // Got a candidate so hand it over for class processing.
1236 snamelen = suffix - superstr - sizeof(kOSObjPrefix) + 2;
1237 supername = (char *) malloc(snamelen + 1);
1238 bcopy(superstr + sizeof(kOSObjPrefix) - 2, supername, snamelen);
1239 supername[snamelen] = '\0';
0b4e3aa0
A
1240 }
1241
1242 do {
1243 break_if(getClass(file->fClassList, cname),
1244 ("Duplicate class %s in %s\n", cname, file->fPath));
9bccf70c 1245
0b4e3aa0
A
1246 snprintf(strbuffer, sizeof(strbuffer), "%s%s", kVTablePrefix, cname);
1247 newClass.fVTableSym = findSymbolByName(file, strbuffer);
1248 break_if(!newClass.fVTableSym,
1249 ("Can't find vtable %s in %s\n", cname, file->fPath));
1250
1251 newClass.fFile = file;
1252 newClass.fSuperName = supername;
1253 newClass.fPatchedVTable = NULL;
1254
1255 // Can't use cname as it may be a stack variable
1256 // However the vtable's string has the class name as a suffix
1257 // so why don't we use that rather than mallocing a string.
1258 classname = symbolname(file, newClass.fVTableSym)
1259 + sizeof(kVTablePrefix) - 1;
9bccf70c 1260 break_if(!addClass(file, &newClass, classname),
0b4e3aa0
A
1261 ("recordClass - no memory?\n"));
1262
9bccf70c
A
1263 supername = NULL;
1264 result = true;
0b4e3aa0 1265 } while (0);
9bccf70c
A
1266
1267finish:
0b4e3aa0
A
1268 if (supername)
1269 free(supername);
1270
9bccf70c 1271 return result;
0b4e3aa0
A
1272}
1273
9bccf70c 1274
0b4e3aa0
A
1275static Boolean getMetaClassGraph(struct fileRecord *file)
1276{
1277 const struct nlist *sym;
0b4e3aa0
A
1278 int i, nsyms;
1279
1280 // Search the symbol table for the local symbols that are generated
1281 // by the metaclass system. There are three metaclass variables
1282 // that are relevant.
1283 //
1284 // <ClassName>.metaClass A pointer to the meta class structure.
1285 // <ClassName>.superClass A pointer to the super class's meta class.
1286 // <ClassName>.gMetaClass The meta class structure itself.
1287 // ___vt<ClassName> The VTable for the class <ClassName>.
1288 //
1289 // In this code I'm going to search for any symbols that
9bccf70c 1290 // ends in k31SuperClassSuffix as this indicates this class is a conforming
0b4e3aa0
A
1291 // OSObject subclass and will need to be patched, and it also
1292 // contains a pointer to the super class's meta class structure.
0b4e3aa0
A
1293 sym = file->fLocalSyms;
1294 for (i = 0, nsyms = file->fNLocal; i < nsyms; i++, sym++) {
1295 const char *symname;
9bccf70c 1296 const char *suffix;
0b4e3aa0
A
1297 char classname[1024];
1298 unsigned char n_type = sym->n_type & (N_TYPE | N_EXT);
9bccf70c 1299 int cnamelen;
0b4e3aa0
A
1300
1301 // Check that the symbols is a global and that it has a name.
1302 if (((N_SECT | N_EXT) != n_type && (N_ABS | N_EXT) != n_type)
1303 || !sym->n_un.n_strx)
1304 continue;
1305
9bccf70c 1306 // Only search from the last *sep* in the symbol.
0b4e3aa0 1307 // but skip the leading '_' in all symbols first.
9bccf70c
A
1308 symname = symbolname(file, sym) + 1;
1309 if (symname[0] != kCPPSymbolPrefix[0]
1310 || symname[1] != kCPPSymbolPrefix[1])
1311 continue;
1312
1313 suffix = symname + strlen(symname) - sizeof(k31SuperClassSuffix) + 1;
1314 if (suffix <= symname || strcmp(suffix, k31SuperClassSuffix))
0b4e3aa0
A
1315 continue;
1316
1317 // Got a candidate so hand it over for class processing.
9bccf70c
A
1318 cnamelen = suffix - symname - sizeof(kOSObjPrefix) + 2;
1319 return_if(cnamelen + 1 >= (int) sizeof(classname),
1320 false, ("Symbol %s is too long", symname));
0b4e3aa0 1321
9bccf70c
A
1322 bcopy(symname + sizeof(kOSObjPrefix) - 2, classname, cnamelen);
1323 classname[cnamelen] = '\0';
0b4e3aa0
A
1324 if (!recordClass(file, classname, sym))
1325 return false;
1326 }
1327
1328 return_if(!file->fClassList, false, ("Internal error, "
1329 "getMetaClassGraph(%s) found no classes", file->fPath));
1330
9bccf70c 1331 DEBUG_LOG(("Found %ld classes in %p for %s\n",
0b4e3aa0
A
1332 DataGetLength(file->fClassList)/sizeof(void*),
1333 file->fClassList, file->fPath));
1334
1335 return true;
1336}
1337
1338static Boolean mergeOSObjectsForFile(const struct fileRecord *file)
1339{
1340 int i, nmerged;
1341 Boolean foundDuplicates = false;
1342
9bccf70c 1343 DEBUG_LOG(("Merging file %s\n", file->fPath)); // @@@ gvdl:
0b4e3aa0
A
1344
1345 if (!file->fClassList)
1346 return true;
1347
1348 if (!sMergedFiles) {
1349 sMergedFiles = DataCreate(0);
1350 return_if(!sMergedFiles, false,
1351 ("Unable to allocate memory metaclass list\n", file->fPath));
1352 }
1353
1354 // Check to see if we have already merged this file
1355 nmerged = DataGetLength(sMergedFiles) / sizeof(struct fileRecord *);
1356 for (i = 0; i < nmerged; i++) {
1357 if (file == ((void **) DataGetPtr(sMergedFiles))[i])
1358 return true;
1359 }
1360
1361 if (!sMergeMetaClasses) {
1362 sMergeMetaClasses = DataCreate(0);
1363 return_if(!sMergeMetaClasses, false,
1364 ("Unable to allocate memory metaclass list\n", file->fPath));
1365 }
1366 else { /* perform a duplicate check */
1367 int i, j, cnt1, cnt2;
1368 struct metaClassRecord **list1, **list2;
1369
1370 list1 = (struct metaClassRecord **) DataGetPtr(file->fClassList);
1371 cnt1 = DataGetLength(file->fClassList) / sizeof(*list1);
1372 list2 = (struct metaClassRecord **) DataGetPtr(sMergeMetaClasses);
1373 cnt2 = DataGetLength(sMergeMetaClasses) / sizeof(*list2);
1374
1375 for (i = 0; i < cnt1; i++) {
1376 for (j = 0; j < cnt2; j++) {
1377 if (!strcmp(list1[i]->fClassName, list2[j]->fClassName)) {
1378 errprintf("duplicate class %s in %s & %s\n",
1379 list1[i]->fClassName,
1380 file->fPath, list2[j]->fFile->fPath);
1381 }
1382 }
1383 }
1384 }
1385 if (foundDuplicates)
1386 return false;
1387
1388 return_if(!DataAppendBytes(sMergedFiles, &file, sizeof(file)), false,
1389 ("Unable to allocate memory to merge %s\n", file->fPath));
1390
1391 return_if(!DataAppendData(sMergeMetaClasses, file->fClassList), false,
1392 ("Unable to allocate memory to merge %s\n", file->fPath));
1393
1394 if (file == sKernelFile)
1395 sMergedKernel = true;
1396
1397 return true;
1398}
1399
1400// Returns a pointer to the base of the section offset by the sections
1401// base address. The offset is so that we can add nlist::n_values directly
1402// to this address and get a valid pointer in our memory.
1403static unsigned char *
1404getSectionForSymbol(const struct fileRecord *file, const struct nlist *symb,
1405 void ***endP)
1406{
1407 const struct section *section;
1408 unsigned char sectind;
1409 unsigned char *base;
1410
1411 sectind = symb->n_sect; // Default to symbols section
1412 if ((symb->n_type & N_TYPE) == N_ABS && file->fIsKernel) {
1413 // Absolute symbol so we have to iterate over our sections
1414 for (sectind = 1; sectind <= file->fNSects; sectind++) {
1415 unsigned long start, end;
1416
1417 section = file->fSections[sectind - 1].fSection;
1418 start = section->addr;
1419 end = start + section->size;
1420 if (start <= symb->n_value && symb->n_value < end) {
1421 // Found the relevant section
1422 break;
1423 }
1424 }
1425 }
1426
1427 // Is the vtable in a valid section?
1428 return_if(sectind == NO_SECT || sectind > file->fNSects,
1429 (unsigned char *) -1,
1430 ("%s isn't a valid kext, bad section reference\n", file->fPath));
1431
1432 section = file->fSections[sectind - 1].fSection;
1433
1434 // for when we start walking the vtable so compute offset's now.
1435 base = file->fMachO + section->offset;
1436 *endP = (void **) (base + section->size);
1437
1438 return base - section->addr; // return with addr offset
1439}
1440
1441static Boolean resolveKernelVTable(struct metaClassRecord *metaClass)
1442{
1443 const struct fileRecord *file;
1444 struct patchRecord *patchedVTable;
1445 void **curEntry, **vtableEntries, **endSection;
1446 unsigned char *sectionBase;
1447 struct patchRecord *curPatch;
1448 int classSize;
1449
1450 // Should never occur but it doesn't cost us anything to check.
1451 if (metaClass->fPatchedVTable)
1452 return true;
1453
9bccf70c 1454 DEBUG_LOG(("Kernel vtable %s\n", metaClass->fClassName)); // @@@ gvdl:
0b4e3aa0
A
1455
1456 // Do we have a valid vtable to patch?
1457 return_if(!metaClass->fVTableSym,
1458 false, ("Internal error - no class vtable symbol?\n"));
1459
1460 file = metaClass->fFile;
1461
1462 // If the metaClass we are being to ask is in the kernel then we
1463 // need to do a quick scan to grab the fPatchList in a reliable format
1464 // however we don't need to check the superclass in the kernel
1465 // as the kernel vtables are always correct wrt themselves.
1466 // Note this ends the superclass chain recursion.
1467 return_if(!file->fIsKernel,
1468 false, ("Internal error - resolveKernelVTable not kernel\n"));
1469
1470 if (file->fNoKernelExecutable) {
1471 // Oh dear attempt to map the kernel's VM into my memory space
1472 return_if(file->fNoKernelExecutable, false,
1473 ("Internal error - fNoKernelExecutable not implemented yet\n"));
1474 }
1475
1476 // We are going to need the base and the end
1477 sectionBase = getSectionForSymbol(file, metaClass->fVTableSym, &endSection);
1478 if (-1 == (long) sectionBase)
1479 return false;
1480
1481 vtableEntries = (void **) (sectionBase + metaClass->fVTableSym->n_value);
1482 curEntry = vtableEntries + kVTablePreambleLen;
1483 for (classSize = 0; curEntry < endSection && *curEntry; classSize++)
1484 curEntry++;
1485
1486 return_if(*curEntry, false, ("Bad kernel image, short section\n"));
1487
1488 patchedVTable = (struct patchRecord *)
1489 malloc((classSize + 1) * sizeof(struct patchRecord));
1490 return_if(!patchedVTable, false, ("resolveKernelVTable - no memory\n"));
1491
1492 // Copy the vtable of this class into the patch table
1493 curPatch = patchedVTable;
1494 curEntry = vtableEntries + kVTablePreambleLen;
1495 for (; *curEntry; curEntry++, curPatch++) {
1496 curPatch->fSymbol = (struct nlist *)
1497 findSymbolByAddress(file, *curEntry);
1498 curPatch->fType = kSymbolLocal;
1499 }
1500
1501 // Tag the end of the patch vtable
1502 curPatch->fSymbol = NULL;
1503 metaClass->fPatchedVTable = patchedVTable;
1504
1505 return true;
1506}
1507
9bccf70c
A
1508static const char *addNewString(struct fileRecord *file,
1509 const char *strname, int namelen)
1510{
1511 DataRef strings = 0;
1512 const char *newStr;
1513
1514 namelen++; // Include terminating '\0';
1515
1516 // Make sure we have a string table as well for this symbol
1517 if (file->fNewStringBlocks) {
1518 DataRef *blockTable = (DataRef *) DataGetPtr(file->fNewStringBlocks);
1519 int index = DataGetLength(file->fNewStringBlocks) / sizeof(DataRef*);
1520 strings = blockTable[index - 1];
1521 if (DataRemaining(strings) < namelen)
1522 strings = 0;
1523 }
1524 else
1525 {
1526 file->fNewStringBlocks = DataCreate(0);
1527 return_if(!file->fNewStringBlocks, NULL,
1528 ("Unable to allocate new string table %s\n", file->fPath));
1529 }
1530
1531 if (!strings) {
1532 int size = (namelen + 1023) & ~1023;
1533 if (size < 16 * 1024)
1534 size = 16 * 1024;
1535 strings = DataCreate(size);
1536 return_if(!strings, NULL,
1537 ("Unable to allocate new string block %s\n", file->fPath));
1538 return_if(
1539 !DataAppendBytes(file->fNewStringBlocks, &strings, sizeof(strings)),
1540 false, ("Unable to allocate string table for %s\n", file->fPath));
1541 }
1542
1543 newStr = DataGetEndPtr(strings);
1544 DataAppendBytes(strings, strname, namelen);
1545 return newStr;
1546}
1547
1548// reloc->fPatch must contain a valid pointer
0b4e3aa0
A
1549static struct nlist *
1550getNewSymbol(struct fileRecord *file,
1551 const struct relocRecord *reloc, const char *supername)
1552{
9bccf70c 1553 unsigned int size, i;
0b4e3aa0
A
1554 struct nlist **sym;
1555 struct nlist *msym;
0b4e3aa0 1556 struct relocation_info *rinfo;
9bccf70c 1557 const char *newStr;
0b4e3aa0
A
1558
1559 if (!file->fNewSymbols) {
1560 file->fNewSymbols = DataCreate(0);
1561 return_if(!file->fNewSymbols, NULL,
1562 ("Unable to allocate new symbol table for %s\n", file->fPath));
1563 }
1564
0b4e3aa0
A
1565 rinfo = (struct relocation_info *) reloc->fRInfo;
1566 size = DataGetLength(file->fNewSymbols) / sizeof(struct nlist *);
9bccf70c 1567 sym = (struct nlist **) DataGetPtr(file->fNewSymbols);
0b4e3aa0 1568 for (i = 0; i < size; i++, sym++) {
9bccf70c
A
1569 int symnum = i + file->fSymtab->nsyms;
1570 newStr = symNameByIndex(file, symnum);
1571 if (!strcmp(newStr, supername)) {
1572 rinfo->r_symbolnum = symnum;
0b4e3aa0
A
1573 file->fSymbolsDirty = true;
1574 return *sym;
1575 }
1576 }
1577
1578 // Assert that this is a vaild symbol. I need this condition to be true
1579 // for the later code to make non-zero. So the first time through I'd
1580 // better make sure that it is 0.
1581 return_if(reloc->fSymbol->n_sect, NULL,
1582 ("Undefined symbol entry with non-zero section %s:%s\n",
1583 file->fPath, symbolname(file, reloc->fSymbol)));
1584
9bccf70c 1585 // If we are here we didn't find the symbol so create a new one now
0b4e3aa0
A
1586 msym = (struct nlist *) malloc(sizeof(struct nlist));
1587 return_if(!msym,
9bccf70c
A
1588 NULL, ("Unable to create symbol table entry for %s", file->fPath));
1589 return_if(!DataAppendBytes(file->fNewSymbols, &msym, sizeof(msym)),
0b4e3aa0 1590 NULL, ("Unable to grow symbol table for %s\n", file->fPath));
0b4e3aa0 1591
9bccf70c
A
1592 newStr = addNewString(file, supername, strlen(supername));
1593 if (!newStr)
1594 return NULL;
1595 // If we are here we didn't find the symbol so create a new one now
1596 return_if(!DataAppendBytes(file->fSym2Strings, &newStr, sizeof(newStr)),
1597 NULL, ("Unable to grow symbol table for %s\n", file->fPath));
1598 file->fSymbToStringTable = (const char **) DataGetPtr(file->fSym2Strings);
0b4e3aa0
A
1599
1600 // Offset the string index by the original string table size
1601 // and negate the address to indicate that this is a 'new' symbol
9bccf70c 1602 msym->n_un.n_strx = -1;
0b4e3aa0
A
1603 msym->n_type = (N_EXT | N_UNDF);
1604 msym->n_sect = NO_SECT;
1605 msym->n_desc = 0;
9bccf70c 1606 msym->n_value = (unsigned long) newStr;
0b4e3aa0
A
1607
1608 // Mark the old symbol as being potentially deletable I can use the
1609 // n_sect field as the input symbol must be of type N_UNDF which means
1610 // that the n_sect field must be set to NO_SECT otherwise it is an
9bccf70c
A
1611 // invalid input file.
1612 //
1613 // However the symbol may have been just inserted by the fixOldSymbol path.
1614 // If this is the case then we know it is in use and we don't have to
1615 // mark it as a deletable symbol.
1616 if (reloc->fSymbol->n_un.n_strx >= 0) {
1617 ((struct nlist *) reloc->fSymbol)->n_un.n_strx
1618 = -reloc->fSymbol->n_un.n_strx;
1619 ((struct nlist *) reloc->fSymbol)->n_sect = (unsigned char) -1;
1620 }
0b4e3aa0
A
1621
1622 rinfo->r_symbolnum = i + file->fSymtab->nsyms;
1623 file->fSymbolsDirty = true;
1624 return msym;
1625}
1626
1627static struct nlist *
1628fixOldSymbol(struct fileRecord *file,
1629 const struct relocRecord *reloc, const char *supername)
1630{
1631 unsigned int namelen;
1632 struct nlist *sym = (struct nlist *) reloc->fSymbol;
1633 const char *oldname = symbolname(file, sym);
1634
1635 // assert(sym->n_un.n_strx >= 0);
1636
1637 namelen = strlen(supername);
9bccf70c
A
1638
1639 sym->n_un.n_strx = -sym->n_un.n_strx;
1640 if (oldname && namelen < strlen(oldname))
1641 {
0b4e3aa0
A
1642 // Overwrite old string in string table
1643 strcpy((char *) oldname, supername);
9bccf70c
A
1644 file->fSymbolsDirty = true;
1645 return sym;
0b4e3aa0 1646 }
0b4e3aa0 1647
9bccf70c
A
1648 oldname = addNewString(file, supername, namelen);
1649 if (!oldname)
1650 return NULL;
0b4e3aa0 1651
9bccf70c 1652 file->fSymbToStringTable[sym - file->fSymbolBase] = oldname;
0b4e3aa0
A
1653 file->fSymbolsDirty = true;
1654 return sym;
1655}
1656
1657static enum patchState
1658symbolCompare(const struct fileRecord *file,
1659 const struct nlist *classsym,
1660 const char *supername)
1661{
1662 const char *classname;
1663
1664
1665 // Check to see if the target function is locally defined
1666 // if it is then we can assume this is a local vtable override
1667 if ((classsym->n_type & N_TYPE) != N_UNDF)
1668 return kSymbolLocal;
1669
1670 // Check to see if both symbols point to the same symbol name
1671 // if so then we are still identical.
1672 classname = symbolname(file, classsym);
1673 if (!strcmp(classname, supername))
1674 return kSymbolIdentical;
1675
9bccf70c 1676 // We know that the target's vtable entry is different from the
0b4e3aa0
A
1677 // superclass' vtable entry. This means that we will have to apply a
1678 // patch to the current entry, however before returning lets check to
1679 // see if we have a _RESERVEDnnn field 'cause we can use this as a
1680 // registration point that must align between vtables.
9bccf70c 1681 if (strstr(supername, kReservedNamePrefix))
0b4e3aa0
A
1682 return kSymbolMismatch;
1683
1684 // OK, we have a superclass difference where the superclass doesn't
1685 // reference a pad function so assume that the superclass is correct.
9bccf70c 1686 if (strstr(classname, kReservedNamePrefix))
0b4e3aa0
A
1687 return kSymbolPadUpdate;
1688 else
1689 return kSymbolSuperUpdate;
1690}
1691
1692static Boolean patchVTable(struct metaClassRecord *metaClass)
1693{
1694 struct metaClassRecord *super = NULL;
1695 struct fileRecord *file;
1696 struct patchRecord *patchedVTable;
1697 struct relocRecord **curReloc, **vtableRelocs, **endSection;
1698 unsigned char *sectionBase;
1699 int classSize;
1700
1701 // Should never occur but it doesn't cost us anything to check.
1702 if (metaClass->fPatchedVTable)
1703 return true;
1704
1705 // Do we have a valid vtable to patch?
1706 return_if(!metaClass->fVTableSym,
1707 false, ("Internal error - no class vtable symbol?\n"));
1708
1709 file = metaClass->fFile;
1710
1711 // If the metaClass we are being to ask is in the kernel then we
1712 // need to do a quick scan to grab the fPatchList in a reliable format
1713 // however we don't need to check the superclass in the kernel
1714 // as the kernel vtables are always correct wrt themselves.
1715 // Note this ends the superclass chain recursion.
1716 return_if(file->fIsKernel,
1717 false, ("Internal error - patchVTable shouldn't used for kernel\n"));
1718
1719 if (!metaClass->fSuperName)
1720 return false;
1721
1722 // The class isn't in the kernel so make sure that the super class
1723 // is patched before patching ouselves.
1724 super = getClass(sMergeMetaClasses, metaClass->fSuperName);
9bccf70c 1725 return_if(!super, false, ("Can't find superclass for %s : %s\n",
0b4e3aa0
A
1726 metaClass->fClassName, metaClass->fSuperName));
1727
1728 // Superclass recursion if necessary
1729 if (!super->fPatchedVTable) {
1730 Boolean res;
1731
1732 if (super->fFile->fIsKernel)
1733 res = resolveKernelVTable(super);
1734 else
1735 res = patchVTable(super);
1736 if (!res)
1737 return false;
1738 }
1739
9bccf70c 1740 DEBUG_LOG(("Patching %s\n", metaClass->fClassName)); // @@@ gvdl:
0b4e3aa0
A
1741
1742 // We are going to need the base and the end
1743
1744 sectionBase = getSectionForSymbol(file,
1745 metaClass->fVTableSym, (void ***) &endSection);
1746 if (-1 == (long) sectionBase)
1747 return false;
1748
1749 vtableRelocs = (struct relocRecord **)
1750 (sectionBase + metaClass->fVTableSym->n_value);
1751 curReloc = vtableRelocs + kVTablePreambleLen;
1752 for (classSize = 0; curReloc < endSection && *curReloc; classSize++)
1753 curReloc++;
1754
1755 return_if(*curReloc, false,
1756 ("%s isn't a valid kext, short section\n", file->fPath));
1757
1758 patchedVTable = (struct patchRecord *)
1759 malloc((classSize + 1) * sizeof(struct patchRecord));
1760 return_if(!patchedVTable, false, ("patchedVTable - no memory\n"));
1761
1762 do {
1763 struct patchRecord *curPatch;
1764 struct nlist *symbol;
1765
1766 curPatch = patchedVTable;
1767 curReloc = vtableRelocs + kVTablePreambleLen;
1768
1769 // Grab the super table patches if necessary
1770 // Can't be patching a kernel table as we don't walk super
1771 // class chains in the kernel symbol space.
1772 if (super && super->fPatchedVTable) {
1773 const struct patchRecord *spp;
1774
1775 spp = super->fPatchedVTable;
1776
1777 for ( ; spp->fSymbol; curReloc++, spp++, curPatch++) {
1778 const char *supername =
1779 symbolname(super->fFile, spp->fSymbol);
1780
1781 symbol = (struct nlist *) (*curReloc)->fSymbol;
1782
1783 curPatch->fType = symbolCompare(file, symbol, supername);
1784 switch (curPatch->fType) {
1785 case kSymbolIdentical:
1786 case kSymbolLocal:
1787 break;
1788
1789 case kSymbolSuperUpdate:
1790 symbol = getNewSymbol(file, (*curReloc), supername);
1791 break;
1792
1793 case kSymbolPadUpdate:
1794 symbol = fixOldSymbol(file, (*curReloc), supername);
1795 break;
1796
1797 case kSymbolMismatch:
9bccf70c
A
1798 errprintf("%s is not compatible with its superclass, "
1799 "%s superclass changed?\n",
0b4e3aa0
A
1800 metaClass->fClassName, super->fClassName);
1801 goto abortPatch;
1802
1803 default:
1804 errprintf("Internal error - unknown patch type\n");
1805 goto abortPatch;
1806 }
1807 if (symbol) {
1808 curPatch->fSymbol = symbol;
1809 (*curReloc)->fSymbol = symbol;
1810 }
1811 else
1812 goto abortPatch;
1813 }
1814 }
1815
1816 // Copy the remainder of this class' vtable into the patch table
1817 for (; *curReloc; curReloc++, curPatch++) {
1818 // Local reloc symbols
1819 curPatch->fType = kSymbolLocal;
1820 curPatch->fSymbol = (struct nlist *) (*curReloc)->fSymbol;
1821 }
1822
1823 // Tag the end of the patch vtable
1824 curPatch->fSymbol = NULL;
1825
1826 metaClass->fPatchedVTable = patchedVTable;
1827 return true;
1828 } while(0);
1829
1830abortPatch:
1831 if (patchedVTable)
1832 free(patchedVTable);
1833
1834 return false;
1835}
1836
1837static Boolean growImage(struct fileRecord *file, vm_size_t delta)
1838{
1839#if !KERNEL
1840 file->fMachOSize += delta;
1841 return (file->fMachO + file->fMachOSize <= file->fPadEnd);
1842#else /* KERNEL */
1843 vm_address_t startMachO, endMachO, endMap;
1844 vm_offset_t newMachO;
1845 vm_size_t newsize;
9bccf70c 1846 unsigned long i, last = 0;
0b4e3aa0
A
1847 struct metaClassRecord **classes = NULL;
1848 struct sectionRecord *section;
1849 kern_return_t ret;
1850
1851 startMachO = (vm_address_t) file->fMachO;
1852 endMachO = startMachO + file->fMachOSize + delta;
1853 endMap = (vm_address_t) file->fMap + file->fMapSize;
1854
1855 // Do we have room in the current mapped image
d7e50217 1856 if (endMachO < round_page_32(endMap)) {
0b4e3aa0
A
1857 file->fMachOSize += delta;
1858 return true;
1859 }
1860
1861 newsize = endMachO - startMachO;
d7e50217 1862 if (newsize < round_page_32(file->fMapSize)) {
9bccf70c
A
1863 DEBUG_LOG(("Growing image %s by moving\n", file->fPath));
1864
0b4e3aa0
A
1865 // We have room in the map if we shift the macho image within the
1866 // current map. We will have to patch up pointers into the object.
1867 newMachO = (vm_offset_t) file->fMap;
1868 bcopy((char *) startMachO, (char *) newMachO, file->fMachOSize);
1869 }
1870 else if (file->fIsKmem) {
1871 // kmem_alloced mapping so we can try a kmem_realloc
1872 ret = kmem_realloc(kernel_map,
1873 (vm_address_t) file->fMap,
1874 (vm_size_t) file->fMapSize,
1875 &newMachO,
1876 newsize);
1877 if (KERN_SUCCESS != ret)
1878 return false;
1879
1880 // If the mapping didn't move then just return
1881 if ((vm_address_t) file->fMap == newMachO) {
1882 file->fMachOSize = file->fMapSize = newsize;
1883 return true;
1884 }
1885
9bccf70c 1886 DEBUG_LOG(("Growing image %s by reallocing\n", file->fPath));
0b4e3aa0
A
1887 // We have relocated the kmem image so we are going to have to
1888 // move all of the pointers into the image around.
1889 }
1890 else {
9bccf70c 1891 DEBUG_LOG(("Growing image %s by allocating\n", file->fPath));
0b4e3aa0
A
1892 // The image doesn't have room for us and I can't kmem_realloc
1893 // then I just have to bite the bullet and copy the object code
1894 // into a bigger memory segment
1895 ret = kmem_alloc(kernel_map, &newMachO, newsize);
1896
1897 if (KERN_SUCCESS != ret)
1898 return false;
1899 bcopy((char *) startMachO, (void *) newMachO, file->fMachOSize);
1900 file->fIsKmem = true;
1901 }
1902
1903
1904 file->fMap = file->fMachO = (unsigned char *) newMachO;
1905 file->fMapSize = newsize;
1906 file->fMachOSize += delta; // Increment the image size
1907
1908 // If we are here then we have shifted the object image in memory
1909 // I really should change all of my pointers into the image to machO offsets
1910 // but I have run out of time. So I'm going to very quickly go over the
1911 // cached data structures and add adjustments to the addresses that are
1912 // affected. I wonder how long it will take me to get them all.
1913 //
1914 // For every pointer into the MachO I need to add an adjustment satisfying
1915 // the following simultanous equations
1916 // addr_old = macho_old + fixed_offset
1917 // addr_new = macho_new + fixed_offset therefore:
1918 // addr_new = addr_old + (macho_new - macho_old)
1919#define REBASE(addr, delta) ( ((vm_address_t) (addr)) += (delta) )
1920 delta = newMachO - startMachO;
1921
9bccf70c 1922 // Rebase the cached-in object 'struct symtab_command' pointer
0b4e3aa0
A
1923 REBASE(file->fSymtab, delta);
1924
9bccf70c 1925 // Rebase the cached-in object 'struct nlist' pointer for all symbols
0b4e3aa0
A
1926 REBASE(file->fSymbolBase, delta);
1927
9bccf70c 1928 // Rebase the cached-in object 'struct nlist' pointer for local symbols
0b4e3aa0
A
1929 REBASE(file->fLocalSyms, delta);
1930
9bccf70c 1931 // Rebase the cached-in object 'char' pointer for the string table
0b4e3aa0
A
1932 REBASE(file->fStringBase, delta);
1933
1934 // Ok now we have to go over all of the relocs one last time
1935 // to clean up the pad updates which had their string index negated
1936 // to indicate that we have finished with them.
1937 section = file->fSections;
9bccf70c 1938 for (i = 0, last = file->fNSects; i < last; i++, section++)
0b4e3aa0
A
1939 REBASE(section->fSection, delta);
1940
1941 // We only ever grow images that contain class lists so dont bother
1942 // the check if file->fClassList is non-zero 'cause it can't be
1943 // assert(file->fClassList);
9bccf70c 1944 last = DataGetLength(file->fClassList)
0b4e3aa0
A
1945 / sizeof(struct metaClassRecord *);
1946 classes = (struct metaClassRecord **) DataGetPtr(file->fClassList);
9bccf70c 1947 for (i = 0; i < last; i++) {
0b4e3aa0
A
1948 struct patchRecord *patch;
1949
1950 for (patch = classes[i]->fPatchedVTable; patch->fSymbol; patch++) {
1951 vm_address_t symAddr = (vm_address_t) patch->fSymbol;
9bccf70c
A
1952
1953 // Only need to rebase if the symbol is part of the image
1954 // If this is a new symbol then it was independantly allocated
0b4e3aa0
A
1955 if (symAddr >= startMachO && symAddr < endMachO)
1956 REBASE(patch->fSymbol, delta);
1957 }
1958 }
1959
9bccf70c
A
1960 // Finally rebase all of the string table pointers
1961 last = file->fSymtab->nsyms;
1962 for (i = 0; i < last; i++)
1963 REBASE(file->fSymbToStringTable[i], delta);
0b4e3aa0
A
1964
1965#undef REBASE
1966
1967 return true;
1968
1969#endif /* KERNEL */
1970}
1971
1972static Boolean
1973prepareFileForLink(struct fileRecord *file)
1974{
1975 unsigned long i, last, numnewsyms, newsymsize, newstrsize;
1976 struct sectionRecord *section;
1977 struct nlist **symp, *sym;
9bccf70c 1978 DataRef newStrings, *stringBlocks;
0b4e3aa0
A
1979
1980 // If we didn't even do a pseudo 'relocate' and dirty the image
1981 // then we can just return now.
1982 if (!file->fImageDirty)
1983 return true;
1984
1985DEBUG_LOG(("Linking 2 %s\n", file->fPath)); // @@@ gvdl:
1986
1987 // We have to go over all of the relocs to repair the damage
1988 // that we have done to the image when we did our 'relocation'
1989 section = file->fSections;
1990 for (i = 0, last = file->fNSects; i < last; i++, section++) {
1991 unsigned char *sectionBase;
1992 struct relocRecord *rec;
1993 unsigned long j, nreloc;
1994
1995 if (section->fRelocCache) {
1996 sectionBase = file->fMachO + section->fSection->offset;
1997 nreloc = section->fSection->nreloc;
1998 rec = (struct relocRecord *) DataGetPtr(section->fRelocCache);
1999
2000 // We will need to repair the reloc list
2001 for (j = 0; j < nreloc; j++, rec++) {
2002 void **entry;
2003 struct nlist *sym;
2004
2005 // Repair Damage to object image
2006 entry = (void **) (sectionBase + rec->fRInfo->r_address);
2007 *entry = rec->fValue;
2008
2009 // Check if the symbol that this relocation entry points
2010 // to is marked as erasable
2011 sym = (struct nlist *) rec->fSymbol;
2012 if (sym && sym->n_type == (N_EXT | N_UNDF)
2013 && sym->n_sect == (unsigned char) -1) {
9bccf70c 2014 // It is in use so we better clear the mark
0b4e3aa0
A
2015 sym->n_un.n_strx = -sym->n_un.n_strx;
2016 sym->n_sect = NO_SECT;
2017 }
2018 }
2019
2020 // Clean up the fRelocCache we don't need it any more.
2021 DataRelease(section->fRelocCache);
2022 section->fRelocCache = 0;
2023 }
2024 }
2025 file->fImageDirty = false; // Image is clean
2026
2027 // If we didn't dirty the symbol table then just return
2028 if (!file->fSymbolsDirty)
2029 return true;
2030
2031 // calculate total file size increase and check against padding
9bccf70c
A
2032 if (file->fNewSymbols) {
2033 numnewsyms = DataGetLength(file->fNewSymbols);
2034 symp = (struct nlist **) DataGetPtr(file->fNewSymbols);
2035 }
2036 else {
2037 numnewsyms = 0;
2038 symp = 0;
2039 }
0b4e3aa0 2040 numnewsyms /= sizeof(struct nlist *);
9bccf70c
A
2041 file->fSymtab->nsyms += numnewsyms;
2042
2043 // old sting size + 30% rounded up to nearest page
2044 newstrsize = file->fSymtab->strsize * 21 / 16;
2045 newstrsize = (newstrsize + PAGE_MASK) & ~PAGE_MASK;
2046 newStrings = DataCreate(newstrsize);
2047 return_if(!newStrings, false,
2048 ("Unable to allocate a copy aside buffer, no memory\n"));
2049
0b4e3aa0 2050 newsymsize = numnewsyms * sizeof(struct nlist);
9bccf70c
A
2051 file->fStringBase += newsymsize;
2052 file->fSymtab->stroff += newsymsize;
2053
2054 last = file->fSymtab->nsyms - numnewsyms;
2055 newstrsize = 0;
2056 DataAppendBytes(newStrings, &newstrsize, 4); // Leading nuls
2057 sym = file->fSymbolBase;
2058
2059 // Pre-compute an already offset new symbol pointer. The offset is the
2060 // orignal symbol table.
2061 symp -= last;
2062 for (i = 0; i < file->fSymtab->nsyms; i++, sym++) {
2063 const char *str = symNameByIndex(file, i);
2064 int len = strlen(str) + 1;
2065 unsigned int strx;
2066
2067 // Rebase sym in the new symbol region
2068 if (i >= last)
2069 sym = symp[i];
2070
2071 if (sym->n_un.n_strx < 0 && sym->n_type == (N_EXT | N_UNDF)
2072 && (unsigned char) -1 == sym->n_sect) {
2073 // after patching we find that this symbol is no longer in
2074 // use. So invalidate it by converting it into an N_ABS
2075 // symbol, remove the external bit and null out the name.
2076 bzero(sym, sizeof(*sym));
2077 sym->n_type = N_ABS;
2078 }
2079 else {
2080 // Repair the symbol for the getNewSymbol case.
2081 if (-1 == sym->n_un.n_strx)
2082 sym->n_value = 0;
2083
2084 // Record the offset of the string in the new table
2085 strx = DataGetLength(newStrings);
2086 return_if(!DataAppendBytes(newStrings, str, len), false,
2087 ("Unable to append string, no memory\n"));
2088
2089 sym->n_un.n_strx = strx;
2090 file->fSymbToStringTable[i] = file->fStringBase + strx;
2091 }
2092 }
2093
2094 // Don't need the new strings any more
2095 last = DataGetLength(file->fNewStringBlocks) / sizeof(DataRef);
2096 stringBlocks = (DataRef *) DataGetPtr(file->fNewStringBlocks);
2097 for (i = 0; i < last; i++)
2098 DataRelease(stringBlocks[i]);
2099
2100 DataRelease(file->fNewStringBlocks);
2101 file->fNewStringBlocks = 0;
2102
2103 newstrsize = DataGetLength(newStrings);
0b4e3aa0 2104 newstrsize = (newstrsize + 3) & ~3; // Round to nearest word
9bccf70c
A
2105 return_if(
2106 !growImage(file, newsymsize + newstrsize - file->fSymtab->strsize),
0b4e3aa0
A
2107 false, ("Unable to patch the extension, no memory\n", file->fPath));
2108
2109 // Push out the new symbol table if necessary
2110 if (numnewsyms) {
2111 caddr_t base;
2112
9bccf70c 2113 // Append the new symbols to the original symbol table.
0b4e3aa0 2114 base = (caddr_t) file->fSymbolBase
9bccf70c 2115 + (file->fSymtab->nsyms - numnewsyms) * sizeof(struct nlist);
0b4e3aa0
A
2116 symp = (struct nlist **) DataGetPtr(file->fNewSymbols);
2117 for (i = 0; i < numnewsyms; i++, base += sizeof(struct nlist), symp++)
2118 bcopy(*symp, base, sizeof(struct nlist));
0b4e3aa0
A
2119
2120 DataRelease(file->fNewSymbols);
2121 file->fNewSymbols = 0;
2122 }
2123
2124 // Push out the new string table if necessary
9bccf70c
A
2125 if (newStrings) {
2126 unsigned long *base = (unsigned long *) file->fStringBase;
2127 unsigned long actuallen = DataGetLength(newStrings);
0b4e3aa0
A
2128
2129 // Set the last word in string table to zero before copying data
9bccf70c 2130 base[(newstrsize / sizeof(unsigned long)) - 1] = 0;
0b4e3aa0 2131
9bccf70c
A
2132 // Now copy the new strings back to the end of the file
2133 bcopy((caddr_t) DataGetPtr(newStrings), file->fStringBase, actuallen);
0b4e3aa0 2134
9bccf70c 2135 file->fSymtab->strsize = newstrsize;
0b4e3aa0 2136
9bccf70c 2137 DataRelease(newStrings);
0b4e3aa0
A
2138 }
2139
0b4e3aa0
A
2140 file->fSymbolsDirty = false;
2141
2142 return true;
2143}
2144
2145Boolean
2146#if KERNEL
2147kld_file_map(const char *pathName,
2148 unsigned char *map,
2149 size_t mapSize,
2150 Boolean isKmem)
2151#else
2152kld_file_map(const char *pathName)
2153#endif /* KERNEL */
2154{
2155 struct fileRecord file, *fp = 0;
2156
2157 // Already done no need to repeat
2158 fp = getFile(pathName);
2159 if (fp)
2160 return true;
2161
2162 bzero(&file, sizeof(file));
0b4e3aa0
A
2163
2164#if KERNEL
2165 file.fMap = map;
2166 file.fMapSize = mapSize;
2167 file.fIsKmem = isKmem;
2168#else
9bccf70c 2169 if (!mapObjectFile(&file, pathName))
0b4e3aa0
A
2170 return false;
2171#endif /* KERNEL */
2172
2173 do {
2174 const struct machOMapping {
2175 struct mach_header h;
2176 struct load_command c[1];
2177 } *machO;
2178 const struct load_command *cmd;
9bccf70c 2179 int i;
0b4e3aa0 2180
9bccf70c 2181 if (!findBestArch(&file, pathName))
0b4e3aa0 2182 break;
9bccf70c 2183
0b4e3aa0
A
2184 machO = (const struct machOMapping *) file.fMachO;
2185 if (file.fMachOSize < machO->h.sizeofcmds)
2186 break;
2187
9bccf70c
A
2188 file.fIsKernel = (MH_EXECUTE == machO->h.filetype);
2189
0b4e3aa0
A
2190 // If the file type is MH_EXECUTE then this must be a kernel
2191 // as all Kernel extensions must be of type MH_OBJECT
2192 for (i = 0, cmd = &machO->c[0]; i < machO->h.ncmds; i++) {
9bccf70c 2193 if (cmd->cmd == LC_SYMTAB)
0b4e3aa0 2194 file.fSymtab = (struct symtab_command *) cmd;
9bccf70c
A
2195 else if (cmd->cmd == LC_SEGMENT) {
2196 struct segment_command *seg = (struct segment_command *) cmd;
2197 int nsects = seg->nsects;
2198
2199 if (nsects)
2200 return_if(!parseSegments(&file, seg),
2201 false, ("%s isn't a valid mach-o, bad segment",
2202 pathName));
2203 else if (file.fIsKernel) {
2204#if KERNEL
2205 // We don't need to look for the LinkEdit segment unless
2206 // we are running in the kernel environment.
2207 if (!strcmp(kLinkEditSegName, seg->segname))
2208 file.fLinkEditSeg = seg;
2209#endif
2210 }
2211 }
0b4e3aa0
A
2212
2213 cmd = (struct load_command *) ((UInt8 *) cmd + cmd->cmdsize);
2214 }
2215 break_if(!file.fSymtab,
9bccf70c 2216 ("%s isn't a valid mach-o, no symbols\n", pathName));
0b4e3aa0 2217
9bccf70c
A
2218 if (!parseSymtab(&file, pathName))
2219 break;
0b4e3aa0 2220
9bccf70c 2221 fp = addFile(&file, pathName);
0b4e3aa0
A
2222 if (!fp)
2223 break;
2224
9bccf70c 2225 if (file.fFoundOSObject && !getMetaClassGraph(fp))
0b4e3aa0
A
2226 break;
2227
2228 if (file.fIsKernel)
2229 sKernelFile = fp;
9bccf70c 2230
0b4e3aa0 2231#if KERNEL
9bccf70c
A
2232 // Automatically load the kernel's link edit segment if we are
2233 // attempting to load a driver.
0b4e3aa0
A
2234 if (!sKernelFile) {
2235 extern struct mach_header _mh_execute_header;
2236 extern struct segment_command *getsegbyname(char *seg_name);
2237
2238 struct segment_command *sg;
2239 size_t kernelSize;
2240 Boolean ret;
2241
2242 sg = (struct segment_command *) getsegbyname(kLinkEditSegName);
2243 break_if(!sg, ("Can't find kernel link edit segment\n"));
2244
2245 kernelSize = sg->vmaddr + sg->vmsize - (size_t) &_mh_execute_header;
2246 ret = kld_file_map(kld_basefile_name,
2247 (unsigned char *) &_mh_execute_header, kernelSize,
2248 /* isKmem */ false);
2249 break_if(!ret, ("kld can't map kernel file"));
2250 }
2251#endif /* KERNEL */
2252
2253 return true;
2254 } while(0);
2255
9bccf70c
A
2256 // Failure path, then clean up
2257 if (fp)
2258 // @@@ gvdl: for the time being leak the file ref in the file table
2259 removeFile(fp);
2260 else
2261 unmapFile(&file);
0b4e3aa0
A
2262
2263 return false;
2264}
2265
2266void *kld_file_getaddr(const char *pathName, long *size)
2267{
2268 struct fileRecord *file = getFile(pathName);
2269
2270 if (!file)
2271 return 0;
2272
2273 if (size)
2274 *size = file->fMachOSize;
2275
2276 return file->fMachO;
2277}
2278
2279void *kld_file_lookupsymbol(const char *pathName, const char *symname)
2280{
2281 struct fileRecord *file = getFile(pathName);
2282 const struct nlist *sym;
2283 const struct section *section;
2284 unsigned char *sectionBase;
2285 unsigned char sectind;
2286
2287 return_if(!file,
2288 NULL, ("Unknown file %s\n", pathName));
2289
2290 sym = findSymbolByName(file, symname);
2291
2292 // May be a non-extern symbol so look for it there
2293 if (!sym) {
0b4e3aa0
A
2294 unsigned int i, nsyms;
2295
2296 sym = file->fSymbolBase;
0b4e3aa0
A
2297 for (i = 0, nsyms = file->fSymtab->nsyms; i < nsyms; i++, sym++) {
2298 if ( (sym->n_type & N_EXT) ) {
2299 sym = 0;
2300 break; // Terminate search when we hit an extern
2301 }
2302 if ( (sym->n_type & N_STAB) )
2303 continue;
9bccf70c 2304 if ( !strcmp(symname, symNameByIndex(file, i)) )
0b4e3aa0
A
2305 break;
2306 }
2307 }
2308
2309 return_if(!sym,
2310 NULL, ("Unknown symbol %s in %s\n", symname, pathName));
2311
2312 // Is the vtable in a valid section?
2313 sectind = sym->n_sect;
2314 return_if(sectind == NO_SECT || sectind > file->fNSects, NULL,
2315 ("Malformed object file, invalid section reference for %s in %s\n",
2316 symname, pathName));
2317
2318 section = file->fSections[sectind - 1].fSection;
2319 sectionBase = file->fMachO + section->offset - section->addr;
2320
2321 return (void *) (sectionBase + sym->n_value);
2322}
2323
2324Boolean kld_file_merge_OSObjects(const char *pathName)
2325{
2326 struct fileRecord *file = getFile(pathName);
2327
2328 return_if(!file,
2329 false, ("Internal error - unable to find file %s\n", pathName));
2330
2331 return mergeOSObjectsForFile(file);
2332}
2333
2334Boolean kld_file_patch_OSObjects(const char *pathName)
2335{
2336 struct fileRecord *file = getFile(pathName);
2337 struct metaClassRecord **classes;
2338 unsigned long i, last;
2339
2340 return_if(!file,
2341 false, ("Internal error - unable to find file %s\n", pathName));
2342
9bccf70c 2343 DEBUG_LOG(("Patch file %s\n", pathName)); // @@@ gvdl:
0b4e3aa0
A
2344
2345 // If we don't have any classes we can return now.
2346 if (!file->fClassList)
2347 return true;
2348
2349 // If we haven't alread merged the kernel then do it now
2350 if (!sMergedKernel && sKernelFile)
2351 mergeOSObjectsForFile(sKernelFile);
2352 return_if(!sMergedKernel, false, ("Internal error no kernel?\n"));
2353
2354 if (!mergeOSObjectsForFile(file))
2355 return false;
2356
2357 // Patch all of the classes in this executable
2358 last = DataGetLength(file->fClassList) / sizeof(void *);
2359 classes = (struct metaClassRecord **) DataGetPtr(file->fClassList);
2360 for (i = 0; i < last; i++) {
9bccf70c
A
2361 if (!patchVTable(classes[i])) {
2362 // RY: Set a flag in the file list to invalidate this data.
2363 // I would remove the file from the list, but that seems to be
2364 // not worth the effort.
2365 file->fIgnoreFile = TRUE;
2366
2367 return false;
2368 }
0b4e3aa0
A
2369 }
2370
2371 return true;
2372}
2373
2374Boolean kld_file_prepare_for_link()
2375{
2376 if (sMergedFiles) {
2377 unsigned long i, nmerged = 0;
2378 struct fileRecord **files;
2379
2380 // Check to see if we have already merged this file
2381 nmerged = DataGetLength(sMergedFiles) / sizeof(struct fileRecord *);
2382 files = (struct fileRecord **) DataGetPtr(sMergedFiles);
9bccf70c
A
2383 for (i = 0; i < nmerged; i++) {
2384 if (!files[i]->fIgnoreFile && !prepareFileForLink(files[i]))
0b4e3aa0
A
2385 return false;
2386 }
2387 }
2388
2389 // Clear down the meta class table and merged file lists
2390 DataRelease(sMergeMetaClasses);
2391 DataRelease(sMergedFiles);
2392 sMergedFiles = sMergeMetaClasses = NULL;
2393 sMergedKernel = false;
2394
2395 return true;
2396}
2397
2398void kld_file_cleanup_all_resources()
2399{
2400 unsigned long i, nfiles;
2401
2402#if KERNEL // @@@ gvdl:
2403 // Debugger("kld_file_cleanup_all_resources");
2404#endif
2405
2406 if (!sFilesTable || !(nfiles = DataGetLength(sFilesTable)))
2407 return; // Nothing to do just return now
2408
2409 nfiles /= sizeof(struct fileRecord *);
2410 for (i = 0; i < nfiles; i++)
2411 removeFile(((void **) DataGetPtr(sFilesTable))[i]);
2412
9bccf70c
A
2413 DataRelease(sFilesTable);
2414 sFilesTable = NULL;
2415
0b4e3aa0
A
2416 // Don't really have to clean up anything more as the whole
2417 // malloc engine is going to be released and I couldn't be bothered.
2418}
2419
9bccf70c 2420
0b4e3aa0 2421#if !KERNEL
9bccf70c
A
2422#if 0
2423static const struct fileRecord *sortFile;
2424static int symCompare(const void *vSym1, const void *vSym2)
2425{
2426 const struct nlist *sym1 = vSym1;
2427 const struct nlist *sym2 = vSym2;
2428
2429 {
2430 unsigned int ind1, ind2;
2431
2432 ind1 = sym1->n_type & N_TYPE;
2433 ind2 = sym2->n_type & N_TYPE;
2434 if (ind1 != ind2) {
2435 // if sym1 is undefined then sym1 must come later than sym2
2436 if (ind1 == N_UNDF)
2437 return 1;
2438 // if sym2 is undefined then sym1 must come earlier than sym2
2439 if (ind2 == N_UNDF)
2440 return -1;
2441 /* drop out if neither are undefined */
2442 }
2443 }
2444
2445 {
2446 const struct fileRecord *file = sortFile;
2447 const char *name1, *name2;
2448
2449 name1 = file->fStringBase + sym1->n_un.n_strx;
2450 name2 = file->fStringBase + sym2->n_un.n_strx;
2451 return strcmp(name1, name2);
2452 }
2453}
2454#endif /* 0 */
2455
0b4e3aa0
A
2456Boolean kld_file_debug_dump(const char *pathName, const char *outName)
2457{
2458 const struct fileRecord *file = getFile(pathName);
2459 int fd;
2460 Boolean ret = false;
2461
2462 return_if(!file, false, ("Unknown file %s for dumping\n", pathName));
2463
2464 fd = open(outName, O_WRONLY|O_CREAT|O_TRUNC, 0666);
2465 return_if(-1 == fd, false, ("Can't create output file %s - %s(%d)\n",
2466 outName, strerror(errno), errno));
2467
2468 do {
9bccf70c
A
2469#if 0
2470 // Sorting doesn't work until I fix the relocs too?
2471
2472 // sort the symbol table appropriately
2473 unsigned int nsyms = file->fSymtab->nsyms
2474 - (file->fLocalSyms - file->fSymbolBase);
2475 sortFile = file;
2476 heapsort((void *) file->fLocalSyms, nsyms, sizeof(struct nlist),
2477 symCompare);
2478#endif
2479
0b4e3aa0
A
2480 break_if(-1 == write(fd, file->fMachO, file->fMachOSize),
2481 ("Can't dump output file %s - %s(%d)\n",
2482 outName, strerror(errno), errno));
2483 ret = true;
2484 } while(0);
2485
2486 close(fd);
2487
2488 return ret;
2489}
9bccf70c 2490
0b4e3aa0
A
2491#endif /* !KERNEL */
2492