Compare commits
10 Commits
Author | SHA1 | Date | |
---|---|---|---|
|
44f7aba37b | ||
|
68aa01d300 | ||
|
fb41052802 | ||
|
822146bf90 | ||
|
d0723c4a76 | ||
|
6451c66518 | ||
baa1574746 | |||
0a9ff172a5 | |||
|
b6dba50071 | ||
|
81bfa9eecc |
@ -1,3 +1,9 @@
|
|||||||
|
0.0.2 (15-nov-2018)
|
||||||
|
* Added possibility to use user defined memory ops with:
|
||||||
|
ndbuf_t *ndbuf_new_wmops(const struct ndbuf_memops *, const size_t);
|
||||||
|
* Fixed bug with pkgconfig %VERSION% value
|
||||||
|
|
||||||
|
|
||||||
0.0.1 (24-nov-2017)
|
0.0.1 (24-nov-2017)
|
||||||
* initial version with all functions done
|
* initial version with all functions done
|
||||||
* tired from reuse this code and ... now it's lib
|
* tired from reuse this code and ... now it's lib
|
||||||
|
@ -1,8 +1,9 @@
|
|||||||
## Sample for libndbuf automaken stuff
|
## Sample for libndbuf automaken stuff
|
||||||
## (c) alex@vapaa.xyz
|
## Distributed under GNU LGPL v2.1 or later
|
||||||
|
## Originally written by Alexander Vdolainen <alex@vapaa.xyz> (c) 2017
|
||||||
## Process this file with automake to produce Makefile.in
|
## Process this file with automake to produce Makefile.in
|
||||||
|
|
||||||
SUBDIRS = include
|
SUBDIRS = include man
|
||||||
|
|
||||||
clean-local:
|
clean-local:
|
||||||
|
|
||||||
|
@ -16,4 +16,5 @@ AC_OUTPUT([
|
|||||||
Makefile
|
Makefile
|
||||||
libndbuf.pc
|
libndbuf.pc
|
||||||
include/Makefile
|
include/Makefile
|
||||||
|
man/Makefile
|
||||||
])
|
])
|
||||||
|
@ -5,12 +5,12 @@
|
|||||||
* (c) Alexander Vdolainen 2016 <avdolainen@zoho.com>
|
* (c) Alexander Vdolainen 2016 <avdolainen@zoho.com>
|
||||||
* (c) Alexander Vdolainen 2017 <alex@vapaa.xyz>
|
* (c) Alexander Vdolainen 2017 <alex@vapaa.xyz>
|
||||||
*
|
*
|
||||||
* libtbusd is free software: you can redistribute it and/or modify it
|
* libndbuf is free software: you can redistribute it and/or modify it
|
||||||
* under the terms of the GNU Lesser General Public License as published
|
* under the terms of the GNU Lesser General Public License as published
|
||||||
* by the Free Software Foundation, either version 2.1 of the License, or
|
* by the Free Software Foundation, either version 2.1 of the License, or
|
||||||
* (at your option) any later version.
|
* (at your option) any later version.
|
||||||
*
|
*
|
||||||
* libtbusd is distributed in the hope that it will be useful, but
|
* libndbuf is distributed in the hope that it will be useful, but
|
||||||
* WITHOUT ANY WARRANTY; without even the implied warranty of
|
* WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
|
||||||
* See the GNU Lesser General Public License for more details.
|
* See the GNU Lesser General Public License for more details.
|
||||||
@ -27,16 +27,33 @@
|
|||||||
|
|
||||||
#define NDBUF_TERMINAT 0xdeadbeef
|
#define NDBUF_TERMINAT 0xdeadbeef
|
||||||
|
|
||||||
|
/* List of supported flags: */
|
||||||
#define NDBUF_BURN (1 << 1) /* burn buffer data before free */
|
#define NDBUF_BURN (1 << 1) /* burn buffer data before free */
|
||||||
#define NDBUF_NREA (1 << 2) /* non reallocatable buffer */
|
#define NDBUF_NREA (1 << 2) /* non reallocatable buffer */
|
||||||
|
#define NDBUF_UCMO (1 << 3) /* user defined memops functions */
|
||||||
|
#define NDBUF_RORB (1 << 4) /* read-only buffer */
|
||||||
|
#define NDBUF_MOLS (1 << 5) /* "memory-ops"-less escan i.e. return
|
||||||
|
* pointers to the raw buffer itself,
|
||||||
|
* instead of alloc/memcpy
|
||||||
|
*/
|
||||||
|
|
||||||
|
struct ndbuf_memops {
|
||||||
|
void *(*alloc)(size_t);
|
||||||
|
uint32_t (*zero)(void *, size_t);
|
||||||
|
void (*free)(void *ptr);
|
||||||
|
};
|
||||||
|
|
||||||
typedef struct __rawdatabuffer_type {
|
typedef struct __rawdatabuffer_type {
|
||||||
char *raw;
|
char *raw;
|
||||||
|
union {
|
||||||
|
void (*freebuf)(char *);
|
||||||
|
const struct ndbuf_memops *mop;
|
||||||
|
};
|
||||||
uint32_t rlength; /* raw buffer allocated length */
|
uint32_t rlength; /* raw buffer allocated length */
|
||||||
uint32_t ulength; /* length of used allocated space */
|
uint32_t ulength; /* length of used allocated space */
|
||||||
uint32_t curr; /* cursor for read/write operations */
|
uint32_t curr; /* cursor for read/write operations */
|
||||||
int flags;
|
size_t blk_size;
|
||||||
void (*freebuf)(char *);
|
uint32_t flags;
|
||||||
} ndbuf_t;
|
} ndbuf_t;
|
||||||
|
|
||||||
/* variadic macro workaround */
|
/* variadic macro workaround */
|
||||||
@ -67,12 +84,15 @@ typedef struct __rawdatabuffer_type {
|
|||||||
/* allocate raw buffer with defaults preallocation */
|
/* allocate raw buffer with defaults preallocation */
|
||||||
ndbuf_t *ndbuf_new(void);
|
ndbuf_t *ndbuf_new(void);
|
||||||
|
|
||||||
/* will do the same as ndbuf_new but will allocate given length */
|
/* will do the same as ndbuf_new but will pre-allocate with the given length */
|
||||||
ndbuf_t *ndbuf_new_palloc(uint32_t);
|
ndbuf_t *ndbuf_new_palloc(uint32_t);
|
||||||
|
|
||||||
/* create ndbuf from raw buffer */
|
/* create ndbuf from raw buffer */
|
||||||
ndbuf_t *ndbuf_new_frombuf(char *buf, size_t buf_len, void (*freebuf)(char *));
|
ndbuf_t *ndbuf_new_frombuf(char *buf, size_t buf_len, void (*freebuf)(char *));
|
||||||
|
|
||||||
|
/* create a new buffer with user defined memops */
|
||||||
|
ndbuf_t *ndbuf_new_wmops(const struct ndbuf_memops *, const size_t);
|
||||||
|
|
||||||
/* free all allocated space and buffer itself */
|
/* free all allocated space and buffer itself */
|
||||||
void ndbuf_free(ndbuf_t *);
|
void ndbuf_free(ndbuf_t *);
|
||||||
|
|
||||||
@ -158,4 +178,7 @@ int ndbuf_cmp(ndbuf_t *, ndbuf_t *);
|
|||||||
*/
|
*/
|
||||||
int ndbuf_memopt(ndbuf_t *);
|
int ndbuf_memopt(ndbuf_t *);
|
||||||
|
|
||||||
|
/* free in a right way: freing allocated memory during escan */
|
||||||
|
void ndbuf_free_item(ndbuf_t *b, void *ip, size_t is);
|
||||||
|
|
||||||
#endif /* __NDBUF_H__ */
|
#endif /* __NDBUF_H__ */
|
||||||
|
@ -7,7 +7,7 @@ includedir=@includedir@
|
|||||||
|
|
||||||
Name: libndbuf
|
Name: libndbuf
|
||||||
Description: Network designed binary buffer pack/unpack library
|
Description: Network designed binary buffer pack/unpack library
|
||||||
Version: @LIBNDBUF_VERSION@
|
Version: @VERSION@
|
||||||
Requires:
|
Requires:
|
||||||
Libs: -L${libdir} -lndbuf
|
Libs: -L${libdir} -lndbuf
|
||||||
Cflags: -I${includedir}
|
Cflags: -I${includedir}
|
||||||
|
2
man/Makefile.am
Normal file
2
man/Makefile.am
Normal file
@ -0,0 +1,2 @@
|
|||||||
|
man_MANS = ndbuf_new.3 ndbuf_new_palloc.3 ndbuf_new_wmops.3 ndbuf_new_frombuf.3 \
|
||||||
|
ndbuf_free.3 ndbuf_free_item.3
|
1
man/ndbuf_free.3
Symbolic link
1
man/ndbuf_free.3
Symbolic link
@ -0,0 +1 @@
|
|||||||
|
ndbuf_new.3
|
1
man/ndbuf_free_item.3
Symbolic link
1
man/ndbuf_free_item.3
Symbolic link
@ -0,0 +1 @@
|
|||||||
|
ndbuf_new.3
|
141
man/ndbuf_new.3
Normal file
141
man/ndbuf_new.3
Normal file
@ -0,0 +1,141 @@
|
|||||||
|
.TH NDBUF_NEW 3 "15 September 2018" "NDBUF" "Binary buffers lib manual"
|
||||||
|
.SH NAME
|
||||||
|
ndbuf_new \- Create a new raw buffer with library defaults
|
||||||
|
.br
|
||||||
|
ndbuf_new_palloc \- Create a new raw buffer with predefined length
|
||||||
|
.br
|
||||||
|
ndbuf_new_frombuf \- Create a new raw buffer with given data
|
||||||
|
.br
|
||||||
|
ndbuf_new_wmops \- Create a new raw buffer with user-defined memory ops
|
||||||
|
.br
|
||||||
|
ndbuf_free \- Free raw buffer structure
|
||||||
|
.br
|
||||||
|
.SH SYNOPSIS
|
||||||
|
.B #include <ndbuf/ndbuf.h>
|
||||||
|
.sp
|
||||||
|
ndbuf_t *ndbuf_new(void);
|
||||||
|
|
||||||
|
ndbuf_t *ndbuf_new_palloc(uint32_t);
|
||||||
|
|
||||||
|
ndbuf_t *ndbuf_new_frombuf(char *buf, size_t buf_len, void (*freebuf)(char *));
|
||||||
|
|
||||||
|
ndbuf_t *ndbuf_new_wmops(const struct ndbuf_memops *, const size_t);
|
||||||
|
|
||||||
|
void ndbuf_free(ndbuf_t *);
|
||||||
|
|
||||||
|
void ndbuf_free_item(ndbuf_t *b, void *ip, size_t is);
|
||||||
|
|
||||||
|
struct ndbuf_memops {
|
||||||
|
.br
|
||||||
|
void *(*alloc)(size_t);
|
||||||
|
.br
|
||||||
|
uint32_t (*zero)(void *, size_t);
|
||||||
|
.br
|
||||||
|
void (*free)(void *ptr);
|
||||||
|
.br
|
||||||
|
};
|
||||||
|
.br
|
||||||
|
.sp
|
||||||
|
.SH DESCRIPTION
|
||||||
|
.B ndbuf_new
|
||||||
|
family of functions are used to create
|
||||||
|
.B ndbuf_t
|
||||||
|
structure with a different options.
|
||||||
|
.br
|
||||||
|
|
||||||
|
.br
|
||||||
|
.B ndbuf_new()
|
||||||
|
will create a
|
||||||
|
.B ndbuf_t
|
||||||
|
with defaults values and default memory chunk preallocated. No special flags are assigned during the creation.
|
||||||
|
.br
|
||||||
|
|
||||||
|
.br
|
||||||
|
.B ndbuf_new_palloc(uint32_t)
|
||||||
|
will create a
|
||||||
|
.B ndbuf_t
|
||||||
|
with defaults values, but buffer will be allocated with the given lenght. No special flags are assigned during the call.
|
||||||
|
.br
|
||||||
|
|
||||||
|
.br
|
||||||
|
.B ndbuf_new_frombuf(char *buf, size_t buf_len, void (*freebuf)(char *))
|
||||||
|
will create a
|
||||||
|
.B ndbuf_t
|
||||||
|
with buffer point to *buf and is assume the length of this buffer is buf_len, function freebuf will be used during
|
||||||
|
.B ndbuf_t
|
||||||
|
freeing if provided. A few flags will be set:
|
||||||
|
.br
|
||||||
|
.B NDBUF_NREA
|
||||||
|
means non-reallocatable buffer. See also
|
||||||
|
.B ndbuf_setflags(3)
|
||||||
|
man page for further flags exaplanations.
|
||||||
|
.br
|
||||||
|
|
||||||
|
.br
|
||||||
|
.B ndbuf_new_wmops(const struct ndbuf_memops *, const size_t)
|
||||||
|
is used to create a
|
||||||
|
.B ndbuf_t
|
||||||
|
with a custom provided memory options and custom memory block size. Provided structure should has a few functions pointers:
|
||||||
|
.br
|
||||||
|
.B alloc(size_t)
|
||||||
|
is used to allocate a memory block
|
||||||
|
.br
|
||||||
|
.B zero(void *, size_t)
|
||||||
|
is used to zero/fill freed or newly allocated memory if
|
||||||
|
.B NDBUF_BURN
|
||||||
|
flag is set.
|
||||||
|
.br
|
||||||
|
.B free(void *)
|
||||||
|
is used to free allocated memory.
|
||||||
|
.br
|
||||||
|
No specific flags are set upon a call.
|
||||||
|
.br
|
||||||
|
|
||||||
|
.br
|
||||||
|
.B ndbuf_free(ndbuf_t *)
|
||||||
|
is used to free a corresponding
|
||||||
|
.B ndbuf_t
|
||||||
|
structure, and connected raw buffer data as well, excepting the following cases:
|
||||||
|
.br
|
||||||
|
.B NDBUF_NREA
|
||||||
|
flag is set and no
|
||||||
|
.B freebuf()
|
||||||
|
function pointer is provided.
|
||||||
|
.br
|
||||||
|
.B NDBUF_RORB
|
||||||
|
flag is set.
|
||||||
|
.br
|
||||||
|
|
||||||
|
.br
|
||||||
|
.B void ndbuf_free_item(ndbuf_t*, void*, size_t)
|
||||||
|
will free allocated item via escan function, if flag
|
||||||
|
.B NDBUF_BURN
|
||||||
|
is set and size in't 0 memset() or zero() custom function will be called prior to free. This function is using default memory allocation/deallocation functions or (if provided) custom ones.
|
||||||
|
.br
|
||||||
|
.SH RETURN VALUE
|
||||||
|
.B ndbuf_new
|
||||||
|
family of functions returns a pointer to newly created
|
||||||
|
.B ndbuf_t
|
||||||
|
structure or
|
||||||
|
.B NULL
|
||||||
|
otherwise (if any error occurs, or invalid options are given).
|
||||||
|
.br
|
||||||
|
No specific errno code is set in case of error.
|
||||||
|
.br
|
||||||
|
.SH BUGS
|
||||||
|
Not known yet.
|
||||||
|
.SH EXAMPLE
|
||||||
|
None.
|
||||||
|
.SH APPLICATION USAGE
|
||||||
|
None.
|
||||||
|
.SH RATIONALE
|
||||||
|
.B ndbuf_new_wmops()
|
||||||
|
is recommended to use in specific environments i.e. when users are required to store and process all data buffers in a secure way (use special memory allocation to avoid memory pages swappnig, avoid memset() optimization etc ...).
|
||||||
|
.SH SEE ALSO
|
||||||
|
.BI ndbuf_setflags(3)
|
||||||
|
.SH COPYRIGHT
|
||||||
|
This software licensed under GNU LGPL v2.1 or later. See COPYING for further details.
|
||||||
|
.br
|
||||||
|
(c) Authors of libndbuf 2017-2018 <http://vapaa.xyz>
|
||||||
|
.SH AUTHOR
|
||||||
|
Alexander Vdolainen (alex@vapaa.xyz)
|
1
man/ndbuf_new_frombuf.3
Symbolic link
1
man/ndbuf_new_frombuf.3
Symbolic link
@ -0,0 +1 @@
|
|||||||
|
ndbuf_new.3
|
1
man/ndbuf_new_palloc.3
Symbolic link
1
man/ndbuf_new_palloc.3
Symbolic link
@ -0,0 +1 @@
|
|||||||
|
ndbuf_new.3
|
1
man/ndbuf_new_wmops.3
Symbolic link
1
man/ndbuf_new_wmops.3
Symbolic link
@ -0,0 +1 @@
|
|||||||
|
ndbuf_new.3
|
274
ndbuf.c
274
ndbuf.c
@ -2,14 +2,14 @@
|
|||||||
* Networking binary buffers pack/unpack library
|
* Networking binary buffers pack/unpack library
|
||||||
*
|
*
|
||||||
* (c) Alexander Vdolainen 2016 <avdolainen@zoho.com>
|
* (c) Alexander Vdolainen 2016 <avdolainen@zoho.com>
|
||||||
* (c) Alexander Vdolainen 2017 <alex@vapaa.xyz>
|
* (c) Alexander Vdolainen 2017, 2018 <alex@vapaa.xyz>
|
||||||
*
|
*
|
||||||
* libtbusd is free software: you can redistribute it and/or modify it
|
* libndbuf is free software: you can redistribute it and/or modify it
|
||||||
* under the terms of the GNU Lesser General Public License as published
|
* under the terms of the GNU Lesser General Public License as published
|
||||||
* by the Free Software Foundation, either version 2.1 of the License, or
|
* by the Free Software Foundation, either version 2.1 of the License, or
|
||||||
* (at your option) any later version.
|
* (at your option) any later version.
|
||||||
*
|
*
|
||||||
* libtbusd is distributed in the hope that it will be useful, but
|
* libndbuf is distributed in the hope that it will be useful, but
|
||||||
* WITHOUT ANY WARRANTY; without even the implied warranty of
|
* WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
|
||||||
* See the GNU Lesser General Public License for more details.
|
* See the GNU Lesser General Public License for more details.
|
||||||
@ -34,7 +34,7 @@ static inline uint64_t ntohll(uint64_t n)
|
|||||||
#ifdef WORDS_BIGENDIAN
|
#ifdef WORDS_BIGENDIAN
|
||||||
return n;
|
return n;
|
||||||
#else /* WORDS_BIGENDIAN */
|
#else /* WORDS_BIGENDIAN */
|
||||||
return (((uint64_t)(n) << 56) | \
|
return (uint64_t)(((uint64_t)(n) << 56) | \
|
||||||
(((uint64_t)(n) << 40) & 0xff000000000000ULL) | \
|
(((uint64_t)(n) << 40) & 0xff000000000000ULL) | \
|
||||||
(((uint64_t)(n) << 24) & 0xff0000000000ULL) | \
|
(((uint64_t)(n) << 24) & 0xff0000000000ULL) | \
|
||||||
(((uint64_t)(n) << 8) & 0xff00000000ULL) | \
|
(((uint64_t)(n) << 8) & 0xff00000000ULL) | \
|
||||||
@ -104,16 +104,51 @@ ndbuf_t *ndbuf_new_palloc(uint32_t alen)
|
|||||||
return b;
|
return b;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
ndbuf_t *ndbuf_new_wmops(const struct ndbuf_memops *o, const size_t blk_len)
|
||||||
|
{
|
||||||
|
ndbuf_t *b = NULL;
|
||||||
|
|
||||||
|
if(!o) return NULL;
|
||||||
|
else { /* avoid nil calls */
|
||||||
|
if(!o->alloc || !o->free) return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
if(!(b = malloc(sizeof(ndbuf_t)))) return NULL;
|
||||||
|
else memset(b, 0, sizeof(ndbuf_t));
|
||||||
|
|
||||||
|
if(blk_len <= 0) b->blk_size = DEFAULT_PREALLOC_SIZE;
|
||||||
|
|
||||||
|
if(!(b->raw = o->alloc(b->blk_size))) {
|
||||||
|
free(b);
|
||||||
|
return NULL;
|
||||||
|
} else {
|
||||||
|
b->rlength = b->blk_size;
|
||||||
|
b->ulength = b->curr = 0;
|
||||||
|
b->mop = o;
|
||||||
|
b->flags |= NDBUF_UCMO;
|
||||||
|
}
|
||||||
|
|
||||||
|
return b;
|
||||||
|
}
|
||||||
|
|
||||||
/* free all allocated space and buffer itself */
|
/* free all allocated space and buffer itself */
|
||||||
void ndbuf_free(ndbuf_t *b)
|
void ndbuf_free(ndbuf_t *b)
|
||||||
{
|
{
|
||||||
|
const struct ndbuf_memops *o = NULL;
|
||||||
if(!b) return;
|
if(!b) return;
|
||||||
|
|
||||||
if((b->flags & NDBUF_BURN) && b->raw)
|
if(b->flags & NDBUF_UCMO) o = b->mop;
|
||||||
memset(b->raw, 0, b->rlength);
|
|
||||||
|
|
||||||
if((b->flags & NDBUF_NREA) && b->freebuf) b->freebuf(b->raw);
|
if(o) {
|
||||||
else if(b->raw) free(b->raw);
|
if((b->flags & NDBUF_BURN) && o->zero) o->zero((void *)b->raw, b->rlength);
|
||||||
|
o->free(b->raw);
|
||||||
|
} else {
|
||||||
|
if((b->flags & NDBUF_BURN) && b->raw)
|
||||||
|
memset(b->raw, 0, b->rlength);
|
||||||
|
|
||||||
|
if((b->flags & NDBUF_NREA) && b->freebuf) b->freebuf(b->raw);
|
||||||
|
else if(b->raw) free(b->raw);
|
||||||
|
}
|
||||||
|
|
||||||
if(b->flags & NDBUF_BURN) memset(b, 0, sizeof(ndbuf_t));
|
if(b->flags & NDBUF_BURN) memset(b, 0, sizeof(ndbuf_t));
|
||||||
free(b);
|
free(b);
|
||||||
@ -210,23 +245,37 @@ static int __rdb_grow(ndbuf_t *b, uint32_t len)
|
|||||||
{
|
{
|
||||||
uint32_t rlen;
|
uint32_t rlen;
|
||||||
char *ne = NULL;
|
char *ne = NULL;
|
||||||
|
const struct ndbuf_memops *o = NULL;
|
||||||
|
size_t bs = b->blk_size;
|
||||||
|
|
||||||
if(!len) return -1;
|
if(!len) return -1;
|
||||||
if(b->rlength + len > NDBUF_MAXLENGTH) return -1;
|
if(b->rlength + len > NDBUF_MAXLENGTH) return -1;
|
||||||
|
|
||||||
rlen = len +
|
if(b->flags & NDBUF_UCMO) o = b->mop;
|
||||||
(len%DEFAULT_PREALLOC_SIZE != 0 ? (DEFAULT_PREALLOC_SIZE - len%DEFAULT_PREALLOC_SIZE) : 0);
|
|
||||||
|
rlen = len + (len%bs != 0 ? (bs - len%bs) : 0);
|
||||||
if(b->rlength + rlen > NDBUF_MAXLENGTH) rlen = len;
|
if(b->rlength + rlen > NDBUF_MAXLENGTH) rlen = len;
|
||||||
|
|
||||||
rlen += b->rlength;
|
rlen += b->rlength;
|
||||||
if(!(ne = malloc(rlen))) return -1;
|
if(o) ne = (char *)o->alloc(rlen);
|
||||||
|
else ne = malloc(rlen);
|
||||||
|
|
||||||
if(b->flags & NDBUF_BURN) memset(ne, 0, rlen);
|
if(!ne) return -1;
|
||||||
|
|
||||||
|
if(b->flags & NDBUF_BURN) {
|
||||||
|
if(o) o->zero((void *)ne, rlen);
|
||||||
|
else memset(ne, 0, rlen);
|
||||||
|
}
|
||||||
|
|
||||||
memcpy(ne, b->raw, b->ulength);
|
memcpy(ne, b->raw, b->ulength);
|
||||||
|
|
||||||
if(b->flags & NDBUF_BURN) memset(b->raw, 0, b->rlength);
|
if(o) {
|
||||||
free(b->raw);
|
if(b->flags & NDBUF_BURN) o->zero((void *)b->raw, b->rlength);
|
||||||
|
o->free((void *)b->raw);
|
||||||
|
} else {
|
||||||
|
if(b->flags & NDBUF_BURN) memset(b->raw, 0, b->rlength);
|
||||||
|
free(b->raw);
|
||||||
|
}
|
||||||
b->raw = ne;
|
b->raw = ne;
|
||||||
b->rlength = rlen;
|
b->rlength = rlen;
|
||||||
|
|
||||||
@ -287,12 +336,25 @@ uint32_t ndbuf_write_u64(ndbuf_t *b, uint64_t uu)
|
|||||||
return sizeof(uint64_t);
|
return sizeof(uint64_t);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static inline int __ndbuf_is_ro(ndbuf_t *b)
|
||||||
|
{
|
||||||
|
return (b->flags & NDBUF_RORB) ? 1 : 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline int __ndbuf_is_nrea(ndbuf_t *b)
|
||||||
|
{
|
||||||
|
return (b->flags & NDBUF_NREA) ? 1 : 0;
|
||||||
|
}
|
||||||
|
|
||||||
/* write raw data with the given length */
|
/* write raw data with the given length */
|
||||||
uint32_t ndbuf_write_raw(ndbuf_t *b, void *wi, uint32_t len)
|
uint32_t ndbuf_write_raw(ndbuf_t *b, void *wi, uint32_t len)
|
||||||
{
|
{
|
||||||
if(!b || !b->raw) return 0;
|
if(!b || !b->raw) return 0;
|
||||||
if(!wi || !len) return 0;
|
if(!wi || !len) return 0;
|
||||||
|
if(__ndbuf_is_ro(b)) return 0; /* read only buffer */
|
||||||
|
|
||||||
if(b->ulength + len >= b->rlength) {
|
if(b->ulength + len >= b->rlength) {
|
||||||
|
if(__ndbuf_is_nrea(b)) return 0; /* non reallocatable buffer */
|
||||||
if(__rdb_grow(b, len)) return 0;
|
if(__rdb_grow(b, len)) return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -306,27 +368,46 @@ uint32_t ndbuf_write_raw(ndbuf_t *b, void *wi, uint32_t len)
|
|||||||
uint32_t ndbuf_write_raw_head(ndbuf_t *b, void *wi, uint32_t len)
|
uint32_t ndbuf_write_raw_head(ndbuf_t *b, void *wi, uint32_t len)
|
||||||
{
|
{
|
||||||
char *ne;
|
char *ne;
|
||||||
|
const struct ndbuf_memops *o = NULL;
|
||||||
uint32_t rlen;
|
uint32_t rlen;
|
||||||
|
size_t bs = b->blk_size;
|
||||||
|
|
||||||
if(!b || !b->raw) return 0;
|
if(!b || !b->raw) return 0;
|
||||||
if(!wi || !len) return 0;
|
if(!wi || !len) return 0;
|
||||||
|
if(__ndbuf_is_ro(b)) return 0; /* read only buffer */
|
||||||
|
|
||||||
if(b->ulength + len > b->rlength) {
|
if(b->ulength + len > b->rlength) {
|
||||||
|
if(__ndbuf_is_nrea(b)) return 0; /* non reallocatable buffer */
|
||||||
/* allocate a new one and copy it right */
|
/* allocate a new one and copy it right */
|
||||||
if(b->rlength + len > NDBUF_MAXLENGTH) return -1;
|
if(b->rlength + len > NDBUF_MAXLENGTH) return -1;
|
||||||
|
|
||||||
rlen = len +
|
rlen = len + (len%bs != 0 ? (bs - len%bs) : 0);
|
||||||
(len%DEFAULT_PREALLOC_SIZE != 0 ? (DEFAULT_PREALLOC_SIZE - len%DEFAULT_PREALLOC_SIZE) : 0);
|
|
||||||
if(b->rlength + rlen > NDBUF_MAXLENGTH) rlen = len;
|
if(b->rlength + rlen > NDBUF_MAXLENGTH) rlen = len;
|
||||||
|
|
||||||
rlen += b->rlength;
|
rlen += b->rlength;
|
||||||
if(!(ne = malloc(rlen))) return -1;
|
|
||||||
|
|
||||||
if(b->flags & NDBUF_BURN) memset(ne, 0, rlen);
|
if(b->flags & NDBUF_UCMO) o = b->mop;
|
||||||
|
|
||||||
|
if(o) ne = (char *)o->alloc(rlen);
|
||||||
|
else ne = malloc(rlen);
|
||||||
|
|
||||||
|
if(!ne) return -1;
|
||||||
|
|
||||||
|
if(b->flags & NDBUF_BURN) {
|
||||||
|
if(o) o->zero((void *)ne, rlen);
|
||||||
|
else memset(ne, 0, rlen);
|
||||||
|
}
|
||||||
|
|
||||||
memcpy((void *)ne + len, b->raw, b->ulength);
|
memcpy((void *)ne + len, b->raw, b->ulength);
|
||||||
|
|
||||||
if(b->flags & NDBUF_BURN) memset(b->raw, 0, b->rlength);
|
if(o) {
|
||||||
free(b->raw);
|
if(b->flags & NDBUF_BURN) o->zero((void *)b->raw, b->rlength);
|
||||||
|
o->free((void *)b->raw);
|
||||||
|
} else {
|
||||||
|
if(b->flags & NDBUF_BURN) memset(b->raw, 0, b->rlength);
|
||||||
|
free(b->raw);
|
||||||
|
}
|
||||||
|
|
||||||
b->raw = ne;
|
b->raw = ne;
|
||||||
b->rlength = rlen;
|
b->rlength = rlen;
|
||||||
} else {
|
} else {
|
||||||
@ -339,8 +420,14 @@ uint32_t ndbuf_write_raw_head(ndbuf_t *b, void *wi, uint32_t len)
|
|||||||
}
|
}
|
||||||
|
|
||||||
/* parse */
|
/* parse */
|
||||||
|
#define __is_moless(x) ((x) & NDBUF_MOLS) ? 1 : 0
|
||||||
|
#define __is_usermo(x) ((x) & NDBUF_UCMO) ? 1 : 0
|
||||||
|
|
||||||
int ndbuf_escan_va(ndbuf_t *b, const char *fmt, int argc, va_list ap)
|
int ndbuf_escan_va(ndbuf_t *b, const char *fmt, int argc, va_list ap)
|
||||||
{
|
{
|
||||||
|
register uint32_t fo = b->flags;
|
||||||
|
register uint32_t len;
|
||||||
|
uint32_t clen;
|
||||||
va_list ap_copy;
|
va_list ap_copy;
|
||||||
union {
|
union {
|
||||||
uint8_t *_u8;
|
uint8_t *_u8;
|
||||||
@ -352,7 +439,6 @@ int ndbuf_escan_va(ndbuf_t *b, const char *fmt, int argc, va_list ap)
|
|||||||
ndbuf_t **_rdb;
|
ndbuf_t **_rdb;
|
||||||
} d;
|
} d;
|
||||||
const char *t, *last;
|
const char *t, *last;
|
||||||
uint32_t len, clen;
|
|
||||||
int r, count;
|
int r, count;
|
||||||
|
|
||||||
va_copy(ap_copy, ap);
|
va_copy(ap_copy, ap);
|
||||||
@ -393,18 +479,26 @@ int ndbuf_escan_va(ndbuf_t *b, const char *fmt, int argc, va_list ap)
|
|||||||
r = -1;
|
r = -1;
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
/* FIXME: Document the following line in manpages %limits% */
|
||||||
if(clen > NDBUF_MAXLENGTH) goto __errrbread;
|
if(clen > NDBUF_MAXLENGTH) goto __errrbread;
|
||||||
if((*d._cstr = malloc(clen + sizeof(char))) == NULL) {
|
|
||||||
r = -ENOMEM;
|
if(__is_moless(fo)) {
|
||||||
break;
|
*d._cstr = b->raw + b->curr;
|
||||||
|
} else {
|
||||||
|
if(__is_usermo(fo)) *d._cstr = b->mop->alloc(clen + sizeof(char));
|
||||||
|
else *d._cstr = malloc(clen + sizeof(char));
|
||||||
|
if(*d._cstr == NULL) {
|
||||||
|
r = -ENOMEM;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
len = ndbuf_read_raw(b, *d._cstr, clen);
|
||||||
|
if(len != clen) {
|
||||||
|
free(*d._cstr);
|
||||||
|
goto __errrbread;
|
||||||
|
}
|
||||||
|
(*d._cstr)[len] = '\0';
|
||||||
|
d._cstr = NULL;
|
||||||
}
|
}
|
||||||
len = ndbuf_read_raw(b, *d._cstr, clen);
|
|
||||||
if(len != clen) {
|
|
||||||
free(*d._cstr);
|
|
||||||
goto __errrbread;
|
|
||||||
}
|
|
||||||
(*d._cstr)[len] = '\0';
|
|
||||||
d._cstr = NULL;
|
|
||||||
r = 0;
|
r = 0;
|
||||||
break;
|
break;
|
||||||
case 'p':
|
case 'p':
|
||||||
@ -414,14 +508,20 @@ int ndbuf_escan_va(ndbuf_t *b, const char *fmt, int argc, va_list ap)
|
|||||||
*d._dp = NULL;
|
*d._dp = NULL;
|
||||||
count++;
|
count++;
|
||||||
|
|
||||||
if((*d._dp = malloc(clen)) == NULL) {
|
if(__is_moless(fo)) {
|
||||||
r = -ENOMEM;
|
*d._dp = b->raw + b->curr;
|
||||||
break;
|
} else {
|
||||||
}
|
if(__is_usermo(fo)) *d._dp = b->mop->alloc(clen);
|
||||||
len = ndbuf_read_raw(b, *d._dp, clen);
|
else *d._dp = malloc(clen);
|
||||||
if(len != clen) {
|
if(*d._dp == NULL) {
|
||||||
free(*d._dp);
|
r = -ENOMEM;
|
||||||
goto __errrbread;
|
break;
|
||||||
|
}
|
||||||
|
len = ndbuf_read_raw(b, *d._dp, clen);
|
||||||
|
if(len != clen) {
|
||||||
|
free(*d._dp);
|
||||||
|
goto __errrbread;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
d._dp = NULL;
|
d._dp = NULL;
|
||||||
r = 0;
|
r = 0;
|
||||||
@ -435,19 +535,25 @@ int ndbuf_escan_va(ndbuf_t *b, const char *fmt, int argc, va_list ap)
|
|||||||
r = -1;
|
r = -1;
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
if((*d._dp = malloc(clen)) == NULL) {
|
if(__is_moless(fo)) {
|
||||||
r = -ENOMEM;
|
*d._dp = b->raw + b->curr;
|
||||||
break;
|
} else {
|
||||||
}
|
if(__is_usermo(fo)) *d._dp = b->mop->alloc(clen);
|
||||||
len = ndbuf_read_raw(b, *d._dp, clen);
|
else *d._dp = malloc(clen);
|
||||||
if(len != clen) {
|
if(*d._dp == NULL) {
|
||||||
free(*d._dp);
|
r = -ENOMEM;
|
||||||
goto __errrbread;
|
break;
|
||||||
|
}
|
||||||
|
len = ndbuf_read_raw(b, *d._dp, clen);
|
||||||
|
if(len != clen) {
|
||||||
|
free(*d._dp);
|
||||||
|
goto __errrbread;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
d._dp = NULL;
|
d._dp = NULL;
|
||||||
r = 0;
|
r = 0;
|
||||||
break;
|
break;
|
||||||
case 'R':
|
case 'R': /* FIXME: what about moless and user defined ops ? */
|
||||||
d._rdb = va_arg(ap, ndbuf_t **);
|
d._rdb = va_arg(ap, ndbuf_t **);
|
||||||
*d._rdb = NULL;
|
*d._rdb = NULL;
|
||||||
|
|
||||||
@ -498,9 +604,14 @@ int ndbuf_escan_va(ndbuf_t *b, const char *fmt, int argc, va_list ap)
|
|||||||
break;
|
break;
|
||||||
case 's':
|
case 's':
|
||||||
d._cstr = va_arg(ap_copy, char **);
|
d._cstr = va_arg(ap_copy, char **);
|
||||||
if(*d._cstr) {
|
if(*d._cstr && !__is_moless(fo)) {
|
||||||
memset(*d._cstr, 0, strlen(*d._cstr));
|
if(__is_usermo(fo)) {
|
||||||
free(*d._cstr);
|
b->mop->zero(*d._cstr, strlen(*d._cstr));
|
||||||
|
b->mop->free(*d._cstr);
|
||||||
|
} else {
|
||||||
|
memset(*d._cstr, 0, strlen(*d._cstr));
|
||||||
|
free(*d._cstr);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
break;
|
break;
|
||||||
case 'R':
|
case 'R':
|
||||||
@ -511,7 +622,10 @@ int ndbuf_escan_va(ndbuf_t *b, const char *fmt, int argc, va_list ap)
|
|||||||
(void)va_arg(ap_copy, size_t);
|
(void)va_arg(ap_copy, size_t);
|
||||||
case 'P':
|
case 'P':
|
||||||
d._dp = va_arg(ap_copy, void **);
|
d._dp = va_arg(ap_copy, void **);
|
||||||
if(*d._dp) free(*d._dp);
|
if(*d._dp && !__is_moless(fo)) {
|
||||||
|
if(__is_usermo(fo)) b->mop->free(*d._dp);
|
||||||
|
else free(*d._dp);
|
||||||
|
}
|
||||||
break;
|
break;
|
||||||
default:
|
default:
|
||||||
(void)va_arg(ap_copy, void *);
|
(void)va_arg(ap_copy, void *);
|
||||||
@ -552,6 +666,8 @@ uint32_t ndbuf_print_va(ndbuf_t *b, const char *fmt, int argc, va_list ap)
|
|||||||
uint32_t len = 0, clen;
|
uint32_t len = 0, clen;
|
||||||
int r, count;
|
int r, count;
|
||||||
|
|
||||||
|
if(__ndbuf_is_ro(b)) return 0; /* read only buffer */
|
||||||
|
|
||||||
for(t = fmt, count = 0; *t != '\0'; t++, count++) {
|
for(t = fmt, count = 0; *t != '\0'; t++, count++) {
|
||||||
if(count > argc && argc != -1) return 0;
|
if(count > argc && argc != -1) return 0;
|
||||||
switch(*t) {
|
switch(*t) {
|
||||||
@ -705,6 +821,8 @@ void ndbuf_setflags(ndbuf_t *b, int af)
|
|||||||
{
|
{
|
||||||
if(!b) return;
|
if(!b) return;
|
||||||
|
|
||||||
|
if(((af & NDBUF_BURN) && (b->flags & NDBUF_UCMO)) && !b->mop->zero) return;
|
||||||
|
|
||||||
b->flags |= af;
|
b->flags |= af;
|
||||||
|
|
||||||
return;
|
return;
|
||||||
@ -715,6 +833,8 @@ void ndbuf_exflags(ndbuf_t *b, int nf)
|
|||||||
{
|
{
|
||||||
if(!b) return;
|
if(!b) return;
|
||||||
|
|
||||||
|
if(b->flags & NDBUF_UCMO) nf |= NDBUF_UCMO;
|
||||||
|
|
||||||
b->flags = nf;
|
b->flags = nf;
|
||||||
|
|
||||||
return;
|
return;
|
||||||
@ -749,24 +869,43 @@ int ndbuf_cmp(ndbuf_t *b1, ndbuf_t *b2)
|
|||||||
/* let the buffer to use actually used bytes, not all allocated space
|
/* let the buffer to use actually used bytes, not all allocated space
|
||||||
* will return 0 on success (or in case if it doesn't required),
|
* will return 0 on success (or in case if it doesn't required),
|
||||||
* ENOMEM or other error if fails
|
* ENOMEM or other error if fails
|
||||||
|
* TODO: ? make this buffer non reallocable ?
|
||||||
*/
|
*/
|
||||||
int ndbuf_memopt(ndbuf_t *b)
|
int ndbuf_memopt(ndbuf_t *b)
|
||||||
{
|
{
|
||||||
uint32_t len;
|
uint32_t len;
|
||||||
|
size_t bs = b->blk_size;
|
||||||
char *ne;
|
char *ne;
|
||||||
|
const struct ndbuf_memops *o = NULL;
|
||||||
|
|
||||||
if(!b || !b->raw) return EINVAL;
|
if(!b || !b->raw) return EINVAL;
|
||||||
|
|
||||||
if((b->rlength - b->ulength) > DEFAULT_PREALLOC_SIZE) {
|
if((b->rlength - b->ulength) > bs) {
|
||||||
len = b->ulength +
|
len = b->ulength +
|
||||||
(b->ulength%DEFAULT_PREALLOC_SIZE != 0 ?
|
(b->ulength%bs != 0 ? (bs - b->ulength%bs) : 0);
|
||||||
(DEFAULT_PREALLOC_SIZE - b->ulength%DEFAULT_PREALLOC_SIZE) : 0);
|
|
||||||
if(!(ne = malloc(len))) return ENOMEM;
|
if(b->flags & NDBUF_UCMO) o = b->mop;
|
||||||
|
|
||||||
|
if(o) ne = (char *)o->alloc(len);
|
||||||
|
else ne = malloc(len);
|
||||||
|
|
||||||
|
if(!ne) return ENOMEM;
|
||||||
|
|
||||||
|
if(b->flags & NDBUF_BURN) {
|
||||||
|
if(o) o->zero((void *)ne, len);
|
||||||
|
else memset(ne, 0, len);
|
||||||
|
}
|
||||||
|
|
||||||
if(b->flags & NDBUF_BURN) memset(ne, 0, len);
|
|
||||||
memcpy(ne, b->raw, b->ulength);
|
memcpy(ne, b->raw, b->ulength);
|
||||||
if(b->flags & NDBUF_BURN) memset(b->raw, 0, b->ulength);
|
|
||||||
free(b->raw);
|
if(o) {
|
||||||
|
if(b->flags & NDBUF_BURN) o->zero((void *)b->raw, b->rlength);
|
||||||
|
o->free((void *)b->raw);
|
||||||
|
} else {
|
||||||
|
if(b->flags & NDBUF_BURN) memset(b->raw, 0, b->rlength);
|
||||||
|
free(b->raw);
|
||||||
|
}
|
||||||
|
|
||||||
b->raw = ne;
|
b->raw = ne;
|
||||||
b->rlength = len;
|
b->rlength = len;
|
||||||
}
|
}
|
||||||
@ -774,4 +913,21 @@ int ndbuf_memopt(ndbuf_t *b)
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void ndbuf_free_item(ndbuf_t *b, void *ip, size_t is)
|
||||||
|
{
|
||||||
|
if(!b || !ip) return;
|
||||||
|
|
||||||
|
if(__is_moless(b->flags)) return;
|
||||||
|
|
||||||
|
if(__is_usermo(b->flags)) {
|
||||||
|
if((b->flags & NDBUF_BURN) && (is > 0)) b->mop->zero(ip, is);
|
||||||
|
b->mop->free(ip);
|
||||||
|
} else {
|
||||||
|
if((b->flags & NDBUF_BURN) && (is > 0)) memset(ip, 0, is);
|
||||||
|
free(ip);
|
||||||
|
}
|
||||||
|
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
#undef DEFAULT_PREALLOC_SIZE
|
#undef DEFAULT_PREALLOC_SIZE
|
||||||
|
Loading…
x
Reference in New Issue
Block a user