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