125 lines
3.1 KiB
C
125 lines
3.1 KiB
C
/*
|
|
* compression handling for cbfstool
|
|
*
|
|
* Copyright (C) 2009 coresystems GmbH
|
|
* written by Patrick Georgi <patrick.georgi@coresystems.de>
|
|
*
|
|
* Adapted from code
|
|
* Copyright (C) 2008 Jordan Crouse <jordan@cosmicpenguin.net>, released
|
|
* under identical license terms
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; version 2 of the License.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*/
|
|
|
|
#include <string.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include "common.h"
|
|
#include "lz4/lib/lz4frame.h"
|
|
#include <commonlib/bsd/compression.h>
|
|
|
|
static int lz4_compress(char *in, int in_len, char *out, int *out_len)
|
|
{
|
|
LZ4F_preferences_t prefs = {
|
|
.compressionLevel = 20,
|
|
.frameInfo = {
|
|
.blockSizeID = max4MB,
|
|
.blockMode = blockIndependent,
|
|
.contentChecksumFlag = noContentChecksum,
|
|
},
|
|
};
|
|
size_t worst_size = LZ4F_compressFrameBound(in_len, &prefs);
|
|
void *bounce = malloc(worst_size);
|
|
if (!bounce)
|
|
return -1;
|
|
*out_len = LZ4F_compressFrame(bounce, worst_size, in, in_len, &prefs);
|
|
if (LZ4F_isError(*out_len) || *out_len >= in_len)
|
|
return -1;
|
|
memcpy(out, bounce, *out_len);
|
|
return 0;
|
|
}
|
|
|
|
static int lz4_decompress(char *in, int in_len, char *out, int out_len,
|
|
size_t *actual_size)
|
|
{
|
|
size_t result = ulz4fn(in, in_len, out, out_len);
|
|
if (result == 0)
|
|
return -1;
|
|
if (actual_size != NULL)
|
|
*actual_size = result;
|
|
return 0;
|
|
}
|
|
|
|
static int lzma_compress(char *in, int in_len, char *out, int *out_len)
|
|
{
|
|
return do_lzma_compress(in, in_len, out, out_len);
|
|
}
|
|
|
|
static int lzma_decompress(char *in, int in_len, char *out, unused int out_len,
|
|
size_t *actual_size)
|
|
{
|
|
return do_lzma_uncompress(out, out_len, in, in_len, actual_size);
|
|
}
|
|
static int none_compress(char *in, int in_len, char *out, int *out_len)
|
|
{
|
|
memcpy(out, in, in_len);
|
|
*out_len = in_len;
|
|
return 0;
|
|
}
|
|
|
|
static int none_decompress(char *in, int in_len, char *out, unused int out_len,
|
|
size_t *actual_size)
|
|
{
|
|
memcpy(out, in, in_len);
|
|
if (actual_size != NULL)
|
|
*actual_size = in_len;
|
|
return 0;
|
|
}
|
|
|
|
comp_func_ptr compression_function(enum comp_algo algo)
|
|
{
|
|
comp_func_ptr compress;
|
|
switch (algo) {
|
|
case CBFS_COMPRESS_NONE:
|
|
compress = none_compress;
|
|
break;
|
|
case CBFS_COMPRESS_LZMA:
|
|
compress = lzma_compress;
|
|
break;
|
|
case CBFS_COMPRESS_LZ4:
|
|
compress = lz4_compress;
|
|
break;
|
|
default:
|
|
ERROR("Unknown compression algorithm %d!\n", algo);
|
|
return NULL;
|
|
}
|
|
return compress;
|
|
}
|
|
|
|
decomp_func_ptr decompression_function(enum comp_algo algo)
|
|
{
|
|
decomp_func_ptr decompress;
|
|
switch (algo) {
|
|
case CBFS_COMPRESS_NONE:
|
|
decompress = none_decompress;
|
|
break;
|
|
case CBFS_COMPRESS_LZMA:
|
|
decompress = lzma_decompress;
|
|
break;
|
|
case CBFS_COMPRESS_LZ4:
|
|
decompress = lz4_decompress;
|
|
break;
|
|
default:
|
|
ERROR("Unknown compression algorithm %d!\n", algo);
|
|
return NULL;
|
|
}
|
|
return decompress;
|
|
}
|