minizip.c
上传用户:rrhhcc
上传日期:2015-12-11
资源大小:54129k
文件大小:11k
源码类别:

通讯编程

开发平台:

Visual C++

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