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 static unsigned int memberinfo_hash(slotinfo_t*c)
  52 {
  53     unsigned int hash = 0;
  54     hash = crc32_add_string(hash, c->name);
  55     return hash;
  56 }
  57
  58 static void* dummy_clone(void*other) {return other;}
  59 static void dummy_destroy(slotinfo_t*c) {}
  60
  61 type_t slotinfo_type = {
  62     hash: (hash_func)slotinfo_hash,
  63     equals: (equals_func)slotinfo_equals,
  64     dup: (dup_func)dummy_clone, // all signatures are static
  65     free: (free_func)dummy_destroy,
  66 };
  67 type_t memberinfo_type = {
  68     hash: (hash_func)memberinfo_hash,
  69     equals: (equals_func)slotinfo_equals,
  70     dup: (dup_func)dummy_clone, // all signatures are static
  71     free: (free_func)dummy_destroy,
  72 };
  73
  74 // ----------------------- resolving ----------------------------------
  75 slotinfo_t* registry_resolve(slotinfo_t*_s)
  76 {
  77     if(!_s || _s->kind != INFOTYPE_UNRESOLVED)
  78         return _s;
  79     unresolvedinfo_t*s = (unresolvedinfo_t*)_s;
  80
  81     if(s->package)
  82         return registry_find(s->package, s->name);
  83
  84     namespace_list_t*l = s->nsset;
  85     while(l) {
  86         slotinfo_t* n = registry_find(l->namespace->name, s->name);
  87         if(n) return n;
  88         l = l->next;
  89     }
  90     return 0;
  91 }
  92
  93 static slotinfo_list_t*unresolved = 0;
  94 static void schedule_for_resolve(slotinfo_t*s)
  95 {
  96     list_append(unresolved, s);
  97 }
  98 static void resolve_on_slot(slotinfo_t*_member)
  99 {
 100     if(_member->kind == INFOTYPE_SLOT) {
 101         varinfo_t*member = (varinfo_t*)_member;
 102         member->type = (classinfo_t*)registry_resolve((slotinfo_t*)member->type);
 103     } else if(_member->kind == INFOTYPE_METHOD) {
 104         methodinfo_t*member = (methodinfo_t*)_member;
 105         member->return_type = (classinfo_t*)registry_resolve((slotinfo_t*)member->return_type);
 106         classinfo_list_t*l = member->params;
 107         while(l) {
 108             l->classinfo = (classinfo_t*)registry_resolve((slotinfo_t*)l->classinfo);
 109             l = l->next;
 110         }
 111     } else fprintf(stderr, "Internal Error: bad slot %s", _member->name);
 112 }
 113 static void resolve_on_class(slotinfo_t*_cls)
 114 {
 115     classinfo_t*cls = (classinfo_t*)_cls;
 116     cls->superclass = (classinfo_t*)registry_resolve((slotinfo_t*)cls->superclass);
 117     DICT_ITERATE_DATA(&cls->members,slotinfo_t*,m) {
 118         resolve_on_slot(m);
 119     }
 120     int t=0;
 121     while(cls->interfaces[t]) {
 122         cls->interfaces[t] = (classinfo_t*)registry_resolve((slotinfo_t*)cls->interfaces[t]);
 123         t++;
 124     }
 125 }
 126 void registry_resolve_all()
 127 {
 128     while(unresolved) {
 129         slotinfo_t*_s = unresolved->slotinfo;
 130         if(_s->kind == INFOTYPE_CLASS) {
 131             resolve_on_class(_s);
 132         } else if(_s->kind == INFOTYPE_METHOD || _s->kind == INFOTYPE_SLOT) {
 133             resolve_on_slot(_s);
 134         } else {
 135             fprintf(stderr, "Internal Error: object %s.%s has bad type\n", _s->package, _s->name);
 136         }
 137         slotinfo_list_t*tofree = unresolved;
 138         unresolved = unresolved->next;
 139         free(tofree);
 140     }
 141 }
 142 // ------------------------- constructors --------------------------------
 143
 144 #define AVERAGE_NUMBER_OF_MEMBERS 8
 145 classinfo_t* classinfo_register(int access, const char*package, const char*name, int num_interfaces)
 146 {
 147     classinfo_t*c = rfx_calloc(sizeof(classinfo_t)+(sizeof(classinfo_t*)*(num_interfaces+1)));
 148     c->interfaces[0] = 0;
 149     c->kind = INFOTYPE_CLASS;
 150     c->access = access;
 151     c->package = package;
 152     c->name = name;
 153     dict_put(registry_classes, c, c);
 154     dict_init2(&c->members, &memberinfo_type, AVERAGE_NUMBER_OF_MEMBERS);
 155
 156     schedule_for_resolve((slotinfo_t*)c);
 157     return c;
 158 }
 159 methodinfo_t* methodinfo_register_onclass(classinfo_t*cls, U8 access, const char*ns, const char*name)
 160 {
 161     NEW(methodinfo_t,m);
 162     m->kind = INFOTYPE_METHOD;
 163     m->access = access;
 164     m->name = name;
 165     m->package = ns;
 166     m->parent = cls;
 167     dict_put(&cls->members, m, m);
 168     return m;
 169 }
 170 varinfo_t* varinfo_register_onclass(classinfo_t*cls, U8 access, const char*ns, const char*name)
 171 {
 172     NEW(varinfo_t,m);
 173     m->kind = INFOTYPE_SLOT;
 174     m->access = access;
 175     m->name = name;
 176     m->package = ns;
 177     m->parent = cls;
 178     dict_put(&cls->members, m, m);
 179     return m;
 180 }
 181 methodinfo_t* methodinfo_register_global(U8 access, const char*package, const char*name)
 182 {
 183     NEW(methodinfo_t, m);
 184     m->kind = INFOTYPE_METHOD;
 185     m->flags = FLAG_STATIC;
 186     m->access = access;
 187     m->package = package;
 188     m->name = name;
 189     m->parent = 0;
 190     dict_put(registry_classes, m, m);
 191
 192     schedule_for_resolve((slotinfo_t*)m);
 193     return m;
 194 }
 195 varinfo_t* varinfo_register_global(U8 access, const char*package, const char*name)
 196 {
 197     NEW(varinfo_t, m);
 198     m->kind = INFOTYPE_SLOT;
 199     m->flags = FLAG_STATIC;
 200     m->access = access;
 201     m->package = package;
 202     m->name = name;
 203     m->parent = 0;
 204     dict_put(registry_classes, m, m);
 205
 206     schedule_for_resolve((slotinfo_t*)m);
 207     return m;
 208 }
 209
 210 // --------------- builtin classes (from builtin.c) ----------------------
 211
 212 void registry_init()
 213 {
 214     if(!registry_classes)
 215         registry_classes = builtin_getclasses();
 216 }
 217 slotinfo_t* registry_find(const char*package, const char*name)
 218 {
 219     assert(registry_classes);
 220     slotinfo_t tmp;
 221     tmp.package = package;
 222     tmp.name = name;
 223     slotinfo_t* c = (slotinfo_t*)dict_lookup(registry_classes, &tmp);
 224     /*if(c)
 225         printf("%s.%s->%08x (%s.%s)\n", package, name, c, c->package, c->name);*/
 226     return c;
 227 }
 228 slotinfo_t* registry_safefind(const char*package, const char*name)
 229 {
 230     slotinfo_t*c = registry_find(package, name);
 231     if(!c) {
 232         printf("%s.%s\n", package, name);
 233     }
 234     assert(c);
 235     return c;
 236 }
 237 void registry_dump()
 238 {
 239     int t;
 240     for(t=0;t<registry_classes->hashsize;t++) {
 241         dictentry_t*e = registry_classes->slots[t];
 242         while(e) {
 243             slotinfo_t*i = (slotinfo_t*)e->key;
 244             printf("[%s] %s.%s\n", access2str(i->access), i->package, i->name);
 245             e = e->next;
 246         }
 247     }
 248 }
 249
 250 memberinfo_t* registry_findmember(classinfo_t*cls, const char*ns, const char*name, char recursive)
 251 {
 252     memberinfo_t tmp;
 253     tmp.name = name;
 254     tmp.package = ns?ns:"";
 255
 256     if(!recursive) {
 257         return (memberinfo_t*)dict_lookup(&cls->members, &tmp);
 258     }
 259     /* look at classes directly extended by this class */
 260     slotinfo_t*m = 0;
 261     classinfo_t*s = cls;
 262
 263     if(recursive>1) // check *only* superclasses
 264         s = s->superclass;
 265
 266     while(s) {
 267         m = (slotinfo_t*)dict_lookup(&s->members, &tmp);
 268         if(m) {
 269             return (memberinfo_t*)m;
 270         }
 271         s = s->superclass;
 272     }
 273     /* look at interfaces, and parent interfaces */
 274     int t=0;
 275     while(cls->interfaces[t]) {
 276         classinfo_t*s = cls->interfaces[t];
 277         if(s->kind != INFOTYPE_UNRESOLVED) {
 278             while(s) {
 279                 m = (slotinfo_t*)dict_lookup(&s->members, &tmp);
 280                 if(m) {
 281                     return (memberinfo_t*)m;
 282                 }
 283                 s = s->superclass;
 284             }
 285         }
 286         t++;
 287     }
 288     return 0;
 289 }
 290
 291 memberinfo_t* registry_findmember_nsset(classinfo_t*cls, namespace_list_t*ns, const char*name, char superclasses)
 292 {
 293     memberinfo_t*m = 0;
 294     while(ns) {
 295         m = registry_findmember(cls, ns->namespace->name, name, superclasses);
 296         if(m) return m;
 297         ns = ns->next;
 298     }
 299     m = registry_findmember(cls, "", name, superclasses);
 300     if(m) return m;
 301     /* TODO: it maybe would be faster to just store the builtin namespace as "" in
 302              builtins.c */
 303     m = registry_findmember(cls, "http://adobe.com/AS3/2006/builtin", name, superclasses);
 304     if(m) return m;
 305     return 0;
 306 }
 307
 308
 309 void registry_fill_multiname(multiname_t*m, namespace_t*n, slotinfo_t*c)
 310 {
 311     m->type = QNAME;
 312     m->ns = n;
 313     m->ns->access = c->access;
 314     m->ns->name = (char*)c->package;
 315     m->name = c->name;
 316     m->namespace_set = 0;
 317 }
 318 multiname_t* classinfo_to_multiname(slotinfo_t*cls)
 319 {
 320     if(!cls)
 321         return 0;
 322     multiname_t*m=0;
 323     namespace_t ns = {cls->access, (char*)cls->package};
 324     return multiname_new(&ns,cls->name);
 325 }
 326
 327 // ----------------------- memberinfo methods ------------------------------
 328
 329 /* hacky code to wrap a variable or function into a "type"
 330    object, but keep a pointer to the "value" */
 331 static dict_t* functionobjects = 0;
 332 classinfo_t* slotinfo_asclass(slotinfo_t*f) {
 333     if(!functionobjects) {
 334         functionobjects = dict_new2(&ptr_type);
 335     } else {
 336         classinfo_t*c = dict_lookup(functionobjects, f);
 337         if(c)
 338             return c;
 339     }
 340
 341     classinfo_t*c = rfx_calloc(sizeof(classinfo_t)+sizeof(classinfo_t*));
 342     c->access = ACCESS_PUBLIC;
 343     c->package = "";
 344     if(f->kind == INFOTYPE_METHOD) {
 345         c->name = "Function";
 346         c->superclass = registry_getobjectclass();
 347     } else if(f->kind == INFOTYPE_CLASS) {
 348         c->name = "Class";
 349         c->superclass = registry_getobjectclass();
 350     } else if(f->kind == INFOTYPE_SLOT) {
 351         c->name = "Object";
 352     } else {
 353         c->name = "undefined";
 354     }
 355
 356     dict_init2(&c->members, &memberinfo_type, 1);
 357     c->data = f;
 358     dict_put(functionobjects, f, c);
 359     return c;
 360 }
 361
 362 classinfo_t* slotinfo_gettype(slotinfo_t*f)
 363 {
 364     if(f) {
 365        if(f->kind == INFOTYPE_METHOD) {
 366            return slotinfo_asclass(f);
 367        } else if(f->kind == INFOTYPE_SLOT) {
 368            varinfo_t*v = (varinfo_t*)f;
 369            return v->type;
 370        } else
 371            return 0;
 372     } else {
 373        return TYPE_ANY;
 374     }
 375 }
 376
 377 // ----------------------- package handling ---------------------------
 378 char registry_ispackage(char*package)
 379 {
 380     /* crude approximation of "the real thing", but sufficient for now */
 381     return !strncmp(package, "flash", 5);
 382 }
 383 // ----------------------- builtin types ------------------------------
 384
 385 char registry_isfunctionclass(classinfo_t*c) {
 386     return (c && c->package && c->name &&
 387             !strcmp(c->package, "") && !strcmp(c->name, "Function"));
 388 }
 389 char registry_isclassclass(classinfo_t*c) {
 390     return (c && c->package && c->name &&
 391             !strcmp(c->package, "") && !strcmp(c->name, "Class"));
 392 }
 393
 394 classinfo_t* registry_getobjectclass() {
 395     static classinfo_t*c = 0;
 396     if(!c) c = (classinfo_t*)registry_safefind("", "Object");
 397     return c;
 398 }
 399 classinfo_t* registry_getstringclass() {
 400     static classinfo_t*c = 0;
 401     if(!c) c = (classinfo_t*)registry_safefind("", "String");
 402     return c;
 403 }
 404 classinfo_t* registry_getarrayclass() {
 405     static classinfo_t*c = 0;
 406     if(!c) c = (classinfo_t*)registry_safefind("", "Array");
 407     return c;
 408 }
 409 classinfo_t* registry_getintclass() {
 410     static classinfo_t*c = 0;
 411     if(!c) c = (classinfo_t*)registry_safefind("", "int");
 412     return c;
 413 }
 414 classinfo_t* registry_getuintclass() {
 415     static classinfo_t*c = 0;
 416     if(!c) c = (classinfo_t*)registry_safefind("", "uint");
 417     return c;
 418 }
 419 classinfo_t* registry_getbooleanclass() {
 420     static classinfo_t*c = 0;
 421     if(!c) c = (classinfo_t*)registry_safefind("", "Boolean");
 422     return c;
 423 }
 424 classinfo_t* registry_getnumberclass() {
 425     static classinfo_t*c = 0;
 426     if(!c) c = (classinfo_t*)registry_safefind("", "Number");
 427     return c;
 428 }
 429 classinfo_t* registry_getregexpclass() {
 430     static classinfo_t*c = 0;
 431     if(!c) c = (classinfo_t*)registry_safefind("", "RegExp");
 432     return c;
 433 }
 434 classinfo_t* registry_getnamespaceclass() {
 435     static classinfo_t*c = 0;
 436     if(!c) c = (classinfo_t*)registry_safefind("", "Namespace");
 437     return c;
 438 }
 439 classinfo_t* registry_getMovieClip() {
 440     static classinfo_t*c = 0;
 441     if(!c) c = (classinfo_t*)registry_safefind("flash.display", "MovieClip");
 442     return c;
 443 }
 444
 445 // ----------------------- builtin dummy types -------------------------
 446 classinfo_t nullclass = {
 447     INFOTYPE_CLASS,0,0,ACCESS_PACKAGE, "", "null", 0, 0, 0
 448 };
 449 classinfo_t* registry_getnullclass() {
 450     return &nullclass;
 451 }
 452
 453 namespace_t access2namespace(U8 access, char*package)
 454 {
 455     namespace_t ns;
 456     ns.access = access;
 457     ns.name = package;
 458     return ns;
 459 }
 460
 461 char* infotypename(slotinfo_t*s)
 462 {
 463     if(s->kind == INFOTYPE_CLASS) return "class";
 464     else if(s->kind == INFOTYPE_SLOT) return "member";
 465     else if(s->kind == INFOTYPE_METHOD) return "method";
 466     else return "object";
 467 }
 468