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  static int filetime(f, tmzip, dt)
      75      const 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  #if defined(unix) || defined(__APPLE__)
      98  static int filetime(f, tmzip, dt)
      99      const 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    (void)dt;
     104    int ret=0;
     105    struct stat s;        /* results of stat() */
     106    struct tm* filedate;
     107    time_t tm_t=0;
     108  
     109    if (strcmp(f,"-")!=0)
     110    {
     111      char name[MAXFILENAME+1];
     112      size_t len = strlen(f);
     113      if (len > MAXFILENAME)
     114        len = MAXFILENAME;
     115  
     116      strncpy(name, f,MAXFILENAME-1);
     117      /* strncpy doesnt append the trailing NULL, of the string is too long. */
     118      name[ MAXFILENAME ] = '\0';
     119  
     120      if (name[len - 1] == '/')
     121        name[len - 1] = '\0';
     122      /* not all systems allow stat'ing a file with / appended */
     123      if (stat(name,&s)==0)
     124      {
     125        tm_t = s.st_mtime;
     126        ret = 1;
     127      }
     128    }
     129    filedate = localtime(&tm_t);
     130  
     131    tmzip->tm_sec  = filedate->tm_sec;
     132    tmzip->tm_min  = filedate->tm_min;
     133    tmzip->tm_hour = filedate->tm_hour;
     134    tmzip->tm_mday = filedate->tm_mday;
     135    tmzip->tm_mon  = filedate->tm_mon ;
     136    tmzip->tm_year = filedate->tm_year;
     137  
     138    return ret;
     139  }
     140  #else
     141  uLong filetime(f, tmzip, dt)
     142      const char *f;          /* name of file to get info on */
     143      tm_zip *tmzip;             /* return value: access, modific. and creation times */
     144      uLong *dt;             /* dostime */
     145  {
     146      return 0;
     147  }
     148  #endif
     149  #endif
     150  
     151  
     152  
     153  
     154  static int check_exist_file(filename)
     155      const char* filename;
     156  {
     157      FILE* ftestexist;
     158      int ret = 1;
     159      ftestexist = FOPEN_FUNC(filename,"rb");
     160      if (ftestexist==NULL)
     161          ret = 0;
     162      else
     163          fclose(ftestexist);
     164      return ret;
     165  }
     166  
     167  static void do_banner()
     168  {
     169      printf("MiniZip 1.1, demo of zLib + MiniZip64 package, written by Gilles Vollant\n");
     170      printf("more info on MiniZip at http://www.winimage.com/zLibDll/minizip.html\n\n");
     171  }
     172  
     173  static void do_help()
     174  {
     175      printf("Usage : minizip [-o] [-a] [-0 to -9] [-p password] [-j] file.zip [files_to_add]\n\n" \
     176             "  -o  Overwrite existing file.zip\n" \
     177             "  -a  Append to existing file.zip\n" \
     178             "  -0  Store only\n" \
     179             "  -1  Compress faster\n" \
     180             "  -9  Compress better\n\n" \
     181             "  -j  exclude path. store only the file name.\n\n");
     182  }
     183  
     184  /* calculate the CRC32 of a file,
     185     because to encrypt a file, we need known the CRC32 of the file before */
     186  static int getFileCrc(const char* filenameinzip,void*buf,unsigned long size_buf,unsigned long* result_crc)
     187  {
     188     unsigned long calculate_crc=0;
     189     int err=ZIP_OK;
     190     FILE * fin = FOPEN_FUNC(filenameinzip,"rb");
     191  
     192     unsigned long size_read = 0;
     193     unsigned long total_read = 0;
     194     if (fin==NULL)
     195     {
     196         err = ZIP_ERRNO;
     197     }
     198  
     199      if (err == ZIP_OK)
     200          do
     201          {
     202              err = ZIP_OK;
     203              size_read = fread(buf,1,size_buf,fin);
     204              if (size_read < size_buf)
     205                  if (feof(fin)==0)
     206              {
     207                  printf("error in reading %s\n",filenameinzip);
     208                  err = ZIP_ERRNO;
     209              }
     210  
     211              if (size_read>0)
     212                  calculate_crc = crc32_z(calculate_crc,buf,size_read);
     213              total_read += size_read;
     214  
     215          } while ((err == ZIP_OK) && (size_read>0));
     216  
     217      if (fin)
     218          fclose(fin);
     219  
     220      *result_crc=calculate_crc;
     221      printf("file %s crc %lx\n", filenameinzip, calculate_crc);
     222      return err;
     223  }
     224  
     225  static int isLargeFile(const char* filename)
     226  {
     227    int largeFile = 0;
     228    ZPOS64_T pos = 0;
     229    FILE* pFile = FOPEN_FUNC(filename, "rb");
     230  
     231    if(pFile != NULL)
     232    {
     233      FSEEKO_FUNC(pFile, 0, SEEK_END);
     234      pos = (ZPOS64_T)FTELLO_FUNC(pFile);
     235  
     236                  printf("File : %s is %lld bytes\n", filename, pos);
     237  
     238      if(pos >= 0xffffffff)
     239       largeFile = 1;
     240  
     241                  fclose(pFile);
     242    }
     243  
     244   return largeFile;
     245  }
     246  
     247  int main(argc,argv)
     248      int argc;
     249      char *argv[];
     250  {
     251      int i;
     252      int opt_overwrite=0;
     253      int opt_compress_level=Z_DEFAULT_COMPRESSION;
     254      int opt_exclude_path=0;
     255      int zipfilenamearg = 0;
     256      char filename_try[MAXFILENAME+16];
     257      int zipok;
     258      int err=0;
     259      size_t size_buf=0;
     260      void* buf=NULL;
     261      const char* password=NULL;
     262  
     263  
     264      do_banner();
     265      if (argc==1)
     266      {
     267          do_help();
     268          return 0;
     269      }
     270      else
     271      {
     272          for (i=1;i<argc;i++)
     273          {
     274              if ((*argv[i])=='-')
     275              {
     276                  const char *p=argv[i]+1;
     277  
     278                  while ((*p)!='\0')
     279                  {
     280                      char c=*(p++);;
     281                      if ((c=='o') || (c=='O'))
     282                          opt_overwrite = 1;
     283                      if ((c=='a') || (c=='A'))
     284                          opt_overwrite = 2;
     285                      if ((c>='0') && (c<='9'))
     286                          opt_compress_level = c-'0';
     287                      if ((c=='j') || (c=='J'))
     288                          opt_exclude_path = 1;
     289  
     290                      if (((c=='p') || (c=='P')) && (i+1<argc))
     291                      {
     292                          password=argv[i+1];
     293                          i++;
     294                      }
     295                  }
     296              }
     297              else
     298              {
     299                  if (zipfilenamearg == 0)
     300                  {
     301                      zipfilenamearg = i ;
     302                  }
     303              }
     304          }
     305      }
     306  
     307      size_buf = WRITEBUFFERSIZE;
     308      buf = (void*)malloc(size_buf);
     309      if (buf==NULL)
     310      {
     311          printf("Error allocating memory\n");
     312          return ZIP_INTERNALERROR;
     313      }
     314  
     315      if (zipfilenamearg==0)
     316      {
     317          zipok=0;
     318      }
     319      else
     320      {
     321          int i,len;
     322          int dot_found=0;
     323  
     324          zipok = 1 ;
     325          strncpy(filename_try, argv[zipfilenamearg],MAXFILENAME-1);
     326          /* strncpy doesnt append the trailing NULL, of the string is too long. */
     327          filename_try[ MAXFILENAME ] = '\0';
     328  
     329          len=(int)strlen(filename_try);
     330          for (i=0;i<len;i++)
     331              if (filename_try[i]=='.')
     332                  dot_found=1;
     333  
     334          if (dot_found==0)
     335              strcat(filename_try,".zip");
     336  
     337          if (opt_overwrite==2)
     338          {
     339              /* if the file don't exist, we not append file */
     340              if (check_exist_file(filename_try)==0)
     341                  opt_overwrite=1;
     342          }
     343          else
     344          if (opt_overwrite==0)
     345              if (check_exist_file(filename_try)!=0)
     346              {
     347                  char rep=0;
     348                  do
     349                  {
     350                      char answer[128];
     351                      int ret;
     352                      printf("The file %s exists. Overwrite ? [y]es, [n]o, [a]ppend : ",filename_try);
     353                      ret = scanf("%1s",answer);
     354                      if (ret != 1)
     355                      {
     356                         exit(EXIT_FAILURE);
     357                      }
     358                      rep = answer[0] ;
     359                      if ((rep>='a') && (rep<='z'))
     360                          rep -= 0x20;
     361                  }
     362                  while ((rep!='Y') && (rep!='N') && (rep!='A'));
     363                  if (rep=='N')
     364                      zipok = 0;
     365                  if (rep=='A')
     366                      opt_overwrite = 2;
     367              }
     368      }
     369  
     370      if (zipok==1)
     371      {
     372          zipFile zf;
     373          int errclose;
     374  #        ifdef USEWIN32IOAPI
     375          zlib_filefunc64_def ffunc;
     376          fill_win32_filefunc64A(&ffunc);
     377          zf = zipOpen2_64(filename_try,(opt_overwrite==2) ? 2 : 0,NULL,&ffunc);
     378  #        else
     379          zf = zipOpen64(filename_try,(opt_overwrite==2) ? 2 : 0);
     380  #        endif
     381  
     382          if (zf == NULL)
     383          {
     384              printf("error opening %s\n",filename_try);
     385              err= ZIP_ERRNO;
     386          }
     387          else
     388              printf("creating %s\n",filename_try);
     389  
     390          for (i=zipfilenamearg+1;(i<argc) && (err==ZIP_OK);i++)
     391          {
     392              if (!((((*(argv[i]))=='-') || ((*(argv[i]))=='/')) &&
     393                    ((argv[i][1]=='o') || (argv[i][1]=='O') ||
     394                     (argv[i][1]=='a') || (argv[i][1]=='A') ||
     395                     (argv[i][1]=='p') || (argv[i][1]=='P') ||
     396                     ((argv[i][1]>='0') || (argv[i][1]<='9'))) &&
     397                    (strlen(argv[i]) == 2)))
     398              {
     399                  FILE * fin;
     400                  size_t size_read;
     401                  const char* filenameinzip = argv[i];
     402                  const char *savefilenameinzip;
     403                  zip_fileinfo zi;
     404                  unsigned long crcFile=0;
     405                  int zip64 = 0;
     406  
     407                  zi.tmz_date.tm_sec = zi.tmz_date.tm_min = zi.tmz_date.tm_hour =
     408                  zi.tmz_date.tm_mday = zi.tmz_date.tm_mon = zi.tmz_date.tm_year = 0;
     409                  zi.dosDate = 0;
     410                  zi.internal_fa = 0;
     411                  zi.external_fa = 0;
     412                  filetime(filenameinzip,&zi.tmz_date,&zi.dosDate);
     413  
     414  /*
     415                  err = zipOpenNewFileInZip(zf,filenameinzip,&zi,
     416                                   NULL,0,NULL,0,NULL / * comment * /,
     417                                   (opt_compress_level != 0) ? Z_DEFLATED : 0,
     418                                   opt_compress_level);
     419  */
     420                  if ((password != NULL) && (err==ZIP_OK))
     421                      err = getFileCrc(filenameinzip,buf,size_buf,&crcFile);
     422  
     423                  zip64 = isLargeFile(filenameinzip);
     424  
     425                                                           /* The path name saved, should not include a leading slash. */
     426                 /*if it did, windows/xp and dynazip couldn't read the zip file. */
     427                   savefilenameinzip = filenameinzip;
     428                   while( savefilenameinzip[0] == '\\' || savefilenameinzip[0] == '/' )
     429                   {
     430                       savefilenameinzip++;
     431                   }
     432  
     433                   /*should the zip file contain any path at all?*/
     434                   if( opt_exclude_path )
     435                   {
     436                       const char *tmpptr;
     437                       const char *lastslash = 0;
     438                       for( tmpptr = savefilenameinzip; *tmpptr; tmpptr++)
     439                       {
     440                           if( *tmpptr == '\\' || *tmpptr == '/')
     441                           {
     442                               lastslash = tmpptr;
     443                           }
     444                       }
     445                       if( lastslash != NULL )
     446                       {
     447                           savefilenameinzip = lastslash+1; // base filename follows last slash.
     448                       }
     449                   }
     450  
     451                   /**/
     452                  err = zipOpenNewFileInZip3_64(zf,savefilenameinzip,&zi,
     453                                   NULL,0,NULL,0,NULL /* comment*/,
     454                                   (opt_compress_level != 0) ? Z_DEFLATED : 0,
     455                                   opt_compress_level,0,
     456                                   /* -MAX_WBITS, DEF_MEM_LEVEL, Z_DEFAULT_STRATEGY, */
     457                                   -MAX_WBITS, DEF_MEM_LEVEL, Z_DEFAULT_STRATEGY,
     458                                   password,crcFile, zip64);
     459  
     460                  if (err != ZIP_OK)
     461                      printf("error in opening %s in zipfile\n",filenameinzip);
     462                  else
     463                  {
     464                      fin = FOPEN_FUNC(filenameinzip,"rb");
     465                      if (fin==NULL)
     466                      {
     467                          err=ZIP_ERRNO;
     468                          printf("error in opening %s for reading\n",filenameinzip);
     469                      }
     470                  }
     471  
     472                  if (err == ZIP_OK)
     473                      do
     474                      {
     475                          err = ZIP_OK;
     476                          size_read = fread(buf,1,size_buf,fin);
     477                          if (size_read < size_buf)
     478                              if (feof(fin)==0)
     479                          {
     480                              printf("error in reading %s\n",filenameinzip);
     481                              err = ZIP_ERRNO;
     482                          }
     483  
     484                          if (size_read>0)
     485                          {
     486                              err = zipWriteInFileInZip (zf,buf,(unsigned)size_read);
     487                              if (err<0)
     488                              {
     489                                  printf("error in writing %s in the zipfile\n",
     490                                                   filenameinzip);
     491                              }
     492  
     493                          }
     494                      } while ((err == ZIP_OK) && (size_read>0));
     495  
     496                  if (fin)
     497                      fclose(fin);
     498  
     499                  if (err<0)
     500                      err=ZIP_ERRNO;
     501                  else
     502                  {
     503                      err = zipCloseFileInZip(zf);
     504                      if (err!=ZIP_OK)
     505                          printf("error in closing %s in the zipfile\n",
     506                                      filenameinzip);
     507                  }
     508              }
     509          }
     510          errclose = zipClose(zf,NULL);
     511          if (errclose != ZIP_OK)
     512              printf("error in closing %s\n",filename_try);
     513      }
     514      else
     515      {
     516         do_help();
     517      }
     518  
     519      free(buf);
     520      return 0;
     521  }