3 * $Id: pkbuf.c,v 1.1 2000/06/17 10:39:19 mdw Exp $
5 * Simple packet buffering
7 * (c) 2000 Straylight/Edgeware
10 /*----- Licensing notice --------------------------------------------------*
12 * This file is part of the mLib utilities library.
14 * mLib is free software; you can redistribute it and/or modify
15 * it under the terms of the GNU Library General Public License as
16 * published by the Free Software Foundation; either version 2 of the
17 * License, or (at your option) any later version.
19 * mLib is distributed in the hope that it will be useful,
20 * but WITHOUT ANY WARRANTY; without even the implied warranty of
21 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
22 * GNU Library General Public License for more details.
24 * You should have received a copy of the GNU Library General Public
25 * License along with mLib; if not, write to the Free
26 * Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
30 /*----- Revision history --------------------------------------------------*
33 * Revision 1.1 2000/06/17 10:39:19 mdw
34 * Experimental new support for packet buffering.
38 /*----- Header files ------------------------------------------------------*/
48 /*----- Main code ---------------------------------------------------------*/
50 /* --- @pkbuf_flush@ --- *
52 * Arguments: @pkbuf *pk@ = pointer to buffer block
53 * @octet *p@ = pointer to where to start searching
54 * @size_t len@ = length of new material added
58 * Use: Flushes any complete packets in a packet buffer. New
59 * material is assumed to have been added starting at @p@. If
60 * @p@ is null, then the scan starts at the beginning of the
61 * buffer, and the size of data already in the buffer is used in
64 * It is assumed that the buffer is initially enabled. You
65 * shouldn't be contributing data to a disabled buffer anyway.
66 * However, the buffer handler may at some point disable itself,
67 * and @pkbuf_flush@ can cope with this eventuality. Any
68 * pending data is left at the start of the buffer and can be
69 * flushed out by calling @pkbuf_flush(b, 0, 0)@ if the buffer
73 void pkbuf_flush(pkbuf *pk, octet *p, size_t len)
78 /* --- Initialize variables as necessary --- */
84 l = p + len - pk->buf;
87 /* --- Now grind through any packets which have accumulated --- */
89 while (l > pk->want) {
92 /* --- Pass a packet to the user handler --- */
95 pk->func(p, sz, pk, &keep, pk->p);
97 /* --- Adjust all the pointers for the next packet --- */
103 /* --- Abort here if disabled --- */
105 if (!(pk->f & PKBUF_ENABLE))
109 /* --- Shunt data around in the buffer --- */
111 if (p > pk->buf && l != 0)
112 memmove(pk->buf, p, l);
116 /* --- @pkbuf_close@ --- *
118 * Arguments: @pkbuf *pk@ = pointer to buffer block
122 * Use: Informs the client that no more data is likely to arrive. If
123 * there is a partial packet in the buffer, it is discarded.
126 void pkbuf_close(pkbuf *pk)
129 fprintf(stderr, "*** destroying buffer, closing down\n");
130 x_free(pk->a, pk->buf);
133 if (pk->f & PKBUF_ENABLE)
134 pk->func(0, 0, pk, 0, pk->p);
137 /* --- @pkbuf_free@ --- *
139 * Arguments: @pkbuf *pk@ = pointer to buffer block
140 * @octet **p@ = output pointer to free space
142 * Returns: Free buffer size.
144 * Use: Returns the free portion of a packet buffer. Data can then
145 * be written to this portion, and split out into packets by
146 * calling @pkbuf_flush@. A buffer is allocated if none
150 size_t pkbuf_free(pkbuf *pk, octet **p)
153 fprintf(stderr, "*** allocating new buffer\n");
154 pk->buf = x_alloc(pk->a, pk->sz);
156 *p = pk->buf + pk->len;
157 return (pk->sz - pk->len);
160 /* --- @pkbuf_snarf@ --- *
162 * Arguments: @pkbuf *pk@ = pointer to buffer block
163 * @const void *p@ = pointer to input data buffer
164 * @size_t sz@ = size of data in input buffer
168 * Use: Snarfs the data from the input buffer and spits it out as
169 * packets. This interface ignores the complexities of dealing
170 * with disablement: you should be using @pkbuf_free@ to
171 * contribute data if you want to cope with that.
174 void pkbuf_snarf(pkbuf *pk, const void *p, size_t sz)
177 while (sz && (pk->f & PKBUF_ENABLE)) {
181 bsz = pkbuf_free(pk, &bp);
185 pkbuf_flush(pk, bp, bsz);
191 /* --- @pkbuf_want@ --- *
193 * Arguments: @pkbuf *pk@ = pointer to buffer block
194 * @size_t want@ = how many octets wanted for next packet
198 * Use: Sets the desired size for the next packet to be read. If
199 * it's larger than the current buffer, the buffer is extended.
202 void pkbuf_want(pkbuf *pk, size_t want)
206 do pk->sz <<= 1; while (want < pk->sz);
209 pk->buf = x_realloc(pk->a, pk->buf, pk->sz);
211 x_free(pk->a, pk->buf);
218 /* --- @pkbuf_init@ --- *
220 * Arguments: @pkbuf *pk@ = pointer to buffer block
221 * @void (*func)(octet *b, size_t sz, pkbuf *pk,@
222 * @size_t *keep, void *p)@ =
224 * @void *p@ = argument pointer for @func@
228 * Use: Initializes a packet buffer block. Any packets are passed to
229 * the provided function for handling.
232 void pkbuf_init(pkbuf *pk,
233 void (*func)(octet */*b*/, size_t /*sz*/,
234 pkbuf */*pk*/, size_t */*keep*/, void */*p*/),
240 pk->f = PKBUF_ENABLE;
244 pk->a = arena_global;
247 /* --- @pkbuf_destroy@ --- *
249 * Arguments: @pkbuf *pk@ = pointer to buffer block
253 * Use: Deallocates a line buffer and frees any resources it owned.
256 void pkbuf_destroy(pkbuf *pk)
259 x_free(pk->a, pk->buf);
264 /*----- That's all, folks -------------------------------------------------*/