1 // example2.c - Simple demonstration of miniz.c's ZIP archive API's.
2 // Note this test deletes the test archive file "__mz_example2_test__.zip" in the current directory, then creates a new one with test data.
3 // Public domain, May 15 2011, Rich Geldreich, richgel99@gmail.com. See "unlicense" statement at the end of tinfl.c.
4 
5 #if defined(__GNUC__)
6   // Ensure we get the 64-bit variants of the CRT's file I/O calls
7   #ifndef _FILE_OFFSET_BITS
8     #define _FILE_OFFSET_BITS 64
9   #endif
10   #ifndef _LARGEFILE64_SOURCE
11     #define _LARGEFILE64_SOURCE 1
12   #endif
13 #endif
14 
15 #include <stdio.h>
16 #include "miniz_zip.h"
17 
18 typedef unsigned char uint8;
19 typedef unsigned short uint16;
20 typedef unsigned int uint;
21 
22 // The string to compress.
23 static const char *s_pTest_str =
24   "MISSION CONTROL I wouldn't worry too much about the computer. First of all, there is still a chance that he is right, despite your tests, and" \
25   "if it should happen again, we suggest eliminating this possibility by allowing the unit to remain in place and seeing whether or not it" \
26   "actually fails. If the computer should turn out to be wrong, the situation is still not alarming. The type of obsessional error he may be" \
27   "guilty of is not unknown among the latest generation of HAL 9000 computers. It has almost always revolved around a single detail, such as" \
28   "the one you have described, and it has never interfered with the integrity or reliability of the computer's performance in other areas." \
29   "No one is certain of the cause of this kind of malfunctioning. It may be over-programming, but it could also be any number of reasons. In any" \
30   "event, it is somewhat analogous to human neurotic behavior. Does this answer your query?  Zero-five-three-Zero, MC, transmission concluded.";
31 
32 static const char *s_pComment = "This is a comment";
33 
main(int argc,char * argv[])34 int main(int argc, char *argv[])
35 {
36   int i, sort_iter;
37   mz_bool status;
38   size_t uncomp_size;
39   mz_zip_archive zip_archive;
40   void *p;
41   const int N = 50;
42   char data[2048];
43   char archive_filename[64];
44   static const char *s_Test_archive_filename = "__mz_example2_test__.zip";
45 
46   assert((strlen(s_pTest_str) + 64) < sizeof(data));
47 
48   printf("miniz.c version: %s\n", MZ_VERSION);
49 
50   (void)argc, (void)argv;
51 
52   // Delete the test archive, so it doesn't keep growing as we run this test
53   remove(s_Test_archive_filename);
54 
55   // Append a bunch of text files to the test archive
56   for (i = (N - 1); i >= 0; --i)
57   {
58     sprintf(archive_filename, "%u.txt", i);
59     sprintf(data, "%u %s %u", (N - 1) - i, s_pTest_str, i);
60 
61     // Add a new file to the archive. Note this is an IN-PLACE operation, so if it fails your archive is probably hosed (its central directory may not be complete) but it should be recoverable using zip -F or -FF. So use caution with this guy.
62     // A more robust way to add a file to an archive would be to read it into memory, perform the operation, then write a new archive out to a temp file and then delete/rename the files.
63     // Or, write a new archive to disk to a temp file, then delete/rename the files. For this test this API is fine.
64     status = mz_zip_add_mem_to_archive_file_in_place(s_Test_archive_filename, archive_filename, data, strlen(data) + 1, s_pComment, (uint16)strlen(s_pComment), MZ_BEST_COMPRESSION);
65     if (!status)
66     {
67       printf("mz_zip_add_mem_to_archive_file_in_place failed!\n");
68       return EXIT_FAILURE;
69     }
70   }
71 
72   // Add a directory entry for testing
73   status = mz_zip_add_mem_to_archive_file_in_place(s_Test_archive_filename, "directory/", NULL, 0, "no comment", (uint16)strlen("no comment"), MZ_BEST_COMPRESSION);
74   if (!status)
75   {
76     printf("mz_zip_add_mem_to_archive_file_in_place failed!\n");
77     return EXIT_FAILURE;
78   }
79 
80   // Now try to open the archive.
81   memset(&zip_archive, 0, sizeof(zip_archive));
82 
83   status = mz_zip_reader_init_file(&zip_archive, s_Test_archive_filename, 0);
84   if (!status)
85   {
86     printf("mz_zip_reader_init_file() failed!\n");
87     return EXIT_FAILURE;
88   }
89 
90   // Get and print information about each file in the archive.
91   for (i = 0; i < (int)mz_zip_reader_get_num_files(&zip_archive); i++)
92   {
93     mz_zip_archive_file_stat file_stat;
94     if (!mz_zip_reader_file_stat(&zip_archive, i, &file_stat))
95     {
96        printf("mz_zip_reader_file_stat() failed!\n");
97        mz_zip_reader_end(&zip_archive);
98        return EXIT_FAILURE;
99     }
100 
101     printf("Filename: \"%s\", Comment: \"%s\", Uncompressed size: %u, Compressed size: %u, Is Dir: %u\n", file_stat.m_filename, file_stat.m_comment, (uint)file_stat.m_uncomp_size, (uint)file_stat.m_comp_size, mz_zip_reader_is_file_a_directory(&zip_archive, i));
102 
103     if (!strcmp(file_stat.m_filename, "directory/"))
104     {
105       if (!mz_zip_reader_is_file_a_directory(&zip_archive, i))
106       {
107         printf("mz_zip_reader_is_file_a_directory() didn't return the expected results!\n");
108         mz_zip_reader_end(&zip_archive);
109         return EXIT_FAILURE;
110       }
111     }
112   }
113 
114   // Close the archive, freeing any resources it was using
115   mz_zip_reader_end(&zip_archive);
116 
117   // Now verify the compressed data
118   for (sort_iter = 0; sort_iter < 2; sort_iter++)
119   {
120     memset(&zip_archive, 0, sizeof(zip_archive));
121     status = mz_zip_reader_init_file(&zip_archive, s_Test_archive_filename, sort_iter ? MZ_ZIP_FLAG_DO_NOT_SORT_CENTRAL_DIRECTORY : 0);
122     if (!status)
123     {
124       printf("mz_zip_reader_init_file() failed!\n");
125       return EXIT_FAILURE;
126     }
127 
128     for (i = 0; i < N; i++)
129     {
130       sprintf(archive_filename, "%u.txt", i);
131       sprintf(data, "%u %s %u", (N - 1) - i, s_pTest_str, i);
132 
133       // Try to extract all the files to the heap.
134       p = mz_zip_reader_extract_file_to_heap(&zip_archive, archive_filename, &uncomp_size, 0);
135       if (!p)
136       {
137         printf("mz_zip_reader_extract_file_to_heap() failed!\n");
138         mz_zip_reader_end(&zip_archive);
139         return EXIT_FAILURE;
140       }
141 
142       // Make sure the extraction really succeeded.
143       if ((uncomp_size != (strlen(data) + 1)) || (memcmp(p, data, strlen(data))))
144       {
145         printf("mz_zip_reader_extract_file_to_heap() failed to extract the proper data\n");
146         mz_free(p);
147         mz_zip_reader_end(&zip_archive);
148         return EXIT_FAILURE;
149       }
150 
151       printf("Successfully extracted file \"%s\", size %u\n", archive_filename, (uint)uncomp_size);
152       printf("File data: \"%s\"\n", (const char *)p);
153 
154       // We're done.
155       mz_free(p);
156     }
157 
158     // Close the archive, freeing any resources it was using
159     mz_zip_reader_end(&zip_archive);
160   }
161 
162   printf("Success.\n");
163   return EXIT_SUCCESS;
164 }
165