#include "../lib/rfxswf.h"
#include "../lib/args.h"
#include "../lib/log.h"
-#include "reloc.h"
#ifdef HAVE_ZLIB_H
#ifdef HAVE_LIBZ
#include "zlib.h"
char* extractframes = 0;
char* extractjpegids = 0;
char* extractpngids = 0;
+char* extractsoundids = 0;
char extractmp3 = 0;
char* extractname = 0;
{"j","jpegs"},
{"p","pngs"},
{"m","mp3"},
+ {"s","sound"},
{"n","name"},
{"f","frame"},
{"V","version"},
extractjpegids = val;
return 1;
}
+ else if(!strcmp(name, "s")) {
+ if(extractsoundids) {
+ fprintf(stderr, "Only one --sound argument is allowed. (Try to use a range, e.g. -s 1,2,3)\n");
+ exit(1);
+ }
+ numextracts++;
+ extractsoundids = val;
+ return 1;
+ }
#ifdef _ZLIB_INCLUDED_
else if(!strcmp(name, "p")) {
if(extractpngids) {
printf("Usage: %s [-v] [-n name] [-ijf ids] file.swf\n", name);
printf("\t-v , --verbose\t\t\t Be more verbose\n");
printf("\t-o , --output filename\t\t set output filename\n");
- printf("\t-n , --name name\t\t instance name of the object to extract\n");
- printf("\t-i , --id IDs\t\t\t ID of the object to extract\n");
- printf("\t-j , --jpeg IDs\t\t\t IDs of the JPEG pictures to extract\n");
+ printf("\t-V , --version\t\t\t Print program version and exit\n\n");
+ printf("SWF Subelement extraction:\n");
+ printf("\t-n , --name name\t\t instance name of the object (SWF Define) to extract\n");
+ printf("\t-i , --id ID\t\t\t ID of the object (SWF Define) to extract\n");
+ printf("\t-f , --frame frames\t\t frame numbers to extract\n");
+ printf("\t-w , --hollow\t\t\t hollow mode: don't remove empty frames\n");
+ printf("\t \t\t\t (use with -f)\n");
+ printf("Picture extraction:\n");
+ printf("\t-j , --jpeg ID\t\t\t Extract JPEG picture(s)\n");
#ifdef _ZLIB_INCLUDED_
- printf("\t-p , --pngs IDs\t\t\t IDs of the PNG pictures to extract\n");
+ printf("\t-p , --pngs ID\t\t\t Extract PNG picture(s)\n");
#endif
+ printf("\n");
+ printf("Sound extraction:\n");
printf("\t-m , --mp3\t\t\t Extract main mp3 stream\n");
- printf("\t-f , --frame frames\t\t frame numbers to extract\n");
- printf("\t-w , --hollow\t\t\t hollow mode: don't remove empty frames (use with -f)\n");
- printf("\t-V , --version\t\t\t Print program version and exit\n");
+ printf("\t-s , --sound ID\t\t\t Extract Sound(s)\n");
}
int args_callback_command(char*name,char*val)
{
void enumerateIDs(TAG*tag, void(*callback)(void*))
{
- U8*data;
+/* U8*data;
int len = tag->len;
if(tag->len>=64) {
len += 6;
memcpy(&data[2], tag->data, tag->len);
}
map_ids_mem(data, len, callback);
+ */
+ int num = swf_GetNumUsedIDs(tag);
+ int *ptr = malloc(sizeof(int)*num);
+ int t;
+ swf_GetUsedIDs(tag, ptr);
+ for(t=0;t<num;t++)
+ callback(&tag->data[ptr[t]]);
}
void extractTag(SWF*swf, char*filename)
void handlejpeg(TAG*tag)
{
char name[80];
+ char*filename = name;
FILE*fi;
- sprintf(name, "pic%d.jpeg", GET16(tag->data));
+
+ sprintf(name, "pic%d.jpg", GET16(tag->data));
+ if(numextracts==1) {
+ filename = destfilename;
+ if(!strcmp(filename,"output.swf"))
+ filename = "output.jpg";
+ }
/* swf jpeg images have two streams, which both start with ff d8 and
end with ff d9. The following code handles sorting the middle
<ff d9 ff d8> bytes out, so that one stream remains */
if(tag->id == ST_DEFINEBITS && tag->len>2 && jpegtables) {
- fi = save_fopen(name, "wb");
+ fi = save_fopen(filename, "wb");
fwrite(jpegtables, 1, jpegtablessize-2, fi); //don't write end tag (ff,d8)
fwrite(&tag->data[2+2], tag->len-2-2, 1, fi); //don't write start tag (ff,d9)
fclose(fi);
}
- if(tag->id == ST_DEFINEBITSJPEG2 && tag->len>2) {
+ else if(tag->id == ST_DEFINEBITSJPEG2 && tag->len>2) {
int end = tag->len;
int pos = findjpegboundary(&tag->data[2], tag->len-2);
if(pos<0)
return;
pos+=2;
- fi = save_fopen(name, "wb");
+ fi = save_fopen(filename, "wb");
fwrite(&tag->data[2], pos-2, 1, fi);
fwrite(&tag->data[pos+4], end-(pos+4), 1, fi);
fclose(fi);
}
- if(tag->id == ST_DEFINEBITSJPEG3 && tag->len>6) {
+ else if(tag->id == ST_DEFINEBITSJPEG3 && tag->len>6) {
U32 end = GET32(&tag->data[2])+6;
int pos = findjpegboundary(&tag->data[6], tag->len-6);
if(pos<0)
return;
pos+=6;
- fi = save_fopen(name, "wb");
+ fi = save_fopen(filename, "wb");
fwrite(&tag->data[6], pos-6, 1, fi);
fwrite(&tag->data[pos+4], end-(pos+4), 1, fi);
fclose(fi);
}
+ else {
+ int id = GET16(tag->data);
+ fprintf(stderr, "Object %d is not a JPEG picture!\n",id);
+ exit(1);
+ }
}
#ifdef _ZLIB_INCLUDED_
void handlelossless(TAG*tag)
{
char name[80];
+ char*filename = name;
FILE*fi;
int width, height;
int crc;
make_crc32_table();
if(tag->id != ST_DEFINEBITSLOSSLESS &&
- tag->id != ST_DEFINEBITSLOSSLESS2)
- return;
+ tag->id != ST_DEFINEBITSLOSSLESS2) {
+ int id = GET16(tag->data);
+ fprintf(stderr, "Object %d is not a PNG picture!\n",id);
+ exit(1);
+ }
id =swf_GetU16(tag);
format = swf_GetU8(tag);
}
sprintf(name, "pic%d.png", id);
- fi = save_fopen(name, "wb");
+ if(numextracts==1) {
+ filename = destfilename;
+ if(!strcmp(filename,"output.swf"))
+ filename = "output.png";
+ }
+ fi = save_fopen(filename, "wb");
fwrite(head,sizeof(head),1,fi);
png_start_chunk(fi, "IHDR", 13);
break;
case ST_SOUNDSTREAMHEAD2:
if((tag->data[1]&0x30) == 0x20) {//mp3 compression
- mp3file = fopen("mainstream.mp3", "wb");
+ mp3file = fopen(filename, "wb");
logf("<notice> Writing mp3 data to %s",filename);
}
else
}
}
+void handledefinesound(TAG*tag)
+{
+ U8 flags;
+ U32 samples;
+ char buf[128];
+ char*filename = buf;
+ FILE*fi;
+ U16 id;
+ id = swf_GetU16(tag); //id
+ sprintf(buf, "sound%d.mp3", id);
+
+ if(numextracts==1) {
+ filename = destfilename;
+ if(!strcmp(filename,"output.swf"))
+ filename = "output.mp3";
+ }
+ flags = swf_GetU8(tag);
+ if((flags>>4)!=2) {
+ printf("Sorry, can only extract MP3 sounds. Sound %d is ADPCM or RAW.\n", id);
+ /* not mp3 */
+ return;
+ }
+ samples = swf_GetU32(tag);
+
+ swf_GetU16(tag); //(only for mp3) numsamples_seek
+
+ fi = save_fopen(filename, "wb");
+ fwrite(&tag->data[tag->pos], tag->len - tag->pos, 1, fi);
+ fclose(fi);
+}
+
int main (int argc,char ** argv)
{
TAG*tag;
processargs(argc, argv);
if(!extractframes && !extractids && ! extractname && !extractjpegids && !extractpngids
- && !extractmp3)
+ && !extractmp3 && !extractsoundids)
listavailable = 1;
if(!filename)
if(tag->id == ST_SOUNDSTREAMHEAD ||
tag->id == ST_SOUNDSTREAMHEAD2 ||
tag->id == ST_SOUNDSTREAMBLOCK) {
- handlesoundstream(tag);
+ if(extractmp3)
+ handlesoundstream(tag);
}
if(tag->id == ST_JPEGTABLES)
if(extractjpegids && is_in_range(id, extractjpegids)) {
handlejpeg(tag);
}
+ if(extractsoundids && is_in_range(id, extractsoundids)) {
+ handledefinesound(tag);
+ }
#ifdef _ZLIB_INCLUDED_
if(extractpngids && is_in_range(id, extractpngids)) {
handlelossless(tag);