-/* --- @buf_init@ --- *
- *
- * Arguments: @buf *b@ = pointer to a buffer block
- * @void *p@ = pointer to a buffer
- * @size_t sz@ = size of the buffer
- *
- * Returns: ---
- *
- * Use: Initializes the buffer block appropriately.
- */
-
-void buf_init(buf *b, void *p, size_t sz)
-{
- b->base = b->p = p;
- b->limit = b->p + sz;
- b->f = 0;
-}
-
-/* --- @buf_break@ --- *
- *
- * Arguments: @buf *b@ = pointer to a buffer block
- *
- * Returns: Some negative value.
- *
- * Use: Marks a buffer as broken.
- */
-
-int buf_break(buf *b) { b->f |= BF_BROKEN; return (-1); }
-
-/* --- @buf_flip@ --- *
- *
- * Arguments: @buf *b@ = pointer to a buffer block
- *
- * Returns: ---
- *
- * Use: Flips a buffer so that if you've just been writing to it,
- * you can now read from the bit you've written.
- */
-
-void buf_flip(buf *b)
-{
- b->limit = b->p;
- b->p = b->base;
-}
-
-/* --- @buf_ensure@ --- *
- *
- * Arguments: @buf *b@ = pointer to a buffer block
- * @size_t sz@ = size of data wanted
- *
- * Returns: Zero if it worked, nonzero if there wasn't enough space.
- *
- * Use: Ensures that there are @sz@ bytes still in the buffer.
- */
-
-int buf_ensure(buf *b, size_t sz) { return (BENSURE(b, sz)); }
-
-/* --- @buf_get@ --- *
- *
- * Arguments: @buf *b@ = pointer to a buffer block
- * @size_t sz@ = size of the buffer
- *
- * Returns: Pointer to the place in the buffer.
- *
- * Use: Reserves a space in the buffer of the requested size, and
- * returns its start address.
- */
-
-void *buf_get(buf *b, size_t sz)
-{
- void *p;
- if (BENSURE(b, sz))
- return (0);
- p = BCUR(b);
- BSTEP(b, sz);
- return (p);
-}
-
-/* --- @buf_put@ --- *
- *
- * Arguments: @buf *b@ = pointer to a buffer block
- * @const void *p@ = pointer to a buffer
- * @size_t sz@ = size of the buffer
- *
- * Returns: Zero if it worked, nonzero if there wasn't enough space.
- *
- * Use: Fetches data from some place and puts it in the buffer
- */
-
-int buf_put(buf *b, const void *p, size_t sz)
-{
- if (BENSURE(b, sz))
- return (-1);
- memcpy(BCUR(b), p, sz);
- BSTEP(b, sz);
- return (0);
-}
-
-/* --- @buf_getbyte@ --- *
- *
- * Arguments: @buf *b@ = pointer to a buffer block
- *
- * Returns: A byte, or less than zero if there wasn't a byte there.
- *
- * Use: Gets a single byte from a buffer.
- */
-
-int buf_getbyte(buf *b)
-{
- if (BENSURE(b, 1))
- return (-1);
- return (*b->p++);
-}
-
-/* --- @buf_putbyte@ --- *
- *
- * Arguments: @buf *b@ = pointer to a buffer block
- * @int ch@ = byte to write
- *
- * Returns: Zero if OK, nonzero if there wasn't enough space.
- *
- * Use: Puts a single byte in a buffer.
- */
-
-int buf_putbyte(buf *b, int ch)
-{
- if (BENSURE(b, 1))
- return (-1);
- *b->p++ = ch;
- return (0);
-}
-
-/* --- @buf_getu16@ --- *
- *
- * Arguments: @buf *b@ = pointer to a buffer block
- * @uint16 *w@ = where to put the word
- *
- * Returns: Zero if OK, or nonzero if there wasn't a word there.
- *
- * Use: Gets a 16-bit word from a buffer.
- */
-
-int buf_getu16(buf *b, uint16 *w)
-{
- if (BENSURE(b, 2))
- return (-1);
- *w = LOAD16(b->p);
- BSTEP(b, 2);
- return (0);
-}
-
-/* --- @buf_putu16@ --- *
- *
- * Arguments: @buf *b@ = pointer to a buffer block
- * @uint16 w@ = word to write
- *
- * Returns: Zero if OK, nonzero if there wasn't enough space.
- *
- * Use: Puts a 16-but word in a buffer.
- */
-
-int buf_putu16(buf *b, uint16 w)
-{
- if (BENSURE(b, 2))
- return (-1);
- STORE16(b->p, w);
- BSTEP(b, 2);
- return (0);
-}
-
-/* --- @buf_getu32@ --- *
- *
- * Arguments: @buf *b@ = pointer to a buffer block
- * @uint32 *w@ = where to put the word
- *
- * Returns: Zero if OK, or nonzero if there wasn't a word there.
- *
- * Use: Gets a 32-bit word from a buffer.
- */
-
-int buf_getu32(buf *b, uint32 *w)
-{
- if (BENSURE(b, 4))
- return (-1);
- *w = LOAD32(b->p);
- BSTEP(b, 4);
- return (0);
-}
-
-/* --- @buf_putu32@ --- *
- *
- * Arguments: @buf *b@ = pointer to a buffer block
- * @uint32 w@ = word to write
- *
- * Returns: Zero if OK, nonzero if there wasn't enough space.
- *
- * Use: Puts a 32-but word in a buffer.
- */
-
-int buf_putu32(buf *b, uint32 w)
-{
- if (BENSURE(b, 4))
- return (-1);
- STORE32(b->p, w);
- BSTEP(b, 4);
- return (0);
-}
-
-/* --- @findz@ --- *
- *
- * Arguments: @buf *b@ = pointer to a buffer block
- * @size_t *nn@ = where to put the length
- *
- * Returns: Zero if OK, nonzero if there wasn't a null byte to be found.
- *
- * Use: Finds a terminating null byte.
- */
-
-static int findz(buf *b, size_t *nn)
-{
- octet *p;
-
- if ((p = memchr(BCUR(b), 0, BLEN(b))) == 0) {
- buf_break(b);
- return (-1);
- }
- *nn = p - BCUR(b);
- return (0);
-}
-
-#define DOSUFFIXES(DO) DO(8) DO(16) DO(32) DO(z)
-
-/* --- @buf_getmem{8,16,32,z} --- *
- *
- * Arguments: @buf *b@ = pointer to a buffer block
- * @size_t *nn@ = where to put the length
- *
- * Returns: Pointer to the buffer data, or null.
- *
- * Use: Gets a chunk of memory from a buffer. The number, @16@ or
- * @32@, is the width of the length; @z@ means null-terminated.
- */
-
-void *buf_getmem8(buf *b, size_t *nn)
-{
- int n;
-
- if ((n = buf_getbyte(b)) < 0) return (0);
- *nn = n;
- return (buf_get(b, n));
-}
-
-void *buf_getmem16(buf *b, size_t *nn)
-{
- uint16 n;
-
- if (buf_getu16(b, &n)) return (0);
- *nn = n;
- return (buf_get(b, n));
-}
-
-void *buf_getmem32(buf *b, size_t *nn)
-{
- uint32 n;
-
- if (buf_getu32(b, &n)) return (0);
- *nn = n;
- return (buf_get(b, n));
-}
-
-void *buf_getmemz(buf *b, size_t *nn)
-{
- if (findz(b, nn)) return (0);
- return (buf_get(b, *nn));
-}
-
-/* --- @buf_putmem{8,16,32,z} --- *
- *
- * Arguments: @buf *b@ = pointer to a buffer block
- * @const void *p@ = pointer to data to write
- * @size_t n@ = length to write
- *
- * Returns: Zero if OK, nonzero if there wasn't enough space.
- *
- * Use: Writes a chunk of data to a buffer. The number, @16@ or
- * @32@, is the width of the length; @z@ means null-terminated.
- */
-
-int buf_putmem8(buf *b, const void *p, size_t n)
-{
- assert(n <= 0xfful);
- if (buf_putbyte(b, n) || buf_put(b, p, n))
- return (-1);
- return (0);
-}
-
-int buf_putmem16(buf *b, const void *p, size_t n)
-{
- assert(n <= 0xfffful);
- if (buf_putu16(b, n) || buf_put(b, p, n))
- return (-1);
- return (0);
-}
-
-int buf_putmem32(buf *b, const void *p, size_t n)
-{
- assert(n <= 0xfffffffful);
- if (buf_putu32(b, n) || buf_put(b, p, n))
- return (-1);
- return (0);
-}
-
-int buf_putmemz(buf *b, const void *p, size_t n)
-{
- octet *q;
-
- assert(!memchr(p, 0, n));
- if ((q = buf_get(b, n + 1)) == 0)
- return (-1);
- memcpy(q, p, n);
- q[n] = 0;
- return (0);
-}
-
-/* --- @buf_getbuf{8,16,32,z}@ --- *
- *
- * Arguments: @buf *b@ = pointer to a buffer block
- * @buf *bb@ = where to put the result
- *
- * Returns: Zero if it worked, nonzero if there wasn't enough space.
- *
- * Use: Gets a block of data from a buffer, and writes its bounds to
- * another buffer. The number, @16@ or @32@, is the width of
- * the length; @z@ means null-terminated.
- */
-
-#define GETBUF(suff) int buf_getbuf##suff(buf *b, buf *bb) \
-{ \
- void *p; \
- size_t n; \
- \
- if ((p = buf_getmem##suff(b, &n)) == 0) \
- return (-1); \
- buf_init(bb, p, n); \
- return (0); \
-}
-DOSUFFIXES(GETBUF)
-
-/* --- @buf_putbuf{8,16,32,z}@ --- *
- *
- * Arguments: @buf *b@ = pointer to a buffer block
- * @buf *bb@ = buffer to write
- *
- * Returns: Zero if it worked, nonzero if there wasn't enough space.
- *
- * Use: Puts the contents of a buffer to a buffer. The number, @16@
- * or @32@, is the width of the length; @z@ means null-
- * terminated.
- */
-
-#define PUTBUF(suff) int buf_putbuf##suff(buf *b, buf *bb) \
- { return (buf_putmem##suff(b, BBASE(bb), BLEN(bb))); }
-DOSUFFIXES(PUTBUF)
-
-/* --- @buf_getstr{8,16,32,z}@ --- *
- *
- * Arguments: @buf *b@ = pointer to a buffer block
- * @dstr *d@ = where to put the result
- *
- * Returns: Zero if it worked, nonzero if there wasn't enough space.
- *
- * Use: Gets a block of data from a buffer, and writes its contents
- * to a string. The number, @16@ or @32@, is the width of the
- * length; @z@ means null-terminated.
- */
-
-#define GETSTR(suff) int buf_getstr##suff(buf *b, dstr *d) \
-{ \
- void *p; \
- size_t n; \
- \
- if ((p = buf_getmem##suff(b, &n)) == 0) \
- return (-1); \
- DPUTM(d, p, n); \
- return (0); \
-}
-DOSUFFIXES(GETSTR)
-
-/* --- @buf_putstr{8,16,32,z}@ --- *
- *
- * Arguments: @buf *b@ = pointer to a buffer block
- * @dstr *d@ = string to write
- *
- * Returns: Zero if it worked, nonzero if there wasn't enough space.
- *
- * Use: Puts a string to a buffer, and writes its bounds to
- * another buffer. The number, @16@ or @32@, is the width of
- * the length; @z@ means null-terminated.
- */
-
-#define PUTSTR(suff) int buf_putstr##suff(buf *b, dstr *d) \
- { return (buf_putmem##suff(b, d->buf, d->len)); }
-DOSUFFIXES(PUTSTR)
-
projects / u / mdw / catacomb / commitdiff