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