| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | #define __STDC_WANT_LIB_EXT1__ 1 |
| |
|
| | #include <errno.h> |
| | #include <sys/stat.h> |
| | #include <time.h> |
| |
|
| | #if defined(_WIN32) || defined(__WIN32__) || defined(_MSC_VER) || \ |
| | defined(__MINGW32__) |
| | |
| | #include <direct.h> |
| |
|
| | #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] == ':') |
| | #define FILESYSTEM_PREFIX_LEN(P) (HAS_DEVICE(P) ? 2 : 0) |
| |
|
| | #else |
| |
|
| | #include <unistd.h> |
| | #define STRCLONE(STR) ((STR) ? strdup(STR) : NULL) |
| |
|
| | #endif |
| |
|
| | #ifdef __MINGW32__ |
| | #include <sys/types.h> |
| | #include <unistd.h> |
| | #endif |
| |
|
| | #define USE_EXTERNAL_MZCRC |
| | #include "miniz.h" |
| | #include "zip.h" |
| |
|
| | #ifdef _MSC_VER |
| | #include <io.h> |
| |
|
| | #define ftruncate(fd, sz) (-(_chsize_s((fd), (sz)) != 0)) |
| | #define fileno _fileno |
| | #endif |
| |
|
| | #if defined(__TINYC__) && (defined(_WIN32) || defined(_WIN64)) |
| | #include <io.h> |
| |
|
| | #define ftruncate(fd, sz) (-(_chsize_s((fd), (sz)) != 0)) |
| | #define fileno _fileno |
| | #endif |
| |
|
| | #ifndef HAS_DEVICE |
| | #define HAS_DEVICE(P) 0 |
| | #endif |
| |
|
| | #ifndef FILESYSTEM_PREFIX_LEN |
| | #define FILESYSTEM_PREFIX_LEN(P) 0 |
| | #endif |
| |
|
| | #ifndef ISSLASH |
| | #define ISSLASH(C) ((C) == '/' || (C) == '\\') |
| | #endif |
| |
|
| | #define CLEANUP(ptr) \ |
| | do { \ |
| | if (ptr) { \ |
| | free((void *)ptr); \ |
| | ptr = NULL; \ |
| | } \ |
| | } while (0) |
| |
|
| | #define UNX_IFDIR 0040000 |
| | #define UNX_IFREG 0100000 |
| | #define UNX_IFSOCK 0140000 |
| | #define UNX_IFLNK 0120000 |
| | #define UNX_IFBLK 0060000 |
| | #define UNX_IFCHR 0020000 |
| | #define UNX_IFIFO 0010000 |
| |
|
| | struct zip_entry_t { |
| | ssize_t 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; |
| | }; |
| |
|
| | enum zip_modify_t { |
| | MZ_KEEP = 0, |
| | MZ_DELETE = 1, |
| | MZ_MOVE = 2, |
| | }; |
| |
|
| | struct zip_entry_mark_t { |
| | ssize_t file_index; |
| | enum zip_modify_t type; |
| | mz_uint64 m_local_header_ofs; |
| | size_t lf_length; |
| | }; |
| |
|
| | static const char *const zip_errlist[33] = { |
| | NULL, |
| | "not initialized\0", |
| | "invalid entry name\0", |
| | "entry not found\0", |
| | "invalid zip mode\0", |
| | "invalid compression level\0", |
| | "no zip 64 support\0", |
| | "memset error\0", |
| | "cannot write data to entry\0", |
| | "cannot initialize tdefl compressor\0", |
| | "invalid index\0", |
| | "header not found\0", |
| | "cannot flush tdefl buffer\0", |
| | "cannot write entry header\0", |
| | "cannot create entry header\0", |
| | "cannot write to central dir\0", |
| | "cannot open file\0", |
| | "invalid entry type\0", |
| | "extracting data using no memory allocation\0", |
| | "file not found\0", |
| | "no permission\0", |
| | "out of memory\0", |
| | "invalid zip archive name\0", |
| | "make dir error\0", |
| | "symlink error\0", |
| | "close archive error\0", |
| | "capacity size too small\0", |
| | "fseek error\0", |
| | "fread error\0", |
| | "fwrite error\0", |
| | "cannot initialize reader\0", |
| | "cannot initialize writer\0", |
| | "cannot initialize writer from reader\0", |
| | }; |
| |
|
| | const char *zip_strerror(int errnum) { |
| | errnum = -errnum; |
| | if (errnum <= 0 || errnum >= 33) { |
| | return NULL; |
| | } |
| |
|
| | return zip_errlist[errnum]; |
| | } |
| |
|
| | static const char *zip_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 (*base == '\0' && ISSLASH(*name) && all_slashes) |
| | --base; |
| |
|
| | return base; |
| | } |
| |
|
| | static int zip_mkpath(char *path) { |
| | char *p; |
| | char npath[MZ_ZIP_MAX_ARCHIVE_FILENAME_SIZE + 1]; |
| | int len = 0; |
| | int has_device = HAS_DEVICE(path); |
| |
|
| | memset(npath, 0, MZ_ZIP_MAX_ARCHIVE_FILENAME_SIZE + 1); |
| | if (has_device) { |
| | |
| | npath[0] = path[0]; |
| | npath[1] = path[1]; |
| | len = 2; |
| | } |
| | for (p = path + len; *p && len < MZ_ZIP_MAX_ARCHIVE_FILENAME_SIZE; p++) { |
| | if (ISSLASH(*p) && ((!has_device && len > 0) || (has_device && len > 2))) { |
| | #if defined(_WIN32) || defined(__WIN32__) || defined(_MSC_VER) || \ |
| | defined(__MINGW32__) |
| | #else |
| | if ('\\' == *p) { |
| | *p = '/'; |
| | } |
| | #endif |
| |
|
| | if (MZ_MKDIR(npath) == -1) { |
| | if (errno != EEXIST) { |
| | return ZIP_EMKDIR; |
| | } |
| | } |
| | } |
| | npath[len++] = *p; |
| | } |
| |
|
| | return 0; |
| | } |
| |
|
| | static char *zip_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++ = c; |
| | } |
| |
|
| | return begin; |
| | } |
| |
|
| | static char *zip_name_normalize(char *name, char *const nname, size_t len) { |
| | size_t offn = 0; |
| | size_t offnn = 0, ncpy = 0; |
| |
|
| | if (name == NULL || nname == NULL || len <= 0) { |
| | return NULL; |
| | } |
| | |
| | while (ISSLASH(*name)) |
| | name++; |
| |
|
| | for (; offn < len; offn++) { |
| | if (ISSLASH(name[offn])) { |
| | if (ncpy > 0 && strcmp(&nname[offnn], ".\0") && |
| | strcmp(&nname[offnn], "..\0")) { |
| | offnn += ncpy; |
| | nname[offnn++] = name[offn]; |
| | } |
| | ncpy = 0; |
| | } else { |
| | nname[offnn + ncpy] = name[offn]; |
| | ncpy++; |
| | } |
| | } |
| |
|
| | |
| | if (ncpy == 0 || !strcmp(&nname[offnn], ".\0") || |
| | !strcmp(&nname[offnn], "..\0")) { |
| | nname[offnn] = 0; |
| | } |
| | return nname; |
| | } |
| |
|
| | static mz_bool zip_name_match(const char *name1, const char *name2) { |
| | char *nname2 = NULL; |
| |
|
| | #ifdef ZIP_RAW_ENTRYNAME |
| | nname2 = STRCLONE(name2); |
| | #else |
| | nname2 = zip_strrpl(name2, strlen(name2), '\\', '/'); |
| | #endif |
| |
|
| | if (!nname2) { |
| | return MZ_FALSE; |
| | } |
| |
|
| | mz_bool res = (strcmp(name1, nname2) == 0) ? MZ_TRUE : MZ_FALSE; |
| | CLEANUP(nname2); |
| | return res; |
| | } |
| |
|
| | static int zip_archive_truncate(mz_zip_archive *pzip) { |
| | mz_zip_internal_state *pState = pzip->m_pState; |
| | mz_uint64 file_size = pzip->m_archive_size; |
| | if ((pzip->m_pWrite == mz_zip_heap_write_func) && (pState->m_pMem)) { |
| | return 0; |
| | } |
| | if (pzip->m_zip_mode == MZ_ZIP_MODE_WRITING_HAS_BEEN_FINALIZED) { |
| | if (pState->m_pFile) { |
| | int fd = fileno(pState->m_pFile); |
| | return ftruncate(fd, file_size); |
| | } |
| | } |
| | return 0; |
| | } |
| |
|
| | static int zip_archive_extract(mz_zip_archive *zip_archive, const char *dir, |
| | int (*on_extract)(const char *filename, |
| | void *arg), |
| | void *arg) { |
| | int err = 0; |
| | mz_uint i, n; |
| | char path[MZ_ZIP_MAX_ARCHIVE_FILENAME_SIZE + 1]; |
| | char symlink_to[MZ_ZIP_MAX_ARCHIVE_FILENAME_SIZE + 1]; |
| | mz_zip_archive_file_stat info; |
| | size_t dirlen = 0, filename_size = MZ_ZIP_MAX_ARCHIVE_FILENAME_SIZE; |
| | mz_uint32 xattr = 0; |
| |
|
| | memset(path, 0, sizeof(path)); |
| | memset(symlink_to, 0, sizeof(symlink_to)); |
| |
|
| | dirlen = strlen(dir); |
| | if (dirlen + 1 > MZ_ZIP_MAX_ARCHIVE_FILENAME_SIZE) { |
| | return ZIP_EINVENTNAME; |
| | } |
| |
|
| | memset((void *)&info, 0, sizeof(mz_zip_archive_file_stat)); |
| |
|
| | #if defined(_MSC_VER) |
| | strcpy_s(path, MZ_ZIP_MAX_ARCHIVE_FILENAME_SIZE, dir); |
| | #else |
| | strcpy(path, dir); |
| | #endif |
| |
|
| | if (!ISSLASH(path[dirlen - 1])) { |
| | #if defined(_WIN32) || defined(__WIN32__) |
| | path[dirlen] = '\\'; |
| | #else |
| | path[dirlen] = '/'; |
| | #endif |
| | ++dirlen; |
| | } |
| |
|
| | if (filename_size > MZ_ZIP_MAX_ARCHIVE_FILENAME_SIZE - dirlen) { |
| | filename_size = MZ_ZIP_MAX_ARCHIVE_FILENAME_SIZE - dirlen; |
| | } |
| | |
| | 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)) { |
| | |
| | err = ZIP_ENOENT; |
| | goto out; |
| | } |
| |
|
| | if (!zip_name_normalize(info.m_filename, info.m_filename, |
| | strlen(info.m_filename))) { |
| | |
| | err = ZIP_EINVENTNAME; |
| | goto out; |
| | } |
| |
|
| | #if defined(_MSC_VER) |
| | strncpy_s(&path[dirlen], filename_size, info.m_filename, filename_size); |
| | #else |
| | strncpy(&path[dirlen], info.m_filename, filename_size); |
| | #endif |
| | err = zip_mkpath(path); |
| | if (err < 0) { |
| | |
| | goto out; |
| | } |
| |
|
| | if ((((info.m_version_made_by >> 8) == 3) || |
| | ((info.m_version_made_by >> 8) == |
| | 19)) |
| | |
| | && info.m_external_attr & |
| | (0x20 << 24)) { |
| | |
| | #if defined(_WIN32) || defined(__WIN32__) || defined(_MSC_VER) || \ |
| | defined(__MINGW32__) |
| | #else |
| | if (info.m_uncomp_size > MZ_ZIP_MAX_ARCHIVE_FILENAME_SIZE || |
| | !mz_zip_reader_extract_to_mem_no_alloc( |
| | zip_archive, i, symlink_to, MZ_ZIP_MAX_ARCHIVE_FILENAME_SIZE, 0, |
| | NULL, 0)) { |
| | err = ZIP_EMEMNOALLOC; |
| | goto out; |
| | } |
| | symlink_to[info.m_uncomp_size] = '\0'; |
| | if (symlink(symlink_to, path) != 0) { |
| | err = ZIP_ESYMLINK; |
| | 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)) { |
| | |
| | err = ZIP_ENOFILE; |
| | goto out; |
| | } |
| | } |
| |
|
| | #if defined(_MSC_VER) || defined(PS4) |
| | (void)xattr; |
| | #else |
| | xattr = (info.m_external_attr >> 16) & 0xFFFF; |
| | if (xattr > 0 && xattr <= MZ_UINT16_MAX) { |
| | if (CHMOD(path, (mode_t)xattr) < 0) { |
| | err = ZIP_ENOPERM; |
| | goto out; |
| | } |
| | } |
| | #endif |
| | } |
| |
|
| | if (on_extract) { |
| | if (on_extract(path, arg) < 0) { |
| | goto out; |
| | } |
| | } |
| | } |
| |
|
| | out: |
| | |
| | if (!mz_zip_reader_end(zip_archive)) { |
| | |
| | err = ZIP_ECLSZIP; |
| | } |
| | return err; |
| | } |
| |
|
| | static inline void zip_archive_finalize(mz_zip_archive *pzip) { |
| | mz_zip_writer_finalize_archive(pzip); |
| | zip_archive_truncate(pzip); |
| | } |
| |
|
| | static ssize_t zip_entry_mark(struct zip_t *zip, |
| | struct zip_entry_mark_t *entry_mark, |
| | const ssize_t n, char *const entries[], |
| | const size_t len) { |
| | ssize_t i = 0; |
| | ssize_t err = 0; |
| | if (!zip || !entry_mark || !entries) { |
| | return ZIP_ENOINIT; |
| | } |
| |
|
| | mz_zip_archive_file_stat file_stat; |
| | mz_uint64 d_pos = UINT64_MAX; |
| | for (i = 0; i < n; ++i) { |
| | if ((err = zip_entry_openbyindex(zip, i))) { |
| | return (ssize_t)err; |
| | } |
| |
|
| | mz_bool name_matches = MZ_FALSE; |
| | { |
| | size_t j; |
| | for (j = 0; j < len; ++j) { |
| | if (zip_name_match(zip->entry.name, entries[j])) { |
| | name_matches = MZ_TRUE; |
| | break; |
| | } |
| | } |
| | } |
| | if (name_matches) { |
| | entry_mark[i].type = MZ_DELETE; |
| | } else { |
| | entry_mark[i].type = MZ_KEEP; |
| | } |
| |
|
| | if (!mz_zip_reader_file_stat(&zip->archive, i, &file_stat)) { |
| | return ZIP_ENOENT; |
| | } |
| |
|
| | zip_entry_close(zip); |
| |
|
| | entry_mark[i].m_local_header_ofs = file_stat.m_local_header_ofs; |
| | entry_mark[i].file_index = (ssize_t)-1; |
| | entry_mark[i].lf_length = 0; |
| | if ((entry_mark[i].type) == MZ_DELETE && |
| | (d_pos > entry_mark[i].m_local_header_ofs)) { |
| | d_pos = entry_mark[i].m_local_header_ofs; |
| | } |
| | } |
| |
|
| | for (i = 0; i < n; ++i) { |
| | if ((entry_mark[i].m_local_header_ofs > d_pos) && |
| | (entry_mark[i].type != MZ_DELETE)) { |
| | entry_mark[i].type = MZ_MOVE; |
| | } |
| | } |
| | return err; |
| | } |
| |
|
| | static ssize_t zip_index_next(mz_uint64 *local_header_ofs_array, |
| | ssize_t cur_index) { |
| | ssize_t new_index = 0, i; |
| | for (i = cur_index - 1; i >= 0; --i) { |
| | if (local_header_ofs_array[cur_index] > local_header_ofs_array[i]) { |
| | new_index = i + 1; |
| | return new_index; |
| | } |
| | } |
| | return new_index; |
| | } |
| |
|
| | static ssize_t zip_sort(mz_uint64 *local_header_ofs_array, ssize_t cur_index) { |
| | ssize_t nxt_index = zip_index_next(local_header_ofs_array, cur_index); |
| |
|
| | if (nxt_index != cur_index) { |
| | mz_uint64 temp = local_header_ofs_array[cur_index]; |
| | ssize_t i; |
| | for (i = cur_index; i > nxt_index; i--) { |
| | local_header_ofs_array[i] = local_header_ofs_array[i - 1]; |
| | } |
| | local_header_ofs_array[nxt_index] = temp; |
| | } |
| | return nxt_index; |
| | } |
| |
|
| | static int zip_index_update(struct zip_entry_mark_t *entry_mark, |
| | ssize_t last_index, ssize_t nxt_index) { |
| | ssize_t j; |
| | for (j = 0; j < last_index; j++) { |
| | if (entry_mark[j].file_index >= nxt_index) { |
| | entry_mark[j].file_index += 1; |
| | } |
| | } |
| | entry_mark[nxt_index].file_index = last_index; |
| | return 0; |
| | } |
| |
|
| | static int zip_entry_finalize(struct zip_t *zip, |
| | struct zip_entry_mark_t *entry_mark, |
| | const ssize_t n) { |
| |
|
| | ssize_t i = 0; |
| | mz_uint64 *local_header_ofs_array = (mz_uint64 *)calloc(n, sizeof(mz_uint64)); |
| | if (!local_header_ofs_array) { |
| | return ZIP_EOOMEM; |
| | } |
| |
|
| | for (i = 0; i < n; ++i) { |
| | local_header_ofs_array[i] = entry_mark[i].m_local_header_ofs; |
| | ssize_t index = zip_sort(local_header_ofs_array, i); |
| |
|
| | if (index != i) { |
| | zip_index_update(entry_mark, i, index); |
| | } |
| | entry_mark[i].file_index = index; |
| | } |
| |
|
| | size_t *length = (size_t *)calloc(n, sizeof(size_t)); |
| | if (!length) { |
| | CLEANUP(local_header_ofs_array); |
| | return ZIP_EOOMEM; |
| | } |
| | for (i = 0; i < n - 1; i++) { |
| | length[i] = |
| | (size_t)(local_header_ofs_array[i + 1] - local_header_ofs_array[i]); |
| | } |
| | length[n - 1] = |
| | (size_t)(zip->archive.m_archive_size - local_header_ofs_array[n - 1]); |
| |
|
| | for (i = 0; i < n; i++) { |
| | entry_mark[i].lf_length = length[entry_mark[i].file_index]; |
| | } |
| |
|
| | CLEANUP(length); |
| | CLEANUP(local_header_ofs_array); |
| | return 0; |
| | } |
| |
|
| | static ssize_t zip_entry_set(struct zip_t *zip, |
| | struct zip_entry_mark_t *entry_mark, ssize_t n, |
| | char *const entries[], const size_t len) { |
| | ssize_t err = 0; |
| |
|
| | if ((err = zip_entry_mark(zip, entry_mark, n, entries, len)) < 0) { |
| | return err; |
| | } |
| | if ((err = zip_entry_finalize(zip, entry_mark, n)) < 0) { |
| | return err; |
| | } |
| | return 0; |
| | } |
| |
|
| | static ssize_t zip_file_move(MZ_FILE *m_pFile, const mz_uint64 to, |
| | const mz_uint64 from, const size_t length, |
| | mz_uint8 *move_buf, const size_t capacity_size) { |
| | if (length > capacity_size) { |
| | return ZIP_ECAPSIZE; |
| | } |
| | if (MZ_FSEEK64(m_pFile, from, SEEK_SET)) { |
| | return ZIP_EFSEEK; |
| | } |
| | if (fread(move_buf, 1, length, m_pFile) != length) { |
| | return ZIP_EFREAD; |
| | } |
| | if (MZ_FSEEK64(m_pFile, to, SEEK_SET)) { |
| | return ZIP_EFSEEK; |
| | } |
| | if (fwrite(move_buf, 1, length, m_pFile) != length) { |
| | return ZIP_EFWRITE; |
| | } |
| | return (ssize_t)length; |
| | } |
| |
|
| | static ssize_t zip_files_move(MZ_FILE *m_pFile, mz_uint64 writen_num, |
| | mz_uint64 read_num, size_t length) { |
| | ssize_t n = 0; |
| | const size_t page_size = 1 << 12; |
| | mz_uint8 *move_buf = (mz_uint8 *)calloc(1, page_size); |
| | if (!move_buf) { |
| | return ZIP_EOOMEM; |
| | } |
| |
|
| | ssize_t moved_length = 0; |
| | ssize_t move_count = 0; |
| | while ((mz_int64)length > 0) { |
| | move_count = (length >= page_size) ? page_size : length; |
| | n = zip_file_move(m_pFile, writen_num, read_num, move_count, move_buf, |
| | page_size); |
| | if (n < 0) { |
| | moved_length = n; |
| | goto cleanup; |
| | } |
| |
|
| | if (n != move_count) { |
| | goto cleanup; |
| | } |
| |
|
| | writen_num += move_count; |
| | read_num += move_count; |
| | length -= move_count; |
| | moved_length += move_count; |
| | } |
| |
|
| | cleanup: |
| | CLEANUP(move_buf); |
| | return moved_length; |
| | } |
| |
|
| | static int zip_central_dir_move(mz_zip_internal_state *pState, int begin, |
| | int end, int entry_num) { |
| | if (begin == entry_num) { |
| | return 0; |
| | } |
| |
|
| | size_t l_size = 0; |
| | size_t r_size = 0; |
| | mz_uint32 d_size = 0; |
| | mz_uint8 *next = NULL; |
| | mz_uint8 *deleted = &MZ_ZIP_ARRAY_ELEMENT( |
| | &pState->m_central_dir, mz_uint8, |
| | MZ_ZIP_ARRAY_ELEMENT(&pState->m_central_dir_offsets, mz_uint32, begin)); |
| | l_size = (size_t)(deleted - (mz_uint8 *)(pState->m_central_dir.m_p)); |
| | if (end == entry_num) { |
| | r_size = 0; |
| | } else { |
| | next = &MZ_ZIP_ARRAY_ELEMENT( |
| | &pState->m_central_dir, mz_uint8, |
| | MZ_ZIP_ARRAY_ELEMENT(&pState->m_central_dir_offsets, mz_uint32, end)); |
| | r_size = pState->m_central_dir.m_size - |
| | (mz_uint32)(next - (mz_uint8 *)(pState->m_central_dir.m_p)); |
| | d_size = (mz_uint32)(next - deleted); |
| | } |
| |
|
| | if (next && l_size == 0) { |
| | memmove(pState->m_central_dir.m_p, next, r_size); |
| | pState->m_central_dir.m_p = MZ_REALLOC(pState->m_central_dir.m_p, r_size); |
| | { |
| | int i; |
| | for (i = end; i < entry_num; i++) { |
| | MZ_ZIP_ARRAY_ELEMENT(&pState->m_central_dir_offsets, mz_uint32, i) -= |
| | d_size; |
| | } |
| | } |
| | } |
| |
|
| | if (next && l_size * r_size != 0) { |
| | memmove(deleted, next, r_size); |
| | { |
| | int i; |
| | for (i = end; i < entry_num; i++) { |
| | MZ_ZIP_ARRAY_ELEMENT(&pState->m_central_dir_offsets, mz_uint32, i) -= |
| | d_size; |
| | } |
| | } |
| | } |
| |
|
| | pState->m_central_dir.m_size = l_size + r_size; |
| | return 0; |
| | } |
| |
|
| | static int zip_central_dir_delete(mz_zip_internal_state *pState, |
| | int *deleted_entry_index_array, |
| | int entry_num) { |
| | int i = 0; |
| | int begin = 0; |
| | int end = 0; |
| | int d_num = 0; |
| | while (i < entry_num) { |
| | while ((i < entry_num) && (!deleted_entry_index_array[i])) { |
| | i++; |
| | } |
| | begin = i; |
| |
|
| | while ((i < entry_num) && (deleted_entry_index_array[i])) { |
| | i++; |
| | } |
| | end = i; |
| | zip_central_dir_move(pState, begin, end, entry_num); |
| | } |
| |
|
| | i = 0; |
| | while (i < entry_num) { |
| | while ((i < entry_num) && (!deleted_entry_index_array[i])) { |
| | i++; |
| | } |
| | begin = i; |
| | if (begin == entry_num) { |
| | break; |
| | } |
| | while ((i < entry_num) && (deleted_entry_index_array[i])) { |
| | i++; |
| | } |
| | end = i; |
| | int k = 0, j; |
| | for (j = end; j < entry_num; j++) { |
| | MZ_ZIP_ARRAY_ELEMENT(&pState->m_central_dir_offsets, mz_uint32, |
| | begin + k) = |
| | (mz_uint32)MZ_ZIP_ARRAY_ELEMENT(&pState->m_central_dir_offsets, |
| | mz_uint32, j); |
| | k++; |
| | } |
| | d_num += end - begin; |
| | } |
| |
|
| | pState->m_central_dir_offsets.m_size = |
| | sizeof(mz_uint32) * (entry_num - d_num); |
| | return 0; |
| | } |
| |
|
| | static ssize_t zip_entries_delete_mark(struct zip_t *zip, |
| | struct zip_entry_mark_t *entry_mark, |
| | int entry_num) { |
| | mz_uint64 writen_num = 0; |
| | mz_uint64 read_num = 0; |
| | size_t deleted_length = 0; |
| | size_t move_length = 0; |
| | int i = 0; |
| | size_t deleted_entry_num = 0; |
| | ssize_t n = 0; |
| |
|
| | mz_bool *deleted_entry_flag_array = |
| | (mz_bool *)calloc(entry_num, sizeof(mz_bool)); |
| | if (deleted_entry_flag_array == NULL) { |
| | return ZIP_EOOMEM; |
| | } |
| |
|
| | mz_zip_internal_state *pState = zip->archive.m_pState; |
| | zip->archive.m_zip_mode = MZ_ZIP_MODE_WRITING; |
| |
|
| | if ((!pState->m_pFile) || MZ_FSEEK64(pState->m_pFile, 0, SEEK_SET)) { |
| | CLEANUP(deleted_entry_flag_array); |
| | return ZIP_ENOENT; |
| | } |
| |
|
| | while (i < entry_num) { |
| | while ((i < entry_num) && (entry_mark[i].type == MZ_KEEP)) { |
| | writen_num += entry_mark[i].lf_length; |
| | read_num = writen_num; |
| | i++; |
| | } |
| |
|
| | while ((i < entry_num) && (entry_mark[i].type == MZ_DELETE)) { |
| | deleted_entry_flag_array[i] = MZ_TRUE; |
| | read_num += entry_mark[i].lf_length; |
| | deleted_length += entry_mark[i].lf_length; |
| | i++; |
| | deleted_entry_num++; |
| | } |
| |
|
| | while ((i < entry_num) && (entry_mark[i].type == MZ_MOVE)) { |
| | move_length += entry_mark[i].lf_length; |
| | mz_uint8 *p = &MZ_ZIP_ARRAY_ELEMENT( |
| | &pState->m_central_dir, mz_uint8, |
| | MZ_ZIP_ARRAY_ELEMENT(&pState->m_central_dir_offsets, mz_uint32, i)); |
| | if (!p) { |
| | CLEANUP(deleted_entry_flag_array); |
| | return ZIP_ENOENT; |
| | } |
| | mz_uint32 offset = MZ_READ_LE32(p + MZ_ZIP_CDH_LOCAL_HEADER_OFS); |
| | offset -= (mz_uint32)deleted_length; |
| | MZ_WRITE_LE32(p + MZ_ZIP_CDH_LOCAL_HEADER_OFS, offset); |
| | i++; |
| | } |
| |
|
| | n = zip_files_move(pState->m_pFile, writen_num, read_num, move_length); |
| | if (n != (ssize_t)move_length) { |
| | CLEANUP(deleted_entry_flag_array); |
| | return n; |
| | } |
| | writen_num += move_length; |
| | read_num += move_length; |
| | } |
| |
|
| | zip->archive.m_archive_size -= (mz_uint64)deleted_length; |
| | zip->archive.m_total_files = |
| | (mz_uint32)entry_num - (mz_uint32)deleted_entry_num; |
| |
|
| | zip_central_dir_delete(pState, deleted_entry_flag_array, entry_num); |
| | CLEANUP(deleted_entry_flag_array); |
| |
|
| | return (ssize_t)deleted_entry_num; |
| | } |
| |
|
| | struct zip_t *zip_open(const char *zipname, int level, char mode) { |
| | int errnum = 0; |
| | return zip_openwitherror(zipname, level, mode, &errnum); |
| | } |
| |
|
| | struct zip_t *zip_openwitherror(const char *zipname, int level, char mode, |
| | int *errnum) { |
| | struct zip_t *zip = NULL; |
| | *errnum = 0; |
| |
|
| | if (!zipname || strlen(zipname) < 1) { |
| | |
| | *errnum = ZIP_EINVZIPNAME; |
| | goto cleanup; |
| | } |
| |
|
| | if (level < 0) |
| | level = MZ_DEFAULT_LEVEL; |
| | if ((level & 0xF) > MZ_UBER_COMPRESSION) { |
| | |
| | *errnum = ZIP_EINVLVL; |
| | goto cleanup; |
| | } |
| |
|
| | zip = (struct zip_t *)calloc((size_t)1, sizeof(struct zip_t)); |
| | if (!zip) { |
| | |
| | *errnum = ZIP_EOOMEM; |
| | goto cleanup; |
| | } |
| |
|
| | zip->level = (mz_uint)level; |
| | switch (mode) { |
| | case 'w': |
| | |
| | if (!mz_zip_writer_init_file_v2(&(zip->archive), zipname, 0, |
| | MZ_ZIP_FLAG_WRITE_ZIP64)) { |
| | |
| | *errnum = ZIP_EWINIT; |
| | goto cleanup; |
| | } |
| | break; |
| |
|
| | case 'r': |
| | if (!mz_zip_reader_init_file_v2( |
| | &(zip->archive), zipname, |
| | zip->level | MZ_ZIP_FLAG_DO_NOT_SORT_CENTRAL_DIRECTORY, 0, 0)) { |
| | |
| | |
| | *errnum = ZIP_ERINIT; |
| | goto cleanup; |
| | } |
| | break; |
| |
|
| | case 'a': |
| | case 'd': |
| | if (!mz_zip_reader_init_file_v2_rpb( |
| | &(zip->archive), zipname, |
| | zip->level | MZ_ZIP_FLAG_DO_NOT_SORT_CENTRAL_DIRECTORY, 0, 0)) { |
| | |
| | |
| | *errnum = ZIP_ERINIT; |
| | goto cleanup; |
| | } |
| | if ((mode == 'a' || mode == 'd')) { |
| | if (!mz_zip_writer_init_from_reader_v2_noreopen(&(zip->archive), zipname, |
| | 0)) { |
| | *errnum = ZIP_EWRINIT; |
| | mz_zip_reader_end(&(zip->archive)); |
| | goto cleanup; |
| | } |
| | } |
| | break; |
| |
|
| | default: |
| | *errnum = ZIP_EINVMODE; |
| | goto cleanup; |
| | } |
| |
|
| | return zip; |
| |
|
| | cleanup: |
| | CLEANUP(zip); |
| | return NULL; |
| | } |
| |
|
| | void zip_close(struct zip_t *zip) { |
| | if (zip) { |
| | mz_zip_archive *pZip = &(zip->archive); |
| | |
| | |
| | if (pZip->m_zip_mode == MZ_ZIP_MODE_WRITING) { |
| | mz_zip_writer_finalize_archive(pZip); |
| | } |
| |
|
| | if (pZip->m_zip_mode == MZ_ZIP_MODE_WRITING || |
| | pZip->m_zip_mode == MZ_ZIP_MODE_WRITING_HAS_BEEN_FINALIZED) { |
| | zip_archive_truncate(pZip); |
| | mz_zip_writer_end(pZip); |
| | } |
| | if (pZip->m_zip_mode == MZ_ZIP_MODE_READING) { |
| | mz_zip_reader_end(pZip); |
| | } |
| |
|
| | CLEANUP(zip); |
| | } |
| | } |
| |
|
| | int zip_is64(struct zip_t *zip) { |
| | if (!zip || !zip->archive.m_pState) { |
| | |
| | return ZIP_ENOINIT; |
| | } |
| |
|
| | return (int)zip->archive.m_pState->m_zip64; |
| | } |
| |
|
| | static int _zip_entry_open(struct zip_t *zip, const char *entryname, |
| | int case_sensitive) { |
| | size_t entrylen = 0; |
| | mz_zip_archive *pzip = NULL; |
| | mz_uint num_alignment_padding_bytes, level; |
| | mz_zip_archive_file_stat stats; |
| | int err = 0; |
| | mz_uint16 dos_time = 0, dos_date = 0; |
| | mz_uint32 extra_size = 0; |
| | mz_uint8 extra_data[MZ_ZIP64_MAX_CENTRAL_EXTRA_FIELD_SIZE]; |
| | mz_uint64 local_dir_header_ofs = 0; |
| |
|
| | if (!zip) { |
| | return ZIP_ENOINIT; |
| | } |
| |
|
| | local_dir_header_ofs = zip->archive.m_archive_size; |
| |
|
| | if (!entryname) { |
| | return ZIP_EINVENTNAME; |
| | } |
| |
|
| | entrylen = strlen(entryname); |
| | if (entrylen == 0) { |
| | return ZIP_EINVENTNAME; |
| | } |
| |
|
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | if (zip->entry.name) { |
| | CLEANUP(zip->entry.name); |
| | } |
| | #ifdef ZIP_RAW_ENTRYNAME |
| | zip->entry.name = STRCLONE(entryname); |
| | #else |
| | zip->entry.name = zip_strrpl(entryname, entrylen, '\\', '/'); |
| | #endif |
| |
|
| | if (!zip->entry.name) { |
| | |
| | return ZIP_EINVENTNAME; |
| | } |
| |
|
| | pzip = &(zip->archive); |
| | if (pzip->m_zip_mode == MZ_ZIP_MODE_READING) { |
| | zip->entry.index = (ssize_t)mz_zip_reader_locate_file( |
| | pzip, zip->entry.name, NULL, |
| | case_sensitive ? MZ_ZIP_FLAG_CASE_SENSITIVE : 0); |
| | if (zip->entry.index < (ssize_t)0) { |
| | err = ZIP_ENOENT; |
| | goto cleanup; |
| | } |
| |
|
| | if (!mz_zip_reader_file_stat(pzip, (mz_uint)zip->entry.index, &stats)) { |
| | err = ZIP_ENOENT; |
| | 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; |
| | #ifndef MINIZ_NO_TIME |
| | zip->entry.m_time = stats.m_time; |
| | #endif |
| |
|
| | return 0; |
| | } |
| |
|
| | level = zip->level & 0xF; |
| |
|
| | zip->entry.index = (ssize_t)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 = level ? MZ_DEFLATED : 0; |
| |
|
| | |
| | #if MZ_PLATFORM == 3 || MZ_PLATFORM == 19 |
| | |
| | zip->entry.external_attr = (mz_uint32)(0100644) << 16; |
| | #else |
| | zip->entry.external_attr = 0; |
| | #endif |
| |
|
| | 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)) { |
| | |
| | err = ZIP_EINVMODE; |
| | goto cleanup; |
| | } |
| | if (zip->level & MZ_ZIP_FLAG_COMPRESSED_DATA) { |
| | |
| | err = ZIP_EINVLVL; |
| | goto cleanup; |
| | } |
| |
|
| | if (!mz_zip_writer_write_zeros(pzip, zip->entry.offset, |
| | num_alignment_padding_bytes)) { |
| | |
| | err = ZIP_EMEMSET; |
| | goto cleanup; |
| | } |
| | local_dir_header_ofs += num_alignment_padding_bytes; |
| |
|
| | zip->entry.m_time = time(NULL); |
| | #ifndef MINIZ_NO_TIME |
| | mz_zip_time_t_to_dos_time(zip->entry.m_time, &dos_time, &dos_date); |
| | #endif |
| |
|
| | |
| | |
| | extra_size = mz_zip_writer_create_zip64_extra_data( |
| | extra_data, NULL, NULL, |
| | (local_dir_header_ofs >= MZ_UINT32_MAX) ? &local_dir_header_ofs : NULL); |
| |
|
| | if (!mz_zip_writer_create_local_dir_header( |
| | pzip, zip->entry.header, entrylen, (mz_uint16)extra_size, 0, 0, 0, |
| | zip->entry.method, |
| | MZ_ZIP_GENERAL_PURPOSE_BIT_FLAG_UTF8 | |
| | MZ_ZIP_LDH_BIT_FLAG_HAS_LOCATOR, |
| | dos_time, dos_date)) { |
| | |
| | err = ZIP_EMEMSET; |
| | goto cleanup; |
| | } |
| |
|
| | zip->entry.header_offset = zip->entry.offset + num_alignment_padding_bytes; |
| |
|
| | if (pzip->m_pWrite(pzip->m_pIO_opaque, zip->entry.header_offset, |
| | zip->entry.header, |
| | sizeof(zip->entry.header)) != sizeof(zip->entry.header)) { |
| | |
| | err = ZIP_EMEMSET; |
| | goto cleanup; |
| | } |
| |
|
| | 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) { |
| | |
| | err = ZIP_EWRTENT; |
| | goto cleanup; |
| | } |
| |
|
| | zip->entry.offset += entrylen; |
| |
|
| | if (pzip->m_pWrite(pzip->m_pIO_opaque, zip->entry.offset, extra_data, |
| | extra_size) != extra_size) { |
| | |
| | err = ZIP_EWRTENT; |
| | goto cleanup; |
| | } |
| | zip->entry.offset += extra_size; |
| |
|
| | 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) { |
| | |
| | err = ZIP_ETDEFLINIT; |
| | goto cleanup; |
| | } |
| | } |
| |
|
| | return 0; |
| |
|
| | cleanup: |
| | CLEANUP(zip->entry.name); |
| | return err; |
| | } |
| |
|
| | int zip_entry_open(struct zip_t *zip, const char *entryname) { |
| | return _zip_entry_open(zip, entryname, 0); |
| | } |
| |
|
| | int zip_entry_opencasesensitive(struct zip_t *zip, const char *entryname) { |
| | return _zip_entry_open(zip, entryname, 1); |
| | } |
| |
|
| | int zip_entry_openbyindex(struct zip_t *zip, size_t index) { |
| | mz_zip_archive *pZip = NULL; |
| | mz_zip_archive_file_stat stats; |
| | mz_uint namelen; |
| | const mz_uint8 *pHeader; |
| | const char *pFilename; |
| |
|
| | if (!zip) { |
| | |
| | return ZIP_ENOINIT; |
| | } |
| |
|
| | pZip = &(zip->archive); |
| | if (pZip->m_zip_mode != MZ_ZIP_MODE_READING) { |
| | |
| | return ZIP_EINVMODE; |
| | } |
| |
|
| | if (index >= (size_t)pZip->m_total_files) { |
| | |
| | return ZIP_EINVIDX; |
| | } |
| |
|
| | 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)))) { |
| | |
| | return ZIP_ENOHDR; |
| | } |
| |
|
| | namelen = MZ_READ_LE16(pHeader + MZ_ZIP_CDH_FILENAME_LEN_OFS); |
| | pFilename = (const char *)pHeader + MZ_ZIP_CENTRAL_DIR_HEADER_SIZE; |
| |
|
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | if (zip->entry.name) { |
| | CLEANUP(zip->entry.name); |
| | } |
| | #ifdef ZIP_RAW_ENTRYNAME |
| | zip->entry.name = STRCLONE(pFilename); |
| | #else |
| | zip->entry.name = zip_strrpl(pFilename, namelen, '\\', '/'); |
| | #endif |
| |
|
| | if (!zip->entry.name) { |
| | |
| | return ZIP_EINVENTNAME; |
| | } |
| |
|
| | if (!mz_zip_reader_file_stat(pZip, (mz_uint)index, &stats)) { |
| | return ZIP_ENOENT; |
| | } |
| |
|
| | zip->entry.index = (ssize_t)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; |
| | #ifndef MINIZ_NO_TIME |
| | zip->entry.m_time = stats.m_time; |
| | #endif |
| |
|
| | return 0; |
| | } |
| |
|
| | 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 = 0, dos_date = 0; |
| | int err = 0; |
| | mz_uint8 *pExtra_data = NULL; |
| | mz_uint32 extra_size = 0; |
| | mz_uint8 extra_data[MZ_ZIP64_MAX_CENTRAL_EXTRA_FIELD_SIZE]; |
| | mz_uint8 local_dir_footer[MZ_ZIP_DATA_DESCRIPTER_SIZE64]; |
| | mz_uint32 local_dir_footer_size = MZ_ZIP_DATA_DESCRIPTER_SIZE64; |
| |
|
| | if (!zip) { |
| | |
| | err = ZIP_ENOINIT; |
| | goto cleanup; |
| | } |
| |
|
| | pzip = &(zip->archive); |
| | if (pzip->m_zip_mode == MZ_ZIP_MODE_READING) { |
| | 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) { |
| | |
| | err = ZIP_ETDEFLBUF; |
| | 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); |
| | #ifndef MINIZ_NO_TIME |
| | mz_zip_time_t_to_dos_time(zip->entry.m_time, &dos_time, &dos_date); |
| | #endif |
| |
|
| | MZ_WRITE_LE32(local_dir_footer + 0, MZ_ZIP_DATA_DESCRIPTOR_ID); |
| | MZ_WRITE_LE32(local_dir_footer + 4, zip->entry.uncomp_crc32); |
| | MZ_WRITE_LE64(local_dir_footer + 8, zip->entry.comp_size); |
| | MZ_WRITE_LE64(local_dir_footer + 16, zip->entry.uncomp_size); |
| |
|
| | if (pzip->m_pWrite(pzip->m_pIO_opaque, zip->entry.offset, local_dir_footer, |
| | local_dir_footer_size) != local_dir_footer_size) { |
| | |
| | err = ZIP_EWRTHDR; |
| | goto cleanup; |
| | } |
| | zip->entry.offset += local_dir_footer_size; |
| |
|
| | pExtra_data = extra_data; |
| | extra_size = mz_zip_writer_create_zip64_extra_data( |
| | extra_data, |
| | (zip->entry.uncomp_size >= MZ_UINT32_MAX) ? &zip->entry.uncomp_size |
| | : NULL, |
| | (zip->entry.comp_size >= MZ_UINT32_MAX) ? &zip->entry.comp_size : NULL, |
| | (zip->entry.header_offset >= MZ_UINT32_MAX) ? &zip->entry.header_offset |
| | : NULL); |
| |
|
| | if ((entrylen) && (zip->entry.name[entrylen - 1] == '/') && |
| | !zip->entry.uncomp_size) { |
| | |
| | zip->entry.external_attr |= MZ_ZIP_DOS_DIR_ATTRIBUTE_BITFLAG; |
| | } |
| |
|
| | if (!mz_zip_writer_add_to_central_dir( |
| | pzip, zip->entry.name, entrylen, pExtra_data, (mz_uint16)extra_size, |
| | "", 0, zip->entry.uncomp_size, zip->entry.comp_size, |
| | zip->entry.uncomp_crc32, zip->entry.method, |
| | MZ_ZIP_GENERAL_PURPOSE_BIT_FLAG_UTF8 | |
| | MZ_ZIP_LDH_BIT_FLAG_HAS_LOCATOR, |
| | dos_time, dos_date, zip->entry.header_offset, |
| | zip->entry.external_attr, NULL, 0)) { |
| | |
| | err = ZIP_EWRTDIR; |
| | goto cleanup; |
| | } |
| |
|
| | pzip->m_total_files++; |
| | pzip->m_archive_size = zip->entry.offset; |
| |
|
| | cleanup: |
| | if (zip) { |
| | zip->entry.m_time = 0; |
| | CLEANUP(zip->entry.name); |
| | } |
| | return err; |
| | } |
| |
|
| | const char *zip_entry_name(struct zip_t *zip) { |
| | if (!zip) { |
| | |
| | return NULL; |
| | } |
| |
|
| | return zip->entry.name; |
| | } |
| |
|
| | ssize_t zip_entry_index(struct zip_t *zip) { |
| | if (!zip) { |
| | |
| | return (ssize_t)ZIP_ENOINIT; |
| | } |
| |
|
| | return zip->entry.index; |
| | } |
| |
|
| | int zip_entry_isdir(struct zip_t *zip) { |
| | if (!zip) { |
| | |
| | return ZIP_ENOINIT; |
| | } |
| |
|
| | if (zip->entry.index < (ssize_t)0) { |
| | |
| | return ZIP_EINVIDX; |
| | } |
| |
|
| | return (int)mz_zip_reader_is_file_a_directory(&zip->archive, |
| | (mz_uint)zip->entry.index); |
| | } |
| |
|
| | unsigned long long zip_entry_size(struct zip_t *zip) { |
| | return zip_entry_uncomp_size(zip); |
| | } |
| |
|
| | unsigned long long zip_entry_uncomp_size(struct zip_t *zip) { |
| | return zip ? zip->entry.uncomp_size : 0; |
| | } |
| |
|
| | unsigned long long zip_entry_comp_size(struct zip_t *zip) { |
| | return zip ? zip->entry.comp_size : 0; |
| | } |
| |
|
| | 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; |
| |
|
| | if (!zip) { |
| | |
| | return ZIP_ENOINIT; |
| | } |
| |
|
| | 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)) { |
| | |
| | return ZIP_EWRTENT; |
| | } |
| | 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) { |
| | |
| | return ZIP_ETDEFLBUF; |
| | } |
| | } |
| | } |
| |
|
| | return 0; |
| | } |
| |
|
| | int zip_entry_fwrite(struct zip_t *zip, const char *filename) { |
| | int err = 0; |
| | size_t n = 0; |
| | MZ_FILE *stream = NULL; |
| | mz_uint8 buf[MZ_ZIP_MAX_IO_BUF_SIZE]; |
| | struct MZ_FILE_STAT_STRUCT file_stat; |
| | mz_uint16 modes; |
| |
|
| | if (!zip) { |
| | |
| | return ZIP_ENOINIT; |
| | } |
| |
|
| | 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) { |
| | |
| | return ZIP_ENOENT; |
| | } |
| |
|
| | #if defined(_WIN32) || defined(__WIN32__) || defined(DJGPP) |
| | (void)modes; |
| | #else |
| | |
| | modes = file_stat.st_mode & |
| | (S_IRWXU | S_IRWXG | S_IRWXO | S_ISUID | S_ISGID | S_ISVTX); |
| | if (S_ISDIR(file_stat.st_mode)) |
| | modes |= UNX_IFDIR; |
| | if (S_ISREG(file_stat.st_mode)) |
| | modes |= UNX_IFREG; |
| | if (S_ISLNK(file_stat.st_mode)) |
| | modes |= UNX_IFLNK; |
| | if (S_ISBLK(file_stat.st_mode)) |
| | modes |= UNX_IFBLK; |
| | if (S_ISCHR(file_stat.st_mode)) |
| | modes |= UNX_IFCHR; |
| | if (S_ISFIFO(file_stat.st_mode)) |
| | modes |= UNX_IFIFO; |
| | if (S_ISSOCK(file_stat.st_mode)) |
| | modes |= UNX_IFSOCK; |
| | zip->entry.external_attr = (modes << 16) | !(file_stat.st_mode & S_IWUSR); |
| | if ((file_stat.st_mode & S_IFMT) == S_IFDIR) { |
| | zip->entry.external_attr |= MZ_ZIP_DOS_DIR_ATTRIBUTE_BITFLAG; |
| | } |
| | #endif |
| |
|
| | zip->entry.m_time = file_stat.st_mtime; |
| |
|
| | if (!(stream = MZ_FOPEN(filename, "rb"))) { |
| | |
| | return ZIP_EOPNFILE; |
| | } |
| |
|
| | while ((n = fread(buf, sizeof(mz_uint8), MZ_ZIP_MAX_IO_BUF_SIZE, stream)) > |
| | 0) { |
| | if (zip_entry_write(zip, buf, n) < 0) { |
| | err = ZIP_EWRTENT; |
| | break; |
| | } |
| | } |
| | fclose(stream); |
| |
|
| | return err; |
| | } |
| |
|
| | 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) { |
| | |
| | return (ssize_t)ZIP_ENOINIT; |
| | } |
| |
|
| | pzip = &(zip->archive); |
| | if (pzip->m_zip_mode != MZ_ZIP_MODE_READING || |
| | zip->entry.index < (ssize_t)0) { |
| | |
| | return (ssize_t)ZIP_ENOENT; |
| | } |
| |
|
| | idx = (mz_uint)zip->entry.index; |
| | if (mz_zip_reader_is_file_a_directory(pzip, idx)) { |
| | |
| | return (ssize_t)ZIP_EINVENTTYPE; |
| | } |
| |
|
| | *buf = mz_zip_reader_extract_to_heap(pzip, idx, &size, 0); |
| | if (*buf && bufsize) { |
| | *bufsize = size; |
| | } |
| | return (ssize_t)size; |
| | } |
| |
|
| | ssize_t zip_entry_noallocread(struct zip_t *zip, void *buf, size_t bufsize) { |
| | mz_zip_archive *pzip = NULL; |
| |
|
| | if (!zip) { |
| | |
| | return (ssize_t)ZIP_ENOINIT; |
| | } |
| |
|
| | pzip = &(zip->archive); |
| | if (pzip->m_zip_mode != MZ_ZIP_MODE_READING || |
| | zip->entry.index < (ssize_t)0) { |
| | |
| | return (ssize_t)ZIP_ENOENT; |
| | } |
| |
|
| | if (!mz_zip_reader_extract_to_mem_no_alloc(pzip, (mz_uint)zip->entry.index, |
| | buf, bufsize, 0, NULL, 0)) { |
| | return (ssize_t)ZIP_EMEMNOALLOC; |
| | } |
| |
|
| | 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; |
| | mz_uint32 xattr = 0; |
| | mz_zip_archive_file_stat info; |
| |
|
| | if (!zip) { |
| | |
| | return ZIP_ENOINIT; |
| | } |
| |
|
| | 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 < (ssize_t)0) { |
| | |
| | return ZIP_ENOENT; |
| | } |
| |
|
| | idx = (mz_uint)zip->entry.index; |
| | if (mz_zip_reader_is_file_a_directory(pzip, idx)) { |
| | |
| | return ZIP_EINVENTTYPE; |
| | } |
| |
|
| | if (!mz_zip_reader_extract_to_file(pzip, idx, filename, 0)) { |
| | return ZIP_ENOFILE; |
| | } |
| |
|
| | #if defined(_MSC_VER) || defined(PS4) |
| | (void)xattr; |
| | #else |
| | if (!mz_zip_reader_file_stat(pzip, idx, &info)) { |
| | |
| | return ZIP_ENOFILE; |
| | } |
| |
|
| | xattr = (info.m_external_attr >> 16) & 0xFFFF; |
| | if (xattr > 0 && xattr <= MZ_UINT16_MAX) { |
| | if (CHMOD(filename, (mode_t)xattr) < 0) { |
| | return ZIP_ENOPERM; |
| | } |
| | } |
| | #endif |
| |
|
| | return 0; |
| | } |
| |
|
| | int zip_entry_extract(struct zip_t *zip, |
| | size_t (*on_extract)(void *arg, uint64_t offset, |
| | const void *buf, size_t bufsize), |
| | void *arg) { |
| | mz_zip_archive *pzip = NULL; |
| | mz_uint idx; |
| |
|
| | if (!zip) { |
| | |
| | return ZIP_ENOINIT; |
| | } |
| |
|
| | pzip = &(zip->archive); |
| | if (pzip->m_zip_mode != MZ_ZIP_MODE_READING || |
| | zip->entry.index < (ssize_t)0) { |
| | |
| | return ZIP_ENOENT; |
| | } |
| |
|
| | idx = (mz_uint)zip->entry.index; |
| | return (mz_zip_reader_extract_to_callback(pzip, idx, on_extract, arg, 0)) |
| | ? 0 |
| | : ZIP_EINVIDX; |
| | } |
| |
|
| | ssize_t zip_entries_total(struct zip_t *zip) { |
| | if (!zip) { |
| | |
| | return ZIP_ENOINIT; |
| | } |
| |
|
| | return (ssize_t)zip->archive.m_total_files; |
| | } |
| |
|
| | ssize_t zip_entries_delete(struct zip_t *zip, char *const entries[], |
| | size_t len) { |
| | ssize_t n = 0; |
| | ssize_t err = 0; |
| | struct zip_entry_mark_t *entry_mark = NULL; |
| |
|
| | if (zip == NULL || (entries == NULL && len != 0)) { |
| | return ZIP_ENOINIT; |
| | } |
| |
|
| | if (entries == NULL && len == 0) { |
| | return 0; |
| | } |
| |
|
| | n = zip_entries_total(zip); |
| |
|
| | entry_mark = (struct zip_entry_mark_t *)calloc( |
| | (size_t)n, sizeof(struct zip_entry_mark_t)); |
| | if (!entry_mark) { |
| | return ZIP_EOOMEM; |
| | } |
| |
|
| | zip->archive.m_zip_mode = MZ_ZIP_MODE_READING; |
| |
|
| | err = zip_entry_set(zip, entry_mark, n, entries, len); |
| | if (err < 0) { |
| | CLEANUP(entry_mark); |
| | return err; |
| | } |
| |
|
| | err = zip_entries_delete_mark(zip, entry_mark, (int)n); |
| | CLEANUP(entry_mark); |
| | return err; |
| | } |
| |
|
| | int zip_stream_extract(const char *stream, size_t size, const char *dir, |
| | int (*on_extract)(const char *filename, void *arg), |
| | void *arg) { |
| | mz_zip_archive zip_archive; |
| | if (!stream || !dir) { |
| | |
| | return ZIP_ENOINIT; |
| | } |
| | if (!memset(&zip_archive, 0, sizeof(mz_zip_archive))) { |
| | |
| | return ZIP_EMEMSET; |
| | } |
| | if (!mz_zip_reader_init_mem(&zip_archive, stream, size, 0)) { |
| | |
| | return ZIP_ENOINIT; |
| | } |
| |
|
| | return zip_archive_extract(&zip_archive, dir, on_extract, arg); |
| | } |
| |
|
| | struct zip_t *zip_stream_open(const char *stream, size_t size, int level, |
| | char mode) { |
| | int errnum = 0; |
| | return zip_stream_openwitherror(stream, size, level, mode, &errnum); |
| | } |
| |
|
| | struct zip_t *zip_stream_openwitherror(const char *stream, size_t size, |
| | int level, char mode, int *errnum) { |
| | struct zip_t *zip = (struct zip_t *)calloc((size_t)1, sizeof(struct zip_t)); |
| | if (!zip) { |
| | |
| | *errnum = ZIP_EOOMEM; |
| | return NULL; |
| | } |
| |
|
| | if (level < 0) { |
| | level = MZ_DEFAULT_LEVEL; |
| | } |
| | if ((level & 0xF) > MZ_UBER_COMPRESSION) { |
| | |
| | *errnum = ZIP_EINVLVL; |
| | goto cleanup; |
| | } |
| | zip->level = (mz_uint)level; |
| |
|
| | if ((stream != NULL) && (size > 0) && (mode == 'r')) { |
| | if (!mz_zip_reader_init_mem(&(zip->archive), stream, size, 0)) { |
| | *errnum = ZIP_ERINIT; |
| | goto cleanup; |
| | } |
| | } else if ((stream == NULL) && (size == 0) && (mode == 'w')) { |
| | |
| | if (!mz_zip_writer_init_heap(&(zip->archive), 0, 1024)) { |
| | |
| | *errnum = ZIP_EWINIT; |
| | goto cleanup; |
| | } |
| | } else { |
| | *errnum = ZIP_EINVMODE; |
| | goto cleanup; |
| | } |
| |
|
| | *errnum = 0; |
| | return zip; |
| |
|
| | cleanup: |
| | CLEANUP(zip); |
| | return NULL; |
| | } |
| |
|
| | ssize_t zip_stream_copy(struct zip_t *zip, void **buf, size_t *bufsize) { |
| | size_t n; |
| |
|
| | if (!zip) { |
| | return (ssize_t)ZIP_ENOINIT; |
| | } |
| | zip_archive_finalize(&(zip->archive)); |
| |
|
| | n = (size_t)zip->archive.m_archive_size; |
| | if (bufsize != NULL) { |
| | *bufsize = n; |
| | } |
| |
|
| | *buf = calloc(sizeof(unsigned char), n); |
| | memcpy(*buf, zip->archive.m_pState->m_pMem, n); |
| |
|
| | return (ssize_t)n; |
| | } |
| |
|
| | void zip_stream_close(struct zip_t *zip) { |
| | if (zip) { |
| | mz_zip_writer_end(&(zip->archive)); |
| | mz_zip_reader_end(&(zip->archive)); |
| | CLEANUP(zip); |
| | } |
| | } |
| |
|
| | int zip_create(const char *zipname, const char *filenames[], size_t len) { |
| | int err = 0; |
| | size_t i; |
| | mz_zip_archive zip_archive; |
| | struct MZ_FILE_STAT_STRUCT file_stat; |
| | mz_uint32 ext_attributes = 0; |
| | mz_uint16 modes; |
| |
|
| | if (!zipname || strlen(zipname) < 1) { |
| | |
| | return ZIP_EINVZIPNAME; |
| | } |
| |
|
| | |
| | if (!memset(&(zip_archive), 0, sizeof(zip_archive))) { |
| | |
| | return ZIP_EMEMSET; |
| | } |
| |
|
| | if (!mz_zip_writer_init_file(&zip_archive, zipname, 0)) { |
| | |
| | return ZIP_ENOINIT; |
| | } |
| |
|
| | if (!memset((void *)&file_stat, 0, sizeof(struct MZ_FILE_STAT_STRUCT))) { |
| | return ZIP_EMEMSET; |
| | } |
| |
|
| | for (i = 0; i < len; ++i) { |
| | const char *name = filenames[i]; |
| | if (!name) { |
| | err = ZIP_EINVENTNAME; |
| | break; |
| | } |
| |
|
| | if (MZ_FILE_STAT(name, &file_stat) != 0) { |
| | |
| | err = ZIP_ENOFILE; |
| | break; |
| | } |
| |
|
| | #if defined(_WIN32) || defined(__WIN32__) || defined(DJGPP) |
| | (void)modes; |
| | #else |
| |
|
| | |
| | modes = file_stat.st_mode & |
| | (S_IRWXU | S_IRWXG | S_IRWXO | S_ISUID | S_ISGID | S_ISVTX); |
| | if (S_ISDIR(file_stat.st_mode)) |
| | modes |= UNX_IFDIR; |
| | if (S_ISREG(file_stat.st_mode)) |
| | modes |= UNX_IFREG; |
| | if (S_ISLNK(file_stat.st_mode)) |
| | modes |= UNX_IFLNK; |
| | if (S_ISBLK(file_stat.st_mode)) |
| | modes |= UNX_IFBLK; |
| | if (S_ISCHR(file_stat.st_mode)) |
| | modes |= UNX_IFCHR; |
| | if (S_ISFIFO(file_stat.st_mode)) |
| | modes |= UNX_IFIFO; |
| | if (S_ISSOCK(file_stat.st_mode)) |
| | modes |= UNX_IFSOCK; |
| | ext_attributes = (modes << 16) | !(file_stat.st_mode & S_IWUSR); |
| | if ((file_stat.st_mode & S_IFMT) == S_IFDIR) { |
| | ext_attributes |= MZ_ZIP_DOS_DIR_ATTRIBUTE_BITFLAG; |
| | } |
| | #endif |
| |
|
| | if (!mz_zip_writer_add_file(&zip_archive, zip_basename(name), name, "", 0, |
| | ZIP_DEFAULT_COMPRESSION_LEVEL, |
| | ext_attributes)) { |
| | |
| | err = ZIP_ENOFILE; |
| | break; |
| | } |
| | } |
| |
|
| | mz_zip_writer_finalize_archive(&zip_archive); |
| | mz_zip_writer_end(&zip_archive); |
| | return err; |
| | } |
| |
|
| | int zip_extract(const char *zipname, const char *dir, |
| | int (*on_extract)(const char *filename, void *arg), void *arg) { |
| | mz_zip_archive zip_archive; |
| |
|
| | if (!zipname || !dir) { |
| | |
| | return ZIP_EINVZIPNAME; |
| | } |
| |
|
| | if (!memset(&zip_archive, 0, sizeof(mz_zip_archive))) { |
| | |
| | return ZIP_EMEMSET; |
| | } |
| |
|
| | |
| | if (!mz_zip_reader_init_file(&zip_archive, zipname, 0)) { |
| | |
| | return ZIP_ENOINIT; |
| | } |
| |
|
| | return zip_archive_extract(&zip_archive, dir, on_extract, arg); |
| | } |
| |
|
| | #if defined(__SSE4_2__) || defined(__AVX512F__) |
| | #include <immintrin.h> |
| | #endif |
| |
|
| | |
| | |
| | |
| | |
| | mz_ulong mz_crc32(mz_ulong init, const uint8_t *buf, size_t len) { |
| | uint32_t crc = ~init; |
| | #if defined(__AVX512F__) && defined(__VPCLMULQDQ__) && defined(__PCLMUL__) |
| | if (len >= 256) { |
| | _Alignas(__m512) static const uint64_t k1k2[] = { |
| | 0x011542778a, 0x01322d1430, 0x011542778a, 0x01322d1430, |
| | 0x011542778a, 0x01322d1430, 0x011542778a, 0x01322d1430, |
| | }; |
| | _Alignas(__m512) static const uint64_t k3k4[] = { |
| | 0x0154442bd4, 0x01c6e41596, 0x0154442bd4, 0x01c6e41596, |
| | 0x0154442bd4, 0x01c6e41596, 0x0154442bd4, 0x01c6e41596, |
| | }; |
| | _Alignas(__m512) static const uint64_t k5k6[] = { |
| | 0x01751997d0, |
| | 0x00ccaa009e, |
| | }; |
| | _Alignas(__m512) static const uint64_t k7k8[] = { |
| | 0x0163cd6124, |
| | 0x0000000000, |
| | }; |
| | _Alignas(__m512) static const uint64_t poly[] = { |
| | 0x01db710641, |
| | 0x01f7011641, |
| | }; |
| | __m512i x0, x1, x2, x3, x4, x5, x6, x7, x8, y5, y6, y7, y8; |
| | __m128i a0, a1, a2, a3; |
| | x1 = _mm512_loadu_si512((__m512i *)(buf + 0x00)); |
| | x2 = _mm512_loadu_si512((__m512i *)(buf + 0x40)); |
| | x3 = _mm512_loadu_si512((__m512i *)(buf + 0x80)); |
| | x4 = _mm512_loadu_si512((__m512i *)(buf + 0xC0)); |
| | x1 = _mm512_xor_si512(x1, _mm512_castsi128_si512(_mm_cvtsi32_si128(crc))); |
| | x0 = _mm512_load_si512((__m512i *)k1k2); |
| | buf += 256; |
| | len -= 256; |
| | while (len >= 256) { |
| | x5 = _mm512_clmulepi64_epi128(x1, x0, 0x00); |
| | x6 = _mm512_clmulepi64_epi128(x2, x0, 0x00); |
| | x7 = _mm512_clmulepi64_epi128(x3, x0, 0x00); |
| | x8 = _mm512_clmulepi64_epi128(x4, x0, 0x00); |
| | x1 = _mm512_clmulepi64_epi128(x1, x0, 0x11); |
| | x2 = _mm512_clmulepi64_epi128(x2, x0, 0x11); |
| | x3 = _mm512_clmulepi64_epi128(x3, x0, 0x11); |
| | x4 = _mm512_clmulepi64_epi128(x4, x0, 0x11); |
| | y5 = _mm512_loadu_si512((__m512i *)(buf + 0x00)); |
| | y6 = _mm512_loadu_si512((__m512i *)(buf + 0x40)); |
| | y7 = _mm512_loadu_si512((__m512i *)(buf + 0x80)); |
| | y8 = _mm512_loadu_si512((__m512i *)(buf + 0xC0)); |
| | x1 = _mm512_xor_si512(x1, x5); |
| | x2 = _mm512_xor_si512(x2, x6); |
| | x3 = _mm512_xor_si512(x3, x7); |
| | x4 = _mm512_xor_si512(x4, x8); |
| | x1 = _mm512_xor_si512(x1, y5); |
| | x2 = _mm512_xor_si512(x2, y6); |
| | x3 = _mm512_xor_si512(x3, y7); |
| | x4 = _mm512_xor_si512(x4, y8); |
| | buf += 256; |
| | len -= 256; |
| | } |
| | x0 = _mm512_load_si512((__m512i *)k3k4); |
| | x5 = _mm512_clmulepi64_epi128(x1, x0, 0x00); |
| | x1 = _mm512_clmulepi64_epi128(x1, x0, 0x11); |
| | x1 = _mm512_xor_si512(x1, x2); |
| | x1 = _mm512_xor_si512(x1, x5); |
| | x5 = _mm512_clmulepi64_epi128(x1, x0, 0x00); |
| | x1 = _mm512_clmulepi64_epi128(x1, x0, 0x11); |
| | x1 = _mm512_xor_si512(x1, x3); |
| | x1 = _mm512_xor_si512(x1, x5); |
| | x5 = _mm512_clmulepi64_epi128(x1, x0, 0x00); |
| | x1 = _mm512_clmulepi64_epi128(x1, x0, 0x11); |
| | x1 = _mm512_xor_si512(x1, x4); |
| | x1 = _mm512_xor_si512(x1, x5); |
| | while (len >= 64) { |
| | x2 = _mm512_loadu_si512((__m512i *)buf); |
| | x5 = _mm512_clmulepi64_epi128(x1, x0, 0x00); |
| | x1 = _mm512_clmulepi64_epi128(x1, x0, 0x11); |
| | x1 = _mm512_xor_si512(x1, x2); |
| | x1 = _mm512_xor_si512(x1, x5); |
| | buf += 64; |
| | len -= 64; |
| | } |
| | a0 = _mm_load_si128((__m128i *)k5k6); |
| | a1 = _mm512_extracti32x4_epi32(x1, 0); |
| | a2 = _mm512_extracti32x4_epi32(x1, 1); |
| | a3 = _mm_clmulepi64_si128(a1, a0, 0x00); |
| | a1 = _mm_clmulepi64_si128(a1, a0, 0x11); |
| | a1 = _mm_xor_si128(a1, a3); |
| | a1 = _mm_xor_si128(a1, a2); |
| | a2 = _mm512_extracti32x4_epi32(x1, 2); |
| | a3 = _mm_clmulepi64_si128(a1, a0, 0x00); |
| | a1 = _mm_clmulepi64_si128(a1, a0, 0x11); |
| | a1 = _mm_xor_si128(a1, a3); |
| | a1 = _mm_xor_si128(a1, a2); |
| | a2 = _mm512_extracti32x4_epi32(x1, 3); |
| | a3 = _mm_clmulepi64_si128(a1, a0, 0x00); |
| | a1 = _mm_clmulepi64_si128(a1, a0, 0x11); |
| | a1 = _mm_xor_si128(a1, a3); |
| | a1 = _mm_xor_si128(a1, a2); |
| | a2 = _mm_clmulepi64_si128(a1, a0, 0x10); |
| | a3 = _mm_setr_epi32(~0, 0, ~0, 0); |
| | a1 = _mm_srli_si128(a1, 8); |
| | a1 = _mm_xor_si128(a1, a2); |
| | a0 = _mm_loadl_epi64((__m128i *)k7k8); |
| | a2 = _mm_srli_si128(a1, 4); |
| | a1 = _mm_and_si128(a1, a3); |
| | a1 = _mm_clmulepi64_si128(a1, a0, 0x00); |
| | a1 = _mm_xor_si128(a1, a2); |
| | a0 = _mm_load_si128((__m128i *)poly); |
| | a2 = _mm_and_si128(a1, a3); |
| | a2 = _mm_clmulepi64_si128(a2, a0, 0x10); |
| | a2 = _mm_and_si128(a2, a3); |
| | a2 = _mm_clmulepi64_si128(a2, a0, 0x00); |
| | a1 = _mm_xor_si128(a1, a2); |
| | crc = _mm_extract_epi32(a1, 1); |
| | } |
| | #endif |
| | #if defined(__SSE4_2__) && defined(__PCLMUL__) |
| | if (len >= 64) { |
| | _Alignas(__m128) static const uint64_t k1k2[] = { |
| | 0x0154442bd4, |
| | 0x01c6e41596, |
| | }; |
| | _Alignas(__m128) static const uint64_t k3k4[] = { |
| | 0x01751997d0, |
| | 0x00ccaa009e, |
| | }; |
| | _Alignas(__m128) static const uint64_t k5k0[] = { |
| | 0x0163cd6124, |
| | 0x0000000000, |
| | }; |
| | _Alignas(__m128) static const uint64_t poly[] = { |
| | 0x01db710641, |
| | 0x01f7011641, |
| | }; |
| | __m128i x0, x1, x2, x3, x4, x5, x6, x7, x8, y5, y6, y7, y8; |
| | x1 = _mm_loadu_si128((__m128i *)(buf + 0x00)); |
| | x2 = _mm_loadu_si128((__m128i *)(buf + 0x10)); |
| | x3 = _mm_loadu_si128((__m128i *)(buf + 0x20)); |
| | x4 = _mm_loadu_si128((__m128i *)(buf + 0x30)); |
| | x1 = _mm_xor_si128(x1, _mm_cvtsi32_si128(crc)); |
| | x0 = _mm_load_si128((__m128i *)k1k2); |
| | buf += 64; |
| | len -= 64; |
| | while (len >= 64) { |
| | x5 = _mm_clmulepi64_si128(x1, x0, 0x00); |
| | x6 = _mm_clmulepi64_si128(x2, x0, 0x00); |
| | x7 = _mm_clmulepi64_si128(x3, x0, 0x00); |
| | x8 = _mm_clmulepi64_si128(x4, x0, 0x00); |
| | x1 = _mm_clmulepi64_si128(x1, x0, 0x11); |
| | x2 = _mm_clmulepi64_si128(x2, x0, 0x11); |
| | x3 = _mm_clmulepi64_si128(x3, x0, 0x11); |
| | x4 = _mm_clmulepi64_si128(x4, x0, 0x11); |
| | y5 = _mm_loadu_si128((__m128i *)(buf + 0x00)); |
| | y6 = _mm_loadu_si128((__m128i *)(buf + 0x10)); |
| | y7 = _mm_loadu_si128((__m128i *)(buf + 0x20)); |
| | y8 = _mm_loadu_si128((__m128i *)(buf + 0x30)); |
| | x1 = _mm_xor_si128(x1, x5); |
| | x2 = _mm_xor_si128(x2, x6); |
| | x3 = _mm_xor_si128(x3, x7); |
| | x4 = _mm_xor_si128(x4, x8); |
| | x1 = _mm_xor_si128(x1, y5); |
| | x2 = _mm_xor_si128(x2, y6); |
| | x3 = _mm_xor_si128(x3, y7); |
| | x4 = _mm_xor_si128(x4, y8); |
| | buf += 64; |
| | len -= 64; |
| | } |
| | x0 = _mm_load_si128((__m128i *)k3k4); |
| | x5 = _mm_clmulepi64_si128(x1, x0, 0x00); |
| | x1 = _mm_clmulepi64_si128(x1, x0, 0x11); |
| | x1 = _mm_xor_si128(x1, x2); |
| | x1 = _mm_xor_si128(x1, x5); |
| | x5 = _mm_clmulepi64_si128(x1, x0, 0x00); |
| | x1 = _mm_clmulepi64_si128(x1, x0, 0x11); |
| | x1 = _mm_xor_si128(x1, x3); |
| | x1 = _mm_xor_si128(x1, x5); |
| | x5 = _mm_clmulepi64_si128(x1, x0, 0x00); |
| | x1 = _mm_clmulepi64_si128(x1, x0, 0x11); |
| | x1 = _mm_xor_si128(x1, x4); |
| | x1 = _mm_xor_si128(x1, x5); |
| | while (len >= 16) { |
| | x2 = _mm_loadu_si128((__m128i *)buf); |
| | x5 = _mm_clmulepi64_si128(x1, x0, 0x00); |
| | x1 = _mm_clmulepi64_si128(x1, x0, 0x11); |
| | x1 = _mm_xor_si128(x1, x2); |
| | x1 = _mm_xor_si128(x1, x5); |
| | buf += 16; |
| | len -= 16; |
| | } |
| | x2 = _mm_clmulepi64_si128(x1, x0, 0x10); |
| | x3 = _mm_setr_epi32(~0, 0, ~0, 0); |
| | x1 = _mm_srli_si128(x1, 8); |
| | x1 = _mm_xor_si128(x1, x2); |
| | x0 = _mm_loadl_epi64((__m128i *)k5k0); |
| | x2 = _mm_srli_si128(x1, 4); |
| | x1 = _mm_and_si128(x1, x3); |
| | x1 = _mm_clmulepi64_si128(x1, x0, 0x00); |
| | x1 = _mm_xor_si128(x1, x2); |
| | x0 = _mm_load_si128((__m128i *)poly); |
| | x2 = _mm_and_si128(x1, x3); |
| | x2 = _mm_clmulepi64_si128(x2, x0, 0x10); |
| | x2 = _mm_and_si128(x2, x3); |
| | x2 = _mm_clmulepi64_si128(x2, x0, 0x00); |
| | x1 = _mm_xor_si128(x1, x2); |
| | crc = _mm_extract_epi32(x1, 1); |
| | } |
| | #endif |
| | static uint32_t tab[256]; |
| | if (!tab[255]) { |
| | |
| | |
| | uint32_t polynomial = 0xedb88320; |
| | for (int d = 0; d < 256; ++d) { |
| | uint32_t r = d; |
| | for (int i = 0; i < 8; ++i) |
| | r = r >> 1 ^ (r & 1 ? polynomial : 0); |
| | tab[d] = r; |
| | } |
| | } |
| | for (size_t i = 0; i < len; ++i) |
| | crc = crc >> 8 ^ tab[(crc & 255) ^ buf[i]]; |
| | return ~crc & 0xffffffff; |
| | } |
| |
|