1 /* $NetBSD: bsd-comp.c,v 1.18 2008/06/15 16:37:21 christos Exp $ */ 2 /* Id: bsd-comp.c,v 1.6 1996/08/28 06:31:58 paulus Exp */ 3 4 /* Because this code is derived from the 4.3BSD compress source: 5 * 6 * 7 * Copyright (c) 1985, 1986 The Regents of the University of California. 8 * All rights reserved. 9 * 10 * This code is derived from software contributed to Berkeley by 11 * James A. Woods, derived from original work by Spencer Thomas 12 * and Joseph Orost. 13 * 14 * Redistribution and use in source and binary forms, with or without 15 * modification, are permitted provided that the following conditions 16 * are met: 17 * 1. Redistributions of source code must retain the above copyright 18 * notice, this list of conditions and the following disclaimer. 19 * 2. Redistributions in binary form must reproduce the above copyright 20 * notice, this list of conditions and the following disclaimer in the 21 * documentation and/or other materials provided with the distribution. 22 * 3. Neither the name of the University nor the names of its contributors 23 * may be used to endorse or promote products derived from this software 24 * without specific prior written permission. 25 * 26 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 27 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 28 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 29 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 30 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 31 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 32 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 33 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 34 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 35 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 36 * SUCH DAMAGE. 37 */ 38 39 /* 40 * This version is for use with mbufs on BSD-derived systems. 41 */ 42 43 #include <sys/cdefs.h> 44 __KERNEL_RCSID(0, "$NetBSD: bsd-comp.c,v 1.18 2008/06/15 16:37:21 christos Exp $"); 45 46 #include <sys/param.h> 47 #include <sys/systm.h> 48 #include <sys/mbuf.h> 49 #include <sys/socket.h> 50 #include <net/if.h> 51 #include <net/if_types.h> 52 #include <net/ppp_defs.h> 53 #include <net/if_ppp.h> 54 55 #define PACKETPTR struct mbuf * 56 #include <net/ppp-comp.h> 57 58 #if DO_BSD_COMPRESS 59 /* 60 * PPP "BSD compress" compression 61 * The differences between this compression and the classic BSD LZW 62 * source are obvious from the requirement that the classic code worked 63 * with files while this handles arbitrarily long streams that 64 * are broken into packets. They are: 65 * 66 * When the code size expands, a block of junk is not emitted by 67 * the compressor and not expected by the decompressor. 68 * 69 * New codes are not necessarily assigned every time an old 70 * code is output by the compressor. This is because a packet 71 * end forces a code to be emitted, but does not imply that a 72 * new sequence has been seen. 73 * 74 * The compression ratio is checked at the first end of a packet 75 * after the appropriate gap. Besides simplifying and speeding 76 * things up, this makes it more likely that the transmitter 77 * and receiver will agree when the dictionary is cleared when 78 * compression is not going well. 79 */ 80 81 /* 82 * A dictionary for doing BSD compress. 83 */ 84 struct bsd_db { 85 int totlen; /* length of this structure */ 86 u_int hsize; /* size of the hash table */ 87 u_char hshift; /* used in hash function */ 88 u_char n_bits; /* current bits/code */ 89 u_char maxbits; 90 u_char debug; 91 u_char unit; 92 uint16_t seqno; /* sequence # of next packet */ 93 u_int hdrlen; /* header length to preallocate */ 94 u_int mru; 95 u_int maxmaxcode; /* largest valid code */ 96 u_int max_ent; /* largest code in use */ 97 u_int in_count; /* uncompressed bytes, aged */ 98 u_int bytes_out; /* compressed bytes, aged */ 99 u_int ratio; /* recent compression ratio */ 100 u_int checkpoint; /* when to next check the ratio */ 101 u_int clear_count; /* times dictionary cleared */ 102 u_int incomp_count; /* incompressible packets */ 103 u_int incomp_bytes; /* incompressible bytes */ 104 u_int uncomp_count; /* uncompressed packets */ 105 u_int uncomp_bytes; /* uncompressed bytes */ 106 u_int comp_count; /* compressed packets */ 107 u_int comp_bytes; /* compressed bytes */ 108 uint16_t *lens; /* array of lengths of codes */ 109 struct bsd_dict { 110 union { /* hash value */ 111 uint32_t fcode; 112 struct { 113 #if BYTE_ORDER == LITTLE_ENDIAN 114 uint16_t prefix; /* preceding code */ 115 u_char suffix; /* last character of new code */ 116 u_char pad; 117 #else 118 u_char pad; 119 u_char suffix; /* last character of new code */ 120 uint16_t prefix; /* preceding code */ 121 #endif 122 } hs; 123 } f; 124 uint16_t codem1; /* output of hash table -1 */ 125 uint16_t cptr; /* map code to hash table entry */ 126 } dict[1]; 127 }; 128 129 #define BSD_OVHD 2 /* BSD compress overhead/packet */ 130 #define BSD_INIT_BITS BSD_MIN_BITS 131 132 static void *bsd_comp_alloc(u_char *options, int opt_len); 133 static void *bsd_decomp_alloc(u_char *options, int opt_len); 134 static void bsd_free(void *state); 135 static int bsd_comp_init(void *state, u_char *options, int opt_len, 136 int unit, int hdrlen, int debug); 137 static int bsd_decomp_init(void *state, u_char *options, int opt_len, 138 int unit, int hdrlen, int mru, int debug); 139 static int bsd_compress(void *state, struct mbuf **mret, 140 struct mbuf *mp, int slen, int maxolen); 141 static void bsd_incomp(void *state, struct mbuf *dmsg); 142 static int bsd_decompress(void *state, struct mbuf *cmp, 143 struct mbuf **dmpp); 144 static void bsd_reset(void *state); 145 static void bsd_comp_stats(void *state, struct compstat *stats); 146 147 /* 148 * Procedures exported to if_ppp.c. 149 */ 150 struct compressor ppp_bsd_compress = { 151 CI_BSD_COMPRESS, /* compress_proto */ 152 bsd_comp_alloc, /* comp_alloc */ 153 bsd_free, /* comp_free */ 154 bsd_comp_init, /* comp_init */ 155 bsd_reset, /* comp_reset */ 156 bsd_compress, /* compress */ 157 bsd_comp_stats, /* comp_stat */ 158 bsd_decomp_alloc, /* decomp_alloc */ 159 bsd_free, /* decomp_free */ 160 bsd_decomp_init, /* decomp_init */ 161 bsd_reset, /* decomp_reset */ 162 bsd_decompress, /* decompress */ 163 bsd_incomp, /* incomp */ 164 bsd_comp_stats, /* decomp_stat */ 165 }; 166 167 /* 168 * the next two codes should not be changed lightly, as they must not 169 * lie within the contiguous general code space. 170 */ 171 #define CLEAR 256 /* table clear output code */ 172 #define FIRST 257 /* first free entry */ 173 #define LAST 255 174 175 #define MAXCODE(b) ((1 << (b)) - 1) 176 #define BADCODEM1 MAXCODE(BSD_MAX_BITS) 177 178 #define BSD_HASH(prefix,suffix,hshift) ((((uint32_t)(suffix)) << (hshift)) \ 179 ^ (uint32_t)(prefix)) 180 #define BSD_KEY(prefix,suffix) ((((uint32_t)(suffix)) << 16) \ 181 + (uint32_t)(prefix)) 182 183 #define CHECK_GAP 10000 /* Ratio check interval */ 184 185 #define RATIO_SCALE_LOG 8 186 #define RATIO_SCALE (1<<RATIO_SCALE_LOG) 187 #define RATIO_MAX (0x7fffffff>>RATIO_SCALE_LOG) 188 189 static void bsd_clear(struct bsd_db *); 190 static int bsd_check(struct bsd_db *); 191 static void *bsd_alloc(u_char *, int, int); 192 static int bsd_init(struct bsd_db *, u_char *, int, int, int, int, 193 int, int); 194 195 /* 196 * clear the dictionary 197 */ 198 static void 199 bsd_clear(struct bsd_db *db) 200 { 201 db->clear_count++; 202 db->max_ent = FIRST-1; 203 db->n_bits = BSD_INIT_BITS; 204 db->ratio = 0; 205 db->bytes_out = 0; 206 db->in_count = 0; 207 db->checkpoint = CHECK_GAP; 208 } 209 210 /* 211 * If the dictionary is full, then see if it is time to reset it. 212 * 213 * Compute the compression ratio using fixed-point arithmetic 214 * with 8 fractional bits. 215 * 216 * Since we have an infinite stream instead of a single file, 217 * watch only the local compression ratio. 218 * 219 * Since both peers must reset the dictionary at the same time even in 220 * the absence of CLEAR codes (while packets are incompressible), they 221 * must compute the same ratio. 222 */ 223 static int /* 1=output CLEAR */ 224 bsd_check(struct bsd_db *db) 225 { 226 u_int new_ratio; 227 228 if (db->in_count >= db->checkpoint) { 229 /* age the ratio by limiting the size of the counts */ 230 if (db->in_count >= RATIO_MAX 231 || db->bytes_out >= RATIO_MAX) { 232 db->in_count -= db->in_count/4; 233 db->bytes_out -= db->bytes_out/4; 234 } 235 236 db->checkpoint = db->in_count + CHECK_GAP; 237 238 if (db->max_ent >= db->maxmaxcode) { 239 /* Reset the dictionary only if the ratio is worse, 240 * or if it looks as if it has been poisoned 241 * by incompressible data. 242 * 243 * This does not overflow, because 244 * db->in_count <= RATIO_MAX. 245 */ 246 new_ratio = db->in_count << RATIO_SCALE_LOG; 247 if (db->bytes_out != 0) 248 new_ratio /= db->bytes_out; 249 250 if (new_ratio < db->ratio || new_ratio < 1 * RATIO_SCALE) { 251 bsd_clear(db); 252 return 1; 253 } 254 db->ratio = new_ratio; 255 } 256 } 257 return 0; 258 } 259 260 /* 261 * Return statistics. 262 */ 263 static void 264 bsd_comp_stats(void *state, struct compstat *stats) 265 { 266 struct bsd_db *db = (struct bsd_db *) state; 267 u_int out; 268 269 stats->unc_bytes = db->uncomp_bytes; 270 stats->unc_packets = db->uncomp_count; 271 stats->comp_bytes = db->comp_bytes; 272 stats->comp_packets = db->comp_count; 273 stats->inc_bytes = db->incomp_bytes; 274 stats->inc_packets = db->incomp_count; 275 stats->ratio = db->in_count; 276 out = db->bytes_out; 277 if (stats->ratio <= 0x7fffff) 278 stats->ratio <<= 8; 279 else 280 out >>= 8; 281 if (out != 0) 282 stats->ratio /= out; 283 } 284 285 /* 286 * Reset state, as on a CCP ResetReq. 287 */ 288 static void 289 bsd_reset(void *state) 290 { 291 struct bsd_db *db = (struct bsd_db *) state; 292 293 db->seqno = 0; 294 bsd_clear(db); 295 db->clear_count = 0; 296 } 297 298 /* 299 * Allocate space for a (de) compressor. 300 */ 301 static void * 302 bsd_alloc(u_char *options, int opt_len, int decomp) 303 { 304 int bits; 305 u_int newlen, hsize, hshift, maxmaxcode; 306 struct bsd_db *db; 307 308 if (opt_len < CILEN_BSD_COMPRESS || options[0] != CI_BSD_COMPRESS 309 || options[1] != CILEN_BSD_COMPRESS 310 || BSD_VERSION(options[2]) != BSD_CURRENT_VERSION) 311 return NULL; 312 bits = BSD_NBITS(options[2]); 313 switch (bits) { 314 case 9: /* needs 82152 for both directions */ 315 case 10: /* needs 84144 */ 316 case 11: /* needs 88240 */ 317 case 12: /* needs 96432 */ 318 hsize = 5003; 319 hshift = 4; 320 break; 321 case 13: /* needs 176784 */ 322 hsize = 9001; 323 hshift = 5; 324 break; 325 case 14: /* needs 353744 */ 326 hsize = 18013; 327 hshift = 6; 328 break; 329 case 15: /* needs 691440 */ 330 hsize = 35023; 331 hshift = 7; 332 break; 333 case 16: /* needs 1366160--far too much, */ 334 /* hsize = 69001; */ /* and 69001 is too big for cptr */ 335 /* hshift = 8; */ /* in struct bsd_db */ 336 /* break; */ 337 default: 338 return NULL; 339 } 340 341 maxmaxcode = MAXCODE(bits); 342 newlen = sizeof(*db) + (hsize-1) * (sizeof(db->dict[0])); 343 db = malloc(newlen, M_DEVBUF, M_NOWAIT|M_ZERO); 344 if (!db) 345 return NULL; 346 347 if (!decomp) { 348 db->lens = NULL; 349 } else { 350 db->lens = malloc((maxmaxcode+1) * sizeof(db->lens[0]), 351 M_DEVBUF, M_NOWAIT); 352 if (!db->lens) { 353 free(db, M_DEVBUF); 354 return NULL; 355 } 356 } 357 358 db->totlen = newlen; 359 db->hsize = hsize; 360 db->hshift = hshift; 361 db->maxmaxcode = maxmaxcode; 362 db->maxbits = bits; 363 364 return (void *) db; 365 } 366 367 static void 368 bsd_free(void *state) 369 { 370 struct bsd_db *db = (struct bsd_db *) state; 371 372 if (db->lens) 373 free(db->lens, M_DEVBUF); 374 free(db, M_DEVBUF); 375 } 376 377 static void * 378 bsd_comp_alloc(u_char *options, int opt_len) 379 { 380 return bsd_alloc(options, opt_len, 0); 381 } 382 383 static void * 384 bsd_decomp_alloc(u_char *options, int opt_len) 385 { 386 return bsd_alloc(options, opt_len, 1); 387 } 388 389 /* 390 * Initialize the database. 391 */ 392 static int 393 bsd_init(struct bsd_db *db, u_char *options, int opt_len, int unit, int hdrlen, 394 int mru, int debug, int decomp) 395 { 396 int i; 397 398 if (opt_len < CILEN_BSD_COMPRESS || options[0] != CI_BSD_COMPRESS 399 || options[1] != CILEN_BSD_COMPRESS 400 || BSD_VERSION(options[2]) != BSD_CURRENT_VERSION 401 || BSD_NBITS(options[2]) != db->maxbits 402 || (decomp && db->lens == NULL)) 403 return 0; 404 405 if (decomp) { 406 i = LAST+1; 407 while (i != 0) 408 db->lens[--i] = 1; 409 } 410 i = db->hsize; 411 while (i != 0) { 412 db->dict[--i].codem1 = BADCODEM1; 413 db->dict[i].cptr = 0; 414 } 415 416 db->unit = unit; 417 db->hdrlen = hdrlen; 418 db->mru = mru; 419 #ifndef DEBUG 420 if (debug) 421 #endif 422 db->debug = 1; 423 424 bsd_reset(db); 425 426 return 1; 427 } 428 429 static int 430 bsd_comp_init(void *state, u_char *options, int opt_len, int unit, int hdrlen, 431 int debug) 432 { 433 return bsd_init((struct bsd_db *) state, options, opt_len, 434 unit, hdrlen, 0, debug, 0); 435 } 436 437 static int 438 bsd_decomp_init(void *state, u_char *options, int opt_len, int unit, int hdrlen, 439 int mru, int debug) 440 { 441 return bsd_init((struct bsd_db *) state, options, opt_len, 442 unit, hdrlen, mru, debug, 1); 443 } 444 445 446 /* 447 * compress a packet 448 * One change from the BSD compress command is that when the 449 * code size expands, we do not output a bunch of padding. 450 */ 451 int /* new slen */ 452 bsd_compress(void *state, 453 struct mbuf **mret /* return compressed mbuf chain here */, 454 struct mbuf *mp /* from here */, 455 int slen, int maxolen) 456 { 457 struct bsd_db *db = (struct bsd_db *) state; 458 int hshift = db->hshift; 459 u_int max_ent = db->max_ent; 460 u_int n_bits = db->n_bits; 461 u_int bitno = 32; 462 uint32_t accm = 0, fcode; 463 struct bsd_dict *dictp; 464 u_char c; 465 int hval, disp, ent, ilen; 466 u_char *rptr, *wptr; 467 u_char *cp_end; 468 int olen; 469 struct mbuf *m; 470 471 #define PUTBYTE(v) { \ 472 ++olen; \ 473 if (wptr) { \ 474 *wptr++ = (v); \ 475 if (wptr >= cp_end) { \ 476 m->m_len = wptr - mtod(m, u_char *); \ 477 MGET(m->m_next, M_DONTWAIT, MT_DATA); \ 478 m = m->m_next; \ 479 if (m) { \ 480 m->m_len = 0; \ 481 if (maxolen - olen > MLEN) \ 482 MCLGET(m, M_DONTWAIT); \ 483 wptr = mtod(m, u_char *); \ 484 cp_end = wptr + M_TRAILINGSPACE(m); \ 485 } else \ 486 wptr = NULL; \ 487 } \ 488 } \ 489 } 490 491 #define OUTPUT(ent) { \ 492 bitno -= n_bits; \ 493 accm |= ((ent) << bitno); \ 494 do { \ 495 PUTBYTE(accm >> 24); \ 496 accm <<= 8; \ 497 bitno += 8; \ 498 } while (bitno <= 24); \ 499 } 500 501 /* 502 * If the protocol is not in the range we're interested in, 503 * just return without compressing the packet. If it is, 504 * the protocol becomes the first byte to compress. 505 */ 506 rptr = mtod(mp, u_char *); 507 ent = PPP_PROTOCOL(rptr); 508 if (ent < 0x21 || ent > 0xf9) { 509 *mret = NULL; 510 return slen; 511 } 512 513 /* Don't generate compressed packets which are larger than 514 the uncompressed packet. */ 515 if (maxolen > slen) 516 maxolen = slen; 517 518 /* Allocate one mbuf to start with. */ 519 MGET(m, M_DONTWAIT, MT_DATA); 520 *mret = m; 521 if (m != NULL) { 522 m->m_len = 0; 523 if (maxolen + db->hdrlen > MLEN) 524 MCLGET(m, M_DONTWAIT); 525 m->m_data += db->hdrlen; 526 wptr = mtod(m, u_char *); 527 cp_end = wptr + M_TRAILINGSPACE(m); 528 } else 529 wptr = cp_end = NULL; 530 531 /* 532 * Copy the PPP header over, changing the protocol, 533 * and install the 2-byte packet sequence number. 534 */ 535 if (wptr) { 536 *wptr++ = PPP_ADDRESS(rptr); /* assumes the ppp header is */ 537 *wptr++ = PPP_CONTROL(rptr); /* all in one mbuf */ 538 *wptr++ = 0; /* change the protocol */ 539 *wptr++ = PPP_COMP; 540 *wptr++ = db->seqno >> 8; 541 *wptr++ = db->seqno; 542 } 543 ++db->seqno; 544 545 olen = 0; 546 rptr += PPP_HDRLEN; 547 slen = mp->m_len - PPP_HDRLEN; 548 ilen = slen + 1; 549 for (;;) { 550 if (slen <= 0) { 551 mp = mp->m_next; 552 if (!mp) 553 break; 554 rptr = mtod(mp, u_char *); 555 slen = mp->m_len; 556 if (!slen) 557 continue; /* handle 0-length buffers */ 558 ilen += slen; 559 } 560 561 slen--; 562 c = *rptr++; 563 fcode = BSD_KEY(ent, c); 564 hval = BSD_HASH(ent, c, hshift); 565 dictp = &db->dict[hval]; 566 567 /* Validate and then check the entry. */ 568 if (dictp->codem1 >= max_ent) 569 goto nomatch; 570 if (dictp->f.fcode == fcode) { 571 ent = dictp->codem1+1; 572 continue; /* found (prefix,suffix) */ 573 } 574 575 /* continue probing until a match or invalid entry */ 576 disp = (hval == 0) ? 1 : hval; 577 do { 578 hval += disp; 579 if (hval >= db->hsize) 580 hval -= db->hsize; 581 dictp = &db->dict[hval]; 582 if (dictp->codem1 >= max_ent) 583 goto nomatch; 584 } while (dictp->f.fcode != fcode); 585 ent = dictp->codem1 + 1; /* finally found (prefix,suffix) */ 586 continue; 587 588 nomatch: 589 OUTPUT(ent); /* output the prefix */ 590 591 /* code -> hashtable */ 592 if (max_ent < db->maxmaxcode) { 593 struct bsd_dict *dictp2; 594 /* expand code size if needed */ 595 if (max_ent >= MAXCODE(n_bits)) 596 db->n_bits = ++n_bits; 597 598 /* Invalidate old hash table entry using 599 * this code, and then take it over. 600 */ 601 dictp2 = &db->dict[max_ent+1]; 602 if (db->dict[dictp2->cptr].codem1 == max_ent) 603 db->dict[dictp2->cptr].codem1 = BADCODEM1; 604 dictp2->cptr = hval; 605 dictp->codem1 = max_ent; 606 dictp->f.fcode = fcode; 607 608 db->max_ent = ++max_ent; 609 } 610 ent = c; 611 } 612 613 OUTPUT(ent); /* output the last code */ 614 db->bytes_out += olen; 615 db->in_count += ilen; 616 if (bitno < 32) 617 ++db->bytes_out; /* count complete bytes */ 618 619 if (bsd_check(db)) 620 OUTPUT(CLEAR); /* do not count the CLEAR */ 621 622 /* 623 * Pad dribble bits of last code with ones. 624 * Do not emit a completely useless byte of ones. 625 */ 626 if (bitno != 32) 627 PUTBYTE((accm | (0xff << (bitno-8))) >> 24); 628 629 if (m != NULL) { 630 m->m_len = wptr - mtod(m, u_char *); 631 m->m_next = NULL; 632 } 633 634 /* 635 * Increase code size if we would have without the packet 636 * boundary and as the decompressor will. 637 */ 638 if (max_ent >= MAXCODE(n_bits) && max_ent < db->maxmaxcode) 639 db->n_bits++; 640 641 db->uncomp_bytes += ilen; 642 ++db->uncomp_count; 643 if (olen + PPP_HDRLEN + BSD_OVHD > maxolen) { 644 /* throw away the compressed stuff if it is longer than uncompressed */ 645 if (*mret != NULL) { 646 m_freem(*mret); 647 *mret = NULL; 648 } 649 ++db->incomp_count; 650 db->incomp_bytes += ilen; 651 } else { 652 ++db->comp_count; 653 db->comp_bytes += olen + BSD_OVHD; 654 } 655 656 return olen + PPP_HDRLEN + BSD_OVHD; 657 #undef OUTPUT 658 #undef PUTBYTE 659 } 660 661 662 /* 663 * Update the "BSD Compress" dictionary on the receiver for 664 * incompressible data by pretending to compress the incoming data. 665 */ 666 static void 667 bsd_incomp(void *state, struct mbuf *dmsg) 668 { 669 struct bsd_db *db = (struct bsd_db *) state; 670 u_int hshift = db->hshift; 671 u_int max_ent = db->max_ent; 672 u_int n_bits = db->n_bits; 673 struct bsd_dict *dictp; 674 uint32_t fcode; 675 u_char c; 676 uint32_t hval, disp; 677 int slen, ilen; 678 u_int bitno = 7; 679 u_char *rptr; 680 u_int ent; 681 682 /* 683 * If the protocol is not in the range we're interested in, 684 * just return without looking at the packet. If it is, 685 * the protocol becomes the first byte to "compress". 686 */ 687 rptr = mtod(dmsg, u_char *); 688 ent = PPP_PROTOCOL(rptr); 689 if (ent < 0x21 || ent > 0xf9) 690 return; 691 692 db->seqno++; 693 ilen = 1; /* count the protocol as 1 byte */ 694 rptr += PPP_HDRLEN; 695 slen = dmsg->m_len - PPP_HDRLEN; 696 for (;;) { 697 if (slen <= 0) { 698 dmsg = dmsg->m_next; 699 if (!dmsg) 700 break; 701 rptr = mtod(dmsg, u_char *); 702 slen = dmsg->m_len; 703 continue; 704 } 705 ilen += slen; 706 707 do { 708 c = *rptr++; 709 fcode = BSD_KEY(ent, c); 710 hval = BSD_HASH(ent, c, hshift); 711 dictp = &db->dict[hval]; 712 713 /* validate and then check the entry */ 714 if (dictp->codem1 >= max_ent) 715 goto nomatch; 716 if (dictp->f.fcode == fcode) { 717 ent = dictp->codem1+1; 718 continue; /* found (prefix,suffix) */ 719 } 720 721 /* continue probing until a match or invalid entry */ 722 disp = (hval == 0) ? 1 : hval; 723 do { 724 hval += disp; 725 if (hval >= db->hsize) 726 hval -= db->hsize; 727 dictp = &db->dict[hval]; 728 if (dictp->codem1 >= max_ent) 729 goto nomatch; 730 } while (dictp->f.fcode != fcode); 731 ent = dictp->codem1+1; 732 continue; /* finally found (prefix,suffix) */ 733 734 nomatch: /* output (count) the prefix */ 735 bitno += n_bits; 736 737 /* code -> hashtable */ 738 if (max_ent < db->maxmaxcode) { 739 struct bsd_dict *dictp2; 740 /* expand code size if needed */ 741 if (max_ent >= MAXCODE(n_bits)) 742 db->n_bits = ++n_bits; 743 744 /* Invalidate previous hash table entry 745 * assigned this code, and then take it over. 746 */ 747 dictp2 = &db->dict[max_ent+1]; 748 if (db->dict[dictp2->cptr].codem1 == max_ent) 749 db->dict[dictp2->cptr].codem1 = BADCODEM1; 750 dictp2->cptr = hval; 751 dictp->codem1 = max_ent; 752 dictp->f.fcode = fcode; 753 754 db->max_ent = ++max_ent; 755 db->lens[max_ent] = db->lens[ent]+1; 756 } 757 ent = c; 758 } while (--slen != 0); 759 } 760 bitno += n_bits; /* output (count) the last code */ 761 db->bytes_out += bitno/8; 762 db->in_count += ilen; 763 (void)bsd_check(db); 764 765 ++db->incomp_count; 766 db->incomp_bytes += ilen; 767 ++db->uncomp_count; 768 db->uncomp_bytes += ilen; 769 770 /* Increase code size if we would have without the packet 771 * boundary and as the decompressor will. 772 */ 773 if (max_ent >= MAXCODE(n_bits) && max_ent < db->maxmaxcode) 774 db->n_bits++; 775 } 776 777 778 /* 779 * Decompress "BSD Compress". 780 * 781 * Because of patent problems, we return DECOMP_ERROR for errors 782 * found by inspecting the input data and for system problems, but 783 * DECOMP_FATALERROR for any errors which could possibly be said to 784 * be being detected "after" decompression. For DECOMP_ERROR, 785 * we can issue a CCP reset-request; for DECOMP_FATALERROR, we may be 786 * infringing a patent of Motorola's if we do, so we take CCP down 787 * instead. 788 * 789 * Given that the frame has the correct sequence number and a good FCS, 790 * errors such as invalid codes in the input most likely indicate a 791 * bug, so we return DECOMP_FATALERROR for them in order to turn off 792 * compression, even though they are detected by inspecting the input. 793 */ 794 int 795 bsd_decompress(void *state, struct mbuf *cmp, struct mbuf **dmpp) 796 { 797 struct bsd_db *db = (struct bsd_db *) state; 798 u_int max_ent = db->max_ent; 799 uint32_t accm = 0; 800 u_int bitno = 32; /* 1st valid bit in accm */ 801 u_int n_bits = db->n_bits; 802 u_int tgtbitno = 32-n_bits; /* bitno when we have a code */ 803 struct bsd_dict *dictp; 804 int explen, i, seq, len; 805 u_int incode, oldcode, finchar; 806 u_char *p, *rptr, *wptr; 807 struct mbuf *m, *dmp, *mret; 808 int adrs, ctrl, ilen; 809 int space, codelen, extra; 810 811 /* 812 * Save the address/control from the PPP header 813 * and then get the sequence number. 814 */ 815 *dmpp = NULL; 816 rptr = mtod(cmp, u_char *); 817 adrs = PPP_ADDRESS(rptr); 818 ctrl = PPP_CONTROL(rptr); 819 rptr += PPP_HDRLEN; 820 len = cmp->m_len - PPP_HDRLEN; 821 seq = 0; 822 for (i = 0; i < 2; ++i) { 823 while (len <= 0) { 824 cmp = cmp->m_next; 825 if (cmp == NULL) 826 return DECOMP_ERROR; 827 rptr = mtod(cmp, u_char *); 828 len = cmp->m_len; 829 } 830 seq = (seq << 8) + *rptr++; 831 --len; 832 } 833 834 /* 835 * Check the sequence number and give up if it differs from 836 * the value we're expecting. 837 */ 838 if (seq != db->seqno) { 839 if (db->debug) 840 printf("bsd_decomp%d: bad sequence # %d, expected %d\n", 841 db->unit, seq, db->seqno - 1); 842 return DECOMP_ERROR; 843 } 844 ++db->seqno; 845 846 /* 847 * Allocate one mbuf to start with. 848 */ 849 MGETHDR(dmp, M_DONTWAIT, MT_DATA); 850 if (dmp == NULL) 851 return DECOMP_ERROR; 852 mret = dmp; 853 dmp->m_len = 0; 854 dmp->m_next = NULL; 855 MCLGET(dmp, M_DONTWAIT); 856 dmp->m_data += db->hdrlen; 857 wptr = mtod(dmp, u_char *); 858 space = M_TRAILINGSPACE(dmp) - PPP_HDRLEN + 1; 859 860 /* 861 * Fill in the ppp header, but not the last byte of the protocol 862 * (that comes from the decompressed data). 863 */ 864 wptr[0] = adrs; 865 wptr[1] = ctrl; 866 wptr[2] = 0; 867 wptr += PPP_HDRLEN - 1; 868 869 ilen = len; 870 oldcode = CLEAR; 871 explen = 0; 872 for (;;) { 873 if (len == 0) { 874 cmp = cmp->m_next; 875 if (!cmp) /* quit at end of message */ 876 break; 877 rptr = mtod(cmp, u_char *); 878 len = cmp->m_len; 879 ilen += len; 880 continue; /* handle 0-length buffers */ 881 } 882 883 /* 884 * Accumulate bytes until we have a complete code. 885 * Then get the next code, relying on the 32-bit, 886 * unsigned accm to mask the result. 887 */ 888 bitno -= 8; 889 accm |= *rptr++ << bitno; 890 --len; 891 if (tgtbitno < bitno) 892 continue; 893 incode = accm >> tgtbitno; 894 accm <<= n_bits; 895 bitno += n_bits; 896 897 if (incode == CLEAR) { 898 /* 899 * The dictionary must only be cleared at 900 * the end of a packet. But there could be an 901 * empty mbuf at the end. 902 */ 903 if (len > 0 || cmp->m_next != NULL) { 904 while ((cmp = cmp->m_next) != NULL) 905 len += cmp->m_len; 906 if (len > 0) { 907 m_freem(mret); 908 if (db->debug) 909 printf("bsd_decomp%d: bad CLEAR\n", db->unit); 910 return DECOMP_FATALERROR; /* probably a bug */ 911 } 912 } 913 bsd_clear(db); 914 explen = ilen = 0; 915 break; 916 } 917 918 if (incode > max_ent + 2 || incode > db->maxmaxcode 919 || (incode > max_ent && oldcode == CLEAR)) { 920 m_freem(mret); 921 if (db->debug) { 922 printf("bsd_decomp%d: bad code 0x%x oldcode=0x%x ", 923 db->unit, incode, oldcode); 924 printf("max_ent=0x%x explen=%d seqno=%d\n", 925 max_ent, explen, db->seqno); 926 } 927 return DECOMP_FATALERROR; /* probably a bug */ 928 } 929 930 /* Special case for KwKwK string. */ 931 if (incode > max_ent) { 932 finchar = oldcode; 933 extra = 1; 934 } else { 935 finchar = incode; 936 extra = 0; 937 } 938 939 codelen = db->lens[finchar]; 940 explen += codelen + extra; 941 if (explen > db->mru + 1) { 942 m_freem(mret); 943 if (db->debug) { 944 printf("bsd_decomp%d: ran out of mru\n", db->unit); 945 #ifdef DEBUG 946 while ((cmp = cmp->m_next) != NULL) 947 len += cmp->m_len; 948 printf(" len=%d, finchar=0x%x, codelen=%d, explen=%d\n", 949 len, finchar, codelen, explen); 950 #endif 951 } 952 return DECOMP_FATALERROR; 953 } 954 955 /* 956 * For simplicity, the decoded characters go in a single mbuf, 957 * so we allocate a single extra cluster mbuf if necessary. 958 */ 959 if ((space -= codelen + extra) < 0) { 960 dmp->m_len = wptr - mtod(dmp, u_char *); 961 MGET(m, M_DONTWAIT, MT_DATA); 962 if (m == NULL) { 963 m_freem(mret); 964 return DECOMP_ERROR; 965 } 966 m->m_len = 0; 967 m->m_next = NULL; 968 dmp->m_next = m; 969 MCLGET(m, M_DONTWAIT); 970 space = M_TRAILINGSPACE(m) - (codelen + extra); 971 if (space < 0) { 972 /* now that's what I call *compression*. */ 973 m_freem(mret); 974 return DECOMP_ERROR; 975 } 976 dmp = m; 977 wptr = mtod(dmp, u_char *); 978 } 979 980 /* 981 * Decode this code and install it in the decompressed buffer. 982 */ 983 p = (wptr += codelen); 984 while (finchar > LAST) { 985 dictp = &db->dict[db->dict[finchar].cptr]; 986 #ifdef DEBUG 987 if (--codelen <= 0 || dictp->codem1 != finchar-1) 988 goto bad; 989 #endif 990 *--p = dictp->f.hs.suffix; 991 finchar = dictp->f.hs.prefix; 992 } 993 *--p = finchar; 994 995 #ifdef DEBUG 996 if (--codelen != 0) 997 printf("bsd_decomp%d: short by %d after code 0x%x, max_ent=0x%x\n", 998 db->unit, codelen, incode, max_ent); 999 #endif 1000 1001 if (extra) /* the KwKwK case again */ 1002 *wptr++ = finchar; 1003 1004 /* 1005 * If not first code in a packet, and 1006 * if not out of code space, then allocate a new code. 1007 * 1008 * Keep the hash table correct so it can be used 1009 * with uncompressed packets. 1010 */ 1011 if (oldcode != CLEAR && max_ent < db->maxmaxcode) { 1012 struct bsd_dict *dictp2; 1013 uint32_t fcode; 1014 uint32_t hval, disp; 1015 1016 fcode = BSD_KEY(oldcode,finchar); 1017 hval = BSD_HASH(oldcode,finchar,db->hshift); 1018 dictp = &db->dict[hval]; 1019 1020 /* look for a free hash table entry */ 1021 if (dictp->codem1 < max_ent) { 1022 disp = (hval == 0) ? 1 : hval; 1023 do { 1024 hval += disp; 1025 if (hval >= db->hsize) 1026 hval -= db->hsize; 1027 dictp = &db->dict[hval]; 1028 } while (dictp->codem1 < max_ent); 1029 } 1030 1031 /* 1032 * Invalidate previous hash table entry 1033 * assigned this code, and then take it over 1034 */ 1035 dictp2 = &db->dict[max_ent+1]; 1036 if (db->dict[dictp2->cptr].codem1 == max_ent) { 1037 db->dict[dictp2->cptr].codem1 = BADCODEM1; 1038 } 1039 dictp2->cptr = hval; 1040 dictp->codem1 = max_ent; 1041 dictp->f.fcode = fcode; 1042 1043 db->max_ent = ++max_ent; 1044 db->lens[max_ent] = db->lens[oldcode]+1; 1045 1046 /* Expand code size if needed. */ 1047 if (max_ent >= MAXCODE(n_bits) && max_ent < db->maxmaxcode) { 1048 db->n_bits = ++n_bits; 1049 tgtbitno = 32-n_bits; 1050 } 1051 } 1052 oldcode = incode; 1053 } 1054 dmp->m_len = wptr - mtod(dmp, u_char *); 1055 1056 /* 1057 * Keep the checkpoint right so that incompressible packets 1058 * clear the dictionary at the right times. 1059 */ 1060 db->bytes_out += ilen; 1061 db->in_count += explen; 1062 if (bsd_check(db) && db->debug) { 1063 printf("bsd_decomp%d: peer should have cleared dictionary\n", 1064 db->unit); 1065 } 1066 1067 ++db->comp_count; 1068 db->comp_bytes += ilen + BSD_OVHD; 1069 ++db->uncomp_count; 1070 db->uncomp_bytes += explen; 1071 1072 *dmpp = mret; 1073 return DECOMP_OK; 1074 1075 #ifdef DEBUG 1076 bad: 1077 if (codelen <= 0) { 1078 printf("bsd_decomp%d: fell off end of chain ", db->unit); 1079 printf("0x%x at 0x%x by 0x%x, max_ent=0x%x\n", 1080 incode, finchar, db->dict[finchar].cptr, max_ent); 1081 } else if (dictp->codem1 != finchar-1) { 1082 printf("bsd_decomp%d: bad code chain 0x%x finchar=0x%x ", 1083 db->unit, incode, finchar); 1084 printf("oldcode=0x%x cptr=0x%x codem1=0x%x\n", oldcode, 1085 db->dict[finchar].cptr, dictp->codem1); 1086 } 1087 m_freem(mret); 1088 return DECOMP_FATALERROR; 1089 #endif /* DEBUG */ 1090 } 1091 #endif /* DO_BSD_COMPRESS */ 1092