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