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 38uLong 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 62uLong 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 104uLong 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 117int 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 130void 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 136void 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 */ 148int 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 186int 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} 421