/* $OpenBSD: compress.c,v 1.8 2007/09/02 15:19:35 deraadt Exp $ */ /* $NetBSD: compress.c,v 1.3 1995/09/28 10:34:13 tls Exp $ */ /* * Copyright (c) 1989, 1993 * The Regents of the University of California. All rights reserved. * * This code is derived from software contributed to Berkeley by * Edward Wang at The University of California, Berkeley. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * 3. Neither the name of the University nor the names of its contributors * may be used to endorse or promote products derived from this software * without specific prior written permission. * * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF * SUCH DAMAGE. */ #ifndef lint #if 0 static char sccsid[] = "@(#)compress.c 8.1 (Berkeley) 6/6/93"; #else static char rcsid[] = "$OpenBSD: compress.c,v 1.8 2007/09/02 15:19:35 deraadt Exp $"; #endif #endif /* not lint */ #include #include #include "ww.h" #include "tt.h" /* special */ #include #include int cc_trace = 0; FILE *cc_trace_fp; /* tunable parameters */ int cc_reverse = 1; int cc_sort = 0; int cc_chop = 0; int cc_token_max = 8; /* <= TOKEN_MAX */ int cc_token_min = 2; /* > tt.tt_put_token_cost */ int cc_npass0 = 1; int cc_npass1 = 1; int cc_bufsize = 1024 * 3; /* XXX, or 80 * 24 * 2 */ int cc_ntoken = 8192; #define cc_weight XXX #ifndef cc_weight int cc_weight = 0; #endif #define TOKEN_MAX 16 struct cc { char string[TOKEN_MAX]; char length; char flag; #ifndef cc_weight short weight; #endif long time; /* time last seen */ short bcount; /* count in this buffer */ short ccount; /* count in compression */ short places; /* places in the buffer */ short code; /* token code */ struct cc *qforw, *qback; struct cc *hforw, **hback; }; short cc_thresholds[TOKEN_MAX + 1]; #define thresh(length) (cc_thresholds[length]) #define threshp(code, count, length) \ ((code) >= 0 || (short) (count) >= cc_thresholds[length]) #ifndef cc_weight short cc_wthresholds[TOKEN_MAX + 1]; #define wthresh(length) (cc_wthresholds[length]) #define wthreshp(weight, length) ((short) (weight) >= cc_wthresholds[length]) #else #define wthreshp(weight, length) (0) #endif #ifndef cc_weight short cc_wlimits[TOKEN_MAX + 1]; #define wlimit(length) (cc_wlimits[length]) #endif #define put_token_score(length) ((length) - tt.tt_put_token_cost) int cc_score_adjustments[TOKEN_MAX + 1][8]; /* XXX, 8 > max of cc_thresholds */ #define score_adjust(score, p) \ do { \ int length = (p)->length; \ int ccount = (p)->ccount; \ if (threshp((p)->code, ccount, length) || \ wthreshp((p)->weight, length)) /* XXX */ \ (score) -= length - tt.tt_put_token_cost; \ else \ (score) += cc_score_adjustments[length][ccount]; \ } while (0) int cc_initial_scores[TOKEN_MAX + 1][8]; /* XXX, 8 > max of cc_thresholds */ struct cc cc_q0a, cc_q0b, cc_q1a, cc_q1b; #define qinsert(p1, p2) \ do { \ struct cc *forw = (p1)->qforw; \ struct cc *back = (p1)->qback; \ back->qforw = forw; \ forw->qback = back; \ forw = (p2)->qforw; \ (p1)->qforw = forw; \ forw->qback = (p1); \ (p2)->qforw = (p1); \ (p1)->qback = (p2); \ } while (0) #define qinsertq(q, p) \ ((q)->qforw == (q) ? 0 : \ ((q)->qback->qforw = (p)->qforw, \ (p)->qforw->qback = (q)->qback, \ (q)->qforw->qback = (p), \ (p)->qforw = (q)->qforw, \ (q)->qforw = (q), \ (q)->qback = (q))) #define H (14) #define HSIZE (1 << H) #define hash(h, c) ((((h) >> H - 8 | (h) << 8) ^ (c)) & HSIZE - 1) char *cc_buffer; struct cc **cc_output; /* the output array */ short *cc_places[TOKEN_MAX + 1]; short *cc_hashcodes; /* for computing hashcodes */ struct cc **cc_htab; /* the hash table */ struct cc **cc_tokens; /* holds all the active tokens */ struct cc_undo { struct cc **pos; struct cc *val; } *cc_undo; long cc_time, cc_time0; char *cc_tt_ob, *cc_tt_obe; ccinit() { int i, j; struct cc *p; if (tt.tt_token_max > cc_token_max) tt.tt_token_max = cc_token_max; if (tt.tt_token_min < cc_token_min) tt.tt_token_min = cc_token_min; if (tt.tt_token_min > tt.tt_token_max) { tt.tt_ntoken = 0; return 0; } if (tt.tt_ntoken > cc_ntoken / 2) /* not likely */ tt.tt_ntoken = cc_ntoken / 2; #define C(x) (sizeof (x) / sizeof *(x)) for (i = 0; i < C(cc_thresholds); i++) { int h = i - tt.tt_put_token_cost; if (h > 0) cc_thresholds[i] = (tt.tt_set_token_cost + 1 + h - 1) / h + 1; else cc_thresholds[i] = 0; } for (i = 0; i < C(cc_score_adjustments); i++) { int t = cc_thresholds[i]; for (j = 0; j < C(*cc_score_adjustments); j++) { if (j >= t) cc_score_adjustments[i][j] = - (i - tt.tt_put_token_cost); else if (j < t - 1) cc_score_adjustments[i][j] = 0; else /* * cost now is * length * (ccount + 1) a * cost before was * set-token-cost + length + * ccount * put-token-cost b * the score adjustment is (b - a) */ cc_score_adjustments[i][j] = tt.tt_set_token_cost + i + j * tt.tt_put_token_cost - i * (j + 1); if (j >= t) cc_initial_scores[i][j] = 0; else /* * - (set-token-cost + * (length - put-token-cost) - * (length - put-token-cost) * ccount) */ cc_initial_scores[i][j] = - (tt.tt_set_token_cost + (i - tt.tt_put_token_cost) - (i - tt.tt_put_token_cost) * j); } } #ifndef cc_weight for (i = 1; i < C(cc_wthresholds); i++) { cc_wthresholds[i] = ((tt.tt_set_token_cost + tt.tt_put_token_cost) / i + i / 5 + 1) * cc_weight + 1; cc_wlimits[i] = cc_wthresholds[i] + cc_weight; } #endif #undef C if ((cc_output = (struct cc **) calloc(cc_bufsize, sizeof *cc_output)) == 0) goto nomem; if ((cc_hashcodes = (short *) calloc(cc_bufsize, sizeof *cc_hashcodes)) == 0) goto nomem; if ((cc_htab = (struct cc **) calloc(HSIZE, sizeof *cc_htab)) == 0) goto nomem; if ((cc_tokens = (struct cc **) malloc(cc_ntoken + tt.tt_token_max - tt.tt_token_min + 1), sizeof *cc_tokens) == 0) goto nomem; if ((cc_undo = (struct cc_undo *) calloc(cc_bufsize, sizeof *cc_undo)) == 0) goto nomem; for (i = tt.tt_token_min; i <= tt.tt_token_max; i++) if ((cc_places[i] = (short *) calloc(cc_bufsize, sizeof **cc_places)) == 0) goto nomem; cc_q0a.qforw = cc_q0a.qback = &cc_q0a; cc_q0b.qforw = cc_q0b.qback = &cc_q0b; cc_q1a.qforw = cc_q1a.qback = &cc_q1a; cc_q1b.qforw = cc_q1b.qback = &cc_q1b; if ((p = (struct cc *) calloc(cc_ntoken, sizeof *p)) == 0) goto nomem; for (i = 0; i < tt.tt_ntoken; i++) { p->code = i; p->time = -1; p->qback = cc_q0a.qback; p->qforw = &cc_q0a; p->qback->qforw = p; cc_q0a.qback = p; p++; } for (; i < cc_ntoken; i++) { p->code = -1; p->time = -1; p->qback = cc_q1a.qback; p->qforw = &cc_q1a; p->qback->qforw = p; cc_q1a.qback = p; p++; } cc_tt_ob = tt_ob; cc_tt_obe = tt_obe; if ((cc_buffer = malloc(cc_bufsize)) == 0) goto nomem; return 0; nomem: wwerrno = WWE_NOMEM; return -1; } ccstart() { int ccflush(); ttflush(); tt_obp = tt_ob = cc_buffer; tt_obe = tt_ob + cc_bufsize; tt.tt_flush = ccflush; if (cc_trace) { cc_trace_fp = fopen("window-trace", "a"); (void) fcntl(fileno(cc_trace_fp), F_SETFD, 1); } ccreset(); } ccreset() { struct cc *p; bzero((char *) cc_htab, HSIZE * sizeof *cc_htab); for (p = cc_q0a.qforw; p != &cc_q0a; p = p->qforw) p->hback = 0; for (p = cc_q1a.qforw; p != &cc_q1a; p = p->qforw) p->hback = 0; } ccend() { ttflush(); tt_obp = tt_ob = cc_tt_ob; tt_obe = cc_tt_obe; tt.tt_flush = 0; if (cc_trace_fp != NULL) { (void) fclose(cc_trace_fp); cc_trace_fp = NULL; } } ccflush() { int bufsize = tt_obp - tt_ob; int n; if (tt_ob != cc_buffer) abort(); if (cc_trace_fp != NULL) { (void) fwrite(tt_ob, 1, bufsize, cc_trace_fp); (void) putc(-1, cc_trace_fp); } tt.tt_flush = 0; (*tt.tt_compress)(1); if (bufsize < tt.tt_token_min) { ttflush(); goto out; } tt_obp = tt_ob = cc_tt_ob; tt_obe = cc_tt_obe; cc_time0 = cc_time; cc_time += bufsize; n = cc_sweep_phase(cc_buffer, bufsize, cc_tokens); cc_compress_phase(cc_output, bufsize, cc_tokens, n); cc_output_phase(cc_buffer, cc_output, bufsize); ttflush(); tt_obp = tt_ob = cc_buffer; tt_obe = cc_buffer + cc_bufsize; out: (*tt.tt_compress)(0); tt.tt_flush = ccflush; } cc_sweep_phase(buffer, bufsize, tokens) char *buffer; struct cc **tokens; { struct cc **pp = tokens; int i, n; #ifdef STATS int nn, ii; #endif #ifdef STATS if (verbose >= 0) time_begin(); if (verbose > 0) printf("Sweep:"); #endif cc_sweep0(buffer, bufsize, tt.tt_token_min - 1); #ifdef STATS ntoken_stat = 0; nn = 0; ii = 0; #endif for (i = tt.tt_token_min; i <= tt.tt_token_max; i++) { #ifdef STATS if (verbose > 0) { if (ii > 7) { printf("\n "); ii = 0; } ii++; printf(" (%d", i); (void) fflush(stdout); } #endif n = cc_sweep(buffer, bufsize, pp, i); pp += n; #ifdef STATS if (verbose > 0) { if (--n > 0) { printf(" %d", n); nn += n; } putchar(')'); } #endif } qinsertq(&cc_q1b, &cc_q1a); #ifdef STATS if (verbose > 0) printf("\n %d tokens, %d candidates\n", ntoken_stat, nn); if (verbose >= 0) time_end(); #endif return pp - tokens; } cc_sweep0(buffer, n, length) char *buffer; { char *p; short *hc; int i; short c; short pc = tt.tt_padc; /* n and length are at least 1 */ p = buffer++; hc = cc_hashcodes; i = n; do { if ((*hc++ = *p++) == pc) hc[-1] = -1; } while (--i); while (--length) { p = buffer++; hc = cc_hashcodes; for (i = n--; --i;) { if ((c = *p++) == pc || *hc < 0) c = -1; else c = hash(*hc, c); *hc++ = c; } } } cc_sweep(buffer, bufsize, tokens, length) char *buffer; struct cc **tokens; int length; { struct cc *p; char *cp; int i; short *hc; short *places = cc_places[length]; struct cc **pp = tokens; short threshold = thresh(length); #ifndef cc_weight short wthreshold = wthresh(length); short limit = wlimit(length); #endif int time; short pc = tt.tt_padc; i = length - 1; bufsize -= i; cp = buffer + i; hc = cc_hashcodes; time = cc_time0; for (i = 0; i < bufsize; i++, time++) { struct cc **h; { short *hc1 = hc; short c = *cp++; short hh; if ((hh = *hc1) < 0 || c == pc) { *hc1++ = -1; hc = hc1; continue; } h = cc_htab + (*hc1++ = hash(hh, c)); hc = hc1; } for (p = *h; p != 0; p = p->hforw) if (p->length == (char) length) { char *p1 = p->string; char *p2 = cp - length; int n = length; do if (*p1++ != *p2++) goto fail; while (--n); break; fail:; } if (p == 0) { p = cc_q1a.qback; if (p == &cc_q1a || p->time >= cc_time0 && p->length == (char) length) continue; if (p->hback != 0) if ((*p->hback = p->hforw) != 0) p->hforw->hback = p->hback; { char *p1 = p->string; char *p2 = cp - length; int n = length; do *p1++ = *p2++; while (--n); } p->length = length; #ifndef cc_weight p->weight = cc_weight; #endif p->time = time; p->bcount = 1; p->ccount = 0; p->flag = 0; if ((p->hforw = *h) != 0) p->hforw->hback = &p->hforw; *h = p; p->hback = h; qinsert(p, &cc_q1a); places[i] = -1; p->places = i; #ifdef STATS ntoken_stat++; #endif } else if (p->time < cc_time0) { #ifndef cc_weight if ((p->weight += p->time - time) < 0) p->weight = cc_weight; else if ((p->weight += cc_weight) > limit) p->weight = limit; #endif p->time = time; p->bcount = 1; p->ccount = 0; if (p->code >= 0) { p->flag = 1; *pp++ = p; } else #ifndef cc_weight if (p->weight >= wthreshold) { p->flag = 1; *pp++ = p; qinsert(p, &cc_q1b); } else #endif { p->flag = 0; qinsert(p, &cc_q1a); } places[i] = -1; p->places = i; #ifdef STATS ntoken_stat++; #endif } else if (p->time + length > time) { /* * overlapping token, don't count as two and * don't update time, but do adjust weight to offset * the difference */ #ifndef cc_weight if (cc_weight != 0) { /* XXX */ p->weight += time - p->time; if (!p->flag && p->weight >= wthreshold) { p->flag = 1; *pp++ = p; qinsert(p, &cc_q1b); } } #endif places[i] = p->places; p->places = i; } else { #ifndef cc_weight if ((p->weight += p->time - time) < 0) p->weight = cc_weight; else if ((p->weight += cc_weight) > limit) p->weight = limit; #endif p->time = time; p->bcount++; if (!p->flag && /* code must be < 0 if flag false here */ (p->bcount >= threshold #ifndef cc_weight || p->weight >= wthreshold #endif )) { p->flag = 1; *pp++ = p; qinsert(p, &cc_q1b); } places[i] = p->places; p->places = i; } } if ((i = pp - tokens) > 0) { *pp = 0; if (cc_reverse) cc_sweep_reverse(tokens, places); if (cc_sort && i > 1) { int cc_token_compare(); qsort((char *) tokens, i, sizeof *tokens, cc_token_compare); } if (cc_chop) { if ((i = i * cc_chop / 100) == 0) i = 1; tokens[i] = 0; } i++; } return i; } cc_sweep_reverse(pp, places) struct cc **pp; short *places; { struct cc *p; short front, back, t; while ((p = *pp++) != 0) { back = -1; t = p->places; /* the list is never empty */ do { front = places[t]; places[t] = back; back = t; } while ((t = front) >= 0); p->places = back; } } cc_compress_phase(output, bufsize, tokens, ntoken) struct cc **output; struct cc **tokens; { int i; bzero((char *) output, bufsize * sizeof *output); for (i = 0; i < cc_npass0; i++) cc_compress_phase1(output, tokens, ntoken, 0); for (i = 0; i < cc_npass1; i++) cc_compress_phase1(output, tokens, ntoken, 1); cc_compress_cleanup(output, bufsize); } cc_compress_phase1(output, tokens, ntoken, flag) struct cc **output; struct cc **tokens; { struct cc **pp; #ifdef STATS int i = 0; int nt = 0, cc = 0, nc = 0; #endif #ifdef STATS if (verbose >= 0) time_begin(); if (verbose > 0) printf("Compress:"); #endif pp = tokens; while (pp < tokens + ntoken) { #ifdef STATS if (verbose > 0) { ntoken_stat = 0; ccount_stat = 0; ncover_stat = 0; if (i > 2) { printf("\n "); i = 0; } i++; printf(" (%d", (*pp)->length); (void) fflush(stdout); } #endif pp += cc_compress(output, pp, flag); #ifdef STATS if (verbose > 0) { printf(" %dt %du %dc)", ntoken_stat, ccount_stat, ncover_stat); nt += ntoken_stat; cc += ccount_stat; nc += ncover_stat; } #endif } #ifdef STATS if (verbose > 0) printf("\n total: (%dt %du %dc)\n", nt, cc, nc); if (verbose >= 0) time_end(); #endif } cc_compress_cleanup(output, bufsize) struct cc **output; { struct cc **end; /* the previous output phase may have been interrupted */ qinsertq(&cc_q0b, &cc_q0a); for (end = output + bufsize; output < end;) { struct cc *p; int length; if ((p = *output) == 0) { output++; continue; } length = p->length; if (!p->flag) { } else if (p->code >= 0) { qinsert(p, &cc_q0b); p->flag = 0; } else if (p->ccount == 0) { *output = 0; } else if (p->ccount >= thresh(length) #ifndef cc_weight || wthreshp(p->weight, length) #endif ) { p->flag = 0; } else { p->ccount = 0; *output = 0; } output += length; } } cc_compress(output, tokens, flag) struct cc **output; struct cc **tokens; char flag; { struct cc **pp = tokens; struct cc *p = *pp++; int length = p->length; int threshold = thresh(length); #ifndef cc_weight short wthreshold = wthresh(length); #endif short *places = cc_places[length]; int *initial_scores = cc_initial_scores[length]; int initial_score0 = put_token_score(length); do { int score; struct cc_undo *undop; int ccount; #ifdef STATS int ncover; #endif int i; ccount = p->ccount; if ((short) ccount >= p->bcount) continue; if (p->code >= 0 || ccount >= threshold) score = 0; #ifndef cc_weight else if (p->weight >= wthreshold) /* allow one fewer match than normal */ /* XXX, should adjust for ccount */ score = - tt.tt_set_token_cost; #endif else score = initial_scores[ccount]; undop = cc_undo; #ifdef STATS ncover = 0; #endif for (i = p->places; i >= 0; i = places[i]) { struct cc **jp; struct cc *x; struct cc **ip = output + i; int score0 = initial_score0; struct cc **iip = ip + length; struct cc_undo *undop1 = undop; if ((x = *(jp = ip)) != 0) goto z; while (--jp >= output) if ((x = *jp) != 0) { if (jp + x->length > ip) goto z; break; } jp = ip + 1; while (jp < iip) { if ((x = *jp) == 0) { jp++; continue; } z: if (x == p) goto undo; #ifdef STATS ncover++; #endif undop->pos = jp; undop->val = x; undop++; *jp = 0; x->ccount--; score_adjust(score0, x); if (score0 < 0 && flag) goto undo; jp += x->length; } undop->pos = ip; undop->val = 0; undop++; *ip = p; ccount++; score += score0; continue; undo: while (--undop >= undop1) if (*undop->pos = x = undop->val) x->ccount++; undop++; } if (score > 0) { #ifdef STATS ccount_stat += ccount - p->ccount; ntoken_stat++; ncover_stat += ncover; #endif p->ccount = ccount; } else { struct cc_undo *u = cc_undo; while (--undop >= u) { struct cc *x; if (*undop->pos = x = undop->val) x->ccount++; } } } while ((p = *pp++) != 0); return pp - tokens; } cc_output_phase(buffer, output, bufsize) char *buffer; struct cc **output; int bufsize; { int i; struct cc *p, *p1; for (i = 0; i < bufsize;) { if ((p = output[i]) == 0) { ttputc(buffer[i]); i++; } else if (p->code >= 0) { if (--p->ccount == 0) qinsert(p, &cc_q0a); (*tt.tt_put_token)(p->code, p->string, p->length); wwntokuse++; wwntoksave += put_token_score(p->length); i += p->length; } else if ((p1 = cc_q0a.qback) != &cc_q0a) { p->code = p1->code; p1->code = -1; qinsert(p1, &cc_q1a); if (--p->ccount == 0) qinsert(p, &cc_q0a); else qinsert(p, &cc_q0b); (*tt.tt_set_token)(p->code, p->string, p->length); wwntokdef++; wwntoksave -= tt.tt_set_token_cost; i += p->length; } else { p->ccount--; ttwrite(p->string, p->length); wwntokbad++; i += p->length; } } wwntokc += bufsize; } cc_token_compare(p1, p2) struct cc **p1, **p2; { return (*p2)->bcount - (*p1)->bcount; }