|
@@ -7,29 +7,40 @@
|
|
|
* ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
|
|
|
* OTHER DEALINGS IN THE SOFTWARE.
|
|
|
*/
|
|
|
-
|
|
|
-#include "zip.h"
|
|
|
-#include "miniz.h"
|
|
|
+#define __STDC_WANT_LIB_EXT1__ 1
|
|
|
|
|
|
#include <errno.h>
|
|
|
#include <sys/stat.h>
|
|
|
#include <time.h>
|
|
|
|
|
|
-#if defined _WIN32 || defined __WIN32__
|
|
|
-/* Win32, DOS */
|
|
|
+#if defined(_WIN32) || defined(__WIN32__) || defined(_MSC_VER) || \
|
|
|
+ defined(__MINGW32__)
|
|
|
+/* Win32, DOS, MSVC, MSVS */
|
|
|
#include <direct.h>
|
|
|
|
|
|
#define MKDIR(DIRNAME) _mkdir(DIRNAME)
|
|
|
#define STRCLONE(STR) ((STR) ? _strdup(STR) : NULL)
|
|
|
#define HAS_DEVICE(P) \
|
|
|
- ((((P)[0] >= 'A' && (P)[0] <= 'Z') || ((P)[0] >= 'a' && (P)[0] <= 'z')) && \
|
|
|
- (P)[1] == ':')
|
|
|
+ ((((P)[0] >= 'A' && (P)[0] <= 'Z') || ((P)[0] >= 'a' && (P)[0] <= 'z')) && \
|
|
|
+ (P)[1] == ':')
|
|
|
#define FILESYSTEM_PREFIX_LEN(P) (HAS_DEVICE(P) ? 2 : 0)
|
|
|
#define ISSLASH(C) ((C) == '/' || (C) == '\\')
|
|
|
|
|
|
#else
|
|
|
+
|
|
|
+#include <unistd.h> // needed for symlink() on BSD
|
|
|
+int symlink(const char *target, const char *linkpath); // needed on Linux
|
|
|
+
|
|
|
#define MKDIR(DIRNAME) mkdir(DIRNAME, 0755)
|
|
|
#define STRCLONE(STR) ((STR) ? strdup(STR) : NULL)
|
|
|
+
|
|
|
+#endif
|
|
|
+
|
|
|
+#include "miniz.h"
|
|
|
+#include "zip.h"
|
|
|
+
|
|
|
+#ifndef HAS_DEVICE
|
|
|
+#define HAS_DEVICE(P) 0
|
|
|
#endif
|
|
|
|
|
|
#ifndef FILESYSTEM_PREFIX_LEN
|
|
@@ -40,601 +51,871 @@
|
|
|
#define ISSLASH(C) ((C) == '/')
|
|
|
#endif
|
|
|
|
|
|
-#define CLEANUP(ptr) \
|
|
|
- do { \
|
|
|
- if (ptr) { \
|
|
|
- free((void *)ptr); \
|
|
|
- ptr = NULL; \
|
|
|
- } \
|
|
|
- } while (0)
|
|
|
-
|
|
|
-static char *basename(const char *name) {
|
|
|
- char const *p;
|
|
|
- char const *base = name += FILESYSTEM_PREFIX_LEN(name);
|
|
|
- int all_slashes = 1;
|
|
|
-
|
|
|
- for (p = name; *p; p++) {
|
|
|
- if (ISSLASH(*p))
|
|
|
- base = p + 1;
|
|
|
- else
|
|
|
- all_slashes = 0;
|
|
|
- }
|
|
|
-
|
|
|
- /* If NAME is all slashes, arrange to return `/'. */
|
|
|
- if (*base == '\0' && ISSLASH(*name) && all_slashes) --base;
|
|
|
-
|
|
|
- return (char *)base;
|
|
|
+#define CLEANUP(ptr) \
|
|
|
+ do { \
|
|
|
+ if (ptr) { \
|
|
|
+ free((void *)ptr); \
|
|
|
+ ptr = NULL; \
|
|
|
+ } \
|
|
|
+ } while (0)
|
|
|
+
|
|
|
+static const char *base_name(const char *name) {
|
|
|
+ char const *p;
|
|
|
+ char const *base = name += FILESYSTEM_PREFIX_LEN(name);
|
|
|
+ int all_slashes = 1;
|
|
|
+
|
|
|
+ for (p = name; *p; p++) {
|
|
|
+ if (ISSLASH(*p))
|
|
|
+ base = p + 1;
|
|
|
+ else
|
|
|
+ all_slashes = 0;
|
|
|
+ }
|
|
|
+
|
|
|
+ /* If NAME is all slashes, arrange to return `/'. */
|
|
|
+ if (*base == '\0' && ISSLASH(*name) && all_slashes)
|
|
|
+ --base;
|
|
|
+
|
|
|
+ return base;
|
|
|
}
|
|
|
|
|
|
static int mkpath(const char *path) {
|
|
|
- char const *p;
|
|
|
- char npath[MAX_PATH + 1] = {0};
|
|
|
- int len = 0;
|
|
|
-
|
|
|
- for (p = path; *p && len < MAX_PATH; p++) {
|
|
|
- if (ISSLASH(*p) && len > 0) {
|
|
|
- if (MKDIR(npath) == -1)
|
|
|
- if (errno != EEXIST) return -1;
|
|
|
- }
|
|
|
- npath[len++] = *p;
|
|
|
- }
|
|
|
-
|
|
|
- return 0;
|
|
|
+ char const *p;
|
|
|
+ char npath[MAX_PATH + 1];
|
|
|
+ int len = 0;
|
|
|
+ int has_device = HAS_DEVICE(path);
|
|
|
+
|
|
|
+ memset(npath, 0, MAX_PATH + 1);
|
|
|
+ if (has_device) {
|
|
|
+ // only on windows
|
|
|
+ npath[0] = path[0];
|
|
|
+ npath[1] = path[1];
|
|
|
+ len = 2;
|
|
|
+ }
|
|
|
+ for (p = path + len; *p && len < MAX_PATH; p++) {
|
|
|
+ if (ISSLASH(*p) && ((!has_device && len > 0) || (has_device && len > 2))) {
|
|
|
+ if (MKDIR(npath) == -1)
|
|
|
+ if (errno != EEXIST)
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+ npath[len++] = *p;
|
|
|
+ }
|
|
|
+
|
|
|
+ return 0;
|
|
|
}
|
|
|
|
|
|
-static char *strrpl(const char *str, char oldchar, char newchar) {
|
|
|
- char *rpl = (char *)malloc(sizeof(char) * (1 + strlen(str)));
|
|
|
- char *begin = rpl;
|
|
|
- char c;
|
|
|
- while((c = *str++)) {
|
|
|
- if (c == oldchar) {
|
|
|
- c = newchar;
|
|
|
- }
|
|
|
- *rpl++ = c;
|
|
|
+static char *strrpl(const char *str, size_t n, char oldchar, char newchar) {
|
|
|
+ char c;
|
|
|
+ size_t i;
|
|
|
+ char *rpl = (char *)calloc((1 + n), sizeof(char));
|
|
|
+ char *begin = rpl;
|
|
|
+ if (!rpl) {
|
|
|
+ return NULL;
|
|
|
+ }
|
|
|
+
|
|
|
+ for (i = 0; (i < n) && (c = *str++); ++i) {
|
|
|
+ if (c == oldchar) {
|
|
|
+ c = newchar;
|
|
|
}
|
|
|
- *rpl = '\0';
|
|
|
+ *rpl++ = c;
|
|
|
+ }
|
|
|
|
|
|
- return begin;
|
|
|
+ return begin;
|
|
|
}
|
|
|
|
|
|
struct zip_entry_t {
|
|
|
- int index;
|
|
|
- const char *name;
|
|
|
- mz_uint64 uncomp_size;
|
|
|
- mz_uint64 comp_size;
|
|
|
- mz_uint32 uncomp_crc32;
|
|
|
- mz_uint64 offset;
|
|
|
- mz_uint8 header[MZ_ZIP_LOCAL_DIR_HEADER_SIZE];
|
|
|
- mz_uint64 header_offset;
|
|
|
- mz_uint16 method;
|
|
|
- mz_zip_writer_add_state state;
|
|
|
- tdefl_compressor comp;
|
|
|
+ int index;
|
|
|
+ char *name;
|
|
|
+ mz_uint64 uncomp_size;
|
|
|
+ mz_uint64 comp_size;
|
|
|
+ mz_uint32 uncomp_crc32;
|
|
|
+ mz_uint64 offset;
|
|
|
+ mz_uint8 header[MZ_ZIP_LOCAL_DIR_HEADER_SIZE];
|
|
|
+ mz_uint64 header_offset;
|
|
|
+ mz_uint16 method;
|
|
|
+ mz_zip_writer_add_state state;
|
|
|
+ tdefl_compressor comp;
|
|
|
+ mz_uint32 external_attr;
|
|
|
+ time_t m_time;
|
|
|
};
|
|
|
|
|
|
struct zip_t {
|
|
|
- mz_zip_archive archive;
|
|
|
- mz_uint level;
|
|
|
- struct zip_entry_t entry;
|
|
|
- char mode;
|
|
|
+ mz_zip_archive archive;
|
|
|
+ mz_uint level;
|
|
|
+ struct zip_entry_t entry;
|
|
|
};
|
|
|
|
|
|
struct zip_t *zip_open(const char *zipname, int level, char mode) {
|
|
|
- struct zip_t *zip = NULL;
|
|
|
-
|
|
|
- if (!zipname || strlen(zipname) < 1) {
|
|
|
- // zip_t archive name is empty or NULL
|
|
|
- goto cleanup;
|
|
|
+ struct zip_t *zip = NULL;
|
|
|
+
|
|
|
+ if (!zipname || strlen(zipname) < 1) {
|
|
|
+ // zip_t archive name is empty or NULL
|
|
|
+ goto cleanup;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (level < 0)
|
|
|
+ level = MZ_DEFAULT_LEVEL;
|
|
|
+ if ((level & 0xF) > MZ_UBER_COMPRESSION) {
|
|
|
+ // Wrong compression level
|
|
|
+ goto cleanup;
|
|
|
+ }
|
|
|
+
|
|
|
+ zip = (struct zip_t *)calloc((size_t)1, sizeof(struct zip_t));
|
|
|
+ if (!zip)
|
|
|
+ goto cleanup;
|
|
|
+
|
|
|
+ zip->level = (mz_uint)level;
|
|
|
+ switch (mode) {
|
|
|
+ case 'w':
|
|
|
+ // Create a new archive.
|
|
|
+ if (!mz_zip_writer_init_file(&(zip->archive), zipname, 0)) {
|
|
|
+ // Cannot initialize zip_archive writer
|
|
|
+ goto cleanup;
|
|
|
}
|
|
|
+ break;
|
|
|
|
|
|
- if (level < 0) level = MZ_DEFAULT_LEVEL;
|
|
|
- if ((level & 0xF) > MZ_UBER_COMPRESSION) {
|
|
|
- // Wrong compression level
|
|
|
- goto cleanup;
|
|
|
+ case 'r':
|
|
|
+ case 'a':
|
|
|
+ if (!mz_zip_reader_init_file(
|
|
|
+ &(zip->archive), zipname,
|
|
|
+ zip->level | MZ_ZIP_FLAG_DO_NOT_SORT_CENTRAL_DIRECTORY)) {
|
|
|
+ // An archive file does not exist or cannot initialize
|
|
|
+ // zip_archive reader
|
|
|
+ goto cleanup;
|
|
|
}
|
|
|
-
|
|
|
- zip = (struct zip_t *)calloc((size_t)1, sizeof(struct zip_t));
|
|
|
- if (!zip) goto cleanup;
|
|
|
-
|
|
|
- zip->level = level;
|
|
|
- zip->mode = mode;
|
|
|
- switch (mode) {
|
|
|
- case 'w':
|
|
|
- // Create a new archive.
|
|
|
- if (!mz_zip_writer_init_file(&(zip->archive), zipname, 0)) {
|
|
|
- // Cannot initialize zip_archive writer
|
|
|
- goto cleanup;
|
|
|
- }
|
|
|
- break;
|
|
|
-
|
|
|
- case 'r':
|
|
|
- case 'a':
|
|
|
- if (!mz_zip_reader_init_file(
|
|
|
- &(zip->archive), zipname,
|
|
|
- level | MZ_ZIP_FLAG_DO_NOT_SORT_CENTRAL_DIRECTORY)) {
|
|
|
- // An archive file does not exist or cannot initialize
|
|
|
- // zip_archive reader
|
|
|
- goto cleanup;
|
|
|
- }
|
|
|
-
|
|
|
- if (mode == 'a' &&
|
|
|
- !mz_zip_writer_init_from_reader(&(zip->archive), zipname)) {
|
|
|
- mz_zip_reader_end(&(zip->archive));
|
|
|
- goto cleanup;
|
|
|
- }
|
|
|
-
|
|
|
- break;
|
|
|
-
|
|
|
- default:
|
|
|
- goto cleanup;
|
|
|
+ if (mode == 'a' &&
|
|
|
+ !mz_zip_writer_init_from_reader(&(zip->archive), zipname)) {
|
|
|
+ mz_zip_reader_end(&(zip->archive));
|
|
|
+ goto cleanup;
|
|
|
}
|
|
|
+ break;
|
|
|
+
|
|
|
+ default:
|
|
|
+ goto cleanup;
|
|
|
+ }
|
|
|
|
|
|
- return zip;
|
|
|
+ return zip;
|
|
|
|
|
|
cleanup:
|
|
|
- CLEANUP(zip);
|
|
|
- return NULL;
|
|
|
+ CLEANUP(zip);
|
|
|
+ return NULL;
|
|
|
}
|
|
|
|
|
|
void zip_close(struct zip_t *zip) {
|
|
|
- if (zip) {
|
|
|
- // Always finalize, even if adding failed for some reason, so we have a
|
|
|
- // valid central directory.
|
|
|
- mz_zip_writer_finalize_archive(&(zip->archive));
|
|
|
+ if (zip) {
|
|
|
+ // Always finalize, even if adding failed for some reason, so we have a
|
|
|
+ // valid central directory.
|
|
|
+ mz_zip_writer_finalize_archive(&(zip->archive));
|
|
|
|
|
|
- mz_zip_writer_end(&(zip->archive));
|
|
|
- mz_zip_reader_end(&(zip->archive));
|
|
|
+ mz_zip_writer_end(&(zip->archive));
|
|
|
+ mz_zip_reader_end(&(zip->archive));
|
|
|
|
|
|
- CLEANUP(zip);
|
|
|
- }
|
|
|
+ CLEANUP(zip);
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
int zip_entry_open(struct zip_t *zip, const char *entryname) {
|
|
|
- char *locname = NULL;
|
|
|
- size_t entrylen = 0;
|
|
|
- mz_zip_archive *pzip = NULL;
|
|
|
- mz_uint num_alignment_padding_bytes, level;
|
|
|
-
|
|
|
- if (!zip || !entryname) {
|
|
|
- return -1;
|
|
|
- }
|
|
|
-
|
|
|
- entrylen = strlen(entryname);
|
|
|
- if (entrylen < 1) {
|
|
|
- return -1;
|
|
|
- }
|
|
|
-
|
|
|
- pzip = &(zip->archive);
|
|
|
- /*
|
|
|
- .ZIP File Format Specification Version: 6.3.3
|
|
|
-
|
|
|
- 4.4.17.1 The name of the file, with optional relative path.
|
|
|
- The path stored MUST not contain a drive or
|
|
|
- device letter, or a leading slash. All slashes
|
|
|
- MUST be forward slashes '/' as opposed to
|
|
|
- backwards slashes '\' for compatibility with Amiga
|
|
|
- and UNIX file systems etc. If input came from standard
|
|
|
- input, there is no file name field.
|
|
|
- */
|
|
|
- locname = strrpl(entryname, '\\', '/');
|
|
|
-
|
|
|
- if (zip->mode == 'r') {
|
|
|
- zip->entry.index = mz_zip_reader_locate_file(pzip, locname, NULL, 0);
|
|
|
- CLEANUP(locname);
|
|
|
- return (zip->entry.index < 0) ? -1 : 0;
|
|
|
- }
|
|
|
-
|
|
|
- zip->entry.index = zip->archive.m_total_files;
|
|
|
- zip->entry.name = locname;
|
|
|
- if (!zip->entry.name) {
|
|
|
- // Cannot parse zip entry name
|
|
|
- return -1;
|
|
|
- }
|
|
|
+ size_t entrylen = 0;
|
|
|
+ mz_zip_archive *pzip = NULL;
|
|
|
+ mz_uint num_alignment_padding_bytes, level;
|
|
|
+ mz_zip_archive_file_stat stats;
|
|
|
+
|
|
|
+ if (!zip || !entryname) {
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ entrylen = strlen(entryname);
|
|
|
+ if (entrylen < 1) {
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ /*
|
|
|
+ .ZIP File Format Specification Version: 6.3.3
|
|
|
+
|
|
|
+ 4.4.17.1 The name of the file, with optional relative path.
|
|
|
+ The path stored MUST not contain a drive or
|
|
|
+ device letter, or a leading slash. All slashes
|
|
|
+ MUST be forward slashes '/' as opposed to
|
|
|
+ backwards slashes '\' for compatibility with Amiga
|
|
|
+ and UNIX file systems etc. If input came from standard
|
|
|
+ input, there is no file name field.
|
|
|
+ */
|
|
|
+ zip->entry.name = strrpl(entryname, entrylen, '\\', '/');
|
|
|
+ if (!zip->entry.name) {
|
|
|
+ // Cannot parse zip entry name
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ pzip = &(zip->archive);
|
|
|
+ if (pzip->m_zip_mode == MZ_ZIP_MODE_READING) {
|
|
|
+ zip->entry.index =
|
|
|
+ mz_zip_reader_locate_file(pzip, zip->entry.name, NULL, 0);
|
|
|
+ if (zip->entry.index < 0) {
|
|
|
+ goto cleanup;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (!mz_zip_reader_file_stat(pzip, (mz_uint)zip->entry.index, &stats)) {
|
|
|
+ goto cleanup;
|
|
|
+ }
|
|
|
+
|
|
|
+ zip->entry.comp_size = stats.m_comp_size;
|
|
|
+ zip->entry.uncomp_size = stats.m_uncomp_size;
|
|
|
+ zip->entry.uncomp_crc32 = stats.m_crc32;
|
|
|
+ zip->entry.offset = stats.m_central_dir_ofs;
|
|
|
+ zip->entry.header_offset = stats.m_local_header_ofs;
|
|
|
+ zip->entry.method = stats.m_method;
|
|
|
+ zip->entry.external_attr = stats.m_external_attr;
|
|
|
+ zip->entry.m_time = stats.m_time;
|
|
|
|
|
|
- zip->entry.comp_size = 0;
|
|
|
- zip->entry.uncomp_size = 0;
|
|
|
- zip->entry.uncomp_crc32 = MZ_CRC32_INIT;
|
|
|
- zip->entry.offset = zip->archive.m_archive_size;
|
|
|
- zip->entry.header_offset = zip->archive.m_archive_size;
|
|
|
- memset(zip->entry.header, 0,
|
|
|
- MZ_ZIP_LOCAL_DIR_HEADER_SIZE * sizeof(mz_uint8));
|
|
|
- zip->entry.method = 0;
|
|
|
-
|
|
|
- num_alignment_padding_bytes =
|
|
|
- mz_zip_writer_compute_padding_needed_for_file_alignment(pzip);
|
|
|
-
|
|
|
- if (!pzip->m_pState || (pzip->m_zip_mode != MZ_ZIP_MODE_WRITING)) {
|
|
|
- // Wrong zip mode
|
|
|
- return -1;
|
|
|
- }
|
|
|
- if (zip->level & MZ_ZIP_FLAG_COMPRESSED_DATA) {
|
|
|
- // Wrong zip compression level
|
|
|
- return -1;
|
|
|
- }
|
|
|
- // no zip64 support yet
|
|
|
- if ((pzip->m_total_files == 0xFFFF) ||
|
|
|
- ((pzip->m_archive_size + num_alignment_padding_bytes +
|
|
|
- MZ_ZIP_LOCAL_DIR_HEADER_SIZE + MZ_ZIP_CENTRAL_DIR_HEADER_SIZE +
|
|
|
- entrylen) > 0xFFFFFFFF)) {
|
|
|
- // No zip64 support yet
|
|
|
- return -1;
|
|
|
- }
|
|
|
- if (!mz_zip_writer_write_zeros(
|
|
|
- pzip, zip->entry.offset,
|
|
|
- num_alignment_padding_bytes + sizeof(zip->entry.header))) {
|
|
|
- // Cannot memset zip entry header
|
|
|
- return -1;
|
|
|
- }
|
|
|
+ return 0;
|
|
|
+ }
|
|
|
+
|
|
|
+ zip->entry.index = (int)zip->archive.m_total_files;
|
|
|
+ zip->entry.comp_size = 0;
|
|
|
+ zip->entry.uncomp_size = 0;
|
|
|
+ zip->entry.uncomp_crc32 = MZ_CRC32_INIT;
|
|
|
+ zip->entry.offset = zip->archive.m_archive_size;
|
|
|
+ zip->entry.header_offset = zip->archive.m_archive_size;
|
|
|
+ memset(zip->entry.header, 0, MZ_ZIP_LOCAL_DIR_HEADER_SIZE * sizeof(mz_uint8));
|
|
|
+ zip->entry.method = 0;
|
|
|
+ zip->entry.external_attr = 0;
|
|
|
+
|
|
|
+ num_alignment_padding_bytes =
|
|
|
+ mz_zip_writer_compute_padding_needed_for_file_alignment(pzip);
|
|
|
+
|
|
|
+ if (!pzip->m_pState || (pzip->m_zip_mode != MZ_ZIP_MODE_WRITING)) {
|
|
|
+ // Wrong zip mode
|
|
|
+ goto cleanup;
|
|
|
+ }
|
|
|
+ if (zip->level & MZ_ZIP_FLAG_COMPRESSED_DATA) {
|
|
|
+ // Wrong zip compression level
|
|
|
+ goto cleanup;
|
|
|
+ }
|
|
|
+ // no zip64 support yet
|
|
|
+ if ((pzip->m_total_files == 0xFFFF) ||
|
|
|
+ ((pzip->m_archive_size + num_alignment_padding_bytes +
|
|
|
+ MZ_ZIP_LOCAL_DIR_HEADER_SIZE + MZ_ZIP_CENTRAL_DIR_HEADER_SIZE +
|
|
|
+ entrylen) > 0xFFFFFFFF)) {
|
|
|
+ // No zip64 support yet
|
|
|
+ goto cleanup;
|
|
|
+ }
|
|
|
+ if (!mz_zip_writer_write_zeros(pzip, zip->entry.offset,
|
|
|
+ num_alignment_padding_bytes +
|
|
|
+ sizeof(zip->entry.header))) {
|
|
|
+ // Cannot memset zip entry header
|
|
|
+ goto cleanup;
|
|
|
+ }
|
|
|
+
|
|
|
+ zip->entry.header_offset += num_alignment_padding_bytes;
|
|
|
+ if (pzip->m_file_offset_alignment) {
|
|
|
+ MZ_ASSERT(
|
|
|
+ (zip->entry.header_offset & (pzip->m_file_offset_alignment - 1)) == 0);
|
|
|
+ }
|
|
|
+ zip->entry.offset += num_alignment_padding_bytes + sizeof(zip->entry.header);
|
|
|
+
|
|
|
+ if (pzip->m_pWrite(pzip->m_pIO_opaque, zip->entry.offset, zip->entry.name,
|
|
|
+ entrylen) != entrylen) {
|
|
|
+ // Cannot write data to zip entry
|
|
|
+ goto cleanup;
|
|
|
+ }
|
|
|
+
|
|
|
+ zip->entry.offset += entrylen;
|
|
|
+ level = zip->level & 0xF;
|
|
|
+ if (level) {
|
|
|
+ zip->entry.state.m_pZip = pzip;
|
|
|
+ zip->entry.state.m_cur_archive_file_ofs = zip->entry.offset;
|
|
|
+ zip->entry.state.m_comp_size = 0;
|
|
|
+
|
|
|
+ if (tdefl_init(&(zip->entry.comp), mz_zip_writer_add_put_buf_callback,
|
|
|
+ &(zip->entry.state),
|
|
|
+ (int)tdefl_create_comp_flags_from_zip_params(
|
|
|
+ (int)level, -15, MZ_DEFAULT_STRATEGY)) !=
|
|
|
+ TDEFL_STATUS_OKAY) {
|
|
|
+ // Cannot initialize the zip compressor
|
|
|
+ goto cleanup;
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ zip->entry.m_time = time(NULL);
|
|
|
+
|
|
|
+ return 0;
|
|
|
|
|
|
- zip->entry.header_offset += num_alignment_padding_bytes;
|
|
|
- if (pzip->m_file_offset_alignment) {
|
|
|
- MZ_ASSERT((zip->entry.header_offset &
|
|
|
- (pzip->m_file_offset_alignment - 1)) == 0);
|
|
|
- }
|
|
|
- zip->entry.offset +=
|
|
|
- num_alignment_padding_bytes + sizeof(zip->entry.header);
|
|
|
+cleanup:
|
|
|
+ CLEANUP(zip->entry.name);
|
|
|
+ return -1;
|
|
|
+}
|
|
|
|
|
|
- if (pzip->m_pWrite(pzip->m_pIO_opaque, zip->entry.offset, zip->entry.name,
|
|
|
- entrylen) != entrylen) {
|
|
|
- // Cannot write data to zip entry
|
|
|
- return -1;
|
|
|
- }
|
|
|
+int zip_entry_openbyindex(struct zip_t *zip, int index) {
|
|
|
+ mz_zip_archive *pZip = NULL;
|
|
|
+ mz_zip_archive_file_stat stats;
|
|
|
+ mz_uint namelen;
|
|
|
+ const mz_uint8 *pHeader;
|
|
|
+ const char *pFilename;
|
|
|
+
|
|
|
+ if (!zip) {
|
|
|
+ // zip_t handler is not initialized
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ pZip = &(zip->archive);
|
|
|
+ if (pZip->m_zip_mode != MZ_ZIP_MODE_READING) {
|
|
|
+ // open by index requires readonly mode
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (index < 0 || (mz_uint)index >= pZip->m_total_files) {
|
|
|
+ // index out of range
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (!(pHeader = &MZ_ZIP_ARRAY_ELEMENT(
|
|
|
+ &pZip->m_pState->m_central_dir, mz_uint8,
|
|
|
+ MZ_ZIP_ARRAY_ELEMENT(&pZip->m_pState->m_central_dir_offsets,
|
|
|
+ mz_uint32, index)))) {
|
|
|
+ // cannot find header in central directory
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ namelen = MZ_READ_LE16(pHeader + MZ_ZIP_CDH_FILENAME_LEN_OFS);
|
|
|
+ pFilename = (const char *)pHeader + MZ_ZIP_CENTRAL_DIR_HEADER_SIZE;
|
|
|
+
|
|
|
+ /*
|
|
|
+ .ZIP File Format Specification Version: 6.3.3
|
|
|
+
|
|
|
+ 4.4.17.1 The name of the file, with optional relative path.
|
|
|
+ The path stored MUST not contain a drive or
|
|
|
+ device letter, or a leading slash. All slashes
|
|
|
+ MUST be forward slashes '/' as opposed to
|
|
|
+ backwards slashes '\' for compatibility with Amiga
|
|
|
+ and UNIX file systems etc. If input came from standard
|
|
|
+ input, there is no file name field.
|
|
|
+ */
|
|
|
+ zip->entry.name = strrpl(pFilename, namelen, '\\', '/');
|
|
|
+ if (!zip->entry.name) {
|
|
|
+ // local entry name is NULL
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (!mz_zip_reader_file_stat(pZip, (mz_uint)index, &stats)) {
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ zip->entry.index = index;
|
|
|
+ zip->entry.comp_size = stats.m_comp_size;
|
|
|
+ zip->entry.uncomp_size = stats.m_uncomp_size;
|
|
|
+ zip->entry.uncomp_crc32 = stats.m_crc32;
|
|
|
+ zip->entry.offset = stats.m_central_dir_ofs;
|
|
|
+ zip->entry.header_offset = stats.m_local_header_ofs;
|
|
|
+ zip->entry.method = stats.m_method;
|
|
|
+ zip->entry.external_attr = stats.m_external_attr;
|
|
|
+ zip->entry.m_time = stats.m_time;
|
|
|
+
|
|
|
+ return 0;
|
|
|
+}
|
|
|
|
|
|
- zip->entry.offset += entrylen;
|
|
|
- level = zip->level & 0xF;
|
|
|
- if (level) {
|
|
|
- zip->entry.state.m_pZip = pzip;
|
|
|
- zip->entry.state.m_cur_archive_file_ofs = zip->entry.offset;
|
|
|
- zip->entry.state.m_comp_size = 0;
|
|
|
-
|
|
|
- if (tdefl_init(&(zip->entry.comp), mz_zip_writer_add_put_buf_callback,
|
|
|
- &(zip->entry.state),
|
|
|
- tdefl_create_comp_flags_from_zip_params(
|
|
|
- level, -15, MZ_DEFAULT_STRATEGY)) !=
|
|
|
- TDEFL_STATUS_OKAY) {
|
|
|
- // Cannot initialize the zip compressor
|
|
|
- return -1;
|
|
|
- }
|
|
|
- }
|
|
|
+int zip_entry_close(struct zip_t *zip) {
|
|
|
+ mz_zip_archive *pzip = NULL;
|
|
|
+ mz_uint level;
|
|
|
+ tdefl_status done;
|
|
|
+ mz_uint16 entrylen;
|
|
|
+ mz_uint16 dos_time, dos_date;
|
|
|
+ int status = -1;
|
|
|
+
|
|
|
+ if (!zip) {
|
|
|
+ // zip_t handler is not initialized
|
|
|
+ goto cleanup;
|
|
|
+ }
|
|
|
+
|
|
|
+ pzip = &(zip->archive);
|
|
|
+ if (pzip->m_zip_mode == MZ_ZIP_MODE_READING) {
|
|
|
+ status = 0;
|
|
|
+ goto cleanup;
|
|
|
+ }
|
|
|
+
|
|
|
+ level = zip->level & 0xF;
|
|
|
+ if (level) {
|
|
|
+ done = tdefl_compress_buffer(&(zip->entry.comp), "", 0, TDEFL_FINISH);
|
|
|
+ if (done != TDEFL_STATUS_DONE && done != TDEFL_STATUS_OKAY) {
|
|
|
+ // Cannot flush compressed buffer
|
|
|
+ goto cleanup;
|
|
|
+ }
|
|
|
+ zip->entry.comp_size = zip->entry.state.m_comp_size;
|
|
|
+ zip->entry.offset = zip->entry.state.m_cur_archive_file_ofs;
|
|
|
+ zip->entry.method = MZ_DEFLATED;
|
|
|
+ }
|
|
|
+
|
|
|
+ entrylen = (mz_uint16)strlen(zip->entry.name);
|
|
|
+ // no zip64 support yet
|
|
|
+ if ((zip->entry.comp_size > 0xFFFFFFFF) || (zip->entry.offset > 0xFFFFFFFF)) {
|
|
|
+ // No zip64 support, yet
|
|
|
+ goto cleanup;
|
|
|
+ }
|
|
|
+
|
|
|
+ mz_zip_time_t_to_dos_time(zip->entry.m_time, &dos_time, &dos_date);
|
|
|
+ if (!mz_zip_writer_create_local_dir_header(
|
|
|
+ pzip, zip->entry.header, entrylen, 0, zip->entry.uncomp_size,
|
|
|
+ zip->entry.comp_size, zip->entry.uncomp_crc32, zip->entry.method, 0,
|
|
|
+ dos_time, dos_date)) {
|
|
|
+ // Cannot create zip entry header
|
|
|
+ goto cleanup;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (pzip->m_pWrite(pzip->m_pIO_opaque, zip->entry.header_offset,
|
|
|
+ zip->entry.header,
|
|
|
+ sizeof(zip->entry.header)) != sizeof(zip->entry.header)) {
|
|
|
+ // Cannot write zip entry header
|
|
|
+ goto cleanup;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (!mz_zip_writer_add_to_central_dir(
|
|
|
+ pzip, zip->entry.name, entrylen, NULL, 0, "", 0,
|
|
|
+ zip->entry.uncomp_size, zip->entry.comp_size, zip->entry.uncomp_crc32,
|
|
|
+ zip->entry.method, 0, dos_time, dos_date, zip->entry.header_offset,
|
|
|
+ zip->entry.external_attr)) {
|
|
|
+ // Cannot write to zip central dir
|
|
|
+ goto cleanup;
|
|
|
+ }
|
|
|
+
|
|
|
+ pzip->m_total_files++;
|
|
|
+ pzip->m_archive_size = zip->entry.offset;
|
|
|
+ status = 0;
|
|
|
|
|
|
- return 0;
|
|
|
+cleanup:
|
|
|
+ if (zip) {
|
|
|
+ zip->entry.m_time = 0;
|
|
|
+ CLEANUP(zip->entry.name);
|
|
|
+ }
|
|
|
+ return status;
|
|
|
}
|
|
|
|
|
|
-int zip_entry_close(struct zip_t *zip) {
|
|
|
- mz_zip_archive *pzip = NULL;
|
|
|
- mz_uint level;
|
|
|
- tdefl_status done;
|
|
|
- mz_uint16 entrylen;
|
|
|
- time_t t;
|
|
|
- struct tm *tm;
|
|
|
- mz_uint16 dos_time, dos_date;
|
|
|
- int status = -1;
|
|
|
-
|
|
|
- if (!zip) {
|
|
|
- // zip_t handler is not initialized
|
|
|
- return -1;
|
|
|
- }
|
|
|
+const char *zip_entry_name(struct zip_t *zip) {
|
|
|
+ if (!zip) {
|
|
|
+ // zip_t handler is not initialized
|
|
|
+ return NULL;
|
|
|
+ }
|
|
|
|
|
|
- if (zip->mode == 'r') {
|
|
|
- return 0;
|
|
|
- }
|
|
|
+ return zip->entry.name;
|
|
|
+}
|
|
|
|
|
|
- pzip = &(zip->archive);
|
|
|
- level = zip->level & 0xF;
|
|
|
- if (level) {
|
|
|
- done = tdefl_compress_buffer(&(zip->entry.comp), "", 0, TDEFL_FINISH);
|
|
|
- if (done != TDEFL_STATUS_DONE && done != TDEFL_STATUS_OKAY) {
|
|
|
- // Cannot flush compressed buffer
|
|
|
- goto cleanup;
|
|
|
- }
|
|
|
- zip->entry.comp_size = zip->entry.state.m_comp_size;
|
|
|
- zip->entry.offset = zip->entry.state.m_cur_archive_file_ofs;
|
|
|
- zip->entry.method = MZ_DEFLATED;
|
|
|
- }
|
|
|
+int zip_entry_index(struct zip_t *zip) {
|
|
|
+ if (!zip) {
|
|
|
+ // zip_t handler is not initialized
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
|
|
|
- entrylen = (mz_uint16)strlen(zip->entry.name);
|
|
|
- t = time(NULL);
|
|
|
- tm = localtime(&t);
|
|
|
- dos_time = (mz_uint16)(((tm->tm_hour) << 11) + ((tm->tm_min) << 5) +
|
|
|
- ((tm->tm_sec) >> 1));
|
|
|
- dos_date = (mz_uint16)(((tm->tm_year + 1900 - 1980) << 9) +
|
|
|
- ((tm->tm_mon + 1) << 5) + tm->tm_mday);
|
|
|
-
|
|
|
- // no zip64 support yet
|
|
|
- if ((zip->entry.comp_size > 0xFFFFFFFF) ||
|
|
|
- (zip->entry.offset > 0xFFFFFFFF)) {
|
|
|
- // No zip64 support, yet
|
|
|
- goto cleanup;
|
|
|
- }
|
|
|
+ return zip->entry.index;
|
|
|
+}
|
|
|
|
|
|
- if (!mz_zip_writer_create_local_dir_header(
|
|
|
- pzip, zip->entry.header, entrylen, 0, zip->entry.uncomp_size,
|
|
|
- zip->entry.comp_size, zip->entry.uncomp_crc32, zip->entry.method, 0,
|
|
|
- dos_time, dos_date)) {
|
|
|
- // Cannot create zip entry header
|
|
|
- goto cleanup;
|
|
|
- }
|
|
|
+int zip_entry_isdir(struct zip_t *zip) {
|
|
|
+ if (!zip) {
|
|
|
+ // zip_t handler is not initialized
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
|
|
|
- if (pzip->m_pWrite(pzip->m_pIO_opaque, zip->entry.header_offset,
|
|
|
- zip->entry.header, sizeof(zip->entry.header)) !=
|
|
|
- sizeof(zip->entry.header)) {
|
|
|
- // Cannot write zip entry header
|
|
|
- goto cleanup;
|
|
|
- }
|
|
|
+ if (zip->entry.index < 0) {
|
|
|
+ // zip entry is not opened
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
|
|
|
- if (!mz_zip_writer_add_to_central_dir(
|
|
|
- pzip, zip->entry.name, entrylen, NULL, 0, "", 0,
|
|
|
- zip->entry.uncomp_size, zip->entry.comp_size,
|
|
|
- zip->entry.uncomp_crc32, zip->entry.method, 0, dos_time, dos_date,
|
|
|
- zip->entry.header_offset, 0)) {
|
|
|
- // Cannot write to zip central dir
|
|
|
- goto cleanup;
|
|
|
- }
|
|
|
+ return (int)mz_zip_reader_is_file_a_directory(&zip->archive,
|
|
|
+ (mz_uint)zip->entry.index);
|
|
|
+}
|
|
|
|
|
|
- pzip->m_total_files++;
|
|
|
- pzip->m_archive_size = zip->entry.offset;
|
|
|
- status = 0;
|
|
|
+unsigned long long zip_entry_size(struct zip_t *zip) {
|
|
|
+ return zip ? zip->entry.uncomp_size : 0;
|
|
|
+}
|
|
|
|
|
|
-cleanup:
|
|
|
- CLEANUP(zip->entry.name);
|
|
|
- return status;
|
|
|
+unsigned int zip_entry_crc32(struct zip_t *zip) {
|
|
|
+ return zip ? zip->entry.uncomp_crc32 : 0;
|
|
|
}
|
|
|
|
|
|
int zip_entry_write(struct zip_t *zip, const void *buf, size_t bufsize) {
|
|
|
- mz_uint level;
|
|
|
- mz_zip_archive *pzip = NULL;
|
|
|
- tdefl_status status;
|
|
|
+ mz_uint level;
|
|
|
+ mz_zip_archive *pzip = NULL;
|
|
|
+ tdefl_status status;
|
|
|
|
|
|
- if (!zip) {
|
|
|
- // zip_t handler is not initialized
|
|
|
- return -1;
|
|
|
- }
|
|
|
+ if (!zip) {
|
|
|
+ // zip_t handler is not initialized
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
|
|
|
- pzip = &(zip->archive);
|
|
|
- if (buf && bufsize > 0) {
|
|
|
- zip->entry.uncomp_size += bufsize;
|
|
|
- zip->entry.uncomp_crc32 = (mz_uint32)mz_crc32(
|
|
|
- zip->entry.uncomp_crc32, (const mz_uint8 *)buf, bufsize);
|
|
|
-
|
|
|
- level = zip->level & 0xF;
|
|
|
- if (!level) {
|
|
|
- if ((pzip->m_pWrite(pzip->m_pIO_opaque, zip->entry.offset, buf,
|
|
|
- bufsize) != bufsize)) {
|
|
|
- // Cannot write buffer
|
|
|
- return -1;
|
|
|
- }
|
|
|
- zip->entry.offset += bufsize;
|
|
|
- zip->entry.comp_size += bufsize;
|
|
|
- } else {
|
|
|
- status = tdefl_compress_buffer(&(zip->entry.comp), buf, bufsize,
|
|
|
- TDEFL_NO_FLUSH);
|
|
|
- if (status != TDEFL_STATUS_DONE && status != TDEFL_STATUS_OKAY) {
|
|
|
- // Cannot compress buffer
|
|
|
- return -1;
|
|
|
- }
|
|
|
- }
|
|
|
+ pzip = &(zip->archive);
|
|
|
+ if (buf && bufsize > 0) {
|
|
|
+ zip->entry.uncomp_size += bufsize;
|
|
|
+ zip->entry.uncomp_crc32 = (mz_uint32)mz_crc32(
|
|
|
+ zip->entry.uncomp_crc32, (const mz_uint8 *)buf, bufsize);
|
|
|
+
|
|
|
+ level = zip->level & 0xF;
|
|
|
+ if (!level) {
|
|
|
+ if ((pzip->m_pWrite(pzip->m_pIO_opaque, zip->entry.offset, buf,
|
|
|
+ bufsize) != bufsize)) {
|
|
|
+ // Cannot write buffer
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+ zip->entry.offset += bufsize;
|
|
|
+ zip->entry.comp_size += bufsize;
|
|
|
+ } else {
|
|
|
+ status = tdefl_compress_buffer(&(zip->entry.comp), buf, bufsize,
|
|
|
+ TDEFL_NO_FLUSH);
|
|
|
+ if (status != TDEFL_STATUS_DONE && status != TDEFL_STATUS_OKAY) {
|
|
|
+ // Cannot compress buffer
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
}
|
|
|
+ }
|
|
|
|
|
|
- return 0;
|
|
|
+ return 0;
|
|
|
}
|
|
|
|
|
|
int zip_entry_fwrite(struct zip_t *zip, const char *filename) {
|
|
|
- int status = 0;
|
|
|
- size_t n = 0;
|
|
|
- FILE *stream = NULL;
|
|
|
- mz_uint8 buf[MZ_ZIP_MAX_IO_BUF_SIZE] = {0};
|
|
|
-
|
|
|
- if (!zip) {
|
|
|
- // zip_t handler is not initialized
|
|
|
- return -1;
|
|
|
- }
|
|
|
+ int status = 0;
|
|
|
+ size_t n = 0;
|
|
|
+ FILE *stream = NULL;
|
|
|
+ mz_uint8 buf[MZ_ZIP_MAX_IO_BUF_SIZE];
|
|
|
+ struct MZ_FILE_STAT_STRUCT file_stat;
|
|
|
+
|
|
|
+ if (!zip) {
|
|
|
+ // zip_t handler is not initialized
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ memset(buf, 0, MZ_ZIP_MAX_IO_BUF_SIZE);
|
|
|
+ memset((void *)&file_stat, 0, sizeof(struct MZ_FILE_STAT_STRUCT));
|
|
|
+ if (MZ_FILE_STAT(filename, &file_stat) != 0) {
|
|
|
+ // problem getting information - check errno
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ if ((file_stat.st_mode & 0200) == 0) {
|
|
|
+ // MS-DOS read-only attribute
|
|
|
+ zip->entry.external_attr |= 0x01;
|
|
|
+ }
|
|
|
+ zip->entry.external_attr |= (mz_uint32)((file_stat.st_mode & 0xFFFF) << 16);
|
|
|
+ zip->entry.m_time = file_stat.st_mtime;
|
|
|
+
|
|
|
+#if defined(_MSC_VER)
|
|
|
+ if (fopen_s(&stream, filename, "rb"))
|
|
|
+#else
|
|
|
+ if (!(stream = fopen(filename, "rb")))
|
|
|
+#endif
|
|
|
+ {
|
|
|
+ // Cannot open filename
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
|
|
|
- stream = fopen(filename, "rb");
|
|
|
- if (!stream) {
|
|
|
- // Cannot open filename
|
|
|
- return -1;
|
|
|
+ while ((n = fread(buf, sizeof(mz_uint8), MZ_ZIP_MAX_IO_BUF_SIZE, stream)) >
|
|
|
+ 0) {
|
|
|
+ if (zip_entry_write(zip, buf, n) < 0) {
|
|
|
+ status = -1;
|
|
|
+ break;
|
|
|
}
|
|
|
+ }
|
|
|
+ fclose(stream);
|
|
|
|
|
|
- while ((n = fread(buf, sizeof(mz_uint8), MZ_ZIP_MAX_IO_BUF_SIZE, stream)) >
|
|
|
- 0) {
|
|
|
- if (zip_entry_write(zip, buf, n) < 0) {
|
|
|
- status = -1;
|
|
|
- break;
|
|
|
- }
|
|
|
- }
|
|
|
- fclose(stream);
|
|
|
+ return status;
|
|
|
+}
|
|
|
|
|
|
- return status;
|
|
|
+ssize_t zip_entry_read(struct zip_t *zip, void **buf, size_t *bufsize) {
|
|
|
+ mz_zip_archive *pzip = NULL;
|
|
|
+ mz_uint idx;
|
|
|
+ size_t size = 0;
|
|
|
+
|
|
|
+ if (!zip) {
|
|
|
+ // zip_t handler is not initialized
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ pzip = &(zip->archive);
|
|
|
+ if (pzip->m_zip_mode != MZ_ZIP_MODE_READING || zip->entry.index < 0) {
|
|
|
+ // the entry is not found or we do not have read access
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ idx = (mz_uint)zip->entry.index;
|
|
|
+ if (mz_zip_reader_is_file_a_directory(pzip, idx)) {
|
|
|
+ // the entry is a directory
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ *buf = mz_zip_reader_extract_to_heap(pzip, idx, &size, 0);
|
|
|
+ if (*buf && bufsize) {
|
|
|
+ *bufsize = size;
|
|
|
+ }
|
|
|
+ return size;
|
|
|
}
|
|
|
|
|
|
-int zip_entry_read(struct zip_t *zip, void **buf, size_t *bufsize) {
|
|
|
- mz_zip_archive *pzip = NULL;
|
|
|
- mz_uint idx;
|
|
|
+ssize_t zip_entry_noallocread(struct zip_t *zip, void *buf, size_t bufsize) {
|
|
|
+ mz_zip_archive *pzip = NULL;
|
|
|
|
|
|
- if (!zip) {
|
|
|
- // zip_t handler is not initialized
|
|
|
- return -1;
|
|
|
- }
|
|
|
+ if (!zip) {
|
|
|
+ // zip_t handler is not initialized
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
|
|
|
- if (zip->mode != 'r' || zip->entry.index < 0) {
|
|
|
- // the entry is not found or we do not have read access
|
|
|
- return -1;
|
|
|
- }
|
|
|
+ pzip = &(zip->archive);
|
|
|
+ if (pzip->m_zip_mode != MZ_ZIP_MODE_READING || zip->entry.index < 0) {
|
|
|
+ // the entry is not found or we do not have read access
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
|
|
|
- pzip = &(zip->archive);
|
|
|
- idx = (mz_uint)zip->entry.index;
|
|
|
- if (mz_zip_reader_is_file_a_directory(pzip, idx)) {
|
|
|
- // the entry is a directory
|
|
|
- return -1;
|
|
|
- }
|
|
|
+ if (!mz_zip_reader_extract_to_mem_no_alloc(pzip, (mz_uint)zip->entry.index,
|
|
|
+ buf, bufsize, 0, NULL, 0)) {
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
|
|
|
- *buf = mz_zip_reader_extract_to_heap(pzip, idx, bufsize, 0);
|
|
|
- return (*buf) ? 0 : -1;
|
|
|
+ return (ssize_t)zip->entry.uncomp_size;
|
|
|
}
|
|
|
|
|
|
int zip_entry_fread(struct zip_t *zip, const char *filename) {
|
|
|
- mz_zip_archive *pzip = NULL;
|
|
|
- mz_uint idx;
|
|
|
-
|
|
|
- if (!zip) {
|
|
|
- // zip_t handler is not initialized
|
|
|
- return -1;
|
|
|
- }
|
|
|
+ mz_zip_archive *pzip = NULL;
|
|
|
+ mz_uint idx;
|
|
|
+ mz_uint32 xattr = 0;
|
|
|
+ mz_zip_archive_file_stat info;
|
|
|
+
|
|
|
+ if (!zip) {
|
|
|
+ // zip_t handler is not initialized
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ memset((void *)&info, 0, sizeof(mz_zip_archive_file_stat));
|
|
|
+ pzip = &(zip->archive);
|
|
|
+ if (pzip->m_zip_mode != MZ_ZIP_MODE_READING || zip->entry.index < 0) {
|
|
|
+ // the entry is not found or we do not have read access
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ idx = (mz_uint)zip->entry.index;
|
|
|
+ if (mz_zip_reader_is_file_a_directory(pzip, idx)) {
|
|
|
+ // the entry is a directory
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (!mz_zip_reader_extract_to_file(pzip, idx, filename, 0)) {
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+#if defined(_MSC_VER)
|
|
|
+#else
|
|
|
+ if (!mz_zip_reader_file_stat(pzip, idx, &info)) {
|
|
|
+ // Cannot get information about zip archive;
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
|
|
|
- if (zip->mode != 'r' || zip->entry.index < 0) {
|
|
|
- // the entry is not found or we do not have read access
|
|
|
- return -1;
|
|
|
- }
|
|
|
-
|
|
|
- pzip = &(zip->archive);
|
|
|
- idx = (mz_uint)zip->entry.index;
|
|
|
- if (mz_zip_reader_is_file_a_directory(pzip, idx)) {
|
|
|
- // the entry is a directory
|
|
|
- return -1;
|
|
|
+ xattr = (info.m_external_attr >> 16) & 0xFFFF;
|
|
|
+ if (xattr > 0) {
|
|
|
+ if (chmod(filename, (mode_t)xattr) < 0) {
|
|
|
+ return -1;
|
|
|
}
|
|
|
+ }
|
|
|
+#endif
|
|
|
|
|
|
- return (mz_zip_reader_extract_to_file(pzip, idx, filename, 0)) ? 0 : -1;
|
|
|
+ return 0;
|
|
|
}
|
|
|
|
|
|
int zip_entry_extract(struct zip_t *zip,
|
|
|
size_t (*on_extract)(void *arg, unsigned long long offset,
|
|
|
const void *buf, size_t bufsize),
|
|
|
void *arg) {
|
|
|
- mz_zip_archive *pzip = NULL;
|
|
|
- mz_uint idx;
|
|
|
-
|
|
|
- if (!zip) {
|
|
|
- // zip_t handler is not initialized
|
|
|
- return -1;
|
|
|
- }
|
|
|
-
|
|
|
- if (zip->mode != 'r' || zip->entry.index < 0) {
|
|
|
- // the entry is not found or we do not have read access
|
|
|
- return -1;
|
|
|
- }
|
|
|
+ mz_zip_archive *pzip = NULL;
|
|
|
+ mz_uint idx;
|
|
|
+
|
|
|
+ if (!zip) {
|
|
|
+ // zip_t handler is not initialized
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ pzip = &(zip->archive);
|
|
|
+ if (pzip->m_zip_mode != MZ_ZIP_MODE_READING || zip->entry.index < 0) {
|
|
|
+ // the entry is not found or we do not have read access
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ idx = (mz_uint)zip->entry.index;
|
|
|
+ return (mz_zip_reader_extract_to_callback(pzip, idx, on_extract, arg, 0))
|
|
|
+ ? 0
|
|
|
+ : -1;
|
|
|
+}
|
|
|
|
|
|
- pzip = &(zip->archive);
|
|
|
- idx = (mz_uint)zip->entry.index;
|
|
|
+int zip_total_entries(struct zip_t *zip) {
|
|
|
+ if (!zip) {
|
|
|
+ // zip_t handler is not initialized
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
|
|
|
- return (mz_zip_reader_extract_to_callback(pzip, idx, on_extract, arg, 0))
|
|
|
- ? 0
|
|
|
- : -1;
|
|
|
+ return (int)zip->archive.m_total_files;
|
|
|
}
|
|
|
|
|
|
int zip_create(const char *zipname, const char *filenames[], size_t len) {
|
|
|
- int status = 0;
|
|
|
- size_t i;
|
|
|
- mz_zip_archive zip_archive;
|
|
|
+ int status = 0;
|
|
|
+ size_t i;
|
|
|
+ mz_zip_archive zip_archive;
|
|
|
+ struct MZ_FILE_STAT_STRUCT file_stat;
|
|
|
+ mz_uint32 ext_attributes = 0;
|
|
|
|
|
|
- if (!zipname || strlen(zipname) < 1) {
|
|
|
- // zip_t archive name is empty or NULL
|
|
|
- return -1;
|
|
|
- }
|
|
|
+ if (!zipname || strlen(zipname) < 1) {
|
|
|
+ // zip_t archive name is empty or NULL
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
|
|
|
- // Create a new archive.
|
|
|
- if (!memset(&(zip_archive), 0, sizeof(zip_archive))) {
|
|
|
- // Cannot memset zip archive
|
|
|
- return -1;
|
|
|
- }
|
|
|
+ // Create a new archive.
|
|
|
+ if (!memset(&(zip_archive), 0, sizeof(zip_archive))) {
|
|
|
+ // Cannot memset zip archive
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
|
|
|
- if (!mz_zip_writer_init_file(&zip_archive, zipname, 0)) {
|
|
|
- // Cannot initialize zip_archive writer
|
|
|
- return -1;
|
|
|
- }
|
|
|
+ if (!mz_zip_writer_init_file(&zip_archive, zipname, 0)) {
|
|
|
+ // Cannot initialize zip_archive writer
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
|
|
|
- for (i = 0; i < len; ++i) {
|
|
|
- const char *name = filenames[i];
|
|
|
- if (!name) {
|
|
|
- status = -1;
|
|
|
- break;
|
|
|
- }
|
|
|
+ memset((void *)&file_stat, 0, sizeof(struct MZ_FILE_STAT_STRUCT));
|
|
|
|
|
|
- if (!mz_zip_writer_add_file(&zip_archive, basename(name), name, "", 0,
|
|
|
- ZIP_DEFAULT_COMPRESSION_LEVEL)) {
|
|
|
- // Cannot add file to zip_archive
|
|
|
- status = -1;
|
|
|
- break;
|
|
|
- }
|
|
|
+ for (i = 0; i < len; ++i) {
|
|
|
+ const char *name = filenames[i];
|
|
|
+ if (!name) {
|
|
|
+ status = -1;
|
|
|
+ break;
|
|
|
}
|
|
|
|
|
|
- mz_zip_writer_finalize_archive(&zip_archive);
|
|
|
- mz_zip_writer_end(&zip_archive);
|
|
|
- return status;
|
|
|
-}
|
|
|
-
|
|
|
-int zip_extract(const char *zipname, const char *dir,
|
|
|
- int (*on_extract)(const char *filename, void *arg), void *arg) {
|
|
|
- int status = -1;
|
|
|
- mz_uint i, n;
|
|
|
- char path[MAX_PATH + 1] = {0};
|
|
|
- mz_zip_archive zip_archive;
|
|
|
- mz_zip_archive_file_stat info;
|
|
|
- size_t dirlen = 0;
|
|
|
-
|
|
|
- if (!memset(&(zip_archive), 0, sizeof(zip_archive))) {
|
|
|
- // Cannot memset zip archive
|
|
|
- return -1;
|
|
|
+ if (MZ_FILE_STAT(name, &file_stat) != 0) {
|
|
|
+ // problem getting information - check errno
|
|
|
+ return -1;
|
|
|
}
|
|
|
|
|
|
- if (!zipname || !dir) {
|
|
|
- // Cannot parse zip archive name
|
|
|
- return -1;
|
|
|
+ if ((file_stat.st_mode & 0200) == 0) {
|
|
|
+ // MS-DOS read-only attribute
|
|
|
+ ext_attributes |= 0x01;
|
|
|
}
|
|
|
+ ext_attributes |= (mz_uint32)((file_stat.st_mode & 0xFFFF) << 16);
|
|
|
|
|
|
- dirlen = strlen(dir);
|
|
|
- if (dirlen + 1 > MAX_PATH) {
|
|
|
- return -1;
|
|
|
+ if (!mz_zip_writer_add_file(&zip_archive, base_name(name), name, "", 0,
|
|
|
+ ZIP_DEFAULT_COMPRESSION_LEVEL,
|
|
|
+ ext_attributes)) {
|
|
|
+ // Cannot add file to zip_archive
|
|
|
+ status = -1;
|
|
|
+ break;
|
|
|
}
|
|
|
+ }
|
|
|
|
|
|
- // Now try to open the archive.
|
|
|
- if (!mz_zip_reader_init_file(&zip_archive, zipname, 0)) {
|
|
|
- // Cannot initialize zip_archive reader
|
|
|
- return -1;
|
|
|
- }
|
|
|
+ mz_zip_writer_finalize_archive(&zip_archive);
|
|
|
+ mz_zip_writer_end(&zip_archive);
|
|
|
+ return status;
|
|
|
+}
|
|
|
|
|
|
- strcpy(path, dir);
|
|
|
- if (!ISSLASH(path[dirlen - 1])) {
|
|
|
-#if defined _WIN32 || defined __WIN32__
|
|
|
- path[dirlen] = '\\';
|
|
|
+int zip_extract(const char *zipname, const char *dir,
|
|
|
+ int (*on_extract)(const char *filename, void *arg), void *arg) {
|
|
|
+ int status = -1;
|
|
|
+ mz_uint i, n;
|
|
|
+ char path[MAX_PATH + 1];
|
|
|
+ char symlink_to[MAX_PATH + 1];
|
|
|
+ mz_zip_archive zip_archive;
|
|
|
+ mz_zip_archive_file_stat info;
|
|
|
+ size_t dirlen = 0;
|
|
|
+ mz_uint32 xattr = 0;
|
|
|
+
|
|
|
+ memset(path, 0, sizeof(path));
|
|
|
+ memset(symlink_to, 0, sizeof(symlink_to));
|
|
|
+ if (!memset(&(zip_archive), 0, sizeof(zip_archive))) {
|
|
|
+ // Cannot memset zip archive
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (!zipname || !dir) {
|
|
|
+ // Cannot parse zip archive name
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ dirlen = strlen(dir);
|
|
|
+ if (dirlen + 1 > MAX_PATH) {
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ // Now try to open the archive.
|
|
|
+ if (!mz_zip_reader_init_file(&zip_archive, zipname, 0)) {
|
|
|
+ // Cannot initialize zip_archive reader
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ memset((void *)&info, 0, sizeof(mz_zip_archive_file_stat));
|
|
|
+
|
|
|
+#if defined(_MSC_VER)
|
|
|
+ strcpy_s(path, MAX_PATH, dir);
|
|
|
#else
|
|
|
- path[dirlen] = '/';
|
|
|
+ strcpy(path, dir);
|
|
|
#endif
|
|
|
- ++dirlen;
|
|
|
- }
|
|
|
|
|
|
- // Get and print information about each file in the archive.
|
|
|
- n = mz_zip_reader_get_num_files(&zip_archive);
|
|
|
- for (i = 0; i < n; ++i) {
|
|
|
- if (!mz_zip_reader_file_stat(&zip_archive, i, &info)) {
|
|
|
- // Cannot get information about zip archive;
|
|
|
- goto out;
|
|
|
- }
|
|
|
- strncpy(&path[dirlen], info.m_filename, MAX_PATH - dirlen);
|
|
|
- if (mkpath(path) < 0) {
|
|
|
- // Cannot make a path
|
|
|
- goto out;
|
|
|
+ if (!ISSLASH(path[dirlen - 1])) {
|
|
|
+#if defined(_WIN32) || defined(__WIN32__)
|
|
|
+ path[dirlen] = '\\';
|
|
|
+#else
|
|
|
+ path[dirlen] = '/';
|
|
|
+#endif
|
|
|
+ ++dirlen;
|
|
|
+ }
|
|
|
+
|
|
|
+ // Get and print information about each file in the archive.
|
|
|
+ n = mz_zip_reader_get_num_files(&zip_archive);
|
|
|
+ for (i = 0; i < n; ++i) {
|
|
|
+ if (!mz_zip_reader_file_stat(&zip_archive, i, &info)) {
|
|
|
+ // Cannot get information about zip archive;
|
|
|
+ goto out;
|
|
|
+ }
|
|
|
+#if defined(_MSC_VER)
|
|
|
+ strncpy_s(&path[dirlen], MAX_PATH - dirlen, info.m_filename,
|
|
|
+ MAX_PATH - dirlen);
|
|
|
+#else
|
|
|
+ strncpy(&path[dirlen], info.m_filename, MAX_PATH - dirlen);
|
|
|
+#endif
|
|
|
+ if (mkpath(path) < 0) {
|
|
|
+ // Cannot make a path
|
|
|
+ goto out;
|
|
|
+ }
|
|
|
+
|
|
|
+ if ((((info.m_version_made_by >> 8) == 3) || ((info.m_version_made_by >> 8) == 19)) // if zip is produced on Unix or macOS (3 and 19 from section 4.4.2.2 of zip standard)
|
|
|
+ && info.m_external_attr & (0x20 << 24)) { // and has sym link attribute (0x80 is file, 0x40 is directory)
|
|
|
+#if defined(_WIN32) || defined(__WIN32__) || defined(_MSC_VER) || \
|
|
|
+ defined(__MINGW32__)
|
|
|
+#else
|
|
|
+ if (info.m_uncomp_size > MAX_PATH || !mz_zip_reader_extract_to_mem_no_alloc(&zip_archive, i, symlink_to, MAX_PATH, 0, NULL, 0)) {
|
|
|
+ goto out;
|
|
|
+ }
|
|
|
+ symlink_to[info.m_uncomp_size] = '\0';
|
|
|
+ if (symlink(symlink_to, path) != 0) {
|
|
|
+ goto out;
|
|
|
+ }
|
|
|
+#endif
|
|
|
+ } else {
|
|
|
+ if (!mz_zip_reader_is_file_a_directory(&zip_archive, i)) {
|
|
|
+ if (!mz_zip_reader_extract_to_file(&zip_archive, i, path, 0)) {
|
|
|
+ // Cannot extract zip archive to file
|
|
|
+ goto out;
|
|
|
}
|
|
|
+ }
|
|
|
|
|
|
- if (!mz_zip_reader_is_file_a_directory(&zip_archive, i)) {
|
|
|
- if (!mz_zip_reader_extract_to_file(&zip_archive, i, path, 0)) {
|
|
|
- // Cannot extract zip archive to file
|
|
|
- goto out;
|
|
|
- }
|
|
|
+#if defined(_MSC_VER)
|
|
|
+#else
|
|
|
+ xattr = (info.m_external_attr >> 16) & 0xFFFF;
|
|
|
+ if (xattr > 0) {
|
|
|
+ if (chmod(path, (mode_t)xattr) < 0) {
|
|
|
+ goto out;
|
|
|
}
|
|
|
+ }
|
|
|
+#endif
|
|
|
+ }
|
|
|
|
|
|
- if (on_extract) {
|
|
|
- if (on_extract(path, arg) < 0) {
|
|
|
- goto out;
|
|
|
- }
|
|
|
- }
|
|
|
+ if (on_extract) {
|
|
|
+ if (on_extract(path, arg) < 0) {
|
|
|
+ goto out;
|
|
|
+ }
|
|
|
}
|
|
|
- status = 0;
|
|
|
+ }
|
|
|
+ status = 0;
|
|
|
|
|
|
out:
|
|
|
- // Close the archive, freeing any resources it was using
|
|
|
- if (!mz_zip_reader_end(&zip_archive)) {
|
|
|
- // Cannot end zip reader
|
|
|
- status = -1;
|
|
|
- }
|
|
|
+ // Close the archive, freeing any resources it was using
|
|
|
+ if (!mz_zip_reader_end(&zip_archive)) {
|
|
|
+ // Cannot end zip reader
|
|
|
+ status = -1;
|
|
|
+ }
|
|
|
|
|
|
- return status;
|
|
|
+ return status;
|
|
|
}
|