• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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
filetime(f,tmzip,dt)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__
filetime(f,tmzip,dt)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
filetime(f,tmzip,dt)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 
check_exist_file(filename)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 
do_banner()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 
do_help()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 */
getFileCrc(const char * filenameinzip,void * buf,unsigned long size_buf,unsigned long * result_crc)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 
isLargeFile(const char * filename)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 
main(argc,argv)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 }
521