/************************************************************************
 *
 * Copyright (C) 2020 IRCAD France
 * Copyright (C) 2020 IHU Strasbourg
 *
 * This file is part of Sight.
 *
 * Sight is free software: you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Sight is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with Sight. If not, see <https://www.gnu.org/licenses/>.
 *
 ***********************************************************************/

/* ioapi.c -- IO base function header for compress/uncompress .zip
   part of the MiniZip project

   Copyright (C) 2012-2017 Nathan Moinvaziri
     https://github.com/nmoinvaz/minizip
   Modifications for Zip64 support
     Copyright (C) 2009-2010 Mathias Svensson
     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 <stdlib.h>
#include <string.h>

#if defined unix || defined __APPLE__
#include <sys/types.h>
#include <unistd.h>
#endif

#include "minizip/ioapi.h"

#if defined(_WIN32)
#  define snprintf _snprintf
#endif

//------------------------------------------------------------------------------

voidpf call_zopen64(const zlib_filefunc64_32_def* pfilefunc, const void* filename, int mode)
{
    if (pfilefunc->zfile_func64.zopen64_file != NULL)
    {
        return (*(pfilefunc->zfile_func64.zopen64_file))(pfilefunc->zfile_func64.opaque, filename, mode);
    }
    return (*(pfilefunc->zopen32_file))(pfilefunc->zfile_func64.opaque, (const char*)filename, mode);
}

//------------------------------------------------------------------------------

voidpf call_zopendisk64(const zlib_filefunc64_32_def* pfilefunc, voidpf filestream, uint32_t number_disk, int mode)
{
    if (pfilefunc->zfile_func64.zopendisk64_file != NULL)
    {
        return (*(pfilefunc->zfile_func64.zopendisk64_file))(pfilefunc->zfile_func64.opaque, filestream, number_disk,
                                                             mode);
    }
    return (*(pfilefunc->zopendisk32_file))(pfilefunc->zfile_func64.opaque, filestream, number_disk, mode);
}

//------------------------------------------------------------------------------

long call_zseek64(const zlib_filefunc64_32_def* pfilefunc, voidpf filestream, uint64_t offset, int origin)
{
    uint32_t offset_truncated = 0;
    if (pfilefunc->zfile_func64.zseek64_file != NULL)
    {
        return (*(pfilefunc->zfile_func64.zseek64_file))(pfilefunc->zfile_func64.opaque, filestream, offset, origin);
    }
    offset_truncated = (uint32_t)offset;
    if (offset_truncated != offset)
    {
        return -1;
    }
    return (*(pfilefunc->zseek32_file))(pfilefunc->zfile_func64.opaque, filestream, offset_truncated, origin);
}

//------------------------------------------------------------------------------

uint64_t call_ztell64(const zlib_filefunc64_32_def* pfilefunc, voidpf filestream)
{
    uint64_t position;
    if (pfilefunc->zfile_func64.zseek64_file != NULL)
    {
        return (*(pfilefunc->zfile_func64.ztell64_file))(pfilefunc->zfile_func64.opaque, filestream);
    }
    position = (*(pfilefunc->ztell32_file))(pfilefunc->zfile_func64.opaque, filestream);
    if ((position) == UINT32_MAX)
    {
        return (uint64_t)-1;
    }
    return position;
}

//------------------------------------------------------------------------------

void fill_zlib_filefunc64_32_def_from_filefunc32(zlib_filefunc64_32_def* p_filefunc64_32,
                                                 const zlib_filefunc_def* p_filefunc32)
{
    p_filefunc64_32->zfile_func64.zopen64_file     = NULL;
    p_filefunc64_32->zfile_func64.zopendisk64_file = NULL;
    p_filefunc64_32->zopen32_file                  = p_filefunc32->zopen_file;
    p_filefunc64_32->zopendisk32_file              = p_filefunc32->zopendisk_file;
    p_filefunc64_32->zfile_func64.zerror_file      = p_filefunc32->zerror_file;
    p_filefunc64_32->zfile_func64.zread_file       = p_filefunc32->zread_file;
    p_filefunc64_32->zfile_func64.zwrite_file      = p_filefunc32->zwrite_file;
    p_filefunc64_32->zfile_func64.ztell64_file     = NULL;
    p_filefunc64_32->zfile_func64.zseek64_file     = NULL;
    p_filefunc64_32->zfile_func64.zclose_file      = p_filefunc32->zclose_file;
    p_filefunc64_32->zfile_func64.zerror_file      = p_filefunc32->zerror_file;
    p_filefunc64_32->zfile_func64.opaque           = p_filefunc32->opaque;
    p_filefunc64_32->zseek32_file                  = p_filefunc32->zseek_file;
    p_filefunc64_32->ztell32_file                  = p_filefunc32->ztell_file;
}

static voidpf ZCALLBACK fopen_file_func(ZIP_UNUSED voidpf opaque, const char* filename, int mode);
static uint32_t ZCALLBACK fread_file_func(voidpf opaque, voidpf stream, void* buf, uint32_t size);
static uint32_t ZCALLBACK fwrite_file_func(voidpf opaque, voidpf stream, const void* buf, uint32_t size);
static uint64_t ZCALLBACK ftell64_file_func(voidpf opaque, voidpf stream);
static long ZCALLBACK fseek64_file_func(voidpf opaque, voidpf stream, uint64_t offset, int origin);
static int ZCALLBACK fclose_file_func(voidpf opaque, voidpf stream);
static int ZCALLBACK ferror_file_func(voidpf opaque, voidpf stream);

typedef struct
{
    FILE* file;
    int filenameLength;
    void* filename;
} FILE_IOPOSIX;

//------------------------------------------------------------------------------

static voidpf file_build_ioposix(FILE* file, const char* filename)
{
    FILE_IOPOSIX* ioposix = NULL;
    if (file == NULL)
    {
        return NULL;
    }
    ioposix                 = (FILE_IOPOSIX*)malloc(sizeof(FILE_IOPOSIX));
    ioposix->file           = file;
    ioposix->filenameLength = (int)strlen(filename) + 1;
    ioposix->filename       = (char*)malloc(ioposix->filenameLength * sizeof(char));
    strncpy((char*)ioposix->filename, filename, ioposix->filenameLength);
    return (voidpf)ioposix;
}

//------------------------------------------------------------------------------

static voidpf ZCALLBACK fopen_file_func(ZIP_UNUSED voidpf opaque, const char* filename, int mode)
{
    FILE* file             = NULL;
    const char* mode_fopen = NULL;
    if ((mode & ZLIB_FILEFUNC_MODE_READWRITEFILTER) == ZLIB_FILEFUNC_MODE_READ)
    {
        mode_fopen = "rb";
    }
    else if (mode & ZLIB_FILEFUNC_MODE_EXISTING)
    {
        mode_fopen = "r+b";
    }
    else if (mode & ZLIB_FILEFUNC_MODE_CREATE)
    {
        mode_fopen = "wb";
    }

    if ((filename != NULL) && (mode_fopen != NULL))
    {
        file = fopen(filename, mode_fopen);
        return file_build_ioposix(file, filename);
    }
    return file;
}

//------------------------------------------------------------------------------

static voidpf ZCALLBACK fopen64_file_func(ZIP_UNUSED voidpf opaque, const void* filename, int mode)
{
    FILE* file             = NULL;
    const char* mode_fopen = NULL;
    if ((mode & ZLIB_FILEFUNC_MODE_READWRITEFILTER) == ZLIB_FILEFUNC_MODE_READ)
    {
        mode_fopen = "rb";
    }
    else if (mode & ZLIB_FILEFUNC_MODE_EXISTING)
    {
        mode_fopen = "r+b";
    }
    else if (mode & ZLIB_FILEFUNC_MODE_CREATE)
    {
        mode_fopen = "wb";
    }

    if ((filename != NULL) && (mode_fopen != NULL))
    {
        file = fopen64((const char*)filename, mode_fopen);
        return file_build_ioposix(file, (const char*)filename);
    }
    return file;
}

//------------------------------------------------------------------------------

static voidpf ZCALLBACK fopendisk64_file_func(voidpf opaque, voidpf stream, uint32_t number_disk, int mode)
{
    FILE_IOPOSIX* ioposix = NULL;
    char* diskFilename    = NULL;
    voidpf ret            = NULL;
    int i                 = 0;

    if (stream == NULL)
    {
        return NULL;
    }
    ioposix      = (FILE_IOPOSIX*)stream;
    diskFilename = (char*)malloc(ioposix->filenameLength * sizeof(char));
    strncpy(diskFilename, (const char*)ioposix->filename, ioposix->filenameLength);
    for (i = ioposix->filenameLength - 1; i >= 0; i -= 1)
    {
        if (diskFilename[i] != '.')
        {
            continue;
        }
        snprintf(&diskFilename[i], ioposix->filenameLength - i, ".z%02u", number_disk + 1);
        break;
    }
    if (i >= 0)
    {
        ret = fopen64_file_func(opaque, diskFilename, mode);
    }
    free(diskFilename);
    return ret;
}

//------------------------------------------------------------------------------

static voidpf ZCALLBACK fopendisk_file_func(voidpf opaque, voidpf stream, uint32_t number_disk, int mode)
{
    FILE_IOPOSIX* ioposix = NULL;
    char* diskFilename    = NULL;
    voidpf ret            = NULL;
    int i                 = 0;

    if (stream == NULL)
    {
        return NULL;
    }
    ioposix      = (FILE_IOPOSIX*)stream;
    diskFilename = (char*)malloc(ioposix->filenameLength * sizeof(char));
    strncpy(diskFilename, (const char*)ioposix->filename, ioposix->filenameLength);
    for (i = ioposix->filenameLength - 1; i >= 0; i -= 1)
    {
        if (diskFilename[i] != '.')
        {
            continue;
        }
        snprintf(&diskFilename[i], ioposix->filenameLength - i, ".z%02u", number_disk + 1);
        break;
    }
    if (i >= 0)
    {
        ret = fopen_file_func(opaque, diskFilename, mode);
    }
    free(diskFilename);
    return ret;
}

//------------------------------------------------------------------------------

static uint32_t ZCALLBACK fread_file_func(ZIP_UNUSED voidpf opaque, voidpf stream, void* buf, uint32_t size)
{
    FILE_IOPOSIX* ioposix = NULL;
    uint32_t read         = (uint32_t)-1;
    if (stream == NULL)
    {
        return read;
    }
    ioposix = (FILE_IOPOSIX*)stream;
    read    = (uint32_t)fread(buf, 1, (size_t)size, ioposix->file);
    return read;
}

//------------------------------------------------------------------------------

static uint32_t ZCALLBACK fwrite_file_func(ZIP_UNUSED voidpf opaque, voidpf stream, const void* buf, uint32_t size)
{
    FILE_IOPOSIX* ioposix = NULL;
    uint32_t written      = (uint32_t)-1;
    if (stream == NULL)
    {
        return written;
    }
    ioposix = (FILE_IOPOSIX*)stream;
    written = (uint32_t)fwrite(buf, 1, (size_t)size, ioposix->file);
    return written;
}

//------------------------------------------------------------------------------

static long ZCALLBACK ftell_file_func(ZIP_UNUSED voidpf opaque, voidpf stream)
{
    FILE_IOPOSIX* ioposix = NULL;
    long ret              = -1;
    if (stream == NULL)
    {
        return ret;
    }
    ioposix = (FILE_IOPOSIX*)stream;
    ret     = ftell(ioposix->file);
    return ret;
}

//------------------------------------------------------------------------------

static uint64_t ZCALLBACK ftell64_file_func(ZIP_UNUSED voidpf opaque, voidpf stream)
{
    FILE_IOPOSIX* ioposix = NULL;
    uint64_t ret          = (uint64_t)-1;
    if (stream == NULL)
    {
        return ret;
    }
    ioposix = (FILE_IOPOSIX*)stream;
    ret     = ftello64(ioposix->file);
    return ret;
}

//------------------------------------------------------------------------------

static long ZCALLBACK fseek_file_func(ZIP_UNUSED voidpf opaque, voidpf stream, uint32_t offset, int origin)
{
    FILE_IOPOSIX* ioposix = NULL;
    int fseek_origin      = 0;
    long ret              = 0;

    if (stream == NULL)
    {
        return -1;
    }
    ioposix = (FILE_IOPOSIX*)stream;

    switch (origin)
    {
        case ZLIB_FILEFUNC_SEEK_CUR:
            fseek_origin = SEEK_CUR;
            break;
        case ZLIB_FILEFUNC_SEEK_END:
            fseek_origin = SEEK_END;
            break;
        case ZLIB_FILEFUNC_SEEK_SET:
            fseek_origin = SEEK_SET;
            break;
        default:
            return -1;
    }
    if (fseek(ioposix->file, offset, fseek_origin) != 0)
    {
        ret = -1;
    }
    return ret;
}

//------------------------------------------------------------------------------

static long ZCALLBACK fseek64_file_func(ZIP_UNUSED voidpf opaque, voidpf stream, uint64_t offset, int origin)
{
    FILE_IOPOSIX* ioposix = NULL;
    int fseek_origin      = 0;
    long ret              = 0;

    if (stream == NULL)
    {
        return -1;
    }
    ioposix = (FILE_IOPOSIX*)stream;

    switch (origin)
    {
        case ZLIB_FILEFUNC_SEEK_CUR:
            fseek_origin = SEEK_CUR;
            break;
        case ZLIB_FILEFUNC_SEEK_END:
            fseek_origin = SEEK_END;
            break;
        case ZLIB_FILEFUNC_SEEK_SET:
            fseek_origin = SEEK_SET;
            break;
        default:
            return -1;
    }

    if (fseeko64(ioposix->file, offset, fseek_origin) != 0)
    {
        ret = -1;
    }

    return ret;
}

//------------------------------------------------------------------------------

static int ZCALLBACK fclose_file_func(ZIP_UNUSED voidpf opaque, voidpf stream)
{
    FILE_IOPOSIX* ioposix = NULL;
    int ret               = -1;
    if (stream == NULL)
    {
        return ret;
    }
    ioposix = (FILE_IOPOSIX*)stream;
    if (ioposix->filename != NULL)
    {
        free(ioposix->filename);
    }
    ret = fclose(ioposix->file);
    free(ioposix);
    return ret;
}

//------------------------------------------------------------------------------

static int ZCALLBACK ferror_file_func(ZIP_UNUSED voidpf opaque, voidpf stream)
{
    FILE_IOPOSIX* ioposix = NULL;
    int ret               = -1;
    if (stream == NULL)
    {
        return ret;
    }
    ioposix = (FILE_IOPOSIX*)stream;
    ret     = ferror(ioposix->file);
    return ret;
}

//------------------------------------------------------------------------------

void fill_fopen_filefunc(zlib_filefunc_def* pzlib_filefunc_def)
{
    pzlib_filefunc_def->zopen_file     = fopen_file_func;
    pzlib_filefunc_def->zopendisk_file = fopendisk_file_func;
    pzlib_filefunc_def->zread_file     = fread_file_func;
    pzlib_filefunc_def->zwrite_file    = fwrite_file_func;
    pzlib_filefunc_def->ztell_file     = ftell_file_func;
    pzlib_filefunc_def->zseek_file     = fseek_file_func;
    pzlib_filefunc_def->zclose_file    = fclose_file_func;
    pzlib_filefunc_def->zerror_file    = ferror_file_func;
    pzlib_filefunc_def->opaque         = NULL;
}

//------------------------------------------------------------------------------

void fill_fopen64_filefunc(zlib_filefunc64_def* pzlib_filefunc_def)
{
    pzlib_filefunc_def->zopen64_file     = fopen64_file_func;
    pzlib_filefunc_def->zopendisk64_file = fopendisk64_file_func;
    pzlib_filefunc_def->zread_file       = fread_file_func;
    pzlib_filefunc_def->zwrite_file      = fwrite_file_func;
    pzlib_filefunc_def->ztell64_file     = ftell64_file_func;
    pzlib_filefunc_def->zseek64_file     = fseek64_file_func;
    pzlib_filefunc_def->zclose_file      = fclose_file_func;
    pzlib_filefunc_def->zerror_file      = ferror_file_func;
    pzlib_filefunc_def->opaque           = NULL;
}
