/* zip.c -- IO on .zip files using zlib

   Version 1.01e, February 12th, 2005

   27 Dec 2004 Rolf Kalbermatter

   Modification to zipOpen2 to support globalComment retrieval.

   Copyright (C) 1998-2005 Gilles Vollant

   Read zip.h for more info

*/

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <time.h>

#include "zlib.h"

#include "zip.h"

#ifdef STDC

#  include <stddef.h>

#  include <string.h>

#  include <stdlib.h>

#endif

#ifdef NO_ERRNO_H

    extern int errno;

#else

#   include <errno.h>

#endif

#ifndef local

#  define local static

#endif

/* compile with -Dlocal if your debugger can't find static symbols */

#ifndef VERSIONMADEBY

# define VERSIONMADEBY   (0x0) /* platform depedent */

#endif

#ifndef Z_BUFSIZE

#define Z_BUFSIZE (16384)

#endif

#ifndef Z_MAXFILENAMEINZIP

#define Z_MAXFILENAMEINZIP (256)

#endif

#ifndef ALLOC

# define ALLOC(size) (malloc(size))

#endif

#ifndef TRYFREE

# define TRYFREE(p) {if (p) free(p);}

#endif

/*

#define SIZECENTRALDIRITEM (0x2e)

#define SIZEZIPLOCALHEADER (0x1e)

*/

/* I've found an old Unix (a SunOS 4.1.3_U1) without all SEEK_* defined.... */

#ifndef SEEK_CUR

#define SEEK_CUR    1

#endif

#ifndef SEEK_END

#define SEEK_END    2

#endif

#ifndef SEEK_SET

#define SEEK_SET    0

#endif

#ifndef DEF_MEM_LEVEL

#if MAX_MEM_LEVEL >= 8

#  define DEF_MEM_LEVEL 8

#else

#  define DEF_MEM_LEVEL  MAX_MEM_LEVEL

#endif

#endif

const char zip_copyright[] =

   " zip 1.01 Copyright 1998-2004 Gilles Vollant - http://www.winimage.com/zLibDll";

#define SIZEDATA_INDATABLOCK (4096-(4*4))

#define LOCALHEADERMAGIC    (0x04034b50)

#define CENTRALHEADERMAGIC  (0x02014b50)

#define ENDHEADERMAGIC      (0x06054b50)

#define FLAG_LOCALHEADER_OFFSET (0x06)

#define CRC_LOCALHEADER_OFFSET  (0x0e)

#define SIZECENTRALHEADER (0x2e) /* 46 */

typedef struct linkedlist_datablock_internal_s

{

  struct linkedlist_datablock_internal_s* next_datablock;

  uLong  avail_in_this_block;

  uLong  filled_in_this_block;

  uLong  unused; /* for future use and alignement */

  unsigned char data[SIZEDATA_INDATABLOCK];

} linkedlist_datablock_internal;

typedef struct linkedlist_data_s

{

    linkedlist_datablock_internal* first_block;

    linkedlist_datablock_internal* last_block;

} linkedlist_data;

typedef struct

{

    z_stream stream;            /* zLib stream structure for inflate */

    int  stream_initialised;    /* 1 is stream is initialised */

    uInt pos_in_buffered_data;  /* last written byte in buffered_data */

    uLong pos_local_header;     /* offset of the local header of the file

                                     currenty writing */

    char* central_header;       /* central header data for the current file */

    uLong size_centralheader;   /* size of the central header for cur file */

    uLong flag;                 /* flag of the file currently writing */

    int  method;                /* compression method of file currenty wr.*/

    int  raw;                   /* 1 for directly writing raw data */

    Byte buffered_data[Z_BUFSIZE];/* buffer contain compressed data to be writ*/

    uLong dosDate;

    uLong crc32;

    int  encrypt;

#ifndef NOCRYPT

    unsigned long keys[3];     /* keys defining the pseudo-random sequence */

    const unsigned long* pcrc_32_tab;

    int crypt_header_size;

#endif

} curfile_info;

typedef struct

{

    zlib_filefunc_def z_filefunc;

    voidpf filestream;        /* io structore of the zipfile */

    linkedlist_data central_dir;/* datablock with central dir in construction*/

    int  in_opened_file_inzip;  /* 1 if a file in the zip is currently writ.*/

    curfile_info ci;            /* info on the file curretly writing */

    uLong begin_pos;            /* position of the beginning of the zipfile */

    uLong add_position_when_writting_offset;

    uLong number_entry;

#ifndef NO_ADDFILEINEXISTINGZIP

    char *globalcomment;

#endif

} zip_internal;

#ifndef NOCRYPT

#define INCLUDECRYPTINGCODE_IFCRYPTALLOWED

#include "crypt.h"

#endif

local linkedlist_datablock_internal* allocate_new_datablock()

{

    linkedlist_datablock_internal* ldi;

    ldi = (linkedlist_datablock_internal*)

                 ALLOC(sizeof(linkedlist_datablock_internal));

    if (ldi!=NULL)

    {

        ldi->next_datablock = NULL ;

        ldi->filled_in_this_block = 0 ;

        ldi->avail_in_this_block = SIZEDATA_INDATABLOCK ;

    }

    return ldi;

}

local void free_datablock(ldi)

    linkedlist_datablock_internal* ldi;

{

    while (ldi!=NULL)

    {

        linkedlist_datablock_internal* ldinext = ldi->next_datablock;

        TRYFREE(ldi);

        ldi = ldinext;

    }

}

local void init_linkedlist(ll)

    linkedlist_data* ll;

{

    ll->first_block = ll->last_block = NULL;

}

local void free_linkedlist(ll)

    linkedlist_data* ll;

{

    free_datablock(ll->first_block);

    ll->first_block = ll->last_block = NULL;

}

local int add_data_in_datablock(ll,buf,len)

    linkedlist_data* ll;

    const void* buf;

    uLong len;

{

    linkedlist_datablock_internal* ldi;

    const unsigned char* from_copy;

    if (ll==NULL)

        return ZIP_INTERNALERROR;

    if (ll->last_block == NULL)

    {

        ll->first_block = ll->last_block = allocate_new_datablock();

        if (ll->first_block == NULL)

            return ZIP_INTERNALERROR;

    }

    ldi = ll->last_block;

    from_copy = (unsigned char*)buf;

    while (len>0)

    {

        uInt copy_this;

        uInt i;

        unsigned char* to_copy;

        if (ldi->avail_in_this_block==0)

        {

            ldi->next_datablock = allocate_new_datablock();

            if (ldi->next_datablock == NULL)

                return ZIP_INTERNALERROR;

            ldi = ldi->next_datablock ;

            ll->last_block = ldi;

        }

        if (ldi->avail_in_this_block < len)

            copy_this = (uInt)ldi->avail_in_this_block;

        else

            copy_this = (uInt)len;

        to_copy = &(ldi->data[ldi->filled_in_this_block]);

        for (i=0;i<copy_this;i++)

            *(to_copy+i)=*(from_copy+i);

        ldi->filled_in_this_block += copy_this;

        ldi->avail_in_this_block -= copy_this;

        from_copy += copy_this ;

        len -= copy_this;

    }

    return ZIP_OK;

}

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

#ifndef NO_ADDFILEINEXISTINGZIP

/* ===========================================================================

   Inputs a long in LSB order to the given file

   nbByte == 1, 2 or 4 (byte, short or long)

*/

local int ziplocal_putValue OF((const zlib_filefunc_def* pzlib_filefunc_def,

                                voidpf filestream, uLong x, int nbByte));

local int ziplocal_putValue (pzlib_filefunc_def, filestream, x, nbByte)

    const zlib_filefunc_def* pzlib_filefunc_def;

    voidpf filestream;

    uLong x;

    int nbByte;

{

    unsigned char buf[4];

    int n;

    for (n = 0; n < nbByte; n++)

    {

        buf[n] = (unsigned char)(x & 0xff);

        x >>= 8;

    }

    if (x != 0)

      {     /* data overflow - hack for ZIP64 (X Roche) */

      for (n = 0; n < nbByte; n++)

        {

          buf[n] = 0xff;

        }

      }

    if (ZWRITE(*pzlib_filefunc_def,filestream,buf,nbByte)!=(uLong)nbByte)

        return ZIP_ERRNO;

    else

        return ZIP_OK;

}

local void ziplocal_putValue_inmemory OF((void* dest, uLong x, int nbByte));

local void ziplocal_putValue_inmemory (dest, x, nbByte)

    void* dest;

    uLong x;

    int nbByte;

{

    unsigned char* buf=(unsigned char*)dest;

    int n;

    for (n = 0; n < nbByte; n++) {

        buf[n] = (unsigned char)(x & 0xff);

        x >>= 8;

    }

    if (x != 0)

    {     /* data overflow - hack for ZIP64 */

       for (n = 0; n < nbByte; n++)

       {

          buf[n] = 0xff;

       }

    }

}

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

local uLong ziplocal_TmzDateToDosDate(ptm,dosDate)

    const tm_zip* ptm;

    uLong dosDate;

{

    uLong year = (uLong)ptm->tm_year;

    if (year>1980)

        year-=1980;

    else if (year>80)

        year-=80;

    return

      (uLong) (((ptm->tm_mday) + (32 * (ptm->tm_mon+1)) + (512 * year)) << 16) |

        ((ptm->tm_sec/2) + (32* ptm->tm_min) + (2048 * (uLong)ptm->tm_hour));

}

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

local int ziplocal_getByte OF((

    const zlib_filefunc_def* pzlib_filefunc_def,

    voidpf filestream,

    int *pi));

local int ziplocal_getByte(pzlib_filefunc_def,filestream,pi)

    const zlib_filefunc_def* pzlib_filefunc_def;

    voidpf filestream;

    int *pi;

{

    unsigned char c;

    int err = (int)ZREAD(*pzlib_filefunc_def,filestream,&c,1);

    if (err==1)

    {

        *pi = (int)c;

        return ZIP_OK;

    }

    else

    {

        if (ZERROR(*pzlib_filefunc_def,filestream))

            return ZIP_ERRNO;

        else

            return ZIP_EOF;

    }

}

/* ===========================================================================

   Reads a long in LSB order from the given gz_stream. Sets

*/

local int ziplocal_getShort OF((

    const zlib_filefunc_def* pzlib_filefunc_def,

    voidpf filestream,

    uLong *pX));

local int ziplocal_getShort (pzlib_filefunc_def,filestream,pX)

    const zlib_filefunc_def* pzlib_filefunc_def;

    voidpf filestream;

    uLong *pX;

{

    uLong x ;

    int i;

    int err;

    err = ziplocal_getByte(pzlib_filefunc_def,filestream,&i);

    x = (uLong)i;

    if (err==ZIP_OK)

        err = ziplocal_getByte(pzlib_filefunc_def,filestream,&i);

    x += ((uLong)i)<<8;

    if (err==ZIP_OK)

        *pX = x;

    else

        *pX = 0;

    return err;

}

local int ziplocal_getLong OF((

    const zlib_filefunc_def* pzlib_filefunc_def,

    voidpf filestream,

    uLong *pX));

local int ziplocal_getLong (pzlib_filefunc_def,filestream,pX)

    const zlib_filefunc_def* pzlib_filefunc_def;

    voidpf filestream;

    uLong *pX;

{

    uLong x ;

    int i;

    int err;

    err = ziplocal_getByte(pzlib_filefunc_def,filestream,&i);

    x = (uLong)i;

    if (err==ZIP_OK)

        err = ziplocal_getByte(pzlib_filefunc_def,filestream,&i);

    x += ((uLong)i)<<8;

    if (err==ZIP_OK)

        err = ziplocal_getByte(pzlib_filefunc_def,filestream,&i);

    x += ((uLong)i)<<16;

    if (err==ZIP_OK)

        err = ziplocal_getByte(pzlib_filefunc_def,filestream,&i);

    x += ((uLong)i)<<24;

    if (err==ZIP_OK)

        *pX = x;

    else

        *pX = 0;

    return err;

}

#ifndef BUFREADCOMMENT

#define BUFREADCOMMENT (0x400)

#endif

/*

  Locate the Central directory of a zipfile (at the end, just before

    the global comment)

*/

local uLong ziplocal_SearchCentralDir OF((

    const zlib_filefunc_def* pzlib_filefunc_def,

    voidpf filestream));

local uLong ziplocal_SearchCentralDir(pzlib_filefunc_def,filestream)

    const zlib_filefunc_def* pzlib_filefunc_def;

    voidpf filestream;

{

    unsigned char* buf;

    uLong uSizeFile;

    uLong uBackRead;

    uLong uMaxBack=0xffff; /* maximum size of global comment */

    uLong uPosFound=0;

    if (ZSEEK(*pzlib_filefunc_def,filestream,0,ZLIB_FILEFUNC_SEEK_END) != 0)

        return 0;

    uSizeFile = ZTELL(*pzlib_filefunc_def,filestream);

    if (uMaxBack>uSizeFile)

        uMaxBack = uSizeFile;

    buf = (unsigned char*)ALLOC(BUFREADCOMMENT+4);

    if (buf==NULL)

        return 0;

    uBackRead = 4;

    while (uBackRead<uMaxBack)

    {

        uLong uReadSize,uReadPos ;

        int i;

        if (uBackRead+BUFREADCOMMENT>uMaxBack)

            uBackRead = uMaxBack;

        else

            uBackRead+=BUFREADCOMMENT;

        uReadPos = uSizeFile-uBackRead ;

        uReadSize = ((BUFREADCOMMENT+4) < (uSizeFile-uReadPos)) ?

                     (BUFREADCOMMENT+4) : (uSizeFile-uReadPos);

        if (ZSEEK(*pzlib_filefunc_def,filestream,uReadPos,ZLIB_FILEFUNC_SEEK_SET)!=0)

            break;

        if (ZREAD(*pzlib_filefunc_def,filestream,buf,uReadSize)!=uReadSize)

            break;

        for (i=(int)uReadSize-3; (i--)>0;)

            if (((*(buf+i))==0x50) && ((*(buf+i+1))==0x4b) &&

                ((*(buf+i+2))==0x05) && ((*(buf+i+3))==0x06))

            {

                uPosFound = uReadPos+i;

                break;

            }

        if (uPosFound!=0)

            break;

    }

    TRYFREE(buf);

    return uPosFound;

}

#endif /* !NO_ADDFILEINEXISTINGZIP*/

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

extern zipFile ZEXPORT zipOpen2 (pathname, append, globalcomment, pzlib_filefunc_def)

    const char *pathname;

    int append;

    zipcharpc* globalcomment;

    zlib_filefunc_def* pzlib_filefunc_def;

{

    zip_internal ziinit;

    zip_internal* zi;

    int err=ZIP_OK;

    if (pzlib_filefunc_def==NULL)

        fill_fopen_filefunc(&ziinit.z_filefunc);

    else

        ziinit.z_filefunc = *pzlib_filefunc_def;

    ziinit.filestream = (*(ziinit.z_filefunc.zopen_file))

                 (ziinit.z_filefunc.opaque,

                  pathname,

                  (append == APPEND_STATUS_CREATE) ?

                  (ZLIB_FILEFUNC_MODE_READ | ZLIB_FILEFUNC_MODE_WRITE | ZLIB_FILEFUNC_MODE_CREATE) :

                    (ZLIB_FILEFUNC_MODE_READ | ZLIB_FILEFUNC_MODE_WRITE | ZLIB_FILEFUNC_MODE_EXISTING));

    if (ziinit.filestream == NULL)

        return NULL;

    ziinit.begin_pos = ZTELL(ziinit.z_filefunc,ziinit.filestream);

    ziinit.in_opened_file_inzip = 0;

    ziinit.ci.stream_initialised = 0;

    ziinit.number_entry = 0;

    ziinit.add_position_when_writting_offset = 0;