|
1 /* |
|
2 minizip.c |
|
3 Version 1.01e, February 12th, 2005 |
|
4 |
|
5 Copyright (C) 1998-2005 Gilles Vollant |
|
6 */ |
|
7 |
|
8 #include <stdio.h> |
|
9 #include <stdlib.h> |
|
10 #include <string.h> |
|
11 #include <time.h> |
|
12 #include <errno.h> |
|
13 #include <fcntl.h> |
|
14 |
|
15 #ifdef unix |
|
16 # include <unistd.h> |
|
17 # include <utime.h> |
|
18 # include <sys/types.h> |
|
19 # include <sys/stat.h> |
|
20 #else |
|
21 # include <direct.h> |
|
22 # include <io.h> |
|
23 #endif |
|
24 |
|
25 #include "zip.h" |
|
26 |
|
27 #ifdef WIN32 |
|
28 #define USEWIN32IOAPI |
|
29 #include "iowin32.h" |
|
30 #endif |
|
31 |
|
32 |
|
33 |
|
34 #define WRITEBUFFERSIZE (16384) |
|
35 #define MAXFILENAME (256) |
|
36 |
|
37 #ifdef WIN32 |
|
38 uLong filetime(f, tmzip, dt) |
|
39 char *f; /* name of file to get info on */ |
|
40 tm_zip *tmzip; /* return value: access, modific. and creation times */ |
|
41 uLong *dt; /* dostime */ |
|
42 { |
|
43 int ret = 0; |
|
44 { |
|
45 FILETIME ftLocal; |
|
46 HANDLE hFind; |
|
47 WIN32_FIND_DATA ff32; |
|
48 |
|
49 hFind = FindFirstFile(f,&ff32); |
|
50 if (hFind != INVALID_HANDLE_VALUE) |
|
51 { |
|
52 FileTimeToLocalFileTime(&(ff32.ftLastWriteTime),&ftLocal); |
|
53 FileTimeToDosDateTime(&ftLocal,((LPWORD)dt)+1,((LPWORD)dt)+0); |
|
54 FindClose(hFind); |
|
55 ret = 1; |
|
56 } |
|
57 } |
|
58 return ret; |
|
59 } |
|
60 #else |
|
61 #ifdef unix |
|
62 uLong filetime(f, tmzip, dt) |
|
63 char *f; /* name of file to get info on */ |
|
64 tm_zip *tmzip; /* return value: access, modific. and creation times */ |
|
65 uLong *dt; /* dostime */ |
|
66 { |
|
67 int ret=0; |
|
68 struct stat s; /* results of stat() */ |
|
69 struct tm* filedate; |
|
70 time_t tm_t=0; |
|
71 |
|
72 if (strcmp(f,"-")!=0) |
|
73 { |
|
74 char name[MAXFILENAME+1]; |
|
75 int len = strlen(f); |
|
76 if (len > MAXFILENAME) |
|
77 len = MAXFILENAME; |
|
78 |
|
79 strncpy(name, f,MAXFILENAME-1); |
|
80 /* strncpy doesnt append the trailing NULL, of the string is too long. */ |
|
81 name[ MAXFILENAME ] = '\0'; |
|
82 |
|
83 if (name[len - 1] == '/') |
|
84 name[len - 1] = '\0'; |
|
85 /* not all systems allow stat'ing a file with / appended */ |
|
86 if (stat(name,&s)==0) |
|
87 { |
|
88 tm_t = s.st_mtime; |
|
89 ret = 1; |
|
90 } |
|
91 } |
|
92 filedate = localtime(&tm_t); |
|
93 |
|
94 tmzip->tm_sec = filedate->tm_sec; |
|
95 tmzip->tm_min = filedate->tm_min; |
|
96 tmzip->tm_hour = filedate->tm_hour; |
|
97 tmzip->tm_mday = filedate->tm_mday; |
|
98 tmzip->tm_mon = filedate->tm_mon ; |
|
99 tmzip->tm_year = filedate->tm_year; |
|
100 |
|
101 return ret; |
|
102 } |
|
103 #else |
|
104 uLong filetime(f, tmzip, dt) |
|
105 char *f; /* name of file to get info on */ |
|
106 tm_zip *tmzip; /* return value: access, modific. and creation times */ |
|
107 uLong *dt; /* dostime */ |
|
108 { |
|
109 return 0; |
|
110 } |
|
111 #endif |
|
112 #endif |
|
113 |
|
114 |
|
115 |
|
116 |
|
117 int check_exist_file(filename) |
|
118 const char* filename; |
|
119 { |
|
120 FILE* ftestexist; |
|
121 int ret = 1; |
|
122 ftestexist = fopen(filename,"rb"); |
|
123 if (ftestexist==NULL) |
|
124 ret = 0; |
|
125 else |
|
126 fclose(ftestexist); |
|
127 return ret; |
|
128 } |
|
129 |
|
130 void do_banner() |
|
131 { |
|
132 printf("MiniZip 1.01b, demo of zLib + Zip package written by Gilles Vollant\n"); |
|
133 printf("more info at http://www.winimage.com/zLibDll/unzip.html\n\n"); |
|
134 } |
|
135 |
|
136 void do_help() |
|
137 { |
|
138 printf("Usage : minizip [-o] [-a] [-0 to -9] [-p password] file.zip [files_to_add]\n\n" \ |
|
139 " -o Overwrite existing file.zip\n" \ |
|
140 " -a Append to existing file.zip\n" \ |
|
141 " -0 Store only\n" \ |
|
142 " -1 Compress faster\n" \ |
|
143 " -9 Compress better\n\n"); |
|
144 } |
|
145 |
|
146 /* calculate the CRC32 of a file, |
|
147 because to encrypt a file, we need known the CRC32 of the file before */ |
|
148 int getFileCrc(const char* filenameinzip,void*buf,unsigned long size_buf,unsigned long* result_crc) |
|
149 { |
|
150 unsigned long calculate_crc=0; |
|
151 int err=ZIP_OK; |
|
152 FILE * fin = fopen(filenameinzip,"rb"); |
|
153 unsigned long size_read = 0; |
|
154 unsigned long total_read = 0; |
|
155 if (fin==NULL) |
|
156 { |
|
157 err = ZIP_ERRNO; |
|
158 } |
|
159 |
|
160 if (err == ZIP_OK) |
|
161 do |
|
162 { |
|
163 err = ZIP_OK; |
|
164 size_read = (int)fread(buf,1,size_buf,fin); |
|
165 if (size_read < size_buf) |
|
166 if (feof(fin)==0) |
|
167 { |
|
168 printf("error in reading %s\n",filenameinzip); |
|
169 err = ZIP_ERRNO; |
|
170 } |
|
171 |
|
172 if (size_read>0) |
|
173 calculate_crc = crc32(calculate_crc,buf,size_read); |
|
174 total_read += size_read; |
|
175 |
|
176 } while ((err == ZIP_OK) && (size_read>0)); |
|
177 |
|
178 if (fin) |
|
179 fclose(fin); |
|
180 |
|
181 *result_crc=calculate_crc; |
|
182 printf("file %s crc %x\n",filenameinzip,calculate_crc); |
|
183 return err; |
|
184 } |
|
185 |
|
186 int main(argc,argv) |
|
187 int argc; |
|
188 char *argv[]; |
|
189 { |
|
190 int i; |
|
191 int opt_overwrite=0; |
|
192 int opt_compress_level=Z_DEFAULT_COMPRESSION; |
|
193 int zipfilenamearg = 0; |
|
194 char filename_try[MAXFILENAME+16]; |
|
195 int zipok; |
|
196 int err=0; |
|
197 int size_buf=0; |
|
198 void* buf=NULL; |
|
199 const char* password=NULL; |
|
200 |
|
201 |
|
202 do_banner(); |
|
203 if (argc==1) |
|
204 { |
|
205 do_help(); |
|
206 return 0; |
|
207 } |
|
208 else |
|
209 { |
|
210 for (i=1;i<argc;i++) |
|
211 { |
|
212 if ((*argv[i])=='-') |
|
213 { |
|
214 const char *p=argv[i]+1; |
|
215 |
|
216 while ((*p)!='\0') |
|
217 { |
|
218 char c=*(p++);; |
|
219 if ((c=='o') || (c=='O')) |
|
220 opt_overwrite = 1; |
|
221 if ((c=='a') || (c=='A')) |
|
222 opt_overwrite = 2; |
|
223 if ((c>='0') && (c<='9')) |
|
224 opt_compress_level = c-'0'; |
|
225 |
|
226 if (((c=='p') || (c=='P')) && (i+1<argc)) |
|
227 { |
|
228 password=argv[i+1]; |
|
229 i++; |
|
230 } |
|
231 } |
|
232 } |
|
233 else |
|
234 if (zipfilenamearg == 0) |
|
235 zipfilenamearg = i ; |
|
236 } |
|
237 } |
|
238 |
|
239 size_buf = WRITEBUFFERSIZE; |
|
240 buf = (void*)malloc(size_buf); |
|
241 if (buf==NULL) |
|
242 { |
|
243 printf("Error allocating memory\n"); |
|
244 return ZIP_INTERNALERROR; |
|
245 } |
|
246 |
|
247 if (zipfilenamearg==0) |
|
248 zipok=0; |
|
249 else |
|
250 { |
|
251 int i,len; |
|
252 int dot_found=0; |
|
253 |
|
254 zipok = 1 ; |
|
255 strncpy(filename_try, argv[zipfilenamearg],MAXFILENAME-1); |
|
256 /* strncpy doesnt append the trailing NULL, of the string is too long. */ |
|
257 filename_try[ MAXFILENAME ] = '\0'; |
|
258 |
|
259 len=(int)strlen(filename_try); |
|
260 for (i=0;i<len;i++) |
|
261 if (filename_try[i]=='.') |
|
262 dot_found=1; |
|
263 |
|
264 if (dot_found==0) |
|
265 strcat(filename_try,".zip"); |
|
266 |
|
267 if (opt_overwrite==2) |
|
268 { |
|
269 /* if the file don't exist, we not append file */ |
|
270 if (check_exist_file(filename_try)==0) |
|
271 opt_overwrite=1; |
|
272 } |
|
273 else |
|
274 if (opt_overwrite==0) |
|
275 if (check_exist_file(filename_try)!=0) |
|
276 { |
|
277 char rep=0; |
|
278 do |
|
279 { |
|
280 char answer[128]; |
|
281 int ret; |
|
282 printf("The file %s exists. Overwrite ? [y]es, [n]o, [a]ppend : ",filename_try); |
|
283 ret = scanf("%1s",answer); |
|
284 if (ret != 1) |
|
285 { |
|
286 exit(EXIT_FAILURE); |
|
287 } |
|
288 rep = answer[0] ; |
|
289 if ((rep>='a') && (rep<='z')) |
|
290 rep -= 0x20; |
|
291 } |
|
292 while ((rep!='Y') && (rep!='N') && (rep!='A')); |
|
293 if (rep=='N') |
|
294 zipok = 0; |
|
295 if (rep=='A') |
|
296 opt_overwrite = 2; |
|
297 } |
|
298 } |
|
299 |
|
300 if (zipok==1) |
|
301 { |
|
302 zipFile zf; |
|
303 int errclose; |
|
304 # ifdef USEWIN32IOAPI |
|
305 zlib_filefunc_def ffunc; |
|
306 fill_win32_filefunc(&ffunc); |
|
307 zf = zipOpen2(filename_try,(opt_overwrite==2) ? 2 : 0,NULL,&ffunc); |
|
308 # else |
|
309 zf = zipOpen(filename_try,(opt_overwrite==2) ? 2 : 0); |
|
310 # endif |
|
311 |
|
312 if (zf == NULL) |
|
313 { |
|
314 printf("error opening %s\n",filename_try); |
|
315 err= ZIP_ERRNO; |
|
316 } |
|
317 else |
|
318 printf("creating %s\n",filename_try); |
|
319 |
|
320 for (i=zipfilenamearg+1;(i<argc) && (err==ZIP_OK);i++) |
|
321 { |
|
322 if (!((((*(argv[i]))=='-') || ((*(argv[i]))=='/')) && |
|
323 ((argv[i][1]=='o') || (argv[i][1]=='O') || |
|
324 (argv[i][1]=='a') || (argv[i][1]=='A') || |
|
325 (argv[i][1]=='p') || (argv[i][1]=='P') || |
|
326 ((argv[i][1]>='0') || (argv[i][1]<='9'))) && |
|
327 (strlen(argv[i]) == 2))) |
|
328 { |
|
329 FILE * fin; |
|
330 int size_read; |
|
331 const char* filenameinzip = argv[i]; |
|
332 zip_fileinfo zi; |
|
333 unsigned long crcFile=0; |
|
334 |
|
335 zi.tmz_date.tm_sec = zi.tmz_date.tm_min = zi.tmz_date.tm_hour = |
|
336 zi.tmz_date.tm_mday = zi.tmz_date.tm_mon = zi.tmz_date.tm_year = 0; |
|
337 zi.dosDate = 0; |
|
338 zi.internal_fa = 0; |
|
339 zi.external_fa = 0; |
|
340 filetime(filenameinzip,&zi.tmz_date,&zi.dosDate); |
|
341 |
|
342 /* |
|
343 err = zipOpenNewFileInZip(zf,filenameinzip,&zi, |
|
344 NULL,0,NULL,0,NULL / * comment * /, |
|
345 (opt_compress_level != 0) ? Z_DEFLATED : 0, |
|
346 opt_compress_level); |
|
347 */ |
|
348 if ((password != NULL) && (err==ZIP_OK)) |
|
349 err = getFileCrc(filenameinzip,buf,size_buf,&crcFile); |
|
350 |
|
351 err = zipOpenNewFileInZip3(zf,filenameinzip,&zi, |
|
352 NULL,0,NULL,0,NULL /* comment*/, |
|
353 (opt_compress_level != 0) ? Z_DEFLATED : 0, |
|
354 opt_compress_level,0, |
|
355 /* -MAX_WBITS, DEF_MEM_LEVEL, Z_DEFAULT_STRATEGY, */ |
|
356 -MAX_WBITS, DEF_MEM_LEVEL, Z_DEFAULT_STRATEGY, |
|
357 password,crcFile); |
|
358 |
|
359 if (err != ZIP_OK) |
|
360 printf("error in opening %s in zipfile\n",filenameinzip); |
|
361 else |
|
362 { |
|
363 fin = fopen(filenameinzip,"rb"); |
|
364 if (fin==NULL) |
|
365 { |
|
366 err=ZIP_ERRNO; |
|
367 printf("error in opening %s for reading\n",filenameinzip); |
|
368 } |
|
369 } |
|
370 |
|
371 if (err == ZIP_OK) |
|
372 do |
|
373 { |
|
374 err = ZIP_OK; |
|
375 size_read = (int)fread(buf,1,size_buf,fin); |
|
376 if (size_read < size_buf) |
|
377 if (feof(fin)==0) |
|
378 { |
|
379 printf("error in reading %s\n",filenameinzip); |
|
380 err = ZIP_ERRNO; |
|
381 } |
|
382 |
|
383 if (size_read>0) |
|
384 { |
|
385 err = zipWriteInFileInZip (zf,buf,size_read); |
|
386 if (err<0) |
|
387 { |
|
388 printf("error in writing %s in the zipfile\n", |
|
389 filenameinzip); |
|
390 } |
|
391 |
|
392 } |
|
393 } while ((err == ZIP_OK) && (size_read>0)); |
|
394 |
|
395 if (fin) |
|
396 fclose(fin); |
|
397 |
|
398 if (err<0) |
|
399 err=ZIP_ERRNO; |
|
400 else |
|
401 { |
|
402 err = zipCloseFileInZip(zf); |
|
403 if (err!=ZIP_OK) |
|
404 printf("error in closing %s in the zipfile\n", |
|
405 filenameinzip); |
|
406 } |
|
407 } |
|
408 } |
|
409 errclose = zipClose(zf,NULL); |
|
410 if (errclose != ZIP_OK) |
|
411 printf("error in closing %s\n",filename_try); |
|
412 } |
|
413 else |
|
414 { |
|
415 do_help(); |
|
416 } |
|
417 |
|
418 free(buf); |
|
419 return 0; |
|
420 } |