root/src/SWFReader.cpp
/* [<][>][^][v][top][bottom][index][help] */
DEFINITIONS
This source file includes following definitions.
- getData
 - getByte
 - getWord
 - getInt
 - getInt64
 - getU30
 - getS24
 - getFloat
 - getDouble
 - getDouble2
 - getHalf
 - getEncodedU32
 - getFixed
 - getString
 - getPString
 - getPStringU30
 - getNBitInt
 - getNBitFixed
 - getNBitFixed2
 - dump
 
#include "SWFReader.h"
#include "SWFTag.h"
#include "SWFAction.h"
#include <cstring>
namespace SWF {
Reader::Reader( const unsigned char *d, size_t l ) {
        data = d;
        length = l;
        pos = 0;
        buf = bits = 0;
        err = 0;
}
bool Reader::getData( void *dst, size_t len ) {
        if( pos+len > length ) {
                err = SWFR_EOF;
                pos = length+1;
                return false;
        } else {
                memcpy( dst, &data[pos], len );
                pos += len;
                return true;
        }
}
uint8_t Reader::getByte() {
        if( pos+1 > length ) {
                err = SWFR_EOF;
                pos = length+1;
                return 0;
        }
        return data[pos++];
}
uint16_t Reader::getWord() {
        if( pos+2 > length ) {
                err = SWFR_EOF;
                pos = length+1;
                return 0;
        }
        // FIXME x86-centric?
        int r = data[pos++];
        r += data[pos++]<<8;
        return r;
}
uint32_t Reader::getInt() {
        if( pos+4 > length ) {
                err = SWFR_EOF;
                pos = length+1;
                return 0;
        }
        int r;
        // FIXME x86-centric?
        r = data[pos++];
        r += data[pos++]<<8;
        r += data[pos++]<<16;
        r += data[pos++]<<24;
        return r;
}
uint64_t Reader::getInt64() {
        if( pos+8 > length ) {
                err = SWFR_EOF;
                pos = length+1;
                return 0;
        }
        uint64_t val;
        val = (uint64_t)getInt() << 32;
        val |= (uint64_t)getInt();
        return( val );
}
uint32_t Reader::getU30() {
        uint32_t r = 0;
        unsigned char c;
        for( int i = 0; i < 5; i++ ) {
                c = data[pos++];
                r |= (c & 0x7F) << (7 * i);
                
                if( !(c & 0x80) ) {
                        return r;
                }
                if( pos > length ) {
                        err = SWFR_EOF;
                        pos = length+1;
                        return 0;
                }
        }
        return r;
}
int Reader::getS24() {
        if( pos+3 > length ) {
                err = SWFR_EOF;
                pos = length+1;
                return 0;
        }
        int r = data[pos++];
        r += data[pos++]<<8;
        r += ((signed char)data[pos++])<<16;
        return r;
}
float Reader::getFloat() {
    /*
        if( pos+4 > length ) {
                err = SWFR_EOF;
                pos = length+1;
                return 0;
        }
        // FIXME x86-centric?
        int r = data[pos++];
        r += data[pos++]<<8;
        r += data[pos++]<<16;
        r += data[pos++]<<24;
        return *((float*)&r);
    */
    union {
        float f;
        uint32_t ul;
    } u;
    u.ul = getInt();
    return u.f;
}
double Reader::getDouble() {
/*      if( pos+8 > length ) {
                err = SWFR_EOF;
                pos = length+1;
                return 0;
        }
        // FIXME x86-centric?
        char value[8];
        value[4] = data[pos++];
        value[5] = data[pos++];
        value[6] = data[pos++];
        value[7] = data[pos++];
        value[0] = data[pos++];
        value[1] = data[pos++];
        value[2] = data[pos++];
        value[3] = data[pos++];
        return *(double*)value;
*/
    union {
        double d;
        uint64_t ull;
    } u;
    u.ull = getInt64();
    return u.d;
}
double Reader::getDouble2() {
        if( pos+8 > length ) {
                err = SWFR_EOF;
                pos = length+1;
                return 0;
        }
        
        union {
                double d;
                char c[8];
        } u;
        
        // FIXME x86-centric?
        for( int i = 0; i < 8; i++ ) {
                u.c[i] =  data[pos++];
        }
        return u.d;
}
float Reader::getHalf() {
        uint16_t r = getWord();
        
        int sign = (r & 0x8000) >> 15;
        int exp = (r & 0x7C00) >> 10;
        int man = (r & 0x3FF);
        if(exp == 0) {
                if(man != 0) {
                        while(!(man & 0x400)) {
                                man <<= 1;
                                exp -= 1;
                        }
                        exp += 1 - 15 + 127;
                        man &= 0x3FF;
                }
        } else if(exp == 0x1F) {
                exp = 0xFF;
        } else {
                exp += -15 + 127;
        }
    union {
        float f;
        uint32_t ul;
    } u;
        u.ul = (sign << 31) | (exp << 23) | (man << 13);
        return u.f;
}
unsigned int Reader::getEncodedU32() {
    
    unsigned int result = data[pos++];
    if (!(result & 0x00000080)) return result;
    result = (result & 0x0000007f) | data[pos++] << 7;
    if (!(result & 0x00004000)) return result;
    result = (result & 0x00003fff) | data[pos++] << 14;
    if (!(result & 0x00200000)) return result;
    result = (result & 0x001fffff) | data[pos++] << 21;
    if (!(result & 0x10000000)) return result;
    result = (result & 0x0fffffff) | data[pos++] << 28;
    return result;
}
double Reader::getFixed( int bytesize, int exp ) {
        /* putFixed/getFixed are deprecated: they implicitly to byteAlign */
        
        double r = 0;
        if( pos+bytesize > length ) {
                err = SWFR_EOF;
                pos = length+1;
                return 0;
        }
        for( int i=0; i<bytesize; i++ ) {
                r += data[pos++]<<(8*i);
        }
        return r / ((double)(1<<exp));
}
char *Reader::getString() {
        byteAlign();
        const char *src = (const char *)&data[pos];
        size_t len = strlen(src) + 1;
        char *dst = new char[len];
        strcpy(dst, src);
        pos += len;
        return( dst );
}
char *Reader::getPString() {
        byteAlign();
        int len = getByte();
        char *dst = new char[len+1];
        getData( dst, len );
        dst[len]=0;
        return( dst );
}
char *Reader::getPStringU30() {
        byteAlign();
        uint32_t len = getU30();
        char *dst = new char[len+1];
        getData( dst, len );
        dst[len]=0;
        return( dst );
}
long Reader::getNBitInt( int n, bool is_signed ) {
        // FIXME: unefficient, maybe incorrect!
        int orig_n = n;
        long r = buf;
        if( n == bits ) {
                bits = buf = 0;
                goto ret;
        }
        if( n > bits ) {
                n -= bits;
                while( n > 8 ) {
                        r <<= 8;
                        r += getByte();
                        n-=8;
                }
                buf = getByte();
                
                if( n > 0 ) {
                        r <<= n;
                        bits = 8-n;
                        r += buf >> (8-n);
                        buf &= (1<<bits)-1;
                }
                goto ret;
        }
                
        r = buf >> (bits-n);
        bits -= n;
        buf &= (1<<bits)-1;
        
ret:    
        long sign = (1<<(orig_n-1));
        if( is_signed && (r&sign)!=0 ) {
                r |= - (long) sign;
        }
        
        return r;
}
double Reader::getNBitFixed( int n, int m, bool is_signed ) {
        double d;
        d = getNBitInt( n, is_signed );
        d /= (double)(1<<m);
        return d;
}
double Reader::getNBitFixed2( int n, int m, bool is_signed ) {
        if( n%8 != 0 ) {
                fprintf(stderr,"WARNING: fixedpoint2 needs a size that is divisible by 8");
                return 0;
        }
        return( getFixed( n/8, m ) );
}
void Reader::dump( int len, int ofs ) {
        const unsigned char *d = &data[pos+ofs];
        
        int i=0;
        printf("DumpingReader %i - %i\n", pos+ofs, pos+ofs+len );
        while( i<len ) {
                for( int n=0; n<4 && i<len; n++ ) {
                        for( int b=7; b>=0; b-- ) {
                                printf( d[i] & (1<<b) ? "1" : "0" );
                        }
                        printf(" 0x%02X %03i  ", d[i], d[i] );
                        printf(" ");
                        i++;
                }
                printf("\n");
        }
}
        
}