src/swfextract.c

   1 /* swfextract.c
   2    Allows to extract parts of the swf into a new file.
   3
   4    Part of the swftools package.
   5
   6    Copyright (c) 2001 Matthias Kramm <kramm@quiss.org>
   7
   8    This file is distributed under the GPL, see file COPYING for details */
   9
  10 #include <unistd.h>
  11 #include <stdio.h>
  12 #include <fcntl.h>
  13 #include "../lib/rfxswf.h"
  14 #include "../lib/args.h"
  15 #include "../lib/log.h"
  16 #ifdef HAVE_ZLIB_H
  17 #ifdef HAVE_LIBZ
  18 #include "zlib.h"
  19 #define _ZLIB_INCLUDED_
  20 #endif
  21 #endif
  22
  23 char * filename = 0;
  24 char * destfilename = "output.swf";
  25 int verbose = 3;
  26
  27 char* extractids = 0;
  28 char* extractframes = 0;
  29 char* extractjpegids = 0;
  30 char* extractfontids = 0;
  31 char* extractpngids = 0;
  32 char* extractsoundids = 0;
  33 char extractmp3 = 0;
  34
  35 char* extractname = 0;
  36
  37 char hollow = 0;
  38 char originalplaceobjects = 0;
  39
  40 int numextracts = 0;
  41
  42 struct options_t options[] =
  43 {
  44  {"o","output"},
  45  {"w","hollow"},
  46  {"v","verbose"},
  47  {"i","id"},
  48  {"j","jpegs"},
  49  {"p","pngs"},
  50  {"P","placeobject"},
  51  {"m","mp3"},
  52  {"s","sound"},
  53  {"n","name"},
  54  {"f","frame"},
  55  {"F","font"},
  56  {"V","version"},
  57  {0,0}
  58 };
  59
  60
  61 int args_callback_option(char*name,char*val)
  62 {
  63     if(!strcmp(name, "V")) {
  64         printf("swfextract - part of %s %s\n", PACKAGE, VERSION);
  65         exit(0);
  66     }
  67     else if(!strcmp(name, "o")) {
  68         destfilename = val;
  69         return 1;
  70     }
  71     else if(!strcmp(name, "i")) {
  72         extractids = val;
  73         numextracts++;
  74         if(extractname) {
  75             fprintf(stderr, "You can only supply either name or id\n");
  76             exit(1);
  77         }
  78         return 1;
  79     }
  80     else if(!strcmp(name, "n")) {
  81         extractname = val;
  82         numextracts++;
  83         if(extractids) {
  84             fprintf(stderr, "You can only supply either name or id\n");
  85             exit(1);
  86         }
  87         return 1;
  88     }
  89     else if(!strcmp(name, "v")) {
  90         verbose ++;
  91         return 0;
  92     }
  93     else if(!strcmp(name, "m")) {
  94         extractmp3 = 1;
  95         numextracts++;
  96         return 0;
  97     }
  98     else if(!strcmp(name, "j")) {
  99         if(extractjpegids) {
 100             fprintf(stderr, "Only one --jpegs argument is allowed. (Try to use a range, e.g. -j 1,2,3)\n");
 101             exit(1);
 102         }
 103         numextracts++;
 104         extractjpegids = val;
 105         return 1;
 106     }
 107     else if(!strcmp(name, "F")) {
 108         if(extractfontids) {
 109             fprintf(stderr, "Only one --font argument is allowed. (Try to use a range, e.g. -s 1,2,3)\n");
 110             exit(1);
 111         }
 112         numextracts++;
 113         extractfontids = val;
 114         return 1;
 115     }
 116     else if(!strcmp(name, "s")) {
 117         if(extractsoundids) {
 118             fprintf(stderr, "Only one --sound argument is allowed. (Try to use a range, e.g. -s 1,2,3)\n");
 119             exit(1);
 120         }
 121         numextracts++;
 122         extractsoundids = val;
 123         return 1;
 124     }
 125 #ifdef _ZLIB_INCLUDED_
 126     else if(!strcmp(name, "p")) {
 127         if(extractpngids) {
 128             fprintf(stderr, "Only one --png argument is allowed. (Try to use a range, e.g. -p 1,2,3)\n");
 129             exit(1);
 130         }
 131         numextracts++;
 132         extractpngids = val;
 133         return 1;
 134     }
 135 #endif
 136     else if(!strcmp(name, "f")) {
 137         numextracts++;
 138         extractframes = val;
 139         return 1;
 140     }
 141     else if(!strcmp(name, "P")) {
 142         originalplaceobjects = 1;
 143         return 0;
 144     }
 145     else if(!strcmp(name, "w")) {
 146         hollow = 1;
 147         return 0;
 148     }
 149     else {
 150         printf("Unknown option: -%s\n", name);
 151         exit(1);
 152     }
 153
 154     return 0;
 155 }
 156 int args_callback_longoption(char*name,char*val)
 157 {
 158     return args_long2shortoption(options, name, val);
 159 }
 160 void args_callback_usage(char*name)
 161 {
 162     printf("Usage: %s [-v] [-n name] [-ijf ids] file.swf\n", name);
 163     printf("\t-v , --verbose\t\t\t Be more verbose\n");
 164     printf("\t-o , --output filename\t\t set output filename\n");
 165     printf("\t-V , --version\t\t\t Print program version and exit\n\n");
 166     printf("SWF Subelement extraction:\n");
 167     printf("\t-n , --name name\t\t instance name of the object (SWF Define) to extract\n");
 168     printf("\t-i , --id ID\t\t\t ID of the object, shape or movieclip to extract\n");
 169     printf("\t-f , --frame frames\t\t frame numbers to extract\n");
 170     printf("\t-w , --hollow\t\t\t hollow mode: don't remove empty frames\n");
 171     printf("\t             \t\t\t (use with -f)\n");
 172     printf("\t-P , --placeobject\t\t\t Insert original placeobject into output file\n");
 173     printf("\t             \t\t\t (use with -i)\n");
 174     printf("SWF Font/Text extraction:\n");
 175     printf("\t-F , --font ID\t\t\t Extract font(s)\n");
 176     printf("Picture extraction:\n");
 177     printf("\t-j , --jpeg ID\t\t\t Extract JPEG picture(s)\n");
 178 #ifdef _ZLIB_INCLUDED_
 179     printf("\t-p , --pngs ID\t\t\t Extract PNG picture(s)\n");
 180 #endif
 181     printf("\n");
 182     printf("Sound extraction:\n");
 183     printf("\t-m , --mp3\t\t\t Extract main mp3 stream\n");
 184     printf("\t-s , --sound ID\t\t\t Extract Sound(s)\n");
 185 }
 186 int args_callback_command(char*name,char*val)
 187 {
 188     if(filename) {
 189         fprintf(stderr, "Only one file allowed. You supplied at least two. (%s and %s)\n",
 190                  filename, name);
 191     }
 192     filename = name;
 193     return 0;
 194 }
 195
 196 U8 mainr,maing,mainb;
 197 /* 1 = used, not expanded,
 198    3 = used, expanded
 199    5 = wanted, not expanded
 200    7 = wanted, expanded
 201  */
 202 char used[65536];
 203 TAG*tags[65536];
 204 int changed;
 205 char * tagused;
 206 int extractname_id = -1;
 207
 208 void idcallback(void*data)
 209 {
 210     if(!(used[GET16(data)]&1)) {
 211         changed = 1;
 212         used[GET16(data)] |= 1;
 213     }
 214 }
 215
 216 void enumerateIDs(TAG*tag, void(*callback)(void*))
 217 {
 218 /*    U8*data;
 219     int len = tag->len;
 220     if(tag->len>=64) {
 221         len += 6;
 222         data = (U8*)malloc(len);
 223         PUT16(data, (tag->id<<6)+63);
 224         *(U8*)&data[2] = tag->len;
 225         *(U8*)&data[3] = tag->len>>8;
 226         *(U8*)&data[4] = tag->len>>16;
 227         *(U8*)&data[5] = tag->len>>24;
 228         memcpy(&data[6], tag->data, tag->len);
 229     } else {
 230         len += 2;
 231         data = (U8*)malloc(len);
 232         PUT16(data, (tag->id<<6)+tag->len);
 233         memcpy(&data[2], tag->data, tag->len);
 234     }
 235     map_ids_mem(data, len, callback);
 236  */
 237     int num = swf_GetNumUsedIDs(tag);
 238     int *ptr = malloc(sizeof(int)*num);
 239     int t;
 240     swf_GetUsedIDs(tag, ptr);
 241     for(t=0;t<num;t++)
 242         callback(&tag->data[ptr[t]]);
 243 }
 244
 245 void extractTag(SWF*swf, char*filename)
 246 {
 247     SWF newswf;
 248     TAG*desttag;
 249     TAG*srctag;
 250     RGBA rgb;
 251     SRECT objectbbox;
 252     char sprite;
 253     int f;
 254     int t;
 255     int tagnum;
 256     int copy = 0;
 257     memset(&newswf,0x00,sizeof(SWF));        // set global movie parameters
 258
 259     newswf.fileVersion    = swf->fileVersion;
 260     newswf.frameRate      = swf->frameRate;
 261     newswf.movieSize      = swf->movieSize;
 262
 263     newswf.firstTag = swf_InsertTag(NULL,ST_SETBACKGROUNDCOLOR);
 264     desttag = newswf.firstTag;
 265     rgb.r = mainr;
 266     rgb.g = maing;
 267     rgb.b = mainb;
 268     swf_SetRGB(desttag,&rgb);
 269
 270     swf_GetRect(0, &objectbbox);
 271
 272     do {
 273        changed = 0;
 274        for(t=0;t<65536;t++) {
 275            if(used[t] && !(used[t]&2)) {
 276              if(tags[t]->id==ST_DEFINESPRITE) {
 277                  TAG*tag = tags[t];
 278                  while(tag->id != ST_END)
 279                  {
 280                      enumerateIDs(tag, idcallback);
 281                      tag = tag->next;
 282                  }
 283              }
 284              else
 285                enumerateIDs(tags[t], idcallback);
 286              used[t] |= 2;
 287            }
 288        }
 289     }
 290     while(changed);
 291
 292     srctag = swf->firstTag;
 293     tagnum = 0;
 294     sprite = 0;
 295     while(srctag && (srctag->id || sprite)) {
 296         int reset = 0;
 297         if(!sprite) {
 298             copy = 0;
 299         }
 300         if(srctag->id == ST_END) {
 301             sprite = 0;
 302         }
 303         if(srctag->id == ST_DEFINESPRITE)
 304             sprite = 1;
 305         if(swf_isDefiningTag(srctag)) {
 306             int id = swf_GetDefineID(srctag);
 307             if(used[id])  {
 308                 SRECT b;
 309                 copy = 1;
 310                 b = swf_GetDefineBBox(srctag);
 311                 swf_ExpandRect2(&objectbbox, &b);
 312             }
 313         } else
 314         if (((((srctag->id == ST_PLACEOBJECT || srctag->id == ST_PLACEOBJECT2) && originalplaceobjects)
 315               || srctag->id == ST_STARTSOUND) && (used[swf_GetPlaceID(srctag)]&4) ) ||
 316               (swf_isPseudoDefiningTag(srctag) && used[swf_GetDefineID(srctag)]) ||
 317               (tagused[tagnum]))
 318         {
 319                 if(copy == 0)
 320                     reset = 1;
 321                 copy = 1;
 322         }
 323         if(srctag->id == ST_REMOVEOBJECT) {
 324             if(!used[swf_GetPlaceID(srctag)])
 325                 copy = 0;
 326         }
 327
 328         if(copy) {
 329             TAG*ttag = (TAG*)malloc(sizeof(TAG));
 330             desttag = swf_InsertTag(desttag, srctag->id);
 331             desttag->len = desttag->memsize = srctag->len;
 332             desttag->data = malloc(srctag->len);
 333             memcpy(desttag->data, srctag->data, srctag->len);
 334             if(reset)
 335                 copy = 0;
 336         }
 337
 338         srctag = srctag->next;
 339         tagnum ++;
 340     }
 341     if(!extractframes && !hollow) {
 342         if(!originalplaceobjects && (extractids||extractname_id>=0)) {
 343             int t;
 344             int s=0;
 345             if((objectbbox.xmin|objectbbox.ymin|objectbbox.xmax|objectbbox.ymax)!=0)
 346                 newswf.movieSize = objectbbox;
 347             if(extractname_id>=0) {
 348                 desttag = swf_InsertTag(desttag, ST_PLACEOBJECT2);
 349                 swf_ObjectPlace(desttag, extractname_id, extractname_id, 0,0,extractname);
 350             } else {
 351                 for(t=0;t<65536;t++) {
 352                     if(is_in_range(t, extractids)) {
 353                         desttag = swf_InsertTag(desttag, ST_PLACEOBJECT2);
 354                         swf_ObjectPlace(desttag, t, t, 0,0,0);
 355                         s++;
 356                         if(s==2)
 357                             printf("warning! You should use the -P when extracting multiple objects\n");
 358                     }
 359                 }
 360             }
 361         }
 362         desttag = swf_InsertTag(desttag,ST_SHOWFRAME);
 363     }
 364     desttag = swf_InsertTag(desttag,ST_END);
 365
 366     f = open(filename, O_TRUNC|O_WRONLY|O_CREAT|O_BINARY, 0644);
 367     if FAILED(swf_WriteSWF(f,&newswf)) fprintf(stderr,"WriteSWF() failed.\n");
 368     close(f);
 369
 370     swf_FreeTags(&newswf);                       // cleanup
 371 }
 372
 373 void listObjects(SWF*swf)
 374 {
 375     TAG*tag;
 376     char first;
 377     int t;
 378     int frame = 0;
 379     char*names[] = {"Shapes","MovieClips","JPEGs","PNGs","Sounds","Fonts"};
 380     printf("Objects in file %s:\n",filename);
 381     for(t=0;t<sizeof(names)/sizeof(names[0]);t++) {
 382         tag = swf->firstTag;
 383         first = 1;
 384         while(tag) {
 385             char show = 0;
 386             char text[80];
 387             if(t == 0 &&
 388                (tag->id == ST_DEFINESHAPE ||
 389                 tag->id == ST_DEFINESHAPE2 ||
 390                 tag->id == ST_DEFINESHAPE3)) {
 391                 show = 1;
 392                 sprintf(text,"%d", swf_GetDefineID(tag));
 393             }
 394
 395             if(tag->id == ST_DEFINESPRITE) {
 396                 if (t == 1)  {
 397                     show = 1;
 398                     sprintf(text,"%d", swf_GetDefineID(tag));
 399                 }
 400
 401                 while(tag->id != ST_END)
 402                     tag = tag->next;
 403             }
 404
 405             if(t == 2 && (tag->id == ST_DEFINEBITS ||
 406                 tag->id == ST_DEFINEBITSJPEG2 ||
 407                 tag->id == ST_DEFINEBITSJPEG3)) {
 408                 show = 1;
 409                 sprintf(text,"%d", swf_GetDefineID(tag));
 410             }
 411
 412             if(t == 3 && (tag->id == ST_DEFINEBITSLOSSLESS ||
 413                 tag->id == ST_DEFINEBITSLOSSLESS2)) {
 414                 show = 1;
 415                 sprintf(text,"%d", swf_GetDefineID(tag));
 416             }
 417
 418
 419             if(t == 4 && (tag->id == ST_DEFINESOUND)) {
 420                 show = 1;
 421                 sprintf(text,"%d", swf_GetDefineID(tag));
 422             }
 423
 424             if(t == 5 && (tag->id == ST_DEFINEFONT || tag->id == ST_DEFINEFONT2)) {
 425                 show = 1;
 426                 sprintf(text,"%d", swf_GetDefineID(tag));
 427             }
 428
 429             if(show) {
 430                 if(!first)
 431                     printf(", ");
 432                 else
 433                     printf("%s: ", names[t]);
 434                 printf("%s", text);
 435                 first = 0;
 436             }
 437             tag=tag->next;
 438         }
 439         if(!first)
 440             printf("\n");
 441     }
 442
 443     if(frame)
 444         printf("Frames: 0-%d\n", frame);
 445     else
 446         printf("Frames: 0\n");
 447 }
 448
 449 void handlefont(SWF*swf, TAG*tag)
 450 {
 451     SWFFONT* f=0;
 452     U16 id;
 453     char name[80];
 454     char*filename = name;
 455     int t;
 456
 457     id = swf_GetDefineID(tag);
 458     sprintf(name, "font%d.swf", id);
 459     if(numextracts==1) {
 460         filename = destfilename;
 461     }
 462
 463     swf_FontExtract(swf, id, &f);
 464     if(!f) {
 465         printf("Couldn't extract font %d\n", id);
 466         return;
 467     }
 468     if(!f->layout)
 469         swf_FontCreateLayout(f);
 470
 471     swf_WriteFont(f, filename);
 472     swf_FontFree(f);
 473 }
 474
 475 U8*jpegtables = 0;
 476 int jpegtablessize;
 477
 478 void handlejpegtables(TAG*tag)
 479 {
 480     if(tag->id == ST_JPEGTABLES) {
 481         jpegtables = tag->data;
 482         jpegtablessize = tag->len;
 483     }
 484 }
 485
 486 FILE* save_fopen(char* name, char* mode)
 487 {
 488     FILE*fi = fopen(name, mode);
 489     if(!fi) {
 490         fprintf(stderr, "Error: Couldn't open %s\n", name);
 491         exit(1);
 492     }
 493     return fi;
 494 }
 495
 496 int findjpegboundary(U8*data, int len)
 497 {
 498     int t;
 499     int pos=-1;
 500     for(t=0;t<len;t++) {
 501         if(data[t  ]==0xff &&
 502            data[t+1]==0xd9 &&
 503            data[t+2]==0xff &&
 504            data[t+3]==0xd8) {
 505             pos = t;
 506         }
 507     }
 508     return pos;
 509 }
 510
 511 /* extract jpeg data out of a tag */
 512 void handlejpeg(TAG*tag)
 513 {
 514     char name[80];
 515     char*filename = name;
 516     FILE*fi;
 517
 518     sprintf(name, "pic%d.jpg", GET16(tag->data));
 519     if(numextracts==1) {
 520         filename = destfilename;
 521         if(!strcmp(filename,"output.swf"))
 522             filename = "output.jpg";
 523     }
 524     /* swf jpeg images have two streams, which both start with ff d8 and
 525        end with ff d9. The following code handles sorting the middle
 526        <ff d9 ff d8> bytes out, so that one stream remains */
 527     if(tag->id == ST_DEFINEBITS && tag->len>2 && jpegtables) {
 528         fi = save_fopen(filename, "wb");
 529         fwrite(jpegtables, 1, jpegtablessize-2, fi); //don't write end tag (ff,d8)
 530         fwrite(&tag->data[2+2], tag->len-2-2, 1, fi); //don't write start tag (ff,d9)
 531         fclose(fi);
 532     }
 533     else if(tag->id == ST_DEFINEBITSJPEG2 && tag->len>2) {
 534         int end = tag->len;
 535         int pos = findjpegboundary(&tag->data[2], tag->len-2);
 536         if(pos<0)
 537             return;
 538         pos+=2;
 539         fi = save_fopen(filename, "wb");
 540         fwrite(&tag->data[2], pos-2, 1, fi);
 541         fwrite(&tag->data[pos+4], end-(pos+4), 1, fi);
 542         fclose(fi);
 543     }
 544     else if(tag->id == ST_DEFINEBITSJPEG3 && tag->len>6) {
 545         U32 end = GET32(&tag->data[2])+6;
 546         int pos = findjpegboundary(&tag->data[6], tag->len-6);
 547         if(pos<0)
 548             return;
 549         pos+=6;
 550         fi = save_fopen(filename, "wb");
 551         fwrite(&tag->data[6], pos-6, 1, fi);
 552         fwrite(&tag->data[pos+4], end-(pos+4), 1, fi);
 553         fclose(fi);
 554     }
 555     else {
 556         int id = GET16(tag->data);
 557         fprintf(stderr, "Object %d is not a JPEG picture!\n",id);
 558         exit(1);
 559     }
 560 }
 561
 562 #ifdef _ZLIB_INCLUDED_
 563 static U32 mycrc32;
 564
 565 static U32*crc32_table = 0;
 566 static void make_crc32_table(void)
 567 {
 568   int t;
 569   if(crc32_table)
 570       return;
 571   crc32_table = (U32*)malloc(1024);
 572
 573   for (t = 0; t < 256; t++) {
 574     U32 c = t;
 575     int s;
 576     for (s = 0; s < 8; s++) {
 577       c = (0xedb88320L*(c&1)) ^ (c >> 1);
 578     }
 579     crc32_table[t] = c;
 580   }
 581 }
 582 static void png_write_byte(FILE*fi, U8 byte)
 583 {
 584     fwrite(&byte,1,1,fi);
 585     mycrc32 = crc32_table[(mycrc32 ^ byte) & 0xff] ^ (mycrc32 >> 8);
 586 }
 587 static void png_start_chunk(FILE*fi, char*type, int len)
 588 {
 589     U8 mytype[4]={0,0,0,0};
 590     U32 mylen = REVERSESWAP32(len);
 591     memcpy(mytype,type,strlen(type));
 592     fwrite(&mylen, 4, 1, fi);
 593     mycrc32=0xffffffff;
 594     png_write_byte(fi,mytype[0]);
 595     png_write_byte(fi,mytype[1]);
 596     png_write_byte(fi,mytype[2]);
 597     png_write_byte(fi,mytype[3]);
 598 }
 599 static void png_write_bytes(FILE*fi, U8*bytes, int len)
 600 {
 601     int t;
 602     for(t=0;t<len;t++)
 603         png_write_byte(fi,bytes[t]);
 604 }
 605 static void png_write_dword(FILE*fi, U32 dword)
 606 {
 607     png_write_byte(fi,dword>>24);
 608     png_write_byte(fi,dword>>16);
 609     png_write_byte(fi,dword>>8);
 610     png_write_byte(fi,dword);
 611 }
 612 static void png_end_chunk(FILE*fi)
 613 {
 614     U32 tmp = REVERSESWAP32((mycrc32^0xffffffff));
 615     fwrite(&tmp,4,1,fi);
 616 }
 617
 618
 619 /* extract a lossless image (png) out of a tag
 620    This routine was originally meant to be a one-pager. I just
 621    didn't know png is _that_ much fun. :) -mk
 622  */
 623 void handlelossless(TAG*tag)
 624 {
 625     char name[80];
 626     char*filename = name;
 627     FILE*fi;
 628     int width, height;
 629     int crc;
 630     int id;
 631     int t;
 632     U8 bpp = 1;
 633     U8 format;
 634     U8 tmp;
 635     U8* data=0;
 636     U8* data2=0;
 637     U8* data3=0;
 638     U32 datalen;
 639     U32 datalen2;
 640     U32 datalen3;
 641     U8 head[] = {137,80,78,71,13,10,26,10};
 642     int cols;
 643     char alpha = tag->id == ST_DEFINEBITSLOSSLESS2;
 644     RGBA* palette;
 645     int pos;
 646     int error;
 647     U32 tmp32;
 648
 649     make_crc32_table();
 650
 651     if(tag->id != ST_DEFINEBITSLOSSLESS &&
 652        tag->id != ST_DEFINEBITSLOSSLESS2) {
 653         int id = GET16(tag->data);
 654         fprintf(stderr, "Object %d is not a PNG picture!\n",id);
 655         exit(1);
 656     }
 657
 658     id =swf_GetU16(tag);
 659     format = swf_GetU8(tag);
 660     if(format == 3) bpp = 8;
 661     if(format == 4) bpp = 16;
 662     if(format == 5) bpp = 32;
 663     if(format!=3 && format!=5) {
 664         if(format==4)
 665         fprintf(stderr, "Can't handle 16-bit palette images yet (image %d)\n",id);
 666         else
 667         fprintf(stderr, "Unknown image type %d in image %d\n", format, id);
 668         return;
 669     }
 670     width = swf_GetU16(tag);
 671     height = swf_GetU16(tag);
 672     if(format == 3) cols = swf_GetU8(tag) + 1;
 673 // this is what format means according to the flash specification. (which is
 674 // clearly wrong)
 675 //    if(format == 4) cols = swf_GetU16(tag) + 1;
 676 //    if(format == 5) cols = swf_GetU32(tag) + 1;
 677     else cols = 0;
 678
 679     logf("<verbose> Width %d", width);
 680     logf("<verbose> Height %d", height);
 681     logf("<verbose> Format %d", format);
 682     logf("<verbose> Cols %d", cols);
 683     logf("<verbose> Bpp %d", bpp);
 684
 685     datalen = (width*height*bpp/8+cols*8);
 686     do {
 687         if(data)
 688             free(data);
 689         datalen+=4096;
 690         data = malloc(datalen);
 691         error = uncompress (data, &datalen, &tag->data[tag->pos], tag->len-tag->pos);
 692     } while(error == Z_BUF_ERROR);
 693     if(error != Z_OK) {
 694         fprintf(stderr, "Zlib error %d (image %d)\n", error, id);
 695         return;
 696     }
 697     logf("<verbose> Uncompressed image is %d bytes (%d colormap)", datalen, (3+alpha)*cols);
 698     pos = 0;
 699     datalen2 = datalen;
 700     data2 = malloc(datalen2);
 701     palette = (RGBA*)malloc(cols*sizeof(RGBA));
 702
 703     for(t=0;t<cols;t++) {
 704         palette[t].r = data[pos++];
 705         palette[t].g = data[pos++];
 706         palette[t].b = data[pos++];
 707         if(alpha) {
 708             palette[t].a = data[pos++];
 709         }
 710     }
 711
 712     sprintf(name, "pic%d.png", id);
 713     if(numextracts==1) {
 714         filename = destfilename;
 715         if(!strcmp(filename,"output.swf"))
 716             filename = "output.png";
 717     }
 718     fi = save_fopen(filename, "wb");
 719     fwrite(head,sizeof(head),1,fi);
 720
 721     png_start_chunk(fi, "IHDR", 13);
 722      png_write_dword(fi,width);
 723      png_write_dword(fi,height);
 724      png_write_byte(fi,8);
 725      if(format == 3)
 726      png_write_byte(fi,3); //indexed
 727      else if(format == 5)
 728      png_write_byte(fi,2); //rgb
 729      else return;
 730
 731      png_write_byte(fi,0); //compression mode
 732      png_write_byte(fi,0); //filter mode
 733      png_write_byte(fi,0); //interlace mode
 734     png_end_chunk(fi);
 735
 736     if(format == 3) {
 737         png_start_chunk(fi, "PLTE", 768);
 738          for(t=0;t<256;t++) {
 739              png_write_byte(fi,palette[t].r);
 740              png_write_byte(fi,palette[t].g);
 741              png_write_byte(fi,palette[t].b);
 742          }
 743         png_end_chunk(fi);
 744     }
 745     {
 746         int pos2 = 0;
 747         int x,y;
 748         int srcwidth = width * (bpp/8);
 749         datalen3 = width*height*4;
 750         data3 = (U8*)malloc(datalen3);
 751         for(y=0;y<height;y++)
 752         {
 753            data3[pos2++]=0; //filter type
 754            if(bpp==32) {
 755             // 32 bit to 24 bit "conversion"
 756             for(x=0;x<width;x++) {
 757                 data3[pos2++]=data[pos+1];
 758                 data3[pos2++]=data[pos+2];
 759                 data3[pos2++]=data[pos+3];
 760                 pos+=4; //ignore padding byte
 761             }
 762            }
 763            else {
 764                 for(x=0;x<srcwidth;x++)
 765                     data3[pos2++]=data[pos++];
 766            }
 767
 768            pos+=((srcwidth+3)&~3)-srcwidth; //align
 769         }
 770         datalen3=pos2;
 771     }
 772
 773     if(compress (data2, &datalen2, data3, datalen3) != Z_OK) {
 774         fprintf(stderr, "zlib error in pic %d\n", id);
 775         return;
 776     }
 777     logf("<verbose> Compressed data is %d bytes", datalen2);
 778     png_start_chunk(fi, "IDAT", datalen2);
 779     png_write_bytes(fi,data2,datalen2);
 780     png_end_chunk(fi);
 781     png_start_chunk(fi, "IEND", 0);
 782     png_end_chunk(fi);
 783
 784     free(data);
 785     free(data2);
 786     free(data3);
 787 }
 788 #endif
 789
 790 FILE*mp3file;
 791 void handlesoundstream(TAG*tag)
 792 {
 793     char*filename = "output.mp3";
 794     if(numextracts==1) {
 795         filename = destfilename;
 796         if(!strcmp(filename,"output.swf"))
 797             filename = "output.mp3";
 798     }
 799     switch(tag->id) {
 800         case ST_SOUNDSTREAMHEAD:
 801             if((tag->data[1]&0x30) == 0x20) { //mp3 compression
 802                 mp3file = fopen(filename, "wb");
 803                 logf("<notice> Writing mp3 data to %s",filename);
 804             }
 805             else
 806                 logf("<error> Soundstream is not mp3");
 807         break;
 808         case ST_SOUNDSTREAMHEAD2:
 809             if((tag->data[1]&0x30) == 0x20) {//mp3 compression
 810                 mp3file = fopen(filename, "wb");
 811                 logf("<notice> Writing mp3 data to %s",filename);
 812             }
 813             else
 814                 logf("<error> Soundstream is not mp3 (2)");
 815         break;
 816         case ST_SOUNDSTREAMBLOCK:
 817             if(mp3file)
 818                 fwrite(&tag->data[4],tag->len-4,1,mp3file);
 819         break;
 820     }
 821 }
 822
 823 void handledefinesound(TAG*tag)
 824 {
 825     U8 flags;
 826     U32 samples;
 827     char buf[128];
 828     char*filename = buf;
 829     FILE*fi;
 830     U16 id;
 831     id = swf_GetU16(tag); //id
 832     sprintf(buf, "sound%d.mp3", id);
 833
 834     if(numextracts==1) {
 835         filename = destfilename;
 836         if(!strcmp(filename,"output.swf"))
 837             filename = "output.mp3";
 838     }
 839     flags = swf_GetU8(tag);
 840     if((flags>>4)!=2) {
 841         printf("Sorry, can only extract MP3 sounds. Sound %d is ADPCM or RAW.\n", id);
 842         /* not mp3 */
 843         return;
 844     }
 845     samples = swf_GetU32(tag);
 846
 847     swf_GetU16(tag); //(only for mp3) numsamples_seek
 848
 849     fi = save_fopen(filename, "wb");
 850     fwrite(&tag->data[tag->pos], tag->len - tag->pos, 1, fi);
 851     fclose(fi);
 852 }
 853
 854 int main (int argc,char ** argv)
 855 {
 856     TAG*tag;
 857     SWF swf;
 858     int f;
 859     int found = 0;
 860     int frame = 0;
 861     int tagnum = 0;
 862     char depths[65536];
 863     char listavailable = 0;
 864     processargs(argc, argv);
 865
 866     if(!extractframes && !extractids && ! extractname && !extractjpegids && !extractpngids
 867         && !extractmp3 && !extractsoundids && !extractfontids)
 868         listavailable = 1;
 869
 870     if(!filename)
 871     {
 872         fprintf(stderr, "You must supply a filename.\n");
 873         return 1;
 874     }
 875     initLog(0,-1,0,0,-1, verbose);
 876
 877     f = open(filename,O_RDONLY);
 878
 879     if (f<0)
 880     {
 881         perror("Couldn't open file: ");
 882         exit(1);
 883     }
 884     if (swf_ReadSWF(f,&swf) < 0)
 885     {
 886         fprintf(stderr, "%s is not a valid SWF file or contains errors.\n",filename);
 887         close(f);
 888         exit(1);
 889     }
 890     close(f);
 891
 892     if(listavailable) {
 893         listObjects(&swf);
 894         swf_FreeTags(&swf);
 895         return 0;
 896     }
 897
 898     tag = swf.firstTag;
 899     tagnum = 0;
 900     while(tag) {
 901         tagnum ++;
 902         tag = tag->next;
 903     }
 904
 905     tagused = (char*)malloc(tagnum);
 906     memset(tagused, 0, tagnum);
 907     memset(used, 0, 65536);
 908     memset(depths, 0, 65536);
 909
 910     tag = swf.firstTag;
 911     tagnum = 0;
 912     while(tag) {
 913         if(swf_isAllowedSpriteTag(tag)) {
 914             int write = 0;
 915             if(extractframes && is_in_range(frame, extractframes)) {
 916                 write = 1;
 917                 if(tag->id == ST_PLACEOBJECT || tag->id == ST_PLACEOBJECT2) {
 918                     depths[swf_GetDepth(tag)] = 1;
 919                 }
 920                 if(tag->id == ST_REMOVEOBJECT || tag->id == ST_REMOVEOBJECT2) {
 921                     int depth = swf_GetDepth(tag);
 922                     if(!depths[depth])
 923                         write = 0;
 924                     depths[swf_GetDepth(tag)] = 0;
 925                 }
 926             } else {
 927                 if((tag->id == ST_REMOVEOBJECT || tag->id == ST_REMOVEOBJECT2) &&
 928                     (depths[swf_GetDepth(tag)]) && hollow) {
 929                     write = 1;
 930                     depths[swf_GetDepth(tag)] = 0;
 931                 }
 932             }
 933             if(write) {
 934                 enumerateIDs(tag, idcallback);
 935                 found = 1;
 936                 tagused[tagnum] = 1;
 937             }
 938         }
 939
 940         if(tag->id == ST_SOUNDSTREAMHEAD ||
 941            tag->id == ST_SOUNDSTREAMHEAD2 ||
 942            tag->id == ST_SOUNDSTREAMBLOCK) {
 943             if(extractmp3)
 944                 handlesoundstream(tag);
 945         }
 946
 947         if(tag->id == ST_JPEGTABLES)
 948             handlejpegtables(tag);
 949
 950         if(swf_isDefiningTag(tag)) {
 951             int id = swf_GetDefineID(tag);
 952             tags[id] = tag;
 953             if(extractids && is_in_range(id, extractids)) {
 954                 used[id] = 5;
 955                 found = 1;
 956             }
 957             if(extractfontids && is_in_range(id, extractfontids)) {
 958                 handlefont(&swf, tag);
 959             }
 960             if(extractjpegids && is_in_range(id, extractjpegids)) {
 961                 handlejpeg(tag);
 962             }
 963             if(extractsoundids && is_in_range(id, extractsoundids)) {
 964                 handledefinesound(tag);
 965             }
 966 #ifdef _ZLIB_INCLUDED_
 967             if(extractpngids && is_in_range(id, extractpngids)) {
 968                 handlelossless(tag);
 969             }
 970 #endif
 971         }
 972         else if (tag->id == ST_SETBACKGROUNDCOLOR) {
 973             mainr = tag->data[0];
 974             maing = tag->data[1];
 975             mainb = tag->data[2];
 976         }
 977         else if(tag->id == ST_PLACEOBJECT2) {
 978             char*name = swf_GetName(tag);
 979             if(name && extractname && !strcmp(name, extractname)) {
 980                 int id = swf_GetPlaceID(tag);
 981                 used[id] = 5;
 982                 found = 1;
 983                 tagused[tagnum] = 1;
 984                 depths[swf_GetDepth(tag)] = 1;
 985                 extractname_id = id;
 986             }
 987         }
 988         else if(tag->id == ST_SHOWFRAME) {
 989             frame ++;
 990             if(hollow) {
 991                 tagused[tagnum] = 1;
 992                 found = 1;
 993             }
 994         }
 995
 996         if(tag->id == ST_DEFINESPRITE) {
 997             while(tag->id != ST_END) {
 998                 tag = tag->next;
 999                 tagnum ++;
1000             }
1001         }
1002         tag = tag->next;
1003         tagnum ++;
1004     }
1005     if (found)
1006         extractTag(&swf, destfilename);
1007
1008     if(mp3file)
1009         fclose(mp3file);
1010
1011     swf_FreeTags(&swf);
1012     return 0;
1013 }
1014