8721e2836e02aeaf501134cbb8a00b01530e0d09
[swftools.git] / 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 }