Commit | Line | Data |
---|---|---|
5ca28f79 L |
1 | /* |
2 | minizip.c | |
3 | Version 1.1, February 14h, 2010 | |
4 | sample part of the MiniZip project - ( http://www.winimage.com/zLibDll/minizip.html ) | |
5 | ||
6 | Copyright (C) 1998-2010 Gilles Vollant (minizip) ( http://www.winimage.com/zLibDll/minizip.html ) | |
7 | ||
8 | Modifications of Unzip for Zip64 | |
9 | Copyright (C) 2007-2008 Even Rouault | |
10 | ||
11 | Modifications for Zip64 support on both zip and unzip | |
12 | Copyright (C) 2009-2010 Mathias Svensson ( http://result42.com ) | |
13 | */ | |
14 | ||
15 | ||
16 | #if (!defined(_WIN32)) && (!defined(WIN32)) && (!defined(__APPLE__)) | |
17 | #ifndef __USE_FILE_OFFSET64 | |
18 | #define __USE_FILE_OFFSET64 | |
19 | #endif | |
20 | #ifndef __USE_LARGEFILE64 | |
21 | #define __USE_LARGEFILE64 | |
22 | #endif | |
23 | #ifndef _LARGEFILE64_SOURCE | |
24 | #define _LARGEFILE64_SOURCE | |
25 | #endif | |
26 | #ifndef _FILE_OFFSET_BIT | |
27 | #define _FILE_OFFSET_BIT 64 | |
28 | #endif | |
29 | #endif | |
30 | ||
31 | #ifdef __APPLE__ | |
32 | // In darwin and perhaps other BSD variants off_t is a 64 bit value, hence no need for specific 64 bit functions | |
33 | #define FOPEN_FUNC(filename, mode) fopen(filename, mode) | |
34 | #define FTELLO_FUNC(stream) ftello(stream) | |
35 | #define FSEEKO_FUNC(stream, offset, origin) fseeko(stream, offset, origin) | |
36 | #else | |
37 | #define FOPEN_FUNC(filename, mode) fopen64(filename, mode) | |
38 | #define FTELLO_FUNC(stream) ftello64(stream) | |
39 | #define FSEEKO_FUNC(stream, offset, origin) fseeko64(stream, offset, origin) | |
40 | #endif | |
41 | ||
42 | ||
43 | ||
44 | #include <stdio.h> | |
45 | #include <stdlib.h> | |
46 | #include <string.h> | |
47 | #include <time.h> | |
48 | #include <errno.h> | |
49 | #include <fcntl.h> | |
50 | ||
51 | #ifdef _WIN32 | |
52 | # include <direct.h> | |
53 | # include <io.h> | |
54 | #else | |
55 | # include <unistd.h> | |
56 | # include <utime.h> | |
57 | # include <sys/types.h> | |
58 | # include <sys/stat.h> | |
59 | #endif | |
60 | ||
61 | #include "zip.h" | |
62 | ||
63 | #ifdef _WIN32 | |
64 | #define USEWIN32IOAPI | |
65 | #include "iowin32.h" | |
66 | #endif | |
67 | ||
68 | ||
69 | ||
70 | #define WRITEBUFFERSIZE (16384) | |
71 | #define MAXFILENAME (256) | |
72 | ||
73 | #ifdef _WIN32 | |
74 | uLong filetime(f, tmzip, dt) | |
75 | char *f; /* name of file to get info on */ | |
76 | tm_zip *tmzip; /* return value: access, modific. and creation times */ | |
77 | uLong *dt; /* dostime */ | |
78 | { | |
79 | int ret = 0; | |
80 | { | |
81 | FILETIME ftLocal; | |
82 | HANDLE hFind; | |
83 | WIN32_FIND_DATAA ff32; | |
84 | ||
85 | hFind = FindFirstFileA(f,&ff32); | |
86 | if (hFind != INVALID_HANDLE_VALUE) | |
87 | { | |
88 | FileTimeToLocalFileTime(&(ff32.ftLastWriteTime),&ftLocal); | |
89 | FileTimeToDosDateTime(&ftLocal,((LPWORD)dt)+1,((LPWORD)dt)+0); | |
90 | FindClose(hFind); | |
91 | ret = 1; | |
92 | } | |
93 | } | |
94 | return ret; | |
95 | } | |
96 | #else | |
97 | #ifdef unix || __APPLE__ | |
98 | uLong filetime(f, tmzip, dt) | |
99 | char *f; /* name of file to get info on */ | |
100 | tm_zip *tmzip; /* return value: access, modific. and creation times */ | |
101 | uLong *dt; /* dostime */ | |
102 | { | |
103 | int ret=0; | |
104 | struct stat s; /* results of stat() */ | |
105 | struct tm* filedate; | |
106 | time_t tm_t=0; | |
107 | ||
108 | if (strcmp(f,"-")!=0) | |
109 | { | |
110 | char name[MAXFILENAME+1]; | |
111 | int len = strlen(f); | |
112 | if (len > MAXFILENAME) | |
113 | len = MAXFILENAME; | |
114 | ||
115 | strncpy(name, f,MAXFILENAME-1); | |
116 | /* strncpy doesnt append the trailing NULL, of the string is too long. */ | |
117 | name[ MAXFILENAME ] = '\0'; | |
118 | ||
119 | if (name[len - 1] == '/') | |
120 | name[len - 1] = '\0'; | |
121 | /* not all systems allow stat'ing a file with / appended */ | |
122 | if (stat(name,&s)==0) | |
123 | { | |
124 | tm_t = s.st_mtime; | |
125 | ret = 1; | |
126 | } | |
127 | } | |
128 | filedate = localtime(&tm_t); | |
129 | ||
130 | tmzip->tm_sec = filedate->tm_sec; | |
131 | tmzip->tm_min = filedate->tm_min; | |
132 | tmzip->tm_hour = filedate->tm_hour; | |
133 | tmzip->tm_mday = filedate->tm_mday; | |
134 | tmzip->tm_mon = filedate->tm_mon ; | |
135 | tmzip->tm_year = filedate->tm_year; | |
136 | ||
137 | return ret; | |
138 | } | |
139 | #else | |
140 | uLong filetime(f, tmzip, dt) | |
141 | char *f; /* name of file to get info on */ | |
142 | tm_zip *tmzip; /* return value: access, modific. and creation times */ | |
143 | uLong *dt; /* dostime */ | |
144 | { | |
145 | return 0; | |
146 | } | |
147 | #endif | |
148 | #endif | |
149 | ||
150 | ||
151 | ||
152 | ||
153 | int check_exist_file(filename) | |
154 | const char* filename; | |
155 | { | |
156 | FILE* ftestexist; | |
157 | int ret = 1; | |
158 | ftestexist = FOPEN_FUNC(filename,"rb"); | |
159 | if (ftestexist==NULL) | |
160 | ret = 0; | |
161 | else | |
162 | fclose(ftestexist); | |
163 | return ret; | |
164 | } | |
165 | ||
166 | void do_banner() | |
167 | { | |
168 | printf("MiniZip 1.1, demo of zLib + MiniZip64 package, written by Gilles Vollant\n"); | |
169 | printf("more info on MiniZip at http://www.winimage.com/zLibDll/minizip.html\n\n"); | |
170 | } | |
171 | ||
172 | void do_help() | |
173 | { | |
174 | printf("Usage : minizip [-o] [-a] [-0 to -9] [-p password] [-j] file.zip [files_to_add]\n\n" \ | |
175 | " -o Overwrite existing file.zip\n" \ | |
176 | " -a Append to existing file.zip\n" \ | |
177 | " -0 Store only\n" \ | |
178 | " -1 Compress faster\n" \ | |
179 | " -9 Compress better\n\n" \ | |
180 | " -j exclude path. store only the file name.\n\n"); | |
181 | } | |
182 | ||
183 | /* calculate the CRC32 of a file, | |
184 | because to encrypt a file, we need known the CRC32 of the file before */ | |
185 | int getFileCrc(const char* filenameinzip,void*buf,unsigned long size_buf,unsigned long* result_crc) | |
186 | { | |
187 | unsigned long calculate_crc=0; | |
188 | int err=ZIP_OK; | |
189 | FILE * fin = FOPEN_FUNC(filenameinzip,"rb"); | |
190 | ||
191 | unsigned long size_read = 0; | |
192 | unsigned long total_read = 0; | |
193 | if (fin==NULL) | |
194 | { | |
195 | err = ZIP_ERRNO; | |
196 | } | |
197 | ||
198 | if (err == ZIP_OK) | |
199 | do | |
200 | { | |
201 | err = ZIP_OK; | |
202 | size_read = (int)fread(buf,1,size_buf,fin); | |
203 | if (size_read < size_buf) | |
204 | if (feof(fin)==0) | |
205 | { | |
206 | printf("error in reading %s\n",filenameinzip); | |
207 | err = ZIP_ERRNO; | |
208 | } | |
209 | ||
210 | if (size_read>0) | |
211 | calculate_crc = crc32(calculate_crc,buf,size_read); | |
212 | total_read += size_read; | |
213 | ||
214 | } while ((err == ZIP_OK) && (size_read>0)); | |
215 | ||
216 | if (fin) | |
217 | fclose(fin); | |
218 | ||
219 | *result_crc=calculate_crc; | |
220 | printf("file %s crc %lx\n", filenameinzip, calculate_crc); | |
221 | return err; | |
222 | } | |
223 | ||
224 | int isLargeFile(const char* filename) | |
225 | { | |
226 | int largeFile = 0; | |
227 | ZPOS64_T pos = 0; | |
228 | FILE* pFile = FOPEN_FUNC(filename, "rb"); | |
229 | ||
230 | if(pFile != NULL) | |
231 | { | |
232 | int n = FSEEKO_FUNC(pFile, 0, SEEK_END); | |
233 | pos = FTELLO_FUNC(pFile); | |
234 | ||
235 | printf("File : %s is %lld bytes\n", filename, pos); | |
236 | ||
237 | if(pos >= 0xffffffff) | |
238 | largeFile = 1; | |
239 | ||
240 | fclose(pFile); | |
241 | } | |
242 | ||
243 | return largeFile; | |
244 | } | |
245 | ||
246 | int main(argc,argv) | |
247 | int argc; | |
248 | char *argv[]; | |
249 | { | |
250 | int i; | |
251 | int opt_overwrite=0; | |
252 | int opt_compress_level=Z_DEFAULT_COMPRESSION; | |
253 | int opt_exclude_path=0; | |
254 | int zipfilenamearg = 0; | |
255 | char filename_try[MAXFILENAME+16]; | |
256 | int zipok; | |
257 | int err=0; | |
258 | int size_buf=0; | |
259 | void* buf=NULL; | |
260 | const char* password=NULL; | |
261 | ||
262 | ||
263 | do_banner(); | |
264 | if (argc==1) | |
265 | { | |
266 | do_help(); | |
267 | return 0; | |
268 | } | |
269 | else | |
270 | { | |
271 | for (i=1;i<argc;i++) | |
272 | { | |
273 | if ((*argv[i])=='-') | |
274 | { | |
275 | const char *p=argv[i]+1; | |
276 | ||
277 | while ((*p)!='\0') | |
278 | { | |
279 | char c=*(p++);; | |
280 | if ((c=='o') || (c=='O')) | |
281 | opt_overwrite = 1; | |
282 | if ((c=='a') || (c=='A')) | |
283 | opt_overwrite = 2; | |
284 | if ((c>='0') && (c<='9')) | |
285 | opt_compress_level = c-'0'; | |
286 | if ((c=='j') || (c=='J')) | |
287 | opt_exclude_path = 1; | |
288 | ||
289 | if (((c=='p') || (c=='P')) && (i+1<argc)) | |
290 | { | |
291 | password=argv[i+1]; | |
292 | i++; | |
293 | } | |
294 | } | |
295 | } | |
296 | else | |
297 | { | |
298 | if (zipfilenamearg == 0) | |
299 | { | |
300 | zipfilenamearg = i ; | |
301 | } | |
302 | } | |
303 | } | |
304 | } | |
305 | ||
306 | size_buf = WRITEBUFFERSIZE; | |
307 | buf = (void*)malloc(size_buf); | |
308 | if (buf==NULL) | |
309 | { | |
310 | printf("Error allocating memory\n"); | |
311 | return ZIP_INTERNALERROR; | |
312 | } | |
313 | ||
314 | if (zipfilenamearg==0) | |
315 | { | |
316 | zipok=0; | |
317 | } | |
318 | else | |
319 | { | |
320 | int i,len; | |
321 | int dot_found=0; | |
322 | ||
323 | zipok = 1 ; | |
324 | strncpy(filename_try, argv[zipfilenamearg],MAXFILENAME-1); | |
325 | /* strncpy doesnt append the trailing NULL, of the string is too long. */ | |
326 | filename_try[ MAXFILENAME ] = '\0'; | |
327 | ||
328 | len=(int)strlen(filename_try); | |
329 | for (i=0;i<len;i++) | |
330 | if (filename_try[i]=='.') | |
331 | dot_found=1; | |
332 | ||
333 | if (dot_found==0) | |
334 | strcat(filename_try,".zip"); | |
335 | ||
336 | if (opt_overwrite==2) | |
337 | { | |
338 | /* if the file don't exist, we not append file */ | |
339 | if (check_exist_file(filename_try)==0) | |
340 | opt_overwrite=1; | |
341 | } | |
342 | else | |
343 | if (opt_overwrite==0) | |
344 | if (check_exist_file(filename_try)!=0) | |
345 | { | |
346 | char rep=0; | |
347 | do | |
348 | { | |
349 | char answer[128]; | |
350 | int ret; | |
351 | printf("The file %s exists. Overwrite ? [y]es, [n]o, [a]ppend : ",filename_try); | |
352 | ret = scanf("%1s",answer); | |
353 | if (ret != 1) | |
354 | { | |
355 | exit(EXIT_FAILURE); | |
356 | } | |
357 | rep = answer[0] ; | |
358 | if ((rep>='a') && (rep<='z')) | |
359 | rep -= 0x20; | |
360 | } | |
361 | while ((rep!='Y') && (rep!='N') && (rep!='A')); | |
362 | if (rep=='N') | |
363 | zipok = 0; | |
364 | if (rep=='A') | |
365 | opt_overwrite = 2; | |
366 | } | |
367 | } | |
368 | ||
369 | if (zipok==1) | |
370 | { | |
371 | zipFile zf; | |
372 | int errclose; | |
373 | # ifdef USEWIN32IOAPI | |
374 | zlib_filefunc64_def ffunc; | |
375 | fill_win32_filefunc64A(&ffunc); | |
376 | zf = zipOpen2_64(filename_try,(opt_overwrite==2) ? 2 : 0,NULL,&ffunc); | |
377 | # else | |
378 | zf = zipOpen64(filename_try,(opt_overwrite==2) ? 2 : 0); | |
379 | # endif | |
380 | ||
381 | if (zf == NULL) | |
382 | { | |
383 | printf("error opening %s\n",filename_try); | |
384 | err= ZIP_ERRNO; | |
385 | } | |
386 | else | |
387 | printf("creating %s\n",filename_try); | |
388 | ||
389 | for (i=zipfilenamearg+1;(i<argc) && (err==ZIP_OK);i++) | |
390 | { | |
391 | if (!((((*(argv[i]))=='-') || ((*(argv[i]))=='/')) && | |
392 | ((argv[i][1]=='o') || (argv[i][1]=='O') || | |
393 | (argv[i][1]=='a') || (argv[i][1]=='A') || | |
394 | (argv[i][1]=='p') || (argv[i][1]=='P') || | |
395 | ((argv[i][1]>='0') || (argv[i][1]<='9'))) && | |
396 | (strlen(argv[i]) == 2))) | |
397 | { | |
398 | FILE * fin; | |
399 | int size_read; | |
400 | const char* filenameinzip = argv[i]; | |
401 | const char *savefilenameinzip; | |
402 | zip_fileinfo zi; | |
403 | unsigned long crcFile=0; | |
404 | int zip64 = 0; | |
405 | ||
406 | zi.tmz_date.tm_sec = zi.tmz_date.tm_min = zi.tmz_date.tm_hour = | |
407 | zi.tmz_date.tm_mday = zi.tmz_date.tm_mon = zi.tmz_date.tm_year = 0; | |
408 | zi.dosDate = 0; | |
409 | zi.internal_fa = 0; | |
410 | zi.external_fa = 0; | |
411 | filetime(filenameinzip,&zi.tmz_date,&zi.dosDate); | |
412 | ||
413 | /* | |
414 | err = zipOpenNewFileInZip(zf,filenameinzip,&zi, | |
415 | NULL,0,NULL,0,NULL / * comment * /, | |
416 | (opt_compress_level != 0) ? Z_DEFLATED : 0, | |
417 | opt_compress_level); | |
418 | */ | |
419 | if ((password != NULL) && (err==ZIP_OK)) | |
420 | err = getFileCrc(filenameinzip,buf,size_buf,&crcFile); | |
421 | ||
422 | zip64 = isLargeFile(filenameinzip); | |
423 | ||
424 | /* The path name saved, should not include a leading slash. */ | |
425 | /*if it did, windows/xp and dynazip couldn't read the zip file. */ | |
426 | savefilenameinzip = filenameinzip; | |
427 | while( savefilenameinzip[0] == '\\' || savefilenameinzip[0] == '/' ) | |
428 | { | |
429 | savefilenameinzip++; | |
430 | } | |
431 | ||
432 | /*should the zip file contain any path at all?*/ | |
433 | if( opt_exclude_path ) | |
434 | { | |
435 | const char *tmpptr; | |
436 | const char *lastslash = 0; | |
437 | for( tmpptr = savefilenameinzip; *tmpptr; tmpptr++) | |
438 | { | |
439 | if( *tmpptr == '\\' || *tmpptr == '/') | |
440 | { | |
441 | lastslash = tmpptr; | |
442 | } | |
443 | } | |
444 | if( lastslash != NULL ) | |
445 | { | |
446 | savefilenameinzip = lastslash+1; // base filename follows last slash. | |
447 | } | |
448 | } | |
449 | ||
450 | /**/ | |
451 | err = zipOpenNewFileInZip3_64(zf,savefilenameinzip,&zi, | |
452 | NULL,0,NULL,0,NULL /* comment*/, | |
453 | (opt_compress_level != 0) ? Z_DEFLATED : 0, | |
454 | opt_compress_level,0, | |
455 | /* -MAX_WBITS, DEF_MEM_LEVEL, Z_DEFAULT_STRATEGY, */ | |
456 | -MAX_WBITS, DEF_MEM_LEVEL, Z_DEFAULT_STRATEGY, | |
457 | password,crcFile, zip64); | |
458 | ||
459 | if (err != ZIP_OK) | |
460 | printf("error in opening %s in zipfile\n",filenameinzip); | |
461 | else | |
462 | { | |
463 | fin = FOPEN_FUNC(filenameinzip,"rb"); | |
464 | if (fin==NULL) | |
465 | { | |
466 | err=ZIP_ERRNO; | |
467 | printf("error in opening %s for reading\n",filenameinzip); | |
468 | } | |
469 | } | |
470 | ||
471 | if (err == ZIP_OK) | |
472 | do | |
473 | { | |
474 | err = ZIP_OK; | |
475 | size_read = (int)fread(buf,1,size_buf,fin); | |
476 | if (size_read < size_buf) | |
477 | if (feof(fin)==0) | |
478 | { | |
479 | printf("error in reading %s\n",filenameinzip); | |
480 | err = ZIP_ERRNO; | |
481 | } | |
482 | ||
483 | if (size_read>0) | |
484 | { | |
485 | err = zipWriteInFileInZip (zf,buf,size_read); | |
486 | if (err<0) | |
487 | { | |
488 | printf("error in writing %s in the zipfile\n", | |
489 | filenameinzip); | |
490 | } | |
491 | ||
492 | } | |
493 | } while ((err == ZIP_OK) && (size_read>0)); | |
494 | ||
495 | if (fin) | |
496 | fclose(fin); | |
497 | ||
498 | if (err<0) | |
499 | err=ZIP_ERRNO; | |
500 | else | |
501 | { | |
502 | err = zipCloseFileInZip(zf); | |
503 | if (err!=ZIP_OK) | |
504 | printf("error in closing %s in the zipfile\n", | |
505 | filenameinzip); | |
506 | } | |
507 | } | |
508 | } | |
509 | errclose = zipClose(zf,NULL); | |
510 | if (errclose != ZIP_OK) | |
511 | printf("error in closing %s\n",filename_try); | |
512 | } | |
513 | else | |
514 | { | |
515 | do_help(); | |
516 | } | |
517 | ||
518 | free(buf); | |
519 | return 0; | |
520 | } |