nuclear@1: /********************************************************************
nuclear@1:  *                                                                  *
nuclear@1:  * THIS FILE IS PART OF THE Ogg CONTAINER SOURCE CODE.              *
nuclear@1:  * USE, DISTRIBUTION AND REPRODUCTION OF THIS LIBRARY SOURCE IS     *
nuclear@1:  * GOVERNED BY A BSD-STYLE SOURCE LICENSE INCLUDED WITH THIS SOURCE *
nuclear@1:  * IN 'COPYING'. PLEASE READ THESE TERMS BEFORE DISTRIBUTING.       *
nuclear@1:  *                                                                  *
nuclear@1:  * THE OggVorbis SOURCE CODE IS (C) COPYRIGHT 1994-2010             *
nuclear@1:  * by the Xiph.Org Foundation http://www.xiph.org/                  *
nuclear@1:  *                                                                  *
nuclear@1:  ********************************************************************
nuclear@1: 
nuclear@1:   function: packing variable sized words into an octet stream
nuclear@1:   last mod: $Id: bitwise.c 18051 2011-08-04 17:56:39Z giles $
nuclear@1: 
nuclear@1:  ********************************************************************/
nuclear@1: 
nuclear@1: /* We're 'LSb' endian; if we write a word but read individual bits,
nuclear@1:    then we'll read the lsb first */
nuclear@1: 
nuclear@1: #include <string.h>
nuclear@1: #include <stdlib.h>
nuclear@1: #include <limits.h>
nuclear@1: #include <ogg/ogg.h>
nuclear@1: 
nuclear@1: #define BUFFER_INCREMENT 256
nuclear@1: 
nuclear@1: static const unsigned long mask[]=
nuclear@1: {0x00000000,0x00000001,0x00000003,0x00000007,0x0000000f,
nuclear@1:  0x0000001f,0x0000003f,0x0000007f,0x000000ff,0x000001ff,
nuclear@1:  0x000003ff,0x000007ff,0x00000fff,0x00001fff,0x00003fff,
nuclear@1:  0x00007fff,0x0000ffff,0x0001ffff,0x0003ffff,0x0007ffff,
nuclear@1:  0x000fffff,0x001fffff,0x003fffff,0x007fffff,0x00ffffff,
nuclear@1:  0x01ffffff,0x03ffffff,0x07ffffff,0x0fffffff,0x1fffffff,
nuclear@1:  0x3fffffff,0x7fffffff,0xffffffff };
nuclear@1: 
nuclear@1: static const unsigned int mask8B[]=
nuclear@1: {0x00,0x80,0xc0,0xe0,0xf0,0xf8,0xfc,0xfe,0xff};
nuclear@1: 
nuclear@1: void oggpack_writeinit(oggpack_buffer *b){
nuclear@1:   memset(b,0,sizeof(*b));
nuclear@1:   b->ptr=b->buffer=_ogg_malloc(BUFFER_INCREMENT);
nuclear@1:   b->buffer[0]='\0';
nuclear@1:   b->storage=BUFFER_INCREMENT;
nuclear@1: }
nuclear@1: 
nuclear@1: void oggpackB_writeinit(oggpack_buffer *b){
nuclear@1:   oggpack_writeinit(b);
nuclear@1: }
nuclear@1: 
nuclear@1: int oggpack_writecheck(oggpack_buffer *b){
nuclear@1:   if(!b->ptr || !b->storage)return -1;
nuclear@1:   return 0;
nuclear@1: }
nuclear@1: 
nuclear@1: int oggpackB_writecheck(oggpack_buffer *b){
nuclear@1:   return oggpack_writecheck(b);
nuclear@1: }
nuclear@1: 
nuclear@1: void oggpack_writetrunc(oggpack_buffer *b,long bits){
nuclear@1:   long bytes=bits>>3;
nuclear@1:   if(b->ptr){
nuclear@1:     bits-=bytes*8;
nuclear@1:     b->ptr=b->buffer+bytes;
nuclear@1:     b->endbit=bits;
nuclear@1:     b->endbyte=bytes;
nuclear@1:     *b->ptr&=mask[bits];
nuclear@1:   }
nuclear@1: }
nuclear@1: 
nuclear@1: void oggpackB_writetrunc(oggpack_buffer *b,long bits){
nuclear@1:   long bytes=bits>>3;
nuclear@1:   if(b->ptr){
nuclear@1:     bits-=bytes*8;
nuclear@1:     b->ptr=b->buffer+bytes;
nuclear@1:     b->endbit=bits;
nuclear@1:     b->endbyte=bytes;
nuclear@1:     *b->ptr&=mask8B[bits];
nuclear@1:   }
nuclear@1: }
nuclear@1: 
nuclear@1: /* Takes only up to 32 bits. */
nuclear@1: void oggpack_write(oggpack_buffer *b,unsigned long value,int bits){
nuclear@1:   if(bits<0 || bits>32) goto err;
nuclear@1:   if(b->endbyte>=b->storage-4){
nuclear@1:     void *ret;
nuclear@1:     if(!b->ptr)return;
nuclear@1:     if(b->storage>LONG_MAX-BUFFER_INCREMENT) goto err;
nuclear@1:     ret=_ogg_realloc(b->buffer,b->storage+BUFFER_INCREMENT);
nuclear@1:     if(!ret) goto err;
nuclear@1:     b->buffer=ret;
nuclear@1:     b->storage+=BUFFER_INCREMENT;
nuclear@1:     b->ptr=b->buffer+b->endbyte;
nuclear@1:   }
nuclear@1: 
nuclear@1:   value&=mask[bits];
nuclear@1:   bits+=b->endbit;
nuclear@1: 
nuclear@1:   b->ptr[0]|=value<<b->endbit;
nuclear@1: 
nuclear@1:   if(bits>=8){
nuclear@1:     b->ptr[1]=(unsigned char)(value>>(8-b->endbit));
nuclear@1:     if(bits>=16){
nuclear@1:       b->ptr[2]=(unsigned char)(value>>(16-b->endbit));
nuclear@1:       if(bits>=24){
nuclear@1:         b->ptr[3]=(unsigned char)(value>>(24-b->endbit));
nuclear@1:         if(bits>=32){
nuclear@1:           if(b->endbit)
nuclear@1:             b->ptr[4]=(unsigned char)(value>>(32-b->endbit));
nuclear@1:           else
nuclear@1:             b->ptr[4]=0;
nuclear@1:         }
nuclear@1:       }
nuclear@1:     }
nuclear@1:   }
nuclear@1: 
nuclear@1:   b->endbyte+=bits/8;
nuclear@1:   b->ptr+=bits/8;
nuclear@1:   b->endbit=bits&7;
nuclear@1:   return;
nuclear@1:  err:
nuclear@1:   oggpack_writeclear(b);
nuclear@1: }
nuclear@1: 
nuclear@1: /* Takes only up to 32 bits. */
nuclear@1: void oggpackB_write(oggpack_buffer *b,unsigned long value,int bits){
nuclear@1:   if(bits<0 || bits>32) goto err;
nuclear@1:   if(b->endbyte>=b->storage-4){
nuclear@1:     void *ret;
nuclear@1:     if(!b->ptr)return;
nuclear@1:     if(b->storage>LONG_MAX-BUFFER_INCREMENT) goto err;
nuclear@1:     ret=_ogg_realloc(b->buffer,b->storage+BUFFER_INCREMENT);
nuclear@1:     if(!ret) goto err;
nuclear@1:     b->buffer=ret;
nuclear@1:     b->storage+=BUFFER_INCREMENT;
nuclear@1:     b->ptr=b->buffer+b->endbyte;
nuclear@1:   }
nuclear@1: 
nuclear@1:   value=(value&mask[bits])<<(32-bits);
nuclear@1:   bits+=b->endbit;
nuclear@1: 
nuclear@1:   b->ptr[0]|=value>>(24+b->endbit);
nuclear@1: 
nuclear@1:   if(bits>=8){
nuclear@1:     b->ptr[1]=(unsigned char)(value>>(16+b->endbit));
nuclear@1:     if(bits>=16){
nuclear@1:       b->ptr[2]=(unsigned char)(value>>(8+b->endbit));
nuclear@1:       if(bits>=24){
nuclear@1:         b->ptr[3]=(unsigned char)(value>>(b->endbit));
nuclear@1:         if(bits>=32){
nuclear@1:           if(b->endbit)
nuclear@1:             b->ptr[4]=(unsigned char)(value<<(8-b->endbit));
nuclear@1:           else
nuclear@1:             b->ptr[4]=0;
nuclear@1:         }
nuclear@1:       }
nuclear@1:     }
nuclear@1:   }
nuclear@1: 
nuclear@1:   b->endbyte+=bits/8;
nuclear@1:   b->ptr+=bits/8;
nuclear@1:   b->endbit=bits&7;
nuclear@1:   return;
nuclear@1:  err:
nuclear@1:   oggpack_writeclear(b);
nuclear@1: }
nuclear@1: 
nuclear@1: void oggpack_writealign(oggpack_buffer *b){
nuclear@1:   int bits=8-b->endbit;
nuclear@1:   if(bits<8)
nuclear@1:     oggpack_write(b,0,bits);
nuclear@1: }
nuclear@1: 
nuclear@1: void oggpackB_writealign(oggpack_buffer *b){
nuclear@1:   int bits=8-b->endbit;
nuclear@1:   if(bits<8)
nuclear@1:     oggpackB_write(b,0,bits);
nuclear@1: }
nuclear@1: 
nuclear@1: static void oggpack_writecopy_helper(oggpack_buffer *b,
nuclear@1:                                      void *source,
nuclear@1:                                      long bits,
nuclear@1:                                      void (*w)(oggpack_buffer *,
nuclear@1:                                                unsigned long,
nuclear@1:                                                int),
nuclear@1:                                      int msb){
nuclear@1:   unsigned char *ptr=(unsigned char *)source;
nuclear@1: 
nuclear@1:   long bytes=bits/8;
nuclear@1:   bits-=bytes*8;
nuclear@1: 
nuclear@1:   if(b->endbit){
nuclear@1:     int i;
nuclear@1:     /* unaligned copy.  Do it the hard way. */
nuclear@1:     for(i=0;i<bytes;i++)
nuclear@1:       w(b,(unsigned long)(ptr[i]),8);
nuclear@1:   }else{
nuclear@1:     /* aligned block copy */
nuclear@1:     if(b->endbyte+bytes+1>=b->storage){
nuclear@1:       void *ret;
nuclear@1:       if(!b->ptr) goto err;
nuclear@1:       if(b->endbyte+bytes+BUFFER_INCREMENT>b->storage) goto err;
nuclear@1:       b->storage=b->endbyte+bytes+BUFFER_INCREMENT;
nuclear@1:       ret=_ogg_realloc(b->buffer,b->storage);
nuclear@1:       if(!ret) goto err;
nuclear@1:       b->buffer=ret;
nuclear@1:       b->ptr=b->buffer+b->endbyte;
nuclear@1:     }
nuclear@1: 
nuclear@1:     memmove(b->ptr,source,bytes);
nuclear@1:     b->ptr+=bytes;
nuclear@1:     b->endbyte+=bytes;
nuclear@1:     *b->ptr=0;
nuclear@1: 
nuclear@1:   }
nuclear@1:   if(bits){
nuclear@1:     if(msb)
nuclear@1:       w(b,(unsigned long)(ptr[bytes]>>(8-bits)),bits);
nuclear@1:     else
nuclear@1:       w(b,(unsigned long)(ptr[bytes]),bits);
nuclear@1:   }
nuclear@1:   return;
nuclear@1:  err:
nuclear@1:   oggpack_writeclear(b);
nuclear@1: }
nuclear@1: 
nuclear@1: void oggpack_writecopy(oggpack_buffer *b,void *source,long bits){
nuclear@1:   oggpack_writecopy_helper(b,source,bits,oggpack_write,0);
nuclear@1: }
nuclear@1: 
nuclear@1: void oggpackB_writecopy(oggpack_buffer *b,void *source,long bits){
nuclear@1:   oggpack_writecopy_helper(b,source,bits,oggpackB_write,1);
nuclear@1: }
nuclear@1: 
nuclear@1: void oggpack_reset(oggpack_buffer *b){
nuclear@1:   if(!b->ptr)return;
nuclear@1:   b->ptr=b->buffer;
nuclear@1:   b->buffer[0]=0;
nuclear@1:   b->endbit=b->endbyte=0;
nuclear@1: }
nuclear@1: 
nuclear@1: void oggpackB_reset(oggpack_buffer *b){
nuclear@1:   oggpack_reset(b);
nuclear@1: }
nuclear@1: 
nuclear@1: void oggpack_writeclear(oggpack_buffer *b){
nuclear@1:   if(b->buffer)_ogg_free(b->buffer);
nuclear@1:   memset(b,0,sizeof(*b));
nuclear@1: }
nuclear@1: 
nuclear@1: void oggpackB_writeclear(oggpack_buffer *b){
nuclear@1:   oggpack_writeclear(b);
nuclear@1: }
nuclear@1: 
nuclear@1: void oggpack_readinit(oggpack_buffer *b,unsigned char *buf,int bytes){
nuclear@1:   memset(b,0,sizeof(*b));
nuclear@1:   b->buffer=b->ptr=buf;
nuclear@1:   b->storage=bytes;
nuclear@1: }
nuclear@1: 
nuclear@1: void oggpackB_readinit(oggpack_buffer *b,unsigned char *buf,int bytes){
nuclear@1:   oggpack_readinit(b,buf,bytes);
nuclear@1: }
nuclear@1: 
nuclear@1: /* Read in bits without advancing the bitptr; bits <= 32 */
nuclear@1: long oggpack_look(oggpack_buffer *b,int bits){
nuclear@1:   unsigned long ret;
nuclear@1:   unsigned long m;
nuclear@1: 
nuclear@1:   if(bits<0 || bits>32) return -1;
nuclear@1:   m=mask[bits];
nuclear@1:   bits+=b->endbit;
nuclear@1: 
nuclear@1:   if(b->endbyte >= b->storage-4){
nuclear@1:     /* not the main path */
nuclear@1:     if(b->endbyte > b->storage-((bits+7)>>3)) return -1;
nuclear@1:     /* special case to avoid reading b->ptr[0], which might be past the end of
nuclear@1:         the buffer; also skips some useless accounting */
nuclear@1:     else if(!bits)return(0L);
nuclear@1:   }
nuclear@1: 
nuclear@1:   ret=b->ptr[0]>>b->endbit;
nuclear@1:   if(bits>8){
nuclear@1:     ret|=b->ptr[1]<<(8-b->endbit);
nuclear@1:     if(bits>16){
nuclear@1:       ret|=b->ptr[2]<<(16-b->endbit);
nuclear@1:       if(bits>24){
nuclear@1:         ret|=b->ptr[3]<<(24-b->endbit);
nuclear@1:         if(bits>32 && b->endbit)
nuclear@1:           ret|=b->ptr[4]<<(32-b->endbit);
nuclear@1:       }
nuclear@1:     }
nuclear@1:   }
nuclear@1:   return(m&ret);
nuclear@1: }
nuclear@1: 
nuclear@1: /* Read in bits without advancing the bitptr; bits <= 32 */
nuclear@1: long oggpackB_look(oggpack_buffer *b,int bits){
nuclear@1:   unsigned long ret;
nuclear@1:   int m=32-bits;
nuclear@1: 
nuclear@1:   if(m<0 || m>32) return -1;
nuclear@1:   bits+=b->endbit;
nuclear@1: 
nuclear@1:   if(b->endbyte >= b->storage-4){
nuclear@1:     /* not the main path */
nuclear@1:     if(b->endbyte > b->storage-((bits+7)>>3)) return -1;
nuclear@1:     /* special case to avoid reading b->ptr[0], which might be past the end of
nuclear@1:         the buffer; also skips some useless accounting */
nuclear@1:     else if(!bits)return(0L);
nuclear@1:   }
nuclear@1: 
nuclear@1:   ret=b->ptr[0]<<(24+b->endbit);
nuclear@1:   if(bits>8){
nuclear@1:     ret|=b->ptr[1]<<(16+b->endbit);
nuclear@1:     if(bits>16){
nuclear@1:       ret|=b->ptr[2]<<(8+b->endbit);
nuclear@1:       if(bits>24){
nuclear@1:         ret|=b->ptr[3]<<(b->endbit);
nuclear@1:         if(bits>32 && b->endbit)
nuclear@1:           ret|=b->ptr[4]>>(8-b->endbit);
nuclear@1:       }
nuclear@1:     }
nuclear@1:   }
nuclear@1:   return ((ret&0xffffffff)>>(m>>1))>>((m+1)>>1);
nuclear@1: }
nuclear@1: 
nuclear@1: long oggpack_look1(oggpack_buffer *b){
nuclear@1:   if(b->endbyte>=b->storage)return(-1);
nuclear@1:   return((b->ptr[0]>>b->endbit)&1);
nuclear@1: }
nuclear@1: 
nuclear@1: long oggpackB_look1(oggpack_buffer *b){
nuclear@1:   if(b->endbyte>=b->storage)return(-1);
nuclear@1:   return((b->ptr[0]>>(7-b->endbit))&1);
nuclear@1: }
nuclear@1: 
nuclear@1: void oggpack_adv(oggpack_buffer *b,int bits){
nuclear@1:   bits+=b->endbit;
nuclear@1: 
nuclear@1:   if(b->endbyte > b->storage-((bits+7)>>3)) goto overflow;
nuclear@1: 
nuclear@1:   b->ptr+=bits/8;
nuclear@1:   b->endbyte+=bits/8;
nuclear@1:   b->endbit=bits&7;
nuclear@1:   return;
nuclear@1: 
nuclear@1:  overflow:
nuclear@1:   b->ptr=NULL;
nuclear@1:   b->endbyte=b->storage;
nuclear@1:   b->endbit=1;
nuclear@1: }
nuclear@1: 
nuclear@1: void oggpackB_adv(oggpack_buffer *b,int bits){
nuclear@1:   oggpack_adv(b,bits);
nuclear@1: }
nuclear@1: 
nuclear@1: void oggpack_adv1(oggpack_buffer *b){
nuclear@1:   if(++(b->endbit)>7){
nuclear@1:     b->endbit=0;
nuclear@1:     b->ptr++;
nuclear@1:     b->endbyte++;
nuclear@1:   }
nuclear@1: }
nuclear@1: 
nuclear@1: void oggpackB_adv1(oggpack_buffer *b){
nuclear@1:   oggpack_adv1(b);
nuclear@1: }
nuclear@1: 
nuclear@1: /* bits <= 32 */
nuclear@1: long oggpack_read(oggpack_buffer *b,int bits){
nuclear@1:   long ret;
nuclear@1:   unsigned long m;
nuclear@1: 
nuclear@1:   if(bits<0 || bits>32) goto err;
nuclear@1:   m=mask[bits];
nuclear@1:   bits+=b->endbit;
nuclear@1: 
nuclear@1:   if(b->endbyte >= b->storage-4){
nuclear@1:     /* not the main path */
nuclear@1:     if(b->endbyte > b->storage-((bits+7)>>3)) goto overflow;
nuclear@1:     /* special case to avoid reading b->ptr[0], which might be past the end of
nuclear@1:         the buffer; also skips some useless accounting */
nuclear@1:     else if(!bits)return(0L);
nuclear@1:   }
nuclear@1: 
nuclear@1:   ret=b->ptr[0]>>b->endbit;
nuclear@1:   if(bits>8){
nuclear@1:     ret|=b->ptr[1]<<(8-b->endbit);
nuclear@1:     if(bits>16){
nuclear@1:       ret|=b->ptr[2]<<(16-b->endbit);
nuclear@1:       if(bits>24){
nuclear@1:         ret|=b->ptr[3]<<(24-b->endbit);
nuclear@1:         if(bits>32 && b->endbit){
nuclear@1:           ret|=b->ptr[4]<<(32-b->endbit);
nuclear@1:         }
nuclear@1:       }
nuclear@1:     }
nuclear@1:   }
nuclear@1:   ret&=m;
nuclear@1:   b->ptr+=bits/8;
nuclear@1:   b->endbyte+=bits/8;
nuclear@1:   b->endbit=bits&7;
nuclear@1:   return ret;
nuclear@1: 
nuclear@1:  overflow:
nuclear@1:  err:
nuclear@1:   b->ptr=NULL;
nuclear@1:   b->endbyte=b->storage;
nuclear@1:   b->endbit=1;
nuclear@1:   return -1L;
nuclear@1: }
nuclear@1: 
nuclear@1: /* bits <= 32 */
nuclear@1: long oggpackB_read(oggpack_buffer *b,int bits){
nuclear@1:   long ret;
nuclear@1:   long m=32-bits;
nuclear@1: 
nuclear@1:   if(m<0 || m>32) goto err;
nuclear@1:   bits+=b->endbit;
nuclear@1: 
nuclear@1:   if(b->endbyte+4>=b->storage){
nuclear@1:     /* not the main path */
nuclear@1:     if(b->endbyte > b->storage-((bits+7)>>3)) goto overflow;
nuclear@1:     /* special case to avoid reading b->ptr[0], which might be past the end of
nuclear@1:         the buffer; also skips some useless accounting */
nuclear@1:     else if(!bits)return(0L);
nuclear@1:   }
nuclear@1: 
nuclear@1:   ret=b->ptr[0]<<(24+b->endbit);
nuclear@1:   if(bits>8){
nuclear@1:     ret|=b->ptr[1]<<(16+b->endbit);
nuclear@1:     if(bits>16){
nuclear@1:       ret|=b->ptr[2]<<(8+b->endbit);
nuclear@1:       if(bits>24){
nuclear@1:         ret|=b->ptr[3]<<(b->endbit);
nuclear@1:         if(bits>32 && b->endbit)
nuclear@1:           ret|=b->ptr[4]>>(8-b->endbit);
nuclear@1:       }
nuclear@1:     }
nuclear@1:   }
nuclear@1:   ret=((ret&0xffffffffUL)>>(m>>1))>>((m+1)>>1);
nuclear@1: 
nuclear@1:   b->ptr+=bits/8;
nuclear@1:   b->endbyte+=bits/8;
nuclear@1:   b->endbit=bits&7;
nuclear@1:   return ret;
nuclear@1: 
nuclear@1:  overflow:
nuclear@1:  err:
nuclear@1:   b->ptr=NULL;
nuclear@1:   b->endbyte=b->storage;
nuclear@1:   b->endbit=1;
nuclear@1:   return -1L;
nuclear@1: }
nuclear@1: 
nuclear@1: long oggpack_read1(oggpack_buffer *b){
nuclear@1:   long ret;
nuclear@1: 
nuclear@1:   if(b->endbyte >= b->storage) goto overflow;
nuclear@1:   ret=(b->ptr[0]>>b->endbit)&1;
nuclear@1: 
nuclear@1:   b->endbit++;
nuclear@1:   if(b->endbit>7){
nuclear@1:     b->endbit=0;
nuclear@1:     b->ptr++;
nuclear@1:     b->endbyte++;
nuclear@1:   }
nuclear@1:   return ret;
nuclear@1: 
nuclear@1:  overflow:
nuclear@1:   b->ptr=NULL;
nuclear@1:   b->endbyte=b->storage;
nuclear@1:   b->endbit=1;
nuclear@1:   return -1L;
nuclear@1: }
nuclear@1: 
nuclear@1: long oggpackB_read1(oggpack_buffer *b){
nuclear@1:   long ret;
nuclear@1: 
nuclear@1:   if(b->endbyte >= b->storage) goto overflow;
nuclear@1:   ret=(b->ptr[0]>>(7-b->endbit))&1;
nuclear@1: 
nuclear@1:   b->endbit++;
nuclear@1:   if(b->endbit>7){
nuclear@1:     b->endbit=0;
nuclear@1:     b->ptr++;
nuclear@1:     b->endbyte++;
nuclear@1:   }
nuclear@1:   return ret;
nuclear@1: 
nuclear@1:  overflow:
nuclear@1:   b->ptr=NULL;
nuclear@1:   b->endbyte=b->storage;
nuclear@1:   b->endbit=1;
nuclear@1:   return -1L;
nuclear@1: }
nuclear@1: 
nuclear@1: long oggpack_bytes(oggpack_buffer *b){
nuclear@1:   return(b->endbyte+(b->endbit+7)/8);
nuclear@1: }
nuclear@1: 
nuclear@1: long oggpack_bits(oggpack_buffer *b){
nuclear@1:   return(b->endbyte*8+b->endbit);
nuclear@1: }
nuclear@1: 
nuclear@1: long oggpackB_bytes(oggpack_buffer *b){
nuclear@1:   return oggpack_bytes(b);
nuclear@1: }
nuclear@1: 
nuclear@1: long oggpackB_bits(oggpack_buffer *b){
nuclear@1:   return oggpack_bits(b);
nuclear@1: }
nuclear@1: 
nuclear@1: unsigned char *oggpack_get_buffer(oggpack_buffer *b){
nuclear@1:   return(b->buffer);
nuclear@1: }
nuclear@1: 
nuclear@1: unsigned char *oggpackB_get_buffer(oggpack_buffer *b){
nuclear@1:   return oggpack_get_buffer(b);
nuclear@1: }
nuclear@1: 
nuclear@1: /* Self test of the bitwise routines; everything else is based on
nuclear@1:    them, so they damned well better be solid. */
nuclear@1: 
nuclear@1: #ifdef _V_SELFTEST
nuclear@1: #include <stdio.h>
nuclear@1: 
nuclear@1: static int ilog(unsigned int v){
nuclear@1:   int ret=0;
nuclear@1:   while(v){
nuclear@1:     ret++;
nuclear@1:     v>>=1;
nuclear@1:   }
nuclear@1:   return(ret);
nuclear@1: }
nuclear@1: 
nuclear@1: oggpack_buffer o;
nuclear@1: oggpack_buffer r;
nuclear@1: 
nuclear@1: void report(char *in){
nuclear@1:   fprintf(stderr,"%s",in);
nuclear@1:   exit(1);
nuclear@1: }
nuclear@1: 
nuclear@1: void cliptest(unsigned long *b,int vals,int bits,int *comp,int compsize){
nuclear@1:   long bytes,i;
nuclear@1:   unsigned char *buffer;
nuclear@1: 
nuclear@1:   oggpack_reset(&o);
nuclear@1:   for(i=0;i<vals;i++)
nuclear@1:     oggpack_write(&o,b[i],bits?bits:ilog(b[i]));
nuclear@1:   buffer=oggpack_get_buffer(&o);
nuclear@1:   bytes=oggpack_bytes(&o);
nuclear@1:   if(bytes!=compsize)report("wrong number of bytes!\n");
nuclear@1:   for(i=0;i<bytes;i++)if(buffer[i]!=comp[i]){
nuclear@1:     for(i=0;i<bytes;i++)fprintf(stderr,"%x %x\n",(int)buffer[i],(int)comp[i]);
nuclear@1:     report("wrote incorrect value!\n");
nuclear@1:   }
nuclear@1:   oggpack_readinit(&r,buffer,bytes);
nuclear@1:   for(i=0;i<vals;i++){
nuclear@1:     int tbit=bits?bits:ilog(b[i]);
nuclear@1:     if(oggpack_look(&r,tbit)==-1)
nuclear@1:       report("out of data!\n");
nuclear@1:     if(oggpack_look(&r,tbit)!=(b[i]&mask[tbit]))
nuclear@1:       report("looked at incorrect value!\n");
nuclear@1:     if(tbit==1)
nuclear@1:       if(oggpack_look1(&r)!=(b[i]&mask[tbit]))
nuclear@1:         report("looked at single bit incorrect value!\n");
nuclear@1:     if(tbit==1){
nuclear@1:       if(oggpack_read1(&r)!=(b[i]&mask[tbit]))
nuclear@1:         report("read incorrect single bit value!\n");
nuclear@1:     }else{
nuclear@1:     if(oggpack_read(&r,tbit)!=(b[i]&mask[tbit]))
nuclear@1:       report("read incorrect value!\n");
nuclear@1:     }
nuclear@1:   }
nuclear@1:   if(oggpack_bytes(&r)!=bytes)report("leftover bytes after read!\n");
nuclear@1: }
nuclear@1: 
nuclear@1: void cliptestB(unsigned long *b,int vals,int bits,int *comp,int compsize){
nuclear@1:   long bytes,i;
nuclear@1:   unsigned char *buffer;
nuclear@1: 
nuclear@1:   oggpackB_reset(&o);
nuclear@1:   for(i=0;i<vals;i++)
nuclear@1:     oggpackB_write(&o,b[i],bits?bits:ilog(b[i]));
nuclear@1:   buffer=oggpackB_get_buffer(&o);
nuclear@1:   bytes=oggpackB_bytes(&o);
nuclear@1:   if(bytes!=compsize)report("wrong number of bytes!\n");
nuclear@1:   for(i=0;i<bytes;i++)if(buffer[i]!=comp[i]){
nuclear@1:     for(i=0;i<bytes;i++)fprintf(stderr,"%x %x\n",(int)buffer[i],(int)comp[i]);
nuclear@1:     report("wrote incorrect value!\n");
nuclear@1:   }
nuclear@1:   oggpackB_readinit(&r,buffer,bytes);
nuclear@1:   for(i=0;i<vals;i++){
nuclear@1:     int tbit=bits?bits:ilog(b[i]);
nuclear@1:     if(oggpackB_look(&r,tbit)==-1)
nuclear@1:       report("out of data!\n");
nuclear@1:     if(oggpackB_look(&r,tbit)!=(b[i]&mask[tbit]))
nuclear@1:       report("looked at incorrect value!\n");
nuclear@1:     if(tbit==1)
nuclear@1:       if(oggpackB_look1(&r)!=(b[i]&mask[tbit]))
nuclear@1:         report("looked at single bit incorrect value!\n");
nuclear@1:     if(tbit==1){
nuclear@1:       if(oggpackB_read1(&r)!=(b[i]&mask[tbit]))
nuclear@1:         report("read incorrect single bit value!\n");
nuclear@1:     }else{
nuclear@1:     if(oggpackB_read(&r,tbit)!=(b[i]&mask[tbit]))
nuclear@1:       report("read incorrect value!\n");
nuclear@1:     }
nuclear@1:   }
nuclear@1:   if(oggpackB_bytes(&r)!=bytes)report("leftover bytes after read!\n");
nuclear@1: }
nuclear@1: 
nuclear@1: int main(void){
nuclear@1:   unsigned char *buffer;
nuclear@1:   long bytes,i;
nuclear@1:   static unsigned long testbuffer1[]=
nuclear@1:     {18,12,103948,4325,543,76,432,52,3,65,4,56,32,42,34,21,1,23,32,546,456,7,
nuclear@1:        567,56,8,8,55,3,52,342,341,4,265,7,67,86,2199,21,7,1,5,1,4};
nuclear@1:   int test1size=43;
nuclear@1: 
nuclear@1:   static unsigned long testbuffer2[]=
nuclear@1:     {216531625L,1237861823,56732452,131,3212421,12325343,34547562,12313212,
nuclear@1:        1233432,534,5,346435231,14436467,7869299,76326614,167548585,
nuclear@1:        85525151,0,12321,1,349528352};
nuclear@1:   int test2size=21;
nuclear@1: 
nuclear@1:   static unsigned long testbuffer3[]=
nuclear@1:     {1,0,14,0,1,0,12,0,1,0,0,0,1,1,0,1,0,1,0,1,0,1,0,1,0,1,0,0,1,1,1,1,1,0,0,1,
nuclear@1:        0,1,30,1,1,1,0,0,1,0,0,0,12,0,11,0,1,0,0,1};
nuclear@1:   int test3size=56;
nuclear@1: 
nuclear@1:   static unsigned long large[]=
nuclear@1:     {2136531625L,2137861823,56732452,131,3212421,12325343,34547562,12313212,
nuclear@1:        1233432,534,5,2146435231,14436467,7869299,76326614,167548585,
nuclear@1:        85525151,0,12321,1,2146528352};
nuclear@1: 
nuclear@1:   int onesize=33;
nuclear@1:   static int one[33]={146,25,44,151,195,15,153,176,233,131,196,65,85,172,47,40,
nuclear@1:                     34,242,223,136,35,222,211,86,171,50,225,135,214,75,172,
nuclear@1:                     223,4};
nuclear@1:   static int oneB[33]={150,101,131,33,203,15,204,216,105,193,156,65,84,85,222,
nuclear@1:                        8,139,145,227,126,34,55,244,171,85,100,39,195,173,18,
nuclear@1:                        245,251,128};
nuclear@1: 
nuclear@1:   int twosize=6;
nuclear@1:   static int two[6]={61,255,255,251,231,29};
nuclear@1:   static int twoB[6]={247,63,255,253,249,120};
nuclear@1: 
nuclear@1:   int threesize=54;
nuclear@1:   static int three[54]={169,2,232,252,91,132,156,36,89,13,123,176,144,32,254,
nuclear@1:                       142,224,85,59,121,144,79,124,23,67,90,90,216,79,23,83,
nuclear@1:                       58,135,196,61,55,129,183,54,101,100,170,37,127,126,10,
nuclear@1:                       100,52,4,14,18,86,77,1};
nuclear@1:   static int threeB[54]={206,128,42,153,57,8,183,251,13,89,36,30,32,144,183,
nuclear@1:                          130,59,240,121,59,85,223,19,228,180,134,33,107,74,98,
nuclear@1:                          233,253,196,135,63,2,110,114,50,155,90,127,37,170,104,
nuclear@1:                          200,20,254,4,58,106,176,144,0};
nuclear@1: 
nuclear@1:   int foursize=38;
nuclear@1:   static int four[38]={18,6,163,252,97,194,104,131,32,1,7,82,137,42,129,11,72,
nuclear@1:                      132,60,220,112,8,196,109,64,179,86,9,137,195,208,122,169,
nuclear@1:                      28,2,133,0,1};
nuclear@1:   static int fourB[38]={36,48,102,83,243,24,52,7,4,35,132,10,145,21,2,93,2,41,
nuclear@1:                         1,219,184,16,33,184,54,149,170,132,18,30,29,98,229,67,
nuclear@1:                         129,10,4,32};
nuclear@1: 
nuclear@1:   int fivesize=45;
nuclear@1:   static int five[45]={169,2,126,139,144,172,30,4,80,72,240,59,130,218,73,62,
nuclear@1:                      241,24,210,44,4,20,0,248,116,49,135,100,110,130,181,169,
nuclear@1:                      84,75,159,2,1,0,132,192,8,0,0,18,22};
nuclear@1:   static int fiveB[45]={1,84,145,111,245,100,128,8,56,36,40,71,126,78,213,226,
nuclear@1:                         124,105,12,0,133,128,0,162,233,242,67,152,77,205,77,
nuclear@1:                         172,150,169,129,79,128,0,6,4,32,0,27,9,0};
nuclear@1: 
nuclear@1:   int sixsize=7;
nuclear@1:   static int six[7]={17,177,170,242,169,19,148};
nuclear@1:   static int sixB[7]={136,141,85,79,149,200,41};
nuclear@1: 
nuclear@1:   /* Test read/write together */
nuclear@1:   /* Later we test against pregenerated bitstreams */
nuclear@1:   oggpack_writeinit(&o);
nuclear@1: 
nuclear@1:   fprintf(stderr,"\nSmall preclipped packing (LSb): ");
nuclear@1:   cliptest(testbuffer1,test1size,0,one,onesize);
nuclear@1:   fprintf(stderr,"ok.");
nuclear@1: 
nuclear@1:   fprintf(stderr,"\nNull bit call (LSb): ");
nuclear@1:   cliptest(testbuffer3,test3size,0,two,twosize);
nuclear@1:   fprintf(stderr,"ok.");
nuclear@1: 
nuclear@1:   fprintf(stderr,"\nLarge preclipped packing (LSb): ");
nuclear@1:   cliptest(testbuffer2,test2size,0,three,threesize);
nuclear@1:   fprintf(stderr,"ok.");
nuclear@1: 
nuclear@1:   fprintf(stderr,"\n32 bit preclipped packing (LSb): ");
nuclear@1:   oggpack_reset(&o);
nuclear@1:   for(i=0;i<test2size;i++)
nuclear@1:     oggpack_write(&o,large[i],32);
nuclear@1:   buffer=oggpack_get_buffer(&o);
nuclear@1:   bytes=oggpack_bytes(&o);
nuclear@1:   oggpack_readinit(&r,buffer,bytes);
nuclear@1:   for(i=0;i<test2size;i++){
nuclear@1:     if(oggpack_look(&r,32)==-1)report("out of data. failed!");
nuclear@1:     if(oggpack_look(&r,32)!=large[i]){
nuclear@1:       fprintf(stderr,"%ld != %ld (%lx!=%lx):",oggpack_look(&r,32),large[i],
nuclear@1:               oggpack_look(&r,32),large[i]);
nuclear@1:       report("read incorrect value!\n");
nuclear@1:     }
nuclear@1:     oggpack_adv(&r,32);
nuclear@1:   }
nuclear@1:   if(oggpack_bytes(&r)!=bytes)report("leftover bytes after read!\n");
nuclear@1:   fprintf(stderr,"ok.");
nuclear@1: 
nuclear@1:   fprintf(stderr,"\nSmall unclipped packing (LSb): ");
nuclear@1:   cliptest(testbuffer1,test1size,7,four,foursize);
nuclear@1:   fprintf(stderr,"ok.");
nuclear@1: 
nuclear@1:   fprintf(stderr,"\nLarge unclipped packing (LSb): ");
nuclear@1:   cliptest(testbuffer2,test2size,17,five,fivesize);
nuclear@1:   fprintf(stderr,"ok.");
nuclear@1: 
nuclear@1:   fprintf(stderr,"\nSingle bit unclipped packing (LSb): ");
nuclear@1:   cliptest(testbuffer3,test3size,1,six,sixsize);
nuclear@1:   fprintf(stderr,"ok.");
nuclear@1: 
nuclear@1:   fprintf(stderr,"\nTesting read past end (LSb): ");
nuclear@1:   oggpack_readinit(&r,(unsigned char *)"\0\0\0\0\0\0\0\0",8);
nuclear@1:   for(i=0;i<64;i++){
nuclear@1:     if(oggpack_read(&r,1)!=0){
nuclear@1:       fprintf(stderr,"failed; got -1 prematurely.\n");
nuclear@1:       exit(1);
nuclear@1:     }
nuclear@1:   }
nuclear@1:   if(oggpack_look(&r,1)!=-1 ||
nuclear@1:      oggpack_read(&r,1)!=-1){
nuclear@1:       fprintf(stderr,"failed; read past end without -1.\n");
nuclear@1:       exit(1);
nuclear@1:   }
nuclear@1:   oggpack_readinit(&r,(unsigned char *)"\0\0\0\0\0\0\0\0",8);
nuclear@1:   if(oggpack_read(&r,30)!=0 || oggpack_read(&r,16)!=0){
nuclear@1:       fprintf(stderr,"failed 2; got -1 prematurely.\n");
nuclear@1:       exit(1);
nuclear@1:   }
nuclear@1: 
nuclear@1:   if(oggpack_look(&r,18)!=0 ||
nuclear@1:      oggpack_look(&r,18)!=0){
nuclear@1:     fprintf(stderr,"failed 3; got -1 prematurely.\n");
nuclear@1:       exit(1);
nuclear@1:   }
nuclear@1:   if(oggpack_look(&r,19)!=-1 ||
nuclear@1:      oggpack_look(&r,19)!=-1){
nuclear@1:     fprintf(stderr,"failed; read past end without -1.\n");
nuclear@1:       exit(1);
nuclear@1:   }
nuclear@1:   if(oggpack_look(&r,32)!=-1 ||
nuclear@1:      oggpack_look(&r,32)!=-1){
nuclear@1:     fprintf(stderr,"failed; read past end without -1.\n");
nuclear@1:       exit(1);
nuclear@1:   }
nuclear@1:   oggpack_writeclear(&o);
nuclear@1:   fprintf(stderr,"ok.\n");
nuclear@1: 
nuclear@1:   /********** lazy, cut-n-paste retest with MSb packing ***********/
nuclear@1: 
nuclear@1:   /* Test read/write together */
nuclear@1:   /* Later we test against pregenerated bitstreams */
nuclear@1:   oggpackB_writeinit(&o);
nuclear@1: 
nuclear@1:   fprintf(stderr,"\nSmall preclipped packing (MSb): ");
nuclear@1:   cliptestB(testbuffer1,test1size,0,oneB,onesize);
nuclear@1:   fprintf(stderr,"ok.");
nuclear@1: 
nuclear@1:   fprintf(stderr,"\nNull bit call (MSb): ");
nuclear@1:   cliptestB(testbuffer3,test3size,0,twoB,twosize);
nuclear@1:   fprintf(stderr,"ok.");
nuclear@1: 
nuclear@1:   fprintf(stderr,"\nLarge preclipped packing (MSb): ");
nuclear@1:   cliptestB(testbuffer2,test2size,0,threeB,threesize);
nuclear@1:   fprintf(stderr,"ok.");
nuclear@1: 
nuclear@1:   fprintf(stderr,"\n32 bit preclipped packing (MSb): ");
nuclear@1:   oggpackB_reset(&o);
nuclear@1:   for(i=0;i<test2size;i++)
nuclear@1:     oggpackB_write(&o,large[i],32);
nuclear@1:   buffer=oggpackB_get_buffer(&o);
nuclear@1:   bytes=oggpackB_bytes(&o);
nuclear@1:   oggpackB_readinit(&r,buffer,bytes);
nuclear@1:   for(i=0;i<test2size;i++){
nuclear@1:     if(oggpackB_look(&r,32)==-1)report("out of data. failed!");
nuclear@1:     if(oggpackB_look(&r,32)!=large[i]){
nuclear@1:       fprintf(stderr,"%ld != %ld (%lx!=%lx):",oggpackB_look(&r,32),large[i],
nuclear@1:               oggpackB_look(&r,32),large[i]);
nuclear@1:       report("read incorrect value!\n");
nuclear@1:     }
nuclear@1:     oggpackB_adv(&r,32);
nuclear@1:   }
nuclear@1:   if(oggpackB_bytes(&r)!=bytes)report("leftover bytes after read!\n");
nuclear@1:   fprintf(stderr,"ok.");
nuclear@1: 
nuclear@1:   fprintf(stderr,"\nSmall unclipped packing (MSb): ");
nuclear@1:   cliptestB(testbuffer1,test1size,7,fourB,foursize);
nuclear@1:   fprintf(stderr,"ok.");
nuclear@1: 
nuclear@1:   fprintf(stderr,"\nLarge unclipped packing (MSb): ");
nuclear@1:   cliptestB(testbuffer2,test2size,17,fiveB,fivesize);
nuclear@1:   fprintf(stderr,"ok.");
nuclear@1: 
nuclear@1:   fprintf(stderr,"\nSingle bit unclipped packing (MSb): ");
nuclear@1:   cliptestB(testbuffer3,test3size,1,sixB,sixsize);
nuclear@1:   fprintf(stderr,"ok.");
nuclear@1: 
nuclear@1:   fprintf(stderr,"\nTesting read past end (MSb): ");
nuclear@1:   oggpackB_readinit(&r,(unsigned char *)"\0\0\0\0\0\0\0\0",8);
nuclear@1:   for(i=0;i<64;i++){
nuclear@1:     if(oggpackB_read(&r,1)!=0){
nuclear@1:       fprintf(stderr,"failed; got -1 prematurely.\n");
nuclear@1:       exit(1);
nuclear@1:     }
nuclear@1:   }
nuclear@1:   if(oggpackB_look(&r,1)!=-1 ||
nuclear@1:      oggpackB_read(&r,1)!=-1){
nuclear@1:       fprintf(stderr,"failed; read past end without -1.\n");
nuclear@1:       exit(1);
nuclear@1:   }
nuclear@1:   oggpackB_readinit(&r,(unsigned char *)"\0\0\0\0\0\0\0\0",8);
nuclear@1:   if(oggpackB_read(&r,30)!=0 || oggpackB_read(&r,16)!=0){
nuclear@1:       fprintf(stderr,"failed 2; got -1 prematurely.\n");
nuclear@1:       exit(1);
nuclear@1:   }
nuclear@1: 
nuclear@1:   if(oggpackB_look(&r,18)!=0 ||
nuclear@1:      oggpackB_look(&r,18)!=0){
nuclear@1:     fprintf(stderr,"failed 3; got -1 prematurely.\n");
nuclear@1:       exit(1);
nuclear@1:   }
nuclear@1:   if(oggpackB_look(&r,19)!=-1 ||
nuclear@1:      oggpackB_look(&r,19)!=-1){
nuclear@1:     fprintf(stderr,"failed; read past end without -1.\n");
nuclear@1:       exit(1);
nuclear@1:   }
nuclear@1:   if(oggpackB_look(&r,32)!=-1 ||
nuclear@1:      oggpackB_look(&r,32)!=-1){
nuclear@1:     fprintf(stderr,"failed; read past end without -1.\n");
nuclear@1:       exit(1);
nuclear@1:   }
nuclear@1:   oggpackB_writeclear(&o);
nuclear@1:   fprintf(stderr,"ok.\n\n");
nuclear@1: 
nuclear@1: 
nuclear@1:   return(0);
nuclear@1: }
nuclear@1: #endif  /* _V_SELFTEST */
nuclear@1: 
nuclear@1: #undef BUFFER_INCREMENT