2 * Copyright (c) 2000, Boris Popov
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 3. All advertising materials mentioning features or use of this software
14 * must display the following acknowledgement:
15 * This product includes software developed by Boris Popov.
16 * 4. Neither the name of the author nor the names of any co-contributors
17 * may be used to endorse or promote products derived from this software
18 * without specific prior written permission.
20 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
21 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
24 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32 * $Id: mbuf.c,v 1.6 2001/02/24 15:56:04 bp Exp $
35 #include <sys/cdefs.h>
36 __FBSDID("$FreeBSD$");
38 #include <sys/types.h>
39 #include <sys/endian.h>
40 #include <arpa/inet.h>
47 #include <netsmb/smb_lib.h>
49 #define MBERROR(format, args...) printf("%s(%d): "format, __FUNCTION__ , \
53 m_get(size_t len, struct mbuf **mpp)
60 m = malloc(M_BASESIZE + len);
63 bzero(m, M_BASESIZE + len);
71 m_free(struct mbuf *m)
77 m_freem(struct mbuf *m0)
89 m_totlen(struct mbuf *m0)
102 m_lineup(struct mbuf *m0, struct mbuf **mpp)
109 if (m0->m_next == NULL) {
113 if ((error = m_get(m_totlen(m0), &nm)) != 0)
115 dp = mtod(nm, char *);
118 bcopy(m0->m_data, dp, len);
129 mb_init(struct mbdata *mbp, size_t size)
134 if ((error = m_get(size, &m)) != 0)
136 return mb_initm(mbp, m);
140 mb_initm(struct mbdata *mbp, struct mbuf *m)
142 bzero(mbp, sizeof(*mbp));
143 mbp->mb_top = mbp->mb_cur = m;
144 mbp->mb_pos = mtod(m, char *);
149 mb_done(struct mbdata *mbp)
152 m_freem(mbp->mb_top);
160 mb_fixhdr(struct mbdata *mbp)
162 struct mbuf *m = mbp->mb_top;
169 mbp->mb_top->m_pkthdr.len = len;
174 m_getm(struct mbuf *top, size_t len, struct mbuf **mpp)
179 for (mp = top; ; mp = mp->m_next) {
180 len -= M_TRAILINGSPACE(mp);
181 if (mp->m_next == NULL)
186 if ((error = m_get(len, &m)) != 0)
195 * Routines to put data in a buffer
197 #define MB_PUT(t) int error; t *p; \
198 if ((error = mb_fit(mbp, sizeof(t), (char**)&p)) != 0) \
202 * Check if object of size 'size' fit to the current position and
203 * allocate new mbuf if not. Advance pointers and increase length of mbuf(s).
204 * Return pointer to the object placeholder or NULL if any error occured.
207 mb_fit(struct mbdata *mbp, size_t size, char **pp)
213 if (M_TRAILINGSPACE(m) < (int)size) {
214 if ((error = m_get(size, &mn)) != 0)
216 mbp->mb_pos = mtod(mn, char *);
217 mbp->mb_cur = m->m_next = mn;
223 mbp->mb_count += size;
228 mb_put_uint8(struct mbdata *mbp, u_int8_t x)
236 mb_put_uint16be(struct mbdata *mbp, u_int16_t x)
244 mb_put_uint16le(struct mbdata *mbp, u_int16_t x)
252 mb_put_uint32be(struct mbdata *mbp, u_int32_t x)
260 mb_put_uint32le(struct mbdata *mbp, u_int32_t x)
268 mb_put_int64be(struct mbdata *mbp, int64_t x)
276 mb_put_int64le(struct mbdata *mbp, int64_t x)
284 mb_put_mem(struct mbdata *mbp, const char *source, size_t size)
294 if ((error = m_getm(m, size, &m)) != 0)
297 cplen = M_TRAILINGSPACE(m);
304 dst = mtod(m, char *) + m->m_len;
306 bcopy(source, dst, cplen);
312 mbp->mb_count += cplen;
314 mbp->mb_pos = mtod(m, char *) + m->m_len;
320 mb_put_mbuf(struct mbdata *mbp, struct mbuf *m)
322 mbp->mb_cur->m_next = m;
324 mbp->mb_count += m->m_len;
325 if (m->m_next == NULL)
329 mbp->mb_pos = mtod(m, char *) + m->m_len;
335 mb_put_pstring(struct mbdata *mbp, const char *s)
337 int error, len = strlen(s);
342 if ((error = mb_put_uint8(mbp, len)) != 0)
344 return mb_put_mem(mbp, s, len);
348 * Routines for fetching data from an mbuf chain
350 #define mb_left(m,p) (mtod(m, char *) + (m)->m_len - (p))
353 mb_get_uint8(struct mbdata *mbp, u_int8_t *x)
355 return mb_get_mem(mbp, x, 1);
359 mb_get_uint16(struct mbdata *mbp, u_int16_t *x)
361 return mb_get_mem(mbp, (char *)x, 2);
365 mb_get_uint16le(struct mbdata *mbp, u_int16_t *x)
368 int error = mb_get_uint16(mbp, &v);
375 mb_get_uint16be(struct mbdata *mbp, u_int16_t *x) {
377 int error = mb_get_uint16(mbp, &v);
384 mb_get_uint32(struct mbdata *mbp, u_int32_t *x)
386 return mb_get_mem(mbp, (char *)x, 4);
390 mb_get_uint32be(struct mbdata *mbp, u_int32_t *x)
395 error = mb_get_uint32(mbp, &v);
401 mb_get_uint32le(struct mbdata *mbp, u_int32_t *x)
406 error = mb_get_uint32(mbp, &v);
412 mb_get_int64(struct mbdata *mbp, int64_t *x)
414 return mb_get_mem(mbp, (char *)x, 8);
418 mb_get_int64be(struct mbdata *mbp, int64_t *x)
423 error = mb_get_int64(mbp, &v);
429 mb_get_int64le(struct mbdata *mbp, int64_t *x)
434 error = mb_get_int64(mbp, &v);
440 mb_get_mem(struct mbdata *mbp, char * target, size_t size)
442 struct mbuf *m = mbp->mb_cur;
447 MBERROR("incomplete copy\n");
450 count = mb_left(m, mbp->mb_pos);
452 mbp->mb_cur = m = m->m_next;
454 mbp->mb_pos = mtod(m, char *);
462 *target++ = *mbp->mb_pos;
464 bcopy(mbp->mb_pos, target, count);
468 mbp->mb_pos += count;