/* $NetBSD: ipcomp_core.c,v 1.18 2001/11/13 00:57:03 lukem Exp $ */ /* $KAME: ipcomp_core.c,v 1.25 2001/07/26 06:53:17 jinmei Exp $ */ /* * Copyright (C) 1999 WIDE Project. * All rights reserved. * * 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 project 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 PROJECT 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 PROJECT 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. */ /* * RFC2393 IP payload compression protocol (IPComp). */ #include __KERNEL_RCSID(0, "$NetBSD: ipcomp_core.c,v 1.18 2001/11/13 00:57:03 lukem Exp $"); #include "opt_inet.h" #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include static void *deflate_alloc __P((void *, u_int, u_int)); static void deflate_free __P((void *, void *)); static int deflate_common __P((struct mbuf *, struct mbuf *, size_t *, int)); static int deflate_compress __P((struct mbuf *, struct mbuf *, size_t *)); static int deflate_decompress __P((struct mbuf *, struct mbuf *, size_t *)); /* * We need to use default window size (2^15 = 32Kbytes as of writing) for * inbound case. Otherwise we get interop problem. * Use negative value to avoid Adler32 checksum. This is an undocumented * feature in zlib (see ipsec wg mailing list archive in January 2000). */ static int deflate_policy = Z_DEFAULT_COMPRESSION; static int deflate_window_out = -12; static const int deflate_window_in = -1 * MAX_WBITS; /* don't change it */ static int deflate_memlevel = MAX_MEM_LEVEL; static const struct ipcomp_algorithm ipcomp_algorithms[] = { { deflate_compress, deflate_decompress, 90 }, }; const struct ipcomp_algorithm * ipcomp_algorithm_lookup(idx) int idx; { if (idx == SADB_X_CALG_DEFLATE) return &ipcomp_algorithms[0]; return NULL; } static void * deflate_alloc(aux, items, siz) void *aux; u_int items; u_int siz; { void *ptr; ptr = malloc(items * siz, M_TEMP, M_NOWAIT); return ptr; } static void deflate_free(aux, ptr) void *aux; void *ptr; { free(ptr, M_TEMP); } static int deflate_common(m, md, lenp, mode) struct mbuf *m; struct mbuf *md; size_t *lenp; int mode; /* 0: compress 1: decompress */ { struct mbuf *mprev; struct mbuf *p; struct mbuf *n = NULL, *n0 = NULL, **np; z_stream zs; int error = 0; int zerror; size_t offset; #define MOREBLOCK() \ do { \ /* keep the reply buffer into our chain */ \ if (n) { \ n->m_len = zs.total_out - offset; \ offset = zs.total_out; \ *np = n; \ np = &n->m_next; \ n = NULL; \ } \ \ /* get a fresh reply buffer */ \ MGET(n, M_DONTWAIT, MT_DATA); \ if (n) { \ MCLGET(n, M_DONTWAIT); \ } \ if (!n) { \ error = ENOBUFS; \ goto fail; \ } \ n->m_len = 0; \ n->m_len = M_TRAILINGSPACE(n); \ n->m_next = NULL; \ /* \ * if this is the first reply buffer, reserve \ * region for ipcomp header. \ */ \ if (*np == NULL) { \ n->m_len -= sizeof(struct ipcomp); \ n->m_data += sizeof(struct ipcomp); \ } \ \ zs.next_out = mtod(n, u_int8_t *); \ zs.avail_out = n->m_len; \ } while (0) for (mprev = m; mprev && mprev->m_next != md; mprev = mprev->m_next) ; if (!mprev) panic("md is not in m in deflate_common"); bzero(&zs, sizeof(zs)); zs.zalloc = deflate_alloc; zs.zfree = deflate_free; zerror = mode ? inflateInit2(&zs, deflate_window_in) : deflateInit2(&zs, deflate_policy, Z_DEFLATED, deflate_window_out, deflate_memlevel, Z_DEFAULT_STRATEGY); if (zerror != Z_OK) { error = ENOBUFS; goto fail; } n0 = n = NULL; np = &n0; offset = 0; zerror = 0; p = md; while (p && p->m_len == 0) { p = p->m_next; } /* input stream and output stream are available */ while (p && zs.avail_in == 0) { /* get input buffer */ if (p && zs.avail_in == 0) { zs.next_in = mtod(p, u_int8_t *); zs.avail_in = p->m_len; p = p->m_next; while (p && p->m_len == 0) { p = p->m_next; } } /* get output buffer */ if (zs.next_out == NULL || zs.avail_out == 0) { MOREBLOCK(); } zerror = mode ? inflate(&zs, Z_NO_FLUSH) : deflate(&zs, Z_NO_FLUSH); if (zerror == Z_STREAM_END) ; /* once more. */ else if (zerror == Z_OK) { /* inflate: Z_OK can indicate the end of decode */ if (mode && !p && zs.avail_out != 0) goto terminate; else ; /* once more. */ } else { if (zs.msg) { ipseclog((LOG_ERR, "ipcomp_%scompress: " "%sflate(Z_NO_FLUSH): %s\n", mode ? "de" : "", mode ? "in" : "de", zs.msg)); } else { ipseclog((LOG_ERR, "ipcomp_%scompress: " "%sflate(Z_NO_FLUSH): unknown error (%d)\n", mode ? "de" : "", mode ? "in" : "de", zerror)); } mode ? inflateEnd(&zs) : deflateEnd(&zs); error = EINVAL; goto fail; } } if (zerror == Z_STREAM_END) goto terminate; /* termination */ while (1) { /* get output buffer */ if (zs.next_out == NULL || zs.avail_out == 0) { MOREBLOCK(); } zerror = mode ? inflate(&zs, Z_FINISH) : deflate(&zs, Z_FINISH); if (zerror == Z_STREAM_END) break; else if (zerror == Z_OK) ; /* once more. */ else { if (zs.msg) { ipseclog((LOG_ERR, "ipcomp_%scompress: " "%sflate(Z_FINISH): %s\n", mode ? "de" : "", mode ? "in" : "de", zs.msg)); } else { ipseclog((LOG_ERR, "ipcomp_%scompress: " "%sflate(Z_FINISH): unknown error (%d)\n", mode ? "de" : "", mode ? "in" : "de", zerror)); } mode ? inflateEnd(&zs) : deflateEnd(&zs); error = EINVAL; goto fail; } } terminate: zerror = mode ? inflateEnd(&zs) : deflateEnd(&zs); if (zerror != Z_OK) { if (zs.msg) { ipseclog((LOG_ERR, "ipcomp_%scompress: " "%sflateEnd: %s\n", mode ? "de" : "", mode ? "in" : "de", zs.msg)); } else { ipseclog((LOG_ERR, "ipcomp_%scompress: " "%sflateEnd: unknown error (%d)\n", mode ? "de" : "", mode ? "in" : "de", zerror)); } error = EINVAL; goto fail; } /* keep the final reply buffer into our chain */ if (n) { n->m_len = zs.total_out - offset; offset = zs.total_out; *np = n; np = &n->m_next; n = NULL; } /* switch the mbuf to the new one */ mprev->m_next = n0; m_freem(md); *lenp = zs.total_out; return 0; fail: if (m) m_freem(m); if (n) m_freem(n); if (n0) m_freem(n0); return error; #undef MOREBLOCK } static int deflate_compress(m, md, lenp) struct mbuf *m; struct mbuf *md; size_t *lenp; { if (!m) panic("m == NULL in deflate_compress"); if (!md) panic("md == NULL in deflate_compress"); if (!lenp) panic("lenp == NULL in deflate_compress"); return deflate_common(m, md, lenp, 0); } static int deflate_decompress(m, md, lenp) struct mbuf *m; struct mbuf *md; size_t *lenp; { if (!m) panic("m == NULL in deflate_decompress"); if (!md) panic("md == NULL in deflate_decompress"); if (!lenp) panic("lenp == NULL in deflate_decompress"); return deflate_common(m, md, lenp, 1); }