1dfdcada3SDoug Rabson /*- 2dfdcada3SDoug Rabson * Copyright (c) 2008 Isilon Inc http://www.isilon.com/ 3dfdcada3SDoug Rabson * Authors: Doug Rabson <dfr@rabson.org> 4dfdcada3SDoug Rabson * Developed with Red Inc: Alfred Perlstein <alfred@freebsd.org> 5dfdcada3SDoug Rabson * 6dfdcada3SDoug Rabson * Redistribution and use in source and binary forms, with or without 7dfdcada3SDoug Rabson * modification, are permitted provided that the following conditions 8dfdcada3SDoug Rabson * are met: 9dfdcada3SDoug Rabson * 1. Redistributions of source code must retain the above copyright 10dfdcada3SDoug Rabson * notice, this list of conditions and the following disclaimer. 11dfdcada3SDoug Rabson * 2. Redistributions in binary form must reproduce the above copyright 12dfdcada3SDoug Rabson * notice, this list of conditions and the following disclaimer in the 13dfdcada3SDoug Rabson * documentation and/or other materials provided with the distribution. 14dfdcada3SDoug Rabson * 15dfdcada3SDoug Rabson * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 16dfdcada3SDoug Rabson * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 17dfdcada3SDoug Rabson * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 18dfdcada3SDoug Rabson * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 19dfdcada3SDoug Rabson * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 20dfdcada3SDoug Rabson * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 21dfdcada3SDoug Rabson * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 22dfdcada3SDoug Rabson * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 23dfdcada3SDoug Rabson * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 24dfdcada3SDoug Rabson * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 25dfdcada3SDoug Rabson * SUCH DAMAGE. 26dfdcada3SDoug Rabson */ 27dfdcada3SDoug Rabson 28dfdcada3SDoug Rabson #include <sys/cdefs.h> 29dfdcada3SDoug Rabson __FBSDID("$FreeBSD$"); 30dfdcada3SDoug Rabson 31dfdcada3SDoug Rabson #include <sys/param.h> 32dfdcada3SDoug Rabson #include <sys/systm.h> 33dfdcada3SDoug Rabson #include <sys/malloc.h> 34dfdcada3SDoug Rabson #include <sys/mbuf.h> 35dfdcada3SDoug Rabson 36dfdcada3SDoug Rabson #include <rpc/types.h> 37dfdcada3SDoug Rabson #include <rpc/xdr.h> 38dfdcada3SDoug Rabson 39dfdcada3SDoug Rabson static void xdrmbuf_destroy(XDR *); 40dfdcada3SDoug Rabson static bool_t xdrmbuf_getlong(XDR *, long *); 41dfdcada3SDoug Rabson static bool_t xdrmbuf_putlong(XDR *, const long *); 42dfdcada3SDoug Rabson static bool_t xdrmbuf_getbytes(XDR *, char *, u_int); 43dfdcada3SDoug Rabson static bool_t xdrmbuf_putbytes(XDR *, const char *, u_int); 44dfdcada3SDoug Rabson /* XXX: w/64-bit pointers, u_int not enough! */ 45dfdcada3SDoug Rabson static u_int xdrmbuf_getpos(XDR *); 46dfdcada3SDoug Rabson static bool_t xdrmbuf_setpos(XDR *, u_int); 47dfdcada3SDoug Rabson static int32_t *xdrmbuf_inline(XDR *, u_int); 48dfdcada3SDoug Rabson 49dfdcada3SDoug Rabson static const struct xdr_ops xdrmbuf_ops = { 50dfdcada3SDoug Rabson xdrmbuf_getlong, 51dfdcada3SDoug Rabson xdrmbuf_putlong, 52dfdcada3SDoug Rabson xdrmbuf_getbytes, 53dfdcada3SDoug Rabson xdrmbuf_putbytes, 54dfdcada3SDoug Rabson xdrmbuf_getpos, 55dfdcada3SDoug Rabson xdrmbuf_setpos, 56dfdcada3SDoug Rabson xdrmbuf_inline, 57dfdcada3SDoug Rabson xdrmbuf_destroy 58dfdcada3SDoug Rabson }; 59dfdcada3SDoug Rabson 60dfdcada3SDoug Rabson /* 61dfdcada3SDoug Rabson * The procedure xdrmbuf_create initializes a stream descriptor for a 62dfdcada3SDoug Rabson * mbuf. 63dfdcada3SDoug Rabson */ 64dfdcada3SDoug Rabson void 65dfdcada3SDoug Rabson xdrmbuf_create(XDR *xdrs, struct mbuf *m, enum xdr_op op) 66dfdcada3SDoug Rabson { 67dfdcada3SDoug Rabson 68dfdcada3SDoug Rabson xdrs->x_op = op; 69dfdcada3SDoug Rabson xdrs->x_ops = &xdrmbuf_ops; 70dfdcada3SDoug Rabson xdrs->x_base = (char *) m; 71dfdcada3SDoug Rabson if (op == XDR_ENCODE) { 72dfdcada3SDoug Rabson m = m_last(m); 73dfdcada3SDoug Rabson xdrs->x_private = m; 74dfdcada3SDoug Rabson xdrs->x_handy = m->m_len; 75dfdcada3SDoug Rabson } else { 76dfdcada3SDoug Rabson xdrs->x_private = m; 77dfdcada3SDoug Rabson xdrs->x_handy = 0; 78dfdcada3SDoug Rabson } 79dfdcada3SDoug Rabson } 80dfdcada3SDoug Rabson 81a9148abdSDoug Rabson void 82a9148abdSDoug Rabson xdrmbuf_append(XDR *xdrs, struct mbuf *madd) 83a9148abdSDoug Rabson { 84a9148abdSDoug Rabson struct mbuf *m; 85a9148abdSDoug Rabson 86a9148abdSDoug Rabson KASSERT(xdrs->x_ops == &xdrmbuf_ops && xdrs->x_op == XDR_ENCODE, 87a9148abdSDoug Rabson ("xdrmbuf_append: invalid XDR stream")); 88a9148abdSDoug Rabson 89a9148abdSDoug Rabson if (m_length(madd, NULL) == 0) { 90a9148abdSDoug Rabson m_freem(madd); 91a9148abdSDoug Rabson return; 92a9148abdSDoug Rabson } 93a9148abdSDoug Rabson 94a9148abdSDoug Rabson m = (struct mbuf *) xdrs->x_private; 95a9148abdSDoug Rabson m->m_next = madd; 96a9148abdSDoug Rabson 97a9148abdSDoug Rabson m = m_last(madd); 98a9148abdSDoug Rabson xdrs->x_private = m; 99a9148abdSDoug Rabson xdrs->x_handy = m->m_len; 100a9148abdSDoug Rabson } 101a9148abdSDoug Rabson 102a9148abdSDoug Rabson struct mbuf * 103a9148abdSDoug Rabson xdrmbuf_getall(XDR *xdrs) 104a9148abdSDoug Rabson { 105a9148abdSDoug Rabson struct mbuf *m0, *m; 106a9148abdSDoug Rabson 107a9148abdSDoug Rabson KASSERT(xdrs->x_ops == &xdrmbuf_ops && xdrs->x_op == XDR_DECODE, 108a9148abdSDoug Rabson ("xdrmbuf_append: invalid XDR stream")); 109a9148abdSDoug Rabson 110a9148abdSDoug Rabson m0 = (struct mbuf *) xdrs->x_base; 111a9148abdSDoug Rabson m = (struct mbuf *) xdrs->x_private; 112a9148abdSDoug Rabson if (m0 != m) { 113a9148abdSDoug Rabson while (m0->m_next != m) 114a9148abdSDoug Rabson m0 = m0->m_next; 115a9148abdSDoug Rabson m0->m_next = NULL; 116a9148abdSDoug Rabson xdrs->x_private = NULL; 117a9148abdSDoug Rabson } else { 118a9148abdSDoug Rabson xdrs->x_base = NULL; 119a9148abdSDoug Rabson xdrs->x_private = NULL; 120a9148abdSDoug Rabson } 121a9148abdSDoug Rabson 122e3c56574SDoug Rabson if (m) 123a9148abdSDoug Rabson m_adj(m, xdrs->x_handy); 124e3c56574SDoug Rabson else 125e3c56574SDoug Rabson MGET(m, M_WAITOK, MT_DATA); 126a9148abdSDoug Rabson return (m); 127a9148abdSDoug Rabson } 128a9148abdSDoug Rabson 129dfdcada3SDoug Rabson static void 130dfdcada3SDoug Rabson xdrmbuf_destroy(XDR *xdrs) 131dfdcada3SDoug Rabson { 132dfdcada3SDoug Rabson 133dfdcada3SDoug Rabson if (xdrs->x_op == XDR_DECODE && xdrs->x_base) { 134dfdcada3SDoug Rabson m_freem((struct mbuf *) xdrs->x_base); 135dfdcada3SDoug Rabson xdrs->x_base = NULL; 136dfdcada3SDoug Rabson xdrs->x_private = NULL; 137dfdcada3SDoug Rabson } 138dfdcada3SDoug Rabson } 139dfdcada3SDoug Rabson 140dfdcada3SDoug Rabson static bool_t 141dfdcada3SDoug Rabson xdrmbuf_getlong(XDR *xdrs, long *lp) 142dfdcada3SDoug Rabson { 143a9148abdSDoug Rabson int32_t *p; 144dfdcada3SDoug Rabson int32_t t; 145dfdcada3SDoug Rabson 146a9148abdSDoug Rabson p = xdrmbuf_inline(xdrs, sizeof(int32_t)); 147a9148abdSDoug Rabson if (p) { 148a9148abdSDoug Rabson t = *p; 149a9148abdSDoug Rabson } else { 150dfdcada3SDoug Rabson xdrmbuf_getbytes(xdrs, (char *) &t, sizeof(int32_t)); 151a9148abdSDoug Rabson } 152a9148abdSDoug Rabson 153dfdcada3SDoug Rabson *lp = ntohl(t); 154dfdcada3SDoug Rabson return (TRUE); 155dfdcada3SDoug Rabson } 156dfdcada3SDoug Rabson 157dfdcada3SDoug Rabson static bool_t 158dfdcada3SDoug Rabson xdrmbuf_putlong(xdrs, lp) 159dfdcada3SDoug Rabson XDR *xdrs; 160dfdcada3SDoug Rabson const long *lp; 161dfdcada3SDoug Rabson { 162a9148abdSDoug Rabson int32_t *p; 163dfdcada3SDoug Rabson int32_t t = htonl(*lp); 164dfdcada3SDoug Rabson 165a9148abdSDoug Rabson p = xdrmbuf_inline(xdrs, sizeof(int32_t)); 166a9148abdSDoug Rabson if (p) { 167a9148abdSDoug Rabson *p = t; 168dfdcada3SDoug Rabson return (TRUE); 169a9148abdSDoug Rabson } else { 170a9148abdSDoug Rabson return (xdrmbuf_putbytes(xdrs, (char *) &t, sizeof(int32_t))); 171a9148abdSDoug Rabson } 172dfdcada3SDoug Rabson } 173dfdcada3SDoug Rabson 174dfdcada3SDoug Rabson static bool_t 175dfdcada3SDoug Rabson xdrmbuf_getbytes(XDR *xdrs, char *addr, u_int len) 176dfdcada3SDoug Rabson { 177dfdcada3SDoug Rabson struct mbuf *m = (struct mbuf *) xdrs->x_private; 178dfdcada3SDoug Rabson size_t sz; 179dfdcada3SDoug Rabson 180dfdcada3SDoug Rabson while (len > 0) { 181dfdcada3SDoug Rabson /* 182dfdcada3SDoug Rabson * Make sure we haven't hit the end. 183dfdcada3SDoug Rabson */ 184dfdcada3SDoug Rabson if (!m) { 185dfdcada3SDoug Rabson return (FALSE); 186dfdcada3SDoug Rabson } 187dfdcada3SDoug Rabson 188dfdcada3SDoug Rabson /* 189dfdcada3SDoug Rabson * See how much we can get from this mbuf. 190dfdcada3SDoug Rabson */ 191dfdcada3SDoug Rabson sz = m->m_len - xdrs->x_handy; 192dfdcada3SDoug Rabson if (sz > len) 193dfdcada3SDoug Rabson sz = len; 194a9148abdSDoug Rabson bcopy(mtod(m, const char *) + xdrs->x_handy, addr, sz); 195dfdcada3SDoug Rabson 196dfdcada3SDoug Rabson addr += sz; 197dfdcada3SDoug Rabson xdrs->x_handy += sz; 198dfdcada3SDoug Rabson len -= sz; 199dfdcada3SDoug Rabson 200dfdcada3SDoug Rabson if (xdrs->x_handy == m->m_len) { 201dfdcada3SDoug Rabson m = m->m_next; 202dfdcada3SDoug Rabson xdrs->x_private = (void *) m; 203dfdcada3SDoug Rabson xdrs->x_handy = 0; 204dfdcada3SDoug Rabson } 205dfdcada3SDoug Rabson } 206dfdcada3SDoug Rabson 207dfdcada3SDoug Rabson return (TRUE); 208dfdcada3SDoug Rabson } 209dfdcada3SDoug Rabson 210dfdcada3SDoug Rabson static bool_t 211dfdcada3SDoug Rabson xdrmbuf_putbytes(XDR *xdrs, const char *addr, u_int len) 212dfdcada3SDoug Rabson { 213dfdcada3SDoug Rabson struct mbuf *m = (struct mbuf *) xdrs->x_private; 214dfdcada3SDoug Rabson struct mbuf *n; 215dfdcada3SDoug Rabson size_t sz; 216dfdcada3SDoug Rabson 217dfdcada3SDoug Rabson while (len > 0) { 218dfdcada3SDoug Rabson sz = M_TRAILINGSPACE(m) + (m->m_len - xdrs->x_handy); 219dfdcada3SDoug Rabson if (sz > len) 220dfdcada3SDoug Rabson sz = len; 221a9148abdSDoug Rabson bcopy(addr, mtod(m, char *) + xdrs->x_handy, sz); 222dfdcada3SDoug Rabson addr += sz; 223dfdcada3SDoug Rabson xdrs->x_handy += sz; 224dfdcada3SDoug Rabson if (xdrs->x_handy > m->m_len) 225dfdcada3SDoug Rabson m->m_len = xdrs->x_handy; 226dfdcada3SDoug Rabson len -= sz; 227dfdcada3SDoug Rabson 228dfdcada3SDoug Rabson if (xdrs->x_handy == m->m_len && M_TRAILINGSPACE(m) == 0) { 229dfdcada3SDoug Rabson if (!m->m_next) { 230dfdcada3SDoug Rabson MGET(n, M_TRYWAIT, m->m_type); 231a9148abdSDoug Rabson if (m->m_flags & M_EXT) 232a9148abdSDoug Rabson MCLGET(n, M_TRYWAIT); 233dfdcada3SDoug Rabson m->m_next = n; 234dfdcada3SDoug Rabson } 235dfdcada3SDoug Rabson m = m->m_next; 236dfdcada3SDoug Rabson xdrs->x_private = (void *) m; 237dfdcada3SDoug Rabson xdrs->x_handy = 0; 238dfdcada3SDoug Rabson } 239dfdcada3SDoug Rabson } 240dfdcada3SDoug Rabson 241dfdcada3SDoug Rabson return (TRUE); 242dfdcada3SDoug Rabson } 243dfdcada3SDoug Rabson 244dfdcada3SDoug Rabson static u_int 245dfdcada3SDoug Rabson xdrmbuf_getpos(XDR *xdrs) 246dfdcada3SDoug Rabson { 247dfdcada3SDoug Rabson struct mbuf *m0 = (struct mbuf *) xdrs->x_base; 248dfdcada3SDoug Rabson struct mbuf *m = (struct mbuf *) xdrs->x_private; 249dfdcada3SDoug Rabson u_int pos = 0; 250dfdcada3SDoug Rabson 251dfdcada3SDoug Rabson while (m0 && m0 != m) { 252dfdcada3SDoug Rabson pos += m0->m_len; 253dfdcada3SDoug Rabson m0 = m0->m_next; 254dfdcada3SDoug Rabson } 255dfdcada3SDoug Rabson KASSERT(m0, ("Corrupted mbuf chain")); 256dfdcada3SDoug Rabson 257dfdcada3SDoug Rabson return (pos + xdrs->x_handy); 258dfdcada3SDoug Rabson } 259dfdcada3SDoug Rabson 260dfdcada3SDoug Rabson static bool_t 261dfdcada3SDoug Rabson xdrmbuf_setpos(XDR *xdrs, u_int pos) 262dfdcada3SDoug Rabson { 263dfdcada3SDoug Rabson struct mbuf *m = (struct mbuf *) xdrs->x_base; 264dfdcada3SDoug Rabson 265dfdcada3SDoug Rabson while (m && pos > m->m_len) { 266dfdcada3SDoug Rabson pos -= m->m_len; 267dfdcada3SDoug Rabson m = m->m_next; 268dfdcada3SDoug Rabson } 269dfdcada3SDoug Rabson KASSERT(m, ("Corrupted mbuf chain")); 270dfdcada3SDoug Rabson 271dfdcada3SDoug Rabson xdrs->x_private = (void *) m; 272dfdcada3SDoug Rabson xdrs->x_handy = pos; 273dfdcada3SDoug Rabson 274dfdcada3SDoug Rabson return (TRUE); 275dfdcada3SDoug Rabson } 276dfdcada3SDoug Rabson 277dfdcada3SDoug Rabson static int32_t * 278dfdcada3SDoug Rabson xdrmbuf_inline(XDR *xdrs, u_int len) 279dfdcada3SDoug Rabson { 280dfdcada3SDoug Rabson struct mbuf *m = (struct mbuf *) xdrs->x_private; 281dfdcada3SDoug Rabson size_t available; 282dfdcada3SDoug Rabson char *p; 283dfdcada3SDoug Rabson 284dfdcada3SDoug Rabson if (xdrs->x_op == XDR_ENCODE) { 285dfdcada3SDoug Rabson available = M_TRAILINGSPACE(m) + (m->m_len - xdrs->x_handy); 286dfdcada3SDoug Rabson } else { 287dfdcada3SDoug Rabson available = m->m_len - xdrs->x_handy; 288dfdcada3SDoug Rabson } 289dfdcada3SDoug Rabson 290dfdcada3SDoug Rabson if (available >= len) { 291dfdcada3SDoug Rabson p = mtod(m, char *) + xdrs->x_handy; 292dfdcada3SDoug Rabson if (((uintptr_t) p) & (sizeof(int32_t) - 1)) 293dfdcada3SDoug Rabson return (0); 294dfdcada3SDoug Rabson xdrs->x_handy += len; 295dfdcada3SDoug Rabson if (xdrs->x_handy > m->m_len) 296dfdcada3SDoug Rabson m->m_len = xdrs->x_handy; 297dfdcada3SDoug Rabson return ((int32_t *) p); 298dfdcada3SDoug Rabson } 299dfdcada3SDoug Rabson 300dfdcada3SDoug Rabson return (0); 301dfdcada3SDoug Rabson } 302