rfxswf cleanups: added prefixes and altered structure name conventions
[swftools.git] / lib / modules / swfaction.c
index d620bac..df53167 100644 (file)
@@ -12,7 +12,6 @@
 */
 
 #include "../rfxswf.h"
-#include "../args.h"
 
 struct Action
 {
@@ -29,9 +28,9 @@ t: target (string)
 l: label (string)
 c: constant pool (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)
@@ -89,7 +88,6 @@ r: register (byte)
 {4,"Trace", 0x26, ""}, //-1
 {4,"GetTime", 0x34, ""}, //+1
 {4,"RandomNumber", 0x30, ""}, //-1,+1
-
 {5,"Modulo", 0x3f,""},
 {5,"BitAnd", 0x60,""},
 {5,"BitLShift", 0x63,""},
@@ -129,135 +127,260 @@ r: register (byte)
 };
 static int definedactions = sizeof(actions)/sizeof(struct Action);
 
-ActionTAG* GetActions(TAG*tag) 
+ActionTAG* swf_GetActions(TAG*tag) 
 {
     U8 op = 1;
     int length;
     ActionTAG tmp;
-    ActionTAG*atag = &tmp;
+    ActionTAG*action = &tmp;
     U8*data;
     while(op)
     {
-       int pos;
-       atag->next = (ActionTAG*)malloc(sizeof(ActionTAG));
-       atag->next->prev = atag;
-       atag->next->next = 0;
-       atag = atag->next;
+        int pos;
+        action->next = (ActionTAG*)malloc(sizeof(ActionTAG));
+        action->next->prev = action;
+        action->next->next = 0;
+        action = action->next;
 
-       op = GetU8(tag);
-       if(op<0x80)
-           length = 0;
-       else
-           length = GetU16(tag);
+        op = swf_GetU8(tag);
+        if(op<0x80)
+            length = 0;
+        else
+            length = swf_GetU16(tag);
 
-       if(length) {
-           int t;
-           data = malloc(length);
-           for(t=0;t<length;t++)
-               data[t] = GetU8(tag);
-       } else {
-         data = 0;
-       }
-       atag->op = op;
-       atag->len = length;
-       atag->data = data;
+        if(length) {
+            int t;
+            data = malloc(length);
+            for(t=0;t<length;t++)
+                data[t] = swf_GetU8(tag);
+        } else {
+          data = 0;
+        }
+        action->op = op;
+        action->len = length;
+        action->data = data;
+        action->parent = tag;
     }
     return tmp.next;
 }
 
+void swf_SetActions(TAG*tag, ActionTAG*action)
+{
+    while(action)
+    {
+        swf_SetU8(tag, action->op);
+        if(action->op & 128)
+          swf_SetU16(tag, action->len);
+
+        swf_SetBlock(tag, action->data, action->len);
+
+        action = action->next;
+    }
+}
+
+int swf_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 '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
+            }
+        }
+    }
+}
+
 /* TODO: this should be in swfdump.c */
-void DumpActions(ActionTAG*atag, char*prefix) 
+void swf_DumpActions(ActionTAG*atag, char*prefix) 
 {
     U8 op;
     int t;
     U8*data;
     char* cp;
     if(!prefix) 
-       prefix="";
+        prefix="";
     while(atag)
     {
-       for(t=0;t<definedactions;t++)
-           if(actions[t].op == atag->op)
-               break;
+        for(t=0;t<definedactions;t++)
+            if(actions[t].op == atag->op)
+                break;
 
-       if(t==definedactions) {
-           printf("%s (%5d bytes) action: %02x\n", prefix, atag->len, op);
-           atag = atag->next;
-           continue;
-       }
-       printf("%s (%5d bytes) action: %s", prefix, atag->len, actions[t].name);
-       cp = actions[t].flags;
-       data = atag->data;
-       if(atag->len) //TODO: check for consistency: should we have a length?
-       while(*cp)
-       {
-           switch(*cp)
-           {
-               case 'f': {
-                   printf(" %d", *(U16*)data);data+=2; //FIXME: le/be
-               } break;
-               case 'u': {
-                   printf(" URL:\"%s\"", data);data+=strlen(data)+1;
-               } break;
-               case 't': {
-                   printf(" Target:\"%s\"", data);data+=strlen(data)+1;
-               } break;
-               case 'l': {
-                   printf(" Label:\"%s\"", data);data+=strlen(data)+1;
-               } break;
-               case 'c': {
-                   printf(" Constant Pool:\"%s\"", data);data+=strlen(data)+1;
-               } break;
-               case 's': {
-                   printf(" +%d", data);data++;
-               } break;
-               case 'm': {
+        if(t==definedactions) {
+            printf("%s (%5d bytes) action: %02x\n", prefix, atag->len, op);
+            atag = atag->next;
+            continue;
+        }
+        printf("%s (%5d bytes) action: %s", prefix, atag->len, actions[t].name);
+        cp = actions[t].flags;
+        data = atag->data;
+        if(atag->len) //TODO: check for consistency: should we have a length?
+        while(*cp)
+        {
+            switch(*cp)
+            {
+                case 'f': {
+                    printf(" %d", *(U16*)data); //FIXME: le/be
+                } break;
+                case 'u': {
+                    printf(" URL:\"%s\"", data);
+                } break;
+                case 't': {
+                    printf(" Target:\"%s\"", data);
+                } break;
+                case 'l': {
+                    printf(" Label:\"%s\"", data);
+                } break;
+                case 'c': {
+                    printf(" Constant Pool:\"%s\"", data);
+                } break;
+                case 's': {
+                    printf(" +%d", data);
+                } break;
+                case 'm': {
 //m: method (byte) url:(0=none, 1=get, 2=datat)/gf2:(1=play)
-                   printf(" %d", data);data++;
-               } break;
-               case 'b': {
-                   printf(" %d", *(U16*)data);data+=2;
-               } break;
-               case 'p': {
-                   U8 type = *data++;
-                   if(type == 0) {
-                       printf(" String:\"%s\"", data);data+=strlen(data)+1;
-                   } else if (type == 1) {
-                       printf(" Double:\"%f\"", *(float*)data);data+=4;
-                   } else if (type == 2) {
-                       printf(" NULL");
-                   } else if (type == 4) {
-                       printf(" register:%d", data++);
-                   } else if (type == 5) {
-                       printf(" %s", data++?"true":"false");
-                   } else if (type == 6) {
-                       printf(" %f", *(double*)data);  data+=8;
-                   } else if (type == 7) {
-                       printf(" %d", *(int*)data); data+=4;
-                   } else if (type == 8) {
-                       printf(" Lookup:%d", data++);
-                   }
-               } break;
-           
-           }
-           cp++;
-       }
+                    printf(" %d", data);
+                } break;
+                case 'b': {
+                    printf(" %d", *(U16*)data);
+                } break;
+                case 'p': {
+                    U8 type = *data;
+                    char*value = data+1;
+                    if(type == 0) {
+                        printf(" String:\"%s\"", value);
+                    } else if (type == 1) {
+                        printf(" Float:\"%f\"", *(float*)value);
+                    } else if (type == 2) {
+                        printf(" NULL");
+                    } else if (type == 4) {
+                        printf(" register:%d", value);
+                    } else if (type == 5) {
+                        printf(" %s", *value?"true":"false");
+                    } else if (type == 6) {
+                        printf(" %f", *(double*)value);
+                    } else if (type == 7) {
+                        printf(" %d", *(int*)value);
+                    } else if (type == 8) {
+                        printf(" Lookup:%d", *value);
+                    }
+                } break;
+            }
+            data += swf_OpAdvance(*cp, data);
+            cp++;
+        }
+
+        if(data < atag->data + atag->len)
+        {
+            int nl = ((atag->data+atag->len)-data);
+            int t;
+            printf(" remainder of %d bytes:\"", nl);
+            for(t=0;t<nl;t++) {
+                if(data[t]<32)
+                    printf("\\%d",data[t]);
+                else
+                    printf("%c", data[t]);
+            }
+            printf("\"");
+        }
+        printf("\n");
+        atag = atag->next;
+    }
+}
+
+int swf_ActionEnumerateURLs(ActionTAG*atag, char*(*callback)(char*))
+{
+    U8 op;
+    int t;
+    U8*data;
+    char* cp;
+    
+    while(atag)
+    {
+
+        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)
+            {
+                char * replacepos = 0;
+                int replacelen = 0;
+                char * replacement;
+                switch(*cp)
+                {
+                    case 'u': {
+                        replacelen = strlen(data);
+                        replacepos = data;
+                        replacement = callback(data); // may be null
+                    } break;
+                    /* everything below may very well
+                       contain an URL, too. However, to extract 
+                       these, we would have to call callback also for
+                       strings which might not contain an url.
+                        TODO: should we check for Strings which start 
+                        with "http://"?
+                        Nope: user can force it by reg.ex. if he wants to /r
+                     */
+                    case 'c': {
+                    } break;
+                    case 'o': {
+                    } break;
+                    case 'p': {
+                        U8 type = *data;
+                        char*value = &data[1];
+                        if(type == 0) { //string
+                        } else if (type == 8) { //lookup
+                        }
+                    } break;
+                }
+                data += swf_OpAdvance(*cp, data);
+                cp++;
+
+                //TODO: apply replacement here.
+            }
+        }
 
-       if(data < atag->data + atag->len)
-       {
-           int nl = ((atag->data+atag->len)-data);
-           int t;
-           printf(" remainder of %d bytes:\"", nl);
-           for(t=0;t<nl;t++) {
-               if(data[t]<32)
-                   printf("\\%d",data[t]);
-               else
-                   printf("%c", data[t]);
-           }
-           printf("\"");
-       }
-       printf("\n");
-       atag = atag->next;
+        atag = atag->next;
     }
 }