|
| 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, [email protected]. 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.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 | + |
| 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 | +} |
0 commit comments