Home | History | Annotate | Download | only in minizip
      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 }
    521