lib/rfxswf.c

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