260 lines
9.2 KiB
C
260 lines
9.2 KiB
C
/**
|
|
* Copyright (c) 2016-present, Yann Collet, Facebook, Inc.
|
|
* All rights reserved.
|
|
*
|
|
* This source code is licensed under the BSD-style license found in the
|
|
* LICENSE file in the root directory of this source tree. An additional grant
|
|
* of patent rights can be found in the PATENTS file in the same directory.
|
|
*/
|
|
|
|
#ifndef ZSTD_LEGACY_H
|
|
#define ZSTD_LEGACY_H
|
|
|
|
#if defined (__cplusplus)
|
|
extern "C" {
|
|
#endif
|
|
|
|
/* *************************************
|
|
* Includes
|
|
***************************************/
|
|
#include "mem.h" /* MEM_STATIC */
|
|
#include "error_private.h" /* ERROR */
|
|
#include "zstd.h" /* ZSTD_inBuffer, ZSTD_outBuffer */
|
|
#include "zstd_v01.h"
|
|
#include "zstd_v02.h"
|
|
#include "zstd_v03.h"
|
|
#include "zstd_v04.h"
|
|
#include "zstd_v05.h"
|
|
#include "zstd_v06.h"
|
|
#include "zstd_v07.h"
|
|
|
|
|
|
/** ZSTD_isLegacy() :
|
|
@return : > 0 if supported by legacy decoder. 0 otherwise.
|
|
return value is the version.
|
|
*/
|
|
MEM_STATIC unsigned ZSTD_isLegacy(const void* src, size_t srcSize)
|
|
{
|
|
U32 magicNumberLE;
|
|
if (srcSize<4) return 0;
|
|
magicNumberLE = MEM_readLE32(src);
|
|
switch(magicNumberLE)
|
|
{
|
|
case ZSTDv01_magicNumberLE:return 1;
|
|
case ZSTDv02_magicNumber : return 2;
|
|
case ZSTDv03_magicNumber : return 3;
|
|
case ZSTDv04_magicNumber : return 4;
|
|
case ZSTDv05_MAGICNUMBER : return 5;
|
|
case ZSTDv06_MAGICNUMBER : return 6;
|
|
case ZSTDv07_MAGICNUMBER : return 7;
|
|
default : return 0;
|
|
}
|
|
}
|
|
|
|
|
|
MEM_STATIC unsigned long long ZSTD_getDecompressedSize_legacy(const void* src, size_t srcSize)
|
|
{
|
|
U32 const version = ZSTD_isLegacy(src, srcSize);
|
|
if (version < 5) return 0; /* no decompressed size in frame header, or not a legacy format */
|
|
if (version==5) {
|
|
ZSTDv05_parameters fParams;
|
|
size_t const frResult = ZSTDv05_getFrameParams(&fParams, src, srcSize);
|
|
if (frResult != 0) return 0;
|
|
return fParams.srcSize;
|
|
}
|
|
if (version==6) {
|
|
ZSTDv06_frameParams fParams;
|
|
size_t const frResult = ZSTDv06_getFrameParams(&fParams, src, srcSize);
|
|
if (frResult != 0) return 0;
|
|
return fParams.frameContentSize;
|
|
}
|
|
if (version==7) {
|
|
ZSTDv07_frameParams fParams;
|
|
size_t const frResult = ZSTDv07_getFrameParams(&fParams, src, srcSize);
|
|
if (frResult != 0) return 0;
|
|
return fParams.frameContentSize;
|
|
}
|
|
return 0; /* should not be possible */
|
|
}
|
|
|
|
|
|
MEM_STATIC size_t ZSTD_decompressLegacy(
|
|
void* dst, size_t dstCapacity,
|
|
const void* src, size_t compressedSize,
|
|
const void* dict,size_t dictSize)
|
|
{
|
|
U32 const version = ZSTD_isLegacy(src, compressedSize);
|
|
switch(version)
|
|
{
|
|
case 1 :
|
|
return ZSTDv01_decompress(dst, dstCapacity, src, compressedSize);
|
|
case 2 :
|
|
return ZSTDv02_decompress(dst, dstCapacity, src, compressedSize);
|
|
case 3 :
|
|
return ZSTDv03_decompress(dst, dstCapacity, src, compressedSize);
|
|
case 4 :
|
|
return ZSTDv04_decompress(dst, dstCapacity, src, compressedSize);
|
|
case 5 :
|
|
{ size_t result;
|
|
ZSTDv05_DCtx* const zd = ZSTDv05_createDCtx();
|
|
if (zd==NULL) return ERROR(memory_allocation);
|
|
result = ZSTDv05_decompress_usingDict(zd, dst, dstCapacity, src, compressedSize, dict, dictSize);
|
|
ZSTDv05_freeDCtx(zd);
|
|
return result;
|
|
}
|
|
case 6 :
|
|
{ size_t result;
|
|
ZSTDv06_DCtx* const zd = ZSTDv06_createDCtx();
|
|
if (zd==NULL) return ERROR(memory_allocation);
|
|
result = ZSTDv06_decompress_usingDict(zd, dst, dstCapacity, src, compressedSize, dict, dictSize);
|
|
ZSTDv06_freeDCtx(zd);
|
|
return result;
|
|
}
|
|
case 7 :
|
|
{ size_t result;
|
|
ZSTDv07_DCtx* const zd = ZSTDv07_createDCtx();
|
|
if (zd==NULL) return ERROR(memory_allocation);
|
|
result = ZSTDv07_decompress_usingDict(zd, dst, dstCapacity, src, compressedSize, dict, dictSize);
|
|
ZSTDv07_freeDCtx(zd);
|
|
return result;
|
|
}
|
|
default :
|
|
return ERROR(prefix_unknown);
|
|
}
|
|
}
|
|
|
|
|
|
MEM_STATIC size_t ZSTD_freeLegacyStreamContext(void* legacyContext, U32 version)
|
|
{
|
|
switch(version)
|
|
{
|
|
default :
|
|
case 1 :
|
|
case 2 :
|
|
case 3 :
|
|
return ERROR(version_unsupported);
|
|
case 4 : return ZBUFFv04_freeDCtx((ZBUFFv04_DCtx*)legacyContext);
|
|
case 5 : return ZBUFFv05_freeDCtx((ZBUFFv05_DCtx*)legacyContext);
|
|
case 6 : return ZBUFFv06_freeDCtx((ZBUFFv06_DCtx*)legacyContext);
|
|
case 7 : return ZBUFFv07_freeDCtx((ZBUFFv07_DCtx*)legacyContext);
|
|
}
|
|
}
|
|
|
|
|
|
MEM_STATIC size_t ZSTD_initLegacyStream(void** legacyContext, U32 prevVersion, U32 newVersion,
|
|
const void* dict, size_t dictSize)
|
|
{
|
|
if (prevVersion != newVersion) ZSTD_freeLegacyStreamContext(*legacyContext, prevVersion);
|
|
switch(newVersion)
|
|
{
|
|
default :
|
|
case 1 :
|
|
case 2 :
|
|
case 3 :
|
|
return 0;
|
|
case 4 :
|
|
{
|
|
ZBUFFv04_DCtx* dctx = (prevVersion != newVersion) ? ZBUFFv04_createDCtx() : (ZBUFFv04_DCtx*)*legacyContext;
|
|
if (dctx==NULL) return ERROR(memory_allocation);
|
|
ZBUFFv04_decompressInit(dctx);
|
|
ZBUFFv04_decompressWithDictionary(dctx, dict, dictSize);
|
|
*legacyContext = dctx;
|
|
return 0;
|
|
}
|
|
case 5 :
|
|
{
|
|
ZBUFFv05_DCtx* dctx = (prevVersion != newVersion) ? ZBUFFv05_createDCtx() : (ZBUFFv05_DCtx*)*legacyContext;
|
|
if (dctx==NULL) return ERROR(memory_allocation);
|
|
ZBUFFv05_decompressInitDictionary(dctx, dict, dictSize);
|
|
*legacyContext = dctx;
|
|
return 0;
|
|
}
|
|
case 6 :
|
|
{
|
|
ZBUFFv06_DCtx* dctx = (prevVersion != newVersion) ? ZBUFFv06_createDCtx() : (ZBUFFv06_DCtx*)*legacyContext;
|
|
if (dctx==NULL) return ERROR(memory_allocation);
|
|
ZBUFFv06_decompressInitDictionary(dctx, dict, dictSize);
|
|
*legacyContext = dctx;
|
|
return 0;
|
|
}
|
|
case 7 :
|
|
{
|
|
ZBUFFv07_DCtx* dctx = (prevVersion != newVersion) ? ZBUFFv07_createDCtx() : (ZBUFFv07_DCtx*)*legacyContext;
|
|
if (dctx==NULL) return ERROR(memory_allocation);
|
|
ZBUFFv07_decompressInitDictionary(dctx, dict, dictSize);
|
|
*legacyContext = dctx;
|
|
return 0;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
|
|
MEM_STATIC size_t ZSTD_decompressLegacyStream(void* legacyContext, U32 version,
|
|
ZSTD_outBuffer* output, ZSTD_inBuffer* input)
|
|
{
|
|
switch(version)
|
|
{
|
|
default :
|
|
case 1 :
|
|
case 2 :
|
|
case 3 :
|
|
return ERROR(version_unsupported);
|
|
case 4 :
|
|
{
|
|
ZBUFFv04_DCtx* dctx = (ZBUFFv04_DCtx*) legacyContext;
|
|
const void* src = (const char*)input->src + input->pos;
|
|
size_t readSize = input->size - input->pos;
|
|
void* dst = (char*)output->dst + output->pos;
|
|
size_t decodedSize = output->size - output->pos;
|
|
size_t const hintSize = ZBUFFv04_decompressContinue(dctx, dst, &decodedSize, src, &readSize);
|
|
output->pos += decodedSize;
|
|
input->pos += readSize;
|
|
return hintSize;
|
|
}
|
|
case 5 :
|
|
{
|
|
ZBUFFv05_DCtx* dctx = (ZBUFFv05_DCtx*) legacyContext;
|
|
const void* src = (const char*)input->src + input->pos;
|
|
size_t readSize = input->size - input->pos;
|
|
void* dst = (char*)output->dst + output->pos;
|
|
size_t decodedSize = output->size - output->pos;
|
|
size_t const hintSize = ZBUFFv05_decompressContinue(dctx, dst, &decodedSize, src, &readSize);
|
|
output->pos += decodedSize;
|
|
input->pos += readSize;
|
|
return hintSize;
|
|
}
|
|
case 6 :
|
|
{
|
|
ZBUFFv06_DCtx* dctx = (ZBUFFv06_DCtx*) legacyContext;
|
|
const void* src = (const char*)input->src + input->pos;
|
|
size_t readSize = input->size - input->pos;
|
|
void* dst = (char*)output->dst + output->pos;
|
|
size_t decodedSize = output->size - output->pos;
|
|
size_t const hintSize = ZBUFFv06_decompressContinue(dctx, dst, &decodedSize, src, &readSize);
|
|
output->pos += decodedSize;
|
|
input->pos += readSize;
|
|
return hintSize;
|
|
}
|
|
case 7 :
|
|
{
|
|
ZBUFFv07_DCtx* dctx = (ZBUFFv07_DCtx*) legacyContext;
|
|
const void* src = (const char*)input->src + input->pos;
|
|
size_t readSize = input->size - input->pos;
|
|
void* dst = (char*)output->dst + output->pos;
|
|
size_t decodedSize = output->size - output->pos;
|
|
size_t const hintSize = ZBUFFv07_decompressContinue(dctx, dst, &decodedSize, src, &readSize);
|
|
output->pos += decodedSize;
|
|
input->pos += readSize;
|
|
return hintSize;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
#if defined (__cplusplus)
|
|
}
|
|
#endif
|
|
|
|
#endif /* ZSTD_LEGACY_H */
|