26RCSIDH(dbuff_h,
"$Id: 32e64f2ff1b14c14d9709f3018d6b56da922a5e6 $")
33#include <freeradius-devel/missing.h>
34#include <freeradius-devel/util/debug.h>
35#include <freeradius-devel/util/nbo.h>
115struct fr_dbuff_marker_s {
126#define FR_DBUFF_ADV_PARENT_CURRENT 0x01
128#define FR_DBUFF_ADV_PARENT_END 0x02
186#define FR_DBUFF_RETURN(_func, ...) \
188 ssize_t _slen = _func(__VA_ARGS__ ); \
189 if (_slen < 0) return _slen; \
198#define FR_DBUFF_ERROR_OFFSET(_slen, _offset) ((_slen < INT32_MAX) ? _slen : _slen - (ssize_t) _offset)
210#define _FR_DBUFF(_dbuff_or_marker, _start, _adv_parent) \
212 .buff = fr_dbuff_buff(_dbuff_or_marker), \
214 .end = fr_dbuff_end(_dbuff_or_marker), \
215 .p = fr_dbuff_current(_dbuff_or_marker), \
216 .is_const = fr_dbuff_ptr(_dbuff_or_marker)->is_const, \
217 .adv_parent = (_adv_parent), \
218 .shifted = fr_dbuff_ptr(_dbuff_or_marker)->shifted, \
219 .extend = fr_dbuff_ptr(_dbuff_or_marker)->extend, \
220 .uctx = fr_dbuff_ptr(_dbuff_or_marker)->uctx, \
221 .parent = fr_dbuff_ptr(_dbuff_or_marker) \
232#define FR_DBUFF(_dbuff_or_marker) _FR_DBUFF(_dbuff_or_marker, fr_dbuff_current(_dbuff_or_marker), 0x00)
241#define FR_DBUFF_ABS(_dbuff_or_marker) _FR_DBUFF(_dbuff_or_marker, fr_dbuff_start(_dbuff_or_marker), 0x00)
250#define FR_DBUFF_BIND_CURRENT(_dbuff_or_marker) _FR_DBUFF(_dbuff_or_marker, fr_dbuff_current(_dbuff_or_marker), FR_DBUFF_ADV_PARENT_CURRENT)
259#define FR_DBUFF_BIND_CURRENT_ABS(_dbuff_or_marker) FR_DBUFF_ABS(_dbuff_or_marker, fr_dbuff_start(_dbuff_or_marker), FR_DBUFF_ADV_PARENT_CURRENT)
270#define FR_DBUFF_BIND_END_ABS(_dbuff_or_marker) _FR_DBUFF(_dbuff_or_marker, fr_dbuff_start(_dbuff_or_marker), FR_DBUFF_ADV_PARENT_END)
277#define _FR_DBUFF_MAX(_dbuff_or_marker, _max, _adv_parent) \
279 .buff = fr_dbuff_buff(_dbuff_or_marker), \
280 .start = fr_dbuff_current(_dbuff_or_marker), \
281 .end = (((fr_dbuff_end(_dbuff_or_marker) - (_max) < fr_dbuff_current(_dbuff_or_marker))) ? fr_dbuff_end(_dbuff_or_marker) : (fr_dbuff_current(_dbuff_or_marker) + (_max))), \
282 .p = fr_dbuff_current(_dbuff_or_marker), \
283 .is_const = fr_dbuff_ptr(_dbuff_or_marker)->is_const, \
284 .adv_parent = _adv_parent, \
285 .shifted = fr_dbuff_ptr(_dbuff_or_marker)->shifted, \
288 .parent = fr_dbuff_ptr(_dbuff_or_marker) \
311#define FR_DBUFF_MAX(_dbuff_or_marker, _max) _FR_DBUFF_MAX(_dbuff_or_marker, _max, 0x00)
328#define FR_DBUFF_MAX_BIND_CURRENT(_dbuff_or_marker, _max) _FR_DBUFF_MAX(_dbuff_or_marker, _max, FR_DBUFF_ADV_PARENT_CURRENT)
334#if defined(__GNUC__) && __GNUC__ >= 11
346 if (
unlikely(end < start)) end = start;
364#define fr_dbuff_init(_out, _start, _len_or_end) \
365_fr_dbuff_init(_out, \
366 (uint8_t const *)(_start), \
367 _Generic((_len_or_end), \
368 size_t : (uint8_t const *)(_start) + (size_t)(_len_or_end), \
369 long : (uint8_t const *)(_start) + (size_t)(_len_or_end), \
370 int : (uint8_t const *)(_start) + (size_t)(_len_or_end), \
371 unsigned int : (uint8_t const *)(_start) + (size_t)(_len_or_end), \
372 uint8_t * : (uint8_t const *)(_len_or_end), \
373 uint8_t const * : (uint8_t const *)(_len_or_end), \
374 char * : (uint8_t const *)(_len_or_end), \
375 char const * : (uint8_t const *)(_len_or_end) \
379 uint8_t const * : true, \
381 char const * : true \
383#if defined(__GNUC__) && __GNUC__ >= 11
387#define FR_DBUFF_INIT(_out, _start, _len_or_end) do { \
388 fr_dbuff_init(_out, _start, _len_or_end); \
389 *(unsigned char *) _start = '\0'; \
408} fr_dbuff_uctx_talloc_t;
422 fr_dbuff_t *dbuff, fr_dbuff_uctx_talloc_t *tctx,
423 size_t init,
size_t max)
427 *tctx = (fr_dbuff_uctx_talloc_t){
444 memset(dbuff, 0,
sizeof(*dbuff));
465 TALLOC_FREE(dbuff->buff);
498 *fctx = (fr_dbuff_uctx_fd_t){
501 .buff_end =
buff + len
524#define FR_DBUFF_TMP(_start, _len_or_end) \
526 .buff_i = (uint8_t const *)(_start), \
527 .start_i = (uint8_t const *)(_start), \
528 .end_i = _Generic((_len_or_end), \
529 size_t : (uint8_t const *)(_start) + (size_t)(_len_or_end), \
530 long : (uint8_t const *)(_start) + (size_t)(_len_or_end), \
531 int : (uint8_t const *)(_start) + (size_t)(_len_or_end), \
532 unsigned int : (uint8_t const *)(_start) + (size_t)(_len_or_end), \
533 uint8_t * : (uint8_t const *)(_len_or_end), \
534 uint8_t const * : (uint8_t const *)(_len_or_end), \
535 char * : (uint8_t const *)(_len_or_end), \
536 char const * : (uint8_t const *)(_len_or_end) \
539 .is_const = _Generic((_start), \
541 uint8_t const * : true, \
543 char const * : true \
566#define FR_DBUFF_TALLOC_THREAD_LOCAL(_out, _init, _max) \
568 static _Thread_local fr_dbuff_thread_local_t *_dbuff_t_local; \
569 if (!_dbuff_t_local) { \
570 fr_dbuff_thread_local_t *dbtl = talloc_zero(NULL, fr_dbuff_thread_local_t); \
571 fr_dbuff_init_talloc(dbtl, &dbtl->dbuff, &dbtl->tctx, _init, _max); \
572 fr_atexit_thread_local(_dbuff_t_local, _dbuff_thread_local_free, dbtl); \
573 *(_out) = &_dbuff_t_local->dbuff; \
575 fr_dbuff_reset_talloc(&_dbuff_t_local->dbuff); \
576 *(_out) = &_dbuff_t_local->dbuff; \
592#define FR_DBUFF_FLAG_EXTENDABLE 0x01
596#define FR_DBUFF_FLAG_EXTENDED 0x02
620#define fr_dbuff_is_extendable(_status) ((_status) & FR_DBUFF_FLAG_EXTENDABLE)
624#define fr_dbuff_was_extended(_status) ((_status) & FR_DBUFF_FLAG_EXTENDED)
630 size_t remaining,
size_t lowat)
640 if (remaining >= lowat) {
645 if (!
in->extend || !(extended =
in->extend(
in, lowat - remaining)))
goto not_extendable;
649 return remaining + extended;
670#define fr_dbuff_extend_lowat(_status, _dbuff_or_marker, _lowat) \
671 _fr_dbuff_extend_lowat(_status, \
672 fr_dbuff_ptr(_dbuff_or_marker), \
673 fr_dbuff_remaining(_dbuff_or_marker), _lowat)
683#define FR_DBUFF_EXTEND_LOWAT_OR_RETURN(_dbuff_or_marker, _lowat) \
685 size_t _remaining = fr_dbuff_extend_lowat(NULL, _dbuff_or_marker, _lowat); \
686 if (_remaining < _lowat) return -(_lowat - _remaining); \
699#define _FR_DBUFF_EXTEND_LOWAT_POS_OR_RETURN(_pos_p, _dbuff_or_marker, _lowat) \
701 size_t _remaining = _fr_dbuff_extend_lowat(NULL, \
702 fr_dbuff_ptr(_dbuff_or_marker), \
703 fr_dbuff_end(_dbuff_or_marker) - (*(_pos_p)), _lowat); \
704 if (_remaining < _lowat) return -(_lowat - _remaining); \
715#define fr_dbuff_extend(_dbuff) fr_dbuff_extend_lowat(NULL, _dbuff, 1)
753#define fr_dbuff_remaining(_dbuff_or_marker) \
754 ((size_t)(fr_dbuff_end(_dbuff_or_marker) < fr_dbuff_current(_dbuff_or_marker) ? \
755 0 : (fr_dbuff_end(_dbuff_or_marker) - fr_dbuff_current(_dbuff_or_marker))))
771#define FR_DBUFF_REMAINING_RETURN(_dbuff_or_marker, _len) \
772 if ((_len) > fr_dbuff_remaining(_dbuff_or_marker)) return -((_len) - fr_dbuff_remaining(_dbuff_or_marker))
777#define fr_dbuff_used(_dbuff_or_marker) \
778 ((size_t)(fr_dbuff_start(_dbuff_or_marker) > fr_dbuff_current(_dbuff_or_marker) ? \
779 0 : (fr_dbuff_current(_dbuff_or_marker) - fr_dbuff_start(_dbuff_or_marker))))
786#define fr_dbuff_len(_dbuff_or_marker) \
787 ((size_t)(fr_dbuff_end(_dbuff_or_marker) - fr_dbuff_start(_dbuff_or_marker)))
796#define fr_dbuff_behind(_dbuff_or_marker) \
797 (fr_dbuff_current(_dbuff_or_marker) > fr_dbuff_current((_dbuff_or_marker)->parent) ? \
798 0 : fr_dbuff_current((_dbuff_or_marker)->parent) - fr_dbuff_current(_dbuff_or_marker))
806#define fr_dbuff_ahead(_dbuff_or_marker) \
807 (fr_dbuff_current((_dbuff_or_marker)->parent) > fr_dbuff_current(_dbuff_or_marker) ? \
808 0 : fr_dbuff_current(_dbuff_or_marker) - fr_dbuff_current((_dbuff_or_marker)->parent))
868#define fr_dbuff_ptr(_dbuff_or_marker) \
869 _Generic((_dbuff_or_marker), \
870 fr_dbuff_t * : ((fr_dbuff_t *)(_dbuff_or_marker)), \
871 fr_dbuff_marker_t * : (((fr_dbuff_marker_t *)(_dbuff_or_marker))->parent) \
879#define fr_dbuff_ptr_const(_dbuff_or_marker) \
880 _Generic((_dbuff_or_marker), \
881 fr_dbuff_t * : ((fr_dbuff_t const *)(_dbuff_or_marker)), \
882 fr_dbuff_t const * : ((fr_dbuff_t const *)(_dbuff_or_marker)), \
883 fr_dbuff_marker_t * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->parent), \
884 fr_dbuff_marker_t const * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->parent) \
892#define fr_dbuff_buff(_dbuff_or_marker) \
893 _Generic((_dbuff_or_marker), \
894 fr_dbuff_t * : (((fr_dbuff_t const *)(_dbuff_or_marker))->buff), \
895 fr_dbuff_t const * : (((fr_dbuff_t const *)(_dbuff_or_marker))->buff), \
896 fr_dbuff_marker_t * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->parent->buff), \
897 fr_dbuff_marker_t const * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->parent->buff) \
908#define fr_dbuff_start(_dbuff_or_marker) \
909 (_Generic((_dbuff_or_marker), \
910 fr_dbuff_t * : (((fr_dbuff_t const *)(_dbuff_or_marker))->start), \
911 fr_dbuff_t const * : (((fr_dbuff_t const *)(_dbuff_or_marker))->start), \
912 fr_dbuff_marker_t * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->parent->start), \
913 fr_dbuff_marker_t const * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->parent->start) \
921#define fr_dbuff_current(_dbuff_or_marker) \
922 (_Generic((_dbuff_or_marker), \
923 fr_dbuff_t * : (((fr_dbuff_t const *)(_dbuff_or_marker))->p), \
924 fr_dbuff_t const * : (((fr_dbuff_t const *)(_dbuff_or_marker))->p), \
925 fr_dbuff_marker_t * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->p), \
926 fr_dbuff_marker_t const * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->p) \
936#define _fr_dbuff_current_ptr(_dbuff_or_marker) \
937 (_Generic((_dbuff_or_marker), \
938 fr_dbuff_t * : &(((fr_dbuff_t *)(_dbuff_or_marker))->p), \
939 fr_dbuff_marker_t * : &(((fr_dbuff_marker_t *)(_dbuff_or_marker))->p) \
948#define fr_dbuff_end(_dbuff_or_marker) \
949 (_Generic((_dbuff_or_marker), \
950 fr_dbuff_t * : (((fr_dbuff_t const *)(_dbuff_or_marker))->end), \
951 fr_dbuff_t const * : (((fr_dbuff_t const *)(_dbuff_or_marker))->end), \
952 fr_dbuff_marker_t * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->parent->end), \
953 fr_dbuff_marker_t const * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->parent->end) \
971 if (dbuff->adv_parent && dbuff->parent) _fr_dbuff_set_recurse(dbuff->parent, dbuff->adv_parent, p);
990 if (
unlikely(p > dbuff->end))
return -(p - dbuff->end);
991 if (
unlikely(p < dbuff->start))
return 0;
994 if (dbuff->adv_parent && dbuff->parent) _fr_dbuff_set_recurse(dbuff->parent, dbuff->adv_parent, p);
1014#define fr_dbuff_set(_dst, _src) \
1016 _fr_dbuff_current_ptr(_dst), fr_dbuff_ptr(_dst), \
1018 fr_dbuff_t * : fr_dbuff_current((fr_dbuff_t const *)(_src)), \
1019 fr_dbuff_marker_t * : fr_dbuff_current((fr_dbuff_marker_t const *)(_src)), \
1020 uint8_t const * : (uint8_t const *)(_src), \
1021 uint8_t * : (uint8_t const *)(_src), \
1022 size_t : (fr_dbuff_start(_dst) + (uintptr_t)(_src)), \
1023 long : (fr_dbuff_start(_dst) + (uintptr_t)(_src)), \
1024 int : (fr_dbuff_start(_dst) + (uintptr_t)(_src)) \
1032#define FR_DBUFF_SET_RETURN(_dst, _src) FR_DBUFF_RETURN(fr_dbuff_set, _dst, _src)
1045 if (
unlikely(p > dbuff->end))
return -(p - dbuff->end);
1046 if (
unlikely(p < dbuff->start))
return 0;
1050 return dbuff->end - p;
1061#define fr_dbuff_set_end(_dst, _end) \
1063 fr_dbuff_ptr(_dst), \
1065 fr_dbuff_t * : fr_dbuff_current((fr_dbuff_t const *)(_end)), \
1066 fr_dbuff_marker_t * : fr_dbuff_current((fr_dbuff_marker_t const *)(_end)), \
1067 uint8_t const * : (uint8_t const *)(_end), \
1068 uint8_t * : (uint8_t const *)(_end) \
1083#define fr_dbuff_advance(_dbuff_or_marker, _len) \
1084 fr_dbuff_set(_dbuff_or_marker, \
1085 (fr_dbuff_current(_dbuff_or_marker) + \
1087 unsigned char : (size_t)(_len), \
1088 unsigned short : (size_t)(_len), \
1089 unsigned int : (size_t)(_len), \
1090 unsigned long : (size_t)(_len), \
1091 unsigned long long : (size_t)(_len), \
1092 int : (size_t)(_len) \
1099#define FR_DBUFF_ADVANCE_RETURN(_dbuff_or_marker, _len) FR_DBUFF_RETURN(fr_dbuff_advance, _dbuff_or_marker, _len)
1119 if (p > dbuff->end) {
1120 size_t rel = p - dbuff->start;
1122 if (!dbuff->extend) {
1124 return -((dbuff->start + rel) - dbuff->end);
1127 dbuff->extend(dbuff, p - dbuff->end);
1128 if ((dbuff->start + rel) > dbuff->end)
goto oos;
1130 *pos_p = dbuff->start + rel;
1135 if (dbuff->adv_parent && dbuff->parent) _fr_dbuff_set_recurse(dbuff->parent, dbuff->adv_parent, *pos_p);
1150#define fr_dbuff_advance_extend(_dbuff_or_marker, _len) \
1151 _fr_dbuff_advance(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), \
1153 unsigned char : (size_t)(_len), \
1154 unsigned short : (size_t)(_len), \
1155 unsigned int : (size_t)(_len), \
1156 unsigned long : (size_t)(_len), \
1157 unsigned long long : (size_t)(_len), \
1158 int : (size_t)(_len) \
1161#define FR_DBUFF_BIND_EXTEND_RETURN(_dbuff_or_marker, _len) FR_DBUFF_RETURN(fr_dbuff_advance_extend, _dbuff_or_marker, _len)
1166#define fr_dbuff_set_to_start(_dbuff_or_marker) \
1167 fr_dbuff_set(_dbuff_or_marker, fr_dbuff_start(_dbuff_or_marker))
1172#define fr_dbuff_set_to_end(_dbuff_or_marker) \
1173 fr_dbuff_set(_dbuff_or_marker, fr_dbuff_end(_dbuff_or_marker))
1223 m->parent->m = m->next;
1226 memset(m, 0,
sizeof(*m));
1285static inline CC_HINT(always_inline)
size_t _fr_dbuff_safecpy(
uint8_t *o_start,
uint8_t *o_end,
1289 size_t i_len = i_end - i_start;
1291 if (
unlikely((o_end < o_start) || (i_end < i_start)))
return 0;
1293 diff = (o_end - o_start) - (i_len);
1294 if (diff < 0)
return 0;
1296 if ((i_start > o_end) || (i_end < o_start)) {
1297 memcpy(o_start, i_start, i_len);
1299 memmove(o_start, i_start, i_len);
1314 _FR_DBUFF_EXTEND_LOWAT_POS_OR_RETURN(pos_p,
out,
inlen);
1316 return _fr_dbuff_set(pos_p,
out, (*pos_p) + _fr_dbuff_safecpy((*pos_p), (*pos_p) +
inlen,
in,
in +
inlen));
1330 memcpy(&our_in, &
in,
sizeof(our_in));
1331 memcpy(&our_in_p, &in_p,
sizeof(our_in_p));
1333 if (
inlen == SIZE_MAX) {
1334 ext_len = _fr_dbuff_extend_lowat(NULL, our_in,
fr_dbuff_end(our_in) - (*our_in_p),
inlen);
1337 _FR_DBUFF_EXTEND_LOWAT_POS_OR_RETURN(our_in_p, our_in,
inlen);
1339 return _fr_dbuff_in_memcpy(pos_p,
out, *our_in_p,
inlen);
1361#define fr_dbuff_in_memcpy(_dbuff_or_marker, _in, _inlen) \
1363 uint8_t * : _fr_dbuff_in_memcpy(_fr_dbuff_current_ptr(_dbuff_or_marker), \
1364 fr_dbuff_ptr(_dbuff_or_marker), \
1365 (uint8_t const *)(_in), \
1367 uint8_t const * : _fr_dbuff_in_memcpy(_fr_dbuff_current_ptr(_dbuff_or_marker), \
1368 fr_dbuff_ptr(_dbuff_or_marker), \
1369 (uint8_t const *)(_in), \
1371 char * : _fr_dbuff_in_memcpy(_fr_dbuff_current_ptr(_dbuff_or_marker), \
1372 fr_dbuff_ptr(_dbuff_or_marker), \
1373 (uint8_t const *)(_in), \
1374 (size_t)(_inlen) == SIZE_MAX ? strlen((char const *)(_in)) : (_inlen)), \
1375 char const * : _fr_dbuff_in_memcpy(_fr_dbuff_current_ptr(_dbuff_or_marker), \
1376 fr_dbuff_ptr(_dbuff_or_marker), \
1377 (uint8_t const *)(_in), \
1378 (size_t)(_inlen) == SIZE_MAX ? strlen((char const *)(_in)) : (_inlen)), \
1379 fr_dbuff_t * : _fr_dbuff_in_memcpy_dbuff(_fr_dbuff_current_ptr(_dbuff_or_marker), \
1380 fr_dbuff_ptr(_dbuff_or_marker), \
1381 &((fr_dbuff_t const *)(_in))->p, \
1382 ((fr_dbuff_t const *)(_in)), \
1384 fr_dbuff_marker_t * : _fr_dbuff_in_memcpy_dbuff(_fr_dbuff_current_ptr(_dbuff_or_marker), \
1385 fr_dbuff_ptr(_dbuff_or_marker), \
1386 &((fr_dbuff_marker_t const *)(_in))->p, \
1387 ((fr_dbuff_marker_t const *)(_in))->parent, _inlen) \
1393#define FR_DBUFF_IN_MEMCPY_RETURN(_dbuff_or_marker, _in, _inlen) FR_DBUFF_RETURN(fr_dbuff_in_memcpy, _dbuff_or_marker, _in, _inlen)
1409 return _fr_dbuff_set(pos_p,
out, (*pos_p) + _fr_dbuff_safecpy((*pos_p), (*pos_p) +
inlen,
in,
in +
inlen));
1423 ext_len = _fr_dbuff_extend_lowat(NULL, our_in,
fr_dbuff_end(our_in) - (*our_in_p),
inlen);
1426 return _fr_dbuff_in_memcpy_partial(pos_p,
out, (*our_in_p),
inlen);
1449#define fr_dbuff_in_memcpy_partial(_out, _in, _inlen) \
1451 uint8_t * : _fr_dbuff_in_memcpy_partial(_fr_dbuff_current_ptr(_out), _out, (uint8_t const *)(_in), _inlen), \
1452 uint8_t const * : _fr_dbuff_in_memcpy_partial(_fr_dbuff_current_ptr(_out), _out, (uint8_t const *)(_in), _inlen), \
1453 char * : _fr_dbuff_in_memcpy_partial(_fr_dbuff_current_ptr(_out), _out, (uint8_t const *)(_in), _inlen == SIZE_MAX ? strlen((char const *)(_in)) : _inlen), \
1454 char const * : _fr_dbuff_in_memcpy_partial(_fr_dbuff_current_ptr(_out), _out, (uint8_t const *)(_in), _inlen == SIZE_MAX ? strlen((char const *)(_in)) : _inlen), \
1455 fr_dbuff_t * : _fr_dbuff_in_memcpy_partial_dbuff(_fr_dbuff_current_ptr(_out), fr_dbuff_ptr(_out), &((fr_dbuff_t const *)(_in))->p, ((fr_dbuff_t const *)(_in)), _inlen), \
1456 fr_dbuff_marker_t * : _fr_dbuff_in_memcpy_partial_dbuff(_fr_dbuff_current_ptr(_out), fr_dbuff_ptr(_out), &((fr_dbuff_marker_t const *)(_in))->p, ((fr_dbuff_marker_t const *)(_in))->parent, _inlen) \
1466#define fr_dbuff_in_bytes_partial(_dbuff, ...) \
1467 fr_dbuff_in_memcpy_partial(_dbuff, ((uint8_t []){ __VA_ARGS__ }), sizeof((uint8_t []){ __VA_ARGS__ }))
1476#define fr_dbuff_in_bytes(_dbuff_or_marker, ...) \
1477 fr_dbuff_in_memcpy(_dbuff_or_marker, ((uint8_t []){ __VA_ARGS__ }), sizeof((uint8_t []){ __VA_ARGS__ }))
1483#define FR_DBUFF_IN_BYTES_RETURN(_dbuff_or_marker, ...) \
1484 FR_DBUFF_IN_MEMCPY_RETURN(_dbuff_or_marker, ((uint8_t []){ __VA_ARGS__ }), sizeof((uint8_t []){ __VA_ARGS__ }))
1494 _FR_DBUFF_EXTEND_LOWAT_POS_OR_RETURN(pos_p, dbuff,
inlen);
1496 memset((*pos_p), c,
inlen);
1498 return _fr_dbuff_set(pos_p, dbuff, (*pos_p) +
inlen);
1512#define fr_dbuff_memset(_dbuff_or_marker, _c, _inlen) \
1513 _fr_dbuff_memset(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), _c, _inlen)
1519#define FR_DBUFF_MEMSET_RETURN(_dbuff_or_marker, _c, _inlen) FR_DBUFF_RETURN(fr_dbuff_memset, _dbuff_or_marker, _c, _inlen)
1525#define FR_DBUFF_PARSE_INT_DEF(_type) \
1526static inline ssize_t _fr_dbuff_in_##_type(uint8_t **pos_p, fr_dbuff_t *out, _type##_t num) \
1528 fr_assert(!out->is_const); \
1529 _FR_DBUFF_EXTEND_LOWAT_POS_OR_RETURN(pos_p, out, sizeof(_type##_t)); \
1530 fr_nbo_from_##_type((*pos_p), num); \
1531 return _fr_dbuff_set(pos_p, out, (*pos_p) + sizeof(_type##_t)); \
1533FR_DBUFF_PARSE_INT_DEF(uint16)
1534FR_DBUFF_PARSE_INT_DEF(uint32)
1535FR_DBUFF_PARSE_INT_DEF(uint64)
1536FR_DBUFF_PARSE_INT_DEF(int16)
1537FR_DBUFF_PARSE_INT_DEF(int32)
1538FR_DBUFF_PARSE_INT_DEF(int64)
1555 return _fr_dbuff_in_uint32(pos_p,
out, *(
uint32_t *)(&num));
1566 return _fr_dbuff_in_uint64(pos_p,
out, *(uint64_t *)(&num));
1578#define fr_dbuff_in(_dbuff_or_marker, _in) \
1580 int8_t : fr_dbuff_in_bytes(_dbuff_or_marker, (int8_t)_in), \
1581 int16_t : _fr_dbuff_in_int16(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), (int16_t)_in), \
1582 int32_t : _fr_dbuff_in_int32(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), (int32_t)_in), \
1583 int64_t : _fr_dbuff_in_int64(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), (int64_t)_in), \
1584 uint8_t : fr_dbuff_in_bytes(_dbuff_or_marker, (uint8_t)_in), \
1585 uint16_t : _fr_dbuff_in_uint16(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), (uint16_t)_in), \
1586 uint32_t : _fr_dbuff_in_uint32(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), (uint32_t)_in), \
1587 uint64_t : _fr_dbuff_in_uint64(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), (uint64_t)_in), \
1588 float : _fr_dbuff_in_float(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), (float)_in), \
1589 double : _fr_dbuff_in_double(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), (double)_in) \
1596#define FR_DBUFF_IN_RETURN(_dbuff_or_marker, _in) FR_DBUFF_RETURN(fr_dbuff_in, _dbuff_or_marker, _in)
1604 uint8_t swapped[
sizeof(uint64_t)];
1608 if (ret >
sizeof(uint64_t))
return -1;
1612 return _fr_dbuff_in_memcpy(pos_p, dbuff, (swapped + (
sizeof(uint64_t) - ret)), ret);
1623#define fr_dbuff_in_uint64v(_dbuff_or_marker, _num) \
1624 _fr_dbuff_in_uint64v(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), _num)
1630#define FR_DBUFF_IN_UINT64V(_dbuff_or_marker, _num) FR_DBUFF_RETURN(fr_dbuff_in_uint64v, _dbuff_or_marker, _num)
1667#define fr_dbuff_move(_out, _in, _len) \
1671 fr_dbuff_t * : _fr_dbuff_move_dbuff_to_dbuff((fr_dbuff_t *)_out, \
1672 (fr_dbuff_t *)_in, \
1674 fr_dbuff_marker_t * : _fr_dbuff_move_dbuff_marker_to_dbuff((fr_dbuff_t *)_out, \
1675 (fr_dbuff_marker_t *)_in, \
1678 fr_dbuff_marker_t * : \
1680 fr_dbuff_t * : _fr_dbuff_move_dbuff_to_dbuff_marker((fr_dbuff_marker_t *)_out, \
1681 (fr_dbuff_t *)_in, \
1683 fr_dbuff_marker_t * : _fr_dbuff_move_dbuff_marker_to_dbuff_marker((fr_dbuff_marker_t *)_out, \
1684 (fr_dbuff_marker_t *)_in, \
1700 size_t ext_len, to_copy, remaining;
1702 for (remaining = outlen; remaining > 0; remaining -= to_copy) {
1703 to_copy = remaining;
1704 ext_len = _fr_dbuff_extend_lowat(NULL,
in,
fr_dbuff_end(
in) - (*pos_p), 1);
1705 if (ext_len == 0)
return -remaining;
1706 if (ext_len < to_copy) to_copy = ext_len;
1707 out += _fr_dbuff_set(pos_p,
in,
1708 (*pos_p) + _fr_dbuff_safecpy(
out,
out + to_copy, (*pos_p), (*pos_p) + to_copy));
1719 if (outlen == SIZE_MAX) outlen = _fr_dbuff_extend_lowat(NULL,
out,
fr_dbuff_end(
out) - (*out_p), outlen);
1721 return _fr_dbuff_out_memcpy((*out_p), pos_p,
in, outlen);
1743#define fr_dbuff_out_memcpy(_out, _dbuff_or_marker, _outlen) \
1745 uint8_t * : _fr_dbuff_out_memcpy((uint8_t *)(_out), \
1746 _fr_dbuff_current_ptr(_dbuff_or_marker), \
1747 fr_dbuff_ptr(_dbuff_or_marker), \
1749 fr_dbuff_t * : _fr_dbuff_out_memcpy_dbuff(_fr_dbuff_current_ptr((fr_dbuff_t *)_out), \
1750 fr_dbuff_ptr((fr_dbuff_t *)(_out)), \
1751 _fr_dbuff_current_ptr(_dbuff_or_marker), \
1752 fr_dbuff_ptr(_dbuff_or_marker), _outlen), \
1753 fr_dbuff_marker_t * : _fr_dbuff_out_memcpy_dbuff(_fr_dbuff_current_ptr((fr_dbuff_marker_t *)_out), \
1754 fr_dbuff_ptr((fr_dbuff_marker_t *)(_out)), \
1755 _fr_dbuff_current_ptr(_dbuff_or_marker), \
1756 fr_dbuff_ptr(_dbuff_or_marker), _outlen) \
1763#define FR_DBUFF_OUT_MEMCPY_RETURN(_out, _dbuff_or_marker, _outlen) FR_DBUFF_RETURN(fr_dbuff_out_memcpy, _out, _dbuff_or_marker, _outlen)
1769#define FR_DBUFF_OUT_DEF(_type) \
1770static inline ssize_t _fr_dbuff_out_##_type(_type##_t *out, uint8_t **pos_p, fr_dbuff_t *in) \
1773 FR_DBUFF_EXTEND_LOWAT_OR_RETURN(in, sizeof(_type##_t)); \
1774 *out = fr_nbo_to_##_type((*pos_p)); \
1775 return _fr_dbuff_set(pos_p, in, (*pos_p) + sizeof(_type##_t)); \
1778FR_DBUFF_OUT_DEF(uint16)
1779FR_DBUFF_OUT_DEF(uint32)
1780FR_DBUFF_OUT_DEF(uint64)
1781FR_DBUFF_OUT_DEF(int16)
1782FR_DBUFF_OUT_DEF(int32)
1783FR_DBUFF_OUT_DEF(int64)
1785#define FR_DBUFF_OUT_DEF_NO_SWAP(_type) \
1786static inline ssize_t _fr_dbuff_out_##_type(_type##_t *out, uint8_t **pos_p, fr_dbuff_t *in) \
1789 FR_DBUFF_EXTEND_LOWAT_OR_RETURN(in, sizeof(_type##_t)); \
1791 return _fr_dbuff_set(pos_p, in, (*pos_p) + sizeof(_type##_t)); \
1794FR_DBUFF_OUT_DEF_NO_SWAP(uint8)
1795FR_DBUFF_OUT_DEF_NO_SWAP(int8)
1810#define fr_dbuff_out(_out, _dbuff_or_marker) \
1812 uint8_t * : _fr_dbuff_out_uint8((uint8_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1813 uint16_t * : _fr_dbuff_out_uint16((uint16_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1814 uint32_t * : _fr_dbuff_out_uint32((uint32_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1815 uint64_t * : _fr_dbuff_out_uint64((uint64_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1816 int8_t * : _fr_dbuff_out_int8((int8_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1817 int16_t * : _fr_dbuff_out_int16((int16_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1818 int32_t * : _fr_dbuff_out_int32((int32_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1819 int64_t * : _fr_dbuff_out_int64((int64_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1820 float * : _fr_dbuff_out_uint32((uint32_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1821 double * : _fr_dbuff_out_uint64((uint64_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)) \
1828#ifndef STATIC_ANALYZER
1829#define FR_DBUFF_OUT_RETURN(_out, _dbuff_or_marker) FR_DBUFF_RETURN(fr_dbuff_out, _out, _dbuff_or_marker)
1831#define FR_DBUFF_OUT_RETURN(_out, _dbuff_or_marker) do { *_out = 0; FR_DBUFF_RETURN(fr_dbuff_out, _out, _dbuff_or_marker); } while (0)
1841 fr_assert(length > 0 && length <=
sizeof(uint64_t));
1844 slen = _fr_dbuff_out_memcpy(((
uint8_t *) num) + (8 - length), pos_p, dbuff, length);
1845 if (slen <= 0)
return slen;
1862#define fr_dbuff_out_uint64v(_num, _dbuff_or_marker, _len) \
1863 _fr_dbuff_out_uint64v(_num, _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), _len)
1869#define FR_DBUFF_OUT_UINT64V_RETURN(_num, _dbuff_or_marker, _len) FR_DBUFF_RETURN(fr_dbuff_out_uint64v, _num, _dbuff_or_marker, _len)
1879 fr_assert(length > 0 && length <=
sizeof(uint64_t));
1882 slen = _fr_dbuff_out_memcpy(((
uint8_t *) num) + (8 - length), pos_p, dbuff, length);
1883 if (slen <= 0)
return slen;
1885 if (msb & 0x80) memset(((
uint8_t *)num), 0xff,
sizeof(*num) - length);
1902#define fr_dbuff_out_int64v(_num, _dbuff_or_marker, _len) \
1903 _fr_dbuff_out_int64v(_num, _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), _len)
1909#define FR_DBUFF_OUT_INT64V_RETURN(_num, _dbuff_or_marker, _len) FR_DBUFF_RETURN(fr_dbuff_out_int64v, _num, _dbuff_or_marker, _len)
#define UNCONST(_type, _ptr)
Remove const qualification from a pointer.
struct fr_dbuff_s fr_dbuff_t
A dbuff.
#define fr_dbuff_behind(_dbuff_or_marker)
How many bytes the dbuff or marker is behind its parent.
#define FR_DBUFF_FLAG_EXTENDED
Flag indicating that during the last extend call the dbuff was extended.
fr_dbuff_uctx_talloc_t tctx
Thread local tctx.
int fr_dbuff_reset_talloc(fr_dbuff_t *dbuff)
Reset a talloced buffer to its initial length, clearing any data stored.
static void fr_dbuff_free_talloc(fr_dbuff_t *dbuff)
Free the talloc buffer associated with a dbuff.
static fr_dbuff_t * fr_dbuff_init_fd(fr_dbuff_t *dbuff, fr_dbuff_uctx_fd_t *fctx, uint8_t *buff, size_t len, int fd, size_t max)
Initialise a special dbuff which automatically reads in more data as the buffer is exhausted.
static int _dbuff_thread_local_free(void *dbtl)
static size_t fr_dbuff_marker_release_behind(fr_dbuff_marker_t *m)
Trims the linked list back to the specified pointer and return how many bytes marker was behind p.
struct fr_dbuff_marker_s fr_dbuff_marker_t
A position marker associated with a dbuff.
fr_dbuff_extend_status_t
Whether the buffer is currently extendable and whether it was extended.
@ FR_DBUFF_NOT_EXTENDABLE
dbuff cannot be extended
@ FR_DBUFF_EXTENDED
dbuff was extended in the last extend call but cannot be extended again
@ FR_DBUFF_EXTENDABLE
dbuff can be extended
@ FR_DBUFF_EXTENDABLE_EXTENDED
dbuff was extended in the last extend call and may be extended again
size_t _fr_dbuff_extend_talloc(fr_dbuff_t *dbuff, size_t extension)
Reallocate the current buffer.
#define fr_dbuff_end(_dbuff_or_marker)
Return the current 'end' position of a dbuff or marker.
#define FR_DBUFF_ADV_PARENT_CURRENT
Advance current position of parent.
int fr_dbuff_trim_talloc(fr_dbuff_t *dbuff, size_t len)
Trim a talloced dbuff to the minimum length required to represent the contained string.
static uint8_t * fr_dbuff_marker(fr_dbuff_marker_t *m, fr_dbuff_t *dbuff)
Initialises a new marker pointing to the 'current' position of the dbuff.
#define FR_DBUFF_FLAG_EXTENDABLE
Flag indicating a dbuff is extendable.
size_t _fr_dbuff_extend_fd(fr_dbuff_t *dbuff, size_t extension)
Refresh the buffer with more data from the file.
fr_dbuff_t dbuff
Thread local dbuff.
void fr_dbuff_update(fr_dbuff_t *dbuff, uint8_t *new_buff, size_t new_len)
Update all markers and pointers in the set of dbuffs to point to new_buff.
static void fr_dbuff_marker_release(fr_dbuff_marker_t *m)
Releases the specified marker and any markers added before it.
#define fr_dbuff_is_extendable(_status)
Check if a dbuff can be extended again.
#define fr_dbuff_ahead(_dbuff_or_marker)
How many bytes the dbuff or marker is ahead of its parent.
static size_t fr_dbuff_marker_release_ahead(fr_dbuff_marker_t *m)
Trims the linked list back to the specified pointer and return how many bytes marker was ahead of p.
#define FR_DBUFF_ADV_PARENT_END
Advance end pointer of parent.
static fr_dbuff_t * fr_dbuff_init_talloc(TALLOC_CTX *ctx, fr_dbuff_t *dbuff, fr_dbuff_uctx_talloc_t *tctx, size_t init, size_t max)
Initialise a special dbuff which automatically extends as additional data is written.
size_t(* fr_dbuff_extend_t)(fr_dbuff_t *dbuff, size_t req_extension)
dbuff extension callback
size_t fr_dbuff_shift(fr_dbuff_t *dbuff, size_t shift)
Shift the contents of the dbuff, returning the number of bytes we managed to shift.
Structure to encapsulate a thread local dbuff information.
#define ROUND_UP_DIV(_x, _y)
Get the ceiling value of integer division.
static uint8_t fr_high_bit_pos(uint64_t num)
Find the highest order high bit in an unsigned 64 bit integer.
static uint64_t fr_nbo_to_uint64(uint8_t const data[static sizeof(uint64_t)])
Read an unsigned 64bit integer from wire format (big endian)
static void fr_nbo_from_uint64(uint8_t out[static sizeof(uint64_t)], uint64_t num)
Write out an unsigned 64bit integer in wire format (big endian)
#define fr_nbo_to_int64(_x)
static char buff[sizeof("18446744073709551615")+3]
#define fr_strerror_printf(_fmt,...)
Log to thread local error buffer.
static size_t char fr_sbuff_t size_t inlen
static size_t char ** out