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 #if (!defined(_WIN32)) && (!defined(WIN32)) && (!defined(__APPLE__)) 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 #if defined(__APPLE__) || defined(__HAIKU__) || defined(MINIZIP_FOPEN_NO_64) 32 // In darwin and perhaps other BSD variants off_t is a 64 bit value, hence no need for specific 64 bit functions 33 #define FOPEN_FUNC(filename, mode) fopen(filename, mode) 34 #define FTELLO_FUNC(stream) ftello(stream) 35 #define FSEEKO_FUNC(stream, offset, origin) fseeko(stream, offset, origin) 36 #else 37 #define FOPEN_FUNC(filename, mode) fopen64(filename, mode) 38 #define FTELLO_FUNC(stream) ftello64(stream) 39 #define FSEEKO_FUNC(stream, offset, origin) fseeko64(stream, offset, origin) 40 #endif 41 42 43 44 #include <stdio.h> 45 #include <stdlib.h> 46 #include <string.h> 47 #include <time.h> 48 #include <errno.h> 49 #include <fcntl.h> 50 51 #ifdef _WIN32 52 # include <direct.h> 53 # include <io.h> 54 #else 55 # include <unistd.h> 56 # include <utime.h> 57 # include <sys/types.h> 58 # include <sys/stat.h> 59 #endif 60 61 #include "zip.h" 62 63 #ifdef _WIN32 64 #define USEWIN32IOAPI 65 #include "iowin32.h" 66 #endif 67 68 69 70 #define WRITEBUFFERSIZE (16384) 71 #define MAXFILENAME (256) 72 73 #ifdef _WIN32 74 /* f: name of file to get info on, tmzip: return value: access, 75 modification and creation times, dt: dostime */ 76 static int filetime(const char *f, tm_zip *tmzip, uLong *dt) { 77 int ret = 0; 78 { 79 FILETIME ftLocal; 80 HANDLE hFind; 81 WIN32_FIND_DATAA ff32; 82 83 hFind = FindFirstFileA(f,&ff32); 84 if (hFind != INVALID_HANDLE_VALUE) 85 { 86 FileTimeToLocalFileTime(&(ff32.ftLastWriteTime),&ftLocal); 87 FileTimeToDosDateTime(&ftLocal,((LPWORD)dt)+1,((LPWORD)dt)+0); 88 FindClose(hFind); 89 ret = 1; 90 } 91 } 92 return ret; 93 } 94 #else 95 #if defined(unix) || defined(__APPLE__) 96 /* f: name of file to get info on, tmzip: return value: access, 97 modification and creation times, dt: dostime */ 98 static int filetime(const char *f, tm_zip *tmzip, uLong *dt) { 99 (void)dt; 100 int ret=0; 101 struct stat s; /* results of stat() */ 102 struct tm* filedate; 103 time_t tm_t=0; 104 105 if (strcmp(f,"-")!=0) 106 { 107 char name[MAXFILENAME+1]; 108 size_t len = strlen(f); 109 if (len > MAXFILENAME) 110 len = MAXFILENAME; 111 112 strncpy(name, f,MAXFILENAME-1); 113 /* strncpy doesn't append the trailing NULL, of the string is too long. */ 114 name[ MAXFILENAME ] = '\0'; 115 116 if (name[len - 1] == '/') 117 name[len - 1] = '\0'; 118 /* not all systems allow stat'ing a file with / appended */ 119 if (stat(name,&s)==0) 120 { 121 tm_t = s.st_mtime; 122 ret = 1; 123 } 124 } 125 filedate = localtime(&tm_t); 126 127 tmzip->tm_sec = filedate->tm_sec; 128 tmzip->tm_min = filedate->tm_min; 129 tmzip->tm_hour = filedate->tm_hour; 130 tmzip->tm_mday = filedate->tm_mday; 131 tmzip->tm_mon = filedate->tm_mon ; 132 tmzip->tm_year = filedate->tm_year; 133 134 return ret; 135 } 136 #else 137 /* f: name of file to get info on, tmzip: return value: access, 138 modification and creation times, dt: dostime */ 139 static int filetime(const char *f, tm_zip *tmzip, uLong *dt) { 140 (void)f; 141 (void)tmzip; 142 (void)dt; 143 return 0; 144 } 145 #endif 146 #endif 147 148 149 150 151 static int check_exist_file(const char* filename) { 152 FILE* ftestexist; 153 int ret = 1; 154 ftestexist = FOPEN_FUNC(filename,"rb"); 155 if (ftestexist==NULL) 156 ret = 0; 157 else 158 fclose(ftestexist); 159 return ret; 160 } 161 162 static void do_banner(void) { 163 printf("MiniZip 1.1, demo of zLib + MiniZip64 package, written by Gilles Vollant\n"); 164 printf("more info on MiniZip at http://www.winimage.com/zLibDll/minizip.html\n\n"); 165 } 166 167 static void do_help(void) { 168 printf("Usage : minizip [-o] [-a] [-0 to -9] [-p password] [-j] file.zip [files_to_add]\n\n" \ 169 " -o Overwrite existing file.zip\n" \ 170 " -a Append to existing file.zip\n" \ 171 " -0 Store only\n" \ 172 " -1 Compress faster\n" \ 173 " -9 Compress better\n\n" \ 174 " -j exclude path. store only the file name.\n\n"); 175 } 176 177 /* calculate the CRC32 of a file, 178 because to encrypt a file, we need known the CRC32 of the file before */ 179 static int getFileCrc(const char* filenameinzip, void* buf, unsigned long size_buf, unsigned long* result_crc) { 180 unsigned long calculate_crc=0; 181 int err=ZIP_OK; 182 FILE * fin = FOPEN_FUNC(filenameinzip,"rb"); 183 184 unsigned long size_read = 0; 185 /* unsigned long total_read = 0; */ 186 if (fin==NULL) 187 { 188 err = ZIP_ERRNO; 189 } 190 191 if (err == ZIP_OK) 192 do 193 { 194 err = ZIP_OK; 195 size_read = fread(buf,1,size_buf,fin); 196 if (size_read < size_buf) 197 if (feof(fin)==0) 198 { 199 printf("error in reading %s\n",filenameinzip); 200 err = ZIP_ERRNO; 201 } 202 203 if (size_read>0) 204 calculate_crc = crc32_z(calculate_crc,buf,size_read); 205 /* total_read += size_read; */ 206 207 } while ((err == ZIP_OK) && (size_read>0)); 208 209 if (fin) 210 fclose(fin); 211 212 *result_crc=calculate_crc; 213 printf("file %s crc %lx\n", filenameinzip, calculate_crc); 214 return err; 215 } 216 217 static int isLargeFile(const char* filename) { 218 int largeFile = 0; 219 ZPOS64_T pos = 0; 220 FILE* pFile = FOPEN_FUNC(filename, "rb"); 221 222 if(pFile != NULL) 223 { 224 FSEEKO_FUNC(pFile, 0, SEEK_END); 225 pos = (ZPOS64_T)FTELLO_FUNC(pFile); 226 227 printf("File : %s is %llu bytes\n", filename, pos); 228 229 if(pos >= 0xffffffff) 230 largeFile = 1; 231 232 fclose(pFile); 233 } 234 235 return largeFile; 236 } 237 238 int main(int argc, char *argv[]) { 239 int i; 240 int opt_overwrite=0; 241 int opt_compress_level=Z_DEFAULT_COMPRESSION; 242 int opt_exclude_path=0; 243 int zipfilenamearg = 0; 244 char filename_try[MAXFILENAME+16]; 245 int zipok; 246 int err=0; 247 size_t size_buf=0; 248 void* buf=NULL; 249 const char* password=NULL; 250 251 252 do_banner(); 253 if (argc==1) 254 { 255 do_help(); 256 return 0; 257 } 258 else 259 { 260 for (i=1;i<argc;i++) 261 { 262 if ((*argv[i])=='-') 263 { 264 const char *p=argv[i]+1; 265 266 while ((*p)!='\0') 267 { 268 char c=*(p++); 269 if ((c=='o') || (c=='O')) 270 opt_overwrite = 1; 271 if ((c=='a') || (c=='A')) 272 opt_overwrite = 2; 273 if ((c>='0') && (c<='9')) 274 opt_compress_level = c-'0'; 275 if ((c=='j') || (c=='J')) 276 opt_exclude_path = 1; 277 278 if (((c=='p') || (c=='P')) && (i+1<argc)) 279 { 280 password=argv[i+1]; 281 i++; 282 } 283 } 284 } 285 else 286 { 287 if (zipfilenamearg == 0) 288 { 289 zipfilenamearg = i ; 290 } 291 } 292 } 293 } 294 295 size_buf = WRITEBUFFERSIZE; 296 buf = (void*)malloc(size_buf); 297 if (buf==NULL) 298 { 299 printf("Error allocating memory\n"); 300 return ZIP_INTERNALERROR; 301 } 302 303 if (zipfilenamearg==0) 304 { 305 zipok=0; 306 } 307 else 308 { 309 int i,len; 310 int dot_found=0; 311 312 zipok = 1 ; 313 strncpy(filename_try, argv[zipfilenamearg],MAXFILENAME-1); 314 /* strncpy doesn't append the trailing NULL, of the string is too long. */ 315 filename_try[ MAXFILENAME ] = '\0'; 316 317 len=(int)strlen(filename_try); 318 for (i=0;i<len;i++) 319 if (filename_try[i]=='.') 320 dot_found=1; 321 322 if (dot_found==0) 323 strcat(filename_try,".zip"); 324 325 if (opt_overwrite==2) 326 { 327 /* if the file don't exist, we not append file */ 328 if (check_exist_file(filename_try)==0) 329 opt_overwrite=1; 330 } 331 else 332 if (opt_overwrite==0) 333 if (check_exist_file(filename_try)!=0) 334 { 335 char rep=0; 336 do 337 { 338 char answer[128]; 339 int ret; 340 printf("The file %s exists. Overwrite ? [y]es, [n]o, [a]ppend : ",filename_try); 341 ret = scanf("%1s",answer); 342 if (ret != 1) 343 { 344 exit(EXIT_FAILURE); 345 } 346 rep = answer[0] ; 347 if ((rep>='a') && (rep<='z')) 348 rep -= 0x20; 349 } 350 while ((rep!='Y') && (rep!='N') && (rep!='A')); 351 if (rep=='N') 352 zipok = 0; 353 if (rep=='A') 354 opt_overwrite = 2; 355 } 356 } 357 358 if (zipok==1) 359 { 360 zipFile zf; 361 int errclose; 362 # ifdef USEWIN32IOAPI 363 zlib_filefunc64_def ffunc; 364 fill_win32_filefunc64A(&ffunc); 365 zf = zipOpen2_64(filename_try,(opt_overwrite==2) ? 2 : 0,NULL,&ffunc); 366 # else 367 zf = zipOpen64(filename_try,(opt_overwrite==2) ? 2 : 0); 368 # endif 369 370 if (zf == NULL) 371 { 372 printf("error opening %s\n",filename_try); 373 err= ZIP_ERRNO; 374 } 375 else 376 printf("creating %s\n",filename_try); 377 378 for (i=zipfilenamearg+1;(i<argc) && (err==ZIP_OK);i++) 379 { 380 if (!((((*(argv[i]))=='-') || ((*(argv[i]))=='/')) && 381 ((argv[i][1]=='o') || (argv[i][1]=='O') || 382 (argv[i][1]=='a') || (argv[i][1]=='A') || 383 (argv[i][1]=='p') || (argv[i][1]=='P') || 384 ((argv[i][1]>='0') && (argv[i][1]<='9'))) && 385 (strlen(argv[i]) == 2))) 386 { 387 FILE * fin = NULL; 388 size_t size_read; 389 const char* filenameinzip = argv[i]; 390 const char *savefilenameinzip; 391 zip_fileinfo zi; 392 unsigned long crcFile=0; 393 int zip64 = 0; 394 395 zi.tmz_date.tm_sec = zi.tmz_date.tm_min = zi.tmz_date.tm_hour = 396 zi.tmz_date.tm_mday = zi.tmz_date.tm_mon = zi.tmz_date.tm_year = 0; 397 zi.dosDate = 0; 398 zi.internal_fa = 0; 399 zi.external_fa = 0; 400 filetime(filenameinzip,&zi.tmz_date,&zi.dosDate); 401 402 /* 403 err = zipOpenNewFileInZip(zf,filenameinzip,&zi, 404 NULL,0,NULL,0,NULL / * comment * /, 405 (opt_compress_level != 0) ? Z_DEFLATED : 0, 406 opt_compress_level); 407 */ 408 if ((password != NULL) && (err==ZIP_OK)) 409 err = getFileCrc(filenameinzip,buf,size_buf,&crcFile); 410 411 zip64 = isLargeFile(filenameinzip); 412 413 /* The path name saved, should not include a leading slash. */ 414 /*if it did, windows/xp and dynazip couldn't read the zip file. */ 415 savefilenameinzip = filenameinzip; 416 while( savefilenameinzip[0] == '\\' || savefilenameinzip[0] == '/' ) 417 { 418 savefilenameinzip++; 419 } 420 421 /*should the zip file contain any path at all?*/ 422 if( opt_exclude_path ) 423 { 424 const char *tmpptr; 425 const char *lastslash = 0; 426 for( tmpptr = savefilenameinzip; *tmpptr; tmpptr++) 427 { 428 if( *tmpptr == '\\' || *tmpptr == '/') 429 { 430 lastslash = tmpptr; 431 } 432 } 433 if( lastslash != NULL ) 434 { 435 savefilenameinzip = lastslash+1; // base filename follows last slash. 436 } 437 } 438 439 /**/ 440 err = zipOpenNewFileInZip3_64(zf,savefilenameinzip,&zi, 441 NULL,0,NULL,0,NULL /* comment*/, 442 (opt_compress_level != 0) ? Z_DEFLATED : 0, 443 opt_compress_level,0, 444 /* -MAX_WBITS, DEF_MEM_LEVEL, Z_DEFAULT_STRATEGY, */ 445 -MAX_WBITS, DEF_MEM_LEVEL, Z_DEFAULT_STRATEGY, 446 password,crcFile, zip64); 447 448 if (err != ZIP_OK) 449 printf("error in opening %s in zipfile\n",filenameinzip); 450 else 451 { 452 fin = FOPEN_FUNC(filenameinzip,"rb"); 453 if (fin==NULL) 454 { 455 err=ZIP_ERRNO; 456 printf("error in opening %s for reading\n",filenameinzip); 457 } 458 } 459 460 if (err == ZIP_OK) 461 do 462 { 463 err = ZIP_OK; 464 size_read = fread(buf,1,size_buf,fin); 465 if (size_read < size_buf) 466 if (feof(fin)==0) 467 { 468 printf("error in reading %s\n",filenameinzip); 469 err = ZIP_ERRNO; 470 } 471 472 if (size_read>0) 473 { 474 err = zipWriteInFileInZip (zf,buf,(unsigned)size_read); 475 if (err<0) 476 { 477 printf("error in writing %s in the zipfile\n", 478 filenameinzip); 479 } 480 481 } 482 } while ((err == ZIP_OK) && (size_read>0)); 483 484 if (fin) 485 fclose(fin); 486 487 if (err<0) 488 err=ZIP_ERRNO; 489 else 490 { 491 err = zipCloseFileInZip(zf); 492 if (err!=ZIP_OK) 493 printf("error in closing %s in the zipfile\n", 494 filenameinzip); 495 } 496 } 497 } 498 errclose = zipClose(zf,NULL); 499 if (errclose != ZIP_OK) 500 printf("error in closing %s\n",filename_try); 501 } 502 else 503 { 504 do_help(); 505 } 506 507 free(buf); 508 return 0; 509 } 510