/*
   * Wrapper for decompressing LZ4-compressed kernel, initramfs, and initrd
   *
   * Copyright (C) 2013, LG Electronics, Kyungsik Lee <kyungsik.lee@lge.com>
   *
   * This program is free software; you can redistribute it and/or modify
   * it under the terms of the GNU General Public License version 2 as
   * published by the Free Software Foundation.
   */
  
  #ifdef STATIC
  #define PREBOOT
  #include "lz4/lz4_decompress.c"
  #else
  #include <linux/decompress/unlz4.h>
  #endif
  #include <linux/types.h>
  #include <linux/lz4.h>
  #include <linux/decompress/mm.h>
  #include <linux/compiler.h>
  
  #include <asm/unaligned.h>
  
  /*
   * Note: Uncompressed chunk size is used in the compressor side
   * (userspace side for compression).
   * It is hardcoded because there is not proper way to extract it
   * from the binary stream which is generated by the preliminary
   * version of LZ4 tool so far.
   */
  #define LZ4_DEFAULT_UNCOMPRESSED_CHUNK_SIZE (8 << 20)
  #define ARCHIVE_MAGICNUMBER 0x184C2102

  STATIC inline int INIT unlz4(u8 *input, long in_len,
  				long (*fill)(void *, unsigned long),
  				long (*flush)(void *, unsigned long),
  				u8 *output, long *posp,

  				void (*error) (char *x))
  {
  	int ret = -1;
  	size_t chunksize = 0;
  	size_t uncomp_chunksize = LZ4_DEFAULT_UNCOMPRESSED_CHUNK_SIZE;
  	u8 *inp;
  	u8 *inp_start;
  	u8 *outp;

  	long size = in_len;

  #ifdef PREBOOT
  	size_t out_len = get_unaligned_le32(input + in_len);
  #endif
  	size_t dest_len;
  
  
  	if (output) {
  		outp = output;
  	} else if (!flush) {
  		error("NULL output pointer and no flush function provided");
  		goto exit_0;
  	} else {
  		outp = large_malloc(uncomp_chunksize);
  		if (!outp) {
  			error("Could not allocate output buffer");
  			goto exit_0;
  		}
  	}
  
  	if (input && fill) {
  		error("Both input pointer and fill function provided,");
  		goto exit_1;
  	} else if (input) {
  		inp = input;
  	} else if (!fill) {
  		error("NULL input pointer and missing fill function");
  		goto exit_1;
  	} else {

  		inp = large_malloc(LZ4_compressBound(uncomp_chunksize));

  		if (!inp) {
  			error("Could not allocate input buffer");
  			goto exit_1;
  		}
  	}
  	inp_start = inp;
  
  	if (posp)
  		*posp = 0;

  	if (fill) {
  		size = fill(inp, 4);
  		if (size < 4) {
  			error("data corrupted");
  			goto exit_2;
  		}
  	}

  
  	chunksize = get_unaligned_le32(inp);
  	if (chunksize == ARCHIVE_MAGICNUMBER) {

  		if (!fill) {
  			inp += 4;
  			size -= 4;
  		}

  	} else {
  		error("invalid header");
  		goto exit_2;
  	}
  
  	if (posp)
  		*posp += 4;
  
  	for (;;) {

  		if (fill) {
  			size = fill(inp, 4);
  			if (size == 0)
  				break;
  			if (size < 4) {
  				error("data corrupted");
  				goto exit_2;
  			}
  		}

  
  		chunksize = get_unaligned_le32(inp);
  		if (chunksize == ARCHIVE_MAGICNUMBER) {

  			if (!fill) {
  				inp += 4;
  				size -= 4;
  			}

  			if (posp)
  				*posp += 4;
  			continue;
  		}

  
  		if (posp)
  			*posp += 4;

  		if (!fill) {
  			inp += 4;
  			size -= 4;
  		} else {

  			if (chunksize > LZ4_compressBound(uncomp_chunksize)) {

  				error("chunk length is longer than allocated");
  				goto exit_2;
  			}

  			size = fill(inp, chunksize);
  			if (size < chunksize) {
  				error("data corrupted");
  				goto exit_2;
  			}

  		}
  #ifdef PREBOOT
  		if (out_len >= uncomp_chunksize) {
  			dest_len = uncomp_chunksize;
  			out_len -= dest_len;
  		} else
  			dest_len = out_len;

  
  		ret = LZ4_decompress_fast(inp, outp, dest_len);
  		chunksize = ret;

  #else
  		dest_len = uncomp_chunksize;

  
  		ret = LZ4_decompress_safe(inp, outp, chunksize, dest_len);
  		dest_len = ret;

  #endif
  		if (ret < 0) {
  			error("Decoding failed");
  			goto exit_2;
  		}

  		ret = -1;

  		if (flush && flush(outp, dest_len) != dest_len)
  			goto exit_2;
  		if (output)
  			outp += dest_len;
  		if (posp)
  			*posp += chunksize;

  		if (!fill) {
  			size -= chunksize;

  			if (size == 0)
  				break;
  			else if (size < 0) {
  				error("data corrupted");
  				goto exit_2;
  			}
  			inp += chunksize;

  		}

  	}
  
  	ret = 0;
  exit_2:
  	if (!input)
  		large_free(inp_start);
  exit_1:
  	if (!output)
  		large_free(outp);
  exit_0:
  	return ret;
  }
  
  #ifdef PREBOOT

  STATIC int INIT __decompress(unsigned char *buf, long in_len,

  			      long (*fill)(void*, unsigned long),
  			      long (*flush)(void*, unsigned long),

  			      unsigned char *output, long out_len,

  			      long *posp,

  			      void (*error)(char *x)

  	)
  {
  	return unlz4(buf, in_len - 4, fill, flush, output, posp, error);
  }
  #endif