/*
 * LZ4 - Fast LZ compression algorithm
 * Copyright (C) 2011 - 2016, Yann Collet.
 * BSD 2 - Clause License (http://www.opensource.org/licenses/bsd - license.php)
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met:
 *      * Redistributions of source code must retain the above copyright
 *        notice, this list of conditions and the following disclaimer.
 *      * Redistributions in binary form must reproduce the above
 * copyright notice, this list of conditions and the following disclaimer
 * in the documentation and/or other materials provided with the
 * distribution.
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 * You can contact the author at :
 *      - LZ4 homepage : http://www.lz4.org
 *      - LZ4 source repository : https://github.com/lz4/lz4
 *
 *      Changed for kernel usage by:
 *      Sven Schmidt <4sschmid@informatik.uni-hamburg.de>
 */

/*-************************************
 *      Dependencies
 **************************************/
#include <linux/lz4.h>
#include "lz4defs.h"
#include <linux/module.h>
#include <linux/kernel.h>
#include <asm/unaligned.h>

static const int LZ4_minLength = (MFLIMIT + 1);
static const int LZ4_64Klimit = ((64 * KB) + (MFLIMIT - 1));

/*-******************************
 *      Compression functions
 ********************************/
static FORCE_INLINE U32 LZ4_hash4(
        U32 sequence,
        tableType_t const tableType)
{
        if (tableType == byU16)
                return ((sequence * 2654435761U)
                        >> ((MINMATCH * 8) - (LZ4_HASHLOG + 1)));
        else
                return ((sequence * 2654435761U)
                        >> ((MINMATCH * 8) - LZ4_HASHLOG));
}

static FORCE_INLINE U32 LZ4_hash5(
        U64 sequence,
        tableType_t const tableType)
{
        const U32 hashLog = (tableType == byU16)
                ? LZ4_HASHLOG + 1
                : LZ4_HASHLOG;

#if LZ4_LITTLE_ENDIAN
        static const U64 prime5bytes = 889523592379ULL;

        return (U32)(((sequence << 24) * prime5bytes) >> (64 - hashLog));
#else
        static const U64 prime8bytes = 11400714785074694791ULL;

        return (U32)(((sequence >> 24) * prime8bytes) >> (64 - hashLog));
#endif
}

static FORCE_INLINE U32 LZ4_hashPosition(
        const void *p,
        tableType_t const tableType)
{
#if LZ4_ARCH64
        if (tableType == byU32)
                return LZ4_hash5(LZ4_read_ARCH(p), tableType);
#endif

        return LZ4_hash4(LZ4_read32(p), tableType);
}

static void LZ4_putPositionOnHash(
        const BYTE *p,
        U32 h,
        void *tableBase,
        tableType_t const tableType,
        const BYTE *srcBase)
{
        switch (tableType) {
        case byPtr:
        {
                const BYTE **hashTable = (const BYTE **)tableBase;

                hashTable[h] = p;
                return;
        }
        case byU32:
        {
                U32 *hashTable = (U32 *) tableBase;

                hashTable[h] = (U32)(p - srcBase);
                return;
        }
        case byU16:
        {
                U16 *hashTable = (U16 *) tableBase;

                hashTable[h] = (U16)(p - srcBase);
                return;
        }
        }
}

static FORCE_INLINE void LZ4_putPosition(
        const BYTE *p,
        void *tableBase,
        tableType_t tableType,
        const BYTE *srcBase)
{
        U32 const h = LZ4_hashPosition(p, tableType);

        LZ4_putPositionOnHash(p, h, tableBase, tableType, srcBase);
}

static const BYTE *LZ4_getPositionOnHash(
        U32 h,
        void *tableBase,
        tableType_t tableType,
        const BYTE *srcBase)
{
        if (tableType == byPtr) {
                const BYTE **hashTable = (const BYTE **) tableBase;

                return hashTable[h];
        }

        if (tableType == byU32) {
                const U32 * const hashTable = (U32 *) tableBase;

                return hashTable[h] + srcBase;
        }

        {
                /* default, to ensure a return */
                const U16 * const hashTable = (U16 *) tableBase;

                return hashTable[h] + srcBase;
        }
}

static FORCE_INLINE const BYTE *LZ4_getPosition(
        const BYTE *p,
        void *tableBase,
        tableType_t tableType,
        const BYTE *srcBase)
{
        U32 const h = LZ4_hashPosition(p, tableType);

        return LZ4_getPositionOnHash(h, tableBase, tableType, srcBase);
}


/*
 * LZ4_compress_generic() :
 * inlined, to ensure branches are decided at compilation time
 */
static FORCE_INLINE int LZ4_compress_generic(
        LZ4_stream_t_internal * const dictPtr,
        const char * const source,
        char * const dest,
        const int inputSize,
        const int maxOutputSize,
        const limitedOutput_directive outputLimited,
        const tableType_t tableType,
        const dict_directive dict,
        const dictIssue_directive dictIssue,
        const U32 acceleration)
{
        const BYTE *ip = (const BYTE *) source;
        const BYTE *base;
        const BYTE *lowLimit;
        const BYTE * const lowRefLimit = ip - dictPtr->dictSize;
        const BYTE * const dictionary = dictPtr->dictionary;
        const BYTE * const dictEnd = dictionary + dictPtr->dictSize;
        const size_t dictDelta = dictEnd - (const BYTE *)source;
        const BYTE *anchor = (const BYTE *) source;
        const BYTE * const iend = ip + inputSize;
        const BYTE * const mflimit = iend - MFLIMIT;
        const BYTE * const matchlimit = iend - LASTLITERALS;

        BYTE *op = (BYTE *) dest;
        BYTE * const olimit = op + maxOutputSize;

        U32 forwardH;
        size_t refDelta = 0;

        /* Init conditions */
        if ((U32)inputSize > (U32)LZ4_MAX_INPUT_SIZE) {
                /* Unsupported inputSize, too large (or negative) */
                return 0;
        }

        switch (dict) {
        case noDict:
        default:
                base = (const BYTE *)source;
                lowLimit = (const BYTE *)source;
                break;
        case withPrefix64k:
                base = (const BYTE *)source - dictPtr->currentOffset;
                lowLimit = (const BYTE *)source - dictPtr->dictSize;
                break;
        case usingExtDict:
                base = (const BYTE *)source - dictPtr->currentOffset;
                lowLimit = (const BYTE *)source;
                break;
        }

        if ((tableType == byU16)
                && (inputSize >= LZ4_64Klimit)) {
                /* Size too large (not within 64K limit) */
                return 0;
        }

        if (inputSize < LZ4_minLength) {
                /* Input too small, no compression (all literals) */
                goto _last_literals;
        }

        /* First Byte */
        LZ4_putPosition(ip, dictPtr->hashTable, tableType, base);
        ip++;
        forwardH = LZ4_hashPosition(ip, tableType);

        /* Main Loop */
        for ( ; ; ) {
                const BYTE *match;
                BYTE *token;

                /* Find a match */
                {
                        const BYTE *forwardIp = ip;
                        unsigned int step = 1;
                        unsigned int searchMatchNb = acceleration << LZ4_SKIPTRIGGER;

                        do {
                                U32 const h = forwardH;

                                ip = forwardIp;
                                forwardIp += step;
                                step = (searchMatchNb++ >> LZ4_SKIPTRIGGER);

                                if (unlikely(forwardIp > mflimit))
                                        goto _last_literals;

                                match = LZ4_getPositionOnHash(h,
                                        dictPtr->hashTable,
                                        tableType, base);

                                if (dict == usingExtDict) {
                                        if (match < (const BYTE *)source) {
                                                refDelta = dictDelta;
                                                lowLimit = dictionary;
                                        } else {
                                                refDelta = 0;
                                                lowLimit = (const BYTE *)source;
                                }        }

                                forwardH = LZ4_hashPosition(forwardIp,
                                        tableType);

                                LZ4_putPositionOnHash(ip, h, dictPtr->hashTable,
                                        tableType, base);
                        } while (((dictIssue == dictSmall)
                                        ? (match < lowRefLimit)
                                        : 0)
                                || ((tableType == byU16)
                                        ? 0
                                        : (match + MAX_DISTANCE < ip))
                                || (LZ4_read32(match + refDelta)
                                        != LZ4_read32(ip)));
                }

                /* Catch up */
                while (((ip > anchor) & (match + refDelta > lowLimit))
                                && (unlikely(ip[-1] == match[refDelta - 1]))) {
                        ip--;
                        match--;
                }

                /* Encode Literals */
                {
                        unsigned const int litLength = (unsigned int)(ip - anchor);

                        token = op++;

                        if ((outputLimited) &&
                                /* Check output buffer overflow */
                                (unlikely(op + litLength +
                                        (2 + 1 + LASTLITERALS) +
                                        (litLength / 255) > olimit)))
                                return 0;

                        if (litLength >= RUN_MASK) {
                                int len = (int)litLength - RUN_MASK;

                                *token = (RUN_MASK << ML_BITS);

                                for (; len >= 255; len -= 255)
                                        *op++ = 255;
                                *op++ = (BYTE)len;
                        } else
                                *token = (BYTE)(litLength << ML_BITS);

                        /* Copy Literals */
                        LZ4_wildCopy(op, anchor, op + litLength);
                        op += litLength;
                }

_next_match:
                /* Encode Offset */
                LZ4_writeLE16(op, (U16)(ip - match));
                op += 2;

                /* Encode MatchLength */
                {
                        unsigned int matchCode;

                        if ((dict == usingExtDict)
                                && (lowLimit == dictionary)) {
                                const BYTE *limit;

                                match += refDelta;
                                limit = ip + (dictEnd - match);

                                if (limit > matchlimit)
                                        limit = matchlimit;

                                matchCode = LZ4_count(ip + MINMATCH,
                                        match + MINMATCH, limit);

                                ip += MINMATCH + matchCode;

                                if (ip == limit) {
                                        unsigned const int more = LZ4_count(ip,
                                                (const BYTE *)source,
                                                matchlimit);

                                        matchCode += more;
                                        ip += more;
                                }
                        } else {
                                matchCode = LZ4_count(ip + MINMATCH,
                                        match + MINMATCH, matchlimit);
                                ip += MINMATCH + matchCode;
                        }

                        if (outputLimited &&
                                /* Check output buffer overflow */
                                (unlikely(op +
                                        (1 + LASTLITERALS) +
                                        (matchCode >> 8) > olimit)))
                                return 0;

                        if (matchCode >= ML_MASK) {
                                *token += ML_MASK;
                                matchCode -= ML_MASK;
                                LZ4_write32(op, 0xFFFFFFFF);

                                while (matchCode >= 4 * 255) {
                                        op += 4;
                                        LZ4_write32(op, 0xFFFFFFFF);
                                        matchCode -= 4 * 255;
                                }

                                op += matchCode / 255;
                                *op++ = (BYTE)(matchCode % 255);
                        } else
                                *token += (BYTE)(matchCode);
                }

                anchor = ip;

                /* Test end of chunk */
                if (ip > mflimit)
                        break;

                /* Fill table */
                LZ4_putPosition(ip - 2, dictPtr->hashTable, tableType, base);

                /* Test next position */
                match = LZ4_getPosition(ip, dictPtr->hashTable,
                        tableType, base);

                if (dict == usingExtDict) {
                        if (match < (const BYTE *)source) {
                                refDelta = dictDelta;
                                lowLimit = dictionary;
                        } else {
                                refDelta = 0;
                                lowLimit = (const BYTE *)source;
                        }
                }

                LZ4_putPosition(ip, dictPtr->hashTable, tableType, base);

                if (((dictIssue == dictSmall) ? (match >= lowRefLimit) : 1)
                        && (match + MAX_DISTANCE >= ip)
                        && (LZ4_read32(match + refDelta) == LZ4_read32(ip))) {
                        token = op++;
                        *token = 0;
                        goto _next_match;
                }

                /* Prepare next loop */
                forwardH = LZ4_hashPosition(++ip, tableType);
        }

_last_literals:
        /* Encode Last Literals */
        {
                size_t const lastRun = (size_t)(iend - anchor);

                if ((outputLimited) &&
                        /* Check output buffer overflow */
                        ((op - (BYTE *)dest) + lastRun + 1 +
                        ((lastRun + 255 - RUN_MASK) / 255) > (U32)maxOutputSize))
                        return 0;

                if (lastRun >= RUN_MASK) {
                        size_t accumulator = lastRun - RUN_MASK;
                        *op++ = RUN_MASK << ML_BITS;
                        for (; accumulator >= 255; accumulator -= 255)
                                *op++ = 255;
                        *op++ = (BYTE) accumulator;
                } else {
                        *op++ = (BYTE)(lastRun << ML_BITS);
                }

                LZ4_memcpy(op, anchor, lastRun);

                op += lastRun;
        }

        /* End */
        return (int) (((char *)op) - dest);
}

static int LZ4_compress_fast_extState(
        void *state,
        const char *source,
        char *dest,
        int inputSize,
        int maxOutputSize,
        int acceleration)
{
        LZ4_stream_t_internal *ctx = &((LZ4_stream_t *)state)->internal_donotuse;
#if LZ4_ARCH64
        const tableType_t tableType = byU32;
#else
        const tableType_t tableType = byPtr;
#endif

        LZ4_resetStream((LZ4_stream_t *)state);

        if (acceleration < 1)
                acceleration = LZ4_ACCELERATION_DEFAULT;

        if (maxOutputSize >= LZ4_COMPRESSBOUND(inputSize)) {
                if (inputSize < LZ4_64Klimit)
                        return LZ4_compress_generic(ctx, source,
                                dest, inputSize, 0,
                                noLimit, byU16, noDict,
                                noDictIssue, acceleration);
                else
                        return LZ4_compress_generic(ctx, source,
                                dest, inputSize, 0,
                                noLimit, tableType, noDict,
                                noDictIssue, acceleration);
        } else {
                if (inputSize < LZ4_64Klimit)
                        return LZ4_compress_generic(ctx, source,
                                dest, inputSize,
                                maxOutputSize, limitedOutput, byU16, noDict,
                                noDictIssue, acceleration);
                else
                        return LZ4_compress_generic(ctx, source,
                                dest, inputSize,
                                maxOutputSize, limitedOutput, tableType, noDict,
                                noDictIssue, acceleration);
        }
}

int LZ4_compress_fast(const char *source, char *dest, int inputSize,
        int maxOutputSize, int acceleration, void *wrkmem)
{
        return LZ4_compress_fast_extState(wrkmem, source, dest, inputSize,
                maxOutputSize, acceleration);
}
EXPORT_SYMBOL(LZ4_compress_fast);

int LZ4_compress_default(const char *source, char *dest, int inputSize,
        int maxOutputSize, void *wrkmem)
{
        return LZ4_compress_fast(source, dest, inputSize,
                maxOutputSize, LZ4_ACCELERATION_DEFAULT, wrkmem);
}
EXPORT_SYMBOL(LZ4_compress_default);

/*-******************************
 *      *_destSize() variant
 ********************************/
static int LZ4_compress_destSize_generic(
        LZ4_stream_t_internal * const ctx,
        const char * const src,
        char * const dst,
        int * const srcSizePtr,
        const int targetDstSize,
        const tableType_t tableType)
{
        const BYTE *ip = (const BYTE *) src;
        const BYTE *base = (const BYTE *) src;
        const BYTE *lowLimit = (const BYTE *) src;
        const BYTE *anchor = ip;
        const BYTE * const iend = ip + *srcSizePtr;
        const BYTE * const mflimit = iend - MFLIMIT;
        const BYTE * const matchlimit = iend - LASTLITERALS;

        BYTE *op = (BYTE *) dst;
        BYTE * const oend = op + targetDstSize;
        BYTE * const oMaxLit = op + targetDstSize - 2 /* offset */
                - 8 /* because 8 + MINMATCH == MFLIMIT */ - 1 /* token */;
        BYTE * const oMaxMatch = op + targetDstSize
                - (LASTLITERALS + 1 /* token */);
        BYTE * const oMaxSeq = oMaxLit - 1 /* token */;

        U32 forwardH;

        /* Init conditions */
        /* Impossible to store anything */
        if (targetDstSize < 1)
                return 0;
        /* Unsupported input size, too large (or negative) */
        if ((U32)*srcSizePtr > (U32)LZ4_MAX_INPUT_SIZE)
                return 0;
        /* Size too large (not within 64K limit) */
        if ((tableType == byU16) && (*srcSizePtr >= LZ4_64Klimit))
                return 0;
        /* Input too small, no compression (all literals) */
        if (*srcSizePtr < LZ4_minLength)
                goto _last_literals;

        /* First Byte */
        *srcSizePtr = 0;
        LZ4_putPosition(ip, ctx->hashTable, tableType, base);
        ip++; forwardH = LZ4_hashPosition(ip, tableType);

        /* Main Loop */
        for ( ; ; ) {
                const BYTE *match;
                BYTE *token;

                /* Find a match */
                {
                        const BYTE *forwardIp = ip;
                        unsigned int step = 1;
                        unsigned int searchMatchNb = 1 << LZ4_SKIPTRIGGER;

                        do {
                                U32 h = forwardH;

                                ip = forwardIp;
                                forwardIp += step;
                                step = (searchMatchNb++ >> LZ4_SKIPTRIGGER);

                                if (unlikely(forwardIp > mflimit))
                                        goto _last_literals;

                                match = LZ4_getPositionOnHash(h, ctx->hashTable,
                                        tableType, base);
                                forwardH = LZ4_hashPosition(forwardIp,
                                        tableType);
                                LZ4_putPositionOnHash(ip, h,
                                        ctx->hashTable, tableType,
                                        base);

                        } while (((tableType == byU16)
                                ? 0
                                : (match + MAX_DISTANCE < ip))
                                || (LZ4_read32(match) != LZ4_read32(ip)));
                }

                /* Catch up */
                while ((ip > anchor)
                        && (match > lowLimit)
                        && (unlikely(ip[-1] == match[-1]))) {
                        ip--;
                        match--;
                }

                /* Encode Literal length */
                {
                        unsigned int litLength = (unsigned int)(ip - anchor);

                        token = op++;
                        if (op + ((litLength + 240) / 255)
                                + litLength > oMaxLit) {
                                /* Not enough space for a last match */
                                op--;
                                goto _last_literals;
                        }
                        if (litLength >= RUN_MASK) {
                                unsigned int len = litLength - RUN_MASK;
                                *token = (RUN_MASK<<ML_BITS);
                                for (; len >= 255; len -= 255)
                                        *op++ = 255;
                                *op++ = (BYTE)len;
                        } else
                                *token = (BYTE)(litLength << ML_BITS);

                        /* Copy Literals */
                        LZ4_wildCopy(op, anchor, op + litLength);
                        op += litLength;
                }

_next_match:
                /* Encode Offset */
                LZ4_writeLE16(op, (U16)(ip - match)); op += 2;

                /* Encode MatchLength */
                {
                        size_t matchLength = LZ4_count(ip + MINMATCH,
                        match + MINMATCH, matchlimit);

                        if (op + ((matchLength + 240)/255) > oMaxMatch) {
                                /* Match description too long : reduce it */
                                matchLength = (15 - 1) + (oMaxMatch - op) * 255;
                        }
                        ip += MINMATCH + matchLength;

                        if (matchLength >= ML_MASK) {
                                *token += ML_MASK;
                                matchLength -= ML_MASK;
                                while (matchLength >= 255) {
                                        matchLength -= 255;
                                        *op++ = 255;
                                }
                                *op++ = (BYTE)matchLength;
                        } else
                                *token += (BYTE)(matchLength);
                }

                anchor = ip;

                /* Test end of block */
                if (ip > mflimit)
                        break;
                if (op > oMaxSeq)
                        break;

                /* Fill table */
                LZ4_putPosition(ip - 2, ctx->hashTable, tableType, base);

                /* Test next position */
                match = LZ4_getPosition(ip, ctx->hashTable, tableType, base);
                LZ4_putPosition(ip, ctx->hashTable, tableType, base);

                if ((match + MAX_DISTANCE >= ip)
                        && (LZ4_read32(match) == LZ4_read32(ip))) {
                        token = op++; *token = 0;
                        goto _next_match;
                }

                /* Prepare next loop */
                forwardH = LZ4_hashPosition(++ip, tableType);
        }

_last_literals:
        /* Encode Last Literals */
        {
                size_t lastRunSize = (size_t)(iend - anchor);

                if (op + 1 /* token */
                        + ((lastRunSize + 240) / 255) /* litLength */
                        + lastRunSize /* literals */ > oend) {
                        /* adapt lastRunSize to fill 'dst' */
                        lastRunSize     = (oend - op) - 1;
                        lastRunSize -= (lastRunSize + 240) / 255;
                }
                ip = anchor + lastRunSize;

                if (lastRunSize >= RUN_MASK) {
                        size_t accumulator = lastRunSize - RUN_MASK;

                        *op++ = RUN_MASK << ML_BITS;
                        for (; accumulator >= 255; accumulator -= 255)
                                *op++ = 255;
                        *op++ = (BYTE) accumulator;
                } else {
                        *op++ = (BYTE)(lastRunSize<<ML_BITS);
                }
                LZ4_memcpy(op, anchor, lastRunSize);
                op += lastRunSize;
        }

        /* End */
        *srcSizePtr = (int) (((const char *)ip) - src);
        return (int) (((char *)op) - dst);
}

static int LZ4_compress_destSize_extState(
        LZ4_stream_t *state,
        const char *src,
        char *dst,
        int *srcSizePtr,
        int targetDstSize)
{
#if LZ4_ARCH64
        const tableType_t tableType = byU32;
#else
        const tableType_t tableType = byPtr;
#endif

        LZ4_resetStream(state);

        if (targetDstSize >= LZ4_COMPRESSBOUND(*srcSizePtr)) {
                /* compression success is guaranteed */
                return LZ4_compress_fast_extState(
                        state, src, dst, *srcSizePtr,
                        targetDstSize, 1);
        } else {
                if (*srcSizePtr < LZ4_64Klimit)
                        return LZ4_compress_destSize_generic(
                                &state->internal_donotuse,
                                src, dst, srcSizePtr,
                                targetDstSize, byU16);
                else
                        return LZ4_compress_destSize_generic(
                                &state->internal_donotuse,
                                src, dst, srcSizePtr,
                                targetDstSize, tableType);
        }
}


int LZ4_compress_destSize(
        const char *src,
        char *dst,
        int *srcSizePtr,
        int targetDstSize,
        void *wrkmem)
{
        return LZ4_compress_destSize_extState(wrkmem, src, dst, srcSizePtr,
                targetDstSize);
}
EXPORT_SYMBOL(LZ4_compress_destSize);

/*-******************************
 *      Streaming functions
 ********************************/
void LZ4_resetStream(LZ4_stream_t *LZ4_stream)
{
        memset(LZ4_stream, 0, sizeof(LZ4_stream_t));
}

int LZ4_loadDict(LZ4_stream_t *LZ4_dict,
        const char *dictionary, int dictSize)
{
        LZ4_stream_t_internal *dict = &LZ4_dict->internal_donotuse;
        const BYTE *p = (const BYTE *)dictionary;
        const BYTE * const dictEnd = p + dictSize;
        const BYTE *base;

        if ((dict->initCheck)
                || (dict->currentOffset > 1 * GB)) {
                /* Uninitialized structure, or reuse overflow */
                LZ4_resetStream(LZ4_dict);
        }

        if (dictSize < (int)HASH_UNIT) {
                dict->dictionary = NULL;
                dict->dictSize = 0;
                return 0;
        }

        if ((dictEnd - p) > 64 * KB)
                p = dictEnd - 64 * KB;
        dict->currentOffset += 64 * KB;
        base = p - dict->currentOffset;
        dict->dictionary = p;
        dict->dictSize = (U32)(dictEnd - p);
        dict->currentOffset += dict->dictSize;

        while (p <= dictEnd - HASH_UNIT) {
                LZ4_putPosition(p, dict->hashTable, byU32, base);
                p += 3;
        }

        return dict->dictSize;
}
EXPORT_SYMBOL(LZ4_loadDict);

static void LZ4_renormDictT(LZ4_stream_t_internal *LZ4_dict,
        const BYTE *src)
{
        if ((LZ4_dict->currentOffset > 0x80000000) ||
                ((uptrval)LZ4_dict->currentOffset > (uptrval)src)) {
                /* address space overflow */
                /* rescale hash table */
                U32 const delta = LZ4_dict->currentOffset - 64 * KB;
                const BYTE *dictEnd = LZ4_dict->dictionary + LZ4_dict->dictSize;
                int i;

                for (i = 0; i < LZ4_HASH_SIZE_U32; i++) {
                        if (LZ4_dict->hashTable[i] < delta)
                                LZ4_dict->hashTable[i] = 0;
                        else
                                LZ4_dict->hashTable[i] -= delta;
                }
                LZ4_dict->currentOffset = 64 * KB;
                if (LZ4_dict->dictSize > 64 * KB)
                        LZ4_dict->dictSize = 64 * KB;
                LZ4_dict->dictionary = dictEnd - LZ4_dict->dictSize;
        }
}

int LZ4_saveDict(LZ4_stream_t *LZ4_dict, char *safeBuffer, int dictSize)
{
        LZ4_stream_t_internal * const dict = &LZ4_dict->internal_donotuse;
        const BYTE * const previousDictEnd = dict->dictionary + dict->dictSize;

        if ((U32)dictSize > 64 * KB) {
                /* useless to define a dictionary > 64 * KB */
                dictSize = 64 * KB;
        }
        if ((U32)dictSize > dict->dictSize)
                dictSize = dict->dictSize;

        memmove(safeBuffer, previousDictEnd - dictSize, dictSize);

        dict->dictionary = (const BYTE *)safeBuffer;
        dict->dictSize = (U32)dictSize;

        return dictSize;
}
EXPORT_SYMBOL(LZ4_saveDict);

int LZ4_compress_fast_continue(LZ4_stream_t *LZ4_stream, const char *source,
        char *dest, int inputSize, int maxOutputSize, int acceleration)
{
        LZ4_stream_t_internal *streamPtr = &LZ4_stream->internal_donotuse;
        const BYTE * const dictEnd = streamPtr->dictionary
                + streamPtr->dictSize;

        const BYTE *smallest = (const BYTE *) source;

        if (streamPtr->initCheck) {
                /* Uninitialized structure detected */
                return 0;
        }

        if ((streamPtr->dictSize > 0) && (smallest > dictEnd))
                smallest = dictEnd;

        LZ4_renormDictT(streamPtr, smallest);

        if (acceleration < 1)
                acceleration = LZ4_ACCELERATION_DEFAULT;

        /* Check overlapping input/dictionary space */
        {
                const BYTE *sourceEnd = (const BYTE *) source + inputSize;

                if ((sourceEnd > streamPtr->dictionary)
                        && (sourceEnd < dictEnd)) {
                        streamPtr->dictSize = (U32)(dictEnd - sourceEnd);
                        if (streamPtr->dictSize > 64 * KB)
                                streamPtr->dictSize = 64 * KB;
                        if (streamPtr->dictSize < 4)
                                streamPtr->dictSize = 0;
                        streamPtr->dictionary = dictEnd - streamPtr->dictSize;
                }
        }

        /* prefix mode : source data follows dictionary */
        if (dictEnd == (const BYTE *)source) {
                int result;

                if ((streamPtr->dictSize < 64 * KB) &&
                        (streamPtr->dictSize < streamPtr->currentOffset)) {
                        result = LZ4_compress_generic(
                                streamPtr, source, dest, inputSize,
                                maxOutputSize, limitedOutput, byU32,
                                withPrefix64k, dictSmall, acceleration);
                } else {
                        result = LZ4_compress_generic(
                                streamPtr, source, dest, inputSize,
                                maxOutputSize, limitedOutput, byU32,
                                withPrefix64k, noDictIssue, acceleration);
                }
                streamPtr->dictSize += (U32)inputSize;
                streamPtr->currentOffset += (U32)inputSize;
                return result;
        }

        /* external dictionary mode */
        {
                int result;

                if ((streamPtr->dictSize < 64 * KB) &&
                        (streamPtr->dictSize < streamPtr->currentOffset)) {
                        result = LZ4_compress_generic(
                                streamPtr, source, dest, inputSize,
                                maxOutputSize, limitedOutput, byU32,
                                usingExtDict, dictSmall, acceleration);
                } else {
                        result = LZ4_compress_generic(
                                streamPtr, source, dest, inputSize,
                                maxOutputSize, limitedOutput, byU32,
                                usingExtDict, noDictIssue, acceleration);
                }
                streamPtr->dictionary = (const BYTE *)source;
                streamPtr->dictSize = (U32)inputSize;
                streamPtr->currentOffset += (U32)inputSize;
                return result;
        }
}
EXPORT_SYMBOL(LZ4_compress_fast_continue);

MODULE_LICENSE("Dual BSD/GPL");
MODULE_DESCRIPTION("LZ4 compressor");