/* ----------------------------------------------------------------------- *
 *
 *   Copyright 2003-2008 H. Peter Anvin - All Rights Reserved
 *
 *   Permission is hereby granted, free of charge, to any person
 *   obtaining a copy of this software and associated documentation
 *   files (the "Software"), to deal in the Software without
 *   restriction, including without limitation the rights to use,
 *   copy, modify, merge, publish, distribute, sublicense, and/or
 *   sell copies of the Software, and to permit persons to whom
 *   the Software is furnished to do so, subject to the following
 *   conditions:
 *
 *   The above copyright notice and this permission notice shall
 *   be included in all copies or substantial portions of the Software.
 *
 *   THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 *   EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
 *   OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 *   NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
 *   HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
 *   WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 *   FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 *   OTHER DEALINGS IN THE SOFTWARE.
 *
 * ----------------------------------------------------------------------- */

#include <errno.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <stdlib.h>
#include <syslinux/zio.h>

#include "file.h"
#include "zlib.h"

/*
 * zopen.c
 *
 * Open an ordinary file, possibly compressed; if so, insert
 * an appropriate decompressor.
 */

int __file_get_block(struct file_info *fp);
int __file_close(struct file_info *fp);

static ssize_t gzip_file_read(struct file_info *, void *, size_t);
static int gzip_file_close(struct file_info *);

static const struct input_dev gzip_file_dev = {
    .dev_magic = __DEV_MAGIC,
    .flags = __DEV_FILE | __DEV_INPUT,
    .fileflags = O_RDONLY,
    .read = gzip_file_read,
    .close = gzip_file_close,
    .open = NULL,
};

static int gzip_file_init(struct file_info *fp)
{
    z_streamp zs = calloc(1, sizeof(z_stream));

    if (!zs)
	return -1;

    fp->i.pvt = zs;

    zs->next_in = (void *)fp->i.datap;
    zs->avail_in = fp->i.nbytes;

    if (inflateInit2(zs, 15 + 32) != Z_OK) {
	errno = EIO;
	return -1;
    }

    fp->iop = &gzip_file_dev;
    fp->i.fd.size = -1;		/* Unknown */

    return 0;
}

static ssize_t gzip_file_read(struct file_info *fp, void *ptr, size_t n)
{
    z_streamp zs = fp->i.pvt;
    int rv;
    ssize_t bytes;
    ssize_t nout = 0;
    unsigned char *p = ptr;

    while (n) {
	zs->next_out = p;
	zs->avail_out = n;

	if (!zs->avail_in && fp->i.fd.handle) {
	    if (__file_get_block(fp))
		return nout ? nout : -1;

	    zs->next_in = (void *)fp->i.datap;
	    zs->avail_in = fp->i.nbytes;
	}

	rv = inflate(zs, Z_SYNC_FLUSH);

	bytes = n - zs->avail_out;
	nout += bytes;
	p += bytes;
	n -= bytes;

	switch (rv) {
	case Z_DATA_ERROR:
	case Z_NEED_DICT:
	case Z_BUF_ERROR:
	case Z_STREAM_ERROR:
	default:
	    errno = EIO;
	    return nout ? nout : -1;
	case Z_MEM_ERROR:
	    errno = ENOMEM;
	    return nout ? nout : -1;
	case Z_STREAM_END:
	    return nout;
	case Z_OK:
	    break;
	}
    }

    return nout;
}

static int gzip_file_close(struct file_info *fp)
{
    z_streamp zs = fp->i.pvt;

    inflateEnd(zs);
    free(zs);
    return __file_close(fp);
}

int zopen(const char *pathname, int flags, ...)
{
    int fd, rv;
    struct file_info *fp;

    /* We don't actually give a hoot about the creation bits... */
    fd = open(pathname, flags, 0);

    if (fd < 0)
	return -1;

    fp = &__file_info[fd];

    /* Need to get the first block into the buffer, but not consumed */
    if (__file_get_block(fp))
	goto err;

    if (fp->i.nbytes >= 14 &&
	(uint8_t) fp->i.buf[0] == 037 &&
	(uint8_t) fp->i.buf[1] == 0213 &&	/* gzip */
	fp->i.buf[2] == 8)	/* deflate */
	rv = gzip_file_init(fp);
    else
	rv = 0;			/* Plain file */

    if (!rv)
	return fd;

err:
    close(fd);
    return -1;
}