OSDN Git Service

2002-07-02 Rodney Brown <rbrown64@csc.com.au>
[pf3gnuchains/gcc-fork.git] / libobjc / selector.c
1 /* GNU Objective C Runtime selector related functions
2    Copyright (C) 1993, 1995, 1996, 1997, 2002 Free Software Foundation, Inc.
3    Contributed by Kresten Krab Thorup
4
5 This file is part of GNU CC.
6
7 GNU CC is free software; you can redistribute it and/or modify it under the
8 terms of the GNU General Public License as published by the Free Software
9 Foundation; either version 2, or (at your option) any later version.
10
11 GNU CC is distributed in the hope that it will be useful, but WITHOUT ANY
12 WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
13 FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
14 details.
15
16 You should have received a copy of the GNU General Public License along with
17 GNU CC; see the file COPYING.  If not, write to the Free Software
18 Foundation, 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.  */
19
20 /* As a special exception, if you link this library with files compiled with
21    GCC to produce an executable, this does not cause the resulting executable
22    to be covered by the GNU General Public License. This exception does not
23    however invalidate any other reasons why the executable file might be
24    covered by the GNU General Public License.  */
25
26 #include "runtime.h"
27 #include "sarray.h"
28 #include "encoding.h"
29
30 /* Initial selector hash table size. Value doesn't matter much */
31 #define SELECTOR_HASH_SIZE 128
32
33 /* Tables mapping selector names to uid and opposite */
34 static struct sarray *__objc_selector_array = 0; /* uid -> sel  !T:MUTEX */
35 static struct sarray *__objc_selector_names = 0; /* uid -> name !T:MUTEX */
36 static cache_ptr      __objc_selector_hash  = 0; /* name -> uid !T:MUTEX */
37
38 static void register_selectors_from_list (MethodList_t);
39
40 /* Number of selectors stored in each of the above tables */
41 unsigned int __objc_selector_max_index = 0;     /* !T:MUTEX */
42
43 void __objc_init_selector_tables ()
44 {
45   __objc_selector_array = sarray_new (SELECTOR_HASH_SIZE, 0);
46   __objc_selector_names = sarray_new (SELECTOR_HASH_SIZE, 0);
47   __objc_selector_hash
48     = hash_new (SELECTOR_HASH_SIZE,
49                 (hash_func_type) hash_string,
50                 (compare_func_type) compare_strings);
51 }  
52
53 /* This routine is given a class and records all of the methods in its class
54    structure in the record table.  */
55 void
56 __objc_register_selectors_from_class (Class class)
57 {
58   MethodList_t method_list;
59
60   method_list = class->methods;
61   while (method_list)
62     {
63       register_selectors_from_list (method_list);
64       method_list = method_list->method_next;
65     }
66 }
67
68
69 /* This routine is given a list of methods and records each of the methods in
70    the record table.  This is the routine that does the actual recording
71    work.
72
73    This one is only called for Class objects.  For categories,
74    class_add_method_list is called.
75    */
76 static void
77 register_selectors_from_list (MethodList_t method_list)
78 {
79   int i = 0;
80   while (i < method_list->method_count)
81     {
82       Method_t method = &method_list->method_list[i];
83       method->method_name 
84         = sel_register_typed_name ((const char *) method->method_name, 
85                                    method->method_types);
86       i += 1;
87     }
88 }
89
90
91 /* Register instance methods as class methods for root classes */
92 void __objc_register_instance_methods_to_class (Class class)
93 {
94   MethodList_t method_list;
95   MethodList_t class_method_list;
96   int max_methods_no = 16;
97   MethodList_t new_list;
98   Method_t curr_method;
99
100   /* Only if a root class. */
101   if (class->super_class)
102     return;
103
104   /* Allocate a method list to hold the new class methods */
105   new_list = objc_calloc (sizeof (struct objc_method_list)
106                             + sizeof (struct objc_method[max_methods_no]), 1);
107   method_list = class->methods;
108   class_method_list = class->class_pointer->methods;
109   curr_method = &new_list->method_list[0];
110
111   /* Iterate through the method lists for the class */
112   while (method_list)
113     {
114       int i;
115
116       /* Iterate through the methods from this method list */
117       for (i = 0; i < method_list->method_count; i++)
118         {
119           Method_t mth = &method_list->method_list[i];
120           if (mth->method_name
121               && ! search_for_method_in_list (class_method_list,
122                                               mth->method_name))
123             {
124               /* This instance method isn't a class method. 
125                   Add it into the new_list. */
126               *curr_method = *mth;
127   
128               /* Reallocate the method list if necessary */
129               if (++new_list->method_count == max_methods_no)
130                 new_list =
131                   objc_realloc (new_list, sizeof (struct objc_method_list)
132                                 + sizeof (struct 
133                                         objc_method[max_methods_no += 16]));
134               curr_method = &new_list->method_list[new_list->method_count];
135             }
136         }
137
138       method_list = method_list->method_next;
139     }
140
141   /* If we created any new class methods
142      then attach the method list to the class */
143   if (new_list->method_count)
144     {
145       new_list =
146         objc_realloc (new_list, sizeof (struct objc_method_list)
147                      + sizeof (struct objc_method[new_list->method_count]));
148       new_list->method_next = class->class_pointer->methods;
149       class->class_pointer->methods = new_list;
150     }
151
152     __objc_update_dispatch_table_for_class (class->class_pointer);
153 }
154
155
156 /* Returns YES iff t1 and t2 have same method types, but we ignore
157    the argframe layout */
158 BOOL
159 sel_types_match (const char *t1, const char *t2)
160 {
161   if (! t1 || ! t2)
162     return NO;
163   while (*t1 && *t2)
164     {
165       if (*t1 == '+') t1++;
166       if (*t2 == '+') t2++;
167       while (isdigit ((unsigned char) *t1)) t1++;
168       while (isdigit ((unsigned char) *t2)) t2++;
169       /* xxx Remove these next two lines when qualifiers are put in
170          all selectors, not just Protocol selectors. */
171       t1 = objc_skip_type_qualifiers (t1);
172       t2 = objc_skip_type_qualifiers (t2);
173       if (! *t1 && ! *t2)
174         return YES;
175       if (*t1 != *t2)
176         return NO;
177       t1++;
178       t2++;
179     }
180   return NO;
181 }
182
183 /* return selector representing name */
184 SEL
185 sel_get_typed_uid (const char *name, const char *types)
186 {
187   struct objc_list *l;
188   sidx i;
189
190   objc_mutex_lock (__objc_runtime_mutex);
191
192   i = (sidx) hash_value_for_key (__objc_selector_hash, name);
193   if (i == 0)
194     {
195       objc_mutex_unlock (__objc_runtime_mutex);
196       return 0;
197     }
198
199   for (l = (struct objc_list *) sarray_get_safe (__objc_selector_array, i);
200        l; l = l->tail)
201     {
202       SEL s = (SEL) l->head;
203       if (types == 0 || s->sel_types == 0)
204         {
205           if (s->sel_types == types)
206             {
207               objc_mutex_unlock (__objc_runtime_mutex);
208               return s;
209             }
210         }
211       else if (sel_types_match (s->sel_types, types))
212         {
213           objc_mutex_unlock (__objc_runtime_mutex);
214           return s;
215         }
216     }
217
218   objc_mutex_unlock (__objc_runtime_mutex);
219   return 0;
220 }
221
222 /* Return selector representing name; prefer a selector with non-NULL type */
223 SEL
224 sel_get_any_typed_uid (const char *name)
225 {
226   struct objc_list *l;
227   sidx i;
228   SEL s = NULL;
229
230   objc_mutex_lock (__objc_runtime_mutex);
231
232   i = (sidx) hash_value_for_key (__objc_selector_hash, name);
233   if (i == 0)
234     {
235       objc_mutex_unlock (__objc_runtime_mutex);
236       return 0;
237     }
238
239   for (l = (struct objc_list *) sarray_get_safe (__objc_selector_array, i);
240        l; l = l->tail)
241     {
242       s = (SEL) l->head;
243       if (s->sel_types)
244         {
245             objc_mutex_unlock (__objc_runtime_mutex);
246             return s;
247         }
248     }
249
250   objc_mutex_unlock (__objc_runtime_mutex);
251   return s;
252 }
253
254 /* return selector representing name */
255 SEL
256 sel_get_any_uid (const char *name)
257 {
258   struct objc_list *l;
259   sidx i;
260
261   objc_mutex_lock (__objc_runtime_mutex);
262
263   i = (sidx) hash_value_for_key (__objc_selector_hash, name);
264   if (soffset_decode (i) == 0)
265     {
266       objc_mutex_unlock (__objc_runtime_mutex);
267       return 0;
268     }
269
270   l = (struct objc_list *) sarray_get_safe (__objc_selector_array, i);
271   objc_mutex_unlock (__objc_runtime_mutex);
272
273   if (l == 0)
274     return 0;
275
276   return (SEL) l->head;
277 }
278
279 /* return selector representing name */
280 SEL
281 sel_get_uid (const char *name)
282 {
283   return sel_register_typed_name (name, 0);
284 }
285
286 /* Get name of selector.  If selector is unknown, the empty string "" 
287    is returned */ 
288 const char *sel_get_name (SEL selector)
289 {
290   const char *ret;
291
292   objc_mutex_lock (__objc_runtime_mutex);
293   if ((soffset_decode ((sidx)selector->sel_id) > 0)
294       && (soffset_decode ((sidx)selector->sel_id) <= __objc_selector_max_index))
295     ret = sarray_get_safe (__objc_selector_names, (sidx) selector->sel_id);
296   else
297     ret = 0;
298   objc_mutex_unlock (__objc_runtime_mutex);
299   return ret;
300 }
301
302 BOOL
303 sel_is_mapped (SEL selector)
304 {
305   unsigned int idx = soffset_decode ((sidx)selector->sel_id);
306   return ((idx > 0) && (idx <= __objc_selector_max_index));
307 }
308
309
310 const char *sel_get_type (SEL selector)
311 {
312   if (selector)
313     return selector->sel_types;
314   else
315     return 0;
316 }
317
318 /* The uninstalled dispatch table */
319 extern struct sarray *__objc_uninstalled_dtable;
320
321 /* Store the passed selector name in the selector record and return its
322    selector value (value returned by sel_get_uid).
323    Assumes that the calling function has locked down __objc_runtime_mutex. */
324 /* is_const parameter tells us if the name and types parameters
325    are really constant or not.  If YES then they are constant and
326    we can just store the pointers.  If NO then we need to copy
327    name and types because the pointers may disappear later on. */
328 SEL
329 __sel_register_typed_name (const char *name, const char *types, 
330                            struct objc_selector *orig, BOOL is_const)
331 {
332   struct objc_selector *j;
333   sidx i;
334   struct objc_list *l;
335
336   i = (sidx) hash_value_for_key (__objc_selector_hash, name);
337   if (soffset_decode (i) != 0)
338     {
339       for (l = (struct objc_list *) sarray_get_safe (__objc_selector_array, i);
340            l; l = l->tail)
341         {
342           SEL s = (SEL) l->head;
343           if (types == 0 || s->sel_types == 0)
344             {
345               if (s->sel_types == types)
346                 {
347                   if (orig)
348                     {
349                       orig->sel_id = (void *) i;
350                       return orig;
351                     }
352                   else
353                     return s;
354                 }
355             }
356           else if (! strcmp (s->sel_types, types))
357             {
358               if (orig)
359                 {
360                   orig->sel_id = (void *) i;
361                   return orig;
362                 }
363               else
364                 return s;
365             }
366         }
367       if (orig)
368         j = orig;
369       else
370         j = objc_malloc (sizeof (struct objc_selector));
371
372       j->sel_id = (void *) i;
373       /* Can we use the pointer or must copy types?  Don't copy if NULL */
374       if ((is_const) || (types == 0))
375         j->sel_types = (const char *) types;
376       else {
377         j->sel_types = (char *) objc_malloc (strlen (types) + 1);
378         strcpy ((char *) j->sel_types, types);
379       }
380       l = (struct objc_list *) sarray_get_safe (__objc_selector_array, i);
381     }
382   else
383     {
384       __objc_selector_max_index += 1;
385       i = soffset_encode (__objc_selector_max_index);
386       if (orig)
387         j = orig;
388       else
389         j = objc_malloc (sizeof (struct objc_selector));
390         
391       j->sel_id = (void *) i;
392       /* Can we use the pointer or must copy types?  Don't copy if NULL */
393       if ((is_const) || (types == 0))
394         j->sel_types = (const char *) types;
395       else {
396         j->sel_types = (char *) objc_malloc (strlen (types) + 1);
397         strcpy ((char *) j->sel_types, types);
398       }
399       l = 0;
400     }
401
402   DEBUG_PRINTF ("Record selector %s[%s] as: %ld\n", name, types, 
403                 soffset_decode (i));
404   
405   {
406     int is_new = (l == 0);
407     const char *new_name;
408
409     /* Can we use the pointer or must copy name?  Don't copy if NULL */
410     if ((is_const) || (name == 0))
411       new_name = name;
412     else {
413       new_name = (char *) objc_malloc (strlen (name) + 1);
414       strcpy ((char *) new_name, name);
415     }
416
417     l = list_cons ((void *) j, l);
418     sarray_at_put_safe (__objc_selector_names, i, (void *) new_name);
419     sarray_at_put_safe (__objc_selector_array, i, (void *) l);
420     if (is_new)
421       hash_add (&__objc_selector_hash, (void *) new_name, (void *) i);
422   }
423
424   sarray_realloc (__objc_uninstalled_dtable, __objc_selector_max_index + 1);
425
426   return (SEL) j;
427 }
428
429 SEL
430 sel_register_name (const char *name)
431 {
432   SEL ret;
433     
434   objc_mutex_lock (__objc_runtime_mutex);
435   /* Assume that name is not constant static memory and needs to be
436      copied before put into a runtime structure.  is_const == NO */
437   ret = __sel_register_typed_name (name, 0, 0, NO);
438   objc_mutex_unlock (__objc_runtime_mutex);
439   
440   return ret;
441 }
442
443 SEL
444 sel_register_typed_name (const char *name, const char *type)
445 {
446   SEL ret;
447     
448   objc_mutex_lock (__objc_runtime_mutex);
449   /* Assume that name and type are not constant static memory and need to
450      be copied before put into a runtime structure.  is_const == NO */
451   ret = __sel_register_typed_name (name, type, 0, NO);
452   objc_mutex_unlock (__objc_runtime_mutex);
453   
454   return ret;
455 }