#include "./bitio.h"
#include "./MD5.h"
+#include "./os.h"
// internal constants
return l;
}
-int swf_SetBlock(TAG * t,U8 * b,int l)
+int swf_SetBlock(TAG * t,const U8 * b,int l)
// Appends Block to the end of Tagdata, returns size
{ U32 newlen = t->len + l;
swf_ResetWriteBits(t);
}
void swf_SetFixed(TAG * t, double f)
{
- U16 fr = (U16)(f-(int)f)*65536;
+ U16 fr = (U16)((f-(int)f)*65536);
swf_SetU16(t, fr);
swf_SetU16(t, (U16)f - (f<0 && fr!=0));
}
}
void swf_SetFixed8(TAG * t, float f)
{
- U8 fr = (U8)(f-(int)f)*256;
+ U8 fr = (U8)((f-(int)f)*256);
swf_SetU8(t, fr);
swf_SetU8(t, (U8)f - (f<0 && fr!=0));
}
+U32 swf_GetU30(TAG*tag)
+{
+ U32 shift = 0;
+ U32 s = 0;
+ int nr=0;
+ while(1) {
+ U8 b = swf_GetU8(tag);
+ nr++;
+ s|=(b&127)<<shift;
+ shift+=7;
+ if(!(b&128) || shift>=32)
+ break;
+ }
+ /*int nr2= swf_SetU30(0, s);
+ if(nr!=nr2) {
+ printf("Unsigned value %d stored in %d bytes, I'd store it in %d bytes\n", s, nr, nr2);
+ }*/
+ return s;
+}
+
+int swf_SetU30(TAG*tag, U32 u)
+{
+ int nr = 0;
+ do {
+ if(tag)
+ swf_SetU8(tag, (u&~0x7f?0x80:0) | (u&0x7F));
+ u>>=7;
+ nr++;
+ } while(u);
+ return nr;
+}
+
+void swf_SetABCU32(TAG*tag, U32 u)
+{
+ do {
+ swf_SetU8(tag, (u&~0x7f?0x80:0) | (u&0x7F));
+ u>>=7;
+ } while(u);
+}
+U32 swf_GetABCU32(TAG*tag)
+{
+ return swf_GetU30(tag);
+}
+void swf_SetABCS32(TAG*tag, S32 v)
+{
+ swf_SetABCU32(tag, v);
+}
+S32 swf_GetABCS32(TAG*tag)
+{
+ return swf_GetABCU32(tag);
+}
+
+#if 0
+
+/*The AVM2 spec is just plain wrong, claiming that S32 values are sign
+extended. They're not.
+This wastes up to 4 bytes for every negative value. */
+
+void swf_SetABCS32(TAG*tag, S32 s)
+{
+ printf("write S32: %d\n", s);
+ S32 neg = s<0?-1:0;
+ U8 sign = s<0?0x40:0;
+ while(1) {
+ U8 val = s&0x7f;
+ U8 vsign = s&0x40;
+ s>>=7;
+ neg>>=7;
+ if(s==neg && vsign==sign) {
+ /* if the value we now write has the same sign as s
+ and all the remaining bits are equal to the sign of s
+ too, stop writing */
+ swf_SetU8(tag, val);
+ printf("put %02x\n", val);
+ break;
+ } else {
+ swf_SetU8(tag, 0x80 | val);
+ printf("put %02x\n", 0x80|val);
+ }
+ };
+}
+int swf_GetS30(TAG*tag)
+{
+ U32 shift = 0;
+ U32 s = 0;
+ int nr=0;
+ while(1) {
+ U8 b = swf_GetU8(tag);
+ nr++;
+ s|=(b&127)<<shift;
+ shift+=7;
+ if(!(b&128) || shift>=32) {
+ if(b&64) {
+ if(shift<32)
+ s|=0xffffffff<<shift;
+ }
+ break;
+ }
+ }
+ /* It's not uncommon for other applications (Flex for all negative numbers, and
+ Flash for -1) to generate a lot more bytes than would be necessary.
+ int nr2= swf_SetS30(0, s);
+ if(nr!=nr2) {
+ printf("Signed value %d stored in %d bytes, I'd store it in %d bytes\n", s, nr, nr2);
+ }*/
+ return s;
+}
+#endif
+
+int swf_SetU30String(TAG*tag, const char*str, int l)
+{
+ int len=0;
+ len+=swf_SetU30(tag, l);
+ len+=l;
+ swf_SetBlock(tag, (void*)str, l);
+ return len;
+}
+double swf_GetD64(TAG*tag)
+{
+ /* FIXME: this is not big-endian compatible */
+ double value = *(double*)&tag->data[tag->pos];
+ swf_GetU32(tag);
+ swf_GetU32(tag);
+ return value;
+}
+int swf_SetD64(TAG*tag, double v)
+{
+ /* FIXME: this is not big-endian compatible */
+ swf_SetU32(tag, ((U32*)&v)[0]);
+ swf_SetU32(tag, ((U32*)&v)[1]);
+ return 8;
+}
+int swf_GetU24(TAG*tag)
+{
+ int b1 = swf_GetU8(tag);
+ int b2 = swf_GetU8(tag);
+ int b3 = swf_GetU8(tag);
+ return b3<<16|b2<<8|b1;
+}
+int swf_GetS24(TAG*tag)
+{
+ int b1 = swf_GetU8(tag);
+ int b2 = swf_GetU8(tag);
+ int b3 = swf_GetU8(tag);
+ if(b3&0x80) {
+ return -1-((b3<<16|b2<<8|b1)^0xffffff);
+ } else {
+ return b3<<16|b2<<8|b1;
+ }
+}
+int swf_SetU24(TAG*tag, U32 v)
+{
+ if(tag) {
+ if(v&0xff000000)
+ fprintf(stderr, "Error: Overflow in swf_SetU24()\n");
+ swf_SetU8(tag, v);
+ swf_SetU8(tag, v>>8);
+ swf_SetU8(tag, v>>16);
+ }
+ return 3;
+}
+int swf_SetS24(TAG*tag, U32 v)
+{
+ if(tag) {
+ if(!(v&0xff000000))
+ return swf_SetU24(tag, v);
+ if((v&0xff000000)!=0xff000000) {
+ fprintf(stderr, "Error: Overflow in swf_SetS24()\n");
+ }
+ swf_SetU8(tag, v);
+ swf_SetU8(tag, v>>8);
+ swf_SetU8(tag, v>>16);
+ }
+ return 3;
+}
+
+
int swf_SetRGB(TAG * t,RGBA * col)
{ if (!t) return -1;
if (col)
swf_SetU16(t,0);
swf_SetString(t, (U8*)md5string);
+}
+
+void swf_SetString(TAG*t, const char* s)
+{
+ if(!s) {
+ swf_SetU8(t, 0);
+ } else {
+ swf_SetBlock(t,s,strlen(s)+1);
+ }
}
int swf_VerifyPassword(TAG * t, const char * password)
return tag;
}
-int swf_DeleteTag(TAG * t)
-{ if (!t) return -1;
+TAG* swf_DeleteTag(SWF*swf, TAG * t)
+{
+ TAG*next = t->next;
+ if (swf && swf->firstTag==t)
+ swf->firstTag = t->next;
if (t->prev) t->prev->next = t->next;
if (t->next) t->next->prev = t->prev;
if (t->data) rfx_free(t->data);
rfx_free(t);
- return 0;
+ return next;
}
TAG * swf_ReadTag(reader_t*reader, TAG * prev)
t->id!=ST_DEFINEBITSJPEG&&t->id!=ST_DEFINEBITSJPEG2&&t->id!=ST_DEFINEBITSJPEG3);
if (writer)
- { if (short_tag)
+ {
+#ifdef MEASURE
+ int oldpos = writer->pos;
+#endif
+
+ if (short_tag)
{ raw[0] = SWAP16(len|((t->id&0x3ff)<<6));
if (writer->write(writer,raw,2)!=2)
{
#ifdef DEBUG_RFXSWF
else if (t->len) fprintf(stderr,"WriteTag(): Tag Data Error, id=%i\n",t->id);
#endif
+
+#ifdef MEASURE
+ writer->flush(writer);
+ printf("TAG %s costs %d bytes\n", swf_TagGetName(t), writer->pos-oldpos);
+#endif
}
return t->len+(short_tag?2:6);
if(t->id == ST_END)
level--;
t = swf_NextTag(t);
- swf_DeleteTag(tmp);
+ swf_DeleteTag(0, tmp);
}
while (t && level);
if(level)
/* read tags and connect to list */
t = &t1;
- while (t) t = swf_ReadTag(reader,t);
+ while (t) {
+ t = swf_ReadTag(reader,t);
+ if(t && t->id == ST_FILEATTRIBUTES) {
+ swf->fileAttributes = swf_GetU32(t);
+ swf_ResetReadBits(t);
+ }
+ }
swf->firstTag = t1.next;
t1.next->prev = NULL;
}
return reader->pos;
}
+SWF* swf_OpenSWF(char*filename)
+{
+ int fi = open(filename, O_RDONLY|O_BINARY);
+ if(fi<0) {
+ fprintf(stderr, "Failed to open %s\n", filename);
+ return 0;
+ }
+ SWF* swf = rfx_alloc(sizeof(SWF));
+ swf_ReadSWF(fi, swf);
+ close(fi);
+ return swf;
+}
+
int swf_ReadSWF(int handle, SWF * swf)
{
reader_t reader;
return swf_ReadSWF2(&reader, swf);
}
+void swf_ReadABCfile(char*filename, SWF*swf)
+{
+ memset(swf, 0, sizeof(SWF));
+ swf->fileVersion=9;
+ swf->fileAttributes=FILEATTRIBUTE_AS3; //as3
+ TAG*tag = swf->firstTag = swf_InsertTag(0, ST_RAWABC);
+ memfile_t*file = memfile_open(filename);
+ swf_SetBlock(tag, file->data, file->len);
+ memfile_close(file);
+}
+
+int no_extra_tags = 0;
+
+int WriteExtraTags(SWF*swf, writer_t*writer)
+{
+ TAG*t = swf->firstTag;
+ TAG* has_fileattributes=0;
+ int has_scenedescription=0;
+ int has_version_8_action=0;
+ int has_version_9_action=0;
+ int len = 0;
+ while(t) {
+ if(t->id == ST_FILEATTRIBUTES)
+ has_fileattributes = t;
+ if(t->id == ST_SCENEDESCRIPTION)
+ has_scenedescription = 1;
+ if(t->id == ST_DOABC)
+ has_version_9_action=1;
+ /* FIXME: this doesn't yet find actionscript in buttons */
+ if(t->id == ST_DOACTION || t->id == ST_DOINITACTION)
+ has_version_8_action=1;
+ if(t->id == ST_PLACEOBJECT2 && t->len && (t->data[0]&0x80))
+ has_version_8_action=1;
+ t = t->next;
+ }
+ if(has_version_8_action && has_version_9_action) {
+ fprintf(stderr, "Warning: File contains both flash 8 and flash 9 actionscript\n");
+ }
+
+ if(swf->fileVersion >= 9) {
+ if(!has_fileattributes) {
+ U32 flags = swf->fileAttributes|FILEATTRIBUTE_AS3; // 16 = has symbolclass tag | 8 = actionscript3 | 1 = usenetwork
+ if(has_version_8_action && !has_version_9_action)
+ flags &= ~FILEATTRIBUTE_AS3;
+ TAG*fileattrib = swf_InsertTag(0, ST_FILEATTRIBUTES);
+ swf_SetU32(fileattrib, flags);
+ if(writer) {
+ if(swf_WriteTag2(writer, fileattrib)<0)
+ return -1;
+ } else {
+ len += swf_WriteTag(-1,fileattrib);
+ }
+ swf_DeleteTag(0, fileattrib);
+ } else {
+ if(swf_WriteTag2(writer, has_fileattributes)<0)
+ return -1;
+ }
+ if(0 && !has_scenedescription) {
+ TAG*scene = swf_InsertTag(0, ST_SCENEDESCRIPTION);
+ swf_SetU16(scene, 1);
+ swf_SetString(scene, (U8*)"Scene 1");
+ swf_SetU8(scene, 0);
+ if(writer) {
+ if(swf_WriteTag2(writer, scene)<0)
+ return -1;
+ } else {
+ len += swf_WriteTag(-1,scene);
+ }
+ swf_DeleteTag(0, scene);
+ }
+ }
+ return len;
+}
+
int swf_WriteSWF2(writer_t*writer, SWF * swf) // Writes SWF to file, returns length or <0 if fails
{ U32 len;
TAG * t;
writer_t zwriter;
int fileSize = 0;
int inSprite = 0;
- int writer_lastpos = 0;
int ret;
+ writer_t*original_writer = writer;
+ int writer_lastpos = 0;
if (!swf) return -1;
if (!writer) return -1; // the caller should provide a nullwriter, not 0, for querying SWF size
- if(writer) writer_lastpos = writer->pos;
-
- // Insert REFLEX Tag
-
-#ifdef INSERT_RFX_TAG
-
- if ((swf->firstTag && swf->firstTag->id != ST_REFLEX) &&
- (!swf->firstTag->next || (swf->firstTag->next->id != ST_REFLEX &&
- (!swf->firstTag->next->next || (swf->firstTag->next->next->id!=ST_REFLEX)))))
- {
- swf_SetBlock(swf_InsertTagBefore(swf, swf->firstTag,ST_REFLEX),(U8*)"rfx",3);
- }
-
-#endif // INSERT_RFX_TAG
-
- if(swf->fileVersion >= 9) {
- if ((!swf->firstTag || swf->firstTag->id != ST_SCENEDESCRIPTION) &&
- (!swf->firstTag ||
- !swf->firstTag->next || swf->firstTag->next->id != ST_SCENEDESCRIPTION) &&
- (!swf->firstTag ||
- !swf->firstTag->next ||
- !swf->firstTag->next->next || swf->firstTag->next->next->id != ST_SCENEDESCRIPTION))
- {
- TAG*scene = swf_InsertTagBefore(swf, swf->firstTag,ST_SCENEDESCRIPTION);
- swf_SetU16(scene, 1);
- swf_SetString(scene, (U8*)"Scene 1");
- swf_SetU8(scene, 0);
- }
- }
-
- if(swf->fileVersion >= 9) {
- if (swf->firstTag && swf->firstTag->id != ST_FILEATTRIBUTES)
- {
- U32 flags = 0x8; // | 128 = usenetwork, | 16 = Actionscript3 | 8 = hasmetadata
- swf_SetU32(swf_InsertTagBefore(swf, swf->firstTag,ST_FILEATTRIBUTES),flags);
- }
- }
+ if(original_writer) writer_lastpos = original_writer->pos;
// Count Frames + File Size
t = swf->firstTag;
frameCount = 0;
+ if(swf->firstTag && !no_extra_tags) {
+ len += WriteExtraTags(swf, 0);
+ }
while(t) {
len += swf_WriteTag(-1,t);
if(t->id == ST_DEFINESPRITE && !swf_IsFolded(t)) inSprite++;
return -1;
}
+ if(swf->firstTag && !no_extra_tags) {
+ WriteExtraTags(swf, writer);
+ }
t = swf->firstTag;
- while (t)
- { if (swf_WriteTag2(writer, t)<0) return -1;
- t = swf_NextTag(t);
+
+ while (t) {
+ if(no_extra_tags || t->id != ST_FILEATTRIBUTES) {
+ if(swf_WriteTag2(writer, t)<0)
+ return -1;
+ }
+ t = t->next;
}
if(swf->compressed==1 || (swf->compressed==0 && swf->fileVersion>=6) || swf->compressed==8) {
if(swf->compressed != 8) {
zwriter.finish(&zwriter);
- return writer->pos - writer_lastpos;
+ return original_writer->pos - writer_lastpos;
}
return (int)fileSize;
} else {