mirror of https://github.com/vasi/pixz
new, simple-but-ugly test writer
parent
6ebbf15e02
commit
62443a29fe
@ -1,6 +1,13 @@
|
|||||||
pixz
|
pixz
|
||||||
pixzlist
|
pixzlist
|
||||||
pixztar
|
pixztar
|
||||||
|
write
|
||||||
|
|
||||||
|
*.tar
|
||||||
|
*.txz
|
||||||
*.o
|
*.o
|
||||||
|
|
||||||
test.out
|
test.out
|
||||||
test.base
|
test.base
|
||||||
|
|
||||||
|
libs
|
||||||
|
@ -0,0 +1,198 @@
|
|||||||
|
#include <stdarg.h>
|
||||||
|
#include <stdbool.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
|
||||||
|
#include <archive.h>
|
||||||
|
#include <archive_entry.h>
|
||||||
|
#include <lzma.h>
|
||||||
|
|
||||||
|
|
||||||
|
#pragma mark DEFINES
|
||||||
|
|
||||||
|
#define CHUNKSIZE 4096
|
||||||
|
#define BLOCKSIZE (1024 * 1024)
|
||||||
|
|
||||||
|
#define CHECK LZMA_CHECK_CRC32
|
||||||
|
|
||||||
|
|
||||||
|
#pragma mark GLOBALS
|
||||||
|
|
||||||
|
FILE *gInFile = NULL, *gOutFile = NULL;
|
||||||
|
uint8_t gBlockBuf[BLOCKSIZE];
|
||||||
|
size_t gBlockSize = 0;
|
||||||
|
lzma_filter gFilters[LZMA_FILTERS_MAX + 1];
|
||||||
|
lzma_stream gStream = LZMA_STREAM_INIT;
|
||||||
|
lzma_index *gIndex = NULL;
|
||||||
|
|
||||||
|
#pragma mark FUNCTION DECLARATIONS
|
||||||
|
|
||||||
|
void die(const char *fmt, ...);
|
||||||
|
|
||||||
|
void stream_edge(lzma_vli backward_size);
|
||||||
|
void write_block(void);
|
||||||
|
void encode_index(void);
|
||||||
|
|
||||||
|
archive_read_callback tar_read;
|
||||||
|
archive_open_callback tar_ok;
|
||||||
|
archive_close_callback tar_ok;
|
||||||
|
|
||||||
|
|
||||||
|
#pragma mark FUNCTION DEFINITIONS
|
||||||
|
|
||||||
|
int main(int argc, char **argv) {
|
||||||
|
if (argc != 3)
|
||||||
|
die("Need two arguments");
|
||||||
|
if (!(gInFile = fopen(argv[1], "r")))
|
||||||
|
die("Can't open input file");
|
||||||
|
if (!(gOutFile = fopen(argv[2], "w")))
|
||||||
|
die("Can't open output file");
|
||||||
|
|
||||||
|
// xz options
|
||||||
|
lzma_options_lzma lzma_opts;
|
||||||
|
if (lzma_lzma_preset(&lzma_opts, LZMA_PRESET_DEFAULT))
|
||||||
|
die("Error setting lzma options");
|
||||||
|
gFilters[0] = (lzma_filter){ .id = LZMA_FILTER_LZMA2,
|
||||||
|
.options = &lzma_opts };
|
||||||
|
gFilters[1] = (lzma_filter){ .id = LZMA_VLI_UNKNOWN, .options = NULL };
|
||||||
|
|
||||||
|
// xz setup (index, header)
|
||||||
|
if (!(gIndex = lzma_index_init(NULL, NULL)))
|
||||||
|
die("Error creating index");
|
||||||
|
stream_edge(LZMA_VLI_UNKNOWN);
|
||||||
|
|
||||||
|
// read archive
|
||||||
|
struct archive *ar = archive_read_new();
|
||||||
|
archive_read_support_compression_none(ar);
|
||||||
|
archive_read_support_format_tar(ar);
|
||||||
|
archive_read_open(ar, NULL, tar_ok, tar_read, tar_ok);
|
||||||
|
struct archive_entry *entry;
|
||||||
|
while (true) {
|
||||||
|
int aerr = archive_read_next_header(ar, &entry);
|
||||||
|
if (aerr == ARCHIVE_EOF) {
|
||||||
|
// TODO
|
||||||
|
break;
|
||||||
|
} else if (aerr != ARCHIVE_OK && aerr != ARCHIVE_WARN) {
|
||||||
|
// Some charset translations warn spuriously
|
||||||
|
fprintf(stderr, "%s\n", archive_error_string(ar));
|
||||||
|
die("Error reading archive entry");
|
||||||
|
}
|
||||||
|
|
||||||
|
printf("%s\n", archive_entry_pathname(entry));
|
||||||
|
}
|
||||||
|
archive_read_finish(ar);
|
||||||
|
fclose(gInFile);
|
||||||
|
|
||||||
|
// xz cleanup (index, footer)
|
||||||
|
write_block(); // write last block, if necessary
|
||||||
|
encode_index();
|
||||||
|
stream_edge(lzma_index_size(gIndex));
|
||||||
|
lzma_index_end(gIndex, NULL);
|
||||||
|
lzma_end(&gStream);
|
||||||
|
fclose(gOutFile);
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
void die(const char *fmt, ...) {
|
||||||
|
va_list args;
|
||||||
|
va_start(args, fmt);
|
||||||
|
vfprintf(stderr, fmt, args);
|
||||||
|
fprintf(stderr, "\n");
|
||||||
|
fflush(stderr);
|
||||||
|
va_end(args);
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
ssize_t tar_read(struct archive *ar, void *ref, const void **bufp) {
|
||||||
|
if (gBlockSize == BLOCKSIZE)
|
||||||
|
write_block();
|
||||||
|
|
||||||
|
uint8_t *buf = gBlockBuf + gBlockSize;
|
||||||
|
size_t rd = fread(buf, 1, CHUNKSIZE, gInFile);
|
||||||
|
if (rd == 0 && feof(gInFile))
|
||||||
|
die("Error reading input file");
|
||||||
|
gBlockSize += rd;
|
||||||
|
*bufp = buf;
|
||||||
|
return rd;
|
||||||
|
}
|
||||||
|
|
||||||
|
int tar_ok(struct archive *ar, void *ref) {
|
||||||
|
return ARCHIVE_OK;
|
||||||
|
}
|
||||||
|
|
||||||
|
void stream_edge(lzma_vli backward_size) {
|
||||||
|
lzma_stream_flags flags = { .version = 0, .check = CHECK,
|
||||||
|
.backward_size = backward_size };
|
||||||
|
uint8_t buf[LZMA_STREAM_HEADER_SIZE];
|
||||||
|
|
||||||
|
lzma_ret (*encoder)(const lzma_stream_flags *flags, uint8_t *buf);
|
||||||
|
encoder = backward_size == LZMA_VLI_UNKNOWN
|
||||||
|
? &lzma_stream_header_encode
|
||||||
|
: &lzma_stream_footer_encode;
|
||||||
|
if ((*encoder)(&flags, buf) != LZMA_OK)
|
||||||
|
die("Error encoding stream edge");
|
||||||
|
|
||||||
|
if (fwrite(buf, LZMA_STREAM_HEADER_SIZE, 1, gOutFile) != 1)
|
||||||
|
die("Error writing stream edge");
|
||||||
|
}
|
||||||
|
|
||||||
|
void write_block(void) {
|
||||||
|
if (gBlockSize == 0)
|
||||||
|
return;
|
||||||
|
|
||||||
|
lzma_block block = { .version = 0, .check = CHECK, .filters = gFilters };
|
||||||
|
block.compressed_size = block.uncompressed_size = LZMA_VLI_UNKNOWN;
|
||||||
|
uint8_t obuf[CHUNKSIZE];
|
||||||
|
|
||||||
|
// header
|
||||||
|
if (lzma_block_header_size(&block) != LZMA_OK)
|
||||||
|
die("Error getting block header size");
|
||||||
|
if (block.header_size > CHUNKSIZE)
|
||||||
|
die("Block header too big"); // can this really happen!?
|
||||||
|
if (lzma_block_header_encode(&block, obuf) != LZMA_OK)
|
||||||
|
die("Error encoding block header");
|
||||||
|
if (fwrite(obuf, block.header_size, 1, gOutFile) != 1)
|
||||||
|
die("Error writing block header");
|
||||||
|
|
||||||
|
// encode
|
||||||
|
if (lzma_block_encoder(&gStream, &block) != LZMA_OK)
|
||||||
|
die("Error creating block encoder");
|
||||||
|
gStream.next_in = gBlockBuf;
|
||||||
|
gStream.avail_in = gBlockSize;
|
||||||
|
lzma_ret err = LZMA_OK;
|
||||||
|
while (err != LZMA_STREAM_END) {
|
||||||
|
gStream.next_out = obuf;
|
||||||
|
gStream.avail_out = CHUNKSIZE;
|
||||||
|
|
||||||
|
err = lzma_code(&gStream, gStream.avail_in ? LZMA_RUN : LZMA_FINISH);
|
||||||
|
if (err != LZMA_OK && err != LZMA_STREAM_END)
|
||||||
|
die("Error encoding block");
|
||||||
|
|
||||||
|
if (fwrite(obuf, CHUNKSIZE - gStream.avail_out, 1, gOutFile) != 1)
|
||||||
|
die("Error writing block data");
|
||||||
|
}
|
||||||
|
|
||||||
|
// index
|
||||||
|
if (lzma_index_append(gIndex, NULL, lzma_block_unpadded_size(&block),
|
||||||
|
block.uncompressed_size) != LZMA_OK)
|
||||||
|
die("Error adding to index");
|
||||||
|
|
||||||
|
gBlockSize = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
void encode_index(void) {
|
||||||
|
if (lzma_index_encoder(&gStream, gIndex) != LZMA_OK)
|
||||||
|
die("Error creating index encoder");
|
||||||
|
uint8_t obuf[CHUNKSIZE];
|
||||||
|
lzma_ret err = LZMA_OK;
|
||||||
|
while (err != LZMA_STREAM_END) {
|
||||||
|
gStream.next_out = obuf;
|
||||||
|
gStream.avail_out = CHUNKSIZE;
|
||||||
|
err = lzma_code(&gStream, LZMA_RUN);
|
||||||
|
if (err != LZMA_OK && err != LZMA_STREAM_END)
|
||||||
|
die("Error encoding index");
|
||||||
|
if (fwrite(obuf, CHUNKSIZE - gStream.avail_out, 1, gOutFile) != 1)
|
||||||
|
die("Error writing index data");
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue