src/swfbytes.c

   1 /* swfbytes.c
   2    A tool for modifying swfs on the tag level
   3
   4    Part of the swftools package.
   5
   6    Copyright (c) 2008/2009 Matthias Kramm <kramm@quiss.org>
   7
   8    This program is free software; you can redistribute it and/or modify
   9    it under the terms of the GNU General Public License as published by
  10    the Free Software Foundation; either version 2 of the License, or
  11    (at your option) any later version.
  12
  13    This program is distributed in the hope that it will be useful,
  14    but WITHOUT ANY WARRANTY; without even the implied warranty of
  15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16    GNU General Public License for more details.
  17
  18    You should have received a copy of the GNU General Public License
  19    along with this program; if not, write to the Free Software
  20    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  21
  22 #include "../config.h"
  23 #include <unistd.h>
  24 #include <stdio.h>
  25 #include <fcntl.h>
  26 #include <stdarg.h>
  27 #include "../lib/rfxswf.h"
  28 #include "../lib/args.h"
  29
  30 static char * filename = 0;
  31 static int verbose;
  32
  33 struct options_t options[] =
  34 {
  35  {"v","verbose"},
  36  {"V","version"},
  37  {0,0}
  38 };
  39
  40 int args_callback_option(char*name,char*val)
  41 {
  42     if(!strcmp(name, "V")) {
  43         printf("swfedit - part of %s %s\n", PACKAGE, VERSION);
  44         return 0;
  45     } else if(!strcmp(name, "v")) {
  46         verbose++;
  47         return 0;
  48     } else {
  49         printf("Unknown option: -%s\n", name);
  50         return 0;
  51     }
  52     return 0;
  53 }
  54 int args_callback_longoption(char*name,char*val)
  55 {
  56     return args_long2shortoption(options, name, val);
  57 }
  58 void args_callback_usage(char*name)
  59 {
  60     printf("Usage: %s [-at] file.swf\n", name);
  61     printf("\t-h , --help\t\t Print help and exit\n");
  62     printf("\t-v , --verbose\t\t Be more verbose\n");
  63     printf("\t-V , --version\t\t Print program version and exit\n");
  64 }
  65 int args_callback_command(char*name,char*val)
  66 {
  67     if(filename) {
  68         fprintf(stderr, "Only one file allowed. You supplied at least two. (%s and %s)\n", filename, name);
  69     }
  70     filename = name;
  71     return 0;
  72 }
  73
  74 void dumpTag(FILE*fo, char*prefix, TAG*tag)
  75 {
  76     int t;
  77     for(t=0;t<tag->len;t++) {
  78         if(!(t&15))
  79             fprintf(fo, "%s| ", prefix);
  80         fprintf(fo, "%02x ", tag->data[t]);
  81         if((t && ((t&15)==15)) || (t==tag->len-1)) {
  82             fprintf(fo, "\n");
  83         }
  84     }
  85 }
  86
  87 void dumpFile(SWF*swf, FILE*fo)
  88 {
  89     TAG* tag = swf->firstTag;
  90     int indent = 0;
  91     int t;
  92     char whitespace[33];
  93     char*prefix = "";
  94
  95     memset(whitespace, 32, 32); whitespace[32] = 0;
  96
  97     fprintf(fo, "Version: %d\n", swf->fileVersion);
  98     fprintf(fo, "FrameRate: %f\n",swf->frameRate/256.0);
  99     fprintf(fo, "FrameCount: %d\n",swf->frameCount);
 100     fprintf(fo, "Width: %.2f\n",(swf->movieSize.xmax-swf->movieSize.xmin)/20.0);
 101     fprintf(fo, "X-Offset: %.2f\n", swf->movieSize.xmin/20.0);
 102     fprintf(fo, "Height: %.2f\n",(swf->movieSize.ymax-swf->movieSize.ymin)/20.0);
 103     fprintf(fo, "Y-Offset: %.2f\n", swf->movieSize.ymin/20.0);
 104     fprintf(fo, "\n");
 105
 106     while(tag)
 107     {
 108         if(swf_isDefiningTag(tag)) {
 109             fprintf(fo, "%s%s <%d>\n", prefix, swf_TagGetName(tag), swf_GetDefineID(tag));
 110             swf_SetTagPos(tag, 2);
 111             dumpTag(fo, prefix, tag);
 112         } else if(swf_isPseudoDefiningTag(tag)) {
 113             fprintf(fo, "%s%s <%d>\n", prefix, swf_TagGetName(tag), swf_GetDefineID(tag));
 114             swf_SetTagPos(tag, 2);
 115             dumpTag(fo, prefix, tag);
 116         } else if(tag->id == ST_PLACEOBJECT || tag->id == ST_PLACEOBJECT2) {
 117             SWFPLACEOBJECT po;
 118             swf_GetPlaceObject(tag, &po);
 119             fprintf(fo, "%s%s <%d>\n",  prefix,swf_TagGetName(tag), po.id);
 120             /* for now */
 121             swf_SetTagPos(tag, 0);
 122             dumpTag(fo, prefix, tag);
 123             swf_PlaceObjectFree(&po);
 124         } else {
 125             fprintf(fo, "%s%s\n", prefix, swf_TagGetName(tag));
 126             dumpTag(fo, prefix, tag);
 127         }
 128
 129         if(tag->id == ST_DEFINESPRITE) {
 130             indent+=4;
 131             if(indent>32)
 132                 indent = 32;
 133             prefix = &whitespace[32-indent];
 134         } else if(tag->id == ST_END) {
 135             indent-=4;
 136             if(indent<0)
 137                 indent = 0;
 138             prefix = &whitespace[32-indent];
 139         }
 140
 141         tag = tag->next;
 142     }
 143 }
 144
 145 static void readline(FILE*fi, char*line, int maxlen) {
 146     int pos = 0;
 147     while(!feof(fi)) {
 148         if(!fread(&line[pos],1,1,fi))
 149             break;
 150
 151         /* cut of preceding whitespace */
 152         if(pos == 0 && (line[0] == 32 || line[0] == '\t'))
 153             continue;
 154
 155         if(line[pos] == 13 || line[pos]==10)
 156             break;
 157         if(pos<maxlen-1)
 158             pos++;
 159     }
 160     line[pos]=0;
 161
 162     /* cut off comments */
 163     char*x=strchr(line,'#');
 164     if(x) {
 165         *x=0;
 166         pos = x-line;
 167     }
 168
 169     /* cut off trailing whitespace */
 170     while(pos>=1 && (line[pos-1]==32 || line[pos-1]==9)) {
 171         pos--;
 172         line[pos]=0;
 173     }
 174 }
 175
 176 int getFloat(char*s)
 177 {
 178     float x;
 179     int n;
 180     while(*s==32 || *s=='\t') s++;
 181     sscanf(s, "%f%n", &x, &n);
 182     if(n==0)
 183         fprintf(stderr, "Not a float: %s\n", s);
 184     return x;
 185 }
 186 int getInt(char*s)
 187 {
 188     int i;
 189     int n;
 190     while(*s==32 || *s=='\t') s++;
 191     sscanf(s, "%d%n", &i, &n);
 192     if(n==0)
 193         fprintf(stderr, "Not an integer: %s\n", s);
 194     return i;
 195 }
 196 int getTwip(char*x)
 197 {
 198     return (int)(getFloat(x)*20);
 199 }
 200
 201
 202 static char**lookup;
 203 int swf_TagNameToID(char*name)
 204 {
 205     int t;
 206     TAG tag;
 207     memset(&tag, 0, sizeof(tag));
 208     if(!lookup) {
 209         lookup = (char**)malloc(sizeof(char*)*65536);
 210         for(t=0;t<65536;t++) {
 211             tag.id = t;
 212             lookup[t] = swf_TagGetName(&tag);
 213         }
 214     }
 215     for(t=0;t<65536;t++) {
 216         if(lookup[t] && !strcasecmp(name, lookup[t]))
 217             return t;
 218     }
 219     fprintf(stderr, "Not a tag name: \"%s\"\n", name);
 220     return -1;
 221 }
 222
 223 void parseFile(FILE*fi, SWF*swf)
 224 {
 225     TAG*tag = 0;
 226     char line[1024];
 227     memset(swf, 0, sizeof(SWF));
 228     while(1) {
 229         char*colon = 0;
 230         readline(fi, line, 1024);
 231         if(!line[0])
 232             break;
 233         colon = strchr(line, ':');
 234         if(colon) {
 235             int num = colon - line;
 236             int n;
 237             if(num == 9 && !strncmp(line, "FrameRate", num)) {
 238                 swf->frameRate = getFloat(colon+1)*256;
 239             } else if(num == 10 && !strncmp(line, "FrameCount", num)) {
 240                 swf->frameCount = getInt(colon+1);
 241             } else if(num == 7 && !strncmp(line, "Version", num)) {
 242                 swf->fileVersion = getInt(colon+1);
 243             } else if(num == 5 && !strncmp(line, "Width", num)) {
 244                 int width = getTwip(colon+1);
 245                 swf->movieSize.xmax += width;
 246             } else if(num == 6 && !strncmp(line, "Height", num)) {
 247                 int height = getTwip(colon+1);
 248                 swf->movieSize.ymax += height;
 249             } else if(num == 8 && !strncmp(line, "X-Offset", num)) {
 250                 int xoffset = getTwip(colon+1);
 251                 swf->movieSize.xmin += xoffset;
 252                 swf->movieSize.xmax += xoffset;
 253             } else if(num == 8 && !strncmp(line, "Y-Offset", num)) {
 254                 int yoffset = getTwip(colon+1);
 255                 swf->movieSize.ymin += yoffset;
 256                 swf->movieSize.ymax += yoffset;
 257             } else {
 258                 fprintf(stderr, "Ignored line \"%s\"\n", line);
 259             }
 260         }
 261     }
 262     while(!feof(fi)) {
 263         char*s, *tagname;
 264         char*br;
 265         readline(fi, line, 1024);
 266         if(!line[0])
 267             continue;
 268         s = strchr(line, ' ');
 269         br = strchr(line, '|');
 270         if(!br) {
 271             int id = 0;
 272             /* DEFINESHAPE <id> ... type line */
 273             if(!s) {
 274                 tagname = strdup(line);
 275             } else {
 276                 tagname = strdup(line);
 277                 tagname[s-line] = 0;
 278             }
 279             id = swf_TagNameToID(tagname);
 280             free(tagname);
 281             if(id<0) {
 282                 fprintf(stderr, "Ignored line \"%s\"\n", line);
 283                 continue;
 284             }
 285             tag = swf_InsertTag(tag, id);
 286             if(!swf->firstTag)
 287                 swf->firstTag = tag;
 288         } else {
 289             /* | 00 34 fe c7 ... type line */
 290             char*p = line;
 291             int num = 0;
 292             int digits = 0;
 293
 294             if(!tag) {
 295                 fprintf(stderr, "Discarded unassignable data %s\n", line);
 296                 continue;
 297             }
 298
 299             while(*p) {
 300                 int n = 0;
 301                 if((*p>='a' && *p<='f') ||
 302                    (*p>='A' && *p<='F')) {
 303                     n = 9 + (*p & 15);
 304                     num = (num<<4) | n;
 305                     digits++;
 306                 }
 307                 else if(*p>='0' && *p<='9') {
 308                     n = *p & 15;
 309                     num = (num<<4) | n;
 310                     digits++;
 311                 } else if(digits) {
 312                     swf_SetU8(tag, num);
 313                     num = 0;
 314                     digits = 0;
 315                 }
 316                 p++;
 317             }
 318             if(digits)
 319                 swf_SetU8(tag, num);
 320         }
 321     }
 322 }
 323
 324 char swf_IsSWF(char*filename)
 325 {
 326     int fi = open(filename,O_RDONLY|O_BINARY);
 327     U8 buf[3] = {0,0,0};
 328     if(fi<0)
 329         return 0;
 330     read(fi, buf, 3);
 331     close(fi);
 332     if((buf[0] == 'F' || buf[0] == 'C') && buf[1] == 'W' && buf[2] == 'S')
 333         return 1;
 334     return 0;
 335 }
 336
 337 int main (int argc,char ** argv)
 338 {
 339     int fi;
 340     TAG*tag = 0;
 341     SWF swf;
 342
 343     processargs(argc, argv);
 344
 345     if(!filename) {
 346         fprintf(stderr, "You must supply a filename.\n");
 347         return -1;
 348     }
 349     if(swf_IsSWF(filename)) {
 350         fi = open(filename,O_RDONLY|O_BINARY);
 351         if (fi<0) {
 352             perror("Couldn't open file: ");
 353             exit(1);
 354         }
 355         if FAILED(swf_ReadSWF(fi,&swf)) {
 356             fprintf(stderr, "%s is not a valid SWF file or contains errors.\n",filename);
 357             close(fi);
 358             exit(1);
 359         }
 360         dumpFile(&swf, stdout);
 361         swf_FreeTags(&swf);
 362     } else {
 363         SWF newswf;
 364         FILE*fi = fopen(filename, "rb");
 365         parseFile(fi, &newswf);
 366         fclose(fi);
 367         int f;
 368         char*sname = "output.swf";
 369         f = open(sname,O_WRONLY|O_CREAT|O_TRUNC|O_BINARY, 0644);
 370         if FAILED(swf_WriteSWF(f,&newswf)) {
 371             fprintf(stderr, "Unable to write output file: %s\n", sname);
 372         }
 373         close(f);
 374     }
 375
 376     return 0;
 377 }