zlib/contrib/minizip/minizip.c

   1 /*
   2    minizip.c
   3    Version 1.01e, February 12th, 2005
   4
   5    Copyright (C) 1998-2005 Gilles Vollant
   6 */
   7
   8 #include <stdio.h>
   9 #include <stdlib.h>
  10 #include <string.h>
  11 #include <time.h>
  12 #include <errno.h>
  13 #include <fcntl.h>
  14
  15 #ifdef unix
  16 # include <unistd.h>
  17 # include <utime.h>
  18 # include <sys/types.h>
  19 # include <sys/stat.h>
  20 #else
  21 # include <direct.h>
  22 # include <io.h>
  23 #endif
  24
  25 #include "zip.h"
  26
  27 #ifdef WIN32
  28 #define USEWIN32IOAPI
  29 #include "iowin32.h"
  30 #endif
  31
  32
  33
  34 #define WRITEBUFFERSIZE (16384)
  35 #define MAXFILENAME (256)
  36
  37 #ifdef WIN32
  38 uLong filetime(f, tmzip, dt)
  39     char *f;                /* name of file to get info on */
  40     tm_zip *tmzip;             /* return value: access, modific. and creation times */
  41     uLong *dt;             /* dostime */
  42 {
  43   int ret = 0;
  44   {
  45       FILETIME ftLocal;
  46       HANDLE hFind;
  47       WIN32_FIND_DATA  ff32;
  48
  49       hFind = FindFirstFile(f,&ff32);
  50       if (hFind != INVALID_HANDLE_VALUE)
  51       {
  52         FileTimeToLocalFileTime(&(ff32.ftLastWriteTime),&ftLocal);
  53         FileTimeToDosDateTime(&ftLocal,((LPWORD)dt)+1,((LPWORD)dt)+0);
  54         FindClose(hFind);
  55         ret = 1;
  56       }
  57   }
  58   return ret;
  59 }
  60 #else
  61 #ifdef unix
  62 uLong filetime(f, tmzip, dt)
  63     char *f;               /* name of file to get info on */
  64     tm_zip *tmzip;         /* return value: access, modific. and creation times */
  65     uLong *dt;             /* dostime */
  66 {
  67   int ret=0;
  68   struct stat s;        /* results of stat() */
  69   struct tm* filedate;
  70   time_t tm_t=0;
  71
  72   if (strcmp(f,"-")!=0)
  73   {
  74     char name[MAXFILENAME+1];
  75     int len = strlen(f);
  76     if (len > MAXFILENAME)
  77       len = MAXFILENAME;
  78
  79     strncpy(name, f,MAXFILENAME-1);
  80     /* strncpy doesnt append the trailing NULL, of the string is too long. */
  81     name[ MAXFILENAME ] = '\0';
  82
  83     if (name[len - 1] == '/')
  84       name[len - 1] = '\0';
  85     /* not all systems allow stat'ing a file with / appended */
  86     if (stat(name,&s)==0)
  87     {
  88       tm_t = s.st_mtime;
  89       ret = 1;
  90     }
  91   }
  92   filedate = localtime(&tm_t);
  93
  94   tmzip->tm_sec  = filedate->tm_sec;
  95   tmzip->tm_min  = filedate->tm_min;
  96   tmzip->tm_hour = filedate->tm_hour;
  97   tmzip->tm_mday = filedate->tm_mday;
  98   tmzip->tm_mon  = filedate->tm_mon ;
  99   tmzip->tm_year = filedate->tm_year;
 100
 101   return ret;
 102 }
 103 #else
 104 uLong filetime(f, tmzip, dt)
 105     char *f;                /* name of file to get info on */
 106     tm_zip *tmzip;             /* return value: access, modific. and creation times */
 107     uLong *dt;             /* dostime */
 108 {
 109     return 0;
 110 }
 111 #endif
 112 #endif
 113
 114
 115
 116
 117 int check_exist_file(filename)
 118     const char* filename;
 119 {
 120     FILE* ftestexist;
 121     int ret = 1;
 122     ftestexist = fopen(filename,"rb");
 123     if (ftestexist==NULL)
 124         ret = 0;
 125     else
 126         fclose(ftestexist);
 127     return ret;
 128 }
 129
 130 void do_banner()
 131 {
 132     printf("MiniZip 1.01b, demo of zLib + Zip package written by Gilles Vollant\n");
 133     printf("more info at http://www.winimage.com/zLibDll/unzip.html\n\n");
 134 }
 135
 136 void do_help()
 137 {
 138     printf("Usage : minizip [-o] [-a] [-0 to -9] [-p password] file.zip [files_to_add]\n\n" \
 139            "  -o  Overwrite existing file.zip\n" \
 140            "  -a  Append to existing file.zip\n" \
 141            "  -0  Store only\n" \
 142            "  -1  Compress faster\n" \
 143            "  -9  Compress better\n\n");
 144 }
 145
 146 /* calculate the CRC32 of a file,
 147    because to encrypt a file, we need known the CRC32 of the file before */
 148 int getFileCrc(const char* filenameinzip,void*buf,unsigned long size_buf,unsigned long* result_crc)
 149 {
 150    unsigned long calculate_crc=0;
 151    int err=ZIP_OK;
 152    FILE * fin = fopen(filenameinzip,"rb");
 153    unsigned long size_read = 0;
 154    unsigned long total_read = 0;
 155    if (fin==NULL)
 156    {
 157        err = ZIP_ERRNO;
 158    }
 159
 160     if (err == ZIP_OK)
 161         do
 162         {
 163             err = ZIP_OK;
 164             size_read = (int)fread(buf,1,size_buf,fin);
 165             if (size_read < size_buf)
 166                 if (feof(fin)==0)
 167             {
 168                 printf("error in reading %s\n",filenameinzip);
 169                 err = ZIP_ERRNO;
 170             }
 171
 172             if (size_read>0)
 173                 calculate_crc = crc32(calculate_crc,buf,size_read);
 174             total_read += size_read;
 175
 176         } while ((err == ZIP_OK) && (size_read>0));
 177
 178     if (fin)
 179         fclose(fin);
 180
 181     *result_crc=calculate_crc;
 182     printf("file %s crc %x\n",filenameinzip,calculate_crc);
 183     return err;
 184 }
 185
 186 int main(argc,argv)
 187     int argc;
 188     char *argv[];
 189 {
 190     int i;
 191     int opt_overwrite=0;
 192     int opt_compress_level=Z_DEFAULT_COMPRESSION;
 193     int zipfilenamearg = 0;
 194     char filename_try[MAXFILENAME+16];
 195     int zipok;
 196     int err=0;
 197     int size_buf=0;
 198     void* buf=NULL;
 199     const char* password=NULL;
 200
 201
 202     do_banner();
 203     if (argc==1)
 204     {
 205         do_help();
 206         return 0;
 207     }
 208     else
 209     {
 210         for (i=1;i<argc;i++)
 211         {
 212             if ((*argv[i])=='-')
 213             {
 214                 const char *p=argv[i]+1;
 215
 216                 while ((*p)!='\0')
 217                 {
 218                     char c=*(p++);;
 219                     if ((c=='o') || (c=='O'))
 220                         opt_overwrite = 1;
 221                     if ((c=='a') || (c=='A'))
 222                         opt_overwrite = 2;
 223                     if ((c>='0') && (c<='9'))
 224                         opt_compress_level = c-'0';
 225
 226                     if (((c=='p') || (c=='P')) && (i+1<argc))
 227                     {
 228                         password=argv[i+1];
 229                         i++;
 230                     }
 231                 }
 232             }
 233             else
 234                 if (zipfilenamearg == 0)
 235                     zipfilenamearg = i ;
 236         }
 237     }
 238
 239     size_buf = WRITEBUFFERSIZE;
 240     buf = (void*)malloc(size_buf);
 241     if (buf==NULL)
 242     {
 243         printf("Error allocating memory\n");
 244         return ZIP_INTERNALERROR;
 245     }
 246
 247     if (zipfilenamearg==0)
 248         zipok=0;
 249     else
 250     {
 251         int i,len;
 252         int dot_found=0;
 253
 254         zipok = 1 ;
 255         strncpy(filename_try, argv[zipfilenamearg],MAXFILENAME-1);
 256         /* strncpy doesnt append the trailing NULL, of the string is too long. */
 257         filename_try[ MAXFILENAME ] = '\0';
 258
 259         len=(int)strlen(filename_try);
 260         for (i=0;i<len;i++)
 261             if (filename_try[i]=='.')
 262                 dot_found=1;
 263
 264         if (dot_found==0)
 265             strcat(filename_try,".zip");
 266
 267         if (opt_overwrite==2)
 268         {
 269             /* if the file don't exist, we not append file */
 270             if (check_exist_file(filename_try)==0)
 271                 opt_overwrite=1;
 272         }
 273         else
 274         if (opt_overwrite==0)
 275             if (check_exist_file(filename_try)!=0)
 276             {
 277                 char rep=0;
 278                 do
 279                 {
 280                     char answer[128];
 281                     int ret;
 282                     printf("The file %s exists. Overwrite ? [y]es, [n]o, [a]ppend : ",filename_try);
 283                     ret = scanf("%1s",answer);
 284                     if (ret != 1)
 285                     {
 286                        exit(EXIT_FAILURE);
 287                     }
 288                     rep = answer[0] ;
 289                     if ((rep>='a') && (rep<='z'))
 290                         rep -= 0x20;
 291                 }
 292                 while ((rep!='Y') && (rep!='N') && (rep!='A'));
 293                 if (rep=='N')
 294                     zipok = 0;
 295                 if (rep=='A')
 296                     opt_overwrite = 2;
 297             }
 298     }
 299
 300     if (zipok==1)
 301     {
 302         zipFile zf;
 303         int errclose;
 304 #        ifdef USEWIN32IOAPI
 305         zlib_filefunc_def ffunc;
 306         fill_win32_filefunc(&ffunc);
 307         zf = zipOpen2(filename_try,(opt_overwrite==2) ? 2 : 0,NULL,&ffunc);
 308 #        else
 309         zf = zipOpen(filename_try,(opt_overwrite==2) ? 2 : 0);
 310 #        endif
 311
 312         if (zf == NULL)
 313         {
 314             printf("error opening %s\n",filename_try);
 315             err= ZIP_ERRNO;
 316         }
 317         else
 318             printf("creating %s\n",filename_try);
 319
 320         for (i=zipfilenamearg+1;(i<argc) && (err==ZIP_OK);i++)
 321         {
 322             if (!((((*(argv[i]))=='-') || ((*(argv[i]))=='/')) &&
 323                   ((argv[i][1]=='o') || (argv[i][1]=='O') ||
 324                    (argv[i][1]=='a') || (argv[i][1]=='A') ||
 325                    (argv[i][1]=='p') || (argv[i][1]=='P') ||
 326                    ((argv[i][1]>='0') || (argv[i][1]<='9'))) &&
 327                   (strlen(argv[i]) == 2)))
 328             {
 329                 FILE * fin;
 330                 int size_read;
 331                 const char* filenameinzip = argv[i];
 332                 zip_fileinfo zi;
 333                 unsigned long crcFile=0;
 334
 335                 zi.tmz_date.tm_sec = zi.tmz_date.tm_min = zi.tmz_date.tm_hour =
 336                 zi.tmz_date.tm_mday = zi.tmz_date.tm_mon = zi.tmz_date.tm_year = 0;
 337                 zi.dosDate = 0;
 338                 zi.internal_fa = 0;
 339                 zi.external_fa = 0;
 340                 filetime(filenameinzip,&zi.tmz_date,&zi.dosDate);
 341
 342 /*
 343                 err = zipOpenNewFileInZip(zf,filenameinzip,&zi,
 344                                  NULL,0,NULL,0,NULL / * comment * /,
 345                                  (opt_compress_level != 0) ? Z_DEFLATED : 0,
 346                                  opt_compress_level);
 347 */
 348                 if ((password != NULL) && (err==ZIP_OK))
 349                     err = getFileCrc(filenameinzip,buf,size_buf,&crcFile);
 350
 351                 err = zipOpenNewFileInZip3(zf,filenameinzip,&zi,
 352                                  NULL,0,NULL,0,NULL /* comment*/,
 353                                  (opt_compress_level != 0) ? Z_DEFLATED : 0,
 354                                  opt_compress_level,0,
 355                                  /* -MAX_WBITS, DEF_MEM_LEVEL, Z_DEFAULT_STRATEGY, */
 356                                  -MAX_WBITS, DEF_MEM_LEVEL, Z_DEFAULT_STRATEGY,
 357                                  password,crcFile);
 358
 359                 if (err != ZIP_OK)
 360                     printf("error in opening %s in zipfile\n",filenameinzip);
 361                 else
 362                 {
 363                     fin = fopen(filenameinzip,"rb");
 364                     if (fin==NULL)
 365                     {
 366                         err=ZIP_ERRNO;
 367                         printf("error in opening %s for reading\n",filenameinzip);
 368                     }
 369                 }
 370
 371                 if (err == ZIP_OK)
 372                     do
 373                     {
 374                         err = ZIP_OK;
 375                         size_read = (int)fread(buf,1,size_buf,fin);
 376                         if (size_read < size_buf)
 377                             if (feof(fin)==0)
 378                         {
 379                             printf("error in reading %s\n",filenameinzip);
 380                             err = ZIP_ERRNO;
 381                         }
 382
 383                         if (size_read>0)
 384                         {
 385                             err = zipWriteInFileInZip (zf,buf,size_read);
 386                             if (err<0)
 387                             {
 388                                 printf("error in writing %s in the zipfile\n",
 389                                                  filenameinzip);
 390                             }
 391
 392                         }
 393                     } while ((err == ZIP_OK) && (size_read>0));
 394
 395                 if (fin)
 396                     fclose(fin);
 397
 398                 if (err<0)
 399                     err=ZIP_ERRNO;
 400                 else
 401                 {
 402                     err = zipCloseFileInZip(zf);
 403                     if (err!=ZIP_OK)
 404                         printf("error in closing %s in the zipfile\n",
 405                                     filenameinzip);
 406                 }
 407             }
 408         }
 409         errclose = zipClose(zf,NULL);
 410         if (errclose != ZIP_OK)
 411             printf("error in closing %s\n",filename_try);
 412     }
 413     else
 414     {
 415        do_help();
 416     }
 417
 418     free(buf);
 419     return 0;
 420 }