workbench/c/Unpack/package.c

   1 /*
   2     Copyright © 2003, The AROS Development Team. All rights reserved.
   3     $Id$
   4 */
   5
   6 #include <exec/memory.h>
   7 #include <dos/dos.h>
   8
   9 #include <proto/exec.h>
  10 #include <proto/dos.h>
  11 #include <aros/macros.h>
  12
  13 #include "support.h"
  14 #include "modes.h"
  15 #include "bzip2.h"
  16 #include "file.h"
  17 #include "package.h"
  18 #include "gui.h"
  19
  20 #define PKG_BUFFER_SIZE (32*1024) /* 32kiB */
  21
  22 /** Low-level functions *****************************************************/
  23
  24 APTR PKG_Open( CONST_STRPTR filename, LONG mode )
  25 {
  26     if( mode != MODE_READ && mode != MODE_WRITE ) return NULL;
  27
  28     return BZ2_Open( filename, mode );
  29 }
  30
  31 void PKG_Close( APTR pkg )
  32 {
  33     if( pkg != NULL ) BZ2_Close( pkg );
  34 }
  35
  36 LONG PKG_Read( APTR pkg, APTR buffer, LONG length )
  37 {
  38     return BZ2_Read( pkg, buffer, length );
  39 }
  40
  41 /** High-level functions ****************************************************/
  42
  43 UBYTE /* version */ PKG_ReadHeader( APTR pkg )
  44 {
  45     UBYTE data[4] = { 0, 0, 0, 0 };
  46     /*LONG  actual  = */ PKG_Read( pkg, data, 4 );
  47
  48     /* FIXME: error handling? naaaah... :S */
  49
  50     return data[3];
  51 }
  52
  53 LONG /* error */ PKG_ExtractFile( APTR pkg )
  54 {
  55     LONG   pathLength, dataLength, rc, result;
  56     STRPTR path   = NULL;
  57     APTR   buffer = NULL;
  58     BPTR   output = NULL;
  59
  60     /* Read the path length */
  61     rc = PKG_Read( pkg, &pathLength, sizeof( pathLength ) );
  62     pathLength = AROS_BE2LONG(pathLength);
  63
  64     if( rc == -1 ) { result = -1; goto cleanup; }
  65     if( rc == 0  ) { result =  0; goto cleanup; }
  66
  67     /* Read the path */
  68     path = AllocMem( pathLength + 1, MEMF_ANY );
  69     if( path == NULL ) { result = -1; goto cleanup; }
  70     rc = PKG_Read( pkg, path, pathLength + 1 );
  71     if( rc == -1 || rc == 0) { result = -1; goto cleanup; }
  72
  73     /* Read the data lendth */
  74     rc = PKG_Read( pkg, &dataLength, sizeof( dataLength ) );
  75     dataLength = AROS_BE2LONG(dataLength);
  76
  77     if( rc == -1 || rc == 0 ) { result = -1; goto cleanup; }
  78
  79     //Printf( "Extracting %s (%ld bytes)...\n", path, dataLength );
  80
  81     /* Make sure the destination directory exists */
  82     if( !MakeDirs( path ) ) { Printf("E:makedirs\n"); result = -1; goto cleanup; }
  83
  84     /* Read and write the data in pieces */
  85     buffer = AllocMem( PKG_BUFFER_SIZE, MEMF_ANY );
  86     if( buffer == NULL ) { Printf("E:mem\n"); result = -1; goto cleanup; }
  87     output = Open( path, MODE_NEWFILE );
  88     if( output == NULL ) { Printf("E:create\n"); result = -1; goto cleanup; }
  89
  90     {
  91         LONG total = 0;
  92
  93         while( total < dataLength )
  94         {
  95             LONG length = 0;
  96
  97             if( dataLength - total >= PKG_BUFFER_SIZE )
  98             {
  99                 length = PKG_BUFFER_SIZE;
 100             }
 101             else
 102             {
 103                 length = dataLength - total;
 104             }
 105
 106             rc = PKG_Read( pkg, buffer, length );
 107             if( rc == -1 || rc == 0 ) { Printf("E:read\n"); result = -1; goto cleanup; }
 108
 109             rc = FILE_Write( output, buffer, length );
 110             if( rc == -1 ) { Printf("E:write\n"); result = -1; goto cleanup; }
 111
 112             total += length;
 113         }
 114     }
 115
 116     result = 1;
 117
 118 cleanup:
 119     if( path != NULL )   FreeMem( path, pathLength + 1 );
 120     if( buffer != NULL ) FreeMem( buffer, PKG_BUFFER_SIZE );
 121     if( output != NULL ) Close( output );
 122
 123     return result;
 124 }
 125
 126 LONG /* error */ PKG_ExtractEverything( APTR pkg )
 127 {
 128     LONG result;
 129
 130     PKG_ReadHeader( pkg );
 131
 132     result = PKG_ExtractFile( pkg );
 133     while( result != -1 && result != 0 )
 134     {
 135         result = PKG_ExtractFile( pkg );
 136     }
 137
 138     return result;
 139 }