OSDN Git Service

13a2afd181275af6fb0416267746eae60130b0e8
[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*
289 sel_get_name (SEL selector)
290 {
291   const char *ret;
292
293   objc_mutex_lock(__objc_runtime_mutex);
294   if ((soffset_decode((sidx)selector->sel_id) > 0)
295       && (soffset_decode((sidx)selector->sel_id) <= __objc_selector_max_index))
296     ret = sarray_get_safe (__objc_selector_names, (sidx) selector->sel_id);
297   else
298     ret = 0;
299   objc_mutex_unlock(__objc_runtime_mutex);
300   return ret;
301 }
302
303 BOOL
304 sel_is_mapped (SEL selector)
305 {
306   unsigned int idx = soffset_decode ((sidx)selector->sel_id);
307   return ((idx > 0) && (idx <= __objc_selector_max_index));
308 }
309
310
311 const char*
312 sel_get_type (SEL selector)
313 {
314   if (selector)
315     return selector->sel_types;
316   else
317     return 0;
318 }
319
320 /* The uninstalled dispatch table */
321 extern struct sarray* __objc_uninstalled_dtable;
322
323 /* Store the passed selector name in the selector record and return its
324    selector value (value returned by sel_get_uid).
325    Assumes that the calling function has locked down __objc_runtime_mutex. */
326 /* is_const parameter tells us if the name and types parameters
327    are really constant or not.  If YES then they are constant and
328    we can just store the pointers.  If NO then we need to copy
329    name and types because the pointers may disappear later on. */
330 SEL
331 __sel_register_typed_name (const char *name, const char *types, 
332                            struct objc_selector *orig, BOOL is_const)
333 {
334   struct objc_selector* j;
335   sidx i;
336   struct objc_list *l;
337
338   i = (sidx) hash_value_for_key (__objc_selector_hash, name);
339   if (soffset_decode (i) != 0)
340     {
341       for (l = (struct objc_list*)sarray_get_safe (__objc_selector_array, i);
342            l; l = l->tail)
343         {
344           SEL s = (SEL)l->head;
345           if (types == 0 || s->sel_types == 0)
346             {
347               if (s->sel_types == types)
348                 {
349                   if (orig)
350                     {
351                       orig->sel_id = (void*)i;
352                       return orig;
353                     }
354                   else
355                     return s;
356                 }
357             }
358           else if (!strcmp (s->sel_types, types))
359             {
360               if (orig)
361                 {
362                   orig->sel_id = (void*)i;
363                   return orig;
364                 }
365               else
366                 return s;
367             }
368         }
369       if (orig)
370         j = orig;
371       else
372         j = objc_malloc (sizeof (struct objc_selector));
373
374       j->sel_id = (void*)i;
375       /* Can we use the pointer or must copy types?  Don't copy if NULL */
376       if ((is_const) || (types == 0))
377         j->sel_types = (const char*)types;
378       else {
379         j->sel_types = (char *) objc_malloc(strlen(types)+1);
380         strcpy((char *)j->sel_types, types);
381       }
382       l = (struct objc_list*)sarray_get_safe (__objc_selector_array, i);
383     }
384   else
385     {
386       __objc_selector_max_index += 1;
387       i = soffset_encode(__objc_selector_max_index);
388       if (orig)
389         j = orig;
390       else
391         j = objc_malloc (sizeof (struct objc_selector));
392         
393       j->sel_id = (void*)i;
394       /* Can we use the pointer or must copy types?  Don't copy if NULL */
395       if ((is_const) || (types == 0))
396         j->sel_types = (const char*)types;
397       else {
398         j->sel_types = (char *) objc_malloc(strlen(types)+1);
399         strcpy((char *)j->sel_types, types);
400       }
401       l = 0;
402     }
403
404   DEBUG_PRINTF ("Record selector %s[%s] as: %ld\n", name, types, 
405                 soffset_decode (i));
406   
407   {
408     int is_new = (l == 0);
409     const char *new_name;
410
411     /* Can we use the pointer or must copy name?  Don't copy if NULL */
412     if ((is_const) || (name == 0))
413       new_name = name;
414     else {
415       new_name = (char *) objc_malloc(strlen(name)+1);
416       strcpy((char *)new_name, name);
417     }
418
419     l = list_cons ((void*)j, l);
420     sarray_at_put_safe (__objc_selector_names, i, (void *) new_name);
421     sarray_at_put_safe (__objc_selector_array, i, (void *) l);
422     if (is_new)
423       hash_add (&__objc_selector_hash, (void *) new_name, (void *) i);
424   }
425
426   sarray_realloc(__objc_uninstalled_dtable, __objc_selector_max_index+1);
427
428   return (SEL) j;
429 }
430
431 SEL
432 sel_register_name (const char *name)
433 {
434   SEL ret;
435     
436   objc_mutex_lock(__objc_runtime_mutex);
437   /* Assume that name is not constant static memory and needs to be
438      copied before put into a runtime structure.  is_const == NO */
439   ret = __sel_register_typed_name (name, 0, 0, NO);
440   objc_mutex_unlock(__objc_runtime_mutex);
441   
442   return ret;
443 }
444
445 SEL
446 sel_register_typed_name (const char *name, const char *type)
447 {
448   SEL ret;
449     
450   objc_mutex_lock(__objc_runtime_mutex);
451   /* Assume that name and type are not constant static memory and need to
452      be copied before put into a runtime structure.  is_const == NO */
453   ret = __sel_register_typed_name (name, type, 0, NO);
454   objc_mutex_unlock(__objc_runtime_mutex);
455   
456   return ret;
457 }
458