lib/as3/import.c

   1 /* import.c
   2
   3    Extension module for the rfxswf library.
   4    Part of the swftools package.
   5
   6    Copyright (c) 2009 Matthias Kramm <kramm@quiss.org>
   7
   8    This program is free software; you can redistribute it and/or modify
   9    it under the terms of the GNU General Public License as published by
  10    the Free Software Foundation; either version 2 of the License, or
  11    (at your option) any later version.
  12
  13    This program is distributed in the hope that it will be useful,
  14    but WITHOUT ANY WARRANTY; without even the implied warranty of
  15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16    GNU General Public License for more details.
  17
  18    You should have received a copy of the GNU General Public License
  19    along with this program; if not, write to the Free Software
  20    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  21
  22 #include "import.h"
  23 #include "abc.h"
  24 #include "registry.h"
  25 #include "../os.h"
  26
  27 void as3_import_abc(char*filename)
  28 {
  29     TAG*tag = swf_InsertTag(0, ST_RAWABC);
  30     memfile_t*file = memfile_open(filename);
  31     tag->data = file->data;
  32     tag->len = file->len;
  33     abc_file_t*abc = swf_ReadABC(tag);
  34     as3_import_code(abc);
  35     swf_FreeABC(abc);
  36     memfile_close(file);
  37     free(tag);
  38 }
  39
  40 void as3_import_swf(char*filename)
  41 {
  42     SWF* swf = swf_OpenSWF(filename);
  43     if(!swf)
  44         return;
  45     TAG*tag = swf->firstTag;
  46     while(tag) {
  47         if(tag->id == ST_DOABC || tag->id == ST_RAWABC) {
  48             abc_file_t*abc = swf_ReadABC(tag);
  49             as3_import_code(abc);
  50             swf_FreeABC(abc);
  51         }
  52         tag = tag->next;
  53     }
  54     swf_FreeTags(swf);
  55     free(swf);
  56 }
  57
  58 void as3_import_file(char*filename)
  59 {
  60     FILE*fi = fopen(filename, "rb");
  61     if(!fi) return;
  62     char head[3];
  63     fread(head, 3, 1, fi);
  64     fclose(fi);
  65     if(!strncmp(head, "FWS", 3) ||
  66        !strncmp(head, "CWS", 3)) {
  67         as3_import_swf(filename);
  68     } else {
  69         as3_import_abc(filename);
  70     }
  71 }
  72
  73 static int compare_traits(const void*v1, const void*v2)
  74 {
  75     trait_t* x1 = *(trait_t**)v1;
  76     trait_t* x2 = *(trait_t**)v2;
  77     int i = strcmp(x1->name->ns->name, x2->name->ns->name);
  78     if(i)
  79         return i;
  80     return strcmp(x1->name->name, x2->name->name);
  81 }
  82
  83
  84 void as3_import_code(abc_file_t*abc)
  85 {
  86     int t;
  87     for(t=0;t<abc->classes->num;t++) {
  88         abc_class_t*cls = array_getvalue(abc->classes, t);
  89         U8 access = cls->classname->ns->access;
  90         if(access==ACCESS_PRIVATE ||
  91            access==ACCESS_PACKAGEINTERNAL)
  92             continue;
  93         if(!strncmp(cls->classname->ns->name, "__AS3", 5))
  94             continue;
  95
  96         const char*package = strdup(cls->classname->ns->name);
  97         const char*name = strdup(cls->classname->name);
  98
  99         multiname_list_t*i=cls->interfaces;
 100         classinfo_t*c = classinfo_register(access, package, name, list_length(i));
 101         c->flags|=FLAG_BUILTIN;
 102
 103         if(cls->flags & CLASS_FINAL)
 104             c->flags |= FLAG_FINAL;
 105         if(cls->flags & CLASS_INTERFACE)
 106             c->flags |= FLAG_INTERFACE;
 107         if(!(cls->flags & CLASS_SEALED))
 108             c->flags |= FLAG_DYNAMIC;
 109         /*
 110           FIXME
 111            int nr = 0;
 112         while(i) {
 113             c->interfaces[nr++] = i->multiname;
 114             i = i->next;
 115         }
 116         c->superclass = ...
 117
 118          */
 119
 120         trait_list_t*l=0;
 121         char is_static = 0;
 122         l = cls->traits;
 123         while(l) {
 124             trait_t*trait = l->trait;
 125             U8 access = trait->name->ns->access;
 126             if(access==ACCESS_PRIVATE)
 127                 goto cont;
 128             const char*name = trait->name->name;
 129             if(registry_findmember(c, name, 0))
 130                 goto cont;
 131             name = strdup(name);
 132
 133             memberinfo_t*s = 0;
 134             if(trait->kind == TRAIT_METHOD) {
 135                 s = (memberinfo_t*)methodinfo_register_onclass(c, access, name);
 136             } else if(trait->kind == TRAIT_SLOT ||
 137                       trait->kind == TRAIT_GETTER) {
 138                 s = (memberinfo_t*)varinfo_register_onclass(c, access, name);
 139             } else {
 140                 goto cont;
 141             }
 142
 143             s->flags = is_static?FLAG_STATIC:0;
 144             s->flags |= FLAG_BUILTIN;
 145             s->parent = c;
 146
 147             cont:
 148             l = l->next;
 149             if(!l && !is_static) {
 150                 l = cls->static_traits;
 151                 is_static = 1;
 152             }
 153         }
 154     }
 155
 156 #   define IS_PUBLIC_MEMBER(trait) ((trait)->kind != TRAIT_CLASS)
 157
 158     /* count public functions */
 159     int num_methods=0;
 160     for(t=0;t<abc->scripts->num;t++) {
 161         trait_list_t*l = ((abc_script_t*)array_getvalue(abc->scripts, t))->traits;
 162         for(;l;l=l->next) {
 163             num_methods += IS_PUBLIC_MEMBER(l->trait);
 164         }
 165     }
 166     trait_t**traits = (trait_t**)malloc(num_methods*sizeof(trait_t*));
 167     num_methods=0;
 168     for(t=0;t<abc->scripts->num;t++) {
 169         trait_list_t*l = ((abc_script_t*)array_getvalue(abc->scripts, t))->traits;
 170         for(;l;l=l->next) {
 171             if(IS_PUBLIC_MEMBER(l->trait)) {
 172                 traits[num_methods++] = l->trait;
 173             }
 174         }
 175     }
 176     qsort(traits, num_methods, sizeof(trait_t*), compare_traits);
 177     for(t=0;t<num_methods;t++) {
 178         trait_t*trait = traits[t];
 179         if(IS_PUBLIC_MEMBER(trait)) {
 180             U8 access = trait->name->ns->access;
 181             const char*package = strdup(trait->name->ns->name);
 182             const char*name = strdup(trait->name->name);
 183             char np = 0;
 184             memberinfo_t*m = 0;
 185             if(trait->kind == TRAIT_METHOD) {
 186                 m = (memberinfo_t*)methodinfo_register_global(access, package, name);
 187             } else {
 188                 m = (memberinfo_t*)varinfo_register_global(access, package, name);
 189             }
 190             m->flags |= FLAG_BUILTIN;
 191             m->return_type = 0;
 192             m->parent = 0;
 193         }
 194     }
 195 }