lib/rfxswf.c

   1 /* vi: set sts=2 sw=2 :*/
   2 /* rfxswf.c
   3
   4    Library for creating and reading SWF files or parts of it.
   5    There's a module directory which provides some extended functionality.
   6    Most modules are included at the bottom of this file.
   7
   8    Part of the swftools package.
   9
  10    Copyright (c) 2000-2003 Rainer Böhme <rfxswf@reflex-studio.de>
  11    Copyright (c) 2003 Matthias Kramm <kramm@quiss.org>
  12
  13    This program is free software; you can redistribute it and/or modify
  14    it under the terms of the GNU General Public License as published by
  15    the Free Software Foundation; either version 2 of the License, or
  16    (at your option) any later version.
  17
  18    This program is distributed in the hope that it will be useful,
  19    but WITHOUT ANY WARRANTY; without even the implied warranty of
  20    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21    GNU General Public License for more details.
  22
  23    You should have received a copy of the GNU General Public License
  24    along with this program; if not, write to the Free Software
  25    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  26
  27 #include "mem.h"
  28 #include "rfxswf.h"
  29 #ifdef HAVE_ZLIB
  30 #include <zlib.h>
  31 #endif // HAVE_ZLIB
  32
  33 #ifndef RFXSWF_DISABLESOUND
  34 #ifdef HAVE_LAME
  35 #include "lame/lame.h"
  36 #endif
  37 #endif
  38
  39 #ifdef HAVE_TIME_H
  40 #include <time.h>
  41 #endif
  42
  43 #ifdef HAVE_IO_H
  44 #include <io.h>
  45 #endif
  46
  47 #include "./bitio.h"
  48 #include "./MD5.h"
  49
  50 // internal constants
  51
  52 #define MALLOC_SIZE     128
  53 #define INSERT_RFX_TAG
  54
  55 #define MEMSIZE(l) (((l/MALLOC_SIZE)+1)*MALLOC_SIZE)
  56
  57 // inline wrapper functions
  58
  59 TAG * swf_NextTag(TAG * t) { return t->next; }
  60 TAG * swf_PrevTag(TAG * t) { return t->prev; }
  61 U16   swf_GetTagID(TAG * t)    { return t->id; }
  62 U32   swf_GetTagLen(TAG * t) { return t->len; }
  63 U8*   swf_GetTagLenPtr(TAG * t) { return &(t->data[t->len]); }
  64 U32   swf_GetTagPos(TAG * t)   { return t->pos; }
  65
  66 void swf_SetTagPos(TAG * t,U32 pos)
  67 { swf_ResetReadBits(t);
  68   if (pos<=t->len) t->pos = pos;
  69   else {
  70 #ifdef DEBUG_RFXSWF
  71     fprintf(stderr,"SetTagPos(%d) out of bounds: TagID = %i\n",pos, t->id);
  72 #endif
  73   }
  74 }
  75
  76 char* swf_GetString(TAG*t)
  77 {
  78     int pos = t->pos;
  79     while(t->pos < t->len && swf_GetU8(t));
  80     /* make sure we always have a trailing zero byte */
  81     if(t->pos == t->len) {
  82       if(t->len == t->memsize) {
  83         swf_ResetWriteBits(t);
  84         swf_SetU8(t, 0);
  85         t->len = t->pos;
  86       }
  87       t->data[t->len] = 0;
  88     }
  89     return (char*)&(t->data[pos]);
  90 }
  91
  92 U8 swf_GetU8(TAG * t)
  93 { swf_ResetReadBits(t);
  94   #ifdef DEBUG_RFXSWF
  95     if (t->pos>=t->len)
  96     { fprintf(stderr,"GetU8() out of bounds: TagID = %i\n",t->id);
  97       return 0;
  98     }
  99   #endif
 100   return t->data[t->pos++];
 101 }
 102
 103 U16 swf_GetU16(TAG * t)
 104 { U16 res;
 105   swf_ResetReadBits(t);
 106   #ifdef DEBUG_RFXSWF
 107     if (t->pos>(t->len-2))
 108     { fprintf(stderr,"GetU16() out of bounds: TagID = %i\n",t->id);
 109       return 0;
 110     }
 111   #endif
 112   res = t->data[t->pos] | (t->data[t->pos+1]<<8);
 113   t->pos+=2;
 114   return res;
 115 }
 116
 117 U32 swf_GetU32(TAG * t)
 118 { U32 res;
 119   swf_ResetReadBits(t);
 120   #ifdef DEBUG_RFXSWF
 121     if (t->pos>(t->len-4))
 122     { fprintf(stderr,"GetU32() out of bounds: TagID = %i\n",t->id);
 123       return 0;
 124     }
 125   #endif
 126   res = t->data[t->pos]        | (t->data[t->pos+1]<<8) |
 127        (t->data[t->pos+2]<<16) | (t->data[t->pos+3]<<24);
 128   t->pos+=4;
 129   return res;
 130 }
 131
 132 int swf_GetBlock(TAG * t,U8 * b,int l)
 133 // returns number of bytes written (<=l)
 134 // b = NULL -> skip data
 135 { swf_ResetReadBits(t);
 136   if ((t->len-t->pos)<l) l=t->len-t->pos;
 137   if (b && l) memcpy(b,&t->data[t->pos],l);
 138   t->pos+=l;
 139   return l;
 140 }
 141
 142 int swf_SetBlock(TAG * t,const U8 * b,int l)
 143 // Appends Block to the end of Tagdata, returns size
 144 { U32 newlen = t->len + l;
 145   swf_ResetWriteBits(t);
 146   if (newlen>t->memsize)
 147   { U32  newmem  = MEMSIZE(newlen);
 148     U8 * newdata = (U8*)(rfx_realloc(t->data,newmem));
 149     t->memsize = newmem;
 150     t->data    = newdata;
 151   }
 152   if (b) memcpy(&t->data[t->len],b,l);
 153   else memset(&t->data[t->len],0x00,l);
 154   t->len+=l;
 155   return l;
 156 }
 157
 158 int swf_SetU8(TAG * t,U8 v)
 159 { swf_ResetWriteBits(t);
 160   if ((t->len+1)>t->memsize) return (swf_SetBlock(t,&v,1)==1)?0:-1;
 161   t->data[t->len++] = v;
 162   return 0;
 163 }
 164
 165 int swf_SetU16(TAG * t,U16 v)
 166 { U8 a[2];
 167   a[0] = v&0xff;
 168   a[1] = v>>8;
 169
 170   swf_ResetWriteBits(t);
 171   if ((t->len+2)>t->memsize) return (swf_SetBlock(t,a,2)==2)?0:-1;
 172   t->data[t->len++] = a[0];
 173   t->data[t->len++] = a[1];
 174   return 0;
 175 }
 176 void swf_SetS16(TAG * t,int v)
 177 {
 178     if(v>32767 || v<-32768) {
 179         fprintf(stderr, "Warning: S16 overflow: %d\n", v);
 180     }
 181     swf_SetU16(t, (S16)v);
 182 }
 183
 184 int swf_SetU32(TAG * t,U32 v)
 185 { U8 a[4];
 186   a[0] = v&0xff;        // to ensure correct handling of non-intel byteorder
 187   a[1] = (v>>8)&0xff;
 188   a[2] = (v>>16)&0xff;
 189   a[3] = (v>>24)&0xff;
 190
 191   swf_ResetWriteBits(t);
 192   if ((t->len+4)>t->memsize) return (swf_SetBlock(t,a,4)==4)?0:-1;
 193   t->data[t->len++] = a[0];
 194   t->data[t->len++] = a[1];
 195   t->data[t->len++] = a[2];
 196   t->data[t->len++] = a[3];
 197   return 0;
 198 }
 199
 200 U32 swf_GetBits(TAG * t,int nbits)
 201 { U32 res = 0;
 202   if (!nbits) return 0;
 203   if (!t->readBit) t->readBit = 0x80;
 204   while (nbits)
 205   { res<<=1;
 206     if (t->data[t->pos]&t->readBit) res|=1;
 207     t->readBit>>=1;
 208     nbits--;
 209     if (!t->readBit)
 210     { if (nbits) t->readBit = 0x80;
 211       #ifdef DEBUG_RFXSWF
 212       if (t->pos>=t->len)
 213       { fprintf(stderr,"GetBits() out of bounds: TagID = %i\n",t->id);
 214         return res;
 215       }
 216       #endif
 217       t->pos++;
 218     }
 219   }
 220   return res;
 221 }
 222
 223 S32 swf_GetSBits(TAG * t,int nbits)
 224 { U32 res = swf_GetBits(t,nbits);
 225   if (res&(1<<(nbits-1))) res|=(0xffffffff<<nbits);
 226   return (S32)res;
 227 }
 228
 229 U32 reader_GetBits(reader_t*reader, int nbits)
 230 { return reader_readbits(reader, nbits);
 231 }
 232 S32 reader_GetSBits(reader_t*reader, int nbits)
 233 { U32 res = reader_readbits(reader, nbits);
 234   if (res&(1<<(nbits-1))) res|=(0xffffffff<<nbits);
 235   return (S32)res;
 236 }
 237
 238 int swf_SetBits(TAG * t,U32 v,int nbits)
 239 { U32 bm = 1<<(nbits-1);
 240
 241   while (nbits)
 242   { if (!t->writeBit)
 243     { if (FAILED(swf_SetU8(t,0))) return -1;
 244       t->writeBit = 0x80;
 245     }
 246     if (v&bm) t->data[t->len-1] |= t->writeBit;
 247     bm>>=1;
 248     t->writeBit>>=1;
 249     nbits--;
 250   }
 251   return 0;
 252 }
 253
 254 // Advanced Data Access Functions
 255
 256 double swf_GetFixed(TAG * t)
 257 {
 258   U16 low =  swf_GetU16(t);
 259   U16 high = swf_GetU16(t);
 260   return high + low*(1/65536.0);
 261 }
 262 void swf_SetFixed(TAG * t, double f)
 263 {
 264   U16 fr = (U16)((f-(int)f)*65536);
 265   swf_SetU16(t, fr);
 266   swf_SetU16(t, (U16)f - (f<0 && fr!=0));
 267 }
 268 float swf_GetFixed8(TAG * t)
 269 {
 270   U8 low =  swf_GetU8(t);
 271   U8 high = swf_GetU8(t);
 272   return (float)(high + low*(1/256.0));
 273 }
 274 void swf_SetFixed8(TAG * t, float f)
 275 {
 276   U8 fr = (U8)((f-(int)f)*256);
 277   swf_SetU8(t, fr);
 278   swf_SetU8(t, (U8)f - (f<0 && fr!=0));
 279 }
 280
 281 int swf_SetRGB(TAG * t,RGBA * col)
 282 { if (!t) return -1;
 283   if (col)
 284   { swf_SetU8(t,col->r);
 285     swf_SetU8(t,col->g);
 286     swf_SetU8(t,col->b);
 287   } else swf_SetBlock(t,NULL,3);
 288   return 0;
 289 }
 290 void swf_GetRGB(TAG * t, RGBA * col)
 291 {
 292     RGBA dummy;
 293     if(!col)
 294         col = &dummy;
 295     col->r = swf_GetU8(t);
 296     col->g = swf_GetU8(t);
 297     col->b = swf_GetU8(t);
 298     col->a = 255;
 299 }
 300
 301 int swf_SetRGBA(TAG * t,RGBA * col)
 302 { if (!t) return -1;
 303   if (col)
 304   { swf_SetU8(t,col->r);
 305     swf_SetU8(t,col->g);
 306     swf_SetU8(t,col->b);
 307     swf_SetU8(t,col->a);
 308   } else swf_SetBlock(t,NULL,4);
 309   return 0;
 310 }
 311 void swf_GetRGBA(TAG * t, RGBA * col)
 312 {
 313     RGBA dummy;
 314     if(!col)
 315         col = &dummy;
 316     col->r = swf_GetU8(t);
 317     col->g = swf_GetU8(t);
 318     col->b = swf_GetU8(t);
 319     col->a = swf_GetU8(t);
 320 }
 321
 322 void swf_GetGradient(TAG * tag, GRADIENT * gradient, char alpha)
 323 {
 324     int t;
 325     if(!tag) {
 326       memset(gradient, 0, sizeof(GRADIENT));
 327       return;
 328     }
 329     U8 num = swf_GetU8(tag) & 15;
 330     if(gradient) {
 331         gradient->num = num;
 332         gradient->rgba = (RGBA*)rfx_calloc(sizeof(RGBA)*gradient->num);
 333         gradient->ratios = (U8*)rfx_calloc(sizeof(gradient->ratios[0])*gradient->num);
 334     }
 335     for(t=0;t<num;t++)
 336     {
 337         U8 ratio = swf_GetU8(tag);
 338         RGBA color;
 339         if(!alpha)
 340             swf_GetRGB(tag, &color);
 341         else
 342             swf_GetRGBA(tag, &color);
 343         if(gradient) {
 344           gradient->ratios[t] = ratio;
 345           gradient->rgba[t] = color;
 346         }
 347     }
 348 }
 349
 350 void swf_SetGradient(TAG * tag, GRADIENT * gradient, char alpha)
 351 {
 352     int t;
 353     if(!tag) {
 354       memset(gradient, 0, sizeof(GRADIENT));
 355       return;
 356     }
 357     swf_SetU8(tag, gradient->num);
 358     for(t=0; t<8 && t<gradient->num; t++)
 359     {
 360         swf_SetU8(tag, gradient->ratios[t]);
 361         if(!alpha)
 362             swf_SetRGB(tag, &gradient->rgba[t]);
 363         else
 364             swf_SetRGBA(tag, &gradient->rgba[t]);
 365     }
 366 }
 367
 368 void swf_FreeGradient(GRADIENT* gradient)
 369 {
 370   if(gradient->ratios)
 371     rfx_free(gradient->ratios);
 372   if(gradient->rgba)
 373     rfx_free(gradient->rgba);
 374   memset(gradient, 0, sizeof(GRADIENT));
 375 }
 376
 377 int swf_CountUBits(U32 v,int nbits)
 378 { int n = 32;
 379   U32 m = 0x80000000;
 380   if(v == 0x00000000) n = 0;
 381   else
 382     while (!(v&m))
 383     { n--;
 384       m>>=1;
 385     }
 386   return (n>nbits)?n:nbits;
 387 }
 388
 389 int swf_CountBits(U32 v,int nbits)
 390 { int n = 33;
 391   U32 m = 0x80000000;
 392   if (v&m)
 393   { if(v == 0xffffffff) n = 1;
 394     else
 395     while (v&m)
 396     { n--;
 397       m>>=1;
 398     }
 399   }
 400   else
 401   { if(v == 0x00000000) n = 0;
 402     else
 403     while (!(v&m))
 404     { n--;
 405       m>>=1;
 406     }
 407   }
 408   return (n>nbits)?n:nbits;
 409 }
 410
 411 int swf_GetRect(TAG * t,SRECT * r)
 412 { int nbits;
 413   SRECT dummy;
 414   if(!t) {r->xmin=r->xmax=r->ymin=r->ymax=0;return 0;}
 415   if (!r) r = &dummy;
 416   nbits = (int) swf_GetBits(t,5);
 417   r->xmin = swf_GetSBits(t,nbits);
 418   r->xmax = swf_GetSBits(t,nbits);
 419   r->ymin = swf_GetSBits(t,nbits);
 420   r->ymax = swf_GetSBits(t,nbits);
 421   return 0;
 422 }
 423
 424 int reader_GetRect(reader_t*reader,SRECT * r)
 425 { int nbits;
 426   SRECT dummy;
 427   if (!r) r = &dummy;
 428   nbits = (int) reader_GetBits(reader,5);
 429   r->xmin = reader_GetSBits(reader,nbits);
 430   r->xmax = reader_GetSBits(reader,nbits);
 431   r->ymin = reader_GetSBits(reader,nbits);
 432   r->ymax = reader_GetSBits(reader,nbits);
 433   return 0;
 434 }
 435
 436 int swf_SetRect(TAG * t,SRECT * r)
 437 { int nbits;
 438
 439   nbits = swf_CountBits(r->xmin,0);
 440   nbits = swf_CountBits(r->xmax,nbits);
 441   nbits = swf_CountBits(r->ymin,nbits);
 442   nbits = swf_CountBits(r->ymax,nbits);
 443   if(nbits>=32) {
 444     fprintf(stderr, "rfxswf: Warning: num_bits overflow in swf_SetRect\n");
 445     nbits=31;
 446   }
 447
 448   swf_SetBits(t,nbits,5);
 449   swf_SetBits(t,r->xmin,nbits);
 450   swf_SetBits(t,r->xmax,nbits);
 451   swf_SetBits(t,r->ymin,nbits);
 452   swf_SetBits(t,r->ymax,nbits);
 453
 454   return 0;
 455 }
 456
 457 SRECT swf_ClipRect(SRECT border, SRECT r)
 458 {
 459     if(r.xmax > border.xmax) r.xmax = border.xmax;
 460     if(r.ymax > border.ymax) r.ymax = border.ymax;
 461     if(r.xmax < border.xmin) r.xmax = border.xmin;
 462     if(r.ymax < border.ymin) r.ymax = border.ymin;
 463
 464     if(r.xmin > border.xmax) r.xmin = border.xmax;
 465     if(r.ymin > border.ymax) r.ymin = border.ymax;
 466     if(r.xmin < border.xmin) r.xmin = border.xmin;
 467     if(r.ymin < border.ymin) r.ymin = border.ymin;
 468     return r;
 469 }
 470
 471 void swf_ExpandRect(SRECT*src, SPOINT add)
 472 {
 473     if((src->xmin | src->ymin | src->xmax | src->ymax)==0) {
 474         src->xmin = add.x;
 475         src->ymin = add.y;
 476         src->xmax = add.x;
 477         src->ymax = add.y;
 478         if((add.x|add.y) == 0) src->xmax++; //make sure the bbox is not NULL anymore
 479         return;
 480     }
 481     if(add.x < src->xmin)
 482         src->xmin = add.x;
 483     if(add.x > src->xmax)
 484         src->xmax = add.x;
 485     if(add.y < src->ymin)
 486         src->ymin = add.y;
 487     if(add.y > src->ymax)
 488         src->ymax = add.y;
 489 }
 490 void swf_ExpandRect2(SRECT*src, SRECT*add)
 491 {
 492     if((add->xmin | add->ymin | add->xmax | add->ymax)==0)
 493         return;
 494     if((src->xmin | src->ymin | src->xmax | src->ymax)==0)
 495         *src = *add;
 496     if(add->xmin < src->xmin)
 497         src->xmin = add->xmin;
 498     if(add->ymin < src->ymin)
 499         src->ymin = add->ymin;
 500     if(add->xmax > src->xmax)
 501         src->xmax = add->xmax;
 502     if(add->ymax > src->ymax)
 503         src->ymax = add->ymax;
 504 }
 505 void swf_ExpandRect3(SRECT*src, SPOINT center, int radius)
 506 {
 507     if((src->xmin | src->ymin | src->xmax | src->ymax)==0) {
 508         src->xmin = center.x-radius;
 509         src->ymin = center.y-radius;
 510         src->xmax = center.x+radius;
 511         src->ymax = center.y+radius;
 512         if((center.x|center.y|radius) == 0) src->xmax++; //make sure the bbox is not NULL anymore
 513         return;
 514     }
 515     if(center.x - radius < src->xmin)
 516         src->xmin = center.x - radius;
 517     if(center.x + radius > src->xmax)
 518         src->xmax = center.x + radius;
 519     if(center.y - radius < src->ymin)
 520         src->ymin = center.y - radius;
 521     if(center.y + radius > src->ymax)
 522         src->ymax = center.y + radius;
 523 }
 524 SPOINT swf_TurnPoint(SPOINT p, MATRIX* m)
 525 {
 526     SPOINT r;
 527     r.x = (int)(m->sx*(1/65536.0)*p.x + m->r1*(1/65536.0)*p.y + 0.5) + m->tx;
 528     r.y = (int)(m->r0*(1/65536.0)*p.x + m->sy*(1/65536.0)*p.y + 0.5) + m->ty;
 529     return r;
 530 }
 531 SRECT swf_TurnRect(SRECT r, MATRIX* m)
 532 {
 533     SRECT g;
 534     SPOINT p1,p2,p3,p4,pp1,pp2,pp3,pp4;
 535     if(!m)
 536       return r;
 537     p1.x = r.xmin;p1.y = r.ymin;
 538     p2.x = r.xmax;p2.y = r.ymin;
 539     p3.x = r.xmin;p3.y = r.ymax;
 540     p4.x = r.xmax;p4.y = r.ymax;
 541     pp1 = swf_TurnPoint(p1, m);
 542     pp2 = swf_TurnPoint(p2, m);
 543     pp3 = swf_TurnPoint(p3, m);
 544     pp4 = swf_TurnPoint(p4, m);
 545     g.xmin = g.xmax = pp1.x;
 546     g.ymin = g.ymax = pp1.y;
 547     swf_ExpandRect(&g, pp2);
 548     swf_ExpandRect(&g, pp3);
 549     swf_ExpandRect(&g, pp4);
 550     return g;
 551 }
 552
 553
 554 int swf_GetMatrix(TAG * t,MATRIX * m)
 555 { MATRIX dummy;
 556   int nbits;
 557
 558   if (!m) m = &dummy;
 559
 560   if (!t)
 561   { m->sx = m->sy = 0x10000;
 562     m->r0 = m->r1 = 0;
 563     m->tx = m->ty = 0;
 564     return -1;
 565   }
 566
 567   swf_ResetReadBits(t);
 568
 569   if (swf_GetBits(t,1))
 570   { nbits = swf_GetBits(t,5);
 571     m->sx = swf_GetSBits(t,nbits);
 572     m->sy = swf_GetSBits(t,nbits);
 573   }
 574   else m->sx = m->sy = 0x10000;
 575
 576   if (swf_GetBits(t,1))
 577   { nbits = swf_GetBits(t,5);
 578     m->r0 = swf_GetSBits(t,nbits);
 579     m->r1 = swf_GetSBits(t,nbits);
 580   }
 581   else m->r0 = m->r1 = 0x0;
 582
 583   nbits = swf_GetBits(t,5);
 584   m->tx = swf_GetSBits(t,nbits);
 585   m->ty = swf_GetSBits(t,nbits);
 586
 587   return 0;
 588 }
 589
 590 int swf_SetMatrix(TAG * t,MATRIX * m)
 591 { int nbits;
 592   MATRIX ma;
 593
 594   if (!m)
 595   { m = &ma;
 596     ma.sx = ma.sy = 0x10000;
 597     ma.r0 = ma.r1 = 0;
 598     ma.tx = ma.ty = 0;
 599   }
 600
 601   swf_ResetWriteBits(t);
 602
 603   if ((m->sx==0x10000)&&(m->sy==0x10000)) swf_SetBits(t,0,1);
 604   else
 605   { swf_SetBits(t,1,1);
 606     nbits = swf_CountBits(m->sx,0);
 607     nbits = swf_CountBits(m->sy,nbits);
 608     if(nbits>=32) {
 609         /* TODO: happens on AMD64 systems for normal values? */
 610         fprintf(stderr,"rfxswf: Error: matrix values too large\n");
 611         nbits = 31;
 612     }
 613     swf_SetBits(t,nbits,5);
 614     swf_SetBits(t,m->sx,nbits);
 615     swf_SetBits(t,m->sy,nbits);
 616   }
 617
 618   if ((!m->r0)&&(!m->r1)) swf_SetBits(t,0,1);
 619   else
 620   { swf_SetBits(t,1,1);
 621     nbits = swf_CountBits(m->r0,0);
 622     nbits = swf_CountBits(m->r1,nbits);
 623     if(nbits>=32) {
 624         fprintf(stderr,"rfxswf: Error: matrix values too large\n");
 625         nbits = 31;
 626     }
 627     swf_SetBits(t,nbits,5);
 628     swf_SetBits(t,m->r0,nbits);
 629     swf_SetBits(t,m->r1,nbits);
 630   }
 631
 632   nbits = swf_CountBits(m->tx,0);
 633   nbits = swf_CountBits(m->ty,nbits);
 634   if(nbits>=32) {
 635       fprintf(stderr,"rfxswf: Error: matrix values too large\n");
 636       nbits = 31;
 637   }
 638   swf_SetBits(t,nbits,5);
 639   swf_SetBits(t,m->tx,nbits);
 640   swf_SetBits(t,m->ty,nbits);
 641
 642   return 0;
 643 }
 644
 645 int swf_GetCXForm(TAG * t,CXFORM * cx,U8 alpha)
 646 { CXFORM cxf;
 647   int hasadd;
 648   int hasmul;
 649   int nbits;
 650
 651   if (!cx) cx = &cxf;
 652
 653   cx->a0 = cx->r0 = cx->g0 = cx->b0 = 256;
 654   cx->a1 = cx->r1 = cx->g1 = cx->b1 = 0;
 655
 656   if (!t) return 0;
 657
 658   swf_ResetReadBits(t);
 659   hasadd = swf_GetBits(t,1);
 660   hasmul = swf_GetBits(t,1);
 661   nbits  = swf_GetBits(t,4);
 662
 663   if (hasmul)
 664   { cx->r0 = (S16)swf_GetSBits(t,nbits);
 665     cx->g0 = (S16)swf_GetSBits(t,nbits);
 666     cx->b0 = (S16)swf_GetSBits(t,nbits);
 667     if (alpha)
 668       cx->a0 = (S16)swf_GetSBits(t,nbits);
 669   }
 670
 671   if (hasadd)
 672   { cx->r1 = (S16)swf_GetSBits(t,nbits);
 673     cx->g1 = (S16)swf_GetSBits(t,nbits);
 674     cx->b1 = (S16)swf_GetSBits(t,nbits);
 675     if (alpha)
 676       cx->a1 = (S16)swf_GetSBits(t,nbits);
 677   }
 678
 679   return 0;
 680 }
 681
 682 int swf_SetCXForm(TAG * t,CXFORM * cx,U8 alpha)
 683 { CXFORM cxf;
 684   int hasadd;
 685   int hasmul;
 686   int nbits;
 687
 688   if (!cx)
 689   { cx = &cxf;
 690     cx->a0 = cx->r0 = cx->g0 = cx->b0 = 256;
 691     cx->a1 = cx->r1 = cx->g1 = cx->b1 = 0;
 692   }
 693
 694   if (!alpha)
 695   { cx->a0 = 256;
 696     cx->a1 = 0;
 697   }
 698
 699   nbits = 0;
 700
 701   hasmul = (cx->a0!=256)||(cx->r0!=256)||(cx->g0!=256)||(cx->b0!=256);
 702   hasadd = cx->a1|cx->r1|cx->g1|cx->b1;
 703
 704   if (hasmul)
 705   { if (alpha) nbits = swf_CountBits((S32)cx->a0,nbits);
 706     nbits = swf_CountBits((S32)cx->r0,nbits);
 707     nbits = swf_CountBits((S32)cx->g0,nbits);
 708     nbits = swf_CountBits((S32)cx->b0,nbits);
 709   }
 710
 711   if (hasadd)
 712   { if (alpha) nbits = swf_CountBits((S32)cx->a1,nbits);
 713     nbits = swf_CountBits((S32)cx->r1,nbits);
 714     nbits = swf_CountBits((S32)cx->g1,nbits);
 715     nbits = swf_CountBits((S32)cx->b1,nbits);
 716   }
 717
 718   swf_ResetWriteBits(t);
 719   swf_SetBits(t,hasadd?1:0,1);
 720   swf_SetBits(t,hasmul?1:0,1);
 721   swf_SetBits(t,nbits,4);
 722
 723   if (hasmul)
 724   { swf_SetBits(t,cx->r0,nbits);
 725     swf_SetBits(t,cx->g0,nbits);
 726     swf_SetBits(t,cx->b0,nbits);
 727     if (alpha) swf_SetBits(t,cx->a0,nbits);
 728   }
 729
 730   if (hasadd)
 731   { swf_SetBits(t,cx->r1,nbits);
 732     swf_SetBits(t,cx->g1,nbits);
 733     swf_SetBits(t,cx->b1,nbits);
 734     if (alpha) swf_SetBits(t,cx->a1,nbits);
 735   }
 736
 737   return 0;
 738 }
 739
 740 //int swf_GetPoint(TAG * t,SPOINT * p) { return 0; }
 741 //int swf_SetPoint(TAG * t,SPOINT * p) { return 0; }
 742
 743 void  swf_SetPassword(TAG * t, const char * password)
 744 {
 745     /* WARNING: crypt_md5 is not reentrant */
 746     char salt[3];
 747     char* md5string;
 748
 749 #if defined(HAVE_LRAND48) && defined(HAVE_SRAND48) && defined(HAVE_TIME_H) && defined(HAVE_TIME)
 750     srand48(time(0));
 751     salt[0] = "abcdefghijklmnopqrstuvwxyz0123456789"[lrand48()%36];
 752     salt[1] = "abcdefghijklmnopqrstuvwxyz0123456789"[lrand48()%36];
 753 #else
 754     salt[0] = 'l';
 755     salt[1] = '8';
 756     fprintf(stderr, "rfxswf: Warning- no usable random generator found\n");
 757     fprintf(stderr, "Your password will be vulnerable to dictionary attacks\n");
 758 #endif
 759     salt[2] = 0;
 760
 761     md5string = crypt_md5(password, salt);
 762
 763     swf_SetU16(t,0);
 764     swf_SetString(t, (U8*)md5string);
 765 }
 766
 767 int swf_VerifyPassword(TAG * t, const char * password)
 768 {
 769     char*md5string1, *md5string2;
 770     char*x;
 771     char*salt;
 772     int n;
 773
 774     if(t->len >= 5 && t->pos==0 &&
 775        t->data[0] == 0 &&
 776        t->data[1] == 0) {
 777       swf_GetU16(t);
 778     } else {
 779       printf("%d %d %d %d\n", t->len, t->pos, t->data[0], t->data[1]);
 780     }
 781
 782     md5string1 = swf_GetString(t);
 783
 784     if(strncmp(md5string1, "$1$",3 )) {
 785         fprintf(stderr, "rfxswf: no salt in pw string\n");
 786         return 0;
 787     }
 788     x = strchr(md5string1+3, '$');
 789     if(!x) {
 790         fprintf(stderr, "rfxswf: invalid salt format in pw string\n");
 791         return 0;
 792     }
 793     n = x-(md5string1+3);
 794     salt = (char*)rfx_alloc(n+1);
 795     memcpy(salt, md5string1+3, n);
 796     salt[n] = 0;
 797
 798     md5string2 = crypt_md5(password, salt);
 799     rfx_free(salt);
 800     if(strcmp(md5string1, md5string2) != 0)
 801         return 0;
 802     return 1;
 803 }
 804
 805 // Tag List Manipulating Functions
 806
 807 TAG * swf_InsertTag(TAG * after,U16 id)
 808 { TAG * t;
 809
 810   t = (TAG *)rfx_calloc(sizeof(TAG));
 811   t->id = id;
 812
 813   if (after)
 814   {
 815     t->prev  = after;
 816     t->next  = after->next;
 817     after->next = t;
 818     if (t->next) t->next->prev = t;
 819   }
 820   return t;
 821 }
 822
 823 TAG * swf_InsertTagBefore(SWF* swf, TAG * before,U16 id)
 824 { TAG * t;
 825
 826   t = (TAG *)rfx_calloc(sizeof(TAG));
 827   t->id = id;
 828
 829   if (before)
 830   {
 831     t->next  = before;
 832     t->prev  = before->prev;
 833     before->prev = t;
 834     if (t->prev) t->prev->next = t;
 835   }
 836   if(swf && swf->firstTag == before) {
 837     swf->firstTag = t;
 838   }
 839   return t;
 840 }
 841
 842 void swf_ClearTag(TAG * t)
 843 {
 844   if (t->data) rfx_free(t->data);
 845   t->data = 0;
 846   t->pos = 0;
 847   t->len = 0;
 848   t->readBit = 0;
 849   t->writeBit = 0;
 850   t->memsize = 0;
 851 }
 852
 853 void swf_ResetTag(TAG*tag, U16 id)
 854 {
 855     tag->len = tag->pos = tag->readBit = tag->writeBit = 0;
 856     tag->id = id;
 857 }
 858
 859 TAG* swf_CopyTag(TAG*tag, TAG*to_copy)
 860 {
 861     tag = swf_InsertTag(tag, to_copy->id);
 862     swf_SetBlock(tag, to_copy->data, to_copy->len);
 863     return tag;
 864 }
 865
 866 int swf_DeleteTag(TAG * t)
 867 { if (!t) return -1;
 868
 869   if (t->prev) t->prev->next = t->next;
 870   if (t->next) t->next->prev = t->prev;
 871
 872   if (t->data) rfx_free(t->data);
 873   rfx_free(t);
 874   return 0;
 875 }
 876
 877 TAG * swf_ReadTag(reader_t*reader, TAG * prev)
 878 { TAG * t;
 879   U16 raw;
 880   U32 len;
 881   int id;
 882
 883   if (reader->read(reader, &raw, 2) !=2 ) return NULL;
 884   raw = SWAP16(raw);
 885
 886   len = raw&0x3f;
 887   id  = raw>>6;
 888
 889   if (len==0x3f)
 890   {
 891       if (reader->read(reader, &len, 4) != 4) return NULL;
 892       len = SWAP32(len);
 893   }
 894
 895   if (id==ST_DEFINESPRITE) len = 2*sizeof(U16);
 896   // Sprite handling fix: Flatten sprite tree
 897
 898   t = (TAG *)rfx_calloc(sizeof(TAG));
 899
 900   t->len = len;
 901   t->id  = id;
 902
 903   if (t->len)
 904   { t->data = (U8*)rfx_alloc(t->len);
 905     t->memsize = t->len;
 906     if (reader->read(reader, t->data, t->len) != t->len) {
 907       fprintf(stderr, "rfxswf: Warning: Short read (tagid %d). File truncated?\n", t->id);
 908       free(t->data);t->data=0;
 909       free(t);
 910       return NULL;
 911     }
 912   }
 913
 914   if (prev)
 915   {
 916     t->prev  = prev;
 917     prev->next = t;
 918   }
 919
 920   return t;
 921 }
 922
 923 int swf_DefineSprite_GetRealSize(TAG * t);
 924
 925 int swf_WriteTag2(writer_t*writer, TAG * t)
 926 // returns tag length in bytes (incl. Header), -1 = Error
 927 // writer = 0 -> no output
 928 { U16 raw[3];
 929   U32 len;
 930   int short_tag;
 931
 932   if (!t) return -1;
 933
 934   len = (t->id==ST_DEFINESPRITE)?swf_DefineSprite_GetRealSize(t):t->len;
 935
 936   short_tag = len<0x3f&&
 937     (t->id!=ST_DEFINEBITSLOSSLESS&&t->id!=ST_DEFINEBITSLOSSLESS2&&t->id!=ST_SOUNDSTREAMBLOCK&&
 938      t->id!=ST_DEFINEBITSJPEG&&t->id!=ST_DEFINEBITSJPEG2&&t->id!=ST_DEFINEBITSJPEG3);
 939
 940   if (writer)
 941   {
 942 #ifdef MEASURE
 943   int oldpos = writer->pos;
 944 #endif
 945
 946     if (short_tag)
 947     { raw[0] = SWAP16(len|((t->id&0x3ff)<<6));
 948       if (writer->write(writer,raw,2)!=2)
 949       {
 950         #ifdef DEBUG_RFXSWF
 951           fprintf(stderr,"WriteTag() failed: Short Header.\n");
 952         #endif
 953         return -1;
 954       }
 955     }
 956     else
 957     {
 958       raw[0] = SWAP16((t->id<<6)|0x3f);
 959       if (writer->write(writer,raw,2)!=2)
 960       {
 961 #ifdef DEBUG_RFXSWF
 962           fprintf(stderr,"WriteTag() failed: Long Header (1).\n");
 963 #endif
 964           return -1;
 965       }
 966
 967       len = SWAP32(len);
 968       if (writer->write(writer,&len,4)!=4)
 969       {
 970         #ifdef DEBUG_RFXSWF
 971           fprintf(stderr,"WriteTag() failed: Long Header (2).\n");
 972         #endif
 973         return -1;
 974       }
 975     }
 976
 977     if (t->data)
 978     { if (writer->write(writer,t->data,t->len)!=t->len)
 979       {
 980         #ifdef DEBUG_RFXSWF
 981           fprintf(stderr,"WriteTag() failed: Data.\n");
 982         #endif
 983         return -1;
 984       }
 985     }
 986     #ifdef DEBUG_RFXSWF
 987       else if (t->len) fprintf(stderr,"WriteTag(): Tag Data Error, id=%i\n",t->id);
 988     #endif
 989
 990 #ifdef MEASURE
 991   writer->flush(writer);
 992   printf("TAG %s costs %d bytes\n", swf_TagGetName(t), writer->pos-oldpos);
 993 #endif
 994   }
 995
 996   return t->len+(short_tag?2:6);
 997 }
 998
 999 int swf_WriteTag(int handle, TAG * t)
1000 {
1001   writer_t writer;
1002   int len = 0;
1003   if(handle<0)
1004     return swf_WriteTag2(0, t);
1005   writer_init_filewriter(&writer, handle);
1006   len = swf_WriteTag2(&writer, t);
1007   writer.finish(&writer);
1008   return len;
1009 }
1010
1011 int swf_DefineSprite_GetRealSize(TAG * t)
1012 // Sprite Handling: Helper function to pack DefineSprite-Tag
1013 { U32 len = t->len;
1014   if(len>4) { // folded sprite
1015       return t->len;
1016   }
1017   do
1018   { t = swf_NextTag(t);
1019     if (t && t->id!=ST_DEFINESPRITE) len += swf_WriteTag(-1, t);
1020     else t = NULL;
1021   } while (t&&(t->id!=ST_END));
1022   return len;
1023 }
1024
1025 void swf_UnFoldSprite(TAG * t)
1026 {
1027   U16 id,tmp;
1028   U32 len;
1029   TAG*next = t;
1030   U16 spriteid,spriteframes;
1031   int level;
1032   if(t->id!=ST_DEFINESPRITE)
1033     return;
1034   if(t->len<=4) // not folded
1035     return;
1036
1037   swf_SetTagPos(t,0);
1038
1039   spriteid = swf_GetU16(t); //id
1040   spriteframes = swf_GetU16(t); //frames
1041
1042   level = 1;
1043
1044   while(1)
1045   {
1046     TAG*it = 0;
1047     tmp = swf_GetU16(t);
1048     len = tmp&0x3f;
1049     id  = tmp>>6;
1050     if(id == ST_END)
1051         level--;
1052     if(id == ST_DEFINESPRITE && len<=4)
1053         level++;
1054
1055     if (len==0x3f)
1056         len = swf_GetU32(t);
1057     it = swf_InsertTag(next, id);
1058     next = it;
1059     it->len = len;
1060     it->id  = id;
1061     if (it->len)
1062     { it->data = (U8*)rfx_alloc(it->len);
1063       it->memsize = it->len;
1064       swf_GetBlock(t, it->data, it->len);
1065     }
1066
1067     if(!level)
1068         break;
1069   }
1070
1071   rfx_free(t->data); t->data = 0;
1072   t->memsize = t->len = t->pos = 0;
1073
1074   swf_SetU16(t, spriteid);
1075   swf_SetU16(t, spriteframes);
1076 }
1077
1078 void swf_FoldSprite(TAG * t)
1079 {
1080   TAG*sprtag=t,*tmp;
1081   U16 id,frames;
1082   int level;
1083   if(t->id!=ST_DEFINESPRITE)
1084       return;
1085   if(!t->len) {
1086       fprintf(stderr, "Error: Sprite has no ID!");
1087       return;
1088   }
1089   if(t->len>4) {
1090     /* sprite is already folded */
1091       return;
1092   }
1093
1094   t->pos = 0;
1095   id = swf_GetU16(t);
1096   rfx_free(t->data);
1097   t->len = t->pos = t->memsize = 0;
1098   t->data = 0;
1099
1100   frames = 0;
1101
1102   t = swf_NextTag(sprtag);
1103   level = 1;
1104
1105   do
1106   {
1107     if(t->id==ST_SHOWFRAME) frames++;
1108     if(t->id == ST_DEFINESPRITE && t->len<=4)
1109         level++;
1110     if(t->id == ST_END)
1111         level--;
1112     t = swf_NextTag(t);
1113   } while(t && level);
1114   if(level)
1115     fprintf(stderr, "rfxswf error: sprite doesn't end(1)\n");
1116
1117   swf_SetU16(sprtag, id);
1118   swf_SetU16(sprtag, frames);
1119
1120   t = swf_NextTag(sprtag);
1121   level = 1;
1122
1123   do
1124   {
1125     if(t->len<0x3f&&
1126         (t->id!=ST_DEFINEBITSLOSSLESS&&t->id!=ST_DEFINEBITSLOSSLESS2&&t->id!=ST_SOUNDSTREAMBLOCK&&
1127          t->id!=ST_DEFINEBITSJPEG&&t->id!=ST_DEFINEBITSJPEG2&&t->id!=ST_DEFINEBITSJPEG3)
1128       ) {
1129         swf_SetU16(sprtag,t->len|(t->id<<6));
1130     } else {
1131         swf_SetU16(sprtag,0x3f|(t->id<<6));
1132         swf_SetU32(sprtag,t->len);
1133     }
1134     if(t->len)
1135         swf_SetBlock(sprtag,t->data, t->len);
1136     tmp = t;
1137     if(t->id == ST_DEFINESPRITE && t->len<=4)
1138         level++;
1139     if(t->id == ST_END)
1140         level--;
1141     t = swf_NextTag(t);
1142     swf_DeleteTag(tmp);
1143   }
1144   while (t && level);
1145   if(level)
1146     fprintf(stderr, "rfxswf error: sprite doesn't end(2)\n");
1147
1148 //  sprtag->next = t;
1149 //  t->prev = sprtag;
1150 }
1151
1152 int swf_IsFolded(TAG * t)
1153 {
1154     return (t->id == ST_DEFINESPRITE && t->len>4);
1155 }
1156
1157 void swf_FoldAll(SWF*swf)
1158 {
1159     TAG*tag = swf->firstTag;
1160     //swf_DumpSWF(stdout, swf);
1161     while(tag) {
1162         if(tag->id == ST_DEFINESPRITE) {
1163             swf_FoldSprite(tag);
1164             //swf_DumpSWF(stdout, swf);
1165         }
1166         tag = swf_NextTag(tag);
1167     }
1168 }
1169
1170 void swf_UnFoldAll(SWF*swf)
1171 {
1172     TAG*tag = swf->firstTag;
1173     while(tag) {
1174         if(tag->id == ST_DEFINESPRITE)
1175             swf_UnFoldSprite(tag);
1176         tag = tag->next;
1177     }
1178 }
1179
1180 void swf_OptimizeTagOrder(SWF*swf)
1181 {
1182   TAG*tag,*next;
1183   TAG*level0;
1184   int level;
1185   int changes;
1186   swf_UnFoldAll(swf);
1187   /* at the moment, we don't actually do optimizing,
1188      only fixing of non-spec-conformant things like
1189      sprite tags */
1190
1191   do {
1192     changes = 0;
1193     level = 0;
1194     level0 = 0;
1195     tag = swf->firstTag;
1196     while(tag) {
1197       next = tag->next;
1198       if(tag->id == ST_DEFINESPRITE) {
1199         if(tag->len>4) {
1200           /* ??? all sprites are supposed to be unfolded */
1201           fprintf(stderr, "librfxswf error - internal error in OptimizeTagOrder/UnfoldAll\n");
1202         }
1203         level++;
1204         if(level==1) {
1205           level0 = tag;
1206           tag = next;
1207           continue;
1208         }
1209       }
1210       if(level>=1) {
1211         /* move non-sprite tags out of sprite */
1212         if(!swf_isAllowedSpriteTag(tag) || level>=2) {
1213           /* remove tag from current position */
1214           tag->prev->next = tag->next;
1215           if(tag->next)
1216             tag->next->prev = tag->prev;
1217
1218           /* insert before tag level0 */
1219           tag->next = level0;
1220           tag->prev = level0->prev;
1221           level0->prev = tag;
1222           if(tag->prev)
1223             tag->prev->next = tag;
1224           else
1225             swf->firstTag = tag;
1226           changes = 1;
1227         }
1228       }
1229       if(tag->id == ST_END) {
1230         level--;
1231       }
1232
1233       tag = next;
1234     }
1235   } while(changes);
1236 }
1237
1238 // Movie Functions
1239
1240 int swf_ReadSWF2(reader_t*reader, SWF * swf)   // Reads SWF to memory (malloc'ed), returns length or <0 if fails
1241 {
1242   if (!swf) return -1;
1243   memset(swf,0x00,sizeof(SWF));
1244
1245   { char b[32];                         // read Header
1246     int len;
1247     TAG * t;
1248     TAG t1;
1249     reader_t zreader;
1250
1251     if ((len = reader->read(reader ,b,8))<8) return -1;
1252
1253     if (b[0]!='F' && b[0]!='C') return -1;
1254     if (b[1]!='W') return -1;
1255     if (b[2]!='S') return -1;
1256     swf->fileVersion = b[3];
1257     swf->compressed  = (b[0]=='C')?1:0;
1258     swf->fileSize    = GET32(&b[4]);
1259
1260     if(swf->compressed) {
1261         reader_init_zlibinflate(&zreader, reader);
1262         reader = &zreader;
1263     }
1264     swf->compressed = 0; // derive from version number from now on
1265
1266     reader_GetRect(reader, &swf->movieSize);
1267     reader->read(reader, &swf->frameRate, 2);
1268     swf->frameRate = SWAP16(swf->frameRate);
1269     reader->read(reader, &swf->frameCount, 2);
1270     swf->frameCount = SWAP16(swf->frameCount);
1271
1272     /* read tags and connect to list */
1273     t = &t1;
1274     while (t) t = swf_ReadTag(reader,t);
1275     swf->firstTag = t1.next;
1276     t1.next->prev = NULL;
1277   }
1278
1279   return reader->pos;
1280 }
1281
1282 int swf_ReadSWF(int handle, SWF * swf)
1283 {
1284   reader_t reader;
1285   reader_init_filereader(&reader, handle);
1286   return swf_ReadSWF2(&reader, swf);
1287 }
1288
1289 int  swf_WriteSWF2(writer_t*writer, SWF * swf)     // Writes SWF to file, returns length or <0 if fails
1290 { U32 len;
1291   TAG * t;
1292   int frameCount=0;
1293   writer_t zwriter;
1294   int fileSize = 0;
1295   int inSprite = 0;
1296   int writer_lastpos = 0;
1297   int ret;
1298
1299   if (!swf) return -1;
1300   if (!writer) return -1; // the caller should provide a nullwriter, not 0, for querying SWF size
1301
1302   if(writer) writer_lastpos = writer->pos;
1303
1304   // Insert REFLEX Tag
1305
1306 #ifdef INSERT_RFX_TAG
1307
1308   if ((swf->firstTag && swf->firstTag->id != ST_REFLEX) &&
1309       (!swf->firstTag->next || (swf->firstTag->next->id != ST_REFLEX &&
1310       (!swf->firstTag->next->next || (swf->firstTag->next->next->id!=ST_REFLEX)))))
1311   {
1312       swf_SetBlock(swf_InsertTagBefore(swf, swf->firstTag,ST_REFLEX),(U8*)"rfx",3);
1313   }
1314
1315 #endif // INSERT_RFX_TAG
1316
1317   if(swf->fileVersion >= 9) {
1318     if ((!swf->firstTag || swf->firstTag->id != ST_SCENEDESCRIPTION) &&
1319         (!swf->firstTag ||
1320          !swf->firstTag->next || swf->firstTag->next->id != ST_SCENEDESCRIPTION) &&
1321         (!swf->firstTag ||
1322          !swf->firstTag->next ||
1323          !swf->firstTag->next->next || swf->firstTag->next->next->id != ST_SCENEDESCRIPTION))
1324     {
1325         TAG*scene = swf_InsertTagBefore(swf, swf->firstTag,ST_SCENEDESCRIPTION);
1326         swf_SetU16(scene, 1);
1327         swf_SetString(scene, (U8*)"Scene 1");
1328         swf_SetU8(scene, 0);
1329     }
1330   }
1331
1332   if(swf->fileVersion >= 9) {
1333       if (swf->firstTag && swf->firstTag->id != ST_FILEATTRIBUTES)
1334       {
1335           U32 flags = 0x8; // | 128 = usenetwork, | 16 = Actionscript3 | 8 = hasmetadata
1336           swf_SetU32(swf_InsertTagBefore(swf, swf->firstTag,ST_FILEATTRIBUTES),flags);
1337       }
1338   }
1339
1340   // Count Frames + File Size
1341
1342   len = 0;
1343   t = swf->firstTag;
1344   frameCount = 0;
1345
1346   while(t) {
1347       len += swf_WriteTag(-1,t);
1348       if(t->id == ST_DEFINESPRITE && !swf_IsFolded(t)) inSprite++;
1349       else if(t->id == ST_END && inSprite) inSprite--;
1350       else if(t->id == ST_END && !inSprite) {
1351         if(t->prev && t->prev->id!=ST_SHOWFRAME)
1352           frameCount++;
1353       }
1354       else if(t->id == ST_SHOWFRAME && !inSprite) frameCount++;
1355       t = swf_NextTag(t);
1356   }
1357
1358   { TAG t1;
1359     char b[64],b4[4];
1360     U32 l;
1361
1362     memset(&t1,0x00,sizeof(TAG));
1363     t1.data    = (U8*)b;
1364     t1.memsize = 64;
1365
1366     { // measure header file size
1367       TAG t2;
1368       char b2[64];
1369       memset(&t2,0x00,sizeof(TAG));
1370       t2.data    = (U8*)b2;
1371       t2.memsize = 64;
1372       swf_SetRect(&t2, &swf->movieSize);
1373       swf_SetU16(&t2, swf->frameRate);
1374       swf_SetU16(&t2, swf->frameCount);
1375       l = swf_GetTagLen(&t2)+8;
1376     }
1377     if(swf->compressed == 8) {
1378       l -= 8;
1379     }
1380
1381     fileSize = l+len;
1382     if(len) {// don't touch headers without tags
1383         swf->fileSize = fileSize;
1384         swf->frameCount = frameCount;
1385     }
1386
1387     if(swf->compressed != 8) {
1388     /* compressed flag set to 8 means "skip first 8
1389        header bytes". This is necessary if the caller wants to
1390        create compressed SWFs himself .
1391        It also means that we don't initialize our own zlib
1392        writer, but assume the caller provided one.
1393      */
1394       if(swf->compressed==1 || (swf->compressed==0 && swf->fileVersion>=6)) {
1395         char*id = "CWS";
1396         writer->write(writer, id, 3);
1397       } else {
1398         char*id = "FWS";
1399         writer->write(writer, id, 3);
1400       }
1401
1402       writer->write(writer, &swf->fileVersion, 1);
1403       PUT32(b4, swf->fileSize);
1404       writer->write(writer, b4, 4);
1405
1406       if(swf->compressed==1 || (swf->compressed==0 && swf->fileVersion>=6)) {
1407         writer_init_zlibdeflate(&zwriter, writer);
1408         writer = &zwriter;
1409       }
1410     }
1411
1412     swf_SetRect(&t1,&swf->movieSize);
1413     swf_SetU16(&t1,swf->frameRate);
1414     swf_SetU16(&t1,swf->frameCount);
1415
1416     ret = writer->write(writer,b,swf_GetTagLen(&t1));
1417     if (ret!=swf_GetTagLen(&t1))
1418     {
1419       #ifdef DEBUG_RFXSWF
1420         fprintf(stderr, "ret:%d\n",ret);
1421         perror("write:");
1422         fprintf(stderr,"WriteSWF() failed: Header.\n");
1423       #endif
1424       return -1;
1425     }
1426
1427     t = swf->firstTag;
1428     while (t)
1429     { if (swf_WriteTag2(writer, t)<0) return -1;
1430       t = swf_NextTag(t);
1431     }
1432     if(swf->compressed==1 || (swf->compressed==0 && swf->fileVersion>=6) || swf->compressed==8) {
1433       if(swf->compressed != 8) {
1434         zwriter.finish(&zwriter);
1435         return writer->pos - writer_lastpos;
1436       }
1437       return (int)fileSize;
1438     } else {
1439       return (int)fileSize;
1440     }
1441   }
1442 }
1443
1444 int  swf_WriteSWF(int handle, SWF * swf)     // Writes SWF to file, returns length or <0 if fails
1445 {
1446   writer_t writer;
1447   int len = 0;
1448
1449   if(handle<0) {
1450     writer_init_nullwriter(&writer);
1451     len = swf_WriteSWF2(&writer, swf);
1452     return len;
1453   }
1454   writer_init_filewriter(&writer, handle);
1455   len = swf_WriteSWF2(&writer, swf);
1456   writer.finish(&writer);
1457   return len;
1458 }
1459
1460 int swf_WriteHeader2(writer_t*writer,SWF * swf)
1461 {
1462   SWF myswf;
1463   memcpy(&myswf,swf,sizeof(SWF));
1464   myswf.firstTag = 0;
1465   return swf_WriteSWF2(writer, &myswf);
1466 }
1467
1468 int swf_WriteHeader(int handle,SWF * swf)
1469 {
1470   SWF myswf;
1471   memcpy(&myswf,swf,sizeof(SWF));
1472   myswf.firstTag = 0;
1473   return swf_WriteSWF(handle, &myswf);
1474 }
1475
1476 int swf_WriteCGI(SWF * swf)
1477 { int len;
1478   char s[1024];
1479
1480   len = swf_WriteSWF(-1,swf);
1481
1482   if (len<0) return -1;
1483
1484   sprintf(s,"Content-type: application/x-shockwave-flash\n"
1485             "Accept-Ranges: bytes\n"
1486             "Content-Length: %lu\n"
1487             "Expires: Thu, 13 Apr 2000 23:59:59 GMT\n"
1488             "\n",len);
1489
1490   write(fileno(stdout),s,strlen(s));
1491   return swf_WriteSWF(fileno(stdout),swf);
1492 }
1493
1494 SWF* swf_CopySWF(SWF*swf)
1495 {
1496     SWF*nswf = (SWF*)rfx_alloc(sizeof(SWF));
1497     TAG*tag, *ntag;
1498     memcpy(nswf, swf, sizeof(SWF));
1499     nswf->firstTag = 0;
1500     tag = swf->firstTag;
1501     ntag = 0;
1502     while(tag) {
1503         ntag = swf_CopyTag(ntag, tag);
1504         if(!nswf->firstTag)
1505             nswf->firstTag = ntag;
1506         tag = tag->next;
1507     }
1508     return nswf;
1509 }
1510
1511 void swf_FreeTags(SWF * swf)                 // Frees all malloc'ed memory for tags
1512 { TAG * t = swf->firstTag;
1513
1514   while (t)
1515   { TAG * tnew = t->next;
1516     if (t->data) rfx_free(t->data);
1517     rfx_free(t);
1518     t = tnew;
1519   }
1520   swf->firstTag = 0;
1521 }
1522
1523 // include advanced functions
1524
1525 //#include "modules/swfdump.c"
1526 //#include "modules/swfshape.c"
1527 //#include "modules/swftext.c"
1528 //#include "modules/swffont.c"
1529 //#include "modules/swfobject.c"
1530 //#include "modules/swfbutton.c"
1531 //#include "modules/swftools.c"
1532 //#include "modules/swfcgi.c"
1533 //#include "modules/swfbits.c"
1534 //#include "modules/swfaction.c"
1535 //#include "modules/swfabc.c"
1536 //#include "modules/swfsound.c"
1537 //#include "modules/swfdraw.c"
1538 //#include "modules/swfrender.c"
1539 //#include "modules/swffilter.c"