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