contrib/minizip/minizip.c

   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 }