changed logf to msg.
[swftools.git] / 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     msg("<verbose> Width %d", width);
680     msg("<verbose> Height %d", height);
681     msg("<verbose> Format %d", format);
682     msg("<verbose> Cols %d", cols);
683     msg("<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     msg("<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     msg("<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                 msg("<notice> Writing mp3 data to %s",filename);
804             }
805             else
806                 msg("<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                 msg("<notice> Writing mp3 data to %s",filename);
812             }
813             else
814                 msg("<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|O_BINARY);
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