int unpack_write_callback(glc_thread_state_t *state)
{
unpack_t unpack = (unpack_t) state->ptr;
if (state->header.type == GLC_MESSAGE_LZO) {
#ifdef __LZO
__sync_fetch_and_add(&unpack->stats.pack_size, state->read_size - sizeof(glc_lzo_header_t));
memcpy(&state->header, &((glc_lzo_header_t *) state->read_data)->header,
sizeof(glc_message_header_t));
__lzo_decompress((unsigned char *) &state->read_data[sizeof(glc_lzo_header_t)],
state->read_size - sizeof(glc_lzo_header_t),
(unsigned char *) state->write_data,
(lzo_uintp) &state->write_size,
NULL);
#else
return ENOTSUP;
#endif
} else if (state->header.type == GLC_MESSAGE_QUICKLZ) {
#ifdef __QUICKLZ
__sync_fetch_and_add(&unpack->stats.pack_size,
state->read_size - sizeof(glc_quicklz_header_t));
memcpy(&state->header, &((glc_quicklz_header_t *) state->read_data)->header,
sizeof(glc_message_header_t));
if (!state->threadptr)
state->threadptr = malloc(sizeof(qlz_state_decompress));
qlz_decompress((const void *) &state->read_data[sizeof(glc_quicklz_header_t)],
(void *) state->write_data,
(qlz_state_decompress *) state->threadptr);
#else
return ENOTSUP;
#endif
} else if (state->header.type == GLC_MESSAGE_LZJB) {
#ifdef __LZJB
__sync_fetch_and_add(&unpack->stats.pack_size, state->read_size - sizeof(glc_lzjb_header_t));
memcpy(&state->header, &((glc_quicklz_header_t *) state->read_data)->header,
sizeof(glc_message_header_t));
lzjb_decompress(&state->read_data[sizeof(glc_lzjb_header_t)],
state->write_data,
state->read_size - sizeof(glc_lzjb_header_t),
state->write_size);
#else
return ENOTSUP;
#endif
} else
return ENOTSUP;
__sync_fetch_and_add(&unpack->stats.unpack_size, state->write_size);
return 0;
}
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;
}
