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[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 = 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 | } |
---|