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