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 asset_bundle_list_t*assets=0;
  31
  32 // ----------------------- class signature ------------------------------
  33
  34 char slotinfo_equals(slotinfo_t*c1, slotinfo_t*c2)
  35 {
  36     if(!!c1 != !!c2)
  37         return 0;
  38     /* notice: access right is *not* respected */
  39     if(!strcmp(c1->name, c2->name) &&
  40        !strcmp(c1->package, c2->package)) {
  41         return 1;
  42     }
  43     return 0;
  44 }
  45 static unsigned int slotinfo_hash(slotinfo_t*c)
  46 {
  47     unsigned int hash = 0;
  48     hash = crc32_add_string(hash, c->package);
  49     hash = crc32_add_string(hash, c->name);
  50     return hash;
  51 }
  52 static unsigned int memberinfo_hash(slotinfo_t*c)
  53 {
  54     unsigned int hash = 0;
  55     hash = crc32_add_string(hash, c->name);
  56     return hash;
  57 }
  58
  59 static void* dummy_clone(void*other) {return other;}
  60 static void dummy_destroy(slotinfo_t*c) {}
  61
  62 type_t slotinfo_type = {
  63     hash: (hash_func)slotinfo_hash,
  64     equals: (equals_func)slotinfo_equals,
  65     dup: (dup_func)dummy_clone, // all signatures are static
  66     free: (free_func)dummy_destroy,
  67 };
  68 type_t memberinfo_type = {
  69     hash: (hash_func)memberinfo_hash,
  70     equals: (equals_func)slotinfo_equals,
  71     dup: (dup_func)dummy_clone, // all signatures are static
  72     free: (free_func)dummy_destroy,
  73 };
  74
  75 // ----------------------- assets -------------------------------------
  76 static void use_asset(asset_bundle_t*a)
  77 {
  78     a->used = 1;
  79     asset_bundle_list_t*l = a->dependencies;
  80     while(l) {
  81         if(!l->asset_bundle->used) {
  82             use_asset(l->asset_bundle);
  83         }
  84         l = l->next;
  85     }
  86 }
  87 void registry_use(slotinfo_t*s)
  88 {
  89     if(!s) return;
  90     if(!(s->flags&FLAG_USED)) {
  91         s->flags |= FLAG_USED;
  92         if(s->kind == INFOTYPE_CLASS) {
  93             classinfo_t*c=(classinfo_t*)s;
  94             if(c->assets) {
  95                 use_asset(c->assets);
  96             }
  97             int t=0;
  98             while(c->interfaces[t]) {
  99                 registry_use((slotinfo_t*)c->interfaces[t]);
 100                 t++;
 101             }
 102             while(c->superclass) {
 103                 c = c->superclass;
 104                 registry_use((slotinfo_t*)c);
 105             }
 106         } else if(s->kind == INFOTYPE_METHOD) {
 107             methodinfo_t*m=(methodinfo_t*)s;
 108             if(m->parent) {
 109                 registry_use((slotinfo_t*)m->parent);
 110             }
 111         } else if(s->kind == INFOTYPE_VAR) {
 112             varinfo_t*v=(varinfo_t*)s;
 113             if(v->parent) {
 114                 registry_use((slotinfo_t*)v->parent);
 115             }
 116         }
 117     }
 118 }
 119 void registry_add_asset(asset_bundle_t*bundle)
 120 {
 121     list_append(assets, bundle);
 122 }
 123 asset_bundle_list_t*registry_getassets()
 124 {
 125     return assets;
 126 }
 127 // ----------------------- resolving ----------------------------------
 128 slotinfo_t* registry_resolve(slotinfo_t*_s)
 129 {
 130     if(!_s || _s->kind != INFOTYPE_UNRESOLVED)
 131         return _s;
 132     unresolvedinfo_t*s = (unresolvedinfo_t*)_s;
 133
 134     if(s->package)
 135         return registry_find(s->package, s->name);
 136
 137     namespace_list_t*l = s->nsset;
 138     while(l) {
 139         slotinfo_t* n = registry_find(l->namespace->name, s->name);
 140         if(n) return n;
 141         l = l->next;
 142     }
 143     return 0;
 144 }
 145
 146 static slotinfo_list_t*unresolved = 0;
 147 static void schedule_for_resolve(slotinfo_t*s)
 148 {
 149     list_append(unresolved, s);
 150 }
 151 static void resolve_on_slot(slotinfo_t*_member)
 152 {
 153     if(_member->kind == INFOTYPE_VAR) {
 154         varinfo_t*member = (varinfo_t*)_member;
 155         member->type = (classinfo_t*)registry_resolve((slotinfo_t*)member->type);
 156     } else if(_member->kind == INFOTYPE_METHOD) {
 157         methodinfo_t*member = (methodinfo_t*)_member;
 158         member->return_type = (classinfo_t*)registry_resolve((slotinfo_t*)member->return_type);
 159         classinfo_list_t*l = member->params;
 160         while(l) {
 161             l->classinfo = (classinfo_t*)registry_resolve((slotinfo_t*)l->classinfo);
 162             l = l->next;
 163         }
 164     } else fprintf(stderr, "Internal Error: bad slot %s", _member->name);
 165 }
 166 static void resolve_on_class(slotinfo_t*_cls)
 167 {
 168     classinfo_t*cls = (classinfo_t*)_cls;
 169     cls->superclass = (classinfo_t*)registry_resolve((slotinfo_t*)cls->superclass);
 170
 171     DICT_ITERATE_DATA(&cls->members,slotinfo_t*,m) {
 172         resolve_on_slot(m);
 173     }
 174     DICT_ITERATE_DATA(&cls->static_members,slotinfo_t*,m2) {
 175         resolve_on_slot(m2);
 176     }
 177
 178     int t=0;
 179     while(cls->interfaces[t]) {
 180         cls->interfaces[t] = (classinfo_t*)registry_resolve((slotinfo_t*)cls->interfaces[t]);
 181         t++;
 182     }
 183 }
 184 void registry_resolve_all()
 185 {
 186     while(unresolved) {
 187         slotinfo_t*_s = unresolved->slotinfo;
 188         if(_s->kind == INFOTYPE_CLASS) {
 189             resolve_on_class(_s);
 190         } else if(_s->kind == INFOTYPE_METHOD || _s->kind == INFOTYPE_VAR) {
 191             resolve_on_slot(_s);
 192         } else {
 193             fprintf(stderr, "Internal Error: object %s.%s has bad type\n", _s->package, _s->name);
 194         }
 195         slotinfo_list_t*tofree = unresolved;
 196         unresolved = unresolved->next;
 197         free(tofree);
 198     }
 199 }
 200 // ------------------------- constructors --------------------------------
 201
 202 #define AVERAGE_NUMBER_OF_MEMBERS 8
 203 classinfo_t* classinfo_register(int access, const char*package, const char*name, int num_interfaces)
 204 {
 205     classinfo_t*c = rfx_calloc(sizeof(classinfo_t)+(sizeof(classinfo_t*)*(num_interfaces+1)));
 206     c->interfaces[0] = 0;
 207     c->kind = INFOTYPE_CLASS;
 208     c->access = access;
 209     c->package = package;
 210     c->name = name;
 211     dict_put(registry_classes, c, c);
 212     dict_init2(&c->members, &memberinfo_type, AVERAGE_NUMBER_OF_MEMBERS);
 213     dict_init2(&c->static_members, &memberinfo_type, AVERAGE_NUMBER_OF_MEMBERS);
 214
 215     schedule_for_resolve((slotinfo_t*)c);
 216     return c;
 217 }
 218 methodinfo_t* methodinfo_register_onclass(classinfo_t*cls, U8 access, const char*ns, const char*name, char is_static)
 219 {
 220     NEW(methodinfo_t,m);
 221     m->kind = INFOTYPE_METHOD;
 222     m->access = access;
 223     m->name = name;
 224     m->package = ns;
 225     m->parent = cls;
 226     if(!is_static)
 227         dict_put(&cls->members, m, m);
 228     else
 229         dict_put(&cls->static_members, m, m);
 230     return m;
 231 }
 232 varinfo_t* varinfo_register_onclass(classinfo_t*cls, U8 access, const char*ns, const char*name, char is_static)
 233 {
 234     NEW(varinfo_t,m);
 235     m->kind = INFOTYPE_VAR;
 236     m->access = access;
 237     m->name = name;
 238     m->package = ns;
 239     m->parent = cls;
 240     if(!is_static)
 241         dict_put(&cls->members, m, m);
 242     else
 243         dict_put(&cls->static_members, m, m);
 244     return m;
 245 }
 246 methodinfo_t* methodinfo_register_global(U8 access, const char*package, const char*name)
 247 {
 248     NEW(methodinfo_t, m);
 249     m->kind = INFOTYPE_METHOD;
 250     m->flags = FLAG_STATIC;
 251     m->access = access;
 252     m->package = package;
 253     m->name = name;
 254     m->parent = 0;
 255     dict_put(registry_classes, m, m);
 256
 257     schedule_for_resolve((slotinfo_t*)m);
 258     return m;
 259 }
 260 varinfo_t* varinfo_register_global(U8 access, const char*package, const char*name)
 261 {
 262     NEW(varinfo_t, m);
 263     m->kind = INFOTYPE_VAR;
 264     m->flags = FLAG_STATIC;
 265     m->access = access;
 266     m->package = package;
 267     m->name = name;
 268     m->parent = 0;
 269     dict_put(registry_classes, m, m);
 270
 271     schedule_for_resolve((slotinfo_t*)m);
 272     return m;
 273 }
 274
 275 // --------------- builtin classes (from builtin.c) ----------------------
 276
 277 void registry_init()
 278 {
 279     if(!registry_classes)
 280         registry_classes = builtin_getclasses();
 281 }
 282 slotinfo_t* registry_find(const char*package, const char*name)
 283 {
 284     assert(registry_classes);
 285     slotinfo_t tmp;
 286     tmp.package = package;
 287     tmp.name = name;
 288     slotinfo_t* c = (slotinfo_t*)dict_lookup(registry_classes, &tmp);
 289     /*if(c)
 290         printf("%s.%s->%08x (%s.%s)\n", package, name, c, c->package, c->name);*/
 291     return c;
 292 }
 293 slotinfo_t* registry_safefind(const char*package, const char*name)
 294 {
 295     slotinfo_t*c = registry_find(package, name);
 296     if(!c) {
 297         printf("%s.%s\n", package, name);
 298     }
 299     assert(c);
 300     return c;
 301 }
 302 void registry_dump()
 303 {
 304     int t;
 305     for(t=0;t<registry_classes->hashsize;t++) {
 306         dictentry_t*e = registry_classes->slots[t];
 307         while(e) {
 308             slotinfo_t*i = (slotinfo_t*)e->key;
 309             printf("[%s] %s.%s\n", access2str(i->access), i->package, i->name);
 310             e = e->next;
 311         }
 312     }
 313 }
 314
 315 memberinfo_t* registry_findmember(classinfo_t*cls, const char*ns, const char*name, char recursive, char is_static)
 316 {
 317     memberinfo_t tmp;
 318     tmp.name = name;
 319     tmp.package = ns?ns:"";
 320
 321     if(!recursive) {
 322         if(!is_static)
 323             return (memberinfo_t*)dict_lookup(&cls->members, &tmp);
 324         else
 325             return (memberinfo_t*)dict_lookup(&cls->static_members, &tmp);
 326     }
 327     /* look at classes directly extended by this class */
 328     slotinfo_t*m = 0;
 329     classinfo_t*s = cls;
 330
 331     if(recursive>1) // check *only* superclasses
 332         s = s->superclass;
 333
 334     while(s) {
 335         if(s->kind == INFOTYPE_UNRESOLVED)
 336             break;
 337
 338         if(!is_static) {
 339             m = (slotinfo_t*)dict_lookup(&s->members, &tmp);
 340             if(m) return (memberinfo_t*)m;
 341         }
 342         m = (slotinfo_t*)dict_lookup(&s->static_members, &tmp);
 343         if(m) return (memberinfo_t*)m;
 344
 345         s = s->superclass;
 346     }
 347     /* look at interfaces, and parent interfaces */
 348     int t=0;
 349     while(cls->interfaces[t]) {
 350         classinfo_t*s = cls->interfaces[t];
 351         if(s->kind != INFOTYPE_UNRESOLVED) {
 352             while(s) {
 353                 if(!is_static) {
 354                     m = (slotinfo_t*)dict_lookup(&s->members, &tmp);
 355                     if(m) return (memberinfo_t*)m;
 356                 }
 357                 m = (slotinfo_t*)dict_lookup(&s->static_members, &tmp);
 358                 if(m) return (memberinfo_t*)m;
 359
 360                 s = s->superclass;
 361             }
 362         }
 363         t++;
 364     }
 365     return 0;
 366 }
 367
 368 memberinfo_t* registry_findmember_nsset(classinfo_t*cls, namespace_list_t*ns, const char*name, char superclasses, char is_static)
 369 {
 370     memberinfo_t*m = 0;
 371     while(ns) {
 372         m = registry_findmember(cls, ns->namespace->name, name, superclasses, is_static);
 373         if(m) return m;
 374         ns = ns->next;
 375     }
 376     m = registry_findmember(cls, "", name, superclasses, is_static);
 377     if(m) return m;
 378     /* TODO: it maybe would be faster to just store the builtin namespace as "" in
 379              builtins.c (update: some members (e.g. XML.length) are present both for
 380             "" and "http:...builtin") */
 381     m = registry_findmember(cls, "http://adobe.com/AS3/2006/builtin", name, superclasses, is_static);
 382     if(m) return m;
 383     return 0;
 384 }
 385
 386
 387 void registry_fill_multiname(multiname_t*m, namespace_t*n, slotinfo_t*c)
 388 {
 389     m->type = QNAME;
 390     m->ns = n;
 391     m->ns->access = c->access;
 392     m->ns->name = (char*)c->package;
 393     m->name = c->name;
 394     m->namespace_set = 0;
 395 }
 396 multiname_t* classinfo_to_multiname(slotinfo_t*cls)
 397 {
 398     if(!cls)
 399         return 0;
 400     multiname_t*m=0;
 401     namespace_t ns = {cls->access, (char*)cls->package};
 402     return multiname_new(&ns,cls->name);
 403 }
 404
 405 // ----------------------- memberinfo methods ------------------------------
 406
 407 /* hacky code to wrap a variable or function into a "type"
 408    object, but keep a pointer to the "value" */
 409 static dict_t* functionobjects = 0;
 410 classinfo_t* slotinfo_asclass(slotinfo_t*f) {
 411     if(!functionobjects) {
 412         functionobjects = dict_new2(&ptr_type);
 413     } else {
 414         classinfo_t*c = dict_lookup(functionobjects, f);
 415         if(c)
 416             return c;
 417     }
 418
 419     classinfo_t*c = rfx_calloc(sizeof(classinfo_t)+sizeof(classinfo_t*));
 420     c->access = ACCESS_PUBLIC;
 421     c->package = "";
 422     if(f->kind == INFOTYPE_METHOD) {
 423         c->name = "Function";
 424         c->superclass = registry_getobjectclass();
 425     } else if(f->kind == INFOTYPE_CLASS) {
 426         c->name = "Class";
 427         c->superclass = registry_getobjectclass();
 428     } else if(f->kind == INFOTYPE_VAR) {
 429         c->name = "Object";
 430     } else {
 431         c->name = "undefined";
 432     }
 433
 434     dict_init2(&c->members, &memberinfo_type, 1);
 435     dict_init2(&c->static_members, &memberinfo_type, 1);
 436     c->data = f;
 437     dict_put(functionobjects, f, c);
 438     return c;
 439 }
 440
 441 classinfo_t* slotinfo_gettype(slotinfo_t*f)
 442 {
 443     if(f) {
 444        if(f->kind == INFOTYPE_METHOD) {
 445            return slotinfo_asclass(f);
 446        } else if(f->kind == INFOTYPE_VAR) {
 447            varinfo_t*v = (varinfo_t*)f;
 448            return v->type;
 449        } else
 450            return 0;
 451     } else {
 452        return TYPE_ANY;
 453     }
 454 }
 455
 456 // ----------------------- package handling ---------------------------
 457 char registry_ispackage(const char*package)
 458 {
 459     /* crude approximation of "the real thing", but sufficient for now */
 460     return !strncmp(package, "flash", 5);
 461 }
 462 // ----------------------- builtin types ------------------------------
 463
 464 char registry_isfunctionclass(classinfo_t*c) {
 465     return (c && c->package && c->name &&
 466             !strcmp(c->package, "") && !strcmp(c->name, "Function"));
 467 }
 468 char registry_isclassclass(classinfo_t*c) {
 469     return (c && c->package && c->name &&
 470             !strcmp(c->package, "") && !strcmp(c->name, "Class"));
 471 }
 472
 473 classinfo_t* registry_getobjectclass() {
 474     static classinfo_t*c = 0;
 475     if(!c) c = (classinfo_t*)registry_safefind("", "Object");
 476     return c;
 477 }
 478 classinfo_t* registry_getstringclass() {
 479     static classinfo_t*c = 0;
 480     if(!c) c = (classinfo_t*)registry_safefind("", "String");
 481     return c;
 482 }
 483 classinfo_t* registry_getarrayclass() {
 484     static classinfo_t*c = 0;
 485     if(!c) c = (classinfo_t*)registry_safefind("", "Array");
 486     return c;
 487 }
 488 classinfo_t* registry_getintclass() {
 489     static classinfo_t*c = 0;
 490     if(!c) c = (classinfo_t*)registry_safefind("", "int");
 491     return c;
 492 }
 493 classinfo_t* registry_getuintclass() {
 494     static classinfo_t*c = 0;
 495     if(!c) c = (classinfo_t*)registry_safefind("", "uint");
 496     return c;
 497 }
 498 classinfo_t* registry_getbooleanclass() {
 499     static classinfo_t*c = 0;
 500     if(!c) c = (classinfo_t*)registry_safefind("", "Boolean");
 501     return c;
 502 }
 503 classinfo_t* registry_getnumberclass() {
 504     static classinfo_t*c = 0;
 505     if(!c) c = (classinfo_t*)registry_safefind("", "Number");
 506     return c;
 507 }
 508 classinfo_t* registry_getregexpclass() {
 509     static classinfo_t*c = 0;
 510     if(!c) c = (classinfo_t*)registry_safefind("", "RegExp");
 511     return c;
 512 }
 513 classinfo_t* registry_getdateclass() {
 514     static classinfo_t*c = 0;
 515     if(!c) c = (classinfo_t*)registry_safefind("", "Date");
 516     return c;
 517 }
 518 classinfo_t* registry_getxmlclass() {
 519     static classinfo_t*c = 0;
 520     if(!c) c = (classinfo_t*)registry_safefind("", "XML");
 521     return c;
 522 }
 523 classinfo_t* registry_getxmllistclass() {
 524     static classinfo_t*c = 0;
 525     if(!c) c = (classinfo_t*)registry_safefind("", "XMLList");
 526     return c;
 527 }
 528 classinfo_t* registry_getnamespaceclass() {
 529     static classinfo_t*c = 0;
 530     if(!c) c = (classinfo_t*)registry_safefind("", "Namespace");
 531     return c;
 532 }
 533 classinfo_t* registry_getMovieClip() {
 534     static classinfo_t*c = 0;
 535     if(!c) c = (classinfo_t*)registry_safefind("flash.display", "MovieClip");
 536     return c;
 537 }
 538
 539 // ----------------------- builtin dummy types -------------------------
 540 classinfo_t nullclass = {
 541     INFOTYPE_CLASS,0,0,ACCESS_PACKAGE, "", "null", 0, 0, 0
 542 };
 543 classinfo_t* registry_getnullclass() {
 544     return &nullclass;
 545 }
 546 classinfo_t voidclass = {
 547     INFOTYPE_CLASS,0,0,ACCESS_PACKAGE, "", "void", 0, 0, 0
 548 };
 549 classinfo_t* registry_getvoidclass() {
 550     return &voidclass;
 551 }
 552
 553 namespace_t access2namespace(U8 access, char*package)
 554 {
 555     namespace_t ns;
 556     ns.access = access;
 557     ns.name = package;
 558     return ns;
 559 }
 560
 561 char* infotypename(slotinfo_t*s)
 562 {
 563     if(s->kind == INFOTYPE_CLASS) return "class";
 564     else if(s->kind == INFOTYPE_VAR) return "var";
 565     else if(s->kind == INFOTYPE_METHOD) return "function";
 566     else return "object";
 567 }
 568
 569 void slotinfo_dump(slotinfo_t*s)
 570 {
 571     if(s->package[0]) {
 572         printf("%s %s.%s", infotypename(s), s->package, s->name);
 573     } else {
 574         printf("%s %s", infotypename(s), s->name);
 575     }
 576     if(s->kind == INFOTYPE_CLASS) {
 577         classinfo_t*c = (classinfo_t*)s;
 578     }
 579     else if(s->kind == INFOTYPE_VAR) {
 580         varinfo_t*v = (varinfo_t*)s;
 581         printf(":%s", v->type?v->type->name:"*");
 582         if(v->value)
 583             printf("=%s", constant_tostring(v->value));
 584         if(v->slot)
 585             printf(" (slot:%d)", v->slot);
 586     }
 587     else if(s->kind == INFOTYPE_METHOD) {
 588         methodinfo_t*m = (methodinfo_t*)s;
 589     }
 590     else {
 591     }
 592     printf("\n");
 593 }
 594