added hashing for everything
authorkramm <kramm>
Sun, 23 Nov 2008 13:57:48 +0000 (13:57 +0000)
committerkramm <kramm>
Sun, 23 Nov 2008 13:57:48 +0000 (13:57 +0000)
lib/as3/pool.c
lib/as3/pool.h

index fc92648..ccb6ac7 100644 (file)
    along with this program; if not, write to the Free Software
    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
 
+#include <assert.h>
 #include "pool.h"
 
-char* access2str(int type)
+
+// ----------------------------- float ----------------------------------
+
+double undefined_float = 0.0;
+
+void* float_clone(const void*_v) {
+    if(_v==&undefined_float)
+        return &undefined_float;
+    const double*v1=_v;
+    double*v2 = malloc(sizeof(double));
+    *v2 = *v1;
+    return v2;
+}
+unsigned int float_hash(const void*_v) {
+    const unsigned char*b=_v;
+    unsigned int h=0;
+    int t;
+    for(t=0;t<8;t++)
+        h = crc32_add_byte(h, b[t]);
+    return h;
+}
+void float_destroy(void*_v) {
+    double*v = (double*)_v;
+    if(v!=&undefined_float)
+        free(v);
+}
+char float_equals(const void*_v1, const void*_v2) {
+    const double*v1=_v1;
+    const double*v2=_v2;
+    if(v1==&undefined_float || v2==&undefined_float)
+        return 0;
+    return *v1==*v2;
+}
+
+type_t float_type = {
+    dup: float_clone,
+    hash: float_hash,
+    free: float_destroy,
+    equals: float_equals
+};
+
+// ----------------------------- uint ----------------------------------
+
+int undefined_uint = 0;
+
+ptroff_t uint_clone(const void*v) {
+    return (ptroff_t)v;
+}
+unsigned int uint_hash(const void*v) {return (ptroff_t)v;}
+void uint_destroy(void*v) {}
+char uint_equals(const void*v1, const void*v2) {return v1==v2;}
+
+type_t uint_type = {
+    dup: (dup_func)uint_clone,
+    hash: (hash_func)uint_hash,
+    free: (free_func)uint_destroy,
+    equals: (equals_func)uint_equals
+};
+
+// ----------------------------- namespace ----------------------------------
+
+unsigned int namespace_hash(namespace_t*n)
 {
-    if(type==0x08) return "";
-    else if(type==0x16) return "package";
-    else if(type==0x17) return "packageinternal";
-    else if(type==0x18) return "protected";
-    else if(type==0x19) return "explicit";
-    else if(type==0x1A) return "staticprotected";
-    else if(type==0x05) return "private";
-    else return "undefined";
+    unsigned int hash = 0;
+    hash = crc32_add_byte(hash, n->access);
+    hash = crc32_add_string(hash, n->name);
+    return hash;
 }
 
-char* multiname_to_string(multiname_t*m)
+unsigned char namespace_equals(const namespace_t*n1, const namespace_t*n2)
 {
-    char*mname = 0;
-    if(!m)
-        return strdup("--<UNDEFINED_MULTINAME>--");
-    if(m->type==0x07 || m->type==0x0d) {
-        mname = malloc(strlen(m->ns->name)+strlen(m->name)+32);
-        sprintf(mname, "<%s>\0", access2str(m->ns->access));
-        strcat(mname, m->ns->name);
-        strcat(mname, "::");
-        strcat(mname, m->name);
-    } else if(m->type==0x0f || m->type==0x10) {
-        mname = strdup(m->name);
-    } else if(m->type==0x11 || m->type==0x12) {
-        mname = strdup("");
-    } else if(m->type==0x09 || m->type==0x0e) {
-        char*s = namespace_set_to_string(m->namespace_set);
-        mname = malloc(strlen(s)+strlen(m->name)+16);
-        strcpy(mname, s);
-        free(s);
-        strcat(mname, "::");
-        strcat(mname, m->name);
-    } else if(m->type==0x1b || m->type==0x1c) {
-        mname = namespace_set_to_string(m->namespace_set);
-    }
-    return mname;
+    if(n1->access != n2->access)
+        return 0;
+    if(!(n1->name) != !(n2->name))
+        return 0;
+    if(n1->name && n2->name && strcmp(n1->name, n2->name))
+        return 0;
+    return 1;
 }
 
 char* namespace_to_string(namespace_t*ns)
 {
     char*access = 0;
     U8 type = ns->access;
-    if(type==0x08) access = "";
-    else if(type==0x16) access = "Package";
-    else if(type==0x17) access = "PackageInternal";
-    else if(type==0x18) access = "Protected";
-    else if(type==0x19) access = "Explicit";
-    else if(type==0x1A) access = "StaticProtected";
-    else if(type==0x05) access = "Private";
-    else {
-        fprintf(stderr, "Undefined namespace type %02x\n", type);
-        access = "UnknownNS";
-    }
+    access = access2str(type);
     char*string = malloc(strlen(access)+strlen(ns->name)+3);
-    sprintf(string, "<%s>%s", access, ns->name);
+    sprintf(string, "[%s]%s", access, ns->name);
     return string;
 }
 
+namespace_t* namespace_clone(namespace_t*other)
+{
+    NEW(namespace_t,n);
+    n->access = other->access;
+    n->name = strdup(other->name);
+    return n;
+}
+
+namespace_t* namespace_new(U8 access, const char*name)
+{
+    namespace_t*ns = malloc(sizeof(namespace_t));
+    memset(ns, 0, sizeof(namespace_t));
+
+    if(access==0) { // autodetect access
+       char*n = strdup(name);
+       if(n[0] == '[') {
+           char*bracket = strchr(n, ']');
+           if(bracket) {
+               *bracket = 0;
+               char*a = n+1;
+               name += (bracket-n)+1;
+               if(!strcmp(a, "")) access=0x16;
+                else if(!strcmp(a, "undefined")) access=0x08; // public??
+               else if(!strcmp(a, "package")) access=0x16;
+               else if(!strcmp(a, "packageinternal")) access=0x17;
+               else if(!strcmp(a, "protected")) access=0x18;
+               else if(!strcmp(a, "explicit")) access=0x19;
+               else if(!strcmp(a, "staticprotected")) access=0x1a;
+               else if(!strcmp(a, "private")) access=0x05;
+               else {
+                   fprintf(stderr, "Undefined access level: [%s]\n", a);
+                    return 0;
+               }
+           }
+       } else {
+           access = 0x16;
+       }
+       free(n);
+    }
+    ns->access = access;
+    ns->name = strdup(name);
+    return ns;
+}
+
+namespace_t* namespace_new_undefined(const char*name) {
+    return namespace_new(0x08, name); // public?
+}
+namespace_t* namespace_new_package(const char*name) {
+    return namespace_new(0x16 , name);
+}
+namespace_t* namespace_new_packageinternal(const char*name) {
+    return namespace_new(0x17, name);
+}
+namespace_t* namespace_new_protected(const char*name) {
+    return namespace_new(0x18, name);
+}
+namespace_t* namespace_new_explicit(const char*name) {
+    return namespace_new(0x19, name);
+}
+namespace_t* namespace_new_staticprotected(const char*name) {
+    return namespace_new(0x1a, name);
+}
+namespace_t* namespace_new_private(const char*name) {
+    return namespace_new(0x05, name);
+}
+
+void namespace_destroy(namespace_t*n)
+{
+    free(n->name);n->name=0;
+    n->access=0x00;
+    free(n);
+}
+
+type_t namespace_type = {
+    dup: (dup_func)namespace_clone,
+    hash: (hash_func)namespace_hash,
+    free: (free_func)namespace_destroy,
+    equals: (equals_func)namespace_equals
+};
+
+// ---------------------------namespace sets --------------------------------
+
+unsigned int namespace_set_hash(namespace_set_t*set)
+{
+    namespace_list_t*l = set->namespaces;
+    unsigned int hash = 0;
+    while(l) {
+        hash = crc32_add_byte(hash, l->namespace->access);
+        hash = crc32_add_string(hash, l->namespace->name);
+        l = l->next;
+    }
+    return hash;
+}
+
+int namespace_set_equals(namespace_set_t*m1, namespace_set_t*m2)
+{
+    namespace_list_t*l1 = m1->namespaces;
+    namespace_list_t*l2 = m2->namespaces;
+    while(l1 && l2) {
+        if(l1->namespace->access != l2->namespace->access)
+            return 0;
+        if(!(l1->namespace->name) != !(l2->namespace->name))
+            return 0;
+        if(strcmp(l1->namespace->name, l2->namespace->name))
+            return 0;
+        l1 = l1->next;
+        l2 = l2->next;
+    }
+    if(l1||l2)
+        return 0;
+    return 1;
+}
+
+namespace_set_t* namespace_set_clone(namespace_set_t*other)
+{
+    NEW(namespace_set_t,set);
+    set->namespaces = list_new();
+    namespace_list_t*l = other->namespaces;
+    while(l) {
+        list_append(set->namespaces, namespace_clone(l->namespace));
+        l = l->next;
+    }
+    return set;
+}
+namespace_set_t* namespace_set_new()
+{
+    NEW(namespace_set_t,set);
+    set->namespaces = list_new();
+    return set;
+}
 char* namespace_set_to_string(namespace_set_t*set)
 {
     /* TODO: is the order of the namespaces important (does it
@@ -114,98 +275,179 @@ char* namespace_set_to_string(namespace_set_t*set)
     return desc;
 }
 
-
-int pool_register_string(pool_t*pool, const char*s)
+void namespace_set_destroy(namespace_set_t*set)
 {
-    return array_append_if_new(pool->strings, s, 0);
+    namespace_list_t*l = set->namespaces;
+    while(l) {
+        namespace_destroy(l->namespace);l->namespace=0;
+        l = l->next;
+    }
+    list_free(set->namespaces);
+    free(set);
 }
 
-int pool_register_namespace(pool_t*pool, namespace_t*ns)
-{
-    if(!ns)
-        return 0;
-    char*name = namespace_to_string(ns);
-    int i = array_append_if_new(pool->namespaces, name, ns);
-    free(name);
-    return i;
-}
-int pool_register_namespace_set(pool_t*pool, namespace_set_t*set)
+type_t namespace_set_type = {
+    dup: (dup_func)namespace_set_clone,
+    hash: (hash_func)namespace_set_hash,
+    free: (free_func)namespace_set_destroy,
+    equals: (equals_func)namespace_set_equals
+};
+
+// ----------------------------- multiname ----------------------------------
+
+unsigned int multiname_hash(multiname_t*m)
 {
-    char*name = namespace_set_to_string(set);
-    int i = array_append_if_new(pool->namespace_sets, name, set);
-    free(name);
-    return i;
+    unsigned int hash = crc32_add_byte(0, m->type);
+    if(m->name) {
+        hash = crc32_add_string(hash, m->name);
+    }
+    if(m->ns) {
+        hash = crc32_add_byte(hash, m->ns->access);
+        hash = crc32_add_string(hash, m->ns->name);
+    }
+    if(m->namespace_set) {
+        namespace_list_t*l = m->namespace_set->namespaces;
+        while(l) {
+            hash = crc32_add_byte(hash, l->namespace->access);
+            hash = crc32_add_string(hash, l->namespace->name);
+            l = l->next;
+        }
+    }
+    return hash;
 }
-int pool_register_multiname(pool_t*pool, multiname_t*n)
+
+int multiname_equals(multiname_t*m1, multiname_t*m2)
 {
-    if(!n)
+    if(m1->type!=m2->type)
         return 0;
-    char*name = multiname_to_string(n);
-    int i =array_append_if_new(pool->multinames, name, n);
-    free(name);
-    return i;
-}
-int pool_register_multiname2(pool_t*pool, char*name)
-{
-    multiname_t*n = multiname_fromstring(name);
-    int i =array_append_if_new(pool->multinames, name, n);
-    return i;
-}
-int pool_find_namespace(pool_t*pool, namespace_t*ns)
-{
-    if(!ns)
+
+    if((!m1->name) != (!m2->name))
         return 0;
-    char*name = namespace_to_string(ns);
-    int i = array_find(pool->namespaces, name);
-    if(i<0) {
-        fprintf(stderr, "Couldn't find namespace \"%s\" %08x in constant pool\n", name, ns);
+    if((!m1->ns) != (!m2->ns))
         return 0;
+    if((!m1->namespace_set) != (!m2->namespace_set))
+        return 0;
+
+    if(m1->name && m2->name && strcmp(m1->name,m2->name))
+        return 0;
+    if(m1->ns && m2->ns) {
+        if(!namespace_equals(m1->ns, m2->ns))
+            return 0;
     }
-    free(name);
-    return i;
+    if(m1->namespace_set && m2->namespace_set) {
+        if(!namespace_set_equals(m1->namespace_set, m2->namespace_set))
+            return 0;
+    }
+    return 1;
 }
-int pool_find_namespace_set(pool_t*pool, namespace_set_t*set)
+
+multiname_t* multiname_new(namespace_t*ns, const char*name)
 {
-    char*name = namespace_set_to_string(set);
-    int i = array_find(pool->namespace_sets, name);
-    free(name);
-    if(i<0) {
-        fprintf(stderr, "Couldn't find namespace_set \"%s\" in constant pool\n", name);
-        return 0;
-    }
-    return i;
+    NEW(multiname_t,m);
+    m->type = QNAME;
+    m->ns = namespace_clone(ns);
+    m->name = strdup(name);
+    return m;
 }
-int pool_find_string(pool_t*pool, const char*s)
+
+multiname_t* multiname_clone(multiname_t*other)
 {
-    int i = array_find(pool->strings, s);
-    if(i<0) {
-        fprintf(stderr, "Couldn't find string \"%s\" in constant pool\n", s);
-        *(int*)0=0;
-        return 0;
-    }
-    return i;
+    NEW(multiname_t,m);
+    m->type = other->type;
+    if(other->ns)
+        m->ns = namespace_clone(other->ns);
+    if(other->namespace_set)
+        m->namespace_set = namespace_set_clone(other->namespace_set);
+    if(other->name)
+        m->name = strdup(other->name);
+    return m;
 }
-int pool_find_multiname(pool_t*pool, multiname_t*name)
+
+
+char* access2str(int type)
 {
-    char*n = multiname_to_string(name);
-    int i = array_find(pool->multinames, n);
-    if(i<0) {
-        fprintf(stderr, "Couldn't find multiname \"%s\" in constant pool\n", n);
-        return 0;
+    if(type==0x08) return "access08";
+    else if(type==0x16) return "package";
+    else if(type==0x17) return "packageinternal";
+    else if(type==0x18) return "protected";
+    else if(type==0x19) return "explicit";
+    else if(type==0x1A) return "staticprotected";
+    else if(type==0x05) return "private";
+    else {
+        fprintf(stderr, "Undefined access type %02x\n", type);
+        return "undefined";
     }
-    return i;
 }
 
-multiname_t*pool_lookup_multiname(pool_t*pool, int i)
+char* multiname_to_string(multiname_t*m)
 {
-    multiname_t*m = (multiname_t*)array_getvalue(pool->multinames, i);
-    return m;
+    char*mname = 0;
+    if(!m || m->type==0xff)
+        return strdup("--<UNDEFINED_MULTINAME>--");
+
+    int namelen = m->name?strlen(m->name):1;
+
+    if(m->type==QNAME || m->type==QNAMEA) {
+        mname = malloc(strlen(m->ns->name)+namelen+32);
+        strcpy(mname, "<q");
+        if(m->type == QNAMEA)
+            strcat(mname, ",attr");
+        strcat(mname, ">[");
+        strcat(mname,access2str(m->ns->access));
+        strcat(mname, "]");
+        strcat(mname, m->ns->name);
+        strcat(mname, "::");
+        if(m->name)
+            strcat(mname, m->name);
+        else
+            strcat(mname, "*");
+    } else if(m->type==RTQNAME || m->type==RTQNAMEA) {
+        mname = malloc(namelen+32);
+        strcpy(mname, "<rt");
+        if(m->type == RTQNAMEA) 
+            strcat(mname, ",attr");
+        strcat(mname, ">");
+        if(m->name)
+            strcat(mname, m->name);
+        else
+            strcat(mname, "*");
+    } else if(m->type==RTQNAMEL) {
+        mname = strdup("<l>");
+    } else if(m->type==RTQNAMELA) {
+        mname = strdup("<l,attr>");
+    } else if(m->type==MULTINAME || m->type==MULTINAMEA) {
+        char*s = namespace_set_to_string(m->namespace_set);
+        mname = malloc(strlen(s)+namelen+16);
+        if(m->type == MULTINAME)
+            strcpy(mname,"<multi>");
+        else //MULTINAMEA
+            strcpy(mname,"<multi,attr>");
+        strcat(mname, s);
+        strcat(mname, "::");
+        if(m->name)
+            strcat(mname, m->name);
+        else
+            strcat(mname, "*");
+        free(s);
+    } else if(m->type==MULTINAMEL || m->type==MULTINAMELA) {
+        char*s = namespace_set_to_string(m->namespace_set);
+        mname = malloc(strlen(s)+16);
+        if(m->type == MULTINAMEL)
+            strcpy(mname,"<l,multi>");
+        else //MULTINAMELA
+            strcpy(mname,"<l,multi,attr>");
+        strcat(mname,s);
+        free(s);
+    } else {
+        fprintf(stderr, "Invalid multiname type: %02x\n", m->type);
+    }
+    return mname;
 }
 
 multiname_t* multiname_fromstring(const char*name2)
 {
     if(!name2)
-       name2 = "::";
+        return 0;
     char*n = strdup(name2);
     char*p = strstr(n, "::");
     char*namespace=0,*name=0;
@@ -238,29 +480,212 @@ multiname_t* multiname_fromstring(const char*name2)
     return m;
 }
 
+void multiname_destroy(multiname_t*m)
+{
+    if(m->name) {
+        free((void*)m->name);m->name = 0;
+    }
+    if(m->ns) {
+        namespace_destroy(m->ns);m->ns = 0;
+    }
+    if(m->namespace_set) {
+        namespace_set_destroy(m->namespace_set);m->namespace_set = 0;
+    }
+    free(m);
+}
+
+type_t multiname_type = {
+    dup: (dup_func)multiname_clone,
+    hash: (hash_func)multiname_hash,
+    free: (free_func)multiname_destroy,
+    equals: (equals_func)multiname_equals
+};
+
+// ------------------------------- pool -------------------------------------
+
+int pool_register_uint(pool_t*p, unsigned int i)
+{
+    return array_append_if_new(p->x_uints, (void*)(ptroff_t)i, 0);
+}
+int pool_register_int(pool_t*p, int i)
+{
+    return array_append_if_new(p->x_ints, (void*)(ptroff_t)i, 0);
+}
+int pool_register_float(pool_t*p, double d)
+{
+    return array_append_if_new(p->x_floats, &d, 0);
+}
+int pool_register_string(pool_t*pool, const char*s)
+{
+    if(!s) return 0;
+    return array_append_if_new(pool->x_strings, s, 0);
+}
+int pool_register_namespace(pool_t*pool, namespace_t*ns)
+{
+    if(!ns) return 0;
+    return array_append_if_new(pool->x_namespaces, ns, 0);
+}
+int pool_register_namespace_set(pool_t*pool, namespace_set_t*set)
+{
+    if(!set) return 0;
+    return array_append_if_new(pool->x_namespace_sets, set, 0);
+}
+int pool_register_multiname(pool_t*pool, multiname_t*n)
+{
+    if(!n) return 0;
+    return array_append_if_new(pool->x_multinames, n, 0);
+}
+int pool_register_multiname2(pool_t*pool, char*name)
+{
+    if(!name) return 0;
+    multiname_t*n = multiname_fromstring(name);
+    return array_append_if_new(pool->x_multinames, n, 0);
+}
+
+
+int pool_find_uint(pool_t*pool, unsigned int x)
+{
+    int i = array_find(pool->x_uints, (void*)(ptroff_t)x);
+    if(i<0) {
+        fprintf(stderr, "Couldn't find uint \"%d\" in constant pool\n", x);
+        return 0;
+    }
+    return i;
+}
+int pool_find_int(pool_t*pool, int x)
+{
+    int i = array_find(pool->x_ints, (void*)(ptroff_t)x);
+    if(i<0) {
+        fprintf(stderr, "Couldn't find int \"%d\" in constant pool\n", x);
+        return 0;
+    }
+    return i;
+}
+int pool_find_float(pool_t*pool, double x)
+{
+    int i = array_find(pool->x_ints, &x);
+    if(i<0) {
+        fprintf(stderr, "Couldn't find int \"%d\" in constant pool\n", x);
+        return 0;
+    }
+    return i;
+}
+int pool_find_namespace(pool_t*pool, namespace_t*ns)
+{
+    if(!ns)
+        return 0;
+    int i = array_find(pool->x_namespaces, ns);
+    if(i<0) {
+        char*s = namespace_to_string(ns);
+        fprintf(stderr, "Couldn't find namespace \"%s\" %08x in constant pool\n", s, ns);
+        free(s);
+        return 0;
+    }
+    return i;
+}
+int pool_find_namespace_set(pool_t*pool, namespace_set_t*set)
+{
+    int i = array_find(pool->x_namespace_sets, set);
+    if(i<0) {
+        char*s = namespace_set_to_string(set);
+        fprintf(stderr, "Couldn't find namespace_set \"%s\" in constant pool\n", s);
+        free(s);
+        return 0;
+    }
+    return i;
+}
+int pool_find_string(pool_t*pool, const char*s)
+{
+    int i = array_find(pool->x_strings, s);
+    if(i<0) {
+        fprintf(stderr, "Couldn't find string \"%s\" in constant pool\n", s);
+        *(int*)0=0;
+        return 0;
+    }
+    return i;
+}
+int pool_find_multiname(pool_t*pool, multiname_t*name)
+{
+    int i = array_find(pool->x_multinames, name);
+    if(i<0) {
+        char*s = multiname_to_string(name);
+        fprintf(stderr, "Couldn't find multiname \"%s\" in constant pool\n", s);
+        free(s);
+        return 0;
+    }
+    return i;
+}
+
+int pool_lookup_int(pool_t*pool, int i)
+{
+    return *(int*)array_getkey(pool->x_ints, i);
+}
+unsigned int pool_lookup_uint(pool_t*pool, int i)
+{
+    return *(unsigned int*)array_getkey(pool->x_uints, i);
+}
+double pool_lookup_float(pool_t*pool, int i)
+{
+    return *(double*)array_getkey(pool->x_floats, i);
+}
+char*pool_lookup_string(pool_t*pool, int i)
+{
+    return (char*)array_getkey(pool->x_strings, i);
+}
+namespace_t*pool_lookup_namespace(pool_t*pool, int i)
+{
+    return (namespace_t*)array_getkey(pool->x_namespaces, i);
+}
+namespace_set_t*pool_lookup_namespace_set(pool_t*pool, int i)
+{
+    return (namespace_set_t*)array_getkey(pool->x_namespace_sets, i);
+}
+multiname_t*pool_lookup_multiname(pool_t*pool, int i)
+{
+    return (multiname_t*)array_getkey(pool->x_multinames, i);
+}
+
 pool_t*pool_new()
 {
     NEW(pool_t, p);
 
-    p->ints = array_new();
-    array_append(p->ints, 0, (void*)(ptroff_t)0);
-    p->uints = array_new();
-    array_append(p->uints, 0, (void*)(ptroff_t)0);
-    p->floats = array_new();
-    array_append(p->floats, 0, 0);
-    p->strings = array_new();
-    array_append(p->strings, "--<UNDEFINED_STRING>--", 0);
-    p->namespaces = array_new();
-    array_append(p->namespaces, "--<UNDEFINED_NAMESPACE>--", 0);
-    p->namespace_sets = array_new();
-    array_append(p->namespace_sets, "--<UNDEFINED_NSSET>--", 0);
-    p->multinames = array_new();
-    array_append(p->multinames, "--<UNDEFINED_MULTINAME>--", 0);
+    p->x_ints = array_new2(&uint_type);
+    p->x_uints = array_new2(&uint_type);
+    p->x_floats = array_new2(&float_type);
+    p->x_strings = array_new2(&charptr_type);
+    p->x_namespaces = array_new2(&namespace_type);
+    p->x_namespace_sets = array_new2(&namespace_set_type);
+    p->x_multinames = array_new2(&multiname_type);
+
+    /* add a zero-index entry in each list */
+  
+    array_append(p->x_ints, &undefined_uint, 0);
+    array_append(p->x_uints, &undefined_uint, 0);
+    array_append(p->x_floats, &undefined_float, 0);
+
+    pool_register_string(p, "--<UNDEFINED_STRING>--");
+
+    namespace_t*ns = namespace_new(0,"--<UNDEFINED NAMESPACE>--");
+    pool_register_namespace(p, ns);
+    namespace_destroy(ns);
+
+    namespace_set_t*nsset = namespace_set_new();
+    list_append(nsset->namespaces, namespace_new(0, "--<UNDEFINED NAMESPACE SET>--"));
+    pool_register_namespace_set(p, nsset);
+    namespace_set_destroy(nsset);
+
+    namespace_t*mns = namespace_new(0,"nons");
+    multiname_t*mname = multiname_new(mns,"--<UNDEFINED MULTINAME>--");
+    mname->type = 0xff;
+    pool_register_multiname(p, mname);
+    multiname_destroy(mname);
+    namespace_destroy(mns);
 
     return p;
 }
 
 #define DEBUG if(0)
+//#define DEBUG
 
 void pool_read(pool_t*pool, TAG*tag)
 {
@@ -270,7 +695,7 @@ void pool_read(pool_t*pool, TAG*tag)
     for(t=1;t<num_ints;t++) {
         S32 v = swf_GetS30(tag);
         DEBUG printf("int %d) %d\n", t, v);
-        array_append(pool->ints, 0, (void*)(ptroff_t)v);
+        array_append(pool->x_ints, (void*)(ptroff_t)v, 0);
     }
 
     int num_uints = swf_GetU30(tag);
@@ -278,7 +703,7 @@ void pool_read(pool_t*pool, TAG*tag)
     for(t=1;t<num_uints;t++) {
         U32 v = swf_GetU30(tag);
         DEBUG printf("uint %d) %d\n", t, v);
-        array_append(pool->uints, 0, (void*)(ptroff_t)v);
+        array_append(pool->x_uints, (void*)(ptroff_t)v, 0);
     }
     
     int num_floats = swf_GetU30(tag);
@@ -286,7 +711,7 @@ void pool_read(pool_t*pool, TAG*tag)
     for(t=1;t<num_floats;t++) {
         double d = swf_GetD64(tag);
         DEBUG printf("float %d) %f\n", t, d);
-        array_append(pool->floats, 0, 0);
+        array_append(pool->x_floats, &d, 0);
     }
     
     int num_strings = swf_GetU30(tag);
@@ -296,20 +721,20 @@ void pool_read(pool_t*pool, TAG*tag)
        char*s = malloc(len+1);
        swf_GetBlock(tag, s, len);
        s[len] = 0;
-       array_append(pool->strings, s, 0);
-       DEBUG printf("%d) \"%s\"\n", t, pool->strings->d[t].name);
+       array_append(pool->x_strings, s, 0);
+        free(s);
+       DEBUG printf("%d) \"%s\"\n", t, pool->x_strings->d[t].name);
     }
     int num_namespaces = swf_GetU30(tag);
     DEBUG printf("%d namespaces\n", num_namespaces);
     for(t=1;t<num_namespaces;t++) {
        U8 type = swf_GetU8(tag);
        int namenr = swf_GetU30(tag);
-       const char*name = array_getkey(pool->strings, namenr);
-        NEW(namespace_t, ns);
-        memset(ns, 0, sizeof(namespace_t));
-        ns->access = type;
-        ns->name = strdup(name);
-       array_append(pool->namespaces, name, ns);
+       const char*name = array_getkey(pool->x_strings, namenr);
+        namespace_t*ns = namespace_new(type, name);
+       array_append(pool->x_namespaces, ns, 0);
+       DEBUG printf("%d) \"%s\"\n", t, namespace_to_string(ns));
+        namespace_destroy(ns);
     }
     int num_sets = swf_GetU30(tag);
     DEBUG printf("%d namespace sets\n", num_sets);
@@ -320,74 +745,162 @@ void pool_read(pool_t*pool, TAG*tag)
         NEW(namespace_set_t, nsset);
         for(s=0;s<count;s++) {
             int nsnr = swf_GetU30(tag);
-            namespace_t*ns = (namespace_t*)array_getvalue(pool->namespaces, nsnr);
-            list_append(nsset->namespaces, ns);
+            namespace_t*ns = (namespace_t*)array_getkey(pool->x_namespaces, nsnr);
+            list_append(nsset->namespaces, namespace_clone(ns));
         }
-        char*desc = namespace_set_to_string(nsset);
-
-        array_append(pool->namespace_sets, desc, nsset);
-        DEBUG printf("set %d) %s\n", t, desc);
+        array_append(pool->x_namespace_sets, nsset, 0);
+        DEBUG printf("set %d) %s\n", t, namespace_set_to_string(nsset));
+        namespace_set_destroy(nsset);
     }
 
     int num_multinames = swf_GetU30(tag);
     DEBUG printf("%d multinames\n", num_multinames);
     for(t=1;t<num_multinames;t++) {
-        
-        NEW(multiname_t,m);
-       m->type = swf_GetU8(tag);
-       if(m->type==0x07 || m->type==0x0d) {
-           int namespace_index = swf_GetU30(tag); // 0 = '*' (any)
-            m->ns = (namespace_t*)array_getvalue(pool->namespaces, namespace_index);
+        multiname_t m;
+        memset(&m, 0, sizeof(multiname_t));
+       m.type = swf_GetU8(tag);
+       if(m.type==0x07 || m.type==0x0d) {
+           int namespace_index = swf_GetU30(tag);
+            m.ns = (namespace_t*)array_getkey(pool->x_namespaces, namespace_index);
             int name_index = swf_GetU30(tag);
-           m->name = array_getkey(pool->strings, name_index);
-       } else if(m->type==0x0f || m->type==0x10) {
-            int name_index = swf_GetU30(tag); // 0 = '*' (any name)
-           m->name = array_getkey(pool->strings, name_index);
-       } else if(m->type==0x11 || m->type==0x12) {
-       } else if(m->type==0x09 || m->type==0x0e) {
+            if(name_index) // 0 = '*' (any)
+               m.name = array_getkey(pool->x_strings, name_index);
+       } else if(m.type==0x0f || m.type==0x10) {
+            int name_index = swf_GetU30(tag);
+            if(name_index) // 0 = '*' (any name)
+               m.name = array_getkey(pool->x_strings, name_index);
+       } else if(m.type==0x11 || m.type==0x12) {
+       } else if(m.type==0x09 || m.type==0x0e) {
             int name_index = swf_GetU30(tag);
             int namespace_set_index = swf_GetU30(tag);
-           m->name = array_getkey(pool->strings, name_index);
-           m->namespace_set = (namespace_set_t*)array_getvalue(pool->namespace_sets, namespace_set_index);
-        } else if(m->type==0x1b || m->type==0x1c) {
+            if(name_index)
+               m.name = array_getkey(pool->x_strings, name_index);
+           m.namespace_set = (namespace_set_t*)array_getkey(pool->x_namespace_sets, namespace_set_index);
+        } else if(m.type==0x1b || m.type==0x1c) {
             int namespace_set_index = swf_GetU30(tag);
-           m->namespace_set = (namespace_set_t*)array_getvalue(pool->namespace_sets, namespace_set_index);
+           m.namespace_set = (namespace_set_t*)array_getkey(pool->x_namespace_sets, namespace_set_index);
        } else {
-           printf("can't parse type %d multinames yet\n", m->type);
+           printf("can't parse type %d multinames yet\n", m.type);
        }
-       char*mname = multiname_to_string(m);
-        DEBUG printf("multiname %d) %s\n", t, mname);
-       array_append(pool->multinames, mname, m);
-       free(mname);
+        DEBUG printf("multiname %d) %s\n", t, multiname_to_string(&m));
+       array_append(pool->x_multinames, &m, 0);
     }
 } 
 
-void pool_destroy(pool_t*pool)
+void pool_write(pool_t*pool, TAG*tag)
 {
     int t;
-    array_free(pool->ints);
-    array_free(pool->uints);
-    array_free(pool->floats);
+    
+    /* make sure that all namespaces used by multinames / namespace sets
+       and all strings used by namespaces exist */
 
-    for(t=1;t<pool->strings->num;t++) {
-        free((void*)array_getkey(pool->strings, t));
+    for(t=1;t<pool->x_multinames->num;t++) {
+        multiname_t*m = (multiname_t*)array_getkey(pool->x_multinames, t);
+        if(m->ns) {
+            pool_register_namespace(pool, m->ns);
+        }
+        if(m->namespace_set) {
+            pool_register_namespace_set(pool, m->namespace_set);
+        }
+        if(m->name) {
+            pool_register_string(pool, m->name);
+        }
+    }
+    for(t=1;t<pool->x_namespace_sets->num;t++) {
+        namespace_set_t*set = (namespace_set_t*)array_getkey(pool->x_namespace_sets, t);
+        namespace_list_t*i = set->namespaces;
+        while(i) {
+            pool_register_namespace(pool, i->namespace);
+            i = i->next;
+        }
+    }
+    for(t=1;t<pool->x_namespaces->num;t++) {
+       namespace_t*ns= (namespace_t*)array_getkey(pool->x_namespaces, t);
+        array_append_if_new(pool->x_strings, ns->name, 0);
     }
-    array_free(pool->strings);
 
-    for(t=1;t<pool->namespaces->num;t++) {
-        free(array_getvalue(pool->namespaces, t));
+    /* write data */
+    swf_SetU30(tag, pool->x_ints->num>1?pool->x_ints->num:0);
+    for(t=1;t<pool->x_ints->num;t++) {
+        S32 val = (ptroff_t)array_getkey(pool->x_ints, t);
+        swf_SetS30(tag, val);
+    }
+    swf_SetU30(tag, pool->x_uints->num>1?pool->x_uints->num:0);
+    for(t=1;t<pool->x_uints->num;t++) {
+        swf_SetU30(tag, (ptroff_t)array_getkey(pool->x_uints, t));
+    }
+    swf_SetU30(tag, pool->x_floats->num>1?pool->x_floats->num:0);
+    for(t=1;t<pool->x_floats->num;t++) {
+        array_getvalue(pool->x_floats, t);
+        swf_SetD64(tag, 0.0); // fixme
     }
-    array_free(pool->namespaces);
-   
-    for(t=1;t<pool->namespace_sets->num;t++) {
-        namespace_set_t*set = (namespace_set_t*)array_getvalue(pool->namespace_sets, t);
-        list_free(set->namespaces);
-        free(set);
+    swf_SetU30(tag, pool->x_strings->num>1?pool->x_strings->num:0);
+    for(t=1;t<pool->x_strings->num;t++) {
+       swf_SetU30String(tag, array_getkey(pool->x_strings, t));
     }
-    array_free(pool->namespace_sets);
+    swf_SetU30(tag, pool->x_namespaces->num>1?pool->x_namespaces->num:0);
+    for(t=1;t<pool->x_namespaces->num;t++) {
+       namespace_t*ns= (namespace_t*)array_getkey(pool->x_namespaces, t);
+       const char*name = ns->name;
+       int i = pool_find_string(pool, name);
+       swf_SetU8(tag, ns->access);
+       swf_SetU30(tag, i);
+    }
+    swf_SetU30(tag, pool->x_namespace_sets->num>1?pool->x_namespace_sets->num:0);
+    for(t=1;t<pool->x_namespace_sets->num;t++) {
+        namespace_set_t*set = (namespace_set_t*)array_getkey(pool->x_namespace_sets, t);
+        namespace_list_t*i = set->namespaces; 
+        int len = list_length(i);
+        swf_SetU30(tag, len);
+        while(i) {
+            int index = pool_find_namespace(pool, i->namespace);
+            swf_SetU30(tag, index);
+            i = i->next;
+        }
+    }
+
+    swf_SetU30(tag, pool->x_multinames->num>1?pool->x_multinames->num:0);
+    for(t=1;t<pool->x_multinames->num;t++) {
+       multiname_t*m = (multiname_t*)array_getkey(pool->x_multinames, t);
+       swf_SetU8(tag, m->type);
 
-    for(t=1;t<pool->multinames->num;t++) {
-        free(array_getvalue(pool->multinames, t));
+        if(m->ns) {
+            assert(m->type==0x07 || m->type==0x0d);
+           int i = pool_find_namespace(pool, m->ns);
+            if(i<0) fprintf(stderr, "internal error: unregistered namespace %02x %s %s\n", m->ns->access, access2str(m->ns->access), m->ns->name);
+           swf_SetU30(tag, i);
+        } else {
+            assert(m->type!=0x07 && m->type!=0x0d);
+        }
+        if(m->name) {
+            assert(m->type==0x09 || m->type==0x0e || m->type==0x07 || m->type==0x0d || m->type==0x0f || m->type==0x10);
+           int i = pool_find_string(pool, m->name);
+            if(i<0) fprintf(stderr, "internal error: unregistered name\n");
+           swf_SetU30(tag, i);
+        } else {
+            assert(m->type!=0x09 && m->type!=0x0e && m->type!=0x07 && m->type!=0x0d && m->type!=0x0f && m->type!=0x10);
+        }
+        if(m->namespace_set) {
+            assert(m->type==0x09 || m->type==0x0e || m->type==0x1c || m->type==0x1b);
+           int i = pool_find_namespace_set(pool, m->namespace_set);
+            if(i<0) fprintf(stderr, "internal error: unregistered namespace set\n");
+           swf_SetU30(tag, i);
+        } else {
+            assert(m->type!=0x09 && m->type!=0x0e && m->type!=0x1c && m->type!=0x1b);
+        }
     }
-    array_free(pool->multinames);
+}
+
+
+void pool_destroy(pool_t*pool)
+{
+    int t;
+    array_free(pool->x_ints);
+    array_free(pool->x_uints);
+    array_free(pool->x_floats);
+    array_free(pool->x_strings);
+    array_free(pool->x_namespaces);
+    array_free(pool->x_namespace_sets);
+    array_free(pool->x_multinames);
+    free(pool);
 }
index db91e70..1c30062 100644 (file)
@@ -37,13 +37,13 @@ DECLARE_LIST(trait);
 
 /* abc file constant pool */
 struct _pool {
-    array_t*ints;
-    array_t*uints;
-    array_t*floats;
-    array_t*strings;
-    array_t*namespaces;
-    array_t*namespace_sets;
-    array_t*multinames;
+    array_t*x_ints;
+    array_t*x_uints;
+    array_t*x_floats;
+    array_t*x_strings;
+    array_t*x_namespaces;
+    array_t*x_namespace_sets;
+    array_t*x_multinames;
 };
 
 typedef enum multiname_type
@@ -68,6 +68,8 @@ struct _namespace {
 struct _namespace_set {
     namespace_list_t*namespaces;
 };
+
+extern type_t multiname_type;
 struct _multiname {
     multiname_type_t type;
     namespace_t*ns;
@@ -81,9 +83,18 @@ char* multiname_to_string(multiname_t*m);
 char* namespace_to_string(namespace_t*ns);
 
 /* integer -> object */
+int pool_lookup_int(pool_t*pool, int i);
+unsigned int pool_lookup_uint(pool_t*pool, int i);
+double pool_lookup_float(pool_t*pool, int i);
+char*pool_lookup_string(pool_t*pool, int i);
+namespace_t*pool_lookup_namespace(pool_t*pool, int i);
+namespace_set_t*pool_lookup_namespace_set(pool_t*pool, int i);
 multiname_t*pool_lookup_multiname(pool_t*pool, int i);
 
 /* object -> integer (lookup) */
+int pool_find_int(pool_t*pool, int x);
+int pool_find_uint(pool_t*pool, unsigned int x);
+int pool_find_double(pool_t*pool, double x);
 int pool_find_namespace(pool_t*pool, namespace_t*ns);
 int pool_find_namespace_set(pool_t*pool, namespace_set_t*set);
 int pool_find_string(pool_t*pool, const char*s);
@@ -97,10 +108,27 @@ int pool_register_multiname(pool_t*pool, multiname_t*n);
 int pool_register_multiname2(pool_t*pool, char*name);
 
 /* creation */
+namespace_t* namespace_new(U8 access, const char*name);
+namespace_t* namespace_new_package(const char*name);
+namespace_t* namespace_new_packageinternal(const char*name);
+namespace_t* namespace_new_protected(const char*name);
+namespace_t* namespace_new_explicit(const char*name);
+namespace_t* namespace_new_staticprotected(const char*name);
+namespace_t* namespace_new_private(const char*name);
+namespace_t* namespace_clone(namespace_t*ns);
+
 multiname_t* multiname_fromstring(const char*name);
+multiname_t* multiname_new(namespace_t*ns, const char*name);
+multiname_t* multiname_clone(multiname_t*other);
+
+/* freeing */
+void multiname_destroy(multiname_t*m);
+void namespace_destroy(namespace_t*n);
 
+/* pool constructors/destructors */
 pool_t*pool_new();
 void pool_read(pool_t*pool, TAG*tag);
+void pool_write(pool_t*pool, TAG*tag);
 void pool_destroy(pool_t*pool);
 
 #endif