workbench/c/Unpack/bzip2.c

   1 /*
   2     Copyright © 2003, The AROS Development Team. All rights reserved.
   3     $Id$
   4 */
   5
   6 #include <proto/exec.h>
   7 #include <proto/dos.h>
   8 #include <exec/types.h>
   9 #include <exec/memory.h>
  10 #include <dos/dos.h>
  11
  12 #include "modes.h"
  13 #include "file.h"
  14 #include "bzip2.h"
  15 #include "bzip2_private.h"
  16
  17 #define BZIP2_BUFFER_SIZE (32*1024)    /* 32 kiB */
  18
  19 int bz_internal_error;
  20
  21 void *malloc( size_t size )
  22 {
  23     return AllocVec( size, MEMF_ANY );
  24 }
  25
  26 void free( void *p )
  27 {
  28     FreeVec( p );
  29 }
  30
  31 APTR BZ2_Open( CONST_STRPTR path, LONG mode )
  32 {
  33     struct bzFile *bzf = NULL;
  34     int            rc;
  35
  36     if( mode != MODE_READ ) goto error;
  37
  38     bzf = AllocMem( sizeof( struct bzFile ), MEMF_ANY | MEMF_CLEAR );
  39     if( bzf == NULL ) goto error;
  40
  41     bzf->bzf_Buffer = AllocMem( BZIP2_BUFFER_SIZE, MEMF_ANY );
  42     if( bzf->bzf_Buffer == NULL ) goto error;
  43     bzf->bzf_BufferAmount = 0;
  44
  45     bzf->bzf_File = FILE_Open( path, MODE_READ );
  46     if( bzf->bzf_File == BNULL ) goto error;
  47
  48     rc = BZ2_bzDecompressInit( &(bzf->bzf_Stream), 0, 0 );
  49     if( rc != BZ_OK ) goto error;
  50
  51     bzf->bzf_Stream.next_in  = bzf->bzf_Buffer;
  52     bzf->bzf_Stream.avail_in = bzf->bzf_BufferAmount;
  53
  54     return bzf;
  55
  56 error:
  57     //printf( "BZ2_Open: Error!\n" );
  58
  59     BZ2_Close( bzf );
  60
  61     return NULL;
  62 }
  63
  64 void BZ2_Close( APTR file )
  65 {
  66     struct bzFile *bzf = (struct bzFile *) file;
  67
  68     if( bzf != NULL )
  69     {
  70         if( bzf->bzf_File )   Close( bzf->bzf_File );
  71         if( bzf->bzf_Buffer ) FreeMem( bzf->bzf_Buffer, BZIP2_BUFFER_SIZE );
  72
  73         BZ2_bzDecompressEnd( &(bzf->bzf_Stream) );
  74
  75         FreeMem( bzf, sizeof( struct bzFile ) );
  76     }
  77 }
  78
  79 LONG BZ2_Read( APTR file, APTR buffer, LONG length )
  80 {
  81     struct bzFile *bzf  = (struct bzFile *) file;
  82     LONG           read = 0;
  83     int            rc;
  84
  85     bzf->bzf_Stream.next_out  = buffer;
  86     bzf->bzf_Stream.avail_out = length;
  87
  88     while( TRUE )
  89     {
  90         if( bzf->bzf_Stream.avail_in == 0 )
  91         {
  92             read = FILE_Read( bzf->bzf_File, bzf->bzf_Buffer, BZIP2_BUFFER_SIZE );
  93             if( read == -1 ) goto error;
  94             if( read == 0 )  return 0;          /* EOF */
  95             bzf->bzf_BufferAmount     = read;
  96
  97             bzf->bzf_Stream.next_in   = bzf->bzf_Buffer;
  98             bzf->bzf_Stream.avail_in  = bzf->bzf_BufferAmount;
  99         }
 100
 101         rc = BZ2_bzDecompress( &(bzf->bzf_Stream) );
 102
 103         if( rc != BZ_OK && rc != BZ_STREAM_END ) goto error;
 104
 105         if( rc == BZ_STREAM_END )
 106         {
 107             return length - bzf->bzf_Stream.avail_out;
 108         }
 109
 110         if( bzf->bzf_Stream.avail_out == 0 )
 111         {
 112             return length;
 113         }
 114     }
 115
 116 error:
 117
 118     return -1;
 119 }