2 Allows to extract parts of the swf into a new file.
4 Part of the swftools package.
6 Copyright (c) 2001 Matthias Kramm <kramm@quiss.org>
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.
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.
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 */
25 #include "../lib/rfxswf.h"
26 #include "../lib/args.h"
27 #include "../lib/log.h"
31 #define _ZLIB_INCLUDED_
36 char * destfilename = "output.swf";
40 char* extractframes = 0;
41 char* extractjpegids = 0;
42 char* extractfontids = 0;
43 char* extractpngids = 0;
44 char* extractsoundids = 0;
47 char* extractname = 0;
50 char originalplaceobjects = 0;
54 struct options_t options[] =
73 int args_callback_option(char*name,char*val)
75 if(!strcmp(name, "V")) {
76 printf("swfextract - part of %s %s\n", PACKAGE, VERSION);
79 else if(!strcmp(name, "o")) {
83 else if(!strcmp(name, "i")) {
87 fprintf(stderr, "You can only supply either name or id\n");
92 else if(!strcmp(name, "n")) {
96 fprintf(stderr, "You can only supply either name or id\n");
101 else if(!strcmp(name, "v")) {
105 else if(!strcmp(name, "m")) {
110 else if(!strcmp(name, "j")) {
112 fprintf(stderr, "Only one --jpegs argument is allowed. (Try to use a range, e.g. -j 1,2,3)\n");
116 extractjpegids = val;
119 else if(!strcmp(name, "F")) {
121 fprintf(stderr, "Only one --font argument is allowed. (Try to use a range, e.g. -s 1,2,3)\n");
125 extractfontids = val;
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");
134 extractsoundids = val;
137 #ifdef _ZLIB_INCLUDED_
138 else if(!strcmp(name, "p")) {
140 fprintf(stderr, "Only one --png argument is allowed. (Try to use a range, e.g. -p 1,2,3)\n");
148 else if(!strcmp(name, "f")) {
153 else if(!strcmp(name, "P")) {
154 originalplaceobjects = 1;
157 else if(!strcmp(name, "w")) {
162 printf("Unknown option: -%s\n", name);
168 int args_callback_longoption(char*name,char*val)
170 return args_long2shortoption(options, name, val);
172 void args_callback_usage(char*name)
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");
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");
198 int args_callback_command(char*name,char*val)
201 fprintf(stderr, "Only one file allowed. You supplied at least two. (%s and %s)\n",
208 U8 mainr,maing,mainb;
209 /* 1 = used, not expanded,
211 5 = wanted, not expanded
218 int extractname_id = -1;
220 void idcallback(void*data)
222 if(!(used[GET16(data)]&1)) {
224 used[GET16(data)] |= 1;
228 void enumerateIDs(TAG*tag, void(*callback)(void*))
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);
243 data = (U8*)malloc(len);
244 PUT16(data, (tag->id<<6)+tag->len);
245 memcpy(&data[2], tag->data, tag->len);
247 map_ids_mem(data, len, callback);
249 int num = swf_GetNumUsedIDs(tag);
250 int *ptr = malloc(sizeof(int)*num);
252 swf_GetUsedIDs(tag, ptr);
254 callback(&tag->data[ptr[t]]);
257 void extractTag(SWF*swf, char*filename)
269 memset(&newswf,0x00,sizeof(SWF)); // set global movie parameters
271 newswf.fileVersion = swf->fileVersion;
272 newswf.frameRate = swf->frameRate;
273 newswf.movieSize = swf->movieSize;
275 newswf.firstTag = swf_InsertTag(NULL,ST_SETBACKGROUNDCOLOR);
276 desttag = newswf.firstTag;
280 swf_SetRGB(desttag,&rgb);
282 swf_GetRect(0, &objectbbox);
286 for(t=0;t<65536;t++) {
287 if(used[t] && !(used[t]&2)) {
289 msg("<warning> ID %d is referenced, but never defined.", t);
290 } else if(tags[t]->id==ST_DEFINESPRITE) {
292 while(tag->id != ST_END)
294 enumerateIDs(tag, idcallback);
299 enumerateIDs(tags[t], idcallback);
306 srctag = swf->firstTag;
309 while(srctag && (srctag->id || sprite)) {
314 if(srctag->id == ST_END) {
317 if(srctag->id == ST_DEFINESPRITE)
319 if(swf_isDefiningTag(srctag)) {
320 int id = swf_GetDefineID(srctag);
324 b = swf_GetDefineBBox(srctag);
325 swf_ExpandRect2(&objectbbox, &b);
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)]) ||
337 if(srctag->id == ST_REMOVEOBJECT) {
338 if(!used[swf_GetPlaceID(srctag)])
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);
352 srctag = srctag->next;
355 if(!extractframes && !hollow) {
356 if(!originalplaceobjects && (extractids||extractname_id>=0)) {
359 if((objectbbox.xmin|objectbbox.ymin|objectbbox.xmax|objectbbox.ymax)!=0)
360 newswf.movieSize = objectbbox;
361 if(extractname_id>=0) {
362 desttag = swf_InsertTag(desttag, ST_PLACEOBJECT2);
363 swf_ObjectPlace(desttag, extractname_id, extractname_id, 0,0,extractname);
365 for(t=0;t<65536;t++) {
366 if(is_in_range(t, extractids)) {
367 desttag = swf_InsertTag(desttag, ST_PLACEOBJECT2);
368 swf_ObjectPlace(desttag, t, t, 0,0,0);
371 printf("warning! You should use the -P when extracting multiple objects\n");
376 desttag = swf_InsertTag(desttag,ST_SHOWFRAME);
378 desttag = swf_InsertTag(desttag,ST_END);
380 f = open(filename, O_TRUNC|O_WRONLY|O_CREAT|O_BINARY, 0644);
381 if FAILED(swf_WriteSWF(f,&newswf)) fprintf(stderr,"WriteSWF() failed.\n");
384 swf_FreeTags(&newswf); // cleanup
387 int isOfType(int t, TAG*tag)
390 if(t == 0 && (tag->id == ST_DEFINESHAPE ||
391 tag->id == ST_DEFINESHAPE2 ||
392 tag->id == ST_DEFINESHAPE3)) {
395 if(t==1 && tag->id == ST_DEFINESPRITE) {
398 if(t == 2 && (tag->id == ST_DEFINEBITS ||
399 tag->id == ST_DEFINEBITSJPEG2 ||
400 tag->id == ST_DEFINEBITSJPEG3)) {
403 if(t == 3 && (tag->id == ST_DEFINEBITSLOSSLESS ||
404 tag->id == ST_DEFINEBITSLOSSLESS2)) {
407 if(t == 4 && (tag->id == ST_DEFINESOUND)) {
410 if(t == 5 && (tag->id == ST_DEFINEFONT || tag->id == ST_DEFINEFONT2)) {
416 void listObjects(SWF*swf)
422 char*names[] = {"Shape", "MovieClip", "JPEG", "PNG", "Sound", "Font"};
423 printf("Objects in file %s:\n",filename);
425 for(t=0;t<sizeof(names)/sizeof(names[0]);t++) {
427 int lastid = -2, lastprint=-1;
439 printf(" %d %s%s: ID(s) ", nr, names[t], nr>1?"s":"");
444 char show = isOfType(t,tag);
450 id = swf_GetDefineID(tag);
455 if(first || !follow) {
460 if(lastprint + 1 == lastid)
461 printf(", %d, %d", lastid, id);
463 printf("-%d, %d", lastid, id);
473 if(lastprint + 1 == lastid)
474 printf(", %d", lastid);
476 printf("-%d", lastid);
482 printf(" %d Frames: ID(s) 0-%d\n", frame, frame);
484 printf(" 1 Frame: ID(s) 0\n");
487 void handlefont(SWF*swf, TAG*tag)
492 char*filename = name;
495 id = swf_GetDefineID(tag);
496 sprintf(name, "font%d.swf", id);
498 filename = destfilename;
501 swf_FontExtract(swf, id, &f);
503 printf("Couldn't extract font %d\n", id);
507 swf_FontCreateLayout(f);
509 swf_WriteFont(f, filename);
516 void handlejpegtables(TAG*tag)
518 if(tag->id == ST_JPEGTABLES) {
519 jpegtables = tag->data;
520 jpegtablessize = tag->len;
524 FILE* save_fopen(char* name, char* mode)
526 FILE*fi = fopen(name, mode);
528 fprintf(stderr, "Error: Couldn't open %s\n", name);
534 int findjpegboundary(U8*data, int len)
549 /* extract jpeg data out of a tag */
550 void handlejpeg(TAG*tag)
553 char*filename = name;
556 sprintf(name, "pic%d.jpg", GET16(tag->data));
558 filename = destfilename;
559 if(!strcmp(filename,"output.swf"))
560 filename = "output.jpg";
562 /* swf jpeg images have two streams, which both start with ff d8 and
563 end with ff d9. The following code handles sorting the middle
564 <ff d9 ff d8> bytes out, so that one stream remains */
565 if(tag->id == ST_DEFINEBITSJPEG && tag->len>2 && jpegtables) {
566 fi = save_fopen(filename, "wb");
567 fwrite(jpegtables, 1, jpegtablessize-2, fi); //don't write end tag (ff,d8)
568 fwrite(&tag->data[2+2], tag->len-2-2, 1, fi); //don't write start tag (ff,d9)
571 else if(tag->id == ST_DEFINEBITSJPEG2 && tag->len>2) {
573 int pos = findjpegboundary(&tag->data[2], tag->len-2);
576 fi = save_fopen(filename, "wb");
577 fwrite(&tag->data[2], pos-2, 1, fi);
578 fwrite(&tag->data[pos+4], end-(pos+4), 1, fi);
581 fi = save_fopen(filename, "wb");
582 fwrite(&tag->data[2], end-2, 1, fi);
586 else if(tag->id == ST_DEFINEBITSJPEG3 && tag->len>6) {
587 U32 end = GET32(&tag->data[2])+6;
588 int pos = findjpegboundary(&tag->data[6], tag->len-6);
592 fi = save_fopen(filename, "wb");
593 fwrite(&tag->data[6], pos-6, 1, fi);
594 fwrite(&tag->data[pos+4], end-(pos+4), 1, fi);
598 int id = GET16(tag->data);
599 fprintf(stderr, "Object %d is not a JPEG picture!\n",id);
604 #ifdef _ZLIB_INCLUDED_
607 static U32*crc32_table = 0;
608 static void make_crc32_table(void)
613 crc32_table = (U32*)malloc(1024);
615 for (t = 0; t < 256; t++) {
618 for (s = 0; s < 8; s++) {
619 c = (0xedb88320L*(c&1)) ^ (c >> 1);
624 static inline void png_write_byte(FILE*fi, U8 byte)
626 fwrite(&byte,1,1,fi);
627 mycrc32 = crc32_table[(mycrc32 ^ byte) & 0xff] ^ (mycrc32 >> 8);
629 static void png_start_chunk(FILE*fi, char*type, int len)
631 U8 mytype[4]={0,0,0,0};
632 U32 mylen = REVERSESWAP32(len);
633 memcpy(mytype,type,strlen(type));
634 fwrite(&mylen, 4, 1, fi);
636 png_write_byte(fi,mytype[0]);
637 png_write_byte(fi,mytype[1]);
638 png_write_byte(fi,mytype[2]);
639 png_write_byte(fi,mytype[3]);
641 static void png_write_bytes(FILE*fi, U8*bytes, int len)
645 png_write_byte(fi,bytes[t]);
647 static void png_write_dword(FILE*fi, U32 dword)
649 png_write_byte(fi,dword>>24);
650 png_write_byte(fi,dword>>16);
651 png_write_byte(fi,dword>>8);
652 png_write_byte(fi,dword);
654 static void png_end_chunk(FILE*fi)
656 U32 tmp = REVERSESWAP32((mycrc32^0xffffffff));
661 /* extract a lossless image (png) out of a tag
662 This routine was originally meant to be a one-pager. I just
663 didn't know png is _that_ much fun. :) -mk
665 void handlelossless(TAG*tag)
668 char*filename = name;
683 U8 head[] = {137,80,78,71,13,10,26,10};
685 char alpha = tag->id == ST_DEFINEBITSLOSSLESS2;
693 if(tag->id != ST_DEFINEBITSLOSSLESS &&
694 tag->id != ST_DEFINEBITSLOSSLESS2) {
695 int id = GET16(tag->data);
696 fprintf(stderr, "Object %d is not a PNG picture!\n",id);
701 format = swf_GetU8(tag);
702 if(format == 3) bpp = 8;
703 if(format == 4) bpp = 16;
704 if(format == 5) bpp = 32;
705 if(format!=3 && format!=5) {
707 fprintf(stderr, "Can't handle 16-bit palette images yet (image %d)\n",id);
709 fprintf(stderr, "Unknown image type %d in image %d\n", format, id);
712 width = swf_GetU16(tag);
713 height = swf_GetU16(tag);
714 if(format == 3) cols = swf_GetU8(tag) + 1;
715 // this is what format means according to the flash specification. (which is
717 // if(format == 4) cols = swf_GetU16(tag) + 1;
718 // if(format == 5) cols = swf_GetU32(tag) + 1;
721 msg("<verbose> Width %d", width);
722 msg("<verbose> Height %d", height);
723 msg("<verbose> Format %d", format);
724 msg("<verbose> Cols %d", cols);
725 msg("<verbose> Bpp %d", bpp);
727 datalen = (width*height*bpp/8+cols*8);
732 data = malloc(datalen);
733 error = uncompress (data, &datalen, &tag->data[tag->pos], tag->len-tag->pos);
734 } while(error == Z_BUF_ERROR);
736 fprintf(stderr, "Zlib error %d (image %d)\n", error, id);
739 msg("<verbose> Uncompressed image is %d bytes (%d colormap)", datalen, (3+alpha)*cols);
742 data2 = malloc(datalen2);
743 palette = (RGBA*)malloc(cols*sizeof(RGBA));
745 for(t=0;t<cols;t++) {
746 palette[t].r = data[pos++];
747 palette[t].g = data[pos++];
748 palette[t].b = data[pos++];
750 palette[t].a = data[pos++];
754 sprintf(name, "pic%d.png", id);
756 filename = destfilename;
757 if(!strcmp(filename,"output.swf"))
758 filename = "output.png";
760 fi = save_fopen(filename, "wb");
761 fwrite(head,sizeof(head),1,fi);
763 png_start_chunk(fi, "IHDR", 13);
764 png_write_dword(fi,width);
765 png_write_dword(fi,height);
766 png_write_byte(fi,8);
768 png_write_byte(fi,3); //indexed
769 else if(format == 5 && alpha==0)
770 png_write_byte(fi,2); //rgb
771 else if(format == 5 && alpha==1)
772 png_write_byte(fi,6); //rgba
775 png_write_byte(fi,0); //compression mode
776 png_write_byte(fi,0); //filter mode
777 png_write_byte(fi,0); //interlace mode
781 png_start_chunk(fi, "PLTE", 768);
784 png_write_byte(fi,palette[t].r);
785 png_write_byte(fi,palette[t].g);
786 png_write_byte(fi,palette[t].b);
793 int srcwidth = width * (bpp/8);
794 datalen3 = width*height*4;
795 data3 = (U8*)malloc(datalen3);
796 for(y=0;y<height;y++)
798 data3[pos2++]=0; //filter type
801 // 32 bit to 24 bit "conversion"
802 for(x=0;x<width;x++) {
803 data3[pos2++]=data[pos+1];
804 data3[pos2++]=data[pos+2];
805 data3[pos2++]=data[pos+3];
806 pos+=4; //ignore padding byte
809 for(x=0;x<width;x++) {
810 data3[pos2++]=data[pos+1];
811 data3[pos2++]=data[pos+2];
812 data3[pos2++]=data[pos+3];
813 data3[pos2++]=data[pos+0]; //alpha
819 for(x=0;x<srcwidth;x++)
820 data3[pos2++]=data[pos++];
823 pos+=((srcwidth+3)&~3)-srcwidth; //align
828 if(compress (data2, &datalen2, data3, datalen3) != Z_OK) {
829 fprintf(stderr, "zlib error in pic %d\n", id);
832 msg("<verbose> Compressed data is %d bytes", datalen2);
833 png_start_chunk(fi, "IDAT", datalen2);
834 png_write_bytes(fi,data2,datalen2);
836 png_start_chunk(fi, "IEND", 0);
846 void handlesoundstream(TAG*tag)
848 char*filename = "output.mp3";
850 filename = destfilename;
851 if(!strcmp(filename,"output.swf"))
852 filename = "output.mp3";
855 case ST_SOUNDSTREAMHEAD:
856 if((tag->data[1]&0x30) == 0x20) { //mp3 compression
857 mp3file = fopen(filename, "wb");
858 msg("<notice> Writing mp3 data to %s",filename);
861 msg("<error> Soundstream is not mp3");
863 case ST_SOUNDSTREAMHEAD2:
864 if((tag->data[1]&0x30) == 0x20) {//mp3 compression
865 mp3file = fopen(filename, "wb");
866 msg("<notice> Writing mp3 data to %s",filename);
869 msg("<error> Soundstream is not mp3 (2)");
871 case ST_SOUNDSTREAMBLOCK:
873 fwrite(&tag->data[4],tag->len-4,1,mp3file);
878 void handledefinesound(TAG*tag)
888 int rate,bits,stereo;
889 char*rates[] = {"5500","11025","22050","44100"};
890 id = swf_GetU16(tag); //id
892 flags = swf_GetU8(tag);
898 samples = swf_GetU32(tag);
902 if(format == 2) { // mp3
903 swf_GetU16(tag); //numsamples_seek
905 } else if(format == 0) { // raw
906 printf("Sound is RAW, format: %s samples/sec, %d bit, %s\n", rates[rate], bits, stereo?"stereo":"mono");
907 // TODO: convert to WAV
909 } else if(format == 1) { // adpcm
910 printf("Sound is ADPCM, format: %s samples/sec, %d bit, %s\n", rates[rate], bits, stereo?"stereo":"mono");
913 sprintf(buf, "sound%d.%s", id, extension);
915 filename = destfilename;
916 if(!strcmp(filename,"output.swf")) {
917 sprintf(buf, "output.%s", extension);
921 fi = save_fopen(filename, "wb");
922 fwrite(&tag->data[tag->pos], tag->len - tag->pos, 1, fi);
926 int main (int argc,char ** argv)
935 char listavailable = 0;
936 processargs(argc, argv);
938 if(!extractframes && !extractids && ! extractname && !extractjpegids && !extractpngids
939 && !extractmp3 && !extractsoundids && !extractfontids)
944 fprintf(stderr, "You must supply a filename.\n");
947 initLog(0,-1,0,0,-1, verbose);
949 f = open(filename,O_RDONLY|O_BINARY);
953 perror("Couldn't open file: ");
956 if (swf_ReadSWF(f,&swf) < 0)
958 fprintf(stderr, "%s is not a valid SWF file or contains errors.\n",filename);
977 tagused = (char*)malloc(tagnum);
978 memset(tagused, 0, tagnum);
979 memset(used, 0, 65536);
980 memset(depths, 0, 65536);
985 if(swf_isAllowedSpriteTag(tag)) {
987 if(extractframes && is_in_range(frame, extractframes)) {
989 if(tag->id == ST_PLACEOBJECT || tag->id == ST_PLACEOBJECT2) {
990 depths[swf_GetDepth(tag)] = 1;
992 if(tag->id == ST_REMOVEOBJECT || tag->id == ST_REMOVEOBJECT2) {
993 int depth = swf_GetDepth(tag);
996 depths[swf_GetDepth(tag)] = 0;
999 if((tag->id == ST_REMOVEOBJECT || tag->id == ST_REMOVEOBJECT2) &&
1000 (depths[swf_GetDepth(tag)]) && hollow) {
1002 depths[swf_GetDepth(tag)] = 0;
1006 enumerateIDs(tag, idcallback);
1008 tagused[tagnum] = 1;
1012 if(tag->id == ST_SOUNDSTREAMHEAD ||
1013 tag->id == ST_SOUNDSTREAMHEAD2 ||
1014 tag->id == ST_SOUNDSTREAMBLOCK) {
1016 handlesoundstream(tag);
1019 if(tag->id == ST_JPEGTABLES)
1020 handlejpegtables(tag);
1022 if(swf_isDefiningTag(tag)) {
1023 int id = swf_GetDefineID(tag);
1025 if(extractids && is_in_range(id, extractids)) {
1029 if(extractfontids && is_in_range(id, extractfontids)) {
1030 handlefont(&swf, tag);
1032 if(extractjpegids && is_in_range(id, extractjpegids)) {
1035 if(extractsoundids && is_in_range(id, extractsoundids)) {
1036 handledefinesound(tag);
1038 #ifdef _ZLIB_INCLUDED_
1039 if(extractpngids && is_in_range(id, extractpngids)) {
1040 handlelossless(tag);
1044 else if (tag->id == ST_SETBACKGROUNDCOLOR) {
1045 mainr = tag->data[0];
1046 maing = tag->data[1];
1047 mainb = tag->data[2];
1049 else if(tag->id == ST_PLACEOBJECT2) {
1050 char*name = swf_GetName(tag);
1051 if(name && extractname && !strcmp(name, extractname)) {
1052 int id = swf_GetPlaceID(tag);
1055 tagused[tagnum] = 1;
1056 depths[swf_GetDepth(tag)] = 1;
1057 extractname_id = id;
1060 else if(tag->id == ST_SHOWFRAME) {
1063 tagused[tagnum] = 1;
1068 if(tag->id == ST_DEFINESPRITE) {
1069 while(tag->id != ST_END) {
1078 extractTag(&swf, destfilename);