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_VAR) {
 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
 118     DICT_ITERATE_DATA(&cls->members,slotinfo_t*,m) {
 119         resolve_on_slot(m);
 120     }
 121     DICT_ITERATE_DATA(&cls->static_members,slotinfo_t*,m2) {
 122         resolve_on_slot(m2);
 123     }
 124
 125     int t=0;
 126     while(cls->interfaces[t]) {
 127         cls->interfaces[t] = (classinfo_t*)registry_resolve((slotinfo_t*)cls->interfaces[t]);
 128         t++;
 129     }
 130 }
 131 void registry_resolve_all()
 132 {
 133     while(unresolved) {
 134         slotinfo_t*_s = unresolved->slotinfo;
 135         if(_s->kind == INFOTYPE_CLASS) {
 136             resolve_on_class(_s);
 137         } else if(_s->kind == INFOTYPE_METHOD || _s->kind == INFOTYPE_VAR) {
 138             resolve_on_slot(_s);
 139         } else {
 140             fprintf(stderr, "Internal Error: object %s.%s has bad type\n", _s->package, _s->name);
 141         }
 142         slotinfo_list_t*tofree = unresolved;
 143         unresolved = unresolved->next;
 144         free(tofree);
 145     }
 146 }
 147 // ------------------------- constructors --------------------------------
 148
 149 #define AVERAGE_NUMBER_OF_MEMBERS 8
 150 classinfo_t* classinfo_register(int access, const char*package, const char*name, int num_interfaces)
 151 {
 152     classinfo_t*c = rfx_calloc(sizeof(classinfo_t)+(sizeof(classinfo_t*)*(num_interfaces+1)));
 153     c->interfaces[0] = 0;
 154     c->kind = INFOTYPE_CLASS;
 155     c->access = access;
 156     c->package = package;
 157     c->name = name;
 158     dict_put(registry_classes, c, c);
 159     dict_init2(&c->members, &memberinfo_type, AVERAGE_NUMBER_OF_MEMBERS);
 160     dict_init2(&c->static_members, &memberinfo_type, AVERAGE_NUMBER_OF_MEMBERS);
 161
 162     schedule_for_resolve((slotinfo_t*)c);
 163     return c;
 164 }
 165 methodinfo_t* methodinfo_register_onclass(classinfo_t*cls, U8 access, const char*ns, const char*name, char is_static)
 166 {
 167     NEW(methodinfo_t,m);
 168     m->kind = INFOTYPE_METHOD;
 169     m->access = access;
 170     m->name = name;
 171     m->package = ns;
 172     m->parent = cls;
 173     if(!is_static)
 174         dict_put(&cls->members, m, m);
 175     else
 176         dict_put(&cls->static_members, m, m);
 177     return m;
 178 }
 179 varinfo_t* varinfo_register_onclass(classinfo_t*cls, U8 access, const char*ns, const char*name, char is_static)
 180 {
 181     NEW(varinfo_t,m);
 182     m->kind = INFOTYPE_VAR;
 183     m->access = access;
 184     m->name = name;
 185     m->package = ns;
 186     m->parent = cls;
 187     if(!is_static)
 188         dict_put(&cls->members, m, m);
 189     else
 190         dict_put(&cls->static_members, m, m);
 191     return m;
 192 }
 193 methodinfo_t* methodinfo_register_global(U8 access, const char*package, const char*name)
 194 {
 195     NEW(methodinfo_t, m);
 196     m->kind = INFOTYPE_METHOD;
 197     m->flags = FLAG_STATIC;
 198     m->access = access;
 199     m->package = package;
 200     m->name = name;
 201     m->parent = 0;
 202     dict_put(registry_classes, m, m);
 203
 204     schedule_for_resolve((slotinfo_t*)m);
 205     return m;
 206 }
 207 varinfo_t* varinfo_register_global(U8 access, const char*package, const char*name)
 208 {
 209     NEW(varinfo_t, m);
 210     m->kind = INFOTYPE_VAR;
 211     m->flags = FLAG_STATIC;
 212     m->access = access;
 213     m->package = package;
 214     m->name = name;
 215     m->parent = 0;
 216     dict_put(registry_classes, m, m);
 217
 218     schedule_for_resolve((slotinfo_t*)m);
 219     return m;
 220 }
 221
 222 // --------------- builtin classes (from builtin.c) ----------------------
 223
 224 void registry_init()
 225 {
 226     if(!registry_classes)
 227         registry_classes = builtin_getclasses();
 228 }
 229 slotinfo_t* registry_find(const char*package, const char*name)
 230 {
 231     assert(registry_classes);
 232     slotinfo_t tmp;
 233     tmp.package = package;
 234     tmp.name = name;
 235     slotinfo_t* c = (slotinfo_t*)dict_lookup(registry_classes, &tmp);
 236     /*if(c)
 237         printf("%s.%s->%08x (%s.%s)\n", package, name, c, c->package, c->name);*/
 238     return c;
 239 }
 240 slotinfo_t* registry_safefind(const char*package, const char*name)
 241 {
 242     slotinfo_t*c = registry_find(package, name);
 243     if(!c) {
 244         printf("%s.%s\n", package, name);
 245     }
 246     assert(c);
 247     return c;
 248 }
 249 void registry_dump()
 250 {
 251     int t;
 252     for(t=0;t<registry_classes->hashsize;t++) {
 253         dictentry_t*e = registry_classes->slots[t];
 254         while(e) {
 255             slotinfo_t*i = (slotinfo_t*)e->key;
 256             printf("[%s] %s.%s\n", access2str(i->access), i->package, i->name);
 257             e = e->next;
 258         }
 259     }
 260 }
 261
 262 memberinfo_t* registry_findmember(classinfo_t*cls, const char*ns, const char*name, char recursive, char is_static)
 263 {
 264     memberinfo_t tmp;
 265     tmp.name = name;
 266     tmp.package = ns?ns:"";
 267
 268     if(!recursive) {
 269         if(!is_static)
 270             return (memberinfo_t*)dict_lookup(&cls->members, &tmp);
 271         else
 272             return (memberinfo_t*)dict_lookup(&cls->static_members, &tmp);
 273     }
 274     /* look at classes directly extended by this class */
 275     slotinfo_t*m = 0;
 276     classinfo_t*s = cls;
 277
 278     if(recursive>1) // check *only* superclasses
 279         s = s->superclass;
 280
 281     while(s) {
 282         if(s->kind == INFOTYPE_UNRESOLVED)
 283             break;
 284
 285         if(!is_static) {
 286             m = (slotinfo_t*)dict_lookup(&s->members, &tmp);
 287             if(m) return (memberinfo_t*)m;
 288         }
 289         m = (slotinfo_t*)dict_lookup(&s->static_members, &tmp);
 290         if(m) return (memberinfo_t*)m;
 291
 292         s = s->superclass;
 293     }
 294     /* look at interfaces, and parent interfaces */
 295     int t=0;
 296     while(cls->interfaces[t]) {
 297         classinfo_t*s = cls->interfaces[t];
 298         if(s->kind != INFOTYPE_UNRESOLVED) {
 299             while(s) {
 300                 if(!is_static) {
 301                     m = (slotinfo_t*)dict_lookup(&s->members, &tmp);
 302                     if(m) return (memberinfo_t*)m;
 303                 }
 304                 m = (slotinfo_t*)dict_lookup(&s->static_members, &tmp);
 305                 if(m) return (memberinfo_t*)m;
 306
 307                 s = s->superclass;
 308             }
 309         }
 310         t++;
 311     }
 312     return 0;
 313 }
 314
 315 memberinfo_t* registry_findmember_nsset(classinfo_t*cls, namespace_list_t*ns, const char*name, char superclasses, char is_static)
 316 {
 317     memberinfo_t*m = 0;
 318     while(ns) {
 319         m = registry_findmember(cls, ns->namespace->name, name, superclasses, is_static);
 320         if(m) return m;
 321         ns = ns->next;
 322     }
 323     m = registry_findmember(cls, "", name, superclasses, is_static);
 324     if(m) return m;
 325     /* TODO: it maybe would be faster to just store the builtin namespace as "" in
 326              builtins.c (update: some members (e.g. XML.length) are present both for
 327             "" and "http:...builtin") */
 328     m = registry_findmember(cls, "http://adobe.com/AS3/2006/builtin", name, superclasses, is_static);
 329     if(m) return m;
 330     return 0;
 331 }
 332
 333
 334 void registry_fill_multiname(multiname_t*m, namespace_t*n, slotinfo_t*c)
 335 {
 336     m->type = QNAME;
 337     m->ns = n;
 338     m->ns->access = c->access;
 339     m->ns->name = (char*)c->package;
 340     m->name = c->name;
 341     m->namespace_set = 0;
 342 }
 343 multiname_t* classinfo_to_multiname(slotinfo_t*cls)
 344 {
 345     if(!cls)
 346         return 0;
 347     multiname_t*m=0;
 348     namespace_t ns = {cls->access, (char*)cls->package};
 349     return multiname_new(&ns,cls->name);
 350 }
 351
 352 // ----------------------- memberinfo methods ------------------------------
 353
 354 /* hacky code to wrap a variable or function into a "type"
 355    object, but keep a pointer to the "value" */
 356 static dict_t* functionobjects = 0;
 357 classinfo_t* slotinfo_asclass(slotinfo_t*f) {
 358     if(!functionobjects) {
 359         functionobjects = dict_new2(&ptr_type);
 360     } else {
 361         classinfo_t*c = dict_lookup(functionobjects, f);
 362         if(c)
 363             return c;
 364     }
 365
 366     classinfo_t*c = rfx_calloc(sizeof(classinfo_t)+sizeof(classinfo_t*));
 367     c->access = ACCESS_PUBLIC;
 368     c->package = "";
 369     if(f->kind == INFOTYPE_METHOD) {
 370         c->name = "Function";
 371         c->superclass = registry_getobjectclass();
 372     } else if(f->kind == INFOTYPE_CLASS) {
 373         c->name = "Class";
 374         c->superclass = registry_getobjectclass();
 375     } else if(f->kind == INFOTYPE_VAR) {
 376         c->name = "Object";
 377     } else {
 378         c->name = "undefined";
 379     }
 380
 381     dict_init2(&c->members, &memberinfo_type, 1);
 382     dict_init2(&c->static_members, &memberinfo_type, 1);
 383     c->data = f;
 384     dict_put(functionobjects, f, c);
 385     return c;
 386 }
 387
 388 classinfo_t* slotinfo_gettype(slotinfo_t*f)
 389 {
 390     if(f) {
 391        if(f->kind == INFOTYPE_METHOD) {
 392            return slotinfo_asclass(f);
 393        } else if(f->kind == INFOTYPE_VAR) {
 394            varinfo_t*v = (varinfo_t*)f;
 395            return v->type;
 396        } else
 397            return 0;
 398     } else {
 399        return TYPE_ANY;
 400     }
 401 }
 402
 403 // ----------------------- package handling ---------------------------
 404 char registry_ispackage(const char*package)
 405 {
 406     /* crude approximation of "the real thing", but sufficient for now */
 407     return !strncmp(package, "flash", 5);
 408 }
 409 // ----------------------- builtin types ------------------------------
 410
 411 char registry_isfunctionclass(classinfo_t*c) {
 412     return (c && c->package && c->name &&
 413             !strcmp(c->package, "") && !strcmp(c->name, "Function"));
 414 }
 415 char registry_isclassclass(classinfo_t*c) {
 416     return (c && c->package && c->name &&
 417             !strcmp(c->package, "") && !strcmp(c->name, "Class"));
 418 }
 419
 420 classinfo_t* registry_getobjectclass() {
 421     static classinfo_t*c = 0;
 422     if(!c) c = (classinfo_t*)registry_safefind("", "Object");
 423     return c;
 424 }
 425 classinfo_t* registry_getstringclass() {
 426     static classinfo_t*c = 0;
 427     if(!c) c = (classinfo_t*)registry_safefind("", "String");
 428     return c;
 429 }
 430 classinfo_t* registry_getarrayclass() {
 431     static classinfo_t*c = 0;
 432     if(!c) c = (classinfo_t*)registry_safefind("", "Array");
 433     return c;
 434 }
 435 classinfo_t* registry_getintclass() {
 436     static classinfo_t*c = 0;
 437     if(!c) c = (classinfo_t*)registry_safefind("", "int");
 438     return c;
 439 }
 440 classinfo_t* registry_getuintclass() {
 441     static classinfo_t*c = 0;
 442     if(!c) c = (classinfo_t*)registry_safefind("", "uint");
 443     return c;
 444 }
 445 classinfo_t* registry_getbooleanclass() {
 446     static classinfo_t*c = 0;
 447     if(!c) c = (classinfo_t*)registry_safefind("", "Boolean");
 448     return c;
 449 }
 450 classinfo_t* registry_getnumberclass() {
 451     static classinfo_t*c = 0;
 452     if(!c) c = (classinfo_t*)registry_safefind("", "Number");
 453     return c;
 454 }
 455 classinfo_t* registry_getregexpclass() {
 456     static classinfo_t*c = 0;
 457     if(!c) c = (classinfo_t*)registry_safefind("", "RegExp");
 458     return c;
 459 }
 460 classinfo_t* registry_getdateclass() {
 461     static classinfo_t*c = 0;
 462     if(!c) c = (classinfo_t*)registry_safefind("", "Date");
 463     return c;
 464 }
 465 classinfo_t* registry_getxmlclass() {
 466     static classinfo_t*c = 0;
 467     if(!c) c = (classinfo_t*)registry_safefind("", "XML");
 468     return c;
 469 }
 470 classinfo_t* registry_getxmllistclass() {
 471     static classinfo_t*c = 0;
 472     if(!c) c = (classinfo_t*)registry_safefind("", "XMLList");
 473     return c;
 474 }
 475 classinfo_t* registry_getnamespaceclass() {
 476     static classinfo_t*c = 0;
 477     if(!c) c = (classinfo_t*)registry_safefind("", "Namespace");
 478     return c;
 479 }
 480 classinfo_t* registry_getMovieClip() {
 481     static classinfo_t*c = 0;
 482     if(!c) c = (classinfo_t*)registry_safefind("flash.display", "MovieClip");
 483     return c;
 484 }
 485
 486 // ----------------------- builtin dummy types -------------------------
 487 classinfo_t nullclass = {
 488     INFOTYPE_CLASS,0,0,ACCESS_PACKAGE, "", "null", 0, 0, 0
 489 };
 490 classinfo_t* registry_getnullclass() {
 491     return &nullclass;
 492 }
 493 classinfo_t voidclass = {
 494     INFOTYPE_CLASS,0,0,ACCESS_PACKAGE, "", "void", 0, 0, 0
 495 };
 496 classinfo_t* registry_getvoidclass() {
 497     return &voidclass;
 498 }
 499
 500 namespace_t access2namespace(U8 access, char*package)
 501 {
 502     namespace_t ns;
 503     ns.access = access;
 504     ns.name = package;
 505     return ns;
 506 }
 507
 508 char* infotypename(slotinfo_t*s)
 509 {
 510     if(s->kind == INFOTYPE_CLASS) return "class";
 511     else if(s->kind == INFOTYPE_VAR) return "var";
 512     else if(s->kind == INFOTYPE_METHOD) return "function";
 513     else return "object";
 514 }
 515
 516 void slotinfo_dump(slotinfo_t*s)
 517 {
 518     if(s->package[0]) {
 519         printf("%s %s.%s", infotypename(s), s->package, s->name);
 520     } else {
 521         printf("%s %s", infotypename(s), s->name);
 522     }
 523     if(s->kind == INFOTYPE_CLASS) {
 524         classinfo_t*c = (classinfo_t*)s;
 525     }
 526     else if(s->kind == INFOTYPE_VAR) {
 527         varinfo_t*v = (varinfo_t*)s;
 528         printf(":%s", v->type?v->type->name:"*");
 529         if(v->value)
 530             printf("=%s", constant_tostring(v->value));
 531         if(v->slot)
 532             printf(" (slot:%d)", v->slot);
 533     }
 534     else if(s->kind == INFOTYPE_METHOD) {
 535         methodinfo_t*m = (methodinfo_t*)s;
 536     }
 537     else {
 538     }
 539     printf("\n");
 540 }
 541