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 file is distributed under the GPL, see file COPYING for details */
13 #include "../lib/rfxswf.h"
14 #include "../lib/args.h"
19 char * destfilename = "output.swf";
23 char* extractframes = 0;
24 char* extractjpegids = 0;
25 char* extractpngids = 0;
27 char* extractname = 0;
31 struct options_t options[] =
45 int args_callback_option(char*name,char*val)
47 if(!strcmp(name, "V")) {
48 printf("swfextract - part of %s %s\n", PACKAGE, VERSION);
51 else if(!strcmp(name, "o")) {
55 else if(!strcmp(name, "i")) {
58 fprintf(stderr, "You can only supply either name or id\n");
63 else if(!strcmp(name, "n")) {
66 fprintf(stderr, "You can only supply either name or id\n");
71 else if(!strcmp(name, "v")) {
75 else if(!strcmp(name, "j")) {
77 fprintf(stderr, "Only one --jpegs argument is allowed. (Try to use a range, e.g. -j 1,2,3)\n");
83 else if(!strcmp(name, "p")) {
85 fprintf(stderr, "Only one --pngs argument is allowed. (Try to use a range, e.g. -p 1,2,3)\n");
91 else if(!strcmp(name, "f")) {
95 else if(!strcmp(name, "w")) {
100 printf("Unknown option: -%s\n", name);
106 int args_callback_longoption(char*name,char*val)
108 return args_long2shortoption(options, name, val);
110 void args_callback_usage(char*name)
112 printf("Usage: %s [-v] [-n name] [-ijf ids] file.swf\n", name);
113 printf("\t-v , --verbose\t\t\t Be more verbose\n");
114 printf("\t-o , --output filename\t\t set output filename\n");
115 printf("\t-n , --name name\t\t instance name of the object to extract\n");
116 printf("\t-i , --id IDs\t\t\t ID of the object to extract\n");
117 printf("\t-j , --jpeg IDs\t\t\t IDs of the JPEG pictures to extract\n");
118 printf("\t-p , --pngs IDs\t\t\t IDs of the PNG pictures to extract\n");
119 printf("\t-f , --frame frames\t\t frame numbers to extract\n");
120 printf("\t-w , --hollow\t\t\t hollow mode: don't remove empty frames (use with -f)\n");
121 printf("\t-V , --version\t\t\t Print program version and exit\n");
123 int args_callback_command(char*name,char*val)
126 fprintf(stderr, "Only one file allowed. You supplied at least two. (%s and %s)\n",
133 U8 mainr,maing,mainb;
134 /* 1 = used, not expanded,
136 5 = wanted, not expanded
144 void idcallback(void*data)
146 if(!(used[*(U16*)data]&1)) {
148 used[*(U16*)data] |= 1;
152 void enumerateIDs(TAG*tag, void(*callback)(void*))
158 data = (U8*)malloc(len);
159 *(U16*)data = (tag->id<<6)+63;
160 *(U32*)&data[2] = tag->len;
161 memcpy(&data[6], tag->data, tag->len);
164 data = (U8*)malloc(len);
165 *(U16*)data = (tag->id<<6)+tag->len;
166 memcpy(&data[2], tag->data, tag->len);
168 map_ids_mem(data, len, callback);
171 void extractTag(SWF*swf, char*filename)
182 memset(&newswf,0x00,sizeof(SWF)); // set global movie parameters
184 newswf.fileVersion = swf->fileVersion;
185 newswf.frameRate = swf->frameRate;
186 newswf.movieSize = swf->movieSize;
188 newswf.firstTag = swf_InsertTag(NULL,ST_SETBACKGROUNDCOLOR);
189 desttag = newswf.firstTag;
193 swf_SetRGB(desttag,&rgb);
197 for(t=0;t<65536;t++) {
198 if(used[t] && !(used[t]&2)) {
199 if(tags[t]->id==ST_DEFINESPRITE) {
201 while(tag->id != ST_END)
203 enumerateIDs(tag, idcallback);
208 enumerateIDs(tags[t], idcallback);
215 srctag = swf->firstTag;
218 while(srctag && (srctag->id || sprite)) {
223 if(srctag->id == ST_END) {
226 if(srctag->id == ST_DEFINESPRITE)
228 if(swf_isDefiningTag(srctag)) {
229 int id = swf_GetDefineID(srctag);
233 if (((srctag->id == ST_PLACEOBJECT ||
234 srctag->id == ST_PLACEOBJECT2 ||
235 srctag->id == ST_STARTSOUND) && (used[swf_GetPlaceID(srctag)]&4) ) ||
236 (swf_isPseudoDefiningTag(srctag) && used[swf_GetDefineID(srctag)]) ||
243 if(srctag->id == ST_REMOVEOBJECT) {
244 if(!used[swf_GetPlaceID(srctag)])
249 TAG*ttag = (TAG*)malloc(sizeof(TAG));
250 desttag = swf_InsertTag(desttag, srctag->id);
251 desttag->len = desttag->memsize = srctag->len;
252 desttag->data = malloc(srctag->len);
253 memcpy(desttag->data, srctag->data, srctag->len);
258 srctag = srctag->next;
261 if(!extractframes && !hollow)
262 desttag = swf_InsertTag(desttag,ST_SHOWFRAME);
264 desttag = swf_InsertTag(desttag,ST_END);
266 f = open(filename, O_TRUNC|O_WRONLY|O_CREAT, 0644);
267 if FAILED(swf_WriteSWF(f,&newswf)) fprintf(stderr,"WriteSWF() failed.\n");
270 swf_FreeTags(&newswf); // cleanup
273 void listObjects(SWF*swf)
279 char*names[] = {"Shapes","MovieClips","JPEGs","PNGs","Sounds","Frames"};
280 printf("Objects in file %s:\n",filename);
288 (tag->id == ST_DEFINESHAPE ||
289 tag->id == ST_DEFINESHAPE2 ||
290 tag->id == ST_DEFINESHAPE3)) {
292 sprintf(text,"%d", swf_GetDefineID(tag));
295 if(tag->id == ST_DEFINESPRITE) {
298 sprintf(text,"%d", swf_GetDefineID(tag));
301 while(tag->id != ST_END)
305 if(t == 2 && (tag->id == ST_DEFINEBITS ||
306 tag->id == ST_DEFINEBITSJPEG2 ||
307 tag->id == ST_DEFINEBITSJPEG3)) {
309 sprintf(text,"%d", swf_GetDefineID(tag));
312 if(t == 3 && (tag->id == ST_DEFINEBITSLOSSLESS ||
313 tag->id == ST_DEFINEBITSLOSSLESS2)) {
315 sprintf(text,"%d", swf_GetDefineID(tag));
319 if(t == 4 && (tag->id == ST_DEFINESOUND)) {
321 sprintf(text,"%d", swf_GetDefineID(tag));
324 if(t == 5 && (tag->id == ST_SHOWFRAME)) {
326 sprintf(text,"%d", frame);
334 printf("%s: ", names[t]);
348 void handlejpegtables(TAG*tag)
350 if(tag->id == ST_JPEGTABLES) {
351 jpegtables = tag->data;
352 jpegtablessize = tag->len;
356 FILE* save_fopen(char* name, char* mode)
358 FILE*fi = fopen(name, mode);
360 fprintf(stderr, "Error: Couldn't open %s\n", name);
366 int findjpegboundary(U8*data, int len)
381 /* extract jpeg data out of a tag */
382 void handlejpeg(TAG*tag)
386 sprintf(name, "pic%d.jpeg", *(U16*)tag->data);
387 /* swf jpeg images have two streams, which both start with ff d8 and
388 end with ff d9. The following code handles sorting the middle
389 <ff d9 ff d8> bytes out, so that one stream remains */
390 if(tag->id == ST_DEFINEBITS && tag->len>2 && jpegtables) {
391 fi = save_fopen(name, "wb");
392 fwrite(jpegtables, 1, jpegtablessize-2, fi); //don't write end tag (ff,d8)
393 fwrite(&tag->data[2+2], tag->len-2-2, 1, fi); //don't write start tag (ff,d9)
396 if(tag->id == ST_DEFINEBITSJPEG2 && tag->len>2) {
398 int pos = findjpegboundary(&tag->data[2], tag->len-2);
402 fi = save_fopen(name, "wb");
403 fwrite(&tag->data[2], pos-2, 1, fi);
404 fwrite(&tag->data[pos+4], end-(pos+4), 1, fi);
407 if(tag->id == ST_DEFINEBITSJPEG3 && tag->len>6) {
408 U32 end = *(U32*)&tag->data[2]+6;
409 int pos = findjpegboundary(&tag->data[6], tag->len-6);
413 fi = save_fopen(name, "wb");
414 fwrite(&tag->data[6], pos-6, 1, fi);
415 fwrite(&tag->data[pos+4], end-(pos+4), 1, fi);
422 static U32*crc32_table = 0;
423 static void make_crc32_table(void)
428 crc32_table = (U32*)malloc(1024);
430 for (t = 0; t < 256; t++) {
433 for (s = 0; s < 8; s++) {
434 c = (0xedb88320L*(c&1)) ^ (c >> 1);
439 static void png_write_byte(FILE*fi, U8 byte)
441 fwrite(&byte,1,1,fi);
442 mycrc32 = crc32_table[(mycrc32 ^ byte) & 0xff] ^ (mycrc32 >> 8);
444 static void png_start_chunk(FILE*fi, char*type, int len)
446 U8 mytype[4]={0,0,0,0};
447 U32 mylen = REVERSESWAP32(len);
448 memcpy(mytype,type,strlen(type));
449 fwrite(&mylen, 4, 1, fi);
451 png_write_byte(fi,mytype[0]);
452 png_write_byte(fi,mytype[1]);
453 png_write_byte(fi,mytype[2]);
454 png_write_byte(fi,mytype[3]);
456 static void png_write_bytes(FILE*fi, U8*bytes, int len)
460 png_write_byte(fi,bytes[t]);
462 static void png_write_dword(FILE*fi, U32 dword)
464 png_write_byte(fi,dword>>24);
465 png_write_byte(fi,dword>>16);
466 png_write_byte(fi,dword>>8);
467 png_write_byte(fi,dword);
469 static void png_end_chunk(FILE*fi)
471 U32 tmp = REVERSESWAP32((mycrc32^0xffffffff));
476 /* extract a lossless image (png) out of a tag
477 This routine was originally meant to be a one-pager. I just
478 didn't know png is _that_ much fun. :) -mk
480 void handlelossless(TAG*tag)
497 U8 head[] = {137,80,78,71,13,10,26,10};
499 char alpha = tag->id == ST_DEFINEBITSLOSSLESS2;
507 if(tag->id != ST_DEFINEBITSLOSSLESS &&
508 tag->id != ST_DEFINEBITSLOSSLESS2)
512 format = swf_GetU8(tag);
513 if(format == 3) bpp = 8;
514 if(format == 4) bpp = 16;
515 if(format == 5) bpp = 32;
516 if(format!=3 && format!=5) {
518 fprintf(stderr, "Can't handle 16-bit palette images yet (image %d)\n",id);
520 fprintf(stderr, "Unknown image type %d in image %d\n", format, id);
523 width = swf_GetU16(tag);
524 height = swf_GetU16(tag);
525 if(format == 3) cols = swf_GetU8(tag) + 1;
526 // this is what format means according to the flash specification. (which is
528 // if(format == 4) cols = swf_GetU16(tag) + 1;
529 // if(format == 5) cols = swf_GetU32(tag) + 1;
532 logf("<verbose> Width %d", width);
533 logf("<verbose> Height %d", height);
534 logf("<verbose> Format %d", format);
535 logf("<verbose> Cols %d", cols);
536 logf("<verbose> Bpp %d", bpp);
538 datalen = (width*height*bpp/8+cols*8);
543 data = malloc(datalen);
544 error = uncompress (data, &datalen, &tag->data[tag->pos], tag->len-tag->pos);
545 } while(error == Z_BUF_ERROR);
547 fprintf(stderr, "Zlib error %d (image %d)\n", error, id);
550 logf("<verbose> Uncompressed image is %d bytes (%d colormap)", datalen, (3+alpha)*cols);
553 data2 = malloc(datalen2);
554 palette = (RGBA*)malloc(cols*sizeof(RGBA));
556 for(t=0;t<cols;t++) {
557 palette[t].r = data[pos++];
558 palette[t].g = data[pos++];
559 palette[t].b = data[pos++];
561 palette[t].a = data[pos++];
565 sprintf(name, "pic%d.png", id);
566 fi = save_fopen(name, "wb");
567 fwrite(head,sizeof(head),1,fi);
569 png_start_chunk(fi, "IHDR", 13);
570 png_write_dword(fi,width);
571 png_write_dword(fi,height);
572 png_write_byte(fi,8);
574 png_write_byte(fi,3); //indexed
576 png_write_byte(fi,2); //rgb
579 png_write_byte(fi,0); //compression mode
580 png_write_byte(fi,0); //filter mode
581 png_write_byte(fi,0); //interlace mode
585 png_start_chunk(fi, "PLTE", 768);
587 png_write_byte(fi,palette[t].r);
588 png_write_byte(fi,palette[t].g);
589 png_write_byte(fi,palette[t].b);
596 int srcwidth = width * (bpp/8);
597 datalen3 = width*height*4;
598 data3 = (U8*)malloc(datalen3);
599 for(y=0;y<height;y++)
601 data3[pos2++]=0; //filter type
603 // 32 bit to 24 bit "conversion"
604 for(x=0;x<width;x++) {
605 data3[pos2++]=data[pos+1];
606 data3[pos2++]=data[pos+2];
607 data3[pos2++]=data[pos+3];
608 pos+=4; //ignore padding byte
612 for(x=0;x<srcwidth;x++)
613 data3[pos2++]=data[pos++];
616 pos+=((srcwidth+3)&~3)-srcwidth; //align
621 if(compress (data2, &datalen2, data3, datalen3) != Z_OK) {
622 fprintf(stderr, "zlib error in pic %d\n", id);
625 logf("<verbose> Compressed data is %d bytes", datalen2);
626 png_start_chunk(fi, "IDAT", datalen2);
627 png_write_bytes(fi,data2,datalen2);
629 png_start_chunk(fi, "IEND", 0);
637 int main (int argc,char ** argv)
646 char listavailable = 0;
647 processargs(argc, argv);
649 if(!extractframes && !extractids && ! extractname && !extractjpegids && !extractpngids)
654 fprintf(stderr, "You must supply a filename.\n");
657 initLog(0,-1,0,0,-1, verbose);
659 f = open(filename,O_RDONLY);
663 perror("Couldn't open file: ");
666 if (swf_ReadSWF(f,&swf) < 0)
668 fprintf(stderr, "%s is not a valid SWF file or contains errors.\n",filename);
687 tagused = (char*)malloc(tagnum);
688 memset(tagused, 0, tagnum);
689 memset(used, 0, 65536);
690 memset(depths, 0, 65536);
695 if(swf_isAllowedSpriteTag(tag)) {
697 if(extractframes && is_in_range(frame, extractframes)) {
699 if(tag->id == ST_PLACEOBJECT || tag->id == ST_PLACEOBJECT2) {
700 depths[swf_GetDepth(tag)] = 1;
702 if(tag->id == ST_REMOVEOBJECT || tag->id == ST_REMOVEOBJECT2) {
703 int depth = swf_GetDepth(tag);
706 depths[swf_GetDepth(tag)] = 0;
709 if((tag->id == ST_REMOVEOBJECT || tag->id == ST_REMOVEOBJECT2) &&
710 (depths[swf_GetDepth(tag)]) && hollow) {
712 depths[swf_GetDepth(tag)] = 0;
716 enumerateIDs(tag, idcallback);
722 if(tag->id == ST_JPEGTABLES)
723 handlejpegtables(tag);
725 if(swf_isDefiningTag(tag)) {
726 int id = swf_GetDefineID(tag);
728 if(extractids && is_in_range(id, extractids)) {
732 if(extractjpegids && is_in_range(id, extractjpegids)) {
735 if(extractpngids && is_in_range(id, extractpngids)) {
739 else if (tag->id == ST_SETBACKGROUNDCOLOR) {
740 mainr = tag->data[0];
741 maing = tag->data[1];
742 mainb = tag->data[2];
744 else if(tag->id == ST_PLACEOBJECT2) {
745 char*name = swf_GetName(tag);
746 if(name && extractname && !strcmp(name, extractname)) {
747 int id = swf_GetPlaceID(tag);
751 depths[swf_GetDepth(tag)] = 1;
754 else if(tag->id == ST_SHOWFRAME) {
762 if(tag->id == ST_DEFINESPRITE) {
763 while(tag->id != ST_END) {
772 extractTag(&swf, destfilename);