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