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
61uLong 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
85uLong 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
127uLong 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
140int 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
153void 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
159void 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 */
172int 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
210int 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
233int 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}
508