Home | History | Annotate | Line # | Download | only in minizip
      1 /*
      2    miniunz.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 #if (!defined(_WIN32)) && (!defined(WIN32)) && (!defined(__APPLE__))
     16         #ifndef __USE_FILE_OFFSET64
     17                 #define __USE_FILE_OFFSET64
     18         #endif
     19         #ifndef __USE_LARGEFILE64
     20                 #define __USE_LARGEFILE64
     21         #endif
     22         #ifndef _LARGEFILE64_SOURCE
     23                 #define _LARGEFILE64_SOURCE
     24         #endif
     25         #ifndef _FILE_OFFSET_BIT
     26                 #define _FILE_OFFSET_BIT 64
     27         #endif
     28 #endif
     29 
     30 #if defined(__APPLE__) || defined(__HAIKU__) || defined(MINIZIP_FOPEN_NO_64)
     31 // In darwin and perhaps other BSD variants off_t is a 64 bit value, hence no need for specific 64 bit functions
     32 #define FOPEN_FUNC(filename, mode) fopen(filename, mode)
     33 #define FTELLO_FUNC(stream) ftello(stream)
     34 #define FSEEKO_FUNC(stream, offset, origin) fseeko(stream, offset, origin)
     35 #else
     36 #define FOPEN_FUNC(filename, mode) fopen64(filename, mode)
     37 #define FTELLO_FUNC(stream) ftello64(stream)
     38 #define FSEEKO_FUNC(stream, offset, origin) fseeko64(stream, offset, origin)
     39 #endif
     40 
     41 
     42 #include <stdio.h>
     43 #include <stdlib.h>
     44 #include <string.h>
     45 #include <time.h>
     46 #include <errno.h>
     47 #include <fcntl.h>
     48 #include <sys/stat.h>
     49 
     50 #ifdef _WIN32
     51 # include <direct.h>
     52 # include <io.h>
     53 #else
     54 # include <unistd.h>
     55 # include <utime.h>
     56 #endif
     57 
     58 
     59 #include "unzip.h"
     60 
     61 #define CASESENSITIVITY (0)
     62 #define WRITEBUFFERSIZE (8192)
     63 #define MAXFILENAME (256)
     64 
     65 #ifdef _WIN32
     66 #define USEWIN32IOAPI
     67 #include "iowin32.h"
     68 #endif
     69 /*
     70   mini unzip, demo of unzip package
     71 
     72   usage :
     73   Usage : miniunz [-exvlo] file.zip [file_to_extract] [-d extractdir]
     74 
     75   list the file in the zipfile, and print the content of FILE_ID.ZIP or README.TXT
     76     if it exists
     77 */
     78 
     79 
     80 /* change_file_date : change the date/time of a file
     81     filename : the filename of the file where date/time must be modified
     82     dosdate : the new date at the MSDOS format (4 bytes)
     83     tmu_date : the SAME new date at the tm_unz format */
     84 static void change_file_date(const char *filename, uLong dosdate, tm_unz tmu_date) {
     85 #ifdef _WIN32
     86   HANDLE hFile;
     87   FILETIME ftm,ftLocal,ftCreate,ftLastAcc,ftLastWrite;
     88 
     89   hFile = CreateFileA(filename,GENERIC_READ | GENERIC_WRITE,
     90                       0,NULL,OPEN_EXISTING,0,NULL);
     91   GetFileTime(hFile,&ftCreate,&ftLastAcc,&ftLastWrite);
     92   DosDateTimeToFileTime((WORD)(dosdate>>16),(WORD)dosdate,&ftLocal);
     93   LocalFileTimeToFileTime(&ftLocal,&ftm);
     94   SetFileTime(hFile,&ftm,&ftLastAcc,&ftm);
     95   CloseHandle(hFile);
     96 #else
     97 #if defined(unix) || defined(__APPLE__)
     98   (void)dosdate;
     99   struct utimbuf ut;
    100   struct tm newdate;
    101   newdate.tm_sec = tmu_date.tm_sec;
    102   newdate.tm_min=tmu_date.tm_min;
    103   newdate.tm_hour=tmu_date.tm_hour;
    104   newdate.tm_mday=tmu_date.tm_mday;
    105   newdate.tm_mon=tmu_date.tm_mon;
    106   if (tmu_date.tm_year > 1900)
    107       newdate.tm_year=tmu_date.tm_year - 1900;
    108   else
    109       newdate.tm_year=tmu_date.tm_year ;
    110   newdate.tm_isdst=-1;
    111 
    112   ut.actime=ut.modtime=mktime(&newdate);
    113   utime(filename,&ut);
    114 #else
    115   (void)filename;
    116   (void)dosdate;
    117   (void)tmu_date;
    118 #endif
    119 #endif
    120 }
    121 
    122 
    123 /* mymkdir and change_file_date are not 100 % portable
    124    As I don't know well Unix, I wait feedback for the unix portion */
    125 
    126 static int mymkdir(const char* dirname) {
    127     int ret=0;
    128 #ifdef _WIN32
    129     ret = _mkdir(dirname);
    130 #elif unix
    131     ret = mkdir (dirname,0775);
    132 #elif __APPLE__
    133     ret = mkdir (dirname,0775);
    134 #else
    135     (void)dirname;
    136 #endif
    137     return ret;
    138 }
    139 
    140 static int makedir(const char *newdir) {
    141   char *buffer ;
    142   char *p;
    143   size_t len = strlen(newdir);
    144 
    145   if (len == 0)
    146     return 0;
    147 
    148   buffer = (char*)malloc(len+1);
    149         if (buffer==NULL)
    150         {
    151                 printf("Error allocating memory\n");
    152                 return UNZ_INTERNALERROR;
    153         }
    154   strcpy(buffer,newdir);
    155 
    156   if (buffer[len-1] == '/') {
    157     buffer[len-1] = '\0';
    158   }
    159   if (mymkdir(buffer) == 0)
    160     {
    161       free(buffer);
    162       return 1;
    163     }
    164 
    165   p = buffer+1;
    166   while (1)
    167     {
    168       char hold;
    169 
    170       while(*p && *p != '\\' && *p != '/')
    171         p++;
    172       hold = *p;
    173       *p = 0;
    174       if ((mymkdir(buffer) == -1) && (errno == ENOENT))
    175         {
    176           printf("couldn't create directory %s\n",buffer);
    177           free(buffer);
    178           return 0;
    179         }
    180       if (hold == 0)
    181         break;
    182       *p++ = hold;
    183     }
    184   free(buffer);
    185   return 1;
    186 }
    187 
    188 static void do_banner(void) {
    189     printf("MiniUnz 1.1, demo of zLib + Unz package written by Gilles Vollant\n");
    190     printf("more info at http://www.winimage.com/zLibDll/unzip.html\n\n");
    191 }
    192 
    193 static void do_help(void) {
    194     printf("Usage : miniunz [-e] [-x] [-v] [-l] [-o] [-p password] file.zip [file_to_extr.] [-d extractdir]\n\n" \
    195            "  -e  Extract without pathname (junk paths)\n" \
    196            "  -x  Extract with pathname\n" \
    197            "  -v  list files\n" \
    198            "  -l  list files\n" \
    199            "  -d  directory to extract into\n" \
    200            "  -o  overwrite files without prompting\n" \
    201            "  -p  extract encrypted file using password\n\n");
    202 }
    203 
    204 static void Display64BitsSize(ZPOS64_T n, int size_char) {
    205   /* to avoid compatibility problem , we do here the conversion */
    206   char number[21];
    207   int offset=19;
    208   int pos_string = 19;
    209   number[20]=0;
    210   for (;;) {
    211       number[offset]=(char)((n%10)+'0');
    212       if (number[offset] != '0')
    213           pos_string=offset;
    214       n/=10;
    215       if (offset==0)
    216           break;
    217       offset--;
    218   }
    219   {
    220       int size_display_string = 19-pos_string;
    221       while (size_char > size_display_string)
    222       {
    223           size_char--;
    224           printf(" ");
    225       }
    226   }
    227 
    228   printf("%s",&number[pos_string]);
    229 }
    230 
    231 static int do_list(unzFile uf) {
    232     uLong i;
    233     unz_global_info64 gi;
    234     int err;
    235 
    236     err = unzGetGlobalInfo64(uf,&gi);
    237     if (err!=UNZ_OK)
    238         printf("error %d with zipfile in unzGetGlobalInfo \n",err);
    239     printf("  Length  Method     Size Ratio   Date    Time   CRC-32     Name\n");
    240     printf("  ------  ------     ---- -----   ----    ----   ------     ----\n");
    241     for (i=0;i<gi.number_entry;i++)
    242     {
    243         char filename_inzip[256];
    244         unz_file_info64 file_info;
    245         uLong ratio=0;
    246         const char *string_method = "";
    247         char charCrypt=' ';
    248         err = unzGetCurrentFileInfo64(uf,&file_info,filename_inzip,sizeof(filename_inzip),NULL,0,NULL,0);
    249         if (err!=UNZ_OK)
    250         {
    251             printf("error %d with zipfile in unzGetCurrentFileInfo\n",err);
    252             break;
    253         }
    254         if (file_info.uncompressed_size>0)
    255             ratio = (uLong)((file_info.compressed_size*100)/file_info.uncompressed_size);
    256 
    257         /* display a '*' if the file is encrypted */
    258         if ((file_info.flag & 1) != 0)
    259             charCrypt='*';
    260 
    261         if (file_info.compression_method==0)
    262             string_method="Stored";
    263         else
    264         if (file_info.compression_method==Z_DEFLATED)
    265         {
    266             uInt iLevel=(uInt)((file_info.flag & 0x6)/2);
    267             if (iLevel==0)
    268               string_method="Defl:N";
    269             else if (iLevel==1)
    270               string_method="Defl:X";
    271             else if ((iLevel==2) || (iLevel==3))
    272               string_method="Defl:F"; /* 2:fast , 3 : extra fast*/
    273         }
    274         else
    275         if (file_info.compression_method==Z_BZIP2ED)
    276         {
    277               string_method="BZip2 ";
    278         }
    279         else
    280             string_method="Unkn. ";
    281 
    282         Display64BitsSize(file_info.uncompressed_size,7);
    283         printf("  %6s%c",string_method,charCrypt);
    284         Display64BitsSize(file_info.compressed_size,7);
    285         printf(" %3lu%%  %2.2lu-%2.2lu-%2.2lu  %2.2lu:%2.2lu  %8.8lx   %s\n",
    286                 ratio,
    287                 (uLong)file_info.tmu_date.tm_mon + 1,
    288                 (uLong)file_info.tmu_date.tm_mday,
    289                 (uLong)file_info.tmu_date.tm_year % 100,
    290                 (uLong)file_info.tmu_date.tm_hour,(uLong)file_info.tmu_date.tm_min,
    291                 (uLong)file_info.crc,filename_inzip);
    292         if ((i+1)<gi.number_entry)
    293         {
    294             err = unzGoToNextFile(uf);
    295             if (err!=UNZ_OK)
    296             {
    297                 printf("error %d with zipfile in unzGoToNextFile\n",err);
    298                 break;
    299             }
    300         }
    301     }
    302 
    303     return 0;
    304 }
    305 
    306 
    307 static int do_extract_currentfile(unzFile uf, const int* popt_extract_without_path, int* popt_overwrite, const char* password) {
    308     char filename_inzip[256];
    309     char* filename_withoutpath;
    310     char* p;
    311     int err=UNZ_OK;
    312     FILE *fout=NULL;
    313     void* buf;
    314     uInt size_buf;
    315 
    316     unz_file_info64 file_info;
    317     err = unzGetCurrentFileInfo64(uf,&file_info,filename_inzip,sizeof(filename_inzip),NULL,0,NULL,0);
    318 
    319     if (err!=UNZ_OK)
    320     {
    321         printf("error %d with zipfile in unzGetCurrentFileInfo\n",err);
    322         return err;
    323     }
    324 
    325     size_buf = WRITEBUFFERSIZE;
    326     buf = (void*)malloc(size_buf);
    327     if (buf==NULL)
    328     {
    329         printf("Error allocating memory\n");
    330         return UNZ_INTERNALERROR;
    331     }
    332 
    333     p = filename_withoutpath = filename_inzip;
    334     while ((*p) != '\0')
    335     {
    336         if (((*p)=='/') || ((*p)=='\\'))
    337             filename_withoutpath = p+1;
    338         p++;
    339     }
    340 
    341     if ((*filename_withoutpath)=='\0')
    342     {
    343         if ((*popt_extract_without_path)==0)
    344         {
    345             printf("creating directory: %s\n",filename_inzip);
    346             mymkdir(filename_inzip);
    347         }
    348     }
    349     else
    350     {
    351         const char* write_filename;
    352         int skip=0;
    353 
    354         if ((*popt_extract_without_path)==0)
    355             write_filename = filename_inzip;
    356         else
    357             write_filename = filename_withoutpath;
    358 
    359         if (write_filename[0]!='\0')
    360         {
    361             const char* relative_check = write_filename;
    362             while (relative_check[1]!='\0')
    363             {
    364                 if (relative_check[0]=='.' && relative_check[1]=='.')
    365                     write_filename = relative_check;
    366                 relative_check++;
    367             }
    368         }
    369 
    370         while (write_filename[0]=='/' || write_filename[0]=='.')
    371             write_filename++;
    372 
    373         err = unzOpenCurrentFilePassword(uf,password);
    374         if (err!=UNZ_OK)
    375         {
    376             printf("error %d with zipfile in unzOpenCurrentFilePassword\n",err);
    377         }
    378 
    379         if (((*popt_overwrite)==0) && (err==UNZ_OK))
    380         {
    381             char rep=0;
    382             FILE* ftestexist;
    383             ftestexist = FOPEN_FUNC(write_filename,"rb");
    384             if (ftestexist!=NULL)
    385             {
    386                 fclose(ftestexist);
    387                 do
    388                 {
    389                     char answer[128];
    390                     int ret;
    391 
    392                     printf("The file %s exists. Overwrite ? [y]es, [n]o, [A]ll: ",write_filename);
    393                     ret = scanf("%1s",answer);
    394                     if (ret != 1)
    395                     {
    396                        exit(EXIT_FAILURE);
    397                     }
    398                     rep = answer[0] ;
    399                     if ((rep>='a') && (rep<='z'))
    400                         rep -= 0x20;
    401                 }
    402                 while ((rep!='Y') && (rep!='N') && (rep!='A'));
    403             }
    404 
    405             if (rep == 'N')
    406                 skip = 1;
    407 
    408             if (rep == 'A')
    409                 *popt_overwrite=1;
    410         }
    411 
    412         if ((skip==0) && (err==UNZ_OK))
    413         {
    414             fout=FOPEN_FUNC(write_filename,"wb");
    415             /* some zipfile don't contain directory alone before file */
    416             if ((fout==NULL) && ((*popt_extract_without_path)==0) &&
    417                                 (filename_withoutpath!=(char*)filename_inzip))
    418             {
    419                 char c=*(filename_withoutpath-1);
    420                 *(filename_withoutpath-1)='\0';
    421                 makedir(write_filename);
    422                 *(filename_withoutpath-1)=c;
    423                 fout=FOPEN_FUNC(write_filename,"wb");
    424             }
    425 
    426             if (fout==NULL)
    427             {
    428                 printf("error opening %s\n",write_filename);
    429             }
    430         }
    431 
    432         if (fout!=NULL)
    433         {
    434             printf(" extracting: %s\n",write_filename);
    435 
    436             do
    437             {
    438                 err = unzReadCurrentFile(uf,buf,size_buf);
    439                 if (err<0)
    440                 {
    441                     printf("error %d with zipfile in unzReadCurrentFile\n",err);
    442                     break;
    443                 }
    444                 if (err>0)
    445                     if (fwrite(buf,(unsigned)err,1,fout)!=1)
    446                     {
    447                         printf("error in writing extracted file\n");
    448                         err=UNZ_ERRNO;
    449                         break;
    450                     }
    451             }
    452             while (err>0);
    453             if (fout)
    454                     fclose(fout);
    455 
    456             if (err==0)
    457                 change_file_date(write_filename,file_info.dosDate,
    458                                  file_info.tmu_date);
    459         }
    460 
    461         if (err==UNZ_OK)
    462         {
    463             err = unzCloseCurrentFile (uf);
    464             if (err!=UNZ_OK)
    465             {
    466                 printf("error %d with zipfile in unzCloseCurrentFile\n",err);
    467             }
    468         }
    469         else
    470             unzCloseCurrentFile(uf); /* don't lose the error */
    471     }
    472 
    473     free(buf);
    474     return err;
    475 }
    476 
    477 
    478 static int do_extract(unzFile uf, int opt_extract_without_path, int opt_overwrite, const char* password) {
    479     uLong i;
    480     unz_global_info64 gi;
    481     int err;
    482 
    483     err = unzGetGlobalInfo64(uf,&gi);
    484     if (err!=UNZ_OK)
    485         printf("error %d with zipfile in unzGetGlobalInfo \n",err);
    486 
    487     for (i=0;i<gi.number_entry;i++)
    488     {
    489         if (do_extract_currentfile(uf,&opt_extract_without_path,
    490                                       &opt_overwrite,
    491                                       password) != UNZ_OK)
    492             break;
    493 
    494         if ((i+1)<gi.number_entry)
    495         {
    496             err = unzGoToNextFile(uf);
    497             if (err!=UNZ_OK)
    498             {
    499                 printf("error %d with zipfile in unzGoToNextFile\n",err);
    500                 break;
    501             }
    502         }
    503     }
    504 
    505     return 0;
    506 }
    507 
    508 static int do_extract_onefile(unzFile uf, const char* filename, int opt_extract_without_path, int opt_overwrite, const char* password) {
    509     if (unzLocateFile(uf,filename,CASESENSITIVITY)!=UNZ_OK)
    510     {
    511         printf("file %s not found in the zipfile\n",filename);
    512         return 2;
    513     }
    514 
    515     if (do_extract_currentfile(uf,&opt_extract_without_path,
    516                                       &opt_overwrite,
    517                                       password) == UNZ_OK)
    518         return 0;
    519     else
    520         return 1;
    521 }
    522 
    523 
    524 int main(int argc, char *argv[]) {
    525     const char *zipfilename=NULL;
    526     const char *filename_to_extract=NULL;
    527     const char *password=NULL;
    528     char filename_try[MAXFILENAME+16] = "";
    529     int i;
    530     int ret_value=0;
    531     int opt_do_list=0;
    532     int opt_do_extract=1;
    533     int opt_do_extract_withoutpath=0;
    534     int opt_overwrite=0;
    535     int opt_extractdir=0;
    536     const char *dirname=NULL;
    537     unzFile uf=NULL;
    538 
    539     do_banner();
    540     if (argc==1)
    541     {
    542         do_help();
    543         return 0;
    544     }
    545     else
    546     {
    547         for (i=1;i<argc;i++)
    548         {
    549             if ((*argv[i])=='-')
    550             {
    551                 const char *p=argv[i]+1;
    552 
    553                 while ((*p)!='\0')
    554                 {
    555                     char c=*(p++);
    556                     if ((c=='l') || (c=='L'))
    557                         opt_do_list = 1;
    558                     if ((c=='v') || (c=='V'))
    559                         opt_do_list = 1;
    560                     if ((c=='x') || (c=='X'))
    561                         opt_do_extract = 1;
    562                     if ((c=='e') || (c=='E'))
    563                         opt_do_extract = opt_do_extract_withoutpath = 1;
    564                     if ((c=='o') || (c=='O'))
    565                         opt_overwrite=1;
    566                     if ((c=='d') || (c=='D'))
    567                     {
    568                         opt_extractdir=1;
    569                         dirname=argv[i+1];
    570                     }
    571 
    572                     if (((c=='p') || (c=='P')) && (i+1<argc))
    573                     {
    574                         password=argv[i+1];
    575                         i++;
    576                     }
    577                 }
    578             }
    579             else
    580             {
    581                 if (zipfilename == NULL)
    582                     zipfilename = argv[i];
    583                 else if ((filename_to_extract==NULL) && (!opt_extractdir))
    584                         filename_to_extract = argv[i] ;
    585             }
    586         }
    587     }
    588 
    589     if (zipfilename!=NULL)
    590     {
    591 
    592 #        ifdef USEWIN32IOAPI
    593         zlib_filefunc64_def ffunc;
    594 #        endif
    595 
    596         strncpy(filename_try, zipfilename,MAXFILENAME-1);
    597         /* strncpy doesn't append the trailing NULL, of the string is too long. */
    598         filename_try[ MAXFILENAME ] = '\0';
    599 
    600 #        ifdef USEWIN32IOAPI
    601         fill_win32_filefunc64A(&ffunc);
    602         uf = unzOpen2_64(zipfilename,&ffunc);
    603 #        else
    604         uf = unzOpen64(zipfilename);
    605 #        endif
    606         if (uf==NULL)
    607         {
    608             strcat(filename_try,".zip");
    609 #            ifdef USEWIN32IOAPI
    610             uf = unzOpen2_64(filename_try,&ffunc);
    611 #            else
    612             uf = unzOpen64(filename_try);
    613 #            endif
    614         }
    615     }
    616 
    617     if (uf==NULL)
    618     {
    619         printf("Cannot open %s or %s.zip\n",zipfilename,zipfilename);
    620         return 1;
    621     }
    622     printf("%s opened\n",filename_try);
    623 
    624     if (opt_do_list==1)
    625         ret_value = do_list(uf);
    626     else if (opt_do_extract==1)
    627     {
    628 #ifdef _WIN32
    629         if (opt_extractdir && _chdir(dirname))
    630 #else
    631         if (opt_extractdir && chdir(dirname))
    632 #endif
    633         {
    634           printf("Error changing into %s, aborting\n", dirname);
    635           exit(-1);
    636         }
    637 
    638         if (filename_to_extract == NULL)
    639             ret_value = do_extract(uf, opt_do_extract_withoutpath, opt_overwrite, password);
    640         else
    641             ret_value = do_extract_onefile(uf, filename_to_extract, opt_do_extract_withoutpath, opt_overwrite, password);
    642     }
    643 
    644     unzClose(uf);
    645 
    646     return ret_value;
    647 }
    648