src/mz_zip.c

/* zip.c -- Zip manipulation
   Version 2.0.1, October 16th, 2017
   part of the MiniZip project

   Copyright (C) 2010-2017 Nathan Moinvaziri
     Modifications for AES, PKWARE disk spanning
     https://github.com/nmoinvaz/minizip
   Copyright (C) 2009-2010 Mathias Svensson
     Modifications for Zip64 support
     http://result42.com
   Copyright (C) 1998-2010 Gilles Vollant
     http://www.winimage.com/zLibDll/minizip.html

   This program is distributed under the terms of the same license as zlib.
   See the accompanying LICENSE file for the full text of the license.
*/

#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <string.h>
#include <errno.h>

#include "zlib.h"

#include "mz.h"
#include "mz_strm.h"
#ifdef HAVE_AES
#  include "mz_strm_aes.h"
#endif
#ifdef HAVE_BZIP2
#  include "mz_strm_bzip.h"
#endif
#ifdef HAVE_CRYPT
#  include "mz_strm_crypt.h"
#endif
#ifdef HAVE_LZMA
#  include "mz_strm_lzma.h"
#endif
#include "mz_strm_mem.h"
#include "mz_strm_zlib.h"

#include "mz_zip.h"

/***************************************************************************/

#define MZ_ZIP_MAGIC_LOCALHEADER        (0x04034b50)
#define MZ_ZIP_MAGIC_CENTRALHEADER      (0x02014b50)
#define MZ_ZIP_MAGIC_ENDHEADER          (0x06054b50)
#define MZ_ZIP_MAGIC_ENDHEADER64        (0x06064b50)
#define MZ_ZIP_MAGIC_ENDLOCHEADER64     (0x07064b50)
#define MZ_ZIP_MAGIC_DATADESCRIPTOR     (0x08074b50)

#define MZ_ZIP_SIZE_CD_ITEM             (0x2e)
#define MZ_ZIP_SIZE_CD_LOCATOR64        (0x14)
#define MZ_ZIP_SIZE_LOCALHEADER         (0x1e)

/***************************************************************************/

typedef struct mz_zip_s
{
    mz_zip_global global_info;
    mz_zip_file file_info;
    mz_zip_file local_file_info;
    mz_zip_compress compress_info;

    void *stream;                   // main stream
    void *cd_stream;                // memory stream for central directory
    void *compress_stream;          // compression stream
    void *crc32_stream;             // crc32 stream
    void *crypt_stream;             // encryption stream
    void *file_info_stream;         // memory stream for storing file info
    void *local_file_info_stream;   // memory stream for storing file info

    int32_t  open_mode;

    uint32_t number_disk;           // number of the current disk, used for spanning ZIP
    uint32_t number_disk_with_CD;   // number the the disk with central dir, used for spanning ZIP

    uint64_t disk_offset;           // byte before the zip file, (>0 for sfx)

    uint64_t cd_current_pos;        // pos of the current file in the central dir
    uint64_t cd_pos;                // position of the beginning of the central dir
    uint64_t cd_offset;             // offset of start of central directory
    uint64_t cd_size;               // size of the central directory

    uint16_t entry_scanned; 
    uint16_t entry_opened;          // 1 if a file in the zip is currently writ.
    uint64_t entry_read;

#ifdef HAVE_AES
    uint16_t aes_version;
    uint8_t  aes_encryption_mode;
#endif
} mz_zip;

/***************************************************************************/

// Locate the central directory of a zip file (at the end, just before the global comment)
static int32_t mz_zip_search_cd(void *stream, uint64_t *central_pos)
{
    uint8_t buf[1024 + 4];
    uint64_t file_size = 0;
    uint64_t back_read = 0;
    uint64_t max_back = UINT16_MAX; // maximum size of global comment
    uint32_t read_size = sizeof(buf);
    uint64_t read_pos = 0;
    uint32_t i = 0;

    *central_pos = 0;

    if (mz_stream_seek(stream, 0, MZ_STREAM_SEEK_END) != MZ_OK)
        return MZ_STREAM_ERROR;

    file_size = mz_stream_tell(stream);

    if (max_back > file_size)
        max_back = file_size;

    while (back_read < max_back)
    {
        back_read += (sizeof(buf) - 4);
        if (back_read > max_back)
            back_read = max_back;

        read_pos = file_size - back_read;
        if (read_size > (file_size - read_pos))
            read_size = (uint32_t)(file_size - read_pos);

        if (mz_stream_seek(stream, read_pos, MZ_STREAM_SEEK_SET) != MZ_OK)
            break;
        if (mz_stream_read(stream, buf, read_size) != read_size)
            break;

        for (i = read_size - 3; (i--) > 0;)
        {
            if (((*(buf + i))     == (MZ_ZIP_MAGIC_ENDHEADER & 0xff)) &&
                ((*(buf + i + 1)) == (MZ_ZIP_MAGIC_ENDHEADER >> 8 & 0xff)) &&
                ((*(buf + i + 2)) == (MZ_ZIP_MAGIC_ENDHEADER >> 16 & 0xff)) &&
                ((*(buf + i + 3)) == (MZ_ZIP_MAGIC_ENDHEADER >> 24 & 0xff)))
            {
                *central_pos = read_pos + i;
                return MZ_OK;
            }
        }

        if (*central_pos != 0)
            break;
    }

    return MZ_EXIST_ERROR;
}

// Locate the central directory 64 of a zip file (at the end, just before the global comment)
static int32_t mz_zip_search_zip64_cd(void *stream, const uint64_t end_central_offset, uint64_t *central_pos)
{
    uint64_t offset = 0;
    uint32_t value32 = 0;
    int16_t err = MZ_OK;


    *central_pos = 0;

    // Zip64 end of central directory locator
    err = mz_stream_seek(stream, end_central_offset - MZ_ZIP_SIZE_CD_LOCATOR64, MZ_STREAM_SEEK_SET);
    // Read locator signature
    if (err == MZ_OK)
    {
        err = mz_stream_read_uint32(stream, &value32);
        if (value32 != MZ_ZIP_MAGIC_ENDLOCHEADER64)
            err = MZ_FORMAT_ERROR;
    }
    // Number of the disk with the start of the zip64 end of  central directory
    if (err == MZ_OK)
        err = mz_stream_read_uint32(stream, &value32);
    // Relative offset of the zip64 end of central directory record8
    if (err == MZ_OK)
        err = mz_stream_read_uint64(stream, &offset);
    // Total number of disks
    if (err == MZ_OK)
        err = mz_stream_read_uint32(stream, &value32);
    // Goto end of central directory record
    if (err == MZ_OK)
        err = mz_stream_seek(stream, offset, MZ_STREAM_SEEK_SET);
    // The signature
    if (err == MZ_OK)
    {
        err = mz_stream_read_uint32(stream, &value32);
        if (value32 != MZ_ZIP_MAGIC_ENDHEADER64)
            err = MZ_FORMAT_ERROR;
    }

    if (err == MZ_OK)
        *central_pos = offset;

    return err;
}

int mz_zip_read_cd(void *handle)
{
    mz_zip *zip = NULL;
    uint64_t number_entry_cd = 0;
    uint64_t number_entry_cd64 = 0;
    uint64_t number_entry = 0;
    uint64_t central_pos = 0;
    uint64_t central_pos64 = 0;
    uint16_t value16 = 0;
    uint32_t value32 = 0;
    uint64_t value64 = 0;
    int16_t err = MZ_OK;


    zip = (mz_zip *)handle;
    if (zip == NULL)
        return MZ_PARAM_ERROR;

    // Read and cache central directory records
    if (mz_zip_search_cd(zip->stream, &central_pos) == MZ_OK)
    {
        // Read end of central directory info
        err = mz_stream_seek(zip->stream, central_pos, MZ_STREAM_SEEK_SET);
        // The signature, already checked
        if (err == MZ_OK)
            err = mz_stream_read_uint32(zip->stream, &value32);
        // Number of this disk
        if (err == MZ_OK)
            err = mz_stream_read_uint16(zip->stream, &value16);
        zip->number_disk = value16;
        // Number of the disk with the start of the central directory
        if (err == MZ_OK)
            err = mz_stream_read_uint16(zip->stream, &value16);
        zip->global_info.number_disk_with_cd = value16;
        // Total number of entries in the central dir on this disk
        if (err == MZ_OK)
            err = mz_stream_read_uint16(zip->stream, &value16);
        zip->global_info.number_entry = value16;
        // Total number of entries in the central dir
        if (err == MZ_OK)
            err = mz_stream_read_uint16(zip->stream, &value16);
        number_entry_cd = value16;
        if (number_entry_cd != zip->global_info.number_entry)
            err = MZ_FORMAT_ERROR;
        // Size of the central directory
        if (err == MZ_OK)
            err = mz_stream_read_uint32(zip->stream, &value32);
        if (err == MZ_OK)
            zip->cd_size = value32;
        // Offset of start of central directory with respect to the starting disk number
        if (err == MZ_OK)
            err = mz_stream_read_uint32(zip->stream, &value32);
        zip->cd_offset = value32;
        // Zip file global comment length
        if (err == MZ_OK)
            err = mz_stream_read_uint16(zip->stream, &zip->global_info.comment_size);

        if ((err == MZ_OK) && ((number_entry_cd == UINT16_MAX) || (zip->cd_offset == UINT32_MAX)))
        {
            // Format should be Zip64, as the central directory or file size is too large
            if (mz_zip_search_zip64_cd(zip->stream, central_pos, &central_pos64) == MZ_OK)
            {
                central_pos = central_pos64;

                err = mz_stream_seek(zip->stream, central_pos, MZ_STREAM_SEEK_SET);
                // The signature, already checked
                if (err == MZ_OK)
                    err = mz_stream_read_uint32(zip->stream, &value32);
                // Size of zip64 end of central directory record
                if (err == MZ_OK)
                    err = mz_stream_read_uint64(zip->stream, &value64);
                // Version made by
                if (err == MZ_OK)
                    err = mz_stream_read_uint16(zip->stream, &value16);
                // Version needed to extract
                if (err == MZ_OK)
                    err = mz_stream_read_uint16(zip->stream, &value16);
                // Number of this disk
                if (err == MZ_OK)
                    err = mz_stream_read_uint32(zip->stream, &zip->number_disk);
                // Number of the disk with the start of the central directory
                if (err == MZ_OK)
                    err = mz_stream_read_uint32(zip->stream, &zip->global_info.number_disk_with_cd);
                // Total number of entries in the central directory on this disk
                if (err == MZ_OK)
                    err = mz_stream_read_uint64(zip->stream, &number_entry);
                // Total number of entries in the central directory
                if (err == MZ_OK)
                    err = mz_stream_read_uint64(zip->stream, &number_entry_cd64);
                if (number_entry == UINT32_MAX)
                    zip->global_info.number_entry = number_entry_cd64;
                // Size of the central directory
                if (err == MZ_OK)
                    err = mz_stream_read_uint64(zip->stream, &zip->cd_size);
                // Offset of start of central directory with respect to the starting disk number
                if (err == MZ_OK)
                    err = mz_stream_read_uint64(zip->stream, &zip->cd_offset);
            }
            else if ((zip->global_info.number_entry == UINT16_MAX) || (number_entry_cd != zip->global_info.number_entry) ||
                     (zip->cd_size == UINT16_MAX) || (zip->cd_offset == UINT32_MAX))
            {
                err = MZ_FORMAT_ERROR;
            }
        }
    }

    if (err == MZ_OK)
    {
        if (central_pos < zip->cd_offset + zip->cd_size)
            err = MZ_FORMAT_ERROR;
    }

    if ((err == MZ_OK) && (zip->global_info.comment_size > 0))
    {
        zip->global_info.comment = (char *)malloc(zip->global_info.comment_size + 1);
        if (zip->global_info.comment)
        {
            if (mz_stream_read(zip->stream, zip->global_info.comment, 
                    zip->global_info.comment_size) != zip->global_info.comment_size)
                err = MZ_STREAM_ERROR;

            zip->global_info.comment[zip->global_info.comment_size] = 0;
        }
    }

    if (err == MZ_OK)
    {
        zip->disk_offset = central_pos - (zip->cd_offset + zip->cd_size);
        zip->cd_pos = zip->disk_offset;
    }

    return err;
}

int mz_zip_write_cd(void *handle, const char *global_comment, uint16_t version_madeby)
{
    mz_zip *zip = NULL;
    uint16_t comment_size = 0;
    uint64_t zip64_eocd_pos_inzip = 0;
    uint64_t pos = 0;
    uint64_t cd_pos = 0;
    int64_t disk_number = 0;
    int16_t err = MZ_OK;


    zip = (mz_zip *)handle;
    if (zip == NULL)
        return MZ_PARAM_ERROR;

    if (global_comment == NULL)
        global_comment = zip->global_info.comment;
    
    if (mz_stream_get_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, &disk_number) == MZ_OK)
        zip->number_disk_with_CD = (uint32_t)disk_number + 1;
    mz_stream_set_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, -1);

    zip->cd_pos = mz_stream_tell(zip->stream);
    mz_stream_seek(zip->cd_stream, 0, MZ_STREAM_SEEK_END);
    zip->cd_size = (uint32_t)mz_stream_tell(zip->cd_stream);
    mz_stream_seek(zip->cd_stream, 0, MZ_STREAM_SEEK_SET);
    
    err = mz_stream_copy(zip->stream, zip->cd_stream, (int32_t)zip->cd_size);

    mz_stream_close(zip->cd_stream);
    mz_stream_delete(&zip->cd_stream);

    // Write the ZIP64 central directory header
    if (pos >= UINT32_MAX || zip->global_info.number_entry > UINT32_MAX)
    {
        zip64_eocd_pos_inzip = mz_stream_tell(zip->stream);

        err = mz_stream_write_uint32(zip->stream, MZ_ZIP_MAGIC_ENDHEADER64);

        // Size of this 'zip64 end of central directory'
        if (err == MZ_OK)
            err = mz_stream_write_uint64(zip->stream, (uint64_t)44);
        // Version made by
        if (err == MZ_OK)
            err = mz_stream_write_uint16(zip->stream, version_madeby);
        // Version needed
        if (err == MZ_OK)
            err = mz_stream_write_uint16(zip->stream, (uint16_t)45);
        // Number of this disk
        if (err == MZ_OK)
            err = mz_stream_write_uint32(zip->stream, zip->number_disk_with_CD);
        // Number of the disk with the start of the central directory
        if (err == MZ_OK)
            err = mz_stream_write_uint32(zip->stream, zip->number_disk_with_CD);
        // Total number of entries in the central dir on this disk
        if (err == MZ_OK)
            err = mz_stream_write_uint64(zip->stream, zip->global_info.number_entry);
        // Total number of entries in the central dir
        if (err == MZ_OK)
            err = mz_stream_write_uint64(zip->stream, zip->global_info.number_entry);
        // Size of the central directory
        if (err == MZ_OK)
            err = mz_stream_write_uint64(zip->stream, (uint64_t)zip->cd_size);
        // Offset of start of central directory with respect to the starting disk number
        if (err == MZ_OK)
        {
            cd_pos = zip->cd_pos - zip->disk_offset;
            err = mz_stream_write_uint64(zip->stream, cd_pos);
        }
        if (err == MZ_OK)
            err = mz_stream_write_uint32(zip->stream, MZ_ZIP_MAGIC_ENDLOCHEADER64);
        // Number of the disk with the start of the central directory
        if (err == MZ_OK)
            err = mz_stream_write_uint32(zip->stream, zip->number_disk_with_CD);
        // Relative offset to the end of zip64 central directory
        if (err == MZ_OK)
        {
            cd_pos = zip64_eocd_pos_inzip - zip->cd_pos;
            err = mz_stream_write_uint64(zip->stream, cd_pos);
        }
        // Number of the disk with the start of the central directory
        if (err == MZ_OK)
            err = mz_stream_write_uint32(zip->stream, zip->number_disk_with_CD + 1);
    }

    // Write the central directory header

    // Signature 
    if (err == MZ_OK)
        err = mz_stream_write_uint32(zip->stream, MZ_ZIP_MAGIC_ENDHEADER);
    // Number of this disk
    if (err == MZ_OK)
        err = mz_stream_write_uint16(zip->stream, (uint16_t)zip->number_disk_with_CD);
    // Number of the disk with the start of the central directory
    if (err == MZ_OK)
        err = mz_stream_write_uint16(zip->stream, (uint16_t)zip->number_disk_with_CD);
    // Total number of entries in the central dir on this disk
    if (err == MZ_OK)
    {
        if (zip->global_info.number_entry >= UINT16_MAX)
            err = mz_stream_write_uint16(zip->stream, UINT16_MAX);
        else
            err = mz_stream_write_uint16(zip->stream, (uint16_t)zip->global_info.number_entry);
    }
    // Total number of entries in the central dir
    if (err == MZ_OK)
    {
        if (zip->global_info.number_entry >= UINT16_MAX)
            err = mz_stream_write_uint16(zip->stream, UINT16_MAX);
        else
            err = mz_stream_write_uint16(zip->stream, (uint16_t)zip->global_info.number_entry);
    }
    // Size of the central directory
    if (err == MZ_OK)
        err = mz_stream_write_uint32(zip->stream, (uint32_t)zip->cd_size);
    // Offset of start of central directory with respect to the starting disk number
    if (err == MZ_OK)
    {
        cd_pos = zip->cd_pos - zip->disk_offset;
        if (pos >= UINT32_MAX)
            err = mz_stream_write_uint32(zip->stream, UINT32_MAX);
        else
            err = mz_stream_write_uint32(zip->stream, (uint32_t)cd_pos);
    }

    // Write global comment
    if (global_comment != NULL)
        comment_size = (uint16_t)strlen(global_comment);
    if (err == MZ_OK)
        err = mz_stream_write_uint16(zip->stream, comment_size);
    if (err == MZ_OK)
    {
        if (mz_stream_write(zip->stream, global_comment, comment_size) != comment_size)
            err = MZ_STREAM_ERROR;
    }
    return err;
}

extern void* ZEXPORT mz_zip_open(void *stream, int32_t mode)
{
    mz_zip *zip = NULL;
    int16_t err = MZ_OK;


    zip = (mz_zip *)malloc(sizeof(mz_zip));
    if (zip == NULL)
        return NULL;

    memset(zip, 0, sizeof(mz_zip));

    zip->stream = stream;
    zip->open_mode = mode;

    if (zip->open_mode & MZ_STREAM_MODE_WRITE)
    {
        mz_stream_mem_create(&zip->cd_stream);
        mz_stream_mem_set_grow(zip->cd_stream, 1);
        mz_stream_mem_open(zip->cd_stream, NULL, MZ_STREAM_MODE_CREATE);
    }

    if ((zip->open_mode & MZ_STREAM_MODE_READ) || (mode & MZ_STREAM_MODE_APPEND))
    {
        err = mz_zip_read_cd(zip);

        if ((err == MZ_OK) && (mode & MZ_STREAM_MODE_APPEND))
        {
            // Store central directory in memory
            if (mz_stream_seek(zip->stream, zip->cd_offset + zip->disk_offset, MZ_STREAM_SEEK_SET) != MZ_OK)
                err = MZ_STREAM_ERROR;
            if (mz_stream_copy(zip->cd_stream, zip->stream, (uint32_t)zip->cd_size) != MZ_OK)
                err = MZ_STREAM_ERROR;
            if (mz_stream_seek(zip->stream, zip->cd_offset + zip->disk_offset, MZ_STREAM_SEEK_SET) != MZ_OK)
                err = MZ_STREAM_ERROR;
        }
    }

    if (err == MZ_OK)
    {
        mz_stream_mem_create(&zip->file_info_stream);
        mz_stream_mem_set_grow(zip->file_info_stream, 1);
        mz_stream_mem_set_grow_size(zip->file_info_stream, 4096);

        err = mz_stream_mem_open(zip->file_info_stream, NULL, MZ_STREAM_MODE_CREATE);
    }

    if (err == MZ_OK)
    {
        mz_stream_mem_create(&zip->local_file_info_stream);
        mz_stream_mem_set_grow(zip->local_file_info_stream, 1);
        mz_stream_mem_set_grow_size(zip->local_file_info_stream, 4096);

        err = mz_stream_mem_open(zip->local_file_info_stream, NULL, MZ_STREAM_MODE_CREATE);
    }

    if (err != MZ_OK)
    {
        if (zip->file_info_stream != NULL)
            mz_stream_mem_delete(&zip->file_info_stream);
        if (zip->local_file_info_stream != NULL)
            mz_stream_mem_delete(&zip->local_file_info_stream);

        mz_stream_close(zip->cd_stream);
        mz_stream_delete(&zip->cd_stream);

        if (zip->global_info.comment)
            free(zip->global_info.comment);

        free(zip);
        return NULL;
    }

    return zip;
}

extern int ZEXPORT mz_zip_close(void *handle, const char *global_comment, uint16_t version_madeby)
{
    mz_zip *zip = NULL;

    int16_t err = MZ_OK;

    if (handle == NULL)
        return MZ_PARAM_ERROR;
    zip = (mz_zip *)handle;

    if (zip->entry_opened == 1)
    {
        err = mz_zip_entry_close(handle);
        if (err != MZ_OK)
            return err;
    }

    if (zip->open_mode & MZ_STREAM_MODE_WRITE)
        err = mz_zip_write_cd(handle, global_comment, version_madeby);

    mz_stream_mem_close(zip->file_info_stream);
    mz_stream_mem_delete(&zip->file_info_stream);

    mz_stream_mem_close(zip->local_file_info_stream);
    mz_stream_mem_delete(&zip->local_file_info_stream);

    if (zip->global_info.comment)
        free(zip->global_info.comment);

    free(zip);

    return err;
}

extern int ZEXPORT mz_zip_get_global_info(void *handle, mz_zip_global **global_info)
{
    mz_zip *zip = NULL;

    if (handle == NULL || global_info == NULL)
        return MZ_PARAM_ERROR;
    zip = (mz_zip *)handle;
    *global_info = &zip->global_info;
    return MZ_OK;
}

static int mz_zip_entry_get_version_needed(mz_zip_file *file_info)
{
    mz_zip *zip = NULL;
    int16_t version_needed = 20;

    if (file_info == NULL)
        return MZ_PARAM_ERROR;

    if (file_info->version_needed > 0)
        version_needed = file_info->version_needed;

    // Calculate extra field size
    if (file_info->zip_64)
        version_needed = 45;
#ifdef HAVE_AES
    if ((file_info->flag & MZ_ZIP_FLAG_ENCRYPTED) && (file_info->aes_version))
        version_needed = 51;
#endif
#ifdef HAVE_LZMA
    if (file_info->compression_method == MZ_COMPRESS_METHOD_LZMA)
        version_needed = 63;
#endif
    return version_needed;
}

// Get info about the current file in the zip file
static int mz_zip_entry_read_header(void *stream, uint8_t local, mz_zip_file *file_info, void *file_info_stream)
{
    uint32_t magic = 0;
    uint32_t extra_pos = 0;
    uint16_t extra_header_id = 0;
    uint16_t extra_data_size = 0;
    uint16_t value16 = 0;
    uint32_t value32 = 0;
    uint64_t value64 = 0;
    int64_t seek = 0;
    int16_t err = MZ_OK;


    memset(file_info, 0, sizeof(mz_zip_file));

    // Check the magic
    err = mz_stream_read_uint32(stream, &magic);
    if (magic == MZ_ZIP_MAGIC_ENDHEADER || magic == MZ_ZIP_MAGIC_ENDHEADER64)
        err = MZ_END_OF_LIST;
    else if ((local) && (magic != MZ_ZIP_MAGIC_LOCALHEADER))
        err = MZ_FORMAT_ERROR;
    else if ((!local) && (magic != MZ_ZIP_MAGIC_CENTRALHEADER))
        err = MZ_FORMAT_ERROR;

    // Read header fields
    if (err == MZ_OK)
    {
        if (!local)
            err = mz_stream_read_uint16(stream, &file_info->version_madeby);
        if (err == MZ_OK)
            err = mz_stream_read_uint16(stream, &file_info->version_needed);
        if (err == MZ_OK)
            err = mz_stream_read_uint16(stream, &file_info->flag);
        if (err == MZ_OK)
            err = mz_stream_read_uint16(stream, &file_info->compression_method);
        if (err == MZ_OK)
            err = mz_stream_read_uint32(stream, &file_info->dos_date);
        if (err == MZ_OK)
            err = mz_stream_read_uint32(stream, &file_info->crc);
        if (err == MZ_OK)
            err = mz_stream_read_uint32(stream, &value32);
        file_info->compressed_size = value32;
        if (err == MZ_OK)
            err = mz_stream_read_uint32(stream, &value32);
        file_info->uncompressed_size = value32;
        if (err == MZ_OK)
            err = mz_stream_read_uint16(stream, &file_info->filename_size);
        if (err == MZ_OK)
            err = mz_stream_read_uint16(stream, &file_info->extrafield_size);
        if (!local)
        {
            if (err == MZ_OK)
                err = mz_stream_read_uint16(stream, &file_info->comment_size);
            if (err == MZ_OK)
                err = mz_stream_read_uint16(stream, &value16);
            file_info->disk_num_start = value16;
            if (err == MZ_OK)
                err = mz_stream_read_uint16(stream, &file_info->internal_fa);
            if (err == MZ_OK)
                err = mz_stream_read_uint32(stream, &file_info->external_fa);
            if (err == MZ_OK)
                err = mz_stream_read_uint32(stream, &value32);
            file_info->disk_offset = value32;
        }
    }

    if (err == MZ_OK)
        err = mz_stream_seek(file_info_stream, 0, MZ_STREAM_SEEK_SET);

    if ((err == MZ_OK) && (file_info->filename_size > 0))
    {
        // Read filename in our memory stream buffer
        mz_stream_mem_get_buffer(file_info_stream, (void **)&file_info->filename);
        
        if (err == MZ_OK)
            err = mz_stream_copy(file_info_stream, stream, file_info->filename_size);
        if (err == MZ_OK)
            err = mz_stream_write_uint8(file_info_stream, 0);

        seek += file_info->filename_size + 1;
    }

    if ((err == MZ_OK) && (file_info->extrafield_size > 0))
    {
        mz_stream_mem_get_buffer_at(file_info_stream, seek, (void **)&file_info->extrafield);

        err = mz_stream_copy(file_info_stream, stream, file_info->extrafield_size);
        if (err == MZ_OK)
            err = mz_stream_write_uint8(file_info_stream, 0);

        // Seek back and parse the extra field
        if (err == MZ_OK)
            err = mz_stream_seek(file_info_stream, seek, MZ_STREAM_SEEK_SET);

        seek += file_info->extrafield_size + 1;

        while ((err == MZ_OK) && (extra_pos < file_info->extrafield_size))
        {
            err = mz_stream_read_uint16(file_info_stream, &extra_header_id);
            if (err == MZ_OK)
                err = mz_stream_read_uint16(file_info_stream, &extra_data_size);

            // ZIP64 extra field
            if (extra_header_id == 0x0001)
            {
                if ((err == MZ_OK) && (file_info->uncompressed_size == UINT32_MAX))
                    err = mz_stream_read_uint64(file_info_stream, &file_info->uncompressed_size);
                if ((err == MZ_OK) && (file_info->compressed_size == UINT32_MAX))
                    err = mz_stream_read_uint64(file_info_stream, &file_info->compressed_size);
                if ((err == MZ_OK) && (file_info->disk_offset == UINT32_MAX))
                    err = mz_stream_read_uint64(file_info_stream, &value64);
                file_info->disk_offset = value64;
                if ((err == MZ_OK) && (file_info->disk_num_start == UINT32_MAX))
                    err = mz_stream_read_uint32(file_info_stream, &file_info->disk_num_start);
            }
#ifdef HAVE_AES
            // AES extra field
            else if (extra_header_id == 0x9901)
            {
                uint8_t value8 = 0;
                // Verify version info
                err = mz_stream_read_uint16(file_info_stream, &value16);
                // Support AE-1 and AE-2
                if (value16 != 1 && value16 != 2)
                    err = MZ_FORMAT_ERROR;
                file_info->aes_version = value16;
                if (err == MZ_OK)
                    err = mz_stream_read_uint8(file_info_stream, &value8);
                if ((char)value8 != 'A')
                    err = MZ_FORMAT_ERROR;
                if (err == MZ_OK)
                    err = mz_stream_read_uint8(file_info_stream, &value8);
                if ((char)value8 != 'E')
                    err = MZ_FORMAT_ERROR;
                // Get AES encryption strength and actual compression method
                if (err == MZ_OK)
                    err = mz_stream_read_uint8(file_info_stream, &value8);
                file_info->aes_encryption_mode = value8;
                if (err == MZ_OK)
                    err = mz_stream_read_uint16(file_info_stream, &value16);
                file_info->compression_method = value16;
            }
#endif
            else
            {
                err = mz_stream_seek(file_info_stream, extra_data_size, MZ_STREAM_SEEK_CUR);
            }

            extra_pos += 4 + extra_data_size;
        }
    }

    if ((err == MZ_OK) && (file_info->comment_size > 0))
    {
        mz_stream_mem_get_buffer_at(file_info_stream, seek, (void **)&file_info->comment);

        err = mz_stream_copy(file_info_stream, stream, file_info->comment_size);
        if (err == MZ_OK)
            err = mz_stream_write_uint8(file_info_stream, 0);
    }

    return err;
}

static int mz_zip_entry_write_header(void *stream, uint8_t local, mz_zip_file *file_info)
{
    uint16_t extrafield_size = 0;
    uint16_t extrafield_zip64_size = 0;
    uint16_t filename_size = 0;
    uint16_t comment_size = 0;
    int16_t err = MZ_OK;

    if (file_info == NULL)
        return MZ_PARAM_ERROR;

    if (file_info->disk_offset >= UINT32_MAX)
        file_info->zip_64 = 1;

    extrafield_size = file_info->extrafield_size;
    if (file_info->zip_64)
    {
        extrafield_size += 4;
        if (file_info->uncompressed_size >= UINT32_MAX)
            extrafield_zip64_size += 8;
        if (file_info->compressed_size >= UINT32_MAX)
            extrafield_zip64_size += 8;
        if (file_info->disk_offset >= UINT32_MAX)
            extrafield_zip64_size += 8;
        extrafield_size += extrafield_zip64_size;
    }
#ifdef HAVE_AES
    if ((file_info->flag & MZ_ZIP_FLAG_ENCRYPTED) && (file_info->aes_version))
        extrafield_size += 4 + 7;
#endif

    file_info->version_needed = mz_zip_entry_get_version_needed(file_info);

    if (local)
    {
        err = mz_stream_write_uint32(stream, MZ_ZIP_MAGIC_LOCALHEADER);
    }
    else
    {
        err = mz_stream_write_uint32(stream, MZ_ZIP_MAGIC_CENTRALHEADER);
        if (err == MZ_OK)
            err = mz_stream_write_uint16(stream, file_info->version_madeby);
    }

    if (err == MZ_OK)
        err = mz_stream_write_uint16(stream, file_info->version_needed);
    if (err == MZ_OK)
        err = mz_stream_write_uint16(stream, file_info->flag);
    if (err == MZ_OK)
    {
#ifdef HAVE_AES
        if (file_info->aes_version > 0)
            err = mz_stream_write_uint16(stream, MZ_AES_METHOD);
        else
#endif
            err = mz_stream_write_uint16(stream, file_info->compression_method);
    }
    if (err == MZ_OK)
        err = mz_stream_write_uint32(stream, file_info->dos_date);

    if (err == MZ_OK)
        err = mz_stream_write_uint32(stream, file_info->crc); // crc
    if (err == MZ_OK)
    {
        if (file_info->compressed_size >= UINT32_MAX) // compr size
            err = mz_stream_write_uint32(stream, UINT32_MAX);
        else
            err = mz_stream_write_uint32(stream, (uint32_t)file_info->compressed_size);
    }
    if (err == MZ_OK)
    {
        if (file_info->uncompressed_size >= UINT32_MAX) // uncompr size
            err = mz_stream_write_uint32(stream, UINT32_MAX);
        else
            err = mz_stream_write_uint32(stream, (uint32_t)file_info->uncompressed_size);
    }

    filename_size = (uint16_t)strlen(file_info->filename);
    if (err == MZ_OK)
        err = mz_stream_write_uint16(stream, filename_size);
    if (err == MZ_OK)
        err = mz_stream_write_uint16(stream, extrafield_size);

    if (!local)
    {
        if (file_info->comment != NULL)
            comment_size = (uint16_t)strlen(file_info->comment);
        if (err == MZ_OK)
            err = mz_stream_write_uint16(stream, comment_size);
        if (err == MZ_OK)
            err = mz_stream_write_uint16(stream, (uint16_t)file_info->disk_num_start);
        if (err == MZ_OK)
            err = mz_stream_write_uint16(stream, file_info->internal_fa);
        if (err == MZ_OK)
            err = mz_stream_write_uint32(stream, file_info->external_fa);
        if (err == MZ_OK)
        {
            if (file_info->disk_offset >= UINT32_MAX)
                err = mz_stream_write_uint32(stream, UINT32_MAX);
            else
                err = mz_stream_write_uint32(stream, (uint32_t)file_info->disk_offset);
        }
    }

    if (err == MZ_OK)
    {
        if (mz_stream_write(stream, file_info->filename, filename_size) != filename_size)
            err = MZ_STREAM_ERROR;
    }
    if (err == MZ_OK)
    {
        if (mz_stream_write(stream, file_info->extrafield, file_info->extrafield_size) != file_info->extrafield_size)
            err = MZ_STREAM_ERROR;
    }
    // Add ZIP64 extra info header to central directory
    if ((err == MZ_OK) && (file_info->zip_64))
    {
        err = mz_stream_write_uint16(stream, 0x0001);
        if (err == MZ_OK)
            err = mz_stream_write_uint16(stream, extrafield_zip64_size);
        if ((err == MZ_OK) && (file_info->uncompressed_size >= UINT32_MAX))
            err = mz_stream_write_uint64(stream, file_info->uncompressed_size);
        if ((err == MZ_OK) && (file_info->compressed_size >= UINT32_MAX))
            err = mz_stream_write_uint64(stream, file_info->compressed_size);
        if ((err == MZ_OK) && (file_info->disk_offset >= UINT32_MAX))
            err = mz_stream_write_uint64(stream, file_info->disk_offset);
    }

#ifdef HAVE_AES
    // Write AES extra info header to central directory
    if ((err == MZ_OK) && (file_info->flag & MZ_ZIP_FLAG_ENCRYPTED) && (file_info->aes_version))
    {
        err = mz_stream_write_uint16(stream, 0x9901);
        if (err == MZ_OK)
            err = mz_stream_write_uint16(stream, 7);
        if (err == MZ_OK)
            err = mz_stream_write_uint16(stream, file_info->aes_version);
        if (err == MZ_OK)
            err = mz_stream_write_uint8(stream, 'A');
        if (err == MZ_OK)
            err = mz_stream_write_uint8(stream, 'E');
        if (err == MZ_OK)
            err = mz_stream_write_uint8(stream, file_info->aes_encryption_mode);
        if (err == MZ_OK)
            err = mz_stream_write_uint16(stream, file_info->compression_method);
    }
#endif
    if ((err == MZ_OK) && (file_info->comment != NULL))
    {
        if (mz_stream_write(stream, file_info->comment, file_info->comment_size) != MZ_OK)
            err = MZ_STREAM_ERROR;
        free(file_info->comment);
    }

    return err;
}

static int mz_zip_entry_open_int(void *handle, const char *password)
{
    mz_zip *zip = NULL;
    uint64_t extrafield_local_offset = 0;
    uint16_t extrafield_local_size = 0;
    uint32_t size_variable = 0;
    int64_t max_total_in = 0;
    int64_t total_in = 0;
    int64_t footer_size = 0;
    int16_t err = MZ_OK;

    if (handle == NULL)
        return MZ_PARAM_ERROR;
    zip = (mz_zip *)handle;

    if ((zip->file_info.compression_method != MZ_COMPRESS_METHOD_RAW) &&
        (zip->file_info.compression_method != MZ_COMPRESS_METHOD_DEFLATE))
    {
#ifdef HAVE_BZIP2
        if (zip->file_info.compression_method != MZ_COMPRESS_METHOD_BZIP2)
            return MZ_INTERNAL_ERROR;
#elif HAVE_LZMA
        if (unzip->file_info.compression_method != MZ_METHOD_LZMA)
            return MZ_INTERNAL_ERROR;
#else
        return MZ_INTERNAL_ERROR;
#endif
    }

    if ((zip->file_info.flag & MZ_ZIP_FLAG_ENCRYPTED) && (password == NULL))
        return MZ_PARAM_ERROR;

    max_total_in = zip->file_info.compressed_size;

    if ((err == Z_OK) && (zip->file_info.flag & MZ_ZIP_FLAG_ENCRYPTED))
    {
#ifdef HAVE_AES
        if (zip->file_info.aes_version)
        {
            mz_stream_aes_create(&zip->crypt_stream);
            mz_stream_aes_set_password(zip->crypt_stream, password);
            mz_stream_aes_set_encryption_mode(zip->crypt_stream, zip->file_info.aes_encryption_mode);

            if (mz_stream_get_prop_int64(zip->crypt_stream, MZ_STREAM_PROP_FOOTER_SIZE, &footer_size) == MZ_OK)
                max_total_in -= footer_size;
        }
        else
#endif
        {
#ifdef HAVE_CRYPT
            uint8_t verify1 = 0;
            uint8_t verify2 = 0;

            // Info-ZIP modification to ZipCrypto format:
            // If bit 3 of the general purpose bit flag is set, it uses high byte of 16-bit File Time.

            verify1 = (uint8_t)((zip->file_info.dos_date >> 16) & 0xff);
            verify2 = (uint8_t)((zip->file_info.dos_date >> 8) & 0xff);

            mz_stream_crypt_create(&zip->crypt_stream);
            mz_stream_crypt_set_password(zip->crypt_stream, password);
            mz_stream_crypt_set_verify(zip->crypt_stream, verify1, verify2);
#endif
        }
    }

    if (err == MZ_OK)
    {
        if (zip->crypt_stream == NULL)
        {
            mz_stream_passthru_create(&zip->crypt_stream);
            mz_stream_set_base(zip->crypt_stream, zip->stream);
        }
        else
        {
            mz_stream_set_base(zip->crypt_stream, zip->stream);

            err = mz_stream_open(zip->crypt_stream, NULL, zip->open_mode);
        }

        if (mz_stream_get_prop_int64(zip->crypt_stream, MZ_STREAM_PROP_TOTAL_IN, &total_in) == MZ_OK)
            max_total_in -= total_in;
    }

    if (err == MZ_OK)
    {
        if (zip->file_info.compression_method == MZ_COMPRESS_METHOD_RAW)
        {
            mz_stream_passthru_create(&zip->compress_stream);
            mz_stream_set_base(zip->compress_stream, zip->crypt_stream);
        }
        else
        {
            if (zip->file_info.compression_method == MZ_COMPRESS_METHOD_DEFLATE)
                mz_stream_zlib_create(&zip->compress_stream);
#ifdef HAVE_BZIP2
            else if (zip->file_info.compression_method == MZ_COMPRESS_METHOD_BZIP2)
                mz_stream_bzip_create(&zip->compress_stream);
#endif
#ifdef HAVE_LZMA
            else if (zip->file_info.compression_method == MZ_COMPRESS_METHOD_LZMA)
                mz_stream_lzma_create(&zip->compress_stream);
#endif
            else
                err = MZ_PARAM_ERROR;

            if (err == MZ_OK)
            {
                if (zip->open_mode & MZ_STREAM_MODE_WRITE)
                {
                    mz_stream_set_prop_int64(zip->compress_stream, MZ_STREAM_PROP_COMPRESS_LEVEL, zip->compress_info.level);
                    mz_stream_set_prop_int64(zip->compress_stream, MZ_STREAM_PROP_COMPRESS_WINDOW_BITS, zip->compress_info.window_bits);
                    mz_stream_set_prop_int64(zip->compress_stream, MZ_STREAM_PROP_COMPRESS_MEM_LEVEL, zip->compress_info.mem_level);
                    mz_stream_set_prop_int64(zip->compress_stream, MZ_STREAM_PROP_COMPRESS_STRATEGY, zip->compress_info.strategy);
                }
                else
                {
                    if (zip->file_info.flag & MZ_ZIP_FLAG_ENCRYPTED)
                        mz_stream_set_prop_int64(zip->compress_stream, MZ_STREAM_PROP_TOTAL_IN_MAX, max_total_in);
                }

                mz_stream_set_base(zip->compress_stream, zip->crypt_stream);

                err = mz_stream_open(zip->compress_stream, NULL, zip->open_mode);
            }
        }
    }

    if (err == Z_OK)
    {
        mz_stream_crc32_create(&zip->crc32_stream);
        mz_stream_set_base(zip->crc32_stream, zip->compress_stream);

        err = mz_stream_open(zip->crc32_stream, NULL, zip->open_mode);
    }

    if (err == MZ_OK)
    {
        zip->entry_opened = 1;
    }

    return err;
}

extern int ZEXPORT mz_zip_entry_read_open(void *handle, int raw, const char *password)
{
    mz_zip *zip = NULL;
    uint64_t extrafield_local_offset = 0;
    uint16_t extrafield_local_size = 0;
    uint32_t size_variable = 0;
    int64_t max_total_in = 0;
    int64_t total_in = 0;
    int64_t footer_size = 0;
    int16_t err = MZ_OK;

#if !defined(HAVE_CRYPT) && !defined(HAVE_AES)
    if (password != NULL)
        return MZ_PARAM_ERROR;
#endif
    if (handle == NULL)
        return MZ_PARAM_ERROR;

    zip = (mz_zip *)handle;

    if (zip->entry_scanned == 0)
        return MZ_PARAM_ERROR;
    if ((zip->file_info.flag & MZ_ZIP_FLAG_ENCRYPTED) && (password == NULL))
        return MZ_PARAM_ERROR;

    if (zip->file_info.disk_num_start == zip->global_info.number_disk_with_cd)
        mz_stream_set_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, -1);
    else
        mz_stream_set_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, zip->file_info.disk_num_start);

    err = mz_stream_seek(zip->stream, zip->disk_offset + zip->file_info.disk_offset, MZ_STREAM_SEEK_SET);
    if (err == MZ_OK)
        err = mz_zip_entry_read_header(zip->stream, 1, &zip->local_file_info, zip->local_file_info_stream);
    if (err == MZ_OK)
        err = mz_zip_entry_open_int(handle, password);

    return err;
}

extern int ZEXPORT mz_zip_entry_write_open(void *handle, const mz_zip_file *file_info,
    const mz_zip_compress *compress_info, const mz_zip_crypt *crypt_info)
{
    mz_zip *zip = NULL;
    int64_t disk_number = 0;
    int16_t err = MZ_OK;

#if !defined(HAVE_CRYPT) && !defined(HAVE_AES)
    if (crypt_info != NULL)
        return MZ_PARAM_ERROR;
#endif
    if (handle == NULL)
        return MZ_PARAM_ERROR;
    if (file_info == NULL || file_info->filename == NULL)
        return MZ_PARAM_ERROR;
    if (compress_info == NULL)
        return MZ_PARAM_ERROR;
    if (crypt_info == NULL)
        return MZ_PARAM_ERROR;

    zip = (mz_zip *)handle;

    if (zip->entry_opened == 1)
    {
        err = mz_zip_entry_close(handle);
        if (err != MZ_OK)
            return err;
    }

    memcpy(&zip->file_info, file_info, sizeof(mz_zip_file));
    memcpy(&zip->compress_info, compress_info, sizeof(mz_zip_compress));

    zip->file_info.flag |= MZ_ZIP_FLAG_DATA_DESCRIPTOR;
#ifdef HAVE_LZMA
    zip->file_info.flag |= MZ_ZIP_FLAG_LZMA_EOS_MARKER;
#endif
    if ((zip->compress_info.level == 8) || (zip->compress_info.level == 9))
        zip->file_info.flag |= MZ_ZIP_FLAG_DEFLATE_MAX;
    if (zip->compress_info.level == 2)
        zip->file_info.flag |= MZ_ZIP_FLAG_DEFLATE_FAST;
    if (zip->compress_info.level == 1)
        zip->file_info.flag |= MZ_ZIP_FLAG_DEFLATE_SUPER_FAST;

    if (crypt_info->password != NULL)
        zip->file_info.flag |= MZ_ZIP_FLAG_ENCRYPTED;
    else
        zip->file_info.flag &= ~MZ_ZIP_FLAG_ENCRYPTED;

    if (mz_stream_get_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, &disk_number) == MZ_OK)
        zip->file_info.disk_num_start = (uint32_t)disk_number;

    zip->file_info.disk_offset = mz_stream_tell(zip->stream);
    zip->file_info.compressed_size = 0;
    zip->file_info.uncompressed_size = 0;
    zip->file_info.compression_method = compress_info->method;
#ifdef HAVE_AES
    if (crypt_info->aes)
    {
        zip->file_info.aes_version = MZ_AES_VERSION;
        zip->file_info.aes_encryption_mode = MZ_AES_ENCRYPTIONMODE;
    }
#endif
    
    if (err == MZ_OK)
        err = mz_zip_entry_write_header(zip->stream, 1, &zip->file_info);
    if (err == MZ_OK)
        err = mz_zip_entry_open_int(handle, crypt_info->password);

    return err;
}

extern int ZEXPORT mz_zip_entry_read(void *handle, void *buf, uint32_t len)
{
    mz_zip *zip = NULL;
    int32_t read = 0;

    if (handle == NULL)
        return MZ_PARAM_ERROR;

    zip = (mz_zip *)handle;

    if (zip->entry_opened == 0)
        return MZ_PARAM_ERROR;

    if (len == 0)
        return 0;
    if (len > UINT16_MAX) // Zlib limitation
        return MZ_PARAM_ERROR;

    read = mz_stream_read(zip->crc32_stream, buf, len);

    if (read > 0)
        zip->entry_read += read;

    return read;
}

extern int ZEXPORT mz_zip_entry_write(void *handle, const void *buf, uint32_t len)
{
    mz_zip *zip = NULL;

    if (handle == NULL)
        return MZ_PARAM_ERROR;

    zip = (mz_zip *)handle;

    if (zip->entry_opened == 0)
        return MZ_PARAM_ERROR;

    return mz_stream_write(zip->crc32_stream, buf, len);
}

extern int ZEXPORT mz_zip_entry_close_raw(void *handle, uint64_t uncompressed_size, uint32_t crc32)
{
    mz_zip *zip = NULL;
    uint64_t compressed_size = 0;
    uint16_t extrafield_size = 0;
    uint16_t extrafield_zip64_size = 0;
    uint16_t filename_size = 0;
    uint16_t comment_size = 0;
    uint16_t version_needed = 0;
    int16_t err = MZ_OK;

    if (handle == NULL)
        return MZ_PARAM_ERROR;

    zip = (mz_zip *)handle;

    if (zip->entry_opened == 0)
        return MZ_PARAM_ERROR;
  
    mz_stream_close(zip->compress_stream);
    crc32 = mz_stream_crc32_get_value(zip->crc32_stream);

    if ((zip->open_mode & MZ_STREAM_MODE_WRITE) == 0)
    {
#ifdef HAVE_AES
        // AES zip version AE-1 will expect a valid crc as well
        if (zip->aes_version <= 0x0001)
#endif
        {
            if ((zip->entry_read > 0) && (zip->file_info.compression_method != MZ_COMPRESS_METHOD_RAW))
            {
                if (crc32 != zip->file_info.crc)
                    err = MZ_CRC_ERROR;
            }
        }
    }
    else
    {
        if ((zip->compress_info.method != MZ_COMPRESS_METHOD_RAW) || (uncompressed_size == 0))
        {
            mz_stream_get_prop_int64(zip->crc32_stream, MZ_STREAM_PROP_TOTAL_OUT, (int64_t *)&uncompressed_size);
            mz_stream_get_prop_int64(zip->compress_stream, MZ_STREAM_PROP_TOTAL_OUT, (int64_t *)&compressed_size);
        }
    }

    if (zip->file_info.flag & MZ_ZIP_FLAG_ENCRYPTED)
    {
        mz_stream_set_base(zip->crypt_stream, zip->stream);

        err = mz_stream_close(zip->crypt_stream);

        if ((zip->compress_info.method != MZ_COMPRESS_METHOD_RAW) || (uncompressed_size == 0))
            mz_stream_get_prop_int64(zip->crypt_stream, MZ_STREAM_PROP_TOTAL_OUT, (int64_t *)&compressed_size);
    }

    mz_stream_delete(&zip->crypt_stream);

    mz_stream_delete(&zip->compress_stream);
    mz_stream_crc32_delete(&zip->crc32_stream);

    if (zip->open_mode & MZ_STREAM_MODE_WRITE)
    {
        if (err == MZ_OK)
        {
            err = mz_stream_write_uint32(zip->stream, MZ_ZIP_MAGIC_DATADESCRIPTOR);
            if (err == MZ_OK)
                err = mz_stream_write_uint32(zip->stream, crc32);
            if (err == MZ_OK)
            {
                if (zip->file_info.zip_64)
                    err = mz_stream_write_uint64(zip->stream, compressed_size);
                else
                    err = mz_stream_write_uint32(zip->stream, (uint32_t)compressed_size);
            }
            if (err == MZ_OK)
            {
                if (zip->file_info.zip_64)
                    err = mz_stream_write_uint64(zip->stream, uncompressed_size);
                else
                    err = mz_stream_write_uint32(zip->stream, (uint32_t)uncompressed_size);
            }
        }

        zip->file_info.crc = crc32;
        zip->file_info.compressed_size = compressed_size;
        zip->file_info.uncompressed_size = uncompressed_size;

        if (err == MZ_OK)
            err = mz_zip_entry_write_header(zip->cd_stream, 0, &zip->file_info);

        zip->global_info.number_entry += 1;
    }

    zip->entry_opened = 0;

    return err;
}

extern int ZEXPORT mz_zip_entry_close(void *handle)
{
    return mz_zip_entry_close_raw(handle, 0, 0);
}

extern int ZEXPORT mz_zip_entry_get_info(void *handle, mz_zip_file **file_info)
{
    mz_zip *zip = NULL;

    if (handle == NULL)
        return MZ_PARAM_ERROR;

    zip = (mz_zip *)handle;

    if (zip->entry_scanned == 0)
        return MZ_PARAM_ERROR;

    *file_info = &zip->file_info;
    return MZ_OK;
}

extern int ZEXPORT mz_zip_entry_get_local_info(void *handle, mz_zip_file **local_file_info)
{
    mz_zip *zip = NULL;

    if (handle == NULL)
        return MZ_PARAM_ERROR;

    zip = (mz_zip *)handle;

    if (zip->entry_scanned == 0 || zip->entry_opened == 0)
        return MZ_PARAM_ERROR;

    *local_file_info = &zip->local_file_info;
    return MZ_OK;
}

static int mz_zip_goto_next_entry_int(void *handle)
{
    mz_zip *zip = NULL;
    int16_t err = MZ_OK;

    if (handle == NULL)
        return MZ_PARAM_ERROR;

    zip = (mz_zip *)handle;
    zip->entry_scanned = 0;

    mz_stream_set_prop_int64(zip->stream, MZ_STREAM_PROP_DISK_NUMBER, -1);

    err = mz_stream_seek(zip->stream, zip->disk_offset + zip->cd_current_pos, MZ_STREAM_SEEK_SET);
    if (err == MZ_OK)
        err = mz_zip_entry_read_header(zip->stream, 0, &zip->file_info, zip->file_info_stream);
    if (err == MZ_OK)
        zip->entry_scanned = 1;
    return err;
}

extern int ZEXPORT mz_zip_goto_first_entry(void *handle)
{
    mz_zip *zip = NULL;

    if (handle == NULL)
        return MZ_PARAM_ERROR;

    zip = (mz_zip *)handle;
    zip->cd_current_pos = zip->cd_offset;

    return mz_zip_goto_next_entry_int(handle);
}

extern int ZEXPORT mz_zip_goto_next_entry(void *handle)
{
    mz_zip *zip = NULL;

    if (handle == NULL)
        return MZ_PARAM_ERROR;

    zip = (mz_zip *)handle;
    zip->cd_current_pos += MZ_ZIP_SIZE_CD_ITEM + zip->file_info.filename_size +
        zip->file_info.extrafield_size + zip->file_info.comment_size;

    return mz_zip_goto_next_entry_int(handle);
}

extern int ZEXPORT mz_zip_locate_entry(void *handle, const char *filename, mz_filename_compare_cb filename_compare_cb)
{
    mz_zip *zip = NULL;
    int16_t err = MZ_OK;
    int32_t result = 0;

    if (handle == NULL)
        return MZ_PARAM_ERROR;

    zip = (mz_zip *)handle;

    err = mz_zip_goto_first_entry(handle);

    while (err == MZ_OK)
    {
        if (filename_compare_cb != NULL)
            result = filename_compare_cb(handle, zip->file_info.filename, filename);
        else
            result = strcmp(zip->file_info.filename, filename);

        if (result == 0)
            return MZ_OK;

        err = mz_zip_goto_next_entry(handle);
    }

    return err;
}