X-Git-Url: http://git.asbjorn.biz/?p=swftools.git;a=blobdiff_plain;f=lib%2Fas3%2Fimport.c;h=fdb4a1215f43ab252557f99f35610e4c2853efd4;hp=1c3e2bc0639e5531e7d7ab378a92be66d289d185;hb=b0dd28e6cbbab28a953f48bf3fe084a0b62fab63;hpb=d2bcd477230b4ac8d7e69fc253d94250b13eb37a

diff --git a/lib/as3/import.c b/lib/as3/import.c
index 1c3e2bc..fdb4a12 100644
--- a/lib/as3/import.c
+++ b/lib/as3/import.c
@@ -22,8 +22,13 @@
 #include "import.h"
 #include "abc.h"
 #include "registry.h"
+#include "common.h"
+#include "tokenizer.h"
+#include "assets.h"
 #include "../os.h"
 
+static void import_code(void*_abc, char*filename, int pass, asset_bundle_t*a);
+
 void as3_import_abc(char*filename)
 {
     TAG*tag = swf_InsertTag(0, ST_RAWABC);
@@ -31,7 +36,8 @@ void as3_import_abc(char*filename)
     tag->data = file->data;
     tag->len = file->len;
     abc_file_t*abc = swf_ReadABC(tag);
-    as3_import_code(abc);
+    import_code(abc, filename, 0, 0);
+    import_code(abc, filename, 1, 0);
     swf_FreeABC(abc);
     memfile_close(file);
     free(tag);
@@ -42,17 +48,39 @@ void as3_import_swf(char*filename)
     SWF* swf = swf_OpenSWF(filename);
     if(!swf)
         return;
+    swf_FoldAll(swf);
+
     TAG*tag = swf->firstTag;
+
+    asset_resolver_t* assets =  swf_ParseAssets(swf);
+
+    /* pass 1 */
     while(tag) {
         if(tag->id == ST_DOABC || tag->id == ST_RAWABC) {
             abc_file_t*abc = swf_ReadABC(tag);
-            as3_import_code(abc);
+            import_code(abc, filename, 0, 0);
             swf_FreeABC(abc);
         }
         tag = tag->next;
     }
-    swf_FreeTags(swf);
+
+    tag = swf->firstTag;
+    /* pass 2 */
+    while(tag) {
+        if(tag->id == ST_DOABC || tag->id == ST_RAWABC) {
+            abc_file_t*abc = swf_ReadABC(tag); //FIXME: mem leak
+	    swf_ResolveAssets(assets, abc);
+	    NEW(asset_bundle_t, a);
+	    a->file = abc;
+	    registry_add_asset(a);
+            import_code(abc, filename, 1, a);
+        }
+        tag = tag->next;
+    }
+
+    //swf_FreeTags(swf); // FIXME: mem leak
     free(swf);
+
 }
 
 void as3_import_file(char*filename)
@@ -63,10 +91,11 @@ void as3_import_file(char*filename)
     fread(head, 3, 1, fi);
     fclose(fi);
     if(!strncmp(head, "FWS", 3) ||
-       !strncmp(head, "FWC", 3))
+       !strncmp(head, "CWS", 3)) {
         as3_import_swf(filename);
-    else
+    } else {
         as3_import_abc(filename);
+    }
 }
 
 static int compare_traits(const void*v1, const void*v2)
@@ -79,66 +108,140 @@ static int compare_traits(const void*v1, const void*v2)
     return strcmp(x1->name->name, x2->name->name);
 }
 
+static classinfo_t*resolve_class(char*filename, char*what, multiname_t*n)
+{
+    if(!n) return 0;
+    if(!n->name[0]) return 0;
+    if(!strcmp(n->name, "void")) 
+        return &voidclass;
+
+    classinfo_t*c = 0;
+    if(n->ns && n->ns->name) {
+        c = (classinfo_t*)registry_find(n->ns->name, n->name);
+    } else if(n->namespace_set) {
+        namespace_list_t*s = n->namespace_set->namespaces;
+        while(s) {
+            c = (classinfo_t*)registry_find(s->namespace->name, n->name);
+            if(c)
+                break;
+            s = s->next;
+        }
+    }
+
+    if(!c) {
+        as3_warning("import %s: couldn't resolve %s %s.%s", filename, what, n->ns->name, n->name);
+        return 0;
+    }
+    if(c->kind != INFOTYPE_CLASS)
+        as3_warning("import %s: %s %s resolves to something that's not a class", filename, what, n->name);
+    return c;
+}
 
-void as3_import_code(abc_file_t*abc)
+static void import_code(void*_abc, char*filename, int pass, asset_bundle_t*asset_bundle)
 {
+    abc_file_t*abc = _abc;
     int t;
+    if(pass==0) {
+        for(t=0;t<abc->classes->num;t++) {
+            abc_class_t*cls = array_getvalue(abc->classes, t);
+            U8 access = cls->classname->ns->access;
+            if(access==ACCESS_PRIVATE ||
+               access==ACCESS_PACKAGEINTERNAL)
+                continue;
+            //if(!strncmp(cls->classname->ns->name, "__AS3", 5))
+            //    continue;
+
+            const char*package = strdup(cls->classname->ns->name);
+            const char*name = strdup(cls->classname->name);
+
+            multiname_list_t*i=cls->interfaces;
+            classinfo_t*c = classinfo_register(access, package, name, list_length(i));
+            c->flags|=FLAG_BUILTIN;
+
+            if(cls->flags & CLASS_FINAL)
+                c->flags |= FLAG_FINAL;
+            if(cls->flags & CLASS_INTERFACE)
+                c->flags |= FLAG_INTERFACE;
+            if(!(cls->flags & CLASS_SEALED))
+                c->flags |= FLAG_DYNAMIC;
+        }
+        return;
+    }
+    
     for(t=0;t<abc->classes->num;t++) {
         abc_class_t*cls = array_getvalue(abc->classes, t);
-        U8 access = cls->classname->ns->access;
-        if(access==ACCESS_PRIVATE ||
-           access==ACCESS_PACKAGEINTERNAL)
-            continue;
-        if(!strncmp(cls->classname->ns->name, "__AS3", 5))
-            continue;
-
         const char*package = strdup(cls->classname->ns->name);
         const char*name = strdup(cls->classname->name);
-                
-        multiname_list_t*i=cls->interfaces;
-        classinfo_t*c = classinfo_register(access, package, name, list_length(i));
-
-        if(cls->flags & CLASS_FINAL)
-            c->flags |= FLAG_FINAL;
-        if(cls->flags & CLASS_INTERFACE)
-            c->flags |= FLAG_INTERFACE;
-        if(!(cls->flags & CLASS_SEALED))
-            c->flags |= FLAG_DYNAMIC;
-        /*
-          FIXME 
-           int nr = 0;
+        classinfo_t*c = (classinfo_t*)registry_find(package, name);
+        if(!c) continue;
+
+	if(cls->asset) {
+	    c->assets = asset_bundle;
+	}
+
+        int nr = 0;
+        multiname_list_t*i = cls->interfaces;
         while(i) {
-            c->interfaces[nr++] = i->multiname;
+            c->interfaces[nr++] = resolve_class(filename, "interface", i->multiname);
             i = i->next;
         }
-        c->superclass = ...
-         
-         */
+        c->superclass = resolve_class(filename, "superclass", cls->superclass);
       
         trait_list_t*l=0;
         char is_static = 0;
         l = cls->traits;
+        if(!l) {
+            l = cls->static_traits;
+            is_static = 1;
+        }
+        dict_t*names = dict_new();
         while(l) {
             trait_t*trait = l->trait;
             U8 access = trait->name->ns->access;
+
             if(access==ACCESS_PRIVATE)
                 goto cont;
             const char*name = trait->name->name;
-            if(registry_findmember(c, name, 0))
+            char* ns = access==ACCESS_NAMESPACE?strdup(trait->name->ns->name):"";
+        
+	    if(registry_findmember(c, ns, name, 0, is_static))
                 goto cont;
+
             name = strdup(name);
 
             memberinfo_t*s = 0;
             if(trait->kind == TRAIT_METHOD) {
-                s = (memberinfo_t*)methodinfo_register_onclass(c, access, name);
-            } else if(trait->kind == TRAIT_SLOT ||
-                      trait->kind == TRAIT_GETTER) {
-                s = (memberinfo_t*)varinfo_register_onclass(c, access, name);
+                s = (memberinfo_t*)methodinfo_register_onclass(c, access, ns, name, is_static);
+                s->return_type = resolve_class(filename, "return type", trait->method->return_type);
+                dict_put(names, name, 0);
+            } else if(trait->kind == TRAIT_SLOT) {
+                s = (memberinfo_t*)varinfo_register_onclass(c, access, ns, name, is_static);
+                s->type = resolve_class(filename, "type", trait->type_name);
+                dict_put(names, name, 0);
+            } else if(trait->kind == TRAIT_GETTER) {
+                s = (memberinfo_t*)varinfo_register_onclass(c, access, ns, name, is_static);
+                s->type = resolve_class(filename, "type", trait->method->return_type);
+                dict_put(names, name, 0);
+            } else if(trait->kind == TRAIT_CONST) {
+                /* some variables (e.g. XML.length) are apparently both a method and a slot.
+                   needs split of static/non-static first */
+                if(!dict_contains(names, name)) {
+                    varinfo_t*v = (varinfo_t*)varinfo_register_onclass(c, access, ns, name, is_static);
+                    v->type = resolve_class(filename, "type", trait->type_name);
+                    v->flags |= FLAG_CONST;
+                    /* leave this alone for now- it blows up the file too much 
+                    v->value = constant_clone(trait->value);*/
+                    s = (memberinfo_t*)v;
+                    dict_put(names, name, 0);
+                } else 
+                    goto cont;
+
             } else {
                 goto cont;
             }
 
             s->flags = is_static?FLAG_STATIC:0;
+            s->flags |= FLAG_BUILTIN;
             s->parent = c;
 
             cont:
@@ -148,9 +251,10 @@ void as3_import_code(abc_file_t*abc)
                 is_static = 1;
             }
         }
+        dict_destroy(names);
     }
 
-#   define IS_PUBLIC_MEMBER(trait) ((trait)->kind != TRAIT_CLASS)
+#   define IS_PUBLIC_MEMBER(trait) ((trait)->kind != TRAIT_CLASS && (trait)->name->ns->access != ACCESS_PRIVATE)
 
     /* count public functions */
     int num_methods=0;
@@ -181,11 +285,22 @@ void as3_import_code(abc_file_t*abc)
             memberinfo_t*m = 0;
             if(trait->kind == TRAIT_METHOD) {
                 m = (memberinfo_t*)methodinfo_register_global(access, package, name);
+                m->return_type = resolve_class(filename, "return type", trait->method->return_type);
             } else {
-                m = (memberinfo_t*)varinfo_register_global(access, package, name);
+                varinfo_t*v = varinfo_register_global(access, package, name);
+                v->type = resolve_class(filename, "type", trait->type_name);
+                v->value = constant_clone(trait->value);
+                v->flags |= trait->kind==TRAIT_CONST?FLAG_CONST:0;
+                m = (memberinfo_t*)v;
             }
-            m->return_type = 0;
+            m->flags |= FLAG_BUILTIN;
             m->parent = 0;
         }
     }
 }
+
+void as3_import_code(void*_abc)
+{
+    import_code(_abc, "", 0, 0);
+    import_code(_abc, "", 1, 0);
+}