Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line |
|---|---|---|---|
| 8614 | maxcodehac | 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 |
||
| 9 | #include |
||
| 10 | #include |
||
| 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[256]; |
||
| 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 (fread(filename, 1, fnsize, fpZip) == fnsize) { |
||
| 77 | if (fwrite(filename, 1, fnsize, fpOut) == fnsize) { |
||
| 78 | offset += fnsize; |
||
| 79 | } else { |
||
| 80 | err = Z_ERRNO; |
||
| 81 | break; |
||
| 82 | } |
||
| 83 | } else { |
||
| 84 | err = Z_ERRNO; |
||
| 85 | break; |
||
| 86 | } |
||
| 87 | } else { |
||
| 88 | err = Z_STREAM_ERROR; |
||
| 89 | break; |
||
| 90 | } |
||
| 91 | |||
| 92 | /* Extra field */ |
||
| 93 | if (extsize > 0) { |
||
| 94 | if (fread(extra, 1, extsize, fpZip) == extsize) { |
||
| 95 | if (fwrite(extra, 1, extsize, fpOut) == extsize) { |
||
| 96 | offset += extsize; |
||
| 97 | } else { |
||
| 98 | err = Z_ERRNO; |
||
| 99 | break; |
||
| 100 | } |
||
| 101 | } else { |
||
| 102 | err = Z_ERRNO; |
||
| 103 | break; |
||
| 104 | } |
||
| 105 | } |
||
| 106 | |||
| 107 | /* Data */ |
||
| 108 | { |
||
| 109 | int dataSize = cpsize; |
||
| 110 | if (dataSize == 0) { |
||
| 111 | dataSize = uncpsize; |
||
| 112 | } |
||
| 113 | if (dataSize > 0) { |
||
| 114 | char* data = malloc(dataSize); |
||
| 115 | if (data != NULL) { |
||
| 116 | if ((int)fread(data, 1, dataSize, fpZip) == dataSize) { |
||
| 117 | if ((int)fwrite(data, 1, dataSize, fpOut) == dataSize) { |
||
| 118 | offset += dataSize; |
||
| 119 | totalBytes += dataSize; |
||
| 120 | } else { |
||
| 121 | err = Z_ERRNO; |
||
| 122 | } |
||
| 123 | } else { |
||
| 124 | err = Z_ERRNO; |
||
| 125 | } |
||
| 126 | free(data); |
||
| 127 | if (err != Z_OK) { |
||
| 128 | break; |
||
| 129 | } |
||
| 130 | } else { |
||
| 131 | err = Z_MEM_ERROR; |
||
| 132 | break; |
||
| 133 | } |
||
| 134 | } |
||
| 135 | } |
||
| 136 | |||
| 137 | /* Central directory entry */ |
||
| 138 | { |
||
| 139 | char header[46]; |
||
| 140 | char* comment = ""; |
||
| 141 | int comsize = (int) strlen(comment); |
||
| 142 | WRITE_32(header, 0x02014b50); |
||
| 143 | WRITE_16(header + 4, version); |
||
| 144 | WRITE_16(header + 6, version); |
||
| 145 | WRITE_16(header + 8, gpflag); |
||
| 146 | WRITE_16(header + 10, method); |
||
| 147 | WRITE_16(header + 12, filetime); |
||
| 148 | WRITE_16(header + 14, filedate); |
||
| 149 | WRITE_32(header + 16, crc); |
||
| 150 | WRITE_32(header + 20, cpsize); |
||
| 151 | WRITE_32(header + 24, uncpsize); |
||
| 152 | WRITE_16(header + 28, fnsize); |
||
| 153 | WRITE_16(header + 30, extsize); |
||
| 154 | WRITE_16(header + 32, comsize); |
||
| 155 | WRITE_16(header + 34, 0); /* disk # */ |
||
| 156 | WRITE_16(header + 36, 0); /* int attrb */ |
||
| 157 | WRITE_32(header + 38, 0); /* ext attrb */ |
||
| 158 | WRITE_32(header + 42, currentOffset); |
||
| 159 | /* Header */ |
||
| 160 | if (fwrite(header, 1, 46, fpOutCD) == 46) { |
||
| 161 | offsetCD += 46; |
||
| 162 | |||
| 163 | /* Filename */ |
||
| 164 | if (fnsize > 0) { |
||
| 165 | if (fwrite(filename, 1, fnsize, fpOutCD) == fnsize) { |
||
| 166 | offsetCD += fnsize; |
||
| 167 | } else { |
||
| 168 | err = Z_ERRNO; |
||
| 169 | break; |
||
| 170 | } |
||
| 171 | } else { |
||
| 172 | err = Z_STREAM_ERROR; |
||
| 173 | break; |
||
| 174 | } |
||
| 175 | |||
| 176 | /* Extra field */ |
||
| 177 | if (extsize > 0) { |
||
| 178 | if (fwrite(extra, 1, extsize, fpOutCD) == extsize) { |
||
| 179 | offsetCD += extsize; |
||
| 180 | } else { |
||
| 181 | err = Z_ERRNO; |
||
| 182 | break; |
||
| 183 | } |
||
| 184 | } |
||
| 185 | |||
| 186 | /* Comment field */ |
||
| 187 | if (comsize > 0) { |
||
| 188 | if ((int)fwrite(comment, 1, comsize, fpOutCD) == comsize) { |
||
| 189 | offsetCD += comsize; |
||
| 190 | } else { |
||
| 191 | err = Z_ERRNO; |
||
| 192 | break; |
||
| 193 | } |
||
| 194 | } |
||
| 195 | |||
| 196 | |||
| 197 | } else { |
||
| 198 | err = Z_ERRNO; |
||
| 199 | break; |
||
| 200 | } |
||
| 201 | } |
||
| 202 | |||
| 203 | /* Success */ |
||
| 204 | entries++; |
||
| 205 | |||
| 206 | } else { |
||
| 207 | break; |
||
| 208 | } |
||
| 209 | } |
||
| 210 | |||
| 211 | /* Final central directory */ |
||
| 212 | { |
||
| 213 | int entriesZip = entries; |
||
| 214 | char header[22]; |
||
| 215 | char* comment = ""; // "ZIP File recovered by zlib/minizip/mztools"; |
||
| 216 | int comsize = (int) strlen(comment); |
||
| 217 | if (entriesZip > 0xffff) { |
||
| 218 | entriesZip = 0xffff; |
||
| 219 | } |
||
| 220 | WRITE_32(header, 0x06054b50); |
||
| 221 | WRITE_16(header + 4, 0); /* disk # */ |
||
| 222 | WRITE_16(header + 6, 0); /* disk # */ |
||
| 223 | WRITE_16(header + 8, entriesZip); /* hack */ |
||
| 224 | WRITE_16(header + 10, entriesZip); /* hack */ |
||
| 225 | WRITE_32(header + 12, offsetCD); /* size of CD */ |
||
| 226 | WRITE_32(header + 16, offset); /* offset to CD */ |
||
| 227 | WRITE_16(header + 20, comsize); /* comment */ |
||
| 228 | |||
| 229 | /* Header */ |
||
| 230 | if (fwrite(header, 1, 22, fpOutCD) == 22) { |
||
| 231 | |||
| 232 | /* Comment field */ |
||
| 233 | if (comsize > 0) { |
||
| 234 | if ((int)fwrite(comment, 1, comsize, fpOutCD) != comsize) { |
||
| 235 | err = Z_ERRNO; |
||
| 236 | } |
||
| 237 | } |
||
| 238 | |||
| 239 | } else { |
||
| 240 | err = Z_ERRNO; |
||
| 241 | } |
||
| 242 | } |
||
| 243 | |||
| 244 | /* Final merge (file + central directory) */ |
||
| 245 | fclose(fpOutCD); |
||
| 246 | if (err == Z_OK) { |
||
| 247 | fpOutCD = fopen(fileOutTmp, "rb"); |
||
| 248 | if (fpOutCD != NULL) { |
||
| 249 | int nRead; |
||
| 250 | char buffer[8192]; |
||
| 251 | while ( (nRead = (int)fread(buffer, 1, sizeof(buffer), fpOutCD)) > 0) { |
||
| 252 | if ((int)fwrite(buffer, 1, nRead, fpOut) != nRead) { |
||
| 253 | err = Z_ERRNO; |
||
| 254 | break; |
||
| 255 | } |
||
| 256 | } |
||
| 257 | fclose(fpOutCD); |
||
| 258 | } |
||
| 259 | } |
||
| 260 | |||
| 261 | /* Close */ |
||
| 262 | fclose(fpZip); |
||
| 263 | fclose(fpOut); |
||
| 264 | |||
| 265 | /* Wipe temporary file */ |
||
| 266 | (void)remove(fileOutTmp); |
||
| 267 | |||
| 268 | /* Number of recovered entries */ |
||
| 269 | if (err == Z_OK) { |
||
| 270 | if (nRecovered != NULL) { |
||
| 271 | *nRecovered = entries; |
||
| 272 | } |
||
| 273 | if (bytesRecovered != NULL) { |
||
| 274 | *bytesRecovered = totalBytes; |
||
| 275 | } |
||
| 276 | } |
||
| 277 | } else { |
||
| 278 | err = Z_STREAM_ERROR; |
||
| 279 | } |
||
| 280 | return err; |
||
| 281 | } |