int pack_lzjb_write_callback(glc_thread_state_t *state)
{
#ifdef __LZJB
glc_container_message_header_t *container = (glc_container_message_header_t *) state->write_data;
glc_lzjb_header_t *lzjb_header =
(glc_lzjb_header_t *) &state->write_data[sizeof(glc_container_message_header_t)];
size_t compressed_size = lzjb_compress(state->read_data,
&state->write_data[sizeof(glc_lzjb_header_t) +
sizeof(glc_container_message_header_t)],
state->read_size);
lzjb_header->size = (glc_size_t) state->read_size;
memcpy(&lzjb_header->header, &state->header, sizeof(glc_message_header_t));
container->size = compressed_size + sizeof(glc_lzjb_header_t);
container->header.type = GLC_MESSAGE_LZJB;
state->header.type = GLC_MESSAGE_CONTAINER;
__sync_fetch_and_add(&((pack_t) state->ptr)->stats.pack_size,
compressed_size);
return 0;
#else
return ENOTSUP;
#endif
}
int main(int argc, char **argv) {
int compress = 1, firstarg=1;
char *infile, *outfile;
int in_fd, out_fd;
void *inbuf, *outbuf;
off_t insize, outsize;
if (argc>1) {
if (strcmp("-d", argv[1])==0) {
compress = 0;
firstarg++;
} else if (strcmp("-z", argv[1])==0) {
compress = 1;
firstarg++;
}
}
if ((argc-firstarg) < 2) {
fprintf(stderr, "Usage: %s -d|-z <infile> <outfile>\n", argv[0]);
return 1;
}
infile = argv[firstarg];
outfile = argv[firstarg+1];
// snarf in input
in_fd = open(infile, O_RDONLY);
if (in_fd < 0) {
fprintf(stderr, "Can't open %s for reading.\n", infile);
return 1;
}
{
struct stat s;
int st;
st = fstat(in_fd, &s);
if (st!=0 || s.st_size==0) {
fprintf(stderr, "Can't read file size of %s.\n", infile);
return 1;
}
insize = s.st_size;
}
inbuf = mmap(NULL, insize, PROT_READ, MAP_SHARED, in_fd, 0);
if (inbuf == MAP_FAILED) {
fprintf(stderr, "Can't mmap %s\n", infile);
return 1;
}
if (compress) {
uchar_t *dst;
off_t outsize_guess;
out_fd = open(outfile, O_RDWR|O_CREAT, 0666);
if (out_fd < 0) {
fprintf(stderr, "Can't open %s for writing.\n", outfile);
return 1;
}
// compressed file can't be bigger than 9/8 * infile + (size bytes)
outsize_guess = insize + (insize/8) + 17 /*2*NBBY+1*/ + 10 /*size*/;
ftruncate(out_fd, outsize_guess);
outbuf = mmap(NULL, outsize_guess, PROT_READ|PROT_WRITE, MAP_SHARED,
out_fd, 0);
if (outbuf == MAP_FAILED) {
fprintf(stderr, "Can't mmap %s\n", outfile);
return 1;
}
// write the input file size
dst = writesize(insize+1, 1, (uchar_t*)outbuf);
// now write the compressed file
outsize = lzjb_compress(inbuf, dst, insize,
(uchar_t*)outbuf+outsize_guess-dst, 0);
// truncate output
munmap(outbuf, outsize_guess);
ftruncate(out_fd, (dst + outsize) - (uchar_t*)outbuf);
close(out_fd);
} else { // decompress
uchar_t *src;
// read the output file size
outsize = 0;
src = inbuf;
for (src=inbuf; ; src++) {
if ((*src) & 0x80) {
outsize |= (*src) & 0x7F;
break;
} else {
outsize = (outsize | (*src)) << 7;
}
}
outsize--; src++;
if (outsize < 0) {
fprintf(stderr, "Streaming decompression not implemented.\n");
return 1;
}
// create an output file of the appropriate length
out_fd = open(outfile, O_RDWR|O_CREAT, 0666);
if (out_fd < 0) {
fprintf(stderr, "Can't open %s for writing.\n", outfile);
return 1;
}
ftruncate(out_fd, outsize);
outbuf = mmap(NULL, outsize, PROT_READ|PROT_WRITE, MAP_SHARED,
out_fd, 0);
if (outbuf == MAP_FAILED) {
fprintf(stderr, "Can't mmap %s\n", outfile);
return 1;
}
// now decompress!
lzjb_decompress(src, outbuf, (uchar_t*)inbuf+insize-src, outsize, 0);
// unmap, close
munmap(outbuf, outsize);
close(out_fd);
}
munmap(inbuf, insize);
close(in_fd);
return 0;
}
