mirror of
https://git.tukaani.org/xz.git
synced 2024-04-04 12:36:23 +02:00
bab0f01ed9
Hopefully these help a bit when learning the basics of liblzma API. I plan to write detailed examples about both basic and advanced features with lots of comments, but these two examples are good have right now. The examples were written by Daniel Mealha Cabrita. Thanks.
127 lines
3 KiB
C
127 lines
3 KiB
C
/*
|
|
* xz_pipe_comp.c
|
|
* A simple example of pipe-only xz compressor implementation.
|
|
* version: 2010-07-12 - by Daniel Mealha Cabrita
|
|
* Not copyrighted -- provided to the public domain.
|
|
*
|
|
* Compiling:
|
|
* Link with liblzma. GCC example:
|
|
* $ gcc -llzma xz_pipe_comp.c -o xz_pipe_comp
|
|
*
|
|
* Usage example:
|
|
* $ cat some_file | ./xz_pipe_comp > some_file.xz
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <stdint.h>
|
|
#include <inttypes.h>
|
|
#include <stdbool.h>
|
|
#include <lzma.h>
|
|
|
|
|
|
/* COMPRESSION SETTINGS */
|
|
|
|
/* analogous to xz CLI options: -0 to -9 */
|
|
#define COMPRESSION_LEVEL 6
|
|
|
|
/* boolean setting, analogous to xz CLI option: -e */
|
|
#define COMPRESSION_EXTREME true
|
|
|
|
/* see: /usr/include/lzma/check.h LZMA_CHECK_* */
|
|
#define INTEGRITY_CHECK LZMA_CHECK_CRC64
|
|
|
|
|
|
/* read/write buffer sizes */
|
|
#define IN_BUF_MAX 4096
|
|
#define OUT_BUF_MAX 4096
|
|
|
|
/* error codes */
|
|
#define RET_OK 0
|
|
#define RET_ERROR_INIT 1
|
|
#define RET_ERROR_INPUT 2
|
|
#define RET_ERROR_OUTPUT 3
|
|
#define RET_ERROR_COMPRESSION 4
|
|
|
|
|
|
/* note: in_file and out_file must be open already */
|
|
int xz_compress (FILE *in_file, FILE *out_file)
|
|
{
|
|
uint32_t preset = COMPRESSION_LEVEL | (COMPRESSION_EXTREME ? LZMA_PRESET_EXTREME : 0);
|
|
lzma_check check = INTEGRITY_CHECK;
|
|
lzma_stream strm = LZMA_STREAM_INIT; /* alloc and init lzma_stream struct */
|
|
uint8_t in_buf [IN_BUF_MAX];
|
|
uint8_t out_buf [OUT_BUF_MAX];
|
|
size_t in_len; /* length of useful data in in_buf */
|
|
size_t out_len; /* length of useful data in out_buf */
|
|
bool in_finished = false;
|
|
bool out_finished = false;
|
|
lzma_action action;
|
|
lzma_ret ret_xz;
|
|
int ret;
|
|
|
|
ret = RET_OK;
|
|
|
|
/* initialize xz encoder */
|
|
ret_xz = lzma_easy_encoder (&strm, preset, check);
|
|
if (ret_xz != LZMA_OK) {
|
|
fprintf (stderr, "lzma_easy_encoder error: %d\n", (int) ret_xz);
|
|
return RET_ERROR_INIT;
|
|
}
|
|
|
|
while ((! in_finished) && (! out_finished)) {
|
|
/* read incoming data */
|
|
in_len = fread (in_buf, 1, IN_BUF_MAX, in_file);
|
|
|
|
if (feof (in_file)) {
|
|
in_finished = true;
|
|
}
|
|
if (ferror (in_file)) {
|
|
in_finished = true;
|
|
ret = RET_ERROR_INPUT;
|
|
}
|
|
|
|
strm.next_in = in_buf;
|
|
strm.avail_in = in_len;
|
|
|
|
/* if no more data from in_buf, flushes the
|
|
internal xz buffers and closes the xz data
|
|
with LZMA_FINISH */
|
|
action = in_finished ? LZMA_FINISH : LZMA_RUN;
|
|
|
|
/* loop until there's no pending compressed output */
|
|
do {
|
|
/* out_buf is clean at this point */
|
|
strm.next_out = out_buf;
|
|
strm.avail_out = OUT_BUF_MAX;
|
|
|
|
/* compress data */
|
|
ret_xz = lzma_code (&strm, action);
|
|
|
|
if ((ret_xz != LZMA_OK) && (ret_xz != LZMA_STREAM_END)) {
|
|
fprintf (stderr, "lzma_code error: %d\n", (int) ret_xz);
|
|
out_finished = true;
|
|
ret = RET_ERROR_COMPRESSION;
|
|
} else {
|
|
/* write compressed data */
|
|
out_len = OUT_BUF_MAX - strm.avail_out;
|
|
fwrite (out_buf, 1, out_len, out_file);
|
|
if (ferror (out_file)) {
|
|
out_finished = true;
|
|
ret = RET_ERROR_OUTPUT;
|
|
}
|
|
}
|
|
} while (strm.avail_out == 0);
|
|
}
|
|
|
|
lzma_end (&strm);
|
|
return ret;
|
|
}
|
|
|
|
int main ()
|
|
{
|
|
int ret;
|
|
|
|
ret = xz_compress (stdin, stdout);
|
|
return ret;
|
|
}
|
|
|