From 8e13c34f9f544dc16ae479da65d9d82140a2e8c2 Mon Sep 17 00:00:00 2001 From: kramm Date: Sun, 23 Nov 2008 13:57:48 +0000 Subject: [PATCH] added hashing for everything --- lib/as3/pool.c | 881 ++++++++++++++++++++++++++++++++++++++++++++------------ lib/as3/pool.h | 42 ++- 2 files changed, 732 insertions(+), 191 deletions(-) diff --git a/lib/as3/pool.c b/lib/as3/pool.c index fc92648..ccb6ac7 100644 --- a/lib/as3/pool.c +++ b/lib/as3/pool.c @@ -21,68 +21,229 @@ along with this program; if not, write to the Free Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */ +#include #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("----"); - 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("----"); + + 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, "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, "type == RTQNAMEA) + strcat(mname, ",attr"); + strcat(mname, ">"); + if(m->name) + strcat(mname, m->name); + else + strcat(mname, "*"); + } else if(m->type==RTQNAMEL) { + mname = strdup(""); + } else if(m->type==RTQNAMELA) { + mname = strdup(""); + } 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,""); + else //MULTINAMEA + strcpy(mname,""); + 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,""); + else //MULTINAMELA + strcpy(mname,""); + 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, "----", 0); - p->namespaces = array_new(); - array_append(p->namespaces, "----", 0); - p->namespace_sets = array_new(); - array_append(p->namespace_sets, "----", 0); - p->multinames = array_new(); - array_append(p->multinames, "----", 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, "----"); + + namespace_t*ns = namespace_new(0,"----"); + pool_register_namespace(p, ns); + namespace_destroy(ns); + + namespace_set_t*nsset = namespace_set_new(); + list_append(nsset->namespaces, namespace_new(0, "----")); + pool_register_namespace_set(p, nsset); + namespace_set_destroy(nsset); + + namespace_t*mns = namespace_new(0,"nons"); + multiname_t*mname = multiname_new(mns,"----"); + 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;tints, 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;tuints, 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;tfloats, 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;tstrings, 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;snamespaces, 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;ttype = 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;tstrings->num;t++) { - free((void*)array_getkey(pool->strings, t)); + for(t=1;tx_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;tx_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;tx_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;tnamespaces->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;tx_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;tx_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;tx_floats->num;t++) { + array_getvalue(pool->x_floats, t); + swf_SetD64(tag, 0.0); // fixme } - array_free(pool->namespaces); - - for(t=1;tnamespace_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;tx_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;tx_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;tx_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;tx_multinames->num;t++) { + multiname_t*m = (multiname_t*)array_getkey(pool->x_multinames, t); + swf_SetU8(tag, m->type); - for(t=1;tmultinames->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); } diff --git a/lib/as3/pool.h b/lib/as3/pool.h index db91e70..1c30062 100644 --- a/lib/as3/pool.h +++ b/lib/as3/pool.h @@ -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 -- 1.7.10.4