0563e8170ad0e8337bc764160db962d8cd781b0b
[swftools.git] / 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_CountBits(U32 v,int nbits)
303 { int n = 33;
304   U32 m = 0x80000000;
305   if (!v) n = 0; else
306   if (v&m)
307   { while (v&m)
308     { n--;
309       m>>=1;
310       if (!m) break;
311     } 
312   }
313   else
314   { while (!(v&m))
315     { n--;
316       m>>=1;
317       if (!m) break;
318     } 
319   }
320   return (n>nbits)?n:nbits;
321 }
322
323 int swf_GetRect(TAG * t,SRECT * r)
324 { int nbits;
325   SRECT dummy;
326   if(!t) {r->xmin=r->xmax=r->ymin=r->ymax=0;return 0;}
327   if (!r) r = &dummy;
328   nbits = (int) swf_GetBits(t,5);
329   r->xmin = swf_GetSBits(t,nbits);
330   r->xmax = swf_GetSBits(t,nbits);
331   r->ymin = swf_GetSBits(t,nbits);
332   r->ymax = swf_GetSBits(t,nbits);
333   return 0;
334 }
335
336 int reader_GetRect(struct reader_t*reader,SRECT * r)
337 { int nbits;
338   SRECT dummy;
339   if (!r) r = &dummy;
340   nbits = (int) reader_GetBits(reader,5);
341   r->xmin = reader_GetSBits(reader,nbits);
342   r->xmax = reader_GetSBits(reader,nbits);
343   r->ymin = reader_GetSBits(reader,nbits);
344   r->ymax = reader_GetSBits(reader,nbits);
345   return 0;
346 }
347
348 int swf_SetRect(TAG * t,SRECT * r)
349 { int nbits;
350     
351   nbits = swf_CountBits(r->xmin,0);
352   nbits = swf_CountBits(r->xmax,nbits);
353   nbits = swf_CountBits(r->ymin,nbits);
354   nbits = swf_CountBits(r->ymax,nbits);
355   if(nbits>=32) {
356     fprintf(stderr, "rfxswf: Warning: num_bits overflow in swf_SetRect\n");
357     nbits=31;
358   }
359
360   swf_SetBits(t,nbits,5);
361   swf_SetBits(t,r->xmin,nbits);
362   swf_SetBits(t,r->xmax,nbits);
363   swf_SetBits(t,r->ymin,nbits);
364   swf_SetBits(t,r->ymax,nbits);
365
366   return 0;
367 }
368
369 void swf_ExpandRect(SRECT*src, SPOINT add)
370 {
371     if(add.x < src->xmin)
372         src->xmin = add.x;
373     if(add.x > src->xmax)
374         src->xmax = add.x;
375     if(add.y < src->ymin)
376         src->ymin = add.y;
377     if(add.y > src->ymax)
378         src->ymax = add.y;
379 }
380 void swf_ExpandRect2(SRECT*src, SRECT*add)
381 {
382     if((add->xmin | add->ymin | add->xmax | add->ymax)==0)
383         return;
384     if(add->xmin < src->xmin)
385         src->xmin = add->xmin;
386     if(add->ymin < src->ymin)
387         src->ymin = add->ymin;
388     if(add->xmax > src->xmax)
389         src->xmax = add->xmax;
390     if(add->ymax > src->ymax)
391         src->ymax = add->ymax;
392 }
393 SPOINT swf_TurnPoint(SPOINT p, MATRIX* m)
394 {
395     SPOINT r;
396     r.x = (int)(m->sx*(1/65536.0)*p.x + m->r1*(1/65536.0)*p.y + 0.5) + m->tx;
397     r.y = (int)(m->r0*(1/65536.0)*p.x + m->sy*(1/65536.0)*p.y + 0.5) + m->ty;
398     return r;
399 }
400 SRECT swf_TurnRect(SRECT r, MATRIX* m)
401 {
402     SRECT g;
403     SPOINT p1,p2,p3,p4,pp1,pp2,pp3,pp4;
404     p1.x = r.xmin;p1.y = r.ymin;
405     p2.x = r.xmax;p2.y = r.ymin;
406     p3.x = r.xmin;p3.y = r.ymax;
407     p4.x = r.xmax;p4.y = r.ymax;
408     pp1 = swf_TurnPoint(p1, m);
409     pp2 = swf_TurnPoint(p2, m);
410     pp3 = swf_TurnPoint(p3, m);
411     pp4 = swf_TurnPoint(p4, m);
412     g.xmin = g.xmax = pp1.x;
413     g.ymin = g.ymax = pp1.y;
414     swf_ExpandRect(&g, pp2);
415     swf_ExpandRect(&g, pp3);
416     swf_ExpandRect(&g, pp4);
417     return g;
418 }
419         
420
421 int swf_GetMatrix(TAG * t,MATRIX * m)
422 { MATRIX dummy;
423   int nbits;
424     
425   if (!m) m = &dummy;
426   
427   if (!t)
428   { m->sx = m->sy = 0x10000;
429     m->r0 = m->r1 = 0;
430     m->tx = m->ty = 0;
431     return -1;
432   }
433
434   swf_ResetReadBits(t);
435   
436   if (swf_GetBits(t,1))
437   { nbits = swf_GetBits(t,5);
438     m->sx = swf_GetSBits(t,nbits);
439     m->sy = swf_GetSBits(t,nbits);
440   }
441   else m->sx = m->sy = 0x10000;
442   
443   if (swf_GetBits(t,1))
444   { nbits = swf_GetBits(t,5);
445     m->r0 = swf_GetSBits(t,nbits);
446     m->r1 = swf_GetSBits(t,nbits);
447   }
448   else m->r0 = m->r1 = 0x0;
449
450   nbits = swf_GetBits(t,5);
451   m->tx = swf_GetSBits(t,nbits);
452   m->ty = swf_GetSBits(t,nbits);
453   
454   return 0;
455 }
456
457 int swf_SetMatrix(TAG * t,MATRIX * m)
458 { int nbits;
459   MATRIX ma;
460
461   if (!m)
462   { m = &ma;
463     ma.sx = ma.sy = 0x10000;
464     ma.r0 = ma.r1 = 0;
465     ma.tx = ma.ty = 0;
466   }
467
468   swf_ResetWriteBits(t);
469
470   if ((m->sx==0x10000)&&(m->sy==0x10000)) swf_SetBits(t,0,1);
471   else
472   { swf_SetBits(t,1,1);
473     nbits = swf_CountBits(m->sx,0);
474     nbits = swf_CountBits(m->sy,nbits);
475     if(nbits>=32) {
476         fprintf(stderr,"rfxswf: Error: matrix values too large\n");
477         nbits = 31;
478     }
479     swf_SetBits(t,nbits,5);
480     swf_SetBits(t,m->sx,nbits);
481     swf_SetBits(t,m->sy,nbits);
482   }
483
484   if ((!m->r0)&&(!m->r1)) swf_SetBits(t,0,1);
485   else
486   { swf_SetBits(t,1,1);
487     nbits = swf_CountBits(m->r0,0);
488     nbits = swf_CountBits(m->r1,nbits);
489     if(nbits>=32) {
490         fprintf(stderr,"rfxswf: Error: matrix values too large\n");
491         nbits = 31;
492     }
493     swf_SetBits(t,nbits,5);
494     swf_SetBits(t,m->r0,nbits);
495     swf_SetBits(t,m->r1,nbits);
496   }
497
498   nbits = swf_CountBits(m->tx,0);
499   nbits = swf_CountBits(m->ty,nbits);
500   if(nbits>=32) {
501       fprintf(stderr,"rfxswf: Error: matrix values too large\n");
502       nbits = 31;
503   }
504   swf_SetBits(t,nbits,5);
505   swf_SetBits(t,m->tx,nbits);
506   swf_SetBits(t,m->ty,nbits);
507
508   return 0;
509 }
510
511 int swf_GetCXForm(TAG * t,CXFORM * cx,U8 alpha) //FIXME: alpha should be type bool
512 { CXFORM cxf;
513   int hasadd;
514   int hasmul;
515   int nbits;
516     
517   if (!cx) cx = &cxf;
518   
519   cx->a0 = cx->r0 = cx->g0 = cx->b0 = 256;
520   cx->a1 = cx->r1 = cx->g1 = cx->b1 = 0;
521
522   if (!t) return 0;
523   
524   swf_ResetReadBits(t);
525   hasadd = swf_GetBits(t,1);
526   hasmul = swf_GetBits(t,1);
527   nbits  = swf_GetBits(t,4);
528
529   if (hasmul)
530   { cx->r0 = (S16)swf_GetSBits(t,nbits);
531     cx->g0 = (S16)swf_GetSBits(t,nbits);
532     cx->b0 = (S16)swf_GetSBits(t,nbits);
533     if (alpha)
534       cx->a0 = (S16)swf_GetSBits(t,nbits);
535   }
536
537   if (hasadd)
538   { cx->r1 = (S16)swf_GetSBits(t,nbits);
539     cx->g1 = (S16)swf_GetSBits(t,nbits);
540     cx->b1 = (S16)swf_GetSBits(t,nbits);
541     if (alpha)
542       cx->a1 = (S16)swf_GetSBits(t,nbits);
543   }
544   
545   return 0;
546 }
547
548 int swf_SetCXForm(TAG * t,CXFORM * cx,U8 alpha)
549 { CXFORM cxf;
550   int hasadd;
551   int hasmul;
552   int nbits;
553     
554   if (!cx)
555   { cx = &cxf;
556     cx->a0 = cx->r0 = cx->g0 = cx->b0 = 256;
557     cx->a1 = cx->r1 = cx->g1 = cx->b1 = 0;
558   }
559
560   if (!alpha)
561   { cx->a0 = 256;
562     cx->a1 = 0;
563   }
564
565   nbits = 0;
566
567   hasmul = (cx->a0!=256)||(cx->r0!=256)||(cx->g0!=256)||(cx->b0!=256);
568   hasadd = cx->a1|cx->r1|cx->g1|cx->b1;
569
570   if (hasmul)
571   { if (alpha) nbits = swf_CountBits((S32)cx->a0,nbits);
572     nbits = swf_CountBits((S32)cx->r0,nbits);
573     nbits = swf_CountBits((S32)cx->g0,nbits);
574     nbits = swf_CountBits((S32)cx->b0,nbits);
575   }
576
577   if (hasadd)
578   { if (alpha) nbits = swf_CountBits((S32)cx->a1,nbits);
579     nbits = swf_CountBits((S32)cx->r1,nbits);
580     nbits = swf_CountBits((S32)cx->g1,nbits);
581     nbits = swf_CountBits((S32)cx->b1,nbits);
582   }
583   
584   swf_ResetWriteBits(t);
585   swf_SetBits(t,hasadd?1:0,1);
586   swf_SetBits(t,hasmul?1:0,1);
587   swf_SetBits(t,nbits,4);
588
589   if (hasmul)
590   { swf_SetBits(t,cx->r0,nbits);
591     swf_SetBits(t,cx->g0,nbits);
592     swf_SetBits(t,cx->b0,nbits);
593     if (alpha) swf_SetBits(t,cx->a0,nbits);
594   }
595
596   if (hasadd)
597   { swf_SetBits(t,cx->r1,nbits);
598     swf_SetBits(t,cx->g1,nbits);
599     swf_SetBits(t,cx->b1,nbits);
600     if (alpha) swf_SetBits(t,cx->a1,nbits);
601   }
602   
603   return 0;
604 }
605
606 int swf_GetPoint(TAG * t,SPOINT * p) { return 0; }
607 int swf_SetPoint(TAG * t,SPOINT * p) { return 0; }
608
609 // Tag List Manipulating Functions
610
611 int swf_UpdateFrame(TAG * t,S8 delta)
612 // returns number of frames
613 { int res = -1;
614   while (t)
615   { t->frame+=delta;
616     res = t->frame;
617     t = t->next;
618   }
619   return res;
620 }
621
622 TAG * swf_InsertTag(TAG * after,U16 id)     // updates frames, if nescessary
623 { TAG * t;
624
625   t = (TAG *)malloc(sizeof(TAG));
626   if (t)
627   { memset(t,0x00,sizeof(TAG));
628     t->id = id;
629     
630     if (after)
631     { t->frame = after->frame;
632       t->prev  = after;
633       t->next  = after->next;
634       after->next = t;
635       if (t->next) t->next->prev = t;
636       
637       if (id==ST_SHOWFRAME) swf_UpdateFrame(t->next,+1);
638     }
639   }
640   return t;
641 }
642
643 void swf_ClearTag(TAG * t)
644 {
645   if (t->data) free(t->data);
646   t->data = 0;
647   t->pos = 0;
648   t->len = 0;
649   t->readBit = 0;
650   t->writeBit = 0;
651   t->memsize = 0;
652 }
653
654 void swf_ResetTag(TAG*tag, U16 id)
655 {
656     tag->len = tag->pos = tag->readBit = tag->writeBit = 0;
657     tag->id = id;
658 }
659
660 int swf_DeleteTag(TAG * t)
661 { if (!t) return -1;
662
663   if (t->id==ST_SHOWFRAME) swf_UpdateFrame(t->next,-1);
664     
665   if (t->prev) t->prev->next = t->next;
666   if (t->next) t->next->prev = t->prev;
667
668   if (t->data) free(t->data);
669   free(t);
670   return 0;
671 }
672
673 TAG * swf_ReadTag(struct reader_t*reader, TAG * prev)
674 { TAG * t;
675   U16 raw;
676   U32 len;
677   int id;
678
679   if (reader->read(reader, &raw, 2) !=2 ) return NULL;
680   raw = SWAP16(raw);
681
682   len = raw&0x3f;
683   id  = raw>>6;
684
685   if (len==0x3f)
686   {
687       if (reader->read(reader, &len, 4) != 4) return NULL;
688       len = SWAP32(len);
689   }
690
691   if (id==ST_DEFINESPRITE) len = 2*sizeof(U16);
692   // Sprite handling fix: Flaten sprite tree
693
694   t = (TAG *)malloc(sizeof(TAG));
695   
696   if (!t)
697   {
698     #ifdef DEBUG_RFXSWF
699       fprintf(stderr,"Fatal Error: malloc()/realloc() failed (2). (%d bytes)\n", sizeof(TAG));
700     #endif
701     return NULL;
702   }
703
704   memset(t,0x00,sizeof(TAG));
705   
706   t->len = len;
707   t->id  = id;
708
709   if (t->len)
710   { t->data = (U8*)malloc(t->len);
711     if (!t->data)
712     {
713       #ifdef DEBUG_RFXSWF
714         fprintf(stderr,"Fatal Error: malloc()/realloc() failed (3). (%d bytes)\n", t->len);
715       #endif
716       return NULL;
717     }
718     t->memsize = t->len;
719     if (reader->read(reader, t->data, t->len) != t->len) return NULL;
720   }
721
722   if (prev)
723   { t->frame = prev->frame+((prev->id==ST_SHOWFRAME)?1:0);
724     t->prev  = prev;
725     prev->next = t;
726   }
727
728   return t;
729 }
730
731 int swf_DefineSprite_GetRealSize(TAG * t);
732
733 int swf_WriteTag2(struct writer_t*writer, TAG * t)
734 // returns tag length in bytes (incl. Header), -1 = Error
735 // writer = 0 -> no output
736 { U16 raw[3];
737   U32 len;
738   int short_tag;
739
740   if (!t) return -1;
741
742   len = (t->id==ST_DEFINESPRITE)?swf_DefineSprite_GetRealSize(t):t->len;
743
744   short_tag = len<0x3f;
745
746   if (writer)
747   { if (short_tag)
748     { raw[0] = SWAP16(len|((t->id&0x3ff)<<6));
749       if (writer->write(writer,raw,2)!=2)
750       {
751         #ifdef DEBUG_RFXSWF
752           fprintf(stderr,"WriteTag() failed: Short Header.\n");
753         #endif
754         return -1;
755       }
756     }
757     else
758     {
759       raw[0] = SWAP16((t->id<<6)|0x3f);
760       if (writer->write(writer,raw,2)!=2)
761       {
762 #ifdef DEBUG_RFXSWF
763           fprintf(stderr,"WriteTag() failed: Long Header (1).\n");
764 #endif
765           return -1;
766       }
767       
768       len = SWAP32(len);
769       if (writer->write(writer,&len,4)!=4)
770       {
771         #ifdef DEBUG_RFXSWF
772           fprintf(stderr,"WriteTag() failed: Long Header (2).\n");
773         #endif
774         return -1;
775       }
776     }
777     
778     if (t->data)
779     { if (writer->write(writer,t->data,t->len)!=t->len)
780       {
781         #ifdef DEBUG_RFXSWF
782           fprintf(stderr,"WriteTag() failed: Data.\n");
783         #endif
784         return -1;
785       }
786     }
787     #ifdef DEBUG_RFXSWF
788       else if (t->len) fprintf(stderr,"WriteTag(): Tag Data Error, id=%i\n",t->id);
789     #endif
790   }
791
792   return t->len+(short_tag?2:6);
793 }
794
795 int swf_WriteTag(int handle, TAG * t)
796 {
797   struct writer_t writer;
798   if(handle<0)
799     return swf_WriteTag2(0, t);
800   writer_init_filewriter(&writer, handle);
801   return swf_WriteTag2(&writer, t);
802 }
803
804 int swf_DefineSprite_GetRealSize(TAG * t)
805 // Sprite Handling: Helper function to pack DefineSprite-Tag
806 { U32 len = t->len;
807   if(len>4) { // folded sprite
808       return t->len;
809   }
810   do
811   { t = swf_NextTag(t);
812     if (t && t->id!=ST_DEFINESPRITE) len += swf_WriteTag(-1, t);
813     else t = NULL;
814   } while (t&&(t->id!=ST_END));
815   return len;
816 }
817
818 void swf_UnFoldSprite(TAG * t)
819 {
820   U16 id,tmp;
821   U32 len;
822   TAG*next = t;
823   U16 spriteid,spriteframes;
824   int level;
825   if(t->id!=ST_DEFINESPRITE)
826     return;
827   if(t->len<=4) // not folded
828     return;
829
830   swf_SetTagPos(t,0);
831
832   spriteid = swf_GetU16(t); //id
833   spriteframes = swf_GetU16(t); //frames
834
835   level = 1;
836
837   while(1)
838   {
839     TAG*it = 0;
840     tmp = swf_GetU16(t);
841     len = tmp&0x3f;
842     id  = tmp>>6;
843     if(id == ST_END)
844         level--;
845     if(id == ST_DEFINESPRITE && len<=4)
846         level++;
847
848     if (len==0x3f)
849         len = swf_GetU32(t);
850     it = swf_InsertTag(next, id);
851     next = it;
852     it->len = len;
853     it->id  = id;
854     if (it->len)
855     { it->data = (U8*)malloc(it->len);
856       it->memsize = it->len;
857       swf_GetBlock(t, it->data, it->len);
858     }
859
860     if(!level)
861         break;
862   }
863   
864   free(t->data); t->data = 0;
865   t->memsize = t->len = t->pos = 0;
866
867   swf_SetU16(t, spriteid);
868   swf_SetU16(t, spriteframes);
869 }
870
871 void swf_FoldSprite(TAG * t)
872 {
873   TAG*sprtag=t,*tmp;
874   U16 id,frames,tmpid;
875   int level;
876   if(t->id!=ST_DEFINESPRITE)
877       return;
878   if(!t->len) {
879       fprintf(stderr, "Error: Sprite has no ID!");
880       return;
881   }
882   if(t->len>4) {
883     /* sprite is already folded */
884       return;
885   }
886
887   t->pos = 0;
888   id = swf_GetU16(t);
889   free(t->data);
890   t->len = t->pos = t->memsize = 0;
891   t->data = 0;
892
893   frames = 0;
894
895   t = swf_NextTag(sprtag);
896   level = 1;
897
898   do 
899   { 
900     if(t->id==ST_SHOWFRAME) frames++;
901     if(t->id == ST_DEFINESPRITE && t->len<=4)
902         level++;
903     if(t->id == ST_END)
904         level--;
905     t = swf_NextTag(t);
906   } while(t && level);
907   if(level)
908     fprintf(stderr, "rfxswf error: sprite doesn't end(1)\n");
909
910   swf_SetU16(sprtag, id);
911   swf_SetU16(sprtag, frames);
912
913   t = swf_NextTag(sprtag);
914   level = 1;
915
916   do
917   { 
918     if(t->len<0x3f) {
919         swf_SetU16(sprtag,t->len|(t->id<<6));
920     } else {
921         swf_SetU16(sprtag,0x3f|(t->id<<6));
922         swf_SetU32(sprtag,t->len);
923     }
924     if(t->len)
925         swf_SetBlock(sprtag,t->data, t->len);
926     tmp = t;
927     if(t->id == ST_DEFINESPRITE && t->len<=4)
928         level++;
929     if(t->id == ST_END)
930         level--;
931     t = swf_NextTag(t);
932     swf_DeleteTag(tmp);
933   } 
934   while (t && level);
935   if(level)
936     fprintf(stderr, "rfxswf error: sprite doesn't end(2)\n");
937
938 //  sprtag->next = t;
939 //  t->prev = sprtag;
940 }
941
942 int swf_IsFolded(TAG * t)
943 {
944     return (t->id == ST_DEFINESPRITE && t->len>4);
945 }
946
947 void swf_FoldAll(SWF*swf)
948 {
949     TAG*tag = swf->firstTag;
950     while(tag) {
951         if(tag->id == ST_DEFINESPRITE)
952             swf_FoldSprite(tag);
953         tag = swf_NextTag(tag);
954     }
955 }
956
957 void swf_UnFoldAll(SWF*swf)
958 {
959     TAG*tag = swf->firstTag;
960     while(tag) {
961         if(tag->id == ST_DEFINESPRITE)
962             swf_UnFoldSprite(tag);
963         tag = tag->next;
964     }
965 }
966
967 void swf_OptimizeTagOrder(SWF*swf)
968 {
969   TAG*tag,*next;
970   TAG*level0;
971   int level;
972   int changes;
973   swf_UnFoldAll(swf);
974   /* at the moment, we don't actually do optimizing,
975      only fixing of non-spec-conformant things like
976      sprite tags */
977
978   do {
979     changes = 0;
980     level = 0;
981     level0 = 0;
982     tag = swf->firstTag;
983     while(tag) {
984       next = tag->next;
985       if(tag->id == ST_DEFINESPRITE) {
986         if(tag->len>4) {
987           /* ??? all sprites are supposed to be unfolded */
988           fprintf(stderr, "librfxswf error - internal error in OptimizeTagOrder/UnfoldAll\n");
989         }
990         level++;
991         if(level==1) {
992           level0 = tag;
993           tag = next;
994           continue;
995         }
996       }
997       if(level>=1) {
998         /* move non-sprite tags out of sprite */
999         if(!swf_isAllowedSpriteTag(tag) || level>=2) {
1000           /* remove tag from current position */
1001           tag->prev->next = tag->next;
1002           if(tag->next)
1003             tag->next->prev = tag->prev;
1004
1005           /* insert before tag level0 */
1006           tag->next = level0;
1007           tag->prev = level0->prev;
1008           level0->prev = tag;
1009           tag->prev->next = tag;
1010           changes = 1;
1011         }
1012       }
1013       if(tag->id == ST_END) {
1014         level--;
1015       }
1016
1017       tag = next;
1018     }
1019   } while(changes);
1020 }
1021
1022 // Movie Functions
1023
1024 int swf_ReadSWF2(struct reader_t*reader, SWF * swf)   // Reads SWF to memory (malloc'ed), returns length or <0 if fails
1025 {     
1026   if (!swf) return -1;
1027   memset(swf,0x00,sizeof(SWF));
1028
1029   { char b[32];                         // read Header
1030     int len;
1031     TAG * t;
1032     TAG t1;
1033     struct reader_t zreader;
1034     
1035     if ((len = reader->read(reader ,b,8))<8) return -1;
1036
1037     if (b[0]!='F' && b[0]!='C') return -1;
1038     if (b[1]!='W') return -1;
1039     if (b[2]!='S') return -1;
1040     swf->fileVersion = b[3];
1041     swf->compressed  = (b[0]=='C')?1:0;
1042     swf->fileSize    = GET32(&b[4]);
1043     
1044     if(swf->compressed) {
1045         reader_init_zlibinflate(&zreader, reader);
1046         reader = &zreader;
1047     }
1048
1049     reader_GetRect(reader, &swf->movieSize);
1050     reader->read(reader, &swf->frameRate, 2);
1051     swf->frameRate = SWAP16(swf->frameRate);
1052     reader->read(reader, &swf->frameCount, 2);
1053     swf->frameCount = SWAP16(swf->frameCount);
1054
1055     /* read tags and connect to list */
1056     t = &t1;
1057     while (t) t = swf_ReadTag(reader,t);
1058     swf->firstTag = t1.next;
1059     t1.next->prev = NULL;
1060   }
1061   
1062   return reader->pos;
1063 }
1064
1065 int swf_ReadSWF(int handle, SWF * swf)
1066 {
1067   struct reader_t reader;
1068   reader_init_filereader(&reader, handle);
1069   return swf_ReadSWF2(&reader, swf);
1070 }
1071
1072 int  swf_WriteSWF2(struct writer_t*writer, SWF * swf)     // Writes SWF to file, returns length or <0 if fails
1073 { U32 len;
1074   TAG * t;
1075   int frameCount=0;
1076   struct writer_t zwriter;
1077   int fileSize = 0;
1078     
1079   if (!swf) return -1;
1080
1081   // Insert REFLEX Tag
1082
1083 #ifdef INSERT_RFX_TAG
1084
1085   if (swf->firstTag && swf_NextTag(swf->firstTag))
1086     if (swf_GetTagID(swf_NextTag(swf->firstTag))!=ST_REFLEX)
1087       swf_SetBlock(swf_InsertTag(swf->firstTag,ST_REFLEX),"rfx",3);
1088
1089 #endif // INSERT_RFX_TAG
1090
1091   // Count Frames + File Size
1092
1093   len = 0;
1094   t = swf->firstTag;
1095   frameCount = 0;
1096
1097   while(t)
1098   { len += swf_WriteTag(-1, t);
1099     if (t->id==ST_SHOWFRAME) frameCount++;
1100     t = swf_NextTag(t);
1101   }
1102   
1103   { TAG t1;
1104     char b[64],b4[4];
1105     U32 l;
1106
1107     memset(&t1,0x00,sizeof(TAG));
1108     t1.data    = (U8*)b;
1109     t1.memsize = 64;
1110     
1111     { // measure header file size
1112       TAG t2;
1113       char b2[64];
1114       memset(&t2,0x00,sizeof(TAG));
1115       t2.data    = (U8*)b2;
1116       t2.memsize = 64;
1117       swf_SetRect(&t2, &swf->movieSize);
1118       swf_SetU16(&t2, swf->frameRate);
1119       swf_SetU16(&t2, swf->frameCount);
1120       l = swf_GetTagLen(&t2)+8;
1121     }
1122
1123     fileSize = l+len;
1124     if(len) {// don't touch headers without tags
1125         swf->fileSize = fileSize;
1126         swf->frameCount = frameCount;
1127     }
1128
1129     if(swf->compressed != 8) {
1130     /* compressed flag set to 8 means "skip first 8 
1131        header bytes". This is necessary if the caller wants to
1132        create compressed SWFs himself */
1133       if(swf->compressed) {
1134         char*id = "CWS";
1135         writer->write(writer, id, 3);
1136       }
1137       else {
1138         char*id = "FWS";
1139         writer->write(writer, id, 3);
1140       }
1141
1142       writer->write(writer, &swf->fileVersion, 1);
1143       PUT32(b4, swf->fileSize);
1144       writer->write(writer, b4, 4);
1145       
1146       if(swf->compressed) {
1147         writer_init_zlibdeflate(&zwriter, writer);
1148         writer = &zwriter;
1149       }
1150     }
1151     swf_SetRect(&t1,&swf->movieSize);
1152     swf_SetU16(&t1,swf->frameRate);
1153     swf_SetU16(&t1,swf->frameCount);
1154
1155     if (writer)
1156     { 
1157       int ret = writer->write(writer,b,swf_GetTagLen(&t1));
1158       if (ret!=swf_GetTagLen(&t1))
1159       {
1160         #ifdef DEBUG_RFXSWF
1161           fprintf(stderr, "ret:%d\n",ret);
1162           perror("write:");
1163           fprintf(stderr,"WriteSWF() failed: Header.\n");
1164         #endif
1165         return -1;
1166       }
1167
1168       t = swf->firstTag;
1169       while (t)
1170       { if (swf_WriteTag2(writer, t)<0) return -1;
1171         t = swf_NextTag(t);
1172       }
1173       if(swf->compressed != 8)
1174         writer->finish(writer); // flush zlib buffers - only if _we_ initialized that writer.
1175     }
1176   }
1177   return (int)fileSize;
1178 }
1179
1180 int  swf_WriteSWF(int handle, SWF * swf)     // Writes SWF to file, returns length or <0 if fails
1181 {
1182   struct writer_t writer;
1183   swf->compressed = 0;
1184   if(handle<0)
1185     return swf_WriteSWF2(&writer, swf);
1186   writer_init_filewriter(&writer, handle);
1187   return swf_WriteSWF2(&writer, swf);
1188 }
1189
1190 int  swf_WriteSWC(int handle, SWF * swf)     // Writes SWF to file, returns length or <0 if fails
1191 {
1192   struct writer_t writer;
1193   swf->compressed = 1;
1194   if(handle<0)
1195     return swf_WriteSWF2(&writer, swf);
1196   writer_init_filewriter(&writer, handle);
1197   return swf_WriteSWF2(&writer, swf);
1198 }
1199
1200 int swf_WriteHeader2(struct writer_t*writer,SWF * swf)
1201 {
1202   SWF myswf;
1203   memcpy(&myswf,swf,sizeof(SWF));
1204   myswf.firstTag = 0;
1205   return swf_WriteSWF2(writer, &myswf);
1206 }
1207
1208 int swf_WriteHeader(int handle,SWF * swf)
1209 {
1210   SWF myswf;
1211   memcpy(&myswf,swf,sizeof(SWF));
1212   myswf.firstTag = 0;
1213   return swf_WriteSWF(handle, &myswf);
1214 }
1215
1216 int swf_WriteCGI(SWF * swf)
1217 { int len;
1218   char s[1024];
1219     
1220   len = swf_WriteSWF(-1,swf);
1221
1222   if (len<0) return -1;
1223
1224   sprintf(s,"Content-type: application/x-shockwave-flash\n"
1225             "Accept-Ranges: bytes\n"
1226             "Content-Length: %lu\n"
1227             "Expires: Thu, 13 Apr 2000 23:59:59 GMT\n"
1228             "\n",len);
1229             
1230   write(fileno(stdout),s,strlen(s));
1231   return swf_WriteSWF(fileno(stdout),swf);
1232 }
1233
1234 void swf_FreeTags(SWF * swf)                 // Frees all malloc'ed memory for tags
1235 { TAG * t = swf->firstTag;
1236
1237   while (t)
1238   { TAG * tnew = t->next;
1239     if (t->data) free(t->data);
1240     free(t);
1241     t = tnew;
1242   }
1243 }
1244
1245 // include advanced functions
1246
1247 #include "modules/swfdump.c"
1248 #include "modules/swfshape.c"
1249 #include "modules/swftext.c"
1250 #include "modules/swfobject.c"
1251 #include "modules/swfbutton.c"
1252 #include "modules/swftools.c"
1253 #include "modules/swfcgi.c"
1254 #include "modules/swfbits.c"
1255 #include "modules/swfaction.c"
1256 #include "modules/swfsound.c"