added more inner function checks
[swftools.git] / lib / as3 / registry.c
1 /* registry.c
2
3    Routines for compiling Flash2 AVM2 ABC Actionscript
4
5    Extension module for the rfxswf library.
6    Part of the swftools package.
7
8    Copyright (c) 2008 Matthias Kramm <kramm@quiss.org>
9  
10    This program is free software; you can redistribute it and/or modify
11    it under the terms of the GNU General Public License as published by
12    the Free Software Foundation; either version 2 of the License, or
13    (at your option) any later version.
14
15    This program is distributed in the hope that it will be useful,
16    but WITHOUT ANY WARRANTY; without even the implied warranty of
17    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18    GNU General Public License for more details.
19
20    You should have received a copy of the GNU General Public License
21    along with this program; if not, write to the Free Software
22    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
23
24 #include <assert.h>
25 #include "pool.h"
26 #include "registry.h"
27 #include "builtin.h"
28
29 dict_t*registry_classes=0;
30
31 // ----------------------- class signature ------------------------------
32
33 char slotinfo_equals(slotinfo_t*c1, slotinfo_t*c2)
34 {
35     if(!!c1 != !!c2)
36         return 0;
37     /* notice: access right is *not* respected */
38     if(!strcmp(c1->name, c2->name) &&
39        !strcmp(c1->package, c2->package)) {
40         return 1;
41     }
42     return 0;
43 }
44 static unsigned int slotinfo_hash(slotinfo_t*c)
45 {
46     unsigned int hash = 0;
47     hash = crc32_add_string(hash, c->package);
48     hash = crc32_add_string(hash, c->name);
49     return hash;
50 }
51
52 static void* dummy_clone(void*other) {return other;}
53 static void dummy_destroy(slotinfo_t*c) {}
54
55 type_t slotinfo_type = {
56     hash: (hash_func)slotinfo_hash,
57     equals: (equals_func)slotinfo_equals,
58     dup: (dup_func)dummy_clone, // all signatures are static
59     free: (free_func)dummy_destroy,
60 };
61
62 // ------------------------- constructors --------------------------------
63
64 #define AVERAGE_NUMBER_OF_MEMBERS 8
65 classinfo_t* classinfo_register(int access, const char*package, const char*name, int num_interfaces)
66 {
67     classinfo_t*c = rfx_calloc(sizeof(classinfo_t)+(sizeof(classinfo_t*)*(num_interfaces+1)));
68     c->interfaces[0] = 0;
69     c->kind = INFOTYPE_CLASS;
70     c->access = access;
71     c->package = package;
72     c->name = name;
73     dict_put(registry_classes, c, c);
74     dict_init2(&c->members, &slotinfo_type, AVERAGE_NUMBER_OF_MEMBERS);
75     return c;
76 }
77 methodinfo_t* methodinfo_register_onclass(classinfo_t*cls, U8 access, const char*ns, const char*name)
78 {
79     NEW(methodinfo_t,m);
80     m->kind = INFOTYPE_METHOD;
81     m->access = access;
82     m->name = name;
83     m->package = ns;
84     m->parent = cls;
85     dict_put(&cls->members, m, m);
86     return m;
87 }
88 varinfo_t* varinfo_register_onclass(classinfo_t*cls, U8 access, const char*ns, const char*name)
89 {
90     NEW(varinfo_t,m);
91     m->kind = INFOTYPE_SLOT;
92     m->access = access;
93     m->name = name;
94     m->package = ns;
95     m->parent = cls;
96     dict_put(&cls->members, m, m);
97     return m;
98 }
99 methodinfo_t* methodinfo_register_global(U8 access, const char*package, const char*name)
100 {
101     NEW(methodinfo_t, m);
102     m->kind = INFOTYPE_METHOD;
103     m->flags = FLAG_STATIC;
104     m->access = access;
105     m->package = package;
106     m->name = name;
107     m->parent = 0;
108     dict_put(registry_classes, m, m);
109     return m;
110 }
111 varinfo_t* varinfo_register_global(U8 access, const char*package, const char*name)
112 {
113     NEW(varinfo_t, m);
114     m->kind = INFOTYPE_SLOT;
115     m->flags = FLAG_STATIC;
116     m->access = access;
117     m->package = package;
118     m->name = name;
119     m->parent = 0;
120     dict_put(registry_classes, m, m);
121     return m;
122 }
123
124 // --------------- builtin classes (from builtin.c) ----------------------
125
126 void registry_init()
127 {
128     if(!registry_classes)
129         registry_classes = builtin_getclasses();
130 }
131 slotinfo_t* registry_find(const char*package, const char*name)
132 {
133     assert(registry_classes);
134     slotinfo_t tmp;
135     tmp.package = package;
136     tmp.name = name;
137     slotinfo_t* c = (slotinfo_t*)dict_lookup(registry_classes, &tmp);
138     /*if(c)
139         printf("%s.%s->%08x (%s.%s)\n", package, name, c, c->package, c->name);*/
140     return c;
141 }
142 slotinfo_t* registry_safefind(const char*package, const char*name)
143 {
144     slotinfo_t*c = registry_find(package, name);
145     if(!c) {
146         printf("%s.%s\n", package, name);
147     }
148     assert(c);
149     return c;
150 }
151 void registry_dump()
152 {
153     int t;
154     for(t=0;t<registry_classes->hashsize;t++) {
155         dictentry_t*e = registry_classes->slots[t];
156         while(e) {
157             slotinfo_t*i = (slotinfo_t*)e->key;
158             printf("[%s] %s.%s\n", access2str(i->access), i->package, i->name);
159             e = e->next;
160         }
161     }
162 }
163
164 memberinfo_t* registry_findmember(classinfo_t*cls, const char*ns, const char*name, char recursive)
165 {
166     memberinfo_t tmp;
167     tmp.name = name;
168     tmp.package = ns?ns:"";
169
170     if(!recursive) {
171         return (memberinfo_t*)dict_lookup(&cls->members, &tmp);
172     }
173     /* look at classes directly extended by this class */
174     slotinfo_t*m = 0;
175     classinfo_t*s = cls;
176
177     if(recursive>1) // check *only* superclasses
178         s = s->superclass;
179
180     while(s) {
181         m = (slotinfo_t*)dict_lookup(&s->members, &tmp);
182         if(m) {
183             return (memberinfo_t*)m;
184         }
185         s = s->superclass;
186     }
187     /* look at interfaces, and parent interfaces */
188     int t=0;
189     while(cls->interfaces[t]) {
190         classinfo_t*s = cls->interfaces[t];
191         while(s) {
192             m = (slotinfo_t*)dict_lookup(&s->members, &tmp);
193             if(m) {
194                 return (memberinfo_t*)m;
195             }
196             s = s->superclass;
197         }
198         t++;
199     }
200     return 0;
201 }
202
203 memberinfo_t* registry_findmember_nsset(classinfo_t*cls, namespace_list_t*ns, const char*name, char superclasses)
204 {
205     while(ns) {
206         memberinfo_t*m = registry_findmember(cls, ns->namespace->name, name, superclasses);
207         if(m) return m;
208         ns = ns->next;
209     }
210     return registry_findmember(cls, "", name, superclasses);
211 }
212
213
214 void registry_fill_multiname(multiname_t*m, namespace_t*n, slotinfo_t*c)
215 {
216     m->type = QNAME;
217     m->ns = n;
218     m->ns->access = c->access;
219     m->ns->name = (char*)c->package;
220     m->name = c->name;
221     m->namespace_set = 0;
222 }
223 multiname_t* classinfo_to_multiname(slotinfo_t*cls)
224 {
225     if(!cls)
226         return 0;
227     multiname_t*m=0;
228     namespace_t ns = {cls->access, (char*)cls->package};
229     return multiname_new(&ns,cls->name);
230 }
231
232 // ----------------------- memberinfo methods ------------------------------
233
234 /* hacky code to wrap a variable or function into a "type"
235    object, but keep a pointer to the "value" */
236 static dict_t* functionobjects = 0;
237 classinfo_t* slotinfo_asclass(slotinfo_t*f) {
238     if(!functionobjects) {
239         functionobjects = dict_new2(&ptr_type);
240     } else {
241         classinfo_t*c = dict_lookup(functionobjects, f);
242         if(c)
243             return c;
244     }
245
246     classinfo_t*c = rfx_calloc(sizeof(classinfo_t)+sizeof(classinfo_t*));
247     c->access = ACCESS_PUBLIC;
248     c->package = "";
249     if(f->kind == INFOTYPE_METHOD)
250         c->name = "Function";
251     else if(f->kind == INFOTYPE_CLASS)
252         c->name = "Class";
253     else if(f->kind == INFOTYPE_SLOT)
254         c->name = "Object";
255     else {
256         c->name = "undefined";
257     }
258     
259     dict_init2(&c->members, &slotinfo_type, 1);
260     c->data = f;
261     dict_put(functionobjects, f, c);
262     return c;
263 }
264
265 classinfo_t* slotinfo_gettype(slotinfo_t*f)
266 {
267     if(f) {
268        if(f->kind == INFOTYPE_METHOD) {
269            return slotinfo_asclass(f);
270        } else if(f->kind == INFOTYPE_SLOT) {
271            varinfo_t*v = (varinfo_t*)f;
272            return v->type;
273        } else 
274            return 0;
275     } else {
276        return registry_getanytype();
277     }
278 }
279 // ----------------------- builtin types ------------------------------
280 classinfo_t* registry_getanytype() {return 0;}
281
282 char registry_isfunctionclass(classinfo_t*c) {
283     return (c && c->package && c->name && 
284             !strcmp(c->package, "") && !strcmp(c->name, "Function"));
285 }
286 char registry_isclassclass(classinfo_t*c) {
287     return (c && c->package && c->name && 
288             !strcmp(c->package, "") && !strcmp(c->name, "Class"));
289 }
290
291 classinfo_t* registry_getobjectclass() {
292     static classinfo_t*c = 0;
293     if(!c) c = (classinfo_t*)registry_safefind("", "Object");
294     return c;
295 }
296 classinfo_t* registry_getstringclass() {
297     static classinfo_t*c = 0;
298     if(!c) c = (classinfo_t*)registry_safefind("", "String");
299     return c;
300 }
301 classinfo_t* registry_getarrayclass() {
302     static classinfo_t*c = 0;
303     if(!c) c = (classinfo_t*)registry_safefind("", "Array");
304     return c;
305 }
306 classinfo_t* registry_getintclass() {
307     static classinfo_t*c = 0;
308     if(!c) c = (classinfo_t*)registry_safefind("", "int");
309     return c;
310 }
311 classinfo_t* registry_getuintclass() {
312     static classinfo_t*c = 0;
313     if(!c) c = (classinfo_t*)registry_safefind("", "uint");
314     return c;
315 }
316 classinfo_t* registry_getbooleanclass() {
317     static classinfo_t*c = 0;
318     if(!c) c = (classinfo_t*)registry_safefind("", "Boolean");
319     return c;
320 }
321 classinfo_t* registry_getnumberclass() {
322     static classinfo_t*c = 0;
323     if(!c) c = (classinfo_t*)registry_safefind("", "Number");
324     return c;
325 }
326 classinfo_t* registry_getregexpclass() {
327     static classinfo_t*c = 0;
328     if(!c) c = (classinfo_t*)registry_safefind("", "RegExp");
329     return c;
330 }
331 classinfo_t* registry_getMovieClip() {
332     static classinfo_t*c = 0;
333     if(!c) c = (classinfo_t*)registry_safefind("flash.display", "MovieClip");
334     return c;
335 }
336
337 // ----------------------- builtin dummy types -------------------------
338 classinfo_t nullclass = {
339     INFOTYPE_CLASS,0,0,ACCESS_PACKAGE, "", "null", 0, 0, 0
340 };
341 classinfo_t* registry_getnullclass() {
342     return &nullclass;
343 }
344
345 namespace_t access2namespace(U8 access, char*package)
346 {
347     namespace_t ns;
348     ns.access = access;
349     ns.name = package;
350     return ns;
351 }
352
353 char* infotypename(slotinfo_t*s)
354 {
355     if(s->kind == INFOTYPE_CLASS) return "class";
356     else if(s->kind == INFOTYPE_SLOT) return "member";
357     else if(s->kind == INFOTYPE_METHOD) return "method";
358     else return "object";
359 }
360