macutils/macunpack/de_lzh.c

309 lines
6.0 KiB
C

#include "macunpack.h"
#ifdef ZMA
#define DELZH
#endif /* ZMA */
#ifdef LZH
#define DELZH
#endif /* LZH */
#ifdef DELZH
#include "globals.h"
#include "../util/masks.h"
#include "../fileio/wrfile.h"
#include "bits_be.h"
/* This code is valid for bitsused upto 15. */
#define DICBIT 13 /* 12(-lh4-) or 13(-lh5-) */
#define UCHAR_MAX 255
#define THRESHOLD 3
static int decoded;
static int bitsused;
static unsigned int blocksize;
static unsigned int decode_c(void);
static unsigned int decode_p(void);
static void make_table(int nchar, unsigned char bitlen[], int tablebits, unsigned int table[]);
/* lzh compression */
void de_lzh(int32_t ibytes, int32_t obytes, char **data, int bits)
{
unsigned int i, r, c;
int remains;
bit_be_inbytes = ibytes;
bit_be_filestart = *data;
bitsused = bits;
bit_be_init_getbits();
blocksize = 0;
decoded = 0;
r = 0;
for(;;) {
c = decode_c();
if(decoded) {
*data = bit_be_filestart;
return;
}
if(c <= UCHAR_MAX) {
out_ptr[r++] = c;
obytes--;
if(obytes == 0) {
*data = bit_be_filestart;
return;
}
} else {
remains = c - (UCHAR_MAX + 1 - THRESHOLD);
i = (r - decode_p() - 1);
while(--remains >= 0) {
out_ptr[r++] = out_ptr[i++];
obytes--;
if(obytes == 0) {
*data = bit_be_filestart;
return;
}
}
}
}
}
#define MAXMATCH 256
#define NC (UCHAR_MAX + MAXMATCH + 2 - THRESHOLD)
#define CBIT 9
#define CODE_BIT 16
#define NP (DICBIT + 1)
#define NT (CODE_BIT + 3)
#define PBIT 4 /* smallest integer such that (1U << PBIT) > NP */
#define TBIT 5 /* smallest integer such that (1U << TBIT) > NT */
#if NT > NP
# define NPT NT
#else
# define NPT NP
#endif
static unsigned int left[2 * NC - 1], right[2 * NC - 1];
static unsigned char c_len[NC], pt_len[NPT];
static unsigned int c_table[4096], pt_table[256];
static void
read_pt_len (int nn, int nbit, int i_special)
{
int i, c, n;
unsigned int mask;
n = bit_be_getbits(nbit);
if (n == 0) {
c = bit_be_getbits(nbit);
for (i = 0; i < nn; i++) {
pt_len[i] = 0;
}
for (i = 0; i < 256; i++) {
pt_table[i] = c;
}
} else {
i = 0;
while (i < n) {
c = bit_be_bitbuf >> (BITBUFSIZ - 3);
if (c == 7) {
mask = (unsigned) 1 << (BITBUFSIZ - 1 - 3);
while (mask & bit_be_bitbuf) {
mask >>= 1;
c++;
}
}
bit_be_fillbuf((c < 7) ? 3 : c - 3);
pt_len[i++] = c;
if (i == i_special) {
c = bit_be_getbits(2);
while (--c >= 0) {
pt_len[i++] = 0;
}
}
}
while (i < nn) {
pt_len[i++] = 0;
}
make_table(nn, pt_len, 8, pt_table);
}
}
static void
read_c_len (void)
{
int i, c, n;
unsigned int mask;
n = bit_be_getbits(CBIT);
if (n == 0) {
c = bit_be_getbits(CBIT);
for (i = 0; i < NC; i++) {
c_len[i] = 0;
}
for (i = 0; i < 4096; i++) {
c_table[i] = c;
}
} else {
i = 0;
while (i < n) {
c = pt_table[bit_be_bitbuf >> (BITBUFSIZ - 8)];
if (c >= NT) {
mask = (unsigned) 1 << (BITBUFSIZ - 1 - 8);
do {
if (bit_be_bitbuf & mask) {
c = right[c];
} else {
c = left [c];
}
mask >>= 1;
} while (c >= NT);
}
bit_be_fillbuf((int)pt_len[c]);
if (c <= 2) {
if (c == 0) {
c = 1;
} else if (c == 1) {
c = bit_be_getbits(4) + 3;
} else {
c = bit_be_getbits(CBIT) + 20;
}
while (--c >= 0) {
c_len[i++] = 0;
}
} else {
c_len[i++] = c - 2;
}
}
while (i < NC) {
c_len[i++] = 0;
}
make_table(NC, c_len, 12, c_table);
}
}
static unsigned int
decode_c (void)
{
unsigned int j, mask;
if (blocksize == 0) {
blocksize = bit_be_getbits(16);
if (blocksize == 0) {
decoded = 1;
return 0;
}
read_pt_len(NT, TBIT, 3);
read_c_len();
read_pt_len(bitsused + 1, PBIT, -1);
}
blocksize--;
j = c_table[bit_be_bitbuf >> (BITBUFSIZ - 12)];
if (j >= NC) {
mask = (unsigned) 1 << (BITBUFSIZ - 1 - 12);
do {
if (bit_be_bitbuf & mask) {
j = right[j];
} else {
j = left [j];
}
mask >>= 1;
} while (j >= NC);
}
bit_be_fillbuf((int)c_len[j]);
return j;
}
static unsigned int
decode_p (void)
{
unsigned int j, mask;
j = pt_table[bit_be_bitbuf >> (BITBUFSIZ - 8)];
if (j > bitsused) {
mask = (unsigned) 1 << (BITBUFSIZ - 1 - 8);
do {
if (bit_be_bitbuf & mask) {
j = right[j];
} else {
j = left [j];
}
mask >>= 1;
} while (j > bitsused);
}
bit_be_fillbuf((int)pt_len[j]);
if (j != 0) {
j = ((unsigned) 1 << (j - 1)) + bit_be_getbits((int) (j - 1));
}
return j;
}
static void
make_table (int nchar, unsigned char bitlen[], int tablebits, unsigned int table[])
{
unsigned int count[17], weight[17], start[18], *p;
unsigned int i, k, len, ch, jutbits, avail, nextcode, mask;
for (i = 1; i <= 16; i++) {
count[i] = 0;
}
for (i = 0; i < nchar; i++) {
count[bitlen[i]]++;
}
start[1] = 0;
for (i = 1; i <= 16; i++) {
start[i + 1] = start[i] + (count[i] << (16 - i));
}
jutbits = 16 - tablebits;
for (i = 1; i <= tablebits; i++) {
start[i] >>= jutbits;
weight[i] = (unsigned) 1 << (tablebits - i);
}
while (i <= 16) {
weight[i] = (unsigned) 1 << (16 - i);
i++;
}
i = start[tablebits + 1] >> jutbits;
if (i != (unsigned int)((unsigned) 1 << 16)) {
k = 1 << tablebits;
while (i != k) {
table[i++] = 0;
}
}
avail = nchar;
mask = (unsigned) 1 << (15 - tablebits);
for (ch = 0; ch < nchar; ch++) {
if ((len = bitlen[ch]) == 0) {
continue;
}
nextcode = start[len] + weight[len];
if (len <= tablebits) {
for (i = start[len]; i < nextcode; i++) {
table[i] = ch;
}
} else {
k = start[len];
p = &table[k >> jutbits];
i = len - tablebits;
while (i != 0) {
if (*p == 0) {
right[avail] = left[avail] = 0;
*p = avail++;
}
if (k & mask) {
p = &right[*p];
} else {
p = &left[*p];
}
k <<= 1;
i--;
}
*p = ch;
}
start[len] = nextcode;
}
}
#else /* DELZH */
int delzh; /* keep lint and some compilers happy */
#endif /* DELZH */