| /* |
| * 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); |
| } |
| |
| 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); |
| } |
| 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"); |