]>
Commit | Line | Data |
---|---|---|
1 | /* | |
2 | * Copyright (c) 1999-2007 Apple Inc. All Rights Reserved. | |
3 | * | |
4 | * @APPLE_LICENSE_HEADER_START@ | |
5 | * | |
6 | * This file contains Original Code and/or Modifications of Original Code | |
7 | * as defined in and that are subject to the Apple Public Source License | |
8 | * Version 2.0 (the 'License'). You may not use this file except in | |
9 | * compliance with the License. Please obtain a copy of the License at | |
10 | * http://www.opensource.apple.com/apsl/ and read it before using this | |
11 | * file. | |
12 | * | |
13 | * The Original Code and all software distributed under the License are | |
14 | * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER | |
15 | * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES, | |
16 | * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, | |
17 | * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. | |
18 | * Please see the License for the specific language governing rights and | |
19 | * limitations under the License. | |
20 | * | |
21 | * @APPLE_LICENSE_HEADER_END@ | |
22 | */ | |
23 | ||
24 | /*********************************************************************** | |
25 | * objc-runtime-old.m | |
26 | * Support for old-ABI classes and images. | |
27 | **********************************************************************/ | |
28 | ||
29 | /*********************************************************************** | |
30 | * Class loading and connecting (GrP 2004-2-11) | |
31 | * | |
32 | * When images are loaded (during program startup or otherwise), the | |
33 | * runtime needs to load classes and categories from the images, connect | |
34 | * classes to superclasses and categories to parent classes, and call | |
35 | * +load methods. | |
36 | * | |
37 | * The Objective-C runtime can cope with classes arriving in any order. | |
38 | * That is, a class may be discovered by the runtime before some | |
39 | * superclass is known. To handle out-of-order class loads, the | |
40 | * runtime uses a "pending class" system. | |
41 | * | |
42 | * (Historical note) | |
43 | * Panther and earlier: many classes arrived out-of-order because of | |
44 | * the poorly-ordered callback from dyld. However, the runtime's | |
45 | * pending mechanism only handled "missing superclass" and not | |
46 | * "present superclass but missing higher class". See Radar #3225652. | |
47 | * Tiger: The runtime's pending mechanism was augmented to handle | |
48 | * arbitrary missing classes. In addition, dyld was rewritten and | |
49 | * now sends the callbacks in strictly bottom-up link order. | |
50 | * The pending mechanism may now be needed only for rare and | |
51 | * hard to construct programs. | |
52 | * (End historical note) | |
53 | * | |
54 | * A class when first seen in an image is considered "unconnected". | |
55 | * It is stored in `unconnected_class_hash`. If all of the class's | |
56 | * superclasses exist and are already "connected", then the new class | |
57 | * can be connected to its superclasses and moved to `class_hash` for | |
58 | * normal use. Otherwise, the class waits in `unconnected_class_hash` | |
59 | * until the superclasses finish connecting. | |
60 | * | |
61 | * A "connected" class is | |
62 | * (1) in `class_hash`, | |
63 | * (2) connected to its superclasses, | |
64 | * (3) has no unconnected superclasses, | |
65 | * (4) is otherwise initialized and ready for use, and | |
66 | * (5) is eligible for +load if +load has not already been called. | |
67 | * | |
68 | * An "unconnected" class is | |
69 | * (1) in `unconnected_class_hash`, | |
70 | * (2) not connected to its superclasses, | |
71 | * (3) has an immediate superclass which is either missing or unconnected, | |
72 | * (4) is not ready for use, and | |
73 | * (5) is not yet eligible for +load. | |
74 | * | |
75 | * Image mapping is NOT CURRENTLY THREAD-SAFE with respect to just about | |
76 | * anything. Image mapping IS RE-ENTRANT in several places: superclass | |
77 | * lookup may cause ZeroLink to load another image, and +load calls may | |
78 | * cause dyld to load another image. | |
79 | * | |
80 | * Image mapping sequence: | |
81 | * | |
82 | * Read all classes in all new images. | |
83 | * Add them all to unconnected_class_hash. | |
84 | * Note any +load implementations before categories are attached. | |
85 | * Attach any pending categories. | |
86 | * Read all categories in all new images. | |
87 | * Attach categories whose parent class exists (connected or not), | |
88 | * and pend the rest. | |
89 | * Mark them all eligible for +load (if implemented), even if the | |
90 | * parent class is missing. | |
91 | * Try to connect all classes in all new images. | |
92 | * If the superclass is missing, pend the class | |
93 | * If the superclass is unconnected, try to recursively connect it | |
94 | * If the superclass is connected: | |
95 | * connect the class | |
96 | * mark the class eligible for +load, if implemented | |
97 | * fix up any pended classrefs referring to the class | |
98 | * connect any pended subclasses of the class | |
99 | * Resolve selector refs and class refs in all new images. | |
100 | * Class refs whose classes still do not exist are pended. | |
101 | * Fix up protocol objects in all new images. | |
102 | * Call +load for classes and categories. | |
103 | * May include classes or categories that are not in these images, | |
104 | * but are newly eligible because of these image. | |
105 | * Class +loads will be called superclass-first because of the | |
106 | * superclass-first nature of the connecting process. | |
107 | * Category +load needs to be deferred until the parent class is | |
108 | * connected and has had its +load called. | |
109 | * | |
110 | * Performance: all classes are read before any categories are read. | |
111 | * Fewer categories need be pended for lack of a parent class. | |
112 | * | |
113 | * Performance: all categories are attempted to be attached before | |
114 | * any classes are connected. Fewer class caches need be flushed. | |
115 | * (Unconnected classes and their respective subclasses are guaranteed | |
116 | * to be un-messageable, so their caches will be empty.) | |
117 | * | |
118 | * Performance: all classes are read before any classes are connected. | |
119 | * Fewer classes need be pended for lack of a superclass. | |
120 | * | |
121 | * Correctness: all selector and class refs are fixed before any | |
122 | * protocol fixups or +load methods. libobjc itself contains selector | |
123 | * and class refs which are used in protocol fixup and +load. | |
124 | * | |
125 | * Correctness: +load methods are scheduled in bottom-up link order. | |
126 | * This constraint is in addition to superclass order. Some +load | |
127 | * implementations expect to use another class in a linked-to library, | |
128 | * even if the two classes don't share a direct superclass relationship. | |
129 | * | |
130 | * Correctness: all classes are scanned for +load before any categories | |
131 | * are attached. Otherwise, if a category implements +load and its class | |
132 | * has no class methods, the class's +load scan would find the category's | |
133 | * +load method, which would then be called twice. | |
134 | * | |
135 | * Correctness: pended class refs are not fixed up until the class is | |
136 | * connected. Classes with missing weak superclasses remain unconnected. | |
137 | * Class refs to classes with missing weak superclasses must be nil. | |
138 | * Therefore class refs to unconnected classes must remain un-fixed. | |
139 | * | |
140 | **********************************************************************/ | |
141 | ||
142 | #if !__OBJC2__ | |
143 | ||
144 | #include "objc-private.h" | |
145 | #include "objc-runtime-old.h" | |
146 | #include "objc-file-old.h" | |
147 | #include "objc-cache-old.h" | |
148 | #include "objc-loadmethod.h" | |
149 | ||
150 | ||
151 | typedef struct _objc_unresolved_category | |
152 | { | |
153 | struct _objc_unresolved_category *next; | |
154 | old_category *cat; // may be nil | |
155 | long version; | |
156 | } _objc_unresolved_category; | |
157 | ||
158 | typedef struct _PendingSubclass | |
159 | { | |
160 | Class subclass; // subclass to finish connecting; may be nil | |
161 | struct _PendingSubclass *next; | |
162 | } PendingSubclass; | |
163 | ||
164 | typedef struct _PendingClassRef | |
165 | { | |
166 | Class *ref; // class reference to fix up; may be nil | |
167 | // (ref & 1) is a metaclass reference | |
168 | struct _PendingClassRef *next; | |
169 | } PendingClassRef; | |
170 | ||
171 | ||
172 | static uintptr_t classHash(void *info, Class data); | |
173 | static int classIsEqual(void *info, Class name, Class cls); | |
174 | static int _objc_defaultClassHandler(const char *clsName); | |
175 | static inline NXMapTable *pendingClassRefsMapTable(void); | |
176 | static inline NXMapTable *pendingSubclassesMapTable(void); | |
177 | static void pendClassInstallation(Class cls, const char *superName); | |
178 | static void pendClassReference(Class *ref, const char *className, bool isMeta); | |
179 | static void resolve_references_to_class(Class cls); | |
180 | static void resolve_subclasses_of_class(Class cls); | |
181 | static void really_connect_class(Class cls, Class supercls); | |
182 | static bool connect_class(Class cls); | |
183 | static void map_method_descs (struct objc_method_description_list * methods, bool copy); | |
184 | static void _objcTweakMethodListPointerForClass(Class cls); | |
185 | static inline void _objc_add_category(Class cls, old_category *category, int version); | |
186 | static bool _objc_add_category_flush_caches(Class cls, old_category *category, int version); | |
187 | static _objc_unresolved_category *reverse_cat(_objc_unresolved_category *cat); | |
188 | static void resolve_categories_for_class(Class cls); | |
189 | static bool _objc_register_category(old_category *cat, int version); | |
190 | ||
191 | ||
192 | // Function called when a class is loaded from an image | |
193 | void (*callbackFunction)(Class, Category) = 0; | |
194 | ||
195 | // Hash table of classes | |
196 | NXHashTable * class_hash = 0; | |
197 | static NXHashTablePrototype classHashPrototype = | |
198 | { | |
199 | (uintptr_t (*) (const void *, const void *)) classHash, | |
200 | (int (*)(const void *, const void *, const void *)) classIsEqual, | |
201 | NXNoEffectFree, 0 | |
202 | }; | |
203 | ||
204 | // Hash table of unconnected classes | |
205 | static NXHashTable *unconnected_class_hash = nil; | |
206 | ||
207 | // Exported copy of class_hash variable (hook for debugging tools) | |
208 | NXHashTable *_objc_debug_class_hash = nil; | |
209 | ||
210 | // Category and class registries | |
211 | // Keys are COPIES of strings, to prevent stale pointers with unloaded bundles | |
212 | // Use NXMapKeyCopyingInsert and NXMapKeyFreeingRemove | |
213 | static NXMapTable * category_hash = nil; | |
214 | ||
215 | // Keys are COPIES of strings, to prevent stale pointers with unloaded bundles | |
216 | // Use NXMapKeyCopyingInsert and NXMapKeyFreeingRemove | |
217 | static NXMapTable * pendingClassRefsMap = nil; | |
218 | static NXMapTable * pendingSubclassesMap = nil; | |
219 | ||
220 | // Protocols | |
221 | static NXMapTable *protocol_map = nil; // name -> protocol | |
222 | static NXMapTable *protocol_ext_map = nil; // protocol -> protocol ext | |
223 | ||
224 | // Function pointer objc_getClass calls through when class is not found | |
225 | static int (*objc_classHandler) (const char *) = _objc_defaultClassHandler; | |
226 | ||
227 | // Function pointer called by objc_getClass and objc_lookupClass when | |
228 | // class is not found. _objc_classLoader is called before objc_classHandler. | |
229 | static BOOL (*_objc_classLoader)(const char *) = nil; | |
230 | ||
231 | ||
232 | /*********************************************************************** | |
233 | * objc_dump_class_hash. Log names of all known classes. | |
234 | **********************************************************************/ | |
235 | void objc_dump_class_hash(void) | |
236 | { | |
237 | NXHashTable *table; | |
238 | unsigned count; | |
239 | Class data; | |
240 | NXHashState state; | |
241 | ||
242 | table = class_hash; | |
243 | count = 0; | |
244 | state = NXInitHashState (table); | |
245 | while (NXNextHashState (table, &state, (void **) &data)) | |
246 | printf ("class %d: %s\n", ++count, data->nameForLogging()); | |
247 | } | |
248 | ||
249 | ||
250 | /*********************************************************************** | |
251 | * _objc_init_class_hash. Return the class lookup table, create it if | |
252 | * necessary. | |
253 | **********************************************************************/ | |
254 | void _objc_init_class_hash(void) | |
255 | { | |
256 | // Do nothing if class hash table already exists | |
257 | if (class_hash) | |
258 | return; | |
259 | ||
260 | // class_hash starts small, with only enough capacity for libobjc itself. | |
261 | // If a second library is found by map_images(), class_hash is immediately | |
262 | // resized to capacity 1024 to cut down on rehashes. | |
263 | // Old numbers: A smallish Foundation+AppKit program will have | |
264 | // about 520 classes. Larger apps (like IB or WOB) have more like | |
265 | // 800 classes. Some customers have massive quantities of classes. | |
266 | // Foundation-only programs aren't likely to notice the ~6K loss. | |
267 | class_hash = NXCreateHashTable(classHashPrototype, 16, nil); | |
268 | _objc_debug_class_hash = class_hash; | |
269 | } | |
270 | ||
271 | ||
272 | /*********************************************************************** | |
273 | * objc_getClassList. Return the known classes. | |
274 | **********************************************************************/ | |
275 | int objc_getClassList(Class *buffer, int bufferLen) | |
276 | { | |
277 | NXHashState state; | |
278 | Class cls; | |
279 | int cnt, num; | |
280 | ||
281 | mutex_locker_t lock(classLock); | |
282 | if (!class_hash) return 0; | |
283 | ||
284 | num = NXCountHashTable(class_hash); | |
285 | if (nil == buffer) return num; | |
286 | ||
287 | cnt = 0; | |
288 | state = NXInitHashState(class_hash); | |
289 | while (cnt < bufferLen && | |
290 | NXNextHashState(class_hash, &state, (void **)&cls)) | |
291 | { | |
292 | buffer[cnt++] = cls; | |
293 | } | |
294 | ||
295 | return num; | |
296 | } | |
297 | ||
298 | ||
299 | /*********************************************************************** | |
300 | * objc_copyClassList | |
301 | * Returns pointers to all classes. | |
302 | * This requires all classes be realized, which is regretfully non-lazy. | |
303 | * | |
304 | * outCount may be nil. *outCount is the number of classes returned. | |
305 | * If the returned array is not nil, it is nil-terminated and must be | |
306 | * freed with free(). | |
307 | * Locking: acquires classLock | |
308 | **********************************************************************/ | |
309 | Class * | |
310 | objc_copyClassList(unsigned int *outCount) | |
311 | { | |
312 | Class *result; | |
313 | unsigned int count; | |
314 | ||
315 | mutex_locker_t lock(classLock); | |
316 | result = nil; | |
317 | count = class_hash ? NXCountHashTable(class_hash) : 0; | |
318 | ||
319 | if (count > 0) { | |
320 | Class cls; | |
321 | NXHashState state = NXInitHashState(class_hash); | |
322 | result = (Class *)malloc((1+count) * sizeof(Class)); | |
323 | count = 0; | |
324 | while (NXNextHashState(class_hash, &state, (void **)&cls)) { | |
325 | result[count++] = cls; | |
326 | } | |
327 | result[count] = nil; | |
328 | } | |
329 | ||
330 | if (outCount) *outCount = count; | |
331 | return result; | |
332 | } | |
333 | ||
334 | ||
335 | /*********************************************************************** | |
336 | * objc_copyProtocolList | |
337 | * Returns pointers to all protocols. | |
338 | * Locking: acquires classLock | |
339 | **********************************************************************/ | |
340 | Protocol * __unsafe_unretained * | |
341 | objc_copyProtocolList(unsigned int *outCount) | |
342 | { | |
343 | int count, i; | |
344 | Protocol *proto; | |
345 | const char *name; | |
346 | NXMapState state; | |
347 | Protocol **result; | |
348 | ||
349 | mutex_locker_t lock(classLock); | |
350 | ||
351 | count = NXCountMapTable(protocol_map); | |
352 | if (count == 0) { | |
353 | if (outCount) *outCount = 0; | |
354 | return nil; | |
355 | } | |
356 | ||
357 | result = (Protocol **)calloc(1 + count, sizeof(Protocol *)); | |
358 | ||
359 | i = 0; | |
360 | state = NXInitMapState(protocol_map); | |
361 | while (NXNextMapState(protocol_map, &state, | |
362 | (const void **)&name, (const void **)&proto)) | |
363 | { | |
364 | result[i++] = proto; | |
365 | } | |
366 | ||
367 | result[i++] = nil; | |
368 | assert(i == count+1); | |
369 | ||
370 | if (outCount) *outCount = count; | |
371 | return result; | |
372 | } | |
373 | ||
374 | ||
375 | /*********************************************************************** | |
376 | * objc_getClasses. Return class lookup table. | |
377 | * | |
378 | * NOTE: This function is very dangerous, since you cannot safely use | |
379 | * the hashtable without locking it, and the lock is private! | |
380 | **********************************************************************/ | |
381 | void *objc_getClasses(void) | |
382 | { | |
383 | OBJC_WARN_DEPRECATED; | |
384 | ||
385 | // Return the class lookup hash table | |
386 | return class_hash; | |
387 | } | |
388 | ||
389 | ||
390 | /*********************************************************************** | |
391 | * classHash. | |
392 | **********************************************************************/ | |
393 | static uintptr_t classHash(void *info, Class data) | |
394 | { | |
395 | // Nil classes hash to zero | |
396 | if (!data) | |
397 | return 0; | |
398 | ||
399 | // Call through to real hash function | |
400 | return _objc_strhash (data->mangledName()); | |
401 | } | |
402 | ||
403 | /*********************************************************************** | |
404 | * classIsEqual. Returns whether the class names match. If we ever | |
405 | * check more than the name, routines like objc_lookUpClass have to | |
406 | * change as well. | |
407 | **********************************************************************/ | |
408 | static int classIsEqual(void *info, Class name, Class cls) | |
409 | { | |
410 | // Standard string comparison | |
411 | return strcmp(name->mangledName(), cls->mangledName()) == 0; | |
412 | } | |
413 | ||
414 | ||
415 | // Unresolved future classes | |
416 | static NXHashTable *future_class_hash = nil; | |
417 | ||
418 | // Resolved future<->original classes | |
419 | static NXMapTable *future_class_to_original_class_map = nil; | |
420 | static NXMapTable *original_class_to_future_class_map = nil; | |
421 | ||
422 | // CF requests about 20 future classes; HIToolbox requests one. | |
423 | #define FUTURE_COUNT 32 | |
424 | ||
425 | ||
426 | /*********************************************************************** | |
427 | * setOriginalClassForFutureClass | |
428 | * Record resolution of a future class. | |
429 | **********************************************************************/ | |
430 | static void setOriginalClassForFutureClass(Class futureClass, | |
431 | Class originalClass) | |
432 | { | |
433 | if (!future_class_to_original_class_map) { | |
434 | future_class_to_original_class_map = | |
435 | NXCreateMapTable(NXPtrValueMapPrototype, FUTURE_COUNT); | |
436 | original_class_to_future_class_map = | |
437 | NXCreateMapTable(NXPtrValueMapPrototype, FUTURE_COUNT); | |
438 | } | |
439 | ||
440 | NXMapInsert (future_class_to_original_class_map, | |
441 | futureClass, originalClass); | |
442 | NXMapInsert (original_class_to_future_class_map, | |
443 | originalClass, futureClass); | |
444 | ||
445 | if (PrintFuture) { | |
446 | _objc_inform("FUTURE: using %p instead of %p for %s", (void*)futureClass, (void*)originalClass, originalClass->name); | |
447 | } | |
448 | } | |
449 | ||
450 | /*********************************************************************** | |
451 | * getOriginalClassForFutureClass | |
452 | * getFutureClassForOriginalClass | |
453 | * Switch between a future class and its corresponding original class. | |
454 | * The future class is the one actually in use. | |
455 | * The original class is the one from disk. | |
456 | **********************************************************************/ | |
457 | /* | |
458 | static Class | |
459 | getOriginalClassForFutureClass(Class futureClass) | |
460 | { | |
461 | if (!future_class_to_original_class_map) return Nil; | |
462 | return NXMapGet (future_class_to_original_class_map, futureClass); | |
463 | } | |
464 | */ | |
465 | static Class | |
466 | getFutureClassForOriginalClass(Class originalClass) | |
467 | { | |
468 | if (!original_class_to_future_class_map) return Nil; | |
469 | return (Class)NXMapGet(original_class_to_future_class_map, originalClass); | |
470 | } | |
471 | ||
472 | ||
473 | /*********************************************************************** | |
474 | * makeFutureClass | |
475 | * Initialize the memory in *cls with an unresolved future class with the | |
476 | * given name. The memory is recorded in future_class_hash. | |
477 | **********************************************************************/ | |
478 | static void makeFutureClass(Class cls, const char *name) | |
479 | { | |
480 | // CF requests about 20 future classes, plus HIToolbox has one. | |
481 | if (!future_class_hash) { | |
482 | future_class_hash = | |
483 | NXCreateHashTable(classHashPrototype, FUTURE_COUNT, nil); | |
484 | } | |
485 | ||
486 | cls->name = strdup(name); | |
487 | NXHashInsert(future_class_hash, cls); | |
488 | ||
489 | if (PrintFuture) { | |
490 | _objc_inform("FUTURE: reserving %p for %s", (void*)cls, name); | |
491 | } | |
492 | } | |
493 | ||
494 | ||
495 | /*********************************************************************** | |
496 | * _objc_allocateFutureClass | |
497 | * Allocate an unresolved future class for the given class name. | |
498 | * Returns any existing allocation if one was already made. | |
499 | * Assumes the named class doesn't exist yet. | |
500 | * Not thread safe. | |
501 | **********************************************************************/ | |
502 | Class _objc_allocateFutureClass(const char *name) | |
503 | { | |
504 | Class cls; | |
505 | ||
506 | if (future_class_hash) { | |
507 | objc_class query; | |
508 | query.name = name; | |
509 | if ((cls = (Class)NXHashGet(future_class_hash, &query))) { | |
510 | // Already have a future class for this name. | |
511 | return cls; | |
512 | } | |
513 | } | |
514 | ||
515 | cls = _calloc_class(sizeof(objc_class)); | |
516 | makeFutureClass(cls, name); | |
517 | return cls; | |
518 | } | |
519 | ||
520 | ||
521 | /*********************************************************************** | |
522 | * objc_getFutureClass. Return the id of the named class. | |
523 | * If the class does not exist, return an uninitialized class | |
524 | * structure that will be used for the class when and if it | |
525 | * does get loaded. | |
526 | * Not thread safe. | |
527 | **********************************************************************/ | |
528 | Class objc_getFutureClass(const char *name) | |
529 | { | |
530 | Class cls; | |
531 | ||
532 | // YES unconnected, NO class handler | |
533 | // (unconnected is OK because it will someday be the real class) | |
534 | cls = look_up_class(name, YES, NO); | |
535 | if (cls) { | |
536 | if (PrintFuture) { | |
537 | _objc_inform("FUTURE: found %p already in use for %s", | |
538 | (void*)cls, name); | |
539 | } | |
540 | return cls; | |
541 | } | |
542 | ||
543 | // No class or future class with that name yet. Make one. | |
544 | // fixme not thread-safe with respect to | |
545 | // simultaneous library load or getFutureClass. | |
546 | return _objc_allocateFutureClass(name); | |
547 | } | |
548 | ||
549 | ||
550 | BOOL _class_isFutureClass(Class cls) | |
551 | { | |
552 | return cls && cls->isFuture(); | |
553 | } | |
554 | ||
555 | bool objc_class::isFuture() | |
556 | { | |
557 | return future_class_hash && NXHashGet(future_class_hash, this); | |
558 | } | |
559 | ||
560 | ||
561 | /*********************************************************************** | |
562 | * _objc_defaultClassHandler. Default objc_classHandler. Does nothing. | |
563 | **********************************************************************/ | |
564 | static int _objc_defaultClassHandler(const char *clsName) | |
565 | { | |
566 | // Return zero so objc_getClass doesn't bother re-searching | |
567 | return 0; | |
568 | } | |
569 | ||
570 | /*********************************************************************** | |
571 | * objc_setClassHandler. Set objc_classHandler to the specified value. | |
572 | * | |
573 | * NOTE: This should probably deal with userSuppliedHandler being nil, | |
574 | * because the objc_classHandler caller does not check... it would bus | |
575 | * error. It would make sense to handle nil by restoring the default | |
576 | * handler. Is anyone hacking with this, though? | |
577 | **********************************************************************/ | |
578 | void objc_setClassHandler(int (*userSuppliedHandler)(const char *)) | |
579 | { | |
580 | OBJC_WARN_DEPRECATED; | |
581 | ||
582 | objc_classHandler = userSuppliedHandler; | |
583 | } | |
584 | ||
585 | ||
586 | /*********************************************************************** | |
587 | * _objc_setClassLoader | |
588 | * Similar to objc_setClassHandler, but objc_classLoader is used for | |
589 | * both objc_getClass() and objc_lookupClass(), and objc_classLoader | |
590 | * pre-empts objc_classHandler. | |
591 | **********************************************************************/ | |
592 | void _objc_setClassLoader(BOOL (*newClassLoader)(const char *)) | |
593 | { | |
594 | _objc_classLoader = newClassLoader; | |
595 | } | |
596 | ||
597 | ||
598 | /*********************************************************************** | |
599 | * objc_getProtocol | |
600 | * Get a protocol by name, or nil. | |
601 | **********************************************************************/ | |
602 | Protocol *objc_getProtocol(const char *name) | |
603 | { | |
604 | mutex_locker_t lock(classLock); | |
605 | if (!protocol_map) return nil; | |
606 | return (Protocol *)NXMapGet(protocol_map, name); | |
607 | } | |
608 | ||
609 | ||
610 | /*********************************************************************** | |
611 | * look_up_class | |
612 | * Map a class name to a class using various methods. | |
613 | * This is the common implementation of objc_lookUpClass and objc_getClass, | |
614 | * and is also used internally to get additional search options. | |
615 | * Sequence: | |
616 | * 1. class_hash | |
617 | * 2. unconnected_class_hash (optional) | |
618 | * 3. classLoader callback | |
619 | * 4. classHandler callback (optional) | |
620 | **********************************************************************/ | |
621 | Class look_up_class(const char *aClassName, bool includeUnconnected, | |
622 | bool includeClassHandler) | |
623 | { | |
624 | bool includeClassLoader = YES; // class loader cannot be skipped | |
625 | Class result = nil; | |
626 | struct objc_class query; | |
627 | ||
628 | query.name = aClassName; | |
629 | ||
630 | retry: | |
631 | ||
632 | if (!result && class_hash) { | |
633 | // Check ordinary classes | |
634 | mutex_locker_t lock(classLock); | |
635 | result = (Class)NXHashGet(class_hash, &query); | |
636 | } | |
637 | ||
638 | if (!result && includeUnconnected && unconnected_class_hash) { | |
639 | // Check not-yet-connected classes | |
640 | mutex_locker_t lock(classLock); | |
641 | result = (Class)NXHashGet(unconnected_class_hash, &query); | |
642 | } | |
643 | ||
644 | if (!result && includeClassLoader && _objc_classLoader) { | |
645 | // Try class loader callback | |
646 | if ((*_objc_classLoader)(aClassName)) { | |
647 | // Re-try lookup without class loader | |
648 | includeClassLoader = NO; | |
649 | goto retry; | |
650 | } | |
651 | } | |
652 | ||
653 | if (!result && includeClassHandler && objc_classHandler) { | |
654 | // Try class handler callback | |
655 | if ((*objc_classHandler)(aClassName)) { | |
656 | // Re-try lookup without class handler or class loader | |
657 | includeClassLoader = NO; | |
658 | includeClassHandler = NO; | |
659 | goto retry; | |
660 | } | |
661 | } | |
662 | ||
663 | return result; | |
664 | } | |
665 | ||
666 | ||
667 | /*********************************************************************** | |
668 | * objc_class::isConnected | |
669 | * Returns TRUE if class cls is connected. | |
670 | * A connected class has either a connected superclass or a nil superclass, | |
671 | * and is present in class_hash. | |
672 | **********************************************************************/ | |
673 | bool objc_class::isConnected() | |
674 | { | |
675 | mutex_locker_t lock(classLock); | |
676 | return NXHashMember(class_hash, this); | |
677 | } | |
678 | ||
679 | ||
680 | /*********************************************************************** | |
681 | * pendingClassRefsMapTable. Return a pointer to the lookup table for | |
682 | * pending class refs. | |
683 | **********************************************************************/ | |
684 | static inline NXMapTable *pendingClassRefsMapTable(void) | |
685 | { | |
686 | // Allocate table if needed | |
687 | if (!pendingClassRefsMap) { | |
688 | pendingClassRefsMap = NXCreateMapTable(NXStrValueMapPrototype, 10); | |
689 | } | |
690 | ||
691 | // Return table pointer | |
692 | return pendingClassRefsMap; | |
693 | } | |
694 | ||
695 | ||
696 | /*********************************************************************** | |
697 | * pendingSubclassesMapTable. Return a pointer to the lookup table for | |
698 | * pending subclasses. | |
699 | **********************************************************************/ | |
700 | static inline NXMapTable *pendingSubclassesMapTable(void) | |
701 | { | |
702 | // Allocate table if needed | |
703 | if (!pendingSubclassesMap) { | |
704 | pendingSubclassesMap = NXCreateMapTable(NXStrValueMapPrototype, 10); | |
705 | } | |
706 | ||
707 | // Return table pointer | |
708 | return pendingSubclassesMap; | |
709 | } | |
710 | ||
711 | ||
712 | /*********************************************************************** | |
713 | * pendClassInstallation | |
714 | * Finish connecting class cls when its superclass becomes connected. | |
715 | * Check for multiple pends of the same class because connect_class does not. | |
716 | **********************************************************************/ | |
717 | static void pendClassInstallation(Class cls, const char *superName) | |
718 | { | |
719 | NXMapTable *table; | |
720 | PendingSubclass *pending; | |
721 | PendingSubclass *oldList; | |
722 | PendingSubclass *l; | |
723 | ||
724 | // Create and/or locate pending class lookup table | |
725 | table = pendingSubclassesMapTable (); | |
726 | ||
727 | // Make sure this class isn't already in the pending list. | |
728 | oldList = (PendingSubclass *)NXMapGet(table, superName); | |
729 | for (l = oldList; l != nil; l = l->next) { | |
730 | if (l->subclass == cls) return; // already here, nothing to do | |
731 | } | |
732 | ||
733 | // Create entry referring to this class | |
734 | pending = (PendingSubclass *)malloc(sizeof(PendingSubclass)); | |
735 | pending->subclass = cls; | |
736 | ||
737 | // Link new entry into head of list of entries for this class | |
738 | pending->next = oldList; | |
739 | ||
740 | // (Re)place entry list in the table | |
741 | NXMapKeyCopyingInsert (table, superName, pending); | |
742 | } | |
743 | ||
744 | ||
745 | /*********************************************************************** | |
746 | * pendClassReference | |
747 | * Fix up a class ref when the class with the given name becomes connected. | |
748 | **********************************************************************/ | |
749 | static void pendClassReference(Class *ref, const char *className, bool isMeta) | |
750 | { | |
751 | NXMapTable *table; | |
752 | PendingClassRef *pending; | |
753 | ||
754 | // Create and/or locate pending class lookup table | |
755 | table = pendingClassRefsMapTable (); | |
756 | ||
757 | // Create entry containing the class reference | |
758 | pending = (PendingClassRef *)malloc(sizeof(PendingClassRef)); | |
759 | pending->ref = ref; | |
760 | if (isMeta) { | |
761 | pending->ref = (Class *)((uintptr_t)pending->ref | 1); | |
762 | } | |
763 | ||
764 | // Link new entry into head of list of entries for this class | |
765 | pending->next = (PendingClassRef *)NXMapGet(table, className); | |
766 | ||
767 | // (Re)place entry list in the table | |
768 | NXMapKeyCopyingInsert (table, className, pending); | |
769 | ||
770 | if (PrintConnecting) { | |
771 | _objc_inform("CONNECT: pended reference to class '%s%s' at %p", | |
772 | className, isMeta ? " (meta)" : "", (void *)ref); | |
773 | } | |
774 | } | |
775 | ||
776 | ||
777 | /*********************************************************************** | |
778 | * resolve_references_to_class | |
779 | * Fix up any pending class refs to this class. | |
780 | **********************************************************************/ | |
781 | static void resolve_references_to_class(Class cls) | |
782 | { | |
783 | PendingClassRef *pending; | |
784 | ||
785 | if (!pendingClassRefsMap) return; // no unresolved refs for any class | |
786 | ||
787 | pending = (PendingClassRef *)NXMapGet(pendingClassRefsMap, cls->name); | |
788 | if (!pending) return; // no unresolved refs for this class | |
789 | ||
790 | NXMapKeyFreeingRemove(pendingClassRefsMap, cls->name); | |
791 | ||
792 | if (PrintConnecting) { | |
793 | _objc_inform("CONNECT: resolving references to class '%s'", cls->name); | |
794 | } | |
795 | ||
796 | while (pending) { | |
797 | PendingClassRef *next = pending->next; | |
798 | if (pending->ref) { | |
799 | bool isMeta = (uintptr_t)pending->ref & 1; | |
800 | Class *ref = | |
801 | (Class *)((uintptr_t)pending->ref & ~(uintptr_t)1); | |
802 | *ref = isMeta ? cls->ISA() : cls; | |
803 | } | |
804 | free(pending); | |
805 | pending = next; | |
806 | } | |
807 | ||
808 | if (NXCountMapTable(pendingClassRefsMap) == 0) { | |
809 | NXFreeMapTable(pendingClassRefsMap); | |
810 | pendingClassRefsMap = nil; | |
811 | } | |
812 | } | |
813 | ||
814 | ||
815 | /*********************************************************************** | |
816 | * resolve_subclasses_of_class | |
817 | * Fix up any pending subclasses of this class. | |
818 | **********************************************************************/ | |
819 | static void resolve_subclasses_of_class(Class cls) | |
820 | { | |
821 | PendingSubclass *pending; | |
822 | ||
823 | if (!pendingSubclassesMap) return; // no unresolved subclasses | |
824 | ||
825 | pending = (PendingSubclass *)NXMapGet(pendingSubclassesMap, cls->name); | |
826 | if (!pending) return; // no unresolved subclasses for this class | |
827 | ||
828 | NXMapKeyFreeingRemove(pendingSubclassesMap, cls->name); | |
829 | ||
830 | // Destroy the pending table if it's now empty, to save memory. | |
831 | if (NXCountMapTable(pendingSubclassesMap) == 0) { | |
832 | NXFreeMapTable(pendingSubclassesMap); | |
833 | pendingSubclassesMap = nil; | |
834 | } | |
835 | ||
836 | if (PrintConnecting) { | |
837 | _objc_inform("CONNECT: resolving subclasses of class '%s'", cls->name); | |
838 | } | |
839 | ||
840 | while (pending) { | |
841 | PendingSubclass *next = pending->next; | |
842 | if (pending->subclass) connect_class(pending->subclass); | |
843 | free(pending); | |
844 | pending = next; | |
845 | } | |
846 | } | |
847 | ||
848 | ||
849 | /*********************************************************************** | |
850 | * really_connect_class | |
851 | * Connect cls to superclass supercls unconditionally. | |
852 | * Also adjust the class hash tables and handle pended subclasses. | |
853 | * | |
854 | * This should be called from connect_class() ONLY. | |
855 | **********************************************************************/ | |
856 | static void really_connect_class(Class cls, | |
857 | Class supercls) | |
858 | { | |
859 | Class oldCls; | |
860 | ||
861 | // Connect superclass pointers. | |
862 | set_superclass(cls, supercls, YES); | |
863 | ||
864 | // Done! | |
865 | cls->info |= CLS_CONNECTED; | |
866 | ||
867 | { | |
868 | mutex_locker_t lock(classLock); | |
869 | ||
870 | // Update hash tables. | |
871 | NXHashRemove(unconnected_class_hash, cls); | |
872 | oldCls = (Class)NXHashInsert(class_hash, cls); | |
873 | ||
874 | // Delete unconnected_class_hash if it is now empty. | |
875 | if (NXCountHashTable(unconnected_class_hash) == 0) { | |
876 | NXFreeHashTable(unconnected_class_hash); | |
877 | unconnected_class_hash = nil; | |
878 | } | |
879 | ||
880 | // No duplicate classes allowed. | |
881 | // Duplicates should have been rejected by _objc_read_classes_from_image | |
882 | assert(!oldCls); | |
883 | } | |
884 | ||
885 | // Fix up pended class refs to this class, if any | |
886 | resolve_references_to_class(cls); | |
887 | ||
888 | // Connect newly-connectable subclasses | |
889 | resolve_subclasses_of_class(cls); | |
890 | ||
891 | // Debugging: if this class has ivars, make sure this class's ivars don't | |
892 | // overlap with its super's. This catches some broken fragile base classes. | |
893 | // Do not use super->instance_size vs. self->ivar[0] to check this. | |
894 | // Ivars may be packed across instance_size boundaries. | |
895 | if (DebugFragileSuperclasses && cls->ivars && cls->ivars->ivar_count) { | |
896 | Class ivar_cls = supercls; | |
897 | ||
898 | // Find closest superclass that has some ivars, if one exists. | |
899 | while (ivar_cls && | |
900 | (!ivar_cls->ivars || ivar_cls->ivars->ivar_count == 0)) | |
901 | { | |
902 | ivar_cls = ivar_cls->superclass; | |
903 | } | |
904 | ||
905 | if (ivar_cls) { | |
906 | // Compare superclass's last ivar to this class's first ivar | |
907 | old_ivar *super_ivar = | |
908 | &ivar_cls->ivars->ivar_list[ivar_cls->ivars->ivar_count - 1]; | |
909 | old_ivar *self_ivar = | |
910 | &cls->ivars->ivar_list[0]; | |
911 | ||
912 | // fixme could be smarter about super's ivar size | |
913 | if (self_ivar->ivar_offset <= super_ivar->ivar_offset) { | |
914 | _objc_inform("WARNING: ivars of superclass '%s' and " | |
915 | "subclass '%s' overlap; superclass may have " | |
916 | "changed since subclass was compiled", | |
917 | ivar_cls->name, cls->name); | |
918 | } | |
919 | } | |
920 | } | |
921 | } | |
922 | ||
923 | ||
924 | /*********************************************************************** | |
925 | * connect_class | |
926 | * Connect class cls to its superclasses, if possible. | |
927 | * If cls becomes connected, move it from unconnected_class_hash | |
928 | * to connected_class_hash. | |
929 | * Returns TRUE if cls is connected. | |
930 | * Returns FALSE if cls could not be connected for some reason | |
931 | * (missing superclass or still-unconnected superclass) | |
932 | **********************************************************************/ | |
933 | static bool connect_class(Class cls) | |
934 | { | |
935 | if (cls->isConnected()) { | |
936 | // This class is already connected to its superclass. | |
937 | // Do nothing. | |
938 | return TRUE; | |
939 | } | |
940 | else if (cls->superclass == nil) { | |
941 | // This class is a root class. | |
942 | // Connect it to itself. | |
943 | ||
944 | if (PrintConnecting) { | |
945 | _objc_inform("CONNECT: class '%s' now connected (root class)", | |
946 | cls->name); | |
947 | } | |
948 | ||
949 | really_connect_class(cls, nil); | |
950 | return TRUE; | |
951 | } | |
952 | else { | |
953 | // This class is not a root class and is not yet connected. | |
954 | // Connect it if its superclass and root class are already connected. | |
955 | // Otherwise, add this class to the to-be-connected list, | |
956 | // pending the completion of its superclass and root class. | |
957 | ||
958 | // At this point, cls->superclass and cls->ISA()->ISA() are still STRINGS | |
959 | char *supercls_name = (char *)cls->superclass; | |
960 | Class supercls; | |
961 | ||
962 | // YES unconnected, YES class handler | |
963 | if (nil == (supercls = look_up_class(supercls_name, YES, YES))) { | |
964 | // Superclass does not exist yet. | |
965 | // pendClassInstallation will handle duplicate pends of this class | |
966 | pendClassInstallation(cls, supercls_name); | |
967 | ||
968 | if (PrintConnecting) { | |
969 | _objc_inform("CONNECT: class '%s' NOT connected (missing super)", cls->name); | |
970 | } | |
971 | return FALSE; | |
972 | } | |
973 | ||
974 | if (! connect_class(supercls)) { | |
975 | // Superclass exists but is not yet connected. | |
976 | // pendClassInstallation will handle duplicate pends of this class | |
977 | pendClassInstallation(cls, supercls_name); | |
978 | ||
979 | if (PrintConnecting) { | |
980 | _objc_inform("CONNECT: class '%s' NOT connected (unconnected super)", cls->name); | |
981 | } | |
982 | return FALSE; | |
983 | } | |
984 | ||
985 | // Superclass exists and is connected. | |
986 | // Connect this class to the superclass. | |
987 | ||
988 | if (PrintConnecting) { | |
989 | _objc_inform("CONNECT: class '%s' now connected", cls->name); | |
990 | } | |
991 | ||
992 | really_connect_class(cls, supercls); | |
993 | return TRUE; | |
994 | } | |
995 | } | |
996 | ||
997 | ||
998 | /*********************************************************************** | |
999 | * _objc_read_categories_from_image. | |
1000 | * Read all categories from the given image. | |
1001 | * Install them on their parent classes, or register them for later | |
1002 | * installation. | |
1003 | * Returns YES if some method caches now need to be flushed. | |
1004 | **********************************************************************/ | |
1005 | static bool _objc_read_categories_from_image (header_info * hi) | |
1006 | { | |
1007 | Module mods; | |
1008 | size_t midx; | |
1009 | bool needFlush = NO; | |
1010 | ||
1011 | if (hi->info()->isReplacement()) { | |
1012 | // Ignore any categories in this image | |
1013 | return NO; | |
1014 | } | |
1015 | ||
1016 | // Major loop - process all modules in the header | |
1017 | mods = hi->mod_ptr; | |
1018 | ||
1019 | // NOTE: The module and category lists are traversed backwards | |
1020 | // to preserve the pre-10.4 processing order. Changing the order | |
1021 | // would have a small chance of introducing binary compatibility bugs. | |
1022 | midx = hi->mod_count; | |
1023 | while (midx-- > 0) { | |
1024 | unsigned int index; | |
1025 | unsigned int total; | |
1026 | ||
1027 | // Nothing to do for a module without a symbol table | |
1028 | if (mods[midx].symtab == nil) | |
1029 | continue; | |
1030 | ||
1031 | // Total entries in symbol table (class entries followed | |
1032 | // by category entries) | |
1033 | total = mods[midx].symtab->cls_def_cnt + | |
1034 | mods[midx].symtab->cat_def_cnt; | |
1035 | ||
1036 | // Minor loop - register all categories from given module | |
1037 | index = total; | |
1038 | while (index-- > mods[midx].symtab->cls_def_cnt) { | |
1039 | old_category *cat = (old_category *)mods[midx].symtab->defs[index]; | |
1040 | needFlush |= _objc_register_category(cat, (int)mods[midx].version); | |
1041 | } | |
1042 | } | |
1043 | ||
1044 | return needFlush; | |
1045 | } | |
1046 | ||
1047 | ||
1048 | /*********************************************************************** | |
1049 | * _objc_read_classes_from_image. | |
1050 | * Read classes from the given image, perform assorted minor fixups, | |
1051 | * scan for +load implementation. | |
1052 | * Does not connect classes to superclasses. | |
1053 | * Does attach pended categories to the classes. | |
1054 | * Adds all classes to unconnected_class_hash. class_hash is unchanged. | |
1055 | **********************************************************************/ | |
1056 | static void _objc_read_classes_from_image(header_info *hi) | |
1057 | { | |
1058 | unsigned int index; | |
1059 | unsigned int midx; | |
1060 | Module mods; | |
1061 | int isBundle = headerIsBundle(hi); | |
1062 | ||
1063 | if (hi->info()->isReplacement()) { | |
1064 | // Ignore any classes in this image | |
1065 | return; | |
1066 | } | |
1067 | ||
1068 | // class_hash starts small, enough only for libobjc itself. | |
1069 | // If other Objective-C libraries are found, immediately resize | |
1070 | // class_hash, assuming that Foundation and AppKit are about | |
1071 | // to add lots of classes. | |
1072 | { | |
1073 | mutex_locker_t lock(classLock); | |
1074 | if (hi->mhdr() != libobjc_header && _NXHashCapacity(class_hash) < 1024) { | |
1075 | _NXHashRehashToCapacity(class_hash, 1024); | |
1076 | } | |
1077 | } | |
1078 | ||
1079 | // Major loop - process all modules in the image | |
1080 | mods = hi->mod_ptr; | |
1081 | for (midx = 0; midx < hi->mod_count; midx += 1) | |
1082 | { | |
1083 | // Skip module containing no classes | |
1084 | if (mods[midx].symtab == nil) | |
1085 | continue; | |
1086 | ||
1087 | // Minor loop - process all the classes in given module | |
1088 | for (index = 0; index < mods[midx].symtab->cls_def_cnt; index += 1) | |
1089 | { | |
1090 | Class newCls, oldCls; | |
1091 | bool rejected; | |
1092 | ||
1093 | // Locate the class description pointer | |
1094 | newCls = (Class)mods[midx].symtab->defs[index]; | |
1095 | ||
1096 | // Classes loaded from Mach-O bundles can be unloaded later. | |
1097 | // Nothing uses this class yet, so cls->setInfo is not needed. | |
1098 | if (isBundle) newCls->info |= CLS_FROM_BUNDLE; | |
1099 | if (isBundle) newCls->ISA()->info |= CLS_FROM_BUNDLE; | |
1100 | ||
1101 | // Use common static empty cache instead of nil | |
1102 | if (newCls->cache == nil) | |
1103 | newCls->cache = (Cache) &_objc_empty_cache; | |
1104 | if (newCls->ISA()->cache == nil) | |
1105 | newCls->ISA()->cache = (Cache) &_objc_empty_cache; | |
1106 | ||
1107 | // Set metaclass version | |
1108 | newCls->ISA()->version = mods[midx].version; | |
1109 | ||
1110 | // methodLists is nil or a single list, not an array | |
1111 | newCls->info |= CLS_NO_METHOD_ARRAY|CLS_NO_PROPERTY_ARRAY; | |
1112 | newCls->ISA()->info |= CLS_NO_METHOD_ARRAY|CLS_NO_PROPERTY_ARRAY; | |
1113 | ||
1114 | // class has no subclasses for cache flushing | |
1115 | newCls->info |= CLS_LEAF; | |
1116 | newCls->ISA()->info |= CLS_LEAF; | |
1117 | ||
1118 | if (mods[midx].version >= 6) { | |
1119 | // class structure has ivar_layout and ext fields | |
1120 | newCls->info |= CLS_EXT; | |
1121 | newCls->ISA()->info |= CLS_EXT; | |
1122 | } | |
1123 | ||
1124 | // Check for +load implementation before categories are attached | |
1125 | if (_class_hasLoadMethod(newCls)) { | |
1126 | newCls->ISA()->info |= CLS_HAS_LOAD_METHOD; | |
1127 | } | |
1128 | ||
1129 | // Install into unconnected_class_hash. | |
1130 | { | |
1131 | mutex_locker_t lock(classLock); | |
1132 | ||
1133 | if (future_class_hash) { | |
1134 | Class futureCls = (Class) | |
1135 | NXHashRemove(future_class_hash, newCls); | |
1136 | if (futureCls) { | |
1137 | // Another class structure for this class was already | |
1138 | // prepared by objc_getFutureClass(). Use it instead. | |
1139 | free((char *)futureCls->name); | |
1140 | memcpy(futureCls, newCls, sizeof(objc_class)); | |
1141 | setOriginalClassForFutureClass(futureCls, newCls); | |
1142 | newCls = futureCls; | |
1143 | ||
1144 | if (NXCountHashTable(future_class_hash) == 0) { | |
1145 | NXFreeHashTable(future_class_hash); | |
1146 | future_class_hash = nil; | |
1147 | } | |
1148 | } | |
1149 | } | |
1150 | ||
1151 | if (!unconnected_class_hash) { | |
1152 | unconnected_class_hash = | |
1153 | NXCreateHashTable(classHashPrototype, 128, nil); | |
1154 | } | |
1155 | ||
1156 | if ((oldCls = (Class)NXHashGet(class_hash, newCls)) || | |
1157 | (oldCls = (Class)NXHashGet(unconnected_class_hash, newCls))) | |
1158 | { | |
1159 | // Another class with this name exists. Complain and reject. | |
1160 | inform_duplicate(newCls->name, oldCls, newCls); | |
1161 | rejected = YES; | |
1162 | } | |
1163 | else { | |
1164 | NXHashInsert(unconnected_class_hash, newCls); | |
1165 | rejected = NO; | |
1166 | } | |
1167 | } | |
1168 | ||
1169 | if (!rejected) { | |
1170 | // Attach pended categories for this class, if any | |
1171 | resolve_categories_for_class(newCls); | |
1172 | } | |
1173 | } | |
1174 | } | |
1175 | } | |
1176 | ||
1177 | ||
1178 | /*********************************************************************** | |
1179 | * _objc_connect_classes_from_image. | |
1180 | * Connect the classes in the given image to their superclasses, | |
1181 | * or register them for later connection if any superclasses are missing. | |
1182 | **********************************************************************/ | |
1183 | static void _objc_connect_classes_from_image(header_info *hi) | |
1184 | { | |
1185 | unsigned int index; | |
1186 | unsigned int midx; | |
1187 | Module mods; | |
1188 | bool replacement = hi->info()->isReplacement(); | |
1189 | ||
1190 | // Major loop - process all modules in the image | |
1191 | mods = hi->mod_ptr; | |
1192 | for (midx = 0; midx < hi->mod_count; midx += 1) | |
1193 | { | |
1194 | // Skip module containing no classes | |
1195 | if (mods[midx].symtab == nil) | |
1196 | continue; | |
1197 | ||
1198 | // Minor loop - process all the classes in given module | |
1199 | for (index = 0; index < mods[midx].symtab->cls_def_cnt; index += 1) | |
1200 | { | |
1201 | Class cls = (Class)mods[midx].symtab->defs[index]; | |
1202 | if (! replacement) { | |
1203 | bool connected; | |
1204 | Class futureCls = getFutureClassForOriginalClass(cls); | |
1205 | if (futureCls) { | |
1206 | // objc_getFutureClass() requested a different class | |
1207 | // struct. Fix up the original struct's superclass | |
1208 | // field for [super ...] use, but otherwise perform | |
1209 | // fixups on the new class struct only. | |
1210 | const char *super_name = (const char *) cls->superclass; | |
1211 | if (super_name) cls->superclass = objc_getClass(super_name); | |
1212 | cls = futureCls; | |
1213 | } | |
1214 | connected = connect_class(cls); | |
1215 | if (connected && callbackFunction) { | |
1216 | (*callbackFunction)(cls, 0); | |
1217 | } | |
1218 | } else { | |
1219 | // Replacement image - fix up superclass only (#3704817) | |
1220 | // And metaclass's superclass (#5351107) | |
1221 | const char *super_name = (const char *) cls->superclass; | |
1222 | if (super_name) { | |
1223 | cls->superclass = objc_getClass(super_name); | |
1224 | // metaclass's superclass is superclass's metaclass | |
1225 | cls->ISA()->superclass = cls->superclass->ISA(); | |
1226 | } else { | |
1227 | // Replacement for a root class | |
1228 | // cls->superclass already nil | |
1229 | // root metaclass's superclass is root class | |
1230 | cls->ISA()->superclass = cls; | |
1231 | } | |
1232 | } | |
1233 | } | |
1234 | } | |
1235 | } | |
1236 | ||
1237 | ||
1238 | /*********************************************************************** | |
1239 | * _objc_map_class_refs_for_image. Convert the class ref entries from | |
1240 | * a class name string pointer to a class pointer. If the class does | |
1241 | * not yet exist, the reference is added to a list of pending references | |
1242 | * to be fixed up at a later date. | |
1243 | **********************************************************************/ | |
1244 | static void fix_class_ref(Class *ref, const char *name, bool isMeta) | |
1245 | { | |
1246 | Class cls; | |
1247 | ||
1248 | // Get pointer to class of this name | |
1249 | // NO unconnected, YES class loader | |
1250 | // (real class with weak-missing superclass is unconnected now) | |
1251 | cls = look_up_class(name, NO, YES); | |
1252 | if (cls) { | |
1253 | // Referenced class exists. Fix up the reference. | |
1254 | *ref = isMeta ? cls->ISA() : cls; | |
1255 | } else { | |
1256 | // Referenced class does not exist yet. Insert nil for now | |
1257 | // (weak-linking) and fix up the reference if the class arrives later. | |
1258 | pendClassReference (ref, name, isMeta); | |
1259 | *ref = nil; | |
1260 | } | |
1261 | } | |
1262 | ||
1263 | static void _objc_map_class_refs_for_image (header_info * hi) | |
1264 | { | |
1265 | Class *cls_refs; | |
1266 | size_t count; | |
1267 | unsigned int index; | |
1268 | ||
1269 | // Locate class refs in image | |
1270 | cls_refs = _getObjcClassRefs (hi, &count); | |
1271 | if (cls_refs) { | |
1272 | // Process each class ref | |
1273 | for (index = 0; index < count; index += 1) { | |
1274 | // Ref is initially class name char* | |
1275 | const char *name = (const char *) cls_refs[index]; | |
1276 | if (!name) continue; | |
1277 | fix_class_ref(&cls_refs[index], name, NO /*never meta*/); | |
1278 | } | |
1279 | } | |
1280 | } | |
1281 | ||
1282 | ||
1283 | /*********************************************************************** | |
1284 | * _objc_remove_pending_class_refs_in_image | |
1285 | * Delete any pending class ref fixups for class refs in the given image, | |
1286 | * because the image is about to be unloaded. | |
1287 | **********************************************************************/ | |
1288 | static void removePendingReferences(Class *refs, size_t count) | |
1289 | { | |
1290 | Class *end = refs + count; | |
1291 | ||
1292 | if (!refs) return; | |
1293 | if (!pendingClassRefsMap) return; | |
1294 | ||
1295 | // Search the pending class ref table for class refs in this range. | |
1296 | // The class refs may have already been stomped with nil, | |
1297 | // so there's no way to recover the original class name. | |
1298 | ||
1299 | { | |
1300 | const char *key; | |
1301 | PendingClassRef *pending; | |
1302 | NXMapState state = NXInitMapState(pendingClassRefsMap); | |
1303 | while(NXNextMapState(pendingClassRefsMap, &state, | |
1304 | (const void **)&key, (const void **)&pending)) | |
1305 | { | |
1306 | for ( ; pending != nil; pending = pending->next) { | |
1307 | if (pending->ref >= refs && pending->ref < end) { | |
1308 | pending->ref = nil; | |
1309 | } | |
1310 | } | |
1311 | } | |
1312 | } | |
1313 | } | |
1314 | ||
1315 | static void _objc_remove_pending_class_refs_in_image(header_info *hi) | |
1316 | { | |
1317 | Class *cls_refs; | |
1318 | size_t count; | |
1319 | ||
1320 | // Locate class refs in this image | |
1321 | cls_refs = _getObjcClassRefs(hi, &count); | |
1322 | removePendingReferences(cls_refs, count); | |
1323 | } | |
1324 | ||
1325 | ||
1326 | /*********************************************************************** | |
1327 | * map_selrefs. For each selector in the specified array, | |
1328 | * replace the name pointer with a uniqued selector. | |
1329 | * If copy is TRUE, all selector data is always copied. This is used | |
1330 | * for registering selectors from unloadable bundles, so the selector | |
1331 | * can still be used after the bundle's data segment is unmapped. | |
1332 | * Returns YES if dst was written to, NO if it was unchanged. | |
1333 | **********************************************************************/ | |
1334 | static inline void map_selrefs(SEL *sels, size_t count, bool copy) | |
1335 | { | |
1336 | size_t index; | |
1337 | ||
1338 | if (!sels) return; | |
1339 | ||
1340 | sel_lock(); | |
1341 | ||
1342 | // Process each selector | |
1343 | for (index = 0; index < count; index += 1) | |
1344 | { | |
1345 | SEL sel; | |
1346 | ||
1347 | // Lookup pointer to uniqued string | |
1348 | sel = sel_registerNameNoLock((const char *) sels[index], copy); | |
1349 | ||
1350 | // Replace this selector with uniqued one (avoid | |
1351 | // modifying the VM page if this would be a NOP) | |
1352 | if (sels[index] != sel) { | |
1353 | sels[index] = sel; | |
1354 | } | |
1355 | } | |
1356 | ||
1357 | sel_unlock(); | |
1358 | } | |
1359 | ||
1360 | ||
1361 | /*********************************************************************** | |
1362 | * map_method_descs. For each method in the specified method list, | |
1363 | * replace the name pointer with a uniqued selector. | |
1364 | * If copy is TRUE, all selector data is always copied. This is used | |
1365 | * for registering selectors from unloadable bundles, so the selector | |
1366 | * can still be used after the bundle's data segment is unmapped. | |
1367 | **********************************************************************/ | |
1368 | static void map_method_descs (struct objc_method_description_list * methods, bool copy) | |
1369 | { | |
1370 | int index; | |
1371 | ||
1372 | if (!methods) return; | |
1373 | ||
1374 | sel_lock(); | |
1375 | ||
1376 | // Process each method | |
1377 | for (index = 0; index < methods->count; index += 1) | |
1378 | { | |
1379 | struct objc_method_description * method; | |
1380 | SEL sel; | |
1381 | ||
1382 | // Get method entry to fix up | |
1383 | method = &methods->list[index]; | |
1384 | ||
1385 | // Lookup pointer to uniqued string | |
1386 | sel = sel_registerNameNoLock((const char *) method->name, copy); | |
1387 | ||
1388 | // Replace this selector with uniqued one (avoid | |
1389 | // modifying the VM page if this would be a NOP) | |
1390 | if (method->name != sel) | |
1391 | method->name = sel; | |
1392 | } | |
1393 | ||
1394 | sel_unlock(); | |
1395 | } | |
1396 | ||
1397 | ||
1398 | /*********************************************************************** | |
1399 | * ext_for_protocol | |
1400 | * Returns the protocol extension for the given protocol. | |
1401 | * Returns nil if the protocol has no extension. | |
1402 | **********************************************************************/ | |
1403 | static old_protocol_ext *ext_for_protocol(old_protocol *proto) | |
1404 | { | |
1405 | if (!proto) return nil; | |
1406 | if (!protocol_ext_map) return nil; | |
1407 | else return (old_protocol_ext *)NXMapGet(protocol_ext_map, proto); | |
1408 | } | |
1409 | ||
1410 | ||
1411 | /*********************************************************************** | |
1412 | * lookup_method | |
1413 | * Search a protocol method list for a selector. | |
1414 | **********************************************************************/ | |
1415 | static struct objc_method_description * | |
1416 | lookup_method(struct objc_method_description_list *mlist, SEL aSel) | |
1417 | { | |
1418 | if (mlist) { | |
1419 | int i; | |
1420 | for (i = 0; i < mlist->count; i++) { | |
1421 | if (mlist->list[i].name == aSel) { | |
1422 | return mlist->list+i; | |
1423 | } | |
1424 | } | |
1425 | } | |
1426 | return nil; | |
1427 | } | |
1428 | ||
1429 | ||
1430 | /*********************************************************************** | |
1431 | * lookup_protocol_method | |
1432 | * Search for a selector in a protocol | |
1433 | * (and optionally recursively all incorporated protocols) | |
1434 | **********************************************************************/ | |
1435 | struct objc_method_description * | |
1436 | lookup_protocol_method(old_protocol *proto, SEL aSel, | |
1437 | bool isRequiredMethod, bool isInstanceMethod, | |
1438 | bool recursive) | |
1439 | { | |
1440 | struct objc_method_description *m = nil; | |
1441 | old_protocol_ext *ext; | |
1442 | ||
1443 | if (isRequiredMethod) { | |
1444 | if (isInstanceMethod) { | |
1445 | m = lookup_method(proto->instance_methods, aSel); | |
1446 | } else { | |
1447 | m = lookup_method(proto->class_methods, aSel); | |
1448 | } | |
1449 | } else if ((ext = ext_for_protocol(proto))) { | |
1450 | if (isInstanceMethod) { | |
1451 | m = lookup_method(ext->optional_instance_methods, aSel); | |
1452 | } else { | |
1453 | m = lookup_method(ext->optional_class_methods, aSel); | |
1454 | } | |
1455 | } | |
1456 | ||
1457 | if (!m && recursive && proto->protocol_list) { | |
1458 | int i; | |
1459 | for (i = 0; !m && i < proto->protocol_list->count; i++) { | |
1460 | m = lookup_protocol_method(proto->protocol_list->list[i], aSel, | |
1461 | isRequiredMethod,isInstanceMethod,true); | |
1462 | } | |
1463 | } | |
1464 | ||
1465 | return m; | |
1466 | } | |
1467 | ||
1468 | ||
1469 | /*********************************************************************** | |
1470 | * protocol_getName | |
1471 | * Returns the name of the given protocol. | |
1472 | **********************************************************************/ | |
1473 | const char *protocol_getName(Protocol *p) | |
1474 | { | |
1475 | old_protocol *proto = oldprotocol(p); | |
1476 | if (!proto) return "nil"; | |
1477 | return proto->protocol_name; | |
1478 | } | |
1479 | ||
1480 | ||
1481 | /*********************************************************************** | |
1482 | * protocol_getMethodDescription | |
1483 | * Returns the description of a named method. | |
1484 | * Searches either required or optional methods. | |
1485 | * Searches either instance or class methods. | |
1486 | **********************************************************************/ | |
1487 | struct objc_method_description | |
1488 | protocol_getMethodDescription(Protocol *p, SEL aSel, | |
1489 | BOOL isRequiredMethod, BOOL isInstanceMethod) | |
1490 | { | |
1491 | struct objc_method_description empty = {nil, nil}; | |
1492 | old_protocol *proto = oldprotocol(p); | |
1493 | struct objc_method_description *desc; | |
1494 | if (!proto) return empty; | |
1495 | ||
1496 | desc = lookup_protocol_method(proto, aSel, | |
1497 | isRequiredMethod, isInstanceMethod, true); | |
1498 | if (desc) return *desc; | |
1499 | else return empty; | |
1500 | } | |
1501 | ||
1502 | ||
1503 | /*********************************************************************** | |
1504 | * protocol_copyMethodDescriptionList | |
1505 | * Returns an array of method descriptions from a protocol. | |
1506 | * Copies either required or optional methods. | |
1507 | * Copies either instance or class methods. | |
1508 | **********************************************************************/ | |
1509 | struct objc_method_description * | |
1510 | protocol_copyMethodDescriptionList(Protocol *p, | |
1511 | BOOL isRequiredMethod, | |
1512 | BOOL isInstanceMethod, | |
1513 | unsigned int *outCount) | |
1514 | { | |
1515 | struct objc_method_description_list *mlist = nil; | |
1516 | old_protocol *proto = oldprotocol(p); | |
1517 | old_protocol_ext *ext; | |
1518 | unsigned int i, count; | |
1519 | struct objc_method_description *result; | |
1520 | ||
1521 | if (!proto) { | |
1522 | if (outCount) *outCount = 0; | |
1523 | return nil; | |
1524 | } | |
1525 | ||
1526 | if (isRequiredMethod) { | |
1527 | if (isInstanceMethod) { | |
1528 | mlist = proto->instance_methods; | |
1529 | } else { | |
1530 | mlist = proto->class_methods; | |
1531 | } | |
1532 | } else if ((ext = ext_for_protocol(proto))) { | |
1533 | if (isInstanceMethod) { | |
1534 | mlist = ext->optional_instance_methods; | |
1535 | } else { | |
1536 | mlist = ext->optional_class_methods; | |
1537 | } | |
1538 | } | |
1539 | ||
1540 | if (!mlist) { | |
1541 | if (outCount) *outCount = 0; | |
1542 | return nil; | |
1543 | } | |
1544 | ||
1545 | count = mlist->count; | |
1546 | result = (struct objc_method_description *) | |
1547 | calloc(count + 1, sizeof(struct objc_method_description)); | |
1548 | for (i = 0; i < count; i++) { | |
1549 | result[i] = mlist->list[i]; | |
1550 | } | |
1551 | ||
1552 | if (outCount) *outCount = count; | |
1553 | return result; | |
1554 | } | |
1555 | ||
1556 | ||
1557 | objc_property_t | |
1558 | protocol_getProperty(Protocol *p, const char *name, | |
1559 | BOOL isRequiredProperty, BOOL isInstanceProperty) | |
1560 | { | |
1561 | old_protocol *proto = oldprotocol(p); | |
1562 | old_protocol_ext *ext; | |
1563 | old_protocol_list *proto_list; | |
1564 | ||
1565 | if (!proto || !name) return nil; | |
1566 | ||
1567 | if (!isRequiredProperty) { | |
1568 | // Only required properties are currently supported | |
1569 | return nil; | |
1570 | } | |
1571 | ||
1572 | if ((ext = ext_for_protocol(proto))) { | |
1573 | old_property_list *plist; | |
1574 | if (isInstanceProperty) plist = ext->instance_properties; | |
1575 | else if (ext->hasClassPropertiesField()) plist = ext->class_properties; | |
1576 | else plist = nil; | |
1577 | ||
1578 | if (plist) { | |
1579 | uint32_t i; | |
1580 | for (i = 0; i < plist->count; i++) { | |
1581 | old_property *prop = property_list_nth(plist, i); | |
1582 | if (0 == strcmp(name, prop->name)) { | |
1583 | return (objc_property_t)prop; | |
1584 | } | |
1585 | } | |
1586 | } | |
1587 | } | |
1588 | ||
1589 | if ((proto_list = proto->protocol_list)) { | |
1590 | int i; | |
1591 | for (i = 0; i < proto_list->count; i++) { | |
1592 | objc_property_t prop = | |
1593 | protocol_getProperty((Protocol *)proto_list->list[i], name, | |
1594 | isRequiredProperty, isInstanceProperty); | |
1595 | if (prop) return prop; | |
1596 | } | |
1597 | } | |
1598 | ||
1599 | return nil; | |
1600 | } | |
1601 | ||
1602 | ||
1603 | objc_property_t * | |
1604 | protocol_copyPropertyList2(Protocol *p, unsigned int *outCount, | |
1605 | BOOL isRequiredProperty, BOOL isInstanceProperty) | |
1606 | { | |
1607 | old_property **result = nil; | |
1608 | old_protocol_ext *ext; | |
1609 | old_property_list *plist; | |
1610 | ||
1611 | old_protocol *proto = oldprotocol(p); | |
1612 | if (! (ext = ext_for_protocol(proto)) || !isRequiredProperty) { | |
1613 | // Only required properties are currently supported. | |
1614 | if (outCount) *outCount = 0; | |
1615 | return nil; | |
1616 | } | |
1617 | ||
1618 | if (isInstanceProperty) plist = ext->instance_properties; | |
1619 | else if (ext->hasClassPropertiesField()) plist = ext->class_properties; | |
1620 | else plist = nil; | |
1621 | ||
1622 | result = copyPropertyList(plist, outCount); | |
1623 | ||
1624 | return (objc_property_t *)result; | |
1625 | } | |
1626 | ||
1627 | objc_property_t *protocol_copyPropertyList(Protocol *p, unsigned int *outCount) | |
1628 | { | |
1629 | return protocol_copyPropertyList2(p, outCount, YES, YES); | |
1630 | } | |
1631 | ||
1632 | ||
1633 | /*********************************************************************** | |
1634 | * protocol_copyProtocolList | |
1635 | * Copies this protocol's incorporated protocols. | |
1636 | * Does not copy those protocol's incorporated protocols in turn. | |
1637 | **********************************************************************/ | |
1638 | Protocol * __unsafe_unretained * | |
1639 | protocol_copyProtocolList(Protocol *p, unsigned int *outCount) | |
1640 | { | |
1641 | unsigned int count = 0; | |
1642 | Protocol **result = nil; | |
1643 | old_protocol *proto = oldprotocol(p); | |
1644 | ||
1645 | if (!proto) { | |
1646 | if (outCount) *outCount = 0; | |
1647 | return nil; | |
1648 | } | |
1649 | ||
1650 | if (proto->protocol_list) { | |
1651 | count = (unsigned int)proto->protocol_list->count; | |
1652 | } | |
1653 | if (count > 0) { | |
1654 | unsigned int i; | |
1655 | result = (Protocol **)malloc((count+1) * sizeof(Protocol *)); | |
1656 | ||
1657 | for (i = 0; i < count; i++) { | |
1658 | result[i] = (Protocol *)proto->protocol_list->list[i]; | |
1659 | } | |
1660 | result[i] = nil; | |
1661 | } | |
1662 | ||
1663 | if (outCount) *outCount = count; | |
1664 | return result; | |
1665 | } | |
1666 | ||
1667 | ||
1668 | BOOL protocol_conformsToProtocol(Protocol *self_gen, Protocol *other_gen) | |
1669 | { | |
1670 | old_protocol *self = oldprotocol(self_gen); | |
1671 | old_protocol *other = oldprotocol(other_gen); | |
1672 | ||
1673 | if (!self || !other) { | |
1674 | return NO; | |
1675 | } | |
1676 | ||
1677 | if (0 == strcmp(self->protocol_name, other->protocol_name)) { | |
1678 | return YES; | |
1679 | } | |
1680 | ||
1681 | if (self->protocol_list) { | |
1682 | int i; | |
1683 | for (i = 0; i < self->protocol_list->count; i++) { | |
1684 | old_protocol *proto = self->protocol_list->list[i]; | |
1685 | if (0 == strcmp(other->protocol_name, proto->protocol_name)) { | |
1686 | return YES; | |
1687 | } | |
1688 | if (protocol_conformsToProtocol((Protocol *)proto, other_gen)) { | |
1689 | return YES; | |
1690 | } | |
1691 | } | |
1692 | } | |
1693 | ||
1694 | return NO; | |
1695 | } | |
1696 | ||
1697 | ||
1698 | BOOL protocol_isEqual(Protocol *self, Protocol *other) | |
1699 | { | |
1700 | if (self == other) return YES; | |
1701 | if (!self || !other) return NO; | |
1702 | ||
1703 | if (!protocol_conformsToProtocol(self, other)) return NO; | |
1704 | if (!protocol_conformsToProtocol(other, self)) return NO; | |
1705 | ||
1706 | return YES; | |
1707 | } | |
1708 | ||
1709 | ||
1710 | /*********************************************************************** | |
1711 | * _protocol_getMethodTypeEncoding | |
1712 | * Return the @encode string for the requested protocol method. | |
1713 | * Returns nil if the compiler did not emit any extended @encode data. | |
1714 | * Locking: runtimeLock must not be held by the caller | |
1715 | **********************************************************************/ | |
1716 | const char * | |
1717 | _protocol_getMethodTypeEncoding(Protocol *proto_gen, SEL sel, | |
1718 | BOOL isRequiredMethod, BOOL isInstanceMethod) | |
1719 | { | |
1720 | old_protocol *proto = oldprotocol(proto_gen); | |
1721 | if (!proto) return nil; | |
1722 | old_protocol_ext *ext = ext_for_protocol(proto); | |
1723 | if (!ext) return nil; | |
1724 | if (ext->size < offsetof(old_protocol_ext, extendedMethodTypes) + sizeof(ext->extendedMethodTypes)) return nil; | |
1725 | if (! ext->extendedMethodTypes) return nil; | |
1726 | ||
1727 | struct objc_method_description *m = | |
1728 | lookup_protocol_method(proto, sel, | |
1729 | isRequiredMethod, isInstanceMethod, false); | |
1730 | if (!m) { | |
1731 | // No method with that name. Search incorporated protocols. | |
1732 | if (proto->protocol_list) { | |
1733 | for (int i = 0; i < proto->protocol_list->count; i++) { | |
1734 | const char *enc = | |
1735 | _protocol_getMethodTypeEncoding((Protocol *)proto->protocol_list->list[i], sel, isRequiredMethod, isInstanceMethod); | |
1736 | if (enc) return enc; | |
1737 | } | |
1738 | } | |
1739 | return nil; | |
1740 | } | |
1741 | ||
1742 | int i = 0; | |
1743 | if (isRequiredMethod && isInstanceMethod) { | |
1744 | i += ((uintptr_t)m - (uintptr_t)proto->instance_methods) / sizeof(proto->instance_methods->list[0]); | |
1745 | goto done; | |
1746 | } else if (proto->instance_methods) { | |
1747 | i += proto->instance_methods->count; | |
1748 | } | |
1749 | ||
1750 | if (isRequiredMethod && !isInstanceMethod) { | |
1751 | i += ((uintptr_t)m - (uintptr_t)proto->class_methods) / sizeof(proto->class_methods->list[0]); | |
1752 | goto done; | |
1753 | } else if (proto->class_methods) { | |
1754 | i += proto->class_methods->count; | |
1755 | } | |
1756 | ||
1757 | if (!isRequiredMethod && isInstanceMethod) { | |
1758 | i += ((uintptr_t)m - (uintptr_t)ext->optional_instance_methods) / sizeof(ext->optional_instance_methods->list[0]); | |
1759 | goto done; | |
1760 | } else if (ext->optional_instance_methods) { | |
1761 | i += ext->optional_instance_methods->count; | |
1762 | } | |
1763 | ||
1764 | if (!isRequiredMethod && !isInstanceMethod) { | |
1765 | i += ((uintptr_t)m - (uintptr_t)ext->optional_class_methods) / sizeof(ext->optional_class_methods->list[0]); | |
1766 | goto done; | |
1767 | } else if (ext->optional_class_methods) { | |
1768 | i += ext->optional_class_methods->count; | |
1769 | } | |
1770 | ||
1771 | done: | |
1772 | return ext->extendedMethodTypes[i]; | |
1773 | } | |
1774 | ||
1775 | ||
1776 | /*********************************************************************** | |
1777 | * objc_allocateProtocol | |
1778 | * Creates a new protocol. The protocol may not be used until | |
1779 | * objc_registerProtocol() is called. | |
1780 | * Returns nil if a protocol with the same name already exists. | |
1781 | * Locking: acquires classLock | |
1782 | **********************************************************************/ | |
1783 | Protocol * | |
1784 | objc_allocateProtocol(const char *name) | |
1785 | { | |
1786 | Class cls = objc_getClass("__IncompleteProtocol"); | |
1787 | assert(cls); | |
1788 | ||
1789 | mutex_locker_t lock(classLock); | |
1790 | ||
1791 | if (NXMapGet(protocol_map, name)) return nil; | |
1792 | ||
1793 | old_protocol *result = (old_protocol *) | |
1794 | calloc(1, sizeof(old_protocol) | |
1795 | + sizeof(old_protocol_ext)); | |
1796 | old_protocol_ext *ext = (old_protocol_ext *)(result+1); | |
1797 | ||
1798 | result->isa = cls; | |
1799 | result->protocol_name = strdup(name); | |
1800 | ext->size = sizeof(old_protocol_ext); | |
1801 | ||
1802 | // fixme reserve name without installing | |
1803 | ||
1804 | NXMapInsert(protocol_ext_map, result, result+1); | |
1805 | ||
1806 | return (Protocol *)result; | |
1807 | } | |
1808 | ||
1809 | ||
1810 | /*********************************************************************** | |
1811 | * objc_registerProtocol | |
1812 | * Registers a newly-constructed protocol. The protocol is now | |
1813 | * ready for use and immutable. | |
1814 | * Locking: acquires classLock | |
1815 | **********************************************************************/ | |
1816 | void objc_registerProtocol(Protocol *proto_gen) | |
1817 | { | |
1818 | old_protocol *proto = oldprotocol(proto_gen); | |
1819 | ||
1820 | Class oldcls = objc_getClass("__IncompleteProtocol"); | |
1821 | Class cls = objc_getClass("Protocol"); | |
1822 | ||
1823 | mutex_locker_t lock(classLock); | |
1824 | ||
1825 | if (proto->isa == cls) { | |
1826 | _objc_inform("objc_registerProtocol: protocol '%s' was already " | |
1827 | "registered!", proto->protocol_name); | |
1828 | return; | |
1829 | } | |
1830 | if (proto->isa != oldcls) { | |
1831 | _objc_inform("objc_registerProtocol: protocol '%s' was not allocated " | |
1832 | "with objc_allocateProtocol!", proto->protocol_name); | |
1833 | return; | |
1834 | } | |
1835 | ||
1836 | proto->isa = cls; | |
1837 | ||
1838 | NXMapKeyCopyingInsert(protocol_map, proto->protocol_name, proto); | |
1839 | } | |
1840 | ||
1841 | ||
1842 | /*********************************************************************** | |
1843 | * protocol_addProtocol | |
1844 | * Adds an incorporated protocol to another protocol. | |
1845 | * No method enforcement is performed. | |
1846 | * `proto` must be under construction. `addition` must not. | |
1847 | * Locking: acquires classLock | |
1848 | **********************************************************************/ | |
1849 | void | |
1850 | protocol_addProtocol(Protocol *proto_gen, Protocol *addition_gen) | |
1851 | { | |
1852 | old_protocol *proto = oldprotocol(proto_gen); | |
1853 | old_protocol *addition = oldprotocol(addition_gen); | |
1854 | ||
1855 | Class cls = objc_getClass("__IncompleteProtocol"); | |
1856 | ||
1857 | if (!proto_gen) return; | |
1858 | if (!addition_gen) return; | |
1859 | ||
1860 | mutex_locker_t lock(classLock); | |
1861 | ||
1862 | if (proto->isa != cls) { | |
1863 | _objc_inform("protocol_addProtocol: modified protocol '%s' is not " | |
1864 | "under construction!", proto->protocol_name); | |
1865 | return; | |
1866 | } | |
1867 | if (addition->isa == cls) { | |
1868 | _objc_inform("protocol_addProtocol: added protocol '%s' is still " | |
1869 | "under construction!", addition->protocol_name); | |
1870 | return; | |
1871 | } | |
1872 | ||
1873 | old_protocol_list *protolist = proto->protocol_list; | |
1874 | if (protolist) { | |
1875 | size_t size = sizeof(old_protocol_list) | |
1876 | + protolist->count * sizeof(protolist->list[0]); | |
1877 | protolist = (old_protocol_list *) | |
1878 | realloc(protolist, size); | |
1879 | } else { | |
1880 | protolist = (old_protocol_list *) | |
1881 | calloc(1, sizeof(old_protocol_list)); | |
1882 | } | |
1883 | ||
1884 | protolist->list[protolist->count++] = addition; | |
1885 | proto->protocol_list = protolist; | |
1886 | } | |
1887 | ||
1888 | ||
1889 | /*********************************************************************** | |
1890 | * protocol_addMethodDescription | |
1891 | * Adds a method to a protocol. The protocol must be under construction. | |
1892 | * Locking: acquires classLock | |
1893 | **********************************************************************/ | |
1894 | static void | |
1895 | _protocol_addMethod(struct objc_method_description_list **list, SEL name, const char *types) | |
1896 | { | |
1897 | if (!*list) { | |
1898 | *list = (struct objc_method_description_list *) | |
1899 | calloc(sizeof(struct objc_method_description_list), 1); | |
1900 | } else { | |
1901 | size_t size = sizeof(struct objc_method_description_list) | |
1902 | + (*list)->count * sizeof(struct objc_method_description); | |
1903 | *list = (struct objc_method_description_list *) | |
1904 | realloc(*list, size); | |
1905 | } | |
1906 | ||
1907 | struct objc_method_description *desc = &(*list)->list[(*list)->count++]; | |
1908 | desc->name = name; | |
1909 | desc->types = strdup(types ?: ""); | |
1910 | } | |
1911 | ||
1912 | void | |
1913 | protocol_addMethodDescription(Protocol *proto_gen, SEL name, const char *types, | |
1914 | BOOL isRequiredMethod, BOOL isInstanceMethod) | |
1915 | { | |
1916 | old_protocol *proto = oldprotocol(proto_gen); | |
1917 | ||
1918 | Class cls = objc_getClass("__IncompleteProtocol"); | |
1919 | ||
1920 | if (!proto_gen) return; | |
1921 | ||
1922 | mutex_locker_t lock(classLock); | |
1923 | ||
1924 | if (proto->isa != cls) { | |
1925 | _objc_inform("protocol_addMethodDescription: protocol '%s' is not " | |
1926 | "under construction!", proto->protocol_name); | |
1927 | return; | |
1928 | } | |
1929 | ||
1930 | if (isRequiredMethod && isInstanceMethod) { | |
1931 | _protocol_addMethod(&proto->instance_methods, name, types); | |
1932 | } else if (isRequiredMethod && !isInstanceMethod) { | |
1933 | _protocol_addMethod(&proto->class_methods, name, types); | |
1934 | } else if (!isRequiredMethod && isInstanceMethod) { | |
1935 | old_protocol_ext *ext = (old_protocol_ext *)(proto+1); | |
1936 | _protocol_addMethod(&ext->optional_instance_methods, name, types); | |
1937 | } else /* !isRequiredMethod && !isInstanceMethod) */ { | |
1938 | old_protocol_ext *ext = (old_protocol_ext *)(proto+1); | |
1939 | _protocol_addMethod(&ext->optional_class_methods, name, types); | |
1940 | } | |
1941 | } | |
1942 | ||
1943 | ||
1944 | /*********************************************************************** | |
1945 | * protocol_addProperty | |
1946 | * Adds a property to a protocol. The protocol must be under construction. | |
1947 | * Locking: acquires classLock | |
1948 | **********************************************************************/ | |
1949 | static void | |
1950 | _protocol_addProperty(old_property_list **plist, const char *name, | |
1951 | const objc_property_attribute_t *attrs, | |
1952 | unsigned int count) | |
1953 | { | |
1954 | if (!*plist) { | |
1955 | *plist = (old_property_list *) | |
1956 | calloc(sizeof(old_property_list), 1); | |
1957 | (*plist)->entsize = sizeof(old_property); | |
1958 | } else { | |
1959 | *plist = (old_property_list *) | |
1960 | realloc(*plist, sizeof(old_property_list) | |
1961 | + (*plist)->count * (*plist)->entsize); | |
1962 | } | |
1963 | ||
1964 | old_property *prop = property_list_nth(*plist, (*plist)->count++); | |
1965 | prop->name = strdup(name); | |
1966 | prop->attributes = copyPropertyAttributeString(attrs, count); | |
1967 | } | |
1968 | ||
1969 | void | |
1970 | protocol_addProperty(Protocol *proto_gen, const char *name, | |
1971 | const objc_property_attribute_t *attrs, | |
1972 | unsigned int count, | |
1973 | BOOL isRequiredProperty, BOOL isInstanceProperty) | |
1974 | { | |
1975 | old_protocol *proto = oldprotocol(proto_gen); | |
1976 | ||
1977 | Class cls = objc_getClass("__IncompleteProtocol"); | |
1978 | ||
1979 | if (!proto) return; | |
1980 | if (!name) return; | |
1981 | ||
1982 | mutex_locker_t lock(classLock); | |
1983 | ||
1984 | if (proto->isa != cls) { | |
1985 | _objc_inform("protocol_addProperty: protocol '%s' is not " | |
1986 | "under construction!", proto->protocol_name); | |
1987 | return; | |
1988 | } | |
1989 | ||
1990 | old_protocol_ext *ext = ext_for_protocol(proto); | |
1991 | ||
1992 | if (isRequiredProperty && isInstanceProperty) { | |
1993 | _protocol_addProperty(&ext->instance_properties, name, attrs, count); | |
1994 | } | |
1995 | else if (isRequiredProperty && !isInstanceProperty) { | |
1996 | _protocol_addProperty(&ext->class_properties, name, attrs, count); | |
1997 | } | |
1998 | // else if (!isRequiredProperty && isInstanceProperty) { | |
1999 | // _protocol_addProperty(&ext->optional_instance_properties, name, attrs, count); | |
2000 | //} | |
2001 | // else /* !isRequiredProperty && !isInstanceProperty) */ { | |
2002 | // _protocol_addProperty(&ext->optional_class_properties, name, attrs, count); | |
2003 | //} | |
2004 | } | |
2005 | ||
2006 | ||
2007 | /*********************************************************************** | |
2008 | * _objc_fixup_protocol_objects_for_image. For each protocol in the | |
2009 | * specified image, selectorize the method names and add to the protocol hash. | |
2010 | **********************************************************************/ | |
2011 | ||
2012 | static bool versionIsExt(uintptr_t version, const char *names, size_t size) | |
2013 | { | |
2014 | // CodeWarrior used isa field for string "Protocol" | |
2015 | // from section __OBJC,__class_names. rdar://4951638 | |
2016 | // gcc (10.4 and earlier) used isa field for version number; | |
2017 | // the only version number used on Mac OS X was 2. | |
2018 | // gcc (10.5 and later) uses isa field for ext pointer | |
2019 | ||
2020 | if (version < 4096 /* not PAGE_SIZE */) { | |
2021 | return NO; | |
2022 | } | |
2023 | ||
2024 | if (version >= (uintptr_t)names && version < (uintptr_t)(names + size)) { | |
2025 | return NO; | |
2026 | } | |
2027 | ||
2028 | return YES; | |
2029 | } | |
2030 | ||
2031 | static void fix_protocol(old_protocol *proto, Class protocolClass, | |
2032 | bool isBundle, const char *names, size_t names_size) | |
2033 | { | |
2034 | uintptr_t version; | |
2035 | if (!proto) return; | |
2036 | ||
2037 | version = (uintptr_t)proto->isa; | |
2038 | ||
2039 | // Set the protocol's isa | |
2040 | proto->isa = protocolClass; | |
2041 | ||
2042 | // Fix up method lists | |
2043 | // fixme share across duplicates | |
2044 | map_method_descs (proto->instance_methods, isBundle); | |
2045 | map_method_descs (proto->class_methods, isBundle); | |
2046 | ||
2047 | // Fix up ext, if any | |
2048 | if (versionIsExt(version, names, names_size)) { | |
2049 | old_protocol_ext *ext = (old_protocol_ext *)version; | |
2050 | NXMapInsert(protocol_ext_map, proto, ext); | |
2051 | map_method_descs (ext->optional_instance_methods, isBundle); | |
2052 | map_method_descs (ext->optional_class_methods, isBundle); | |
2053 | } | |
2054 | ||
2055 | // Record the protocol it if we don't have one with this name yet | |
2056 | // fixme bundles - copy protocol | |
2057 | // fixme unloading | |
2058 | if (!NXMapGet(protocol_map, proto->protocol_name)) { | |
2059 | NXMapKeyCopyingInsert(protocol_map, proto->protocol_name, proto); | |
2060 | if (PrintProtocols) { | |
2061 | _objc_inform("PROTOCOLS: protocol at %p is %s", | |
2062 | proto, proto->protocol_name); | |
2063 | } | |
2064 | } else { | |
2065 | // duplicate - do nothing | |
2066 | if (PrintProtocols) { | |
2067 | _objc_inform("PROTOCOLS: protocol at %p is %s (duplicate)", | |
2068 | proto, proto->protocol_name); | |
2069 | } | |
2070 | } | |
2071 | } | |
2072 | ||
2073 | static void _objc_fixup_protocol_objects_for_image (header_info * hi) | |
2074 | { | |
2075 | Class protocolClass = objc_getClass("Protocol"); | |
2076 | size_t count, i; | |
2077 | old_protocol **protos; | |
2078 | int isBundle = headerIsBundle(hi); | |
2079 | const char *names; | |
2080 | size_t names_size; | |
2081 | ||
2082 | mutex_locker_t lock(classLock); | |
2083 | ||
2084 | // Allocate the protocol registry if necessary. | |
2085 | if (!protocol_map) { | |
2086 | protocol_map = | |
2087 | NXCreateMapTable(NXStrValueMapPrototype, 32); | |
2088 | } | |
2089 | if (!protocol_ext_map) { | |
2090 | protocol_ext_map = | |
2091 | NXCreateMapTable(NXPtrValueMapPrototype, 32); | |
2092 | } | |
2093 | ||
2094 | protos = _getObjcProtocols(hi, &count); | |
2095 | names = _getObjcClassNames(hi, &names_size); | |
2096 | for (i = 0; i < count; i++) { | |
2097 | fix_protocol(protos[i], protocolClass, isBundle, names, names_size); | |
2098 | } | |
2099 | } | |
2100 | ||
2101 | ||
2102 | /*********************************************************************** | |
2103 | * _objc_fixup_selector_refs. Register all of the selectors in each | |
2104 | * image, and fix them all up. | |
2105 | **********************************************************************/ | |
2106 | static void _objc_fixup_selector_refs (const header_info *hi) | |
2107 | { | |
2108 | size_t count; | |
2109 | SEL *sels; | |
2110 | ||
2111 | bool preoptimized = hi->isPreoptimized(); | |
2112 | ||
2113 | if (PrintPreopt) { | |
2114 | if (preoptimized) { | |
2115 | _objc_inform("PREOPTIMIZATION: honoring preoptimized selectors in %s", | |
2116 | hi->fname()); | |
2117 | } | |
2118 | else if (hi->info()->optimizedByDyld()) { | |
2119 | _objc_inform("PREOPTIMIZATION: IGNORING preoptimized selectors in %s", | |
2120 | hi->fname()); | |
2121 | } | |
2122 | } | |
2123 | ||
2124 | if (preoptimized) return; | |
2125 | ||
2126 | sels = _getObjcSelectorRefs (hi, &count); | |
2127 | ||
2128 | map_selrefs(sels, count, headerIsBundle(hi)); | |
2129 | } | |
2130 | ||
2131 | static inline bool _is_threaded() { | |
2132 | #if TARGET_OS_WIN32 | |
2133 | return YES; | |
2134 | #else | |
2135 | return pthread_is_threaded_np() != 0; | |
2136 | #endif | |
2137 | } | |
2138 | ||
2139 | #if !TARGET_OS_WIN32 | |
2140 | /*********************************************************************** | |
2141 | * unmap_image | |
2142 | * Process the given image which is about to be unmapped by dyld. | |
2143 | * mh is mach_header instead of headerType because that's what | |
2144 | * dyld_priv.h says even for 64-bit. | |
2145 | **********************************************************************/ | |
2146 | void | |
2147 | unmap_image(const char *path __unused, const struct mach_header *mh) | |
2148 | { | |
2149 | recursive_mutex_locker_t lock(loadMethodLock); | |
2150 | unmap_image_nolock(mh); | |
2151 | } | |
2152 | ||
2153 | ||
2154 | /*********************************************************************** | |
2155 | * map_images | |
2156 | * Process the given images which are being mapped in by dyld. | |
2157 | * Calls ABI-agnostic code after taking ABI-specific locks. | |
2158 | **********************************************************************/ | |
2159 | void | |
2160 | map_images(unsigned count, const char * const paths[], | |
2161 | const struct mach_header * const mhdrs[]) | |
2162 | { | |
2163 | recursive_mutex_locker_t lock(loadMethodLock); | |
2164 | map_images_nolock(count, paths, mhdrs); | |
2165 | } | |
2166 | ||
2167 | ||
2168 | /*********************************************************************** | |
2169 | * load_images | |
2170 | * Process +load in the given images which are being mapped in by dyld. | |
2171 | * | |
2172 | * Locking: acquires classLock and loadMethodLock | |
2173 | **********************************************************************/ | |
2174 | extern void prepare_load_methods(const headerType *mhdr); | |
2175 | ||
2176 | void | |
2177 | load_images(const char *path __unused, const struct mach_header *mh) | |
2178 | { | |
2179 | recursive_mutex_locker_t lock(loadMethodLock); | |
2180 | ||
2181 | // Discover +load methods | |
2182 | prepare_load_methods((const headerType *)mh); | |
2183 | ||
2184 | // Call +load methods (without classLock - re-entrant) | |
2185 | call_load_methods(); | |
2186 | } | |
2187 | #endif | |
2188 | ||
2189 | ||
2190 | /*********************************************************************** | |
2191 | * _read_images | |
2192 | * Perform metadata processing for hCount images starting with firstNewHeader | |
2193 | **********************************************************************/ | |
2194 | void _read_images(header_info **hList, uint32_t hCount, int totalClasses, int unoptimizedTotalClass) | |
2195 | { | |
2196 | uint32_t i; | |
2197 | bool categoriesLoaded = NO; | |
2198 | ||
2199 | if (!class_hash) _objc_init_class_hash(); | |
2200 | ||
2201 | // Parts of this order are important for correctness or performance. | |
2202 | ||
2203 | // Read classes from all images. | |
2204 | for (i = 0; i < hCount; i++) { | |
2205 | _objc_read_classes_from_image(hList[i]); | |
2206 | } | |
2207 | ||
2208 | // Read categories from all images. | |
2209 | // But not if any other threads are running - they might | |
2210 | // call a category method before the fixups below are complete. | |
2211 | if (!_is_threaded()) { | |
2212 | bool needFlush = NO; | |
2213 | for (i = 0; i < hCount; i++) { | |
2214 | needFlush |= _objc_read_categories_from_image(hList[i]); | |
2215 | } | |
2216 | if (needFlush) flush_marked_caches(); | |
2217 | categoriesLoaded = YES; | |
2218 | } | |
2219 | ||
2220 | // Connect classes from all images. | |
2221 | for (i = 0; i < hCount; i++) { | |
2222 | _objc_connect_classes_from_image(hList[i]); | |
2223 | } | |
2224 | ||
2225 | // Fix up class refs, selector refs, and protocol objects from all images. | |
2226 | for (i = 0; i < hCount; i++) { | |
2227 | _objc_map_class_refs_for_image(hList[i]); | |
2228 | _objc_fixup_selector_refs(hList[i]); | |
2229 | _objc_fixup_protocol_objects_for_image(hList[i]); | |
2230 | } | |
2231 | ||
2232 | // Read categories from all images. | |
2233 | // But not if this is the only thread - it's more | |
2234 | // efficient to attach categories earlier if safe. | |
2235 | if (!categoriesLoaded) { | |
2236 | bool needFlush = NO; | |
2237 | for (i = 0; i < hCount; i++) { | |
2238 | needFlush |= _objc_read_categories_from_image(hList[i]); | |
2239 | } | |
2240 | if (needFlush) flush_marked_caches(); | |
2241 | } | |
2242 | ||
2243 | // Multi-threaded category load MUST BE LAST to avoid a race. | |
2244 | } | |
2245 | ||
2246 | ||
2247 | /*********************************************************************** | |
2248 | * prepare_load_methods | |
2249 | * Schedule +load for classes in this image, any un-+load-ed | |
2250 | * superclasses in other images, and any categories in this image. | |
2251 | **********************************************************************/ | |
2252 | // Recursively schedule +load for cls and any un-+load-ed superclasses. | |
2253 | // cls must already be connected. | |
2254 | static void schedule_class_load(Class cls) | |
2255 | { | |
2256 | if (cls->info & CLS_LOADED) return; | |
2257 | if (cls->superclass) schedule_class_load(cls->superclass); | |
2258 | add_class_to_loadable_list(cls); | |
2259 | cls->info |= CLS_LOADED; | |
2260 | } | |
2261 | ||
2262 | void prepare_load_methods(const headerType *mhdr) | |
2263 | { | |
2264 | Module mods; | |
2265 | unsigned int midx; | |
2266 | ||
2267 | header_info *hi; | |
2268 | for (hi = FirstHeader; hi; hi = hi->getNext()) { | |
2269 | if (mhdr == hi->mhdr()) break; | |
2270 | } | |
2271 | if (!hi) return; | |
2272 | ||
2273 | if (hi->info()->isReplacement()) { | |
2274 | // Ignore any classes in this image | |
2275 | return; | |
2276 | } | |
2277 | ||
2278 | // Major loop - process all modules in the image | |
2279 | mods = hi->mod_ptr; | |
2280 | for (midx = 0; midx < hi->mod_count; midx += 1) | |
2281 | { | |
2282 | unsigned int index; | |
2283 | ||
2284 | // Skip module containing no classes | |
2285 | if (mods[midx].symtab == nil) | |
2286 | continue; | |
2287 | ||
2288 | // Minor loop - process all the classes in given module | |
2289 | for (index = 0; index < mods[midx].symtab->cls_def_cnt; index += 1) | |
2290 | { | |
2291 | // Locate the class description pointer | |
2292 | Class cls = (Class)mods[midx].symtab->defs[index]; | |
2293 | if (cls->info & CLS_CONNECTED) { | |
2294 | schedule_class_load(cls); | |
2295 | } | |
2296 | } | |
2297 | } | |
2298 | ||
2299 | ||
2300 | // Major loop - process all modules in the header | |
2301 | mods = hi->mod_ptr; | |
2302 | ||
2303 | // NOTE: The module and category lists are traversed backwards | |
2304 | // to preserve the pre-10.4 processing order. Changing the order | |
2305 | // would have a small chance of introducing binary compatibility bugs. | |
2306 | midx = (unsigned int)hi->mod_count; | |
2307 | while (midx-- > 0) { | |
2308 | unsigned int index; | |
2309 | unsigned int total; | |
2310 | Symtab symtab = mods[midx].symtab; | |
2311 | ||
2312 | // Nothing to do for a module without a symbol table | |
2313 | if (mods[midx].symtab == nil) | |
2314 | continue; | |
2315 | // Total entries in symbol table (class entries followed | |
2316 | // by category entries) | |
2317 | total = mods[midx].symtab->cls_def_cnt + | |
2318 | mods[midx].symtab->cat_def_cnt; | |
2319 | ||
2320 | // Minor loop - register all categories from given module | |
2321 | index = total; | |
2322 | while (index-- > mods[midx].symtab->cls_def_cnt) { | |
2323 | old_category *cat = (old_category *)symtab->defs[index]; | |
2324 | add_category_to_loadable_list((Category)cat); | |
2325 | } | |
2326 | } | |
2327 | } | |
2328 | ||
2329 | ||
2330 | #if TARGET_OS_WIN32 | |
2331 | ||
2332 | void unload_class(Class cls) | |
2333 | { | |
2334 | } | |
2335 | ||
2336 | #else | |
2337 | ||
2338 | /*********************************************************************** | |
2339 | * _objc_remove_classes_in_image | |
2340 | * Remove all classes in the given image from the runtime, because | |
2341 | * the image is about to be unloaded. | |
2342 | * Things to clean up: | |
2343 | * class_hash | |
2344 | * unconnected_class_hash | |
2345 | * pending subclasses list (only if class is still unconnected) | |
2346 | * loadable class list | |
2347 | * class's method caches | |
2348 | * class refs in all other images | |
2349 | **********************************************************************/ | |
2350 | // Re-pend any class references in refs that point into [start..end) | |
2351 | static void rependClassReferences(Class *refs, size_t count, | |
2352 | uintptr_t start, uintptr_t end) | |
2353 | { | |
2354 | size_t i; | |
2355 | ||
2356 | if (!refs) return; | |
2357 | ||
2358 | // Process each class ref | |
2359 | for (i = 0; i < count; i++) { | |
2360 | if ((uintptr_t)(refs[i]) >= start && (uintptr_t)(refs[i]) < end) { | |
2361 | pendClassReference(&refs[i], refs[i]->name, | |
2362 | refs[i]->info & CLS_META); | |
2363 | refs[i] = nil; | |
2364 | } | |
2365 | } | |
2366 | } | |
2367 | ||
2368 | ||
2369 | void try_free(const void *p) | |
2370 | { | |
2371 | if (p && malloc_size(p)) free((void *)p); | |
2372 | } | |
2373 | ||
2374 | // Deallocate all memory in a method list | |
2375 | static void unload_mlist(old_method_list *mlist) | |
2376 | { | |
2377 | int i; | |
2378 | for (i = 0; i < mlist->method_count; i++) { | |
2379 | try_free(mlist->method_list[i].method_types); | |
2380 | } | |
2381 | try_free(mlist); | |
2382 | } | |
2383 | ||
2384 | static void unload_property_list(old_property_list *proplist) | |
2385 | { | |
2386 | uint32_t i; | |
2387 | ||
2388 | if (!proplist) return; | |
2389 | ||
2390 | for (i = 0; i < proplist->count; i++) { | |
2391 | old_property *prop = property_list_nth(proplist, i); | |
2392 | try_free(prop->name); | |
2393 | try_free(prop->attributes); | |
2394 | } | |
2395 | try_free(proplist); | |
2396 | } | |
2397 | ||
2398 | ||
2399 | // Deallocate all memory in a class. | |
2400 | void unload_class(Class cls) | |
2401 | { | |
2402 | // Free method cache | |
2403 | // This dereferences the cache contents; do this before freeing methods | |
2404 | if (cls->cache && cls->cache != &_objc_empty_cache) { | |
2405 | _cache_free(cls->cache); | |
2406 | } | |
2407 | ||
2408 | // Free ivar lists | |
2409 | if (cls->ivars) { | |
2410 | int i; | |
2411 | for (i = 0; i < cls->ivars->ivar_count; i++) { | |
2412 | try_free(cls->ivars->ivar_list[i].ivar_name); | |
2413 | try_free(cls->ivars->ivar_list[i].ivar_type); | |
2414 | } | |
2415 | try_free(cls->ivars); | |
2416 | } | |
2417 | ||
2418 | // Free fixed-up method lists and method list array | |
2419 | if (cls->methodLists) { | |
2420 | // more than zero method lists | |
2421 | if (cls->info & CLS_NO_METHOD_ARRAY) { | |
2422 | // one method list | |
2423 | unload_mlist((old_method_list *)cls->methodLists); | |
2424 | } | |
2425 | else { | |
2426 | // more than one method list | |
2427 | old_method_list **mlistp; | |
2428 | for (mlistp = cls->methodLists; | |
2429 | *mlistp != nil && *mlistp != END_OF_METHODS_LIST; | |
2430 | mlistp++) | |
2431 | { | |
2432 | unload_mlist(*mlistp); | |
2433 | } | |
2434 | free(cls->methodLists); | |
2435 | } | |
2436 | } | |
2437 | ||
2438 | // Free protocol list | |
2439 | old_protocol_list *protos = cls->protocols; | |
2440 | while (protos) { | |
2441 | old_protocol_list *dead = protos; | |
2442 | protos = protos->next; | |
2443 | try_free(dead); | |
2444 | } | |
2445 | ||
2446 | if ((cls->info & CLS_EXT)) { | |
2447 | if (cls->ext) { | |
2448 | // Free property lists and property list array | |
2449 | if (cls->ext->propertyLists) { | |
2450 | // more than zero property lists | |
2451 | if (cls->info & CLS_NO_PROPERTY_ARRAY) { | |
2452 | // one property list | |
2453 | old_property_list *proplist = | |
2454 | (old_property_list *)cls->ext->propertyLists; | |
2455 | unload_property_list(proplist); | |
2456 | } else { | |
2457 | // more than one property list | |
2458 | old_property_list **plistp; | |
2459 | for (plistp = cls->ext->propertyLists; | |
2460 | *plistp != nil; | |
2461 | plistp++) | |
2462 | { | |
2463 | unload_property_list(*plistp); | |
2464 | } | |
2465 | try_free(cls->ext->propertyLists); | |
2466 | } | |
2467 | } | |
2468 | ||
2469 | // Free weak ivar layout | |
2470 | try_free(cls->ext->weak_ivar_layout); | |
2471 | ||
2472 | // Free ext | |
2473 | try_free(cls->ext); | |
2474 | } | |
2475 | ||
2476 | // Free non-weak ivar layout | |
2477 | try_free(cls->ivar_layout); | |
2478 | } | |
2479 | ||
2480 | // Free class name | |
2481 | try_free(cls->name); | |
2482 | ||
2483 | // Free cls | |
2484 | try_free(cls); | |
2485 | } | |
2486 | ||
2487 | ||
2488 | static void _objc_remove_classes_in_image(header_info *hi) | |
2489 | { | |
2490 | unsigned int index; | |
2491 | unsigned int midx; | |
2492 | Module mods; | |
2493 | ||
2494 | mutex_locker_t lock(classLock); | |
2495 | ||
2496 | // Major loop - process all modules in the image | |
2497 | mods = hi->mod_ptr; | |
2498 | for (midx = 0; midx < hi->mod_count; midx += 1) | |
2499 | { | |
2500 | // Skip module containing no classes | |
2501 | if (mods[midx].symtab == nil) | |
2502 | continue; | |
2503 | ||
2504 | // Minor loop - process all the classes in given module | |
2505 | for (index = 0; index < mods[midx].symtab->cls_def_cnt; index += 1) | |
2506 | { | |
2507 | Class cls; | |
2508 | ||
2509 | // Locate the class description pointer | |
2510 | cls = (Class)mods[midx].symtab->defs[index]; | |
2511 | ||
2512 | // Remove from loadable class list, if present | |
2513 | remove_class_from_loadable_list(cls); | |
2514 | ||
2515 | // Remove from unconnected_class_hash and pending subclasses | |
2516 | if (unconnected_class_hash && NXHashMember(unconnected_class_hash, cls)) { | |
2517 | NXHashRemove(unconnected_class_hash, cls); | |
2518 | if (pendingSubclassesMap) { | |
2519 | // Find this class in its superclass's pending list | |
2520 | char *supercls_name = (char *)cls->superclass; | |
2521 | PendingSubclass *pending = (PendingSubclass *) | |
2522 | NXMapGet(pendingSubclassesMap, supercls_name); | |
2523 | for ( ; pending != nil; pending = pending->next) { | |
2524 | if (pending->subclass == cls) { | |
2525 | pending->subclass = Nil; | |
2526 | break; | |
2527 | } | |
2528 | } | |
2529 | } | |
2530 | } | |
2531 | ||
2532 | // Remove from class_hash | |
2533 | NXHashRemove(class_hash, cls); | |
2534 | ||
2535 | // Free heap memory pointed to by the class | |
2536 | unload_class(cls->ISA()); | |
2537 | unload_class(cls); | |
2538 | } | |
2539 | } | |
2540 | ||
2541 | ||
2542 | // Search all other images for class refs that point back to this range. | |
2543 | // Un-fix and re-pend any such class refs. | |
2544 | ||
2545 | // Get the location of the dying image's __OBJC segment | |
2546 | uintptr_t seg; | |
2547 | unsigned long seg_size; | |
2548 | seg = (uintptr_t)getsegmentdata(hi->mhdr(), "__OBJC", &seg_size); | |
2549 | ||
2550 | header_info *other_hi; | |
2551 | for (other_hi = FirstHeader; other_hi != nil; other_hi = other_hi->getNext()) { | |
2552 | Class *other_refs; | |
2553 | size_t count; | |
2554 | if (other_hi == hi) continue; // skip the image being unloaded | |
2555 | ||
2556 | // Fix class refs in the other image | |
2557 | other_refs = _getObjcClassRefs(other_hi, &count); | |
2558 | rependClassReferences(other_refs, count, seg, seg+seg_size); | |
2559 | } | |
2560 | } | |
2561 | ||
2562 | ||
2563 | /*********************************************************************** | |
2564 | * _objc_remove_categories_in_image | |
2565 | * Remove all categories in the given image from the runtime, because | |
2566 | * the image is about to be unloaded. | |
2567 | * Things to clean up: | |
2568 | * unresolved category list | |
2569 | * loadable category list | |
2570 | **********************************************************************/ | |
2571 | static void _objc_remove_categories_in_image(header_info *hi) | |
2572 | { | |
2573 | Module mods; | |
2574 | unsigned int midx; | |
2575 | ||
2576 | // Major loop - process all modules in the header | |
2577 | mods = hi->mod_ptr; | |
2578 | ||
2579 | for (midx = 0; midx < hi->mod_count; midx++) { | |
2580 | unsigned int index; | |
2581 | unsigned int total; | |
2582 | Symtab symtab = mods[midx].symtab; | |
2583 | ||
2584 | // Nothing to do for a module without a symbol table | |
2585 | if (symtab == nil) continue; | |
2586 | ||
2587 | // Total entries in symbol table (class entries followed | |
2588 | // by category entries) | |
2589 | total = symtab->cls_def_cnt + symtab->cat_def_cnt; | |
2590 | ||
2591 | // Minor loop - check all categories from given module | |
2592 | for (index = symtab->cls_def_cnt; index < total; index++) { | |
2593 | old_category *cat = (old_category *)symtab->defs[index]; | |
2594 | ||
2595 | // Clean up loadable category list | |
2596 | remove_category_from_loadable_list((Category)cat); | |
2597 | ||
2598 | // Clean up category_hash | |
2599 | if (category_hash) { | |
2600 | _objc_unresolved_category *cat_entry = (_objc_unresolved_category *)NXMapGet(category_hash, cat->class_name); | |
2601 | for ( ; cat_entry != nil; cat_entry = cat_entry->next) { | |
2602 | if (cat_entry->cat == cat) { | |
2603 | cat_entry->cat = nil; | |
2604 | break; | |
2605 | } | |
2606 | } | |
2607 | } | |
2608 | } | |
2609 | } | |
2610 | } | |
2611 | ||
2612 | ||
2613 | /*********************************************************************** | |
2614 | * unload_paranoia | |
2615 | * Various paranoid debugging checks that look for poorly-behaving | |
2616 | * unloadable bundles. | |
2617 | * Called by _objc_unmap_image when OBJC_UNLOAD_DEBUG is set. | |
2618 | **********************************************************************/ | |
2619 | static void unload_paranoia(header_info *hi) | |
2620 | { | |
2621 | // Get the location of the dying image's __OBJC segment | |
2622 | uintptr_t seg; | |
2623 | unsigned long seg_size; | |
2624 | seg = (uintptr_t)getsegmentdata(hi->mhdr(), "__OBJC", &seg_size); | |
2625 | ||
2626 | _objc_inform("UNLOAD DEBUG: unloading image '%s' [%p..%p]", | |
2627 | hi->fname(), (void *)seg, (void*)(seg+seg_size)); | |
2628 | ||
2629 | mutex_locker_t lock(classLock); | |
2630 | ||
2631 | // Make sure the image contains no categories on surviving classes. | |
2632 | { | |
2633 | Module mods; | |
2634 | unsigned int midx; | |
2635 | ||
2636 | // Major loop - process all modules in the header | |
2637 | mods = hi->mod_ptr; | |
2638 | ||
2639 | for (midx = 0; midx < hi->mod_count; midx++) { | |
2640 | unsigned int index; | |
2641 | unsigned int total; | |
2642 | Symtab symtab = mods[midx].symtab; | |
2643 | ||
2644 | // Nothing to do for a module without a symbol table | |
2645 | if (symtab == nil) continue; | |
2646 | ||
2647 | // Total entries in symbol table (class entries followed | |
2648 | // by category entries) | |
2649 | total = symtab->cls_def_cnt + symtab->cat_def_cnt; | |
2650 | ||
2651 | // Minor loop - check all categories from given module | |
2652 | for (index = symtab->cls_def_cnt; index < total; index++) { | |
2653 | old_category *cat = (old_category *)symtab->defs[index]; | |
2654 | struct objc_class query; | |
2655 | ||
2656 | query.name = cat->class_name; | |
2657 | if (NXHashMember(class_hash, &query)) { | |
2658 | _objc_inform("UNLOAD DEBUG: dying image contains category '%s(%s)' on surviving class '%s'!", cat->class_name, cat->category_name, cat->class_name); | |
2659 | } | |
2660 | } | |
2661 | } | |
2662 | } | |
2663 | ||
2664 | // Make sure no surviving class is in the dying image. | |
2665 | // Make sure no surviving class has a superclass in the dying image. | |
2666 | // fixme check method implementations too | |
2667 | { | |
2668 | Class cls; | |
2669 | NXHashState state; | |
2670 | ||
2671 | state = NXInitHashState(class_hash); | |
2672 | while (NXNextHashState(class_hash, &state, (void **)&cls)) { | |
2673 | if ((vm_address_t)cls >= seg && | |
2674 | (vm_address_t)cls < seg+seg_size) | |
2675 | { | |
2676 | _objc_inform("UNLOAD DEBUG: dying image contains surviving class '%s'!", cls->name); | |
2677 | } | |
2678 | ||
2679 | if ((vm_address_t)cls->superclass >= seg && | |
2680 | (vm_address_t)cls->superclass < seg+seg_size) | |
2681 | { | |
2682 | _objc_inform("UNLOAD DEBUG: dying image contains superclass '%s' of surviving class '%s'!", cls->superclass->name, cls->name); | |
2683 | } | |
2684 | } | |
2685 | } | |
2686 | } | |
2687 | ||
2688 | ||
2689 | /*********************************************************************** | |
2690 | * _unload_image | |
2691 | * Only handles MH_BUNDLE for now. | |
2692 | * Locking: loadMethodLock acquired by unmap_image | |
2693 | **********************************************************************/ | |
2694 | void _unload_image(header_info *hi) | |
2695 | { | |
2696 | loadMethodLock.assertLocked(); | |
2697 | ||
2698 | // Cleanup: | |
2699 | // Remove image's classes from the class list and free auxiliary data. | |
2700 | // Remove image's unresolved or loadable categories and free auxiliary data | |
2701 | // Remove image's unresolved class refs. | |
2702 | _objc_remove_classes_in_image(hi); | |
2703 | _objc_remove_categories_in_image(hi); | |
2704 | _objc_remove_pending_class_refs_in_image(hi); | |
2705 | if (hi->proto_refs) try_free(hi->proto_refs); | |
2706 | ||
2707 | // Perform various debugging checks if requested. | |
2708 | if (DebugUnload) unload_paranoia(hi); | |
2709 | } | |
2710 | ||
2711 | #endif | |
2712 | ||
2713 | ||
2714 | /*********************************************************************** | |
2715 | * objc_addClass. Add the specified class to the table of known classes, | |
2716 | * after doing a little verification and fixup. | |
2717 | **********************************************************************/ | |
2718 | void objc_addClass (Class cls) | |
2719 | { | |
2720 | OBJC_WARN_DEPRECATED; | |
2721 | ||
2722 | // Synchronize access to hash table | |
2723 | mutex_locker_t lock(classLock); | |
2724 | ||
2725 | // Make sure both the class and the metaclass have caches! | |
2726 | // Clear all bits of the info fields except CLS_CLASS and CLS_META. | |
2727 | // Normally these bits are already clear but if someone tries to cons | |
2728 | // up their own class on the fly they might need to be cleared. | |
2729 | if (cls->cache == nil) { | |
2730 | cls->cache = (Cache) &_objc_empty_cache; | |
2731 | cls->info = CLS_CLASS; | |
2732 | } | |
2733 | ||
2734 | if (cls->ISA()->cache == nil) { | |
2735 | cls->ISA()->cache = (Cache) &_objc_empty_cache; | |
2736 | cls->ISA()->info = CLS_META; | |
2737 | } | |
2738 | ||
2739 | // methodLists should be: | |
2740 | // 1. nil (Tiger and later only) | |
2741 | // 2. A -1 terminated method list array | |
2742 | // In either case, CLS_NO_METHOD_ARRAY remains clear. | |
2743 | // If the user manipulates the method list directly, | |
2744 | // they must use the magic private format. | |
2745 | ||
2746 | // Add the class to the table | |
2747 | (void) NXHashInsert (class_hash, cls); | |
2748 | ||
2749 | // Superclass is no longer a leaf for cache flushing | |
2750 | if (cls->superclass && (cls->superclass->info & CLS_LEAF)) { | |
2751 | cls->superclass->clearInfo(CLS_LEAF); | |
2752 | cls->superclass->ISA()->clearInfo(CLS_LEAF); | |
2753 | } | |
2754 | } | |
2755 | ||
2756 | /*********************************************************************** | |
2757 | * _objcTweakMethodListPointerForClass. | |
2758 | * Change the class's method list pointer to a method list array. | |
2759 | * Does nothing if the method list pointer is already a method list array. | |
2760 | * If the class is currently in use, methodListLock must be held by the caller. | |
2761 | **********************************************************************/ | |
2762 | static void _objcTweakMethodListPointerForClass(Class cls) | |
2763 | { | |
2764 | old_method_list * originalList; | |
2765 | const int initialEntries = 4; | |
2766 | size_t mallocSize; | |
2767 | old_method_list ** ptr; | |
2768 | ||
2769 | // Do nothing if methodLists is already an array. | |
2770 | if (cls->methodLists && !(cls->info & CLS_NO_METHOD_ARRAY)) return; | |
2771 | ||
2772 | // Remember existing list | |
2773 | originalList = (old_method_list *) cls->methodLists; | |
2774 | ||
2775 | // Allocate and zero a method list array | |
2776 | mallocSize = sizeof(old_method_list *) * initialEntries; | |
2777 | ptr = (old_method_list **) calloc(1, mallocSize); | |
2778 | ||
2779 | // Insert the existing list into the array | |
2780 | ptr[initialEntries - 1] = END_OF_METHODS_LIST; | |
2781 | ptr[0] = originalList; | |
2782 | ||
2783 | // Replace existing list with array | |
2784 | cls->methodLists = ptr; | |
2785 | cls->clearInfo(CLS_NO_METHOD_ARRAY); | |
2786 | } | |
2787 | ||
2788 | ||
2789 | /*********************************************************************** | |
2790 | * _objc_insertMethods. | |
2791 | * Adds methods to a class. | |
2792 | * Does not flush any method caches. | |
2793 | * Does not take any locks. | |
2794 | * If the class is already in use, use class_addMethods() instead. | |
2795 | **********************************************************************/ | |
2796 | void _objc_insertMethods(Class cls, old_method_list *mlist, old_category *cat) | |
2797 | { | |
2798 | old_method_list ***list; | |
2799 | old_method_list **ptr; | |
2800 | ptrdiff_t endIndex; | |
2801 | size_t oldSize; | |
2802 | size_t newSize; | |
2803 | ||
2804 | if (!cls->methodLists) { | |
2805 | // cls has no methods - simply use this method list | |
2806 | cls->methodLists = (old_method_list **)mlist; | |
2807 | cls->setInfo(CLS_NO_METHOD_ARRAY); | |
2808 | return; | |
2809 | } | |
2810 | ||
2811 | // Log any existing methods being replaced | |
2812 | if (PrintReplacedMethods) { | |
2813 | int i; | |
2814 | for (i = 0; i < mlist->method_count; i++) { | |
2815 | extern IMP findIMPInClass(Class cls, SEL sel); | |
2816 | SEL sel = sel_registerName((char *)mlist->method_list[i].method_name); | |
2817 | IMP newImp = mlist->method_list[i].method_imp; | |
2818 | IMP oldImp; | |
2819 | ||
2820 | if ((oldImp = findIMPInClass(cls, sel))) { | |
2821 | logReplacedMethod(cls->name, sel, ISMETA(cls), | |
2822 | cat ? cat->category_name : nil, | |
2823 | oldImp, newImp); | |
2824 | } | |
2825 | } | |
2826 | } | |
2827 | ||
2828 | // Create method list array if necessary | |
2829 | _objcTweakMethodListPointerForClass(cls); | |
2830 | ||
2831 | list = &cls->methodLists; | |
2832 | ||
2833 | // Locate unused entry for insertion point | |
2834 | ptr = *list; | |
2835 | while ((*ptr != 0) && (*ptr != END_OF_METHODS_LIST)) | |
2836 | ptr += 1; | |
2837 | ||
2838 | // If array is full, add to it | |
2839 | if (*ptr == END_OF_METHODS_LIST) | |
2840 | { | |
2841 | // Calculate old and new dimensions | |
2842 | endIndex = ptr - *list; | |
2843 | oldSize = (endIndex + 1) * sizeof(void *); | |
2844 | newSize = oldSize + sizeof(old_method_list *); // only increase by 1 | |
2845 | ||
2846 | // Grow the method list array by one. | |
2847 | *list = (old_method_list **)realloc(*list, newSize); | |
2848 | ||
2849 | // Zero out addition part of new array | |
2850 | bzero (&((*list)[endIndex]), newSize - oldSize); | |
2851 | ||
2852 | // Place new end marker | |
2853 | (*list)[(newSize/sizeof(void *)) - 1] = END_OF_METHODS_LIST; | |
2854 | ||
2855 | // Insertion point corresponds to old array end | |
2856 | ptr = &((*list)[endIndex]); | |
2857 | } | |
2858 | ||
2859 | // Right shift existing entries by one | |
2860 | bcopy (*list, (*list) + 1, (uint8_t *)ptr - (uint8_t *)*list); | |
2861 | ||
2862 | // Insert at method list at beginning of array | |
2863 | **list = mlist; | |
2864 | } | |
2865 | ||
2866 | /*********************************************************************** | |
2867 | * _objc_removeMethods. | |
2868 | * Remove methods from a class. | |
2869 | * Does not take any locks. | |
2870 | * Does not flush any method caches. | |
2871 | * If the class is currently in use, use class_removeMethods() instead. | |
2872 | **********************************************************************/ | |
2873 | void _objc_removeMethods(Class cls, old_method_list *mlist) | |
2874 | { | |
2875 | old_method_list ***list; | |
2876 | old_method_list **ptr; | |
2877 | ||
2878 | if (cls->methodLists == nil) { | |
2879 | // cls has no methods | |
2880 | return; | |
2881 | } | |
2882 | if (cls->methodLists == (old_method_list **)mlist) { | |
2883 | // mlist is the class's only method list - erase it | |
2884 | cls->methodLists = nil; | |
2885 | return; | |
2886 | } | |
2887 | if (cls->info & CLS_NO_METHOD_ARRAY) { | |
2888 | // cls has only one method list, and this isn't it - do nothing | |
2889 | return; | |
2890 | } | |
2891 | ||
2892 | // cls has a method list array - search it | |
2893 | ||
2894 | list = &cls->methodLists; | |
2895 | ||
2896 | // Locate list in the array | |
2897 | ptr = *list; | |
2898 | while (*ptr != mlist) { | |
2899 | // fix for radar # 2538790 | |
2900 | if ( *ptr == END_OF_METHODS_LIST ) return; | |
2901 | ptr += 1; | |
2902 | } | |
2903 | ||
2904 | // Remove this entry | |
2905 | *ptr = 0; | |
2906 | ||
2907 | // Left shift the following entries | |
2908 | while (*(++ptr) != END_OF_METHODS_LIST) | |
2909 | *(ptr-1) = *ptr; | |
2910 | *(ptr-1) = 0; | |
2911 | } | |
2912 | ||
2913 | /*********************************************************************** | |
2914 | * _objc_add_category. Install the specified category's methods and | |
2915 | * protocols into the class it augments. | |
2916 | * The class is assumed not to be in use yet: no locks are taken and | |
2917 | * no method caches are flushed. | |
2918 | **********************************************************************/ | |
2919 | static inline void _objc_add_category(Class cls, old_category *category, int version) | |
2920 | { | |
2921 | if (PrintConnecting) { | |
2922 | _objc_inform("CONNECT: attaching category '%s (%s)'", cls->name, category->category_name); | |
2923 | } | |
2924 | ||
2925 | // Augment instance methods | |
2926 | if (category->instance_methods) | |
2927 | _objc_insertMethods (cls, category->instance_methods, category); | |
2928 | ||
2929 | // Augment class methods | |
2930 | if (category->class_methods) | |
2931 | _objc_insertMethods (cls->ISA(), category->class_methods, category); | |
2932 | ||
2933 | // Augment protocols | |
2934 | if ((version >= 5) && category->protocols) | |
2935 | { | |
2936 | if (cls->ISA()->version >= 5) | |
2937 | { | |
2938 | category->protocols->next = cls->protocols; | |
2939 | cls->protocols = category->protocols; | |
2940 | cls->ISA()->protocols = category->protocols; | |
2941 | } | |
2942 | else | |
2943 | { | |
2944 | _objc_inform ("unable to add protocols from category %s...\n", category->category_name); | |
2945 | _objc_inform ("class `%s' must be recompiled\n", category->class_name); | |
2946 | } | |
2947 | } | |
2948 | ||
2949 | // Augment instance properties | |
2950 | if (version >= 7 && category->instance_properties) { | |
2951 | if (cls->ISA()->version >= 6) { | |
2952 | _class_addProperties(cls, category->instance_properties); | |
2953 | } else { | |
2954 | _objc_inform ("unable to add instance properties from category %s...\n", category->category_name); | |
2955 | _objc_inform ("class `%s' must be recompiled\n", category->class_name); | |
2956 | } | |
2957 | } | |
2958 | ||
2959 | // Augment class properties | |
2960 | if (version >= 7 && category->hasClassPropertiesField() && | |
2961 | category->class_properties) | |
2962 | { | |
2963 | if (cls->ISA()->version >= 6) { | |
2964 | _class_addProperties(cls->ISA(), category->class_properties); | |
2965 | } else { | |
2966 | _objc_inform ("unable to add class properties from category %s...\n", category->category_name); | |
2967 | _objc_inform ("class `%s' must be recompiled\n", category->class_name); | |
2968 | } | |
2969 | } | |
2970 | } | |
2971 | ||
2972 | /*********************************************************************** | |
2973 | * _objc_add_category_flush_caches. Install the specified category's | |
2974 | * methods into the class it augments, and flush the class' method cache. | |
2975 | * Return YES if some method caches now need to be flushed. | |
2976 | **********************************************************************/ | |
2977 | static bool _objc_add_category_flush_caches(Class cls, old_category *category, int version) | |
2978 | { | |
2979 | bool needFlush = NO; | |
2980 | ||
2981 | // Install the category's methods into its intended class | |
2982 | { | |
2983 | mutex_locker_t lock(methodListLock); | |
2984 | _objc_add_category (cls, category, version); | |
2985 | } | |
2986 | ||
2987 | // Queue for cache flushing so category's methods can get called | |
2988 | if (category->instance_methods) { | |
2989 | cls->setInfo(CLS_FLUSH_CACHE); | |
2990 | needFlush = YES; | |
2991 | } | |
2992 | if (category->class_methods) { | |
2993 | cls->ISA()->setInfo(CLS_FLUSH_CACHE); | |
2994 | needFlush = YES; | |
2995 | } | |
2996 | ||
2997 | return needFlush; | |
2998 | } | |
2999 | ||
3000 | ||
3001 | /*********************************************************************** | |
3002 | * reverse_cat | |
3003 | * Reverse the given linked list of pending categories. | |
3004 | * The pending category list is built backwards, and needs to be | |
3005 | * reversed before actually attaching the categories to a class. | |
3006 | * Returns the head of the new linked list. | |
3007 | **********************************************************************/ | |
3008 | static _objc_unresolved_category *reverse_cat(_objc_unresolved_category *cat) | |
3009 | { | |
3010 | _objc_unresolved_category *prev; | |
3011 | _objc_unresolved_category *cur; | |
3012 | _objc_unresolved_category *ahead; | |
3013 | ||
3014 | if (!cat) return nil; | |
3015 | ||
3016 | prev = nil; | |
3017 | cur = cat; | |
3018 | ahead = cat->next; | |
3019 | ||
3020 | while (cur) { | |
3021 | ahead = cur->next; | |
3022 | cur->next = prev; | |
3023 | prev = cur; | |
3024 | cur = ahead; | |
3025 | } | |
3026 | ||
3027 | return prev; | |
3028 | } | |
3029 | ||
3030 | ||
3031 | /*********************************************************************** | |
3032 | * resolve_categories_for_class. | |
3033 | * Install all existing categories intended for the specified class. | |
3034 | * cls must be a true class and not a metaclass. | |
3035 | **********************************************************************/ | |
3036 | static void resolve_categories_for_class(Class cls) | |
3037 | { | |
3038 | _objc_unresolved_category * pending; | |
3039 | _objc_unresolved_category * next; | |
3040 | ||
3041 | // Nothing to do if there are no categories at all | |
3042 | if (!category_hash) return; | |
3043 | ||
3044 | // Locate and remove first element in category list | |
3045 | // associated with this class | |
3046 | pending = (_objc_unresolved_category *) | |
3047 | NXMapKeyFreeingRemove (category_hash, cls->name); | |
3048 | ||
3049 | // Traverse the list of categories, if any, registered for this class | |
3050 | ||
3051 | // The pending list is built backwards. Reverse it and walk forwards. | |
3052 | pending = reverse_cat(pending); | |
3053 | ||
3054 | while (pending) { | |
3055 | if (pending->cat) { | |
3056 | // Install the category | |
3057 | // use the non-flush-cache version since we are only | |
3058 | // called from the class intialization code | |
3059 | _objc_add_category(cls, pending->cat, (int)pending->version); | |
3060 | } | |
3061 | ||
3062 | // Delink and reclaim this registration | |
3063 | next = pending->next; | |
3064 | free(pending); | |
3065 | pending = next; | |
3066 | } | |
3067 | } | |
3068 | ||
3069 | ||
3070 | /*********************************************************************** | |
3071 | * _objc_resolve_categories_for_class. | |
3072 | * Public version of resolve_categories_for_class. This was | |
3073 | * exported pre-10.4 for Omni et al. to workaround a problem | |
3074 | * with too-lazy category attachment. | |
3075 | * cls should be a class, but this function can also cope with metaclasses. | |
3076 | **********************************************************************/ | |
3077 | void _objc_resolve_categories_for_class(Class cls) | |
3078 | { | |
3079 | // If cls is a metaclass, get the class. | |
3080 | // resolve_categories_for_class() requires a real class to work correctly. | |
3081 | if (ISMETA(cls)) { | |
3082 | if (strncmp(cls->name, "_%", 2) == 0) { | |
3083 | // Posee's meta's name is smashed and isn't in the class_hash, | |
3084 | // so objc_getClass doesn't work. | |
3085 | const char *baseName = strchr(cls->name, '%'); // get posee's real name | |
3086 | cls = objc_getClass(baseName); | |
3087 | } else { | |
3088 | cls = objc_getClass(cls->name); | |
3089 | } | |
3090 | } | |
3091 | ||
3092 | resolve_categories_for_class(cls); | |
3093 | } | |
3094 | ||
3095 | ||
3096 | /*********************************************************************** | |
3097 | * _objc_register_category. | |
3098 | * Process a category read from an image. | |
3099 | * If the category's class exists, attach the category immediately. | |
3100 | * Classes that need cache flushing are marked but not flushed. | |
3101 | * If the category's class does not exist yet, pend the category for | |
3102 | * later attachment. Pending categories are attached in the order | |
3103 | * they were discovered. | |
3104 | * Returns YES if some method caches now need to be flushed. | |
3105 | **********************************************************************/ | |
3106 | static bool _objc_register_category(old_category *cat, int version) | |
3107 | { | |
3108 | _objc_unresolved_category * new_cat; | |
3109 | _objc_unresolved_category * old; | |
3110 | Class theClass; | |
3111 | ||
3112 | // If the category's class exists, attach the category. | |
3113 | if ((theClass = objc_lookUpClass(cat->class_name))) { | |
3114 | return _objc_add_category_flush_caches(theClass, cat, version); | |
3115 | } | |
3116 | ||
3117 | // If the category's class exists but is unconnected, | |
3118 | // then attach the category to the class but don't bother | |
3119 | // flushing any method caches (because they must be empty). | |
3120 | // YES unconnected, NO class_handler | |
3121 | if ((theClass = look_up_class(cat->class_name, YES, NO))) { | |
3122 | _objc_add_category(theClass, cat, version); | |
3123 | return NO; | |
3124 | } | |
3125 | ||
3126 | ||
3127 | // Category's class does not exist yet. | |
3128 | // Save the category for later attachment. | |
3129 | ||
3130 | if (PrintConnecting) { | |
3131 | _objc_inform("CONNECT: pending category '%s (%s)'", cat->class_name, cat->category_name); | |
3132 | } | |
3133 | ||
3134 | // Create category lookup table if needed | |
3135 | if (!category_hash) | |
3136 | category_hash = NXCreateMapTable(NXStrValueMapPrototype, 128); | |
3137 | ||
3138 | // Locate an existing list of categories, if any, for the class. | |
3139 | old = (_objc_unresolved_category *) | |
3140 | NXMapGet (category_hash, cat->class_name); | |
3141 | ||
3142 | // Register the category to be fixed up later. | |
3143 | // The category list is built backwards, and is reversed again | |
3144 | // by resolve_categories_for_class(). | |
3145 | new_cat = (_objc_unresolved_category *) | |
3146 | malloc(sizeof(_objc_unresolved_category)); | |
3147 | new_cat->next = old; | |
3148 | new_cat->cat = cat; | |
3149 | new_cat->version = version; | |
3150 | (void) NXMapKeyCopyingInsert (category_hash, cat->class_name, new_cat); | |
3151 | ||
3152 | return NO; | |
3153 | } | |
3154 | ||
3155 | ||
3156 | const char ** | |
3157 | _objc_copyClassNamesForImage(header_info *hi, unsigned int *outCount) | |
3158 | { | |
3159 | Module mods; | |
3160 | unsigned int m; | |
3161 | const char **list; | |
3162 | int count; | |
3163 | int allocated; | |
3164 | ||
3165 | list = nil; | |
3166 | count = 0; | |
3167 | allocated = 0; | |
3168 | ||
3169 | mods = hi->mod_ptr; | |
3170 | for (m = 0; m < hi->mod_count; m++) { | |
3171 | int d; | |
3172 | ||
3173 | if (!mods[m].symtab) continue; | |
3174 | ||
3175 | for (d = 0; d < mods[m].symtab->cls_def_cnt; d++) { | |
3176 | Class cls = (Class)mods[m].symtab->defs[d]; | |
3177 | // fixme what about future-ified classes? | |
3178 | if (cls->isConnected()) { | |
3179 | if (count == allocated) { | |
3180 | allocated = allocated*2 + 16; | |
3181 | list = (const char **) | |
3182 | realloc((void *)list, allocated * sizeof(char *)); | |
3183 | } | |
3184 | list[count++] = cls->name; | |
3185 | } | |
3186 | } | |
3187 | } | |
3188 | ||
3189 | if (count > 0) { | |
3190 | // nil-terminate non-empty list | |
3191 | if (count == allocated) { | |
3192 | allocated = allocated+1; | |
3193 | list = (const char **) | |
3194 | realloc((void *)list, allocated * sizeof(char *)); | |
3195 | } | |
3196 | list[count] = nil; | |
3197 | } | |
3198 | ||
3199 | if (outCount) *outCount = count; | |
3200 | return list; | |
3201 | } | |
3202 | ||
3203 | Class gdb_class_getClass(Class cls) | |
3204 | { | |
3205 | const char *className = cls->name; | |
3206 | if(!className || !strlen(className)) return Nil; | |
3207 | Class rCls = look_up_class(className, NO, NO); | |
3208 | return rCls; | |
3209 | ||
3210 | } | |
3211 | ||
3212 | Class gdb_object_getClass(id obj) | |
3213 | { | |
3214 | if (!obj) return nil; | |
3215 | return gdb_class_getClass(obj->getIsa()); | |
3216 | } | |
3217 | ||
3218 | ||
3219 | /*********************************************************************** | |
3220 | * Lock management | |
3221 | **********************************************************************/ | |
3222 | rwlock_t selLock; | |
3223 | mutex_t classLock; | |
3224 | mutex_t methodListLock; | |
3225 | mutex_t cacheUpdateLock; | |
3226 | recursive_mutex_t loadMethodLock; | |
3227 | ||
3228 | void lock_init(void) | |
3229 | { | |
3230 | } | |
3231 | ||
3232 | ||
3233 | #endif |