added action enumeration routines
[swftools.git] / lib / modules / swfaction.c
index 8609686..2338a43 100644 (file)
@@ -26,11 +26,12 @@ f: frame (word)
 u: url (string)
 t: target (string)
 l: label (string)
-c: constant pool (string)
+C: constant pool header (byte)
+c: constant pool entry (string)
 s: skip (byte) (number of actions)
-m: method (byte) url:(0=none, 1=get, 2=post)/gf2:(1=play)
+m: method (byte) GetUrl2:(0=none, 1=get, 2=post)/GotoFrame2:(1=play)
 b: branch (word) (number of bytes)
-p: type(byte), type=0:string, type=1:double
+p (push): type(byte), type=0:string, type=1:double
 {: define function (name (string), num (word), params (num strings), codesize (word)
 o: object (string)
 r: register (byte)
@@ -106,7 +107,7 @@ r: register (byte)
 {5,"GetMember", 0x4e,""},
 {5,"SetMember", 0x4f,""},
 {5,"CallMethod", 0x52,""},
-{5,"Constantpool", 0x88, "c"},
+{5,"Constantpool", 0x88, "Cc"},
 {5,"DefineLocal", 0x3c,""},
 {5,"DefineLocal2", 0x41,""},
 {5,"Makehash", 0x43, ""}, //??
@@ -178,6 +179,53 @@ void SetActions(TAG*tag, ActionTAG*action)
     }
 }
 
+int OpAdvance(char c, char*data)
+{
+    switch (c)
+    {
+       case 'f':
+           return 2;
+       case 'u':
+           return strlen(data)+1;
+       case 't':
+           return strlen(data)+1;
+       case 'l': 
+           return strlen(data)+1;
+       case 'c': 
+           return strlen(data)+1;
+       case 'C': 
+           return 2;
+       case 's':
+           return 1;
+       case 'm':
+           return 1;
+       case 'b':
+           return 2;
+       case 'p': {
+           U8 type = *data++;
+           if(type == 0) {
+               return 1+strlen(data)+1; //string
+           } else if (type == 1) {
+               return 1+4; //float
+           } else if (type == 2) {
+               return 1+0; //NULL
+           } else if (type == 4) {
+               return 1+1; //register
+           } else if (type == 5) {
+               return 1+1; //bool
+           } else if (type == 6) {
+               return 1+8; //double
+           } else if (type == 7) {
+               return 1+4; //int
+           } else if (type == 8) {
+               return 1+1; //lookup
+           }
+           break;
+       }
+    }
+    return 0;
+}
+
 /* TODO: this should be in swfdump.c */
 void DumpActions(ActionTAG*atag, char*prefix) 
 {
@@ -189,6 +237,7 @@ void DumpActions(ActionTAG*atag, char*prefix)
        prefix="";
     while(atag)
     {
+       U8 poollen = 0;
        for(t=0;t<definedactions;t++)
            if(actions[t].op == atag->op)
                break;
@@ -207,53 +256,61 @@ void DumpActions(ActionTAG*atag, char*prefix)
            switch(*cp)
            {
                case 'f': {
-                   printf(" %d", *(U16*)data);data+=2; //FIXME: le/be
+                   printf(" %d", *(U16*)data); //FIXME: le/be
                } break;
                case 'u': {
-                   printf(" URL:\"%s\"", data);data+=strlen(data)+1;
+                   printf(" URL:\"%s\"", data);
                } break;
                case 't': {
-                   printf(" Target:\"%s\"", data);data+=strlen(data)+1;
+                   printf(" Target:\"%s\"", data);
                } break;
                case 'l': {
-                   printf(" Label:\"%s\"", data);data+=strlen(data)+1;
+                   printf(" Label:\"%s\"", data);
                } break;
                case 'c': {
-                   printf(" Constant Pool:\"%s\"", data);data+=strlen(data)+1;
+                   printf(" String:\"%s\"", data);
+               } break;
+               case 'C': {
+                   poollen = *data;
+                   printf("(%d entries)", poollen);
                } break;
                case 's': {
-                   printf(" +%d", data);data++;
+                   printf(" +%d", data);
                } break;
                case 'm': {
-//m: method (byte) url:(0=none, 1=get, 2=datat)/gf2:(1=play)
-                   printf(" %d", data);data++;
+                   //m: method (byte) url:(0=none, 1=get, 2=datat)/gf2:(1=play)
+                   printf(" %d", data);
                } break;
                case 'b': {
-                   printf(" %d", *(U16*)data);data+=2;
+                   printf(" %d", *(U16*)data);
                } break;
                case 'p': {
-                   U8 type = *data++;
+                   U8 type = *data;
+                   char*value = data+1;
                    if(type == 0) {
-                       printf(" String:\"%s\"", data);data+=strlen(data)+1;
+                       printf(" String:\"%s\"", value);
                    } else if (type == 1) {
-                       printf(" Double:\"%f\"", *(float*)data);data+=4;
+                       printf(" Float:\"%f\"", *(float*)value);
                    } else if (type == 2) {
                        printf(" NULL");
                    } else if (type == 4) {
-                       printf(" register:%d", data++);
+                       printf(" register:%d", value);
                    } else if (type == 5) {
-                       printf(" %s", data++?"true":"false");
+                       printf(" %s", *value?"true":"false");
                    } else if (type == 6) {
-                       printf(" %f", *(double*)data);  data+=8;
+                       printf(" %f", *(double*)value);
                    } else if (type == 7) {
-                       printf(" %d", *(int*)data); data+=4;
+                       printf(" %d", *(int*)value);
                    } else if (type == 8) {
-                       printf(" Lookup:%d", data++);
+                       printf(" Lookup:%d", *value);
                    }
                } break;
-           
            }
-           cp++;
+           data += OpAdvance(*cp, data);
+           if(*cp!='c' || !poollen)
+               cp++;
+           if(poollen)
+               poollen--;
        }
 
        if(data < atag->data + atag->len)
@@ -274,3 +331,116 @@ void DumpActions(ActionTAG*atag, char*prefix)
     }
 }
 
+static const char TYPE_URL = 1;
+static const char TYPE_TARGET = 2;
+static const char TYPE_STRING = 4;
+
+int ActionEnumerate(ActionTAG*atag, char*(*callback)(char*), int type)
+{
+    U8 op;
+    int t;
+    U8*data;
+    char* cp;
+    
+    while(atag)
+    {
+       U8 poollen = 0;
+       for(t=0;t<definedactions;t++)
+           if(actions[t].op == atag->op)
+               break;
+
+       if(t==definedactions) {
+           // unknown actiontag
+           atag = atag->next;
+           continue;
+       }
+       cp = actions[t].flags;
+       data = atag->data;
+       if(atag->len) {
+           while(*cp) {
+               U8 * replacepos = 0;
+               int replacelen = 0;
+               U8 * replacement = 0;
+               switch(*cp)
+               {
+                   case 'u': {
+                       if(type&TYPE_URL)
+                       {
+                           replacelen = strlen(data);
+                           replacepos = data;
+                           replacement = callback(data); // may be null
+                       }
+                   } break;
+                   case 't': {
+                       if(type&TYPE_TARGET)
+                       {
+                           replacelen = strlen(data);
+                           replacepos = data;
+                           replacement = callback(data); // may be null
+                       }
+                   } break;
+                   case 'c': {
+                       if(type&TYPE_STRING)
+                       {
+                           replacelen = strlen(data);
+                           replacepos = data;
+                           replacement = callback(data); // may be null
+                       }
+                   } break;
+                   case 'C': {
+                       poollen = (*data);
+                   } break;
+                   case 'o': {
+                   } break;
+                   case 'p': {
+                       U8 datatype = *data;
+                       char*value = &data[1];
+                       if(datatype == 0) { //string
+                           if(type&TYPE_STRING)
+                           {
+                               replacelen = strlen(value);
+                               replacepos = value;
+                               replacement = callback(value); // may be null
+                           }
+                       } else if (datatype == 8) { //lookup
+                       }
+                   } break;
+               }
+               data += OpAdvance(*cp, data);
+               if(*cp!='c' || !poollen)
+                   cp++;
+               if(poollen)
+                   poollen--;
+
+               if(replacement)
+               {
+                   int newlen = strlen(replacement);
+                   char * newdata = malloc(atag->len - replacelen + newlen);
+                   int rpos = replacepos - atag->data;
+                   memcpy(newdata, atag->data, rpos);
+                   memcpy(&newdata[rpos], replacement, newlen);
+                   memcpy(&newdata[rpos+newlen], &replacepos[replacelen],
+                           &data[atag->len] - &replacepos[replacelen]);
+                   free(atag->data);
+                   atag->data = newdata;
+                   data = &atag->data[rpos+newlen+1];
+               }
+           }
+       }
+       atag = atag->next;
+    }
+}
+
+void ActionEnumerateTargets(ActionTAG*atag, char*(*callback)(char*))
+{
+    ActionEnumerate(atag, callback, TYPE_TARGET);
+}
+void ActionEnumerateStrings(ActionTAG*atag, char*(*callback)(char*))
+{
+    ActionEnumerate(atag, callback, TYPE_STRING);
+}
+void ActionEnumerateURLs(ActionTAG*atag, char*(*callback)(char*))
+{
+    ActionEnumerate(atag, callback, TYPE_URL);
+}
+