source/libs/zlib/zlib-src/contrib/minizip/mztools.c

   1 /*
   2   Additional tools for Minizip
   3   Code: Xavier Roche '2004
   4   License: Same as ZLIB (www.gzip.org)
   5 */
   6
   7 /* Code */
   8 #include <stdio.h>
   9 #include <stdlib.h>
  10 #include <string.h>
  11 #include "zlib.h"
  12 #include "unzip.h"
  13
  14 #define READ_8(adr)  ((unsigned char)*(adr))
  15 #define READ_16(adr) ( READ_8(adr) | (READ_8(adr+1) << 8) )
  16 #define READ_32(adr) ( READ_16(adr) | (READ_16((adr)+2) << 16) )
  17
  18 #define WRITE_8(buff, n) do { \
  19   *((unsigned char*)(buff)) = (unsigned char) ((n) & 0xff); \
  20 } while(0)
  21 #define WRITE_16(buff, n) do { \
  22   WRITE_8((unsigned char*)(buff), n); \
  23   WRITE_8(((unsigned char*)(buff)) + 1, (n) >> 8); \
  24 } while(0)
  25 #define WRITE_32(buff, n) do { \
  26   WRITE_16((unsigned char*)(buff), (n) & 0xffff); \
  27   WRITE_16((unsigned char*)(buff) + 2, (n) >> 16); \
  28 } while(0)
  29
  30 extern int ZEXPORT unzRepair(file, fileOut, fileOutTmp, nRecovered, bytesRecovered)
  31 const char* file;
  32 const char* fileOut;
  33 const char* fileOutTmp;
  34 uLong* nRecovered;
  35 uLong* bytesRecovered;
  36 {
  37   int err = Z_OK;
  38   FILE* fpZip = fopen(file, "rb");
  39   FILE* fpOut = fopen(fileOut, "wb");
  40   FILE* fpOutCD = fopen(fileOutTmp, "wb");
  41   if (fpZip != NULL &&  fpOut != NULL) {
  42     int entries = 0;
  43     uLong totalBytes = 0;
  44     char header[30];
  45     char filename[1024];
  46     char extra[1024];
  47     int offset = 0;
  48     int offsetCD = 0;
  49     while ( fread(header, 1, 30, fpZip) == 30 ) {
  50       int currentOffset = offset;
  51
  52       /* File entry */
  53       if (READ_32(header) == 0x04034b50) {
  54         unsigned int version = READ_16(header + 4);
  55         unsigned int gpflag = READ_16(header + 6);
  56         unsigned int method = READ_16(header + 8);
  57         unsigned int filetime = READ_16(header + 10);
  58         unsigned int filedate = READ_16(header + 12);
  59         unsigned int crc = READ_32(header + 14); /* crc */
  60         unsigned int cpsize = READ_32(header + 18); /* compressed size */
  61         unsigned int uncpsize = READ_32(header + 22); /* uncompressed sz */
  62         unsigned int fnsize = READ_16(header + 26); /* file name length */
  63         unsigned int extsize = READ_16(header + 28); /* extra field length */
  64         filename[0] = extra[0] = '\0';
  65
  66         /* Header */
  67         if (fwrite(header, 1, 30, fpOut) == 30) {
  68           offset += 30;
  69         } else {
  70           err = Z_ERRNO;
  71           break;
  72         }
  73
  74         /* Filename */
  75         if (fnsize > 0) {
  76           if (fnsize < sizeof(filename)) {
  77             if (fread(filename, 1, fnsize, fpZip) == fnsize) {
  78                 if (fwrite(filename, 1, fnsize, fpOut) == fnsize) {
  79                 offset += fnsize;
  80               } else {
  81                 err = Z_ERRNO;
  82                 break;
  83               }
  84             } else {
  85               err = Z_ERRNO;
  86               break;
  87             }
  88           } else {
  89             err = Z_ERRNO;
  90             break;
  91           }
  92         } else {
  93           err = Z_STREAM_ERROR;
  94           break;
  95         }
  96
  97         /* Extra field */
  98         if (extsize > 0) {
  99           if (extsize < sizeof(extra)) {
 100             if (fread(extra, 1, extsize, fpZip) == extsize) {
 101               if (fwrite(extra, 1, extsize, fpOut) == extsize) {
 102                 offset += extsize;
 103                 } else {
 104                 err = Z_ERRNO;
 105                 break;
 106               }
 107             } else {
 108               err = Z_ERRNO;
 109               break;
 110             }
 111           } else {
 112             err = Z_ERRNO;
 113             break;
 114           }
 115         }
 116
 117         /* Data */
 118         {
 119           int dataSize = cpsize;
 120           if (dataSize == 0) {
 121             dataSize = uncpsize;
 122           }
 123           if (dataSize > 0) {
 124             char* data = malloc(dataSize);
 125             if (data != NULL) {
 126               if ((int)fread(data, 1, dataSize, fpZip) == dataSize) {
 127                 if ((int)fwrite(data, 1, dataSize, fpOut) == dataSize) {
 128                   offset += dataSize;
 129                   totalBytes += dataSize;
 130                 } else {
 131                   err = Z_ERRNO;
 132                 }
 133               } else {
 134                 err = Z_ERRNO;
 135               }
 136               free(data);
 137               if (err != Z_OK) {
 138                 break;
 139               }
 140             } else {
 141               err = Z_MEM_ERROR;
 142               break;
 143             }
 144           }
 145         }
 146
 147         /* Central directory entry */
 148         {
 149           char header[46];
 150           char* comment = "";
 151           int comsize = (int) strlen(comment);
 152           WRITE_32(header, 0x02014b50);
 153           WRITE_16(header + 4, version);
 154           WRITE_16(header + 6, version);
 155           WRITE_16(header + 8, gpflag);
 156           WRITE_16(header + 10, method);
 157           WRITE_16(header + 12, filetime);
 158           WRITE_16(header + 14, filedate);
 159           WRITE_32(header + 16, crc);
 160           WRITE_32(header + 20, cpsize);
 161           WRITE_32(header + 24, uncpsize);
 162           WRITE_16(header + 28, fnsize);
 163           WRITE_16(header + 30, extsize);
 164           WRITE_16(header + 32, comsize);
 165           WRITE_16(header + 34, 0);     /* disk # */
 166           WRITE_16(header + 36, 0);     /* int attrb */
 167           WRITE_32(header + 38, 0);     /* ext attrb */
 168           WRITE_32(header + 42, currentOffset);
 169           /* Header */
 170           if (fwrite(header, 1, 46, fpOutCD) == 46) {
 171             offsetCD += 46;
 172
 173             /* Filename */
 174             if (fnsize > 0) {
 175               if (fwrite(filename, 1, fnsize, fpOutCD) == fnsize) {
 176                 offsetCD += fnsize;
 177               } else {
 178                 err = Z_ERRNO;
 179                 break;
 180               }
 181             } else {
 182               err = Z_STREAM_ERROR;
 183               break;
 184             }
 185
 186             /* Extra field */
 187             if (extsize > 0) {
 188               if (fwrite(extra, 1, extsize, fpOutCD) == extsize) {
 189                 offsetCD += extsize;
 190               } else {
 191                 err = Z_ERRNO;
 192                 break;
 193               }
 194             }
 195
 196             /* Comment field */
 197             if (comsize > 0) {
 198               if ((int)fwrite(comment, 1, comsize, fpOutCD) == comsize) {
 199                 offsetCD += comsize;
 200               } else {
 201                 err = Z_ERRNO;
 202                 break;
 203               }
 204             }
 205
 206
 207           } else {
 208             err = Z_ERRNO;
 209             break;
 210           }
 211         }
 212
 213         /* Success */
 214         entries++;
 215
 216       } else {
 217         break;
 218       }
 219     }
 220
 221     /* Final central directory  */
 222     {
 223       int entriesZip = entries;
 224       char header[22];
 225       char* comment = ""; // "ZIP File recovered by zlib/minizip/mztools";
 226       int comsize = (int) strlen(comment);
 227       if (entriesZip > 0xffff) {
 228         entriesZip = 0xffff;
 229       }
 230       WRITE_32(header, 0x06054b50);
 231       WRITE_16(header + 4, 0);    /* disk # */
 232       WRITE_16(header + 6, 0);    /* disk # */
 233       WRITE_16(header + 8, entriesZip);   /* hack */
 234       WRITE_16(header + 10, entriesZip);  /* hack */
 235       WRITE_32(header + 12, offsetCD);    /* size of CD */
 236       WRITE_32(header + 16, offset);      /* offset to CD */
 237       WRITE_16(header + 20, comsize);     /* comment */
 238
 239       /* Header */
 240       if (fwrite(header, 1, 22, fpOutCD) == 22) {
 241
 242         /* Comment field */
 243         if (comsize > 0) {
 244           if ((int)fwrite(comment, 1, comsize, fpOutCD) != comsize) {
 245             err = Z_ERRNO;
 246           }
 247         }
 248
 249       } else {
 250         err = Z_ERRNO;
 251       }
 252     }
 253
 254     /* Final merge (file + central directory) */
 255     fclose(fpOutCD);
 256     if (err == Z_OK) {
 257       fpOutCD = fopen(fileOutTmp, "rb");
 258       if (fpOutCD != NULL) {
 259         int nRead;
 260         char buffer[8192];
 261         while ( (nRead = (int)fread(buffer, 1, sizeof(buffer), fpOutCD)) > 0) {
 262           if ((int)fwrite(buffer, 1, nRead, fpOut) != nRead) {
 263             err = Z_ERRNO;
 264             break;
 265           }
 266         }
 267         fclose(fpOutCD);
 268       }
 269     }
 270
 271     /* Close */
 272     fclose(fpZip);
 273     fclose(fpOut);
 274
 275     /* Wipe temporary file */
 276     (void)remove(fileOutTmp);
 277
 278     /* Number of recovered entries */
 279     if (err == Z_OK) {
 280       if (nRecovered != NULL) {
 281         *nRecovered = entries;
 282       }
 283       if (bytesRecovered != NULL) {
 284         *bytesRecovered = totalBytes;
 285       }
 286     }
 287   } else {
 288     err = Z_STREAM_ERROR;
 289   }
 290   return err;
 291 }