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