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
filetime(f,tmzip,dt)38 uLong 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
filetime(f,tmzip,dt)62 uLong 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
filetime(f,tmzip,dt)104 uLong 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
check_exist_file(filename)117 int 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
do_banner()130 void 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
do_help()136 void 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 */
getFileCrc(const char * filenameinzip,void * buf,unsigned long size_buf,unsigned long * result_crc)148 int 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
main(argc,argv)186 int 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