26RCSIDH(dbuff_h,
"$Id: d3eb3a58c12edd930c35eff1b975bab4567740ce $")
33#include <freeradius-devel/missing.h>
34#include <freeradius-devel/util/debug.h>
35#include <freeradius-devel/util/nbo.h>
113struct fr_dbuff_marker_s {
124#define FR_DBUFF_ADV_PARENT_CURRENT 0x01
126#define FR_DBUFF_ADV_PARENT_END 0x02
161 unsigned int is_const : 1;
162 unsigned int adv_parent : 2;
184#define FR_DBUFF_RETURN(_func, ...) \
186 ssize_t _slen = _func(__VA_ARGS__ ); \
187 if (_slen < 0) return _slen; \
196#define FR_DBUFF_ERROR_OFFSET(_slen, _offset) ((_slen < INT32_MAX) ? _slen : _slen - (ssize_t) _offset)
208#define _FR_DBUFF(_dbuff_or_marker, _start, _adv_parent) \
210 .buff = fr_dbuff_buff(_dbuff_or_marker), \
212 .end = fr_dbuff_end(_dbuff_or_marker), \
213 .p = fr_dbuff_current(_dbuff_or_marker), \
214 .is_const = fr_dbuff_ptr(_dbuff_or_marker)->is_const, \
215 .adv_parent = (_adv_parent), \
216 .shifted = fr_dbuff_ptr(_dbuff_or_marker)->shifted, \
217 .extend = fr_dbuff_ptr(_dbuff_or_marker)->extend, \
218 .uctx = fr_dbuff_ptr(_dbuff_or_marker)->uctx, \
219 .parent = fr_dbuff_ptr(_dbuff_or_marker) \
230#define FR_DBUFF(_dbuff_or_marker) _FR_DBUFF(_dbuff_or_marker, fr_dbuff_current(_dbuff_or_marker), 0x00)
239#define FR_DBUFF_ABS(_dbuff_or_marker) _FR_DBUFF(_dbuff_or_marker, fr_dbuff_start(_dbuff_or_marker), 0x00)
248#define FR_DBUFF_BIND_CURRENT(_dbuff_or_marker) _FR_DBUFF(_dbuff_or_marker, fr_dbuff_current(_dbuff_or_marker), FR_DBUFF_ADV_PARENT_CURRENT)
257#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)
268#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)
275#define _FR_DBUFF_MAX(_dbuff_or_marker, _max, _adv_parent) \
277 .buff = fr_dbuff_buff(_dbuff_or_marker), \
278 .start = fr_dbuff_current(_dbuff_or_marker), \
279 .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))), \
280 .p = fr_dbuff_current(_dbuff_or_marker), \
281 .is_const = fr_dbuff_ptr(_dbuff_or_marker)->is_const, \
282 .adv_parent = _adv_parent, \
283 .shifted = fr_dbuff_ptr(_dbuff_or_marker)->shifted, \
286 .parent = fr_dbuff_ptr(_dbuff_or_marker) \
309#define FR_DBUFF_MAX(_dbuff_or_marker, _max) _FR_DBUFF_MAX(_dbuff_or_marker, _max, 0x00)
326#define FR_DBUFF_MAX_BIND_CURRENT(_dbuff_or_marker, _max) _FR_DBUFF_MAX(_dbuff_or_marker, _max, FR_DBUFF_ADV_PARENT_CURRENT)
332#if defined(__GNUC__) && __GNUC__ >= 11
344 if (
unlikely(end < start)) end = start;
362#define fr_dbuff_init(_out, _start, _len_or_end) \
363_fr_dbuff_init(_out, \
364 (uint8_t const *)(_start), \
365 _Generic((_len_or_end), \
366 size_t : (uint8_t const *)(_start) + (size_t)(_len_or_end), \
367 long : (uint8_t const *)(_start) + (size_t)(_len_or_end), \
368 int : (uint8_t const *)(_start) + (size_t)(_len_or_end), \
369 unsigned int : (uint8_t const *)(_start) + (size_t)(_len_or_end), \
370 uint8_t * : (uint8_t const *)(_len_or_end), \
371 uint8_t const * : (uint8_t const *)(_len_or_end), \
372 char * : (uint8_t const *)(_len_or_end), \
373 char const * : (uint8_t const *)(_len_or_end) \
377 uint8_t const * : true, \
379 char const * : true \
381#if defined(__GNUC__) && __GNUC__ >= 11
385#define FR_DBUFF_INIT(_out, _start, _len_or_end) do { \
386 fr_dbuff_init(_out, _start, _len_or_end); \
387 *(unsigned char *) _start = '\0'; \
406} fr_dbuff_uctx_talloc_t;
420 fr_dbuff_t *dbuff, fr_dbuff_uctx_talloc_t *tctx,
421 size_t init,
size_t max)
425 *tctx = (fr_dbuff_uctx_talloc_t){
442 memset(dbuff, 0,
sizeof(*dbuff));
463 TALLOC_FREE(dbuff->buff);
496 *fctx = (fr_dbuff_uctx_fd_t){
499 .buff_end =
buff + len
522#define FR_DBUFF_TMP(_start, _len_or_end) \
524 .buff_i = (uint8_t const *)(_start), \
525 .start_i = (uint8_t const *)(_start), \
526 .end_i = _Generic((_len_or_end), \
527 size_t : (uint8_t const *)(_start) + (size_t)(_len_or_end), \
528 long : (uint8_t const *)(_start) + (size_t)(_len_or_end), \
529 int : (uint8_t const *)(_start) + (size_t)(_len_or_end), \
530 unsigned int : (uint8_t const *)(_start) + (size_t)(_len_or_end), \
531 uint8_t * : (uint8_t const *)(_len_or_end), \
532 uint8_t const * : (uint8_t const *)(_len_or_end), \
533 char * : (uint8_t const *)(_len_or_end), \
534 char const * : (uint8_t const *)(_len_or_end) \
537 .is_const = _Generic((_start), \
539 uint8_t const * : true, \
541 char const * : true \
564#define FR_DBUFF_TALLOC_THREAD_LOCAL(_out, _init, _max) \
566 static _Thread_local fr_dbuff_thread_local_t *_dbuff_t_local; \
567 if (!_dbuff_t_local) { \
568 fr_dbuff_thread_local_t *dbtl = talloc_zero(NULL, fr_dbuff_thread_local_t); \
569 fr_dbuff_init_talloc(dbtl, &dbtl->dbuff, &dbtl->tctx, _init, _max); \
570 fr_atexit_thread_local(_dbuff_t_local, _dbuff_thread_local_free, dbtl); \
571 *(_out) = &_dbuff_t_local->dbuff; \
573 fr_dbuff_reset_talloc(&_dbuff_t_local->dbuff); \
574 *(_out) = &_dbuff_t_local->dbuff; \
590#define FR_DBUFF_FLAG_EXTENDABLE 0x01
594#define FR_DBUFF_FLAG_EXTENDED 0x02
618#define fr_dbuff_is_extendable(_status) ((_status) & FR_DBUFF_FLAG_EXTENDABLE)
622#define fr_dbuff_was_extended(_status) ((_status) & FR_DBUFF_FLAG_EXTENDED)
628 size_t remaining,
size_t lowat)
638 if (remaining >= lowat) {
643 if (!
in->extend || !(extended =
in->extend(
in, lowat - remaining)))
goto not_extendable;
647 return remaining + extended;
668#define fr_dbuff_extend_lowat(_status, _dbuff_or_marker, _lowat) \
669 _fr_dbuff_extend_lowat(_status, \
670 fr_dbuff_ptr(_dbuff_or_marker), \
671 fr_dbuff_remaining(_dbuff_or_marker), _lowat)
681#define FR_DBUFF_EXTEND_LOWAT_OR_RETURN(_dbuff_or_marker, _lowat) \
683 size_t _remaining = fr_dbuff_extend_lowat(NULL, _dbuff_or_marker, _lowat); \
684 if (_remaining < _lowat) return -(_lowat - _remaining); \
697#define _FR_DBUFF_EXTEND_LOWAT_POS_OR_RETURN(_pos_p, _dbuff_or_marker, _lowat) \
699 size_t _remaining = _fr_dbuff_extend_lowat(NULL, \
700 fr_dbuff_ptr(_dbuff_or_marker), \
701 fr_dbuff_end(_dbuff_or_marker) - (*(_pos_p)), _lowat); \
702 if (_remaining < _lowat) return -(_lowat - _remaining); \
713#define fr_dbuff_extend(_dbuff) fr_dbuff_extend_lowat(NULL, _dbuff, 1)
751#define fr_dbuff_remaining(_dbuff_or_marker) \
752 ((size_t)(fr_dbuff_end(_dbuff_or_marker) < fr_dbuff_current(_dbuff_or_marker) ? \
753 0 : (fr_dbuff_end(_dbuff_or_marker) - fr_dbuff_current(_dbuff_or_marker))))
769#define FR_DBUFF_REMAINING_RETURN(_dbuff_or_marker, _len) \
770 if ((_len) > fr_dbuff_remaining(_dbuff_or_marker)) return -((_len) - fr_dbuff_remaining(_dbuff_or_marker))
775#define fr_dbuff_used(_dbuff_or_marker) \
776 ((size_t)(fr_dbuff_start(_dbuff_or_marker) > fr_dbuff_current(_dbuff_or_marker) ? \
777 0 : (fr_dbuff_current(_dbuff_or_marker) - fr_dbuff_start(_dbuff_or_marker))))
784#define fr_dbuff_len(_dbuff_or_marker) \
785 ((size_t)(fr_dbuff_end(_dbuff_or_marker) - fr_dbuff_start(_dbuff_or_marker)))
794#define fr_dbuff_behind(_dbuff_or_marker) \
795 (fr_dbuff_current(_dbuff_or_marker) > fr_dbuff_current((_dbuff_or_marker)->parent) ? \
796 0 : fr_dbuff_current((_dbuff_or_marker)->parent) - fr_dbuff_current(_dbuff_or_marker))
804#define fr_dbuff_ahead(_dbuff_or_marker) \
805 (fr_dbuff_current((_dbuff_or_marker)->parent) > fr_dbuff_current(_dbuff_or_marker) ? \
806 0 : fr_dbuff_current(_dbuff_or_marker) - fr_dbuff_current((_dbuff_or_marker)->parent))
866#define fr_dbuff_ptr(_dbuff_or_marker) \
867 _Generic((_dbuff_or_marker), \
868 fr_dbuff_t * : ((fr_dbuff_t *)(_dbuff_or_marker)), \
869 fr_dbuff_marker_t * : (((fr_dbuff_marker_t *)(_dbuff_or_marker))->parent) \
877#define fr_dbuff_ptr_const(_dbuff_or_marker) \
878 _Generic((_dbuff_or_marker), \
879 fr_dbuff_t * : ((fr_dbuff_t const *)(_dbuff_or_marker)), \
880 fr_dbuff_t const * : ((fr_dbuff_t const *)(_dbuff_or_marker)), \
881 fr_dbuff_marker_t * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->parent), \
882 fr_dbuff_marker_t const * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->parent) \
890#define fr_dbuff_buff(_dbuff_or_marker) \
891 _Generic((_dbuff_or_marker), \
892 fr_dbuff_t * : (((fr_dbuff_t const *)(_dbuff_or_marker))->buff), \
893 fr_dbuff_t const * : (((fr_dbuff_t const *)(_dbuff_or_marker))->buff), \
894 fr_dbuff_marker_t * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->parent->buff), \
895 fr_dbuff_marker_t const * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->parent->buff) \
906#define fr_dbuff_start(_dbuff_or_marker) \
907 (_Generic((_dbuff_or_marker), \
908 fr_dbuff_t * : (((fr_dbuff_t const *)(_dbuff_or_marker))->start), \
909 fr_dbuff_t const * : (((fr_dbuff_t const *)(_dbuff_or_marker))->start), \
910 fr_dbuff_marker_t * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->parent->start), \
911 fr_dbuff_marker_t const * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->parent->start) \
919#define fr_dbuff_current(_dbuff_or_marker) \
920 (_Generic((_dbuff_or_marker), \
921 fr_dbuff_t * : (((fr_dbuff_t const *)(_dbuff_or_marker))->p), \
922 fr_dbuff_t const * : (((fr_dbuff_t const *)(_dbuff_or_marker))->p), \
923 fr_dbuff_marker_t * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->p), \
924 fr_dbuff_marker_t const * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->p) \
934#define _fr_dbuff_current_ptr(_dbuff_or_marker) \
935 (_Generic((_dbuff_or_marker), \
936 fr_dbuff_t * : &(((fr_dbuff_t *)(_dbuff_or_marker))->p), \
937 fr_dbuff_marker_t * : &(((fr_dbuff_marker_t *)(_dbuff_or_marker))->p) \
946#define fr_dbuff_end(_dbuff_or_marker) \
947 (_Generic((_dbuff_or_marker), \
948 fr_dbuff_t * : (((fr_dbuff_t const *)(_dbuff_or_marker))->end), \
949 fr_dbuff_t const * : (((fr_dbuff_t const *)(_dbuff_or_marker))->end), \
950 fr_dbuff_marker_t * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->parent->end), \
951 fr_dbuff_marker_t const * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->parent->end) \
969 if (dbuff->adv_parent && dbuff->parent) _fr_dbuff_set_recurse(dbuff->parent, dbuff->adv_parent, p);
988 if (
unlikely(p > dbuff->end))
return -(p - dbuff->end);
989 if (
unlikely(p < dbuff->start))
return 0;
992 if (dbuff->adv_parent && dbuff->parent) _fr_dbuff_set_recurse(dbuff->parent, dbuff->adv_parent, p);
1012#define fr_dbuff_set(_dst, _src) \
1014 _fr_dbuff_current_ptr(_dst), fr_dbuff_ptr(_dst), \
1016 fr_dbuff_t * : fr_dbuff_current((fr_dbuff_t const *)(_src)), \
1017 fr_dbuff_marker_t * : fr_dbuff_current((fr_dbuff_marker_t const *)(_src)), \
1018 uint8_t const * : (uint8_t const *)(_src), \
1019 uint8_t * : (uint8_t const *)(_src), \
1020 size_t : (fr_dbuff_start(_dst) + (uintptr_t)(_src)), \
1021 long : (fr_dbuff_start(_dst) + (uintptr_t)(_src)), \
1022 int : (fr_dbuff_start(_dst) + (uintptr_t)(_src)) \
1030#define FR_DBUFF_SET_RETURN(_dst, _src) FR_DBUFF_RETURN(fr_dbuff_set, _dst, _src)
1043 if (
unlikely(p > dbuff->end))
return -(p - dbuff->end);
1044 if (
unlikely(p < dbuff->start))
return 0;
1048 return dbuff->end - p;
1059#define fr_dbuff_set_end(_dst, _end) \
1061 fr_dbuff_ptr(_dst), \
1063 fr_dbuff_t * : fr_dbuff_current((fr_dbuff_t const *)(_end)), \
1064 fr_dbuff_marker_t * : fr_dbuff_current((fr_dbuff_marker_t const *)(_end)), \
1065 uint8_t const * : (uint8_t const *)(_end), \
1066 uint8_t * : (uint8_t const *)(_end) \
1081#define fr_dbuff_advance(_dbuff_or_marker, _len) \
1082 fr_dbuff_set(_dbuff_or_marker, \
1083 (fr_dbuff_current(_dbuff_or_marker) + \
1085 unsigned char : (size_t)(_len), \
1086 unsigned short : (size_t)(_len), \
1087 unsigned int : (size_t)(_len), \
1088 unsigned long : (size_t)(_len), \
1089 unsigned long long : (size_t)(_len), \
1090 int : (size_t)(_len) \
1097#define FR_DBUFF_ADVANCE_RETURN(_dbuff_or_marker, _len) FR_DBUFF_RETURN(fr_dbuff_advance, _dbuff_or_marker, _len)
1117 if (p > dbuff->end) {
1118 size_t rel = p - dbuff->start;
1120 if (!dbuff->extend) {
1122 return -((dbuff->start + rel) - dbuff->end);
1125 dbuff->extend(dbuff, p - dbuff->end);
1126 if ((dbuff->start + rel) > dbuff->end)
goto oos;
1128 *pos_p = dbuff->start + rel;
1133 if (dbuff->adv_parent && dbuff->parent) _fr_dbuff_set_recurse(dbuff->parent, dbuff->adv_parent, *pos_p);
1148#define fr_dbuff_advance_extend(_dbuff_or_marker, _len) \
1149 _fr_dbuff_advance(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), \
1151 unsigned char : (size_t)(_len), \
1152 unsigned short : (size_t)(_len), \
1153 unsigned int : (size_t)(_len), \
1154 unsigned long : (size_t)(_len), \
1155 unsigned long long : (size_t)(_len), \
1156 int : (size_t)(_len) \
1159#define FR_DBUFF_BIND_EXTEND_RETURN(_dbuff_or_marker, _len) FR_DBUFF_RETURN(fr_dbuff_advance_extend, _dbuff_or_marker, _len)
1164#define fr_dbuff_set_to_start(_dbuff_or_marker) \
1165 fr_dbuff_set(_dbuff_or_marker, fr_dbuff_start(_dbuff_or_marker))
1170#define fr_dbuff_set_to_end(_dbuff_or_marker) \
1171 fr_dbuff_set(_dbuff_or_marker, fr_dbuff_end(_dbuff_or_marker))
1221 m->parent->m = m->next;
1224 memset(m, 0,
sizeof(*m));
1283static inline CC_HINT(always_inline)
size_t _fr_dbuff_safecpy(
uint8_t *o_start,
uint8_t *o_end,
1287 size_t i_len = i_end - i_start;
1289 if (
unlikely((o_end < o_start) || (i_end < i_start)))
return 0;
1291 diff = (o_end - o_start) - (i_len);
1292 if (diff < 0)
return 0;
1294 if ((i_start > o_end) || (i_end < o_start)) {
1295 memcpy(o_start, i_start, i_len);
1297 memmove(o_start, i_start, i_len);
1312 _FR_DBUFF_EXTEND_LOWAT_POS_OR_RETURN(pos_p,
out,
inlen);
1314 return _fr_dbuff_set(pos_p,
out, (*pos_p) + _fr_dbuff_safecpy((*pos_p), (*pos_p) +
inlen,
in,
in +
inlen));
1328 memcpy(&our_in, &
in,
sizeof(our_in));
1329 memcpy(&our_in_p, &in_p,
sizeof(our_in_p));
1331 if (
inlen == SIZE_MAX) {
1332 ext_len = _fr_dbuff_extend_lowat(NULL, our_in,
fr_dbuff_end(our_in) - (*our_in_p),
inlen);
1335 _FR_DBUFF_EXTEND_LOWAT_POS_OR_RETURN(our_in_p, our_in,
inlen);
1337 return _fr_dbuff_in_memcpy(pos_p,
out, *our_in_p,
inlen);
1359#define fr_dbuff_in_memcpy(_dbuff_or_marker, _in, _inlen) \
1361 uint8_t * : _fr_dbuff_in_memcpy(_fr_dbuff_current_ptr(_dbuff_or_marker), \
1362 fr_dbuff_ptr(_dbuff_or_marker), \
1363 (uint8_t const *)(_in), \
1365 uint8_t const * : _fr_dbuff_in_memcpy(_fr_dbuff_current_ptr(_dbuff_or_marker), \
1366 fr_dbuff_ptr(_dbuff_or_marker), \
1367 (uint8_t const *)(_in), \
1369 char * : _fr_dbuff_in_memcpy(_fr_dbuff_current_ptr(_dbuff_or_marker), \
1370 fr_dbuff_ptr(_dbuff_or_marker), \
1371 (uint8_t const *)(_in), \
1372 (size_t)(_inlen) == SIZE_MAX ? strlen((char const *)(_in)) : (_inlen)), \
1373 char const * : _fr_dbuff_in_memcpy(_fr_dbuff_current_ptr(_dbuff_or_marker), \
1374 fr_dbuff_ptr(_dbuff_or_marker), \
1375 (uint8_t const *)(_in), \
1376 (size_t)(_inlen) == SIZE_MAX ? strlen((char const *)(_in)) : (_inlen)), \
1377 fr_dbuff_t * : _fr_dbuff_in_memcpy_dbuff(_fr_dbuff_current_ptr(_dbuff_or_marker), \
1378 fr_dbuff_ptr(_dbuff_or_marker), \
1379 &((fr_dbuff_t const *)(_in))->p, \
1380 ((fr_dbuff_t const *)(_in)), \
1382 fr_dbuff_marker_t * : _fr_dbuff_in_memcpy_dbuff(_fr_dbuff_current_ptr(_dbuff_or_marker), \
1383 fr_dbuff_ptr(_dbuff_or_marker), \
1384 &((fr_dbuff_marker_t const *)(_in))->p, \
1385 ((fr_dbuff_marker_t const *)(_in))->parent, _inlen) \
1391#define FR_DBUFF_IN_MEMCPY_RETURN(_dbuff_or_marker, _in, _inlen) FR_DBUFF_RETURN(fr_dbuff_in_memcpy, _dbuff_or_marker, _in, _inlen)
1407 return _fr_dbuff_set(pos_p,
out, (*pos_p) + _fr_dbuff_safecpy((*pos_p), (*pos_p) +
inlen,
in,
in +
inlen));
1421 ext_len = _fr_dbuff_extend_lowat(NULL, our_in,
fr_dbuff_end(our_in) - (*our_in_p),
inlen);
1424 return _fr_dbuff_in_memcpy_partial(pos_p,
out, (*our_in_p),
inlen);
1447#define fr_dbuff_in_memcpy_partial(_out, _in, _inlen) \
1449 uint8_t * : _fr_dbuff_in_memcpy_partial(_fr_dbuff_current_ptr(_out), _out, (uint8_t const *)(_in), _inlen), \
1450 uint8_t const * : _fr_dbuff_in_memcpy_partial(_fr_dbuff_current_ptr(_out), _out, (uint8_t const *)(_in), _inlen), \
1451 char * : _fr_dbuff_in_memcpy_partial(_fr_dbuff_current_ptr(_out), _out, (uint8_t const *)(_in), _inlen == SIZE_MAX ? strlen((char const *)(_in)) : _inlen), \
1452 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), \
1453 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), \
1454 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) \
1464#define fr_dbuff_in_bytes_partial(_dbuff, ...) \
1465 fr_dbuff_in_memcpy_partial(_dbuff, ((uint8_t []){ __VA_ARGS__ }), sizeof((uint8_t []){ __VA_ARGS__ }))
1474#define fr_dbuff_in_bytes(_dbuff_or_marker, ...) \
1475 fr_dbuff_in_memcpy(_dbuff_or_marker, ((uint8_t []){ __VA_ARGS__ }), sizeof((uint8_t []){ __VA_ARGS__ }))
1481#define FR_DBUFF_IN_BYTES_RETURN(_dbuff_or_marker, ...) \
1482 FR_DBUFF_IN_MEMCPY_RETURN(_dbuff_or_marker, ((uint8_t []){ __VA_ARGS__ }), sizeof((uint8_t []){ __VA_ARGS__ }))
1492 _FR_DBUFF_EXTEND_LOWAT_POS_OR_RETURN(pos_p, dbuff,
inlen);
1494 memset((*pos_p), c,
inlen);
1496 return _fr_dbuff_set(pos_p, dbuff, (*pos_p) +
inlen);
1510#define fr_dbuff_memset(_dbuff_or_marker, _c, _inlen) \
1511 _fr_dbuff_memset(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), _c, _inlen)
1517#define FR_DBUFF_MEMSET_RETURN(_dbuff_or_marker, _c, _inlen) FR_DBUFF_RETURN(fr_dbuff_memset, _dbuff_or_marker, _c, _inlen)
1523#define FR_DBUFF_PARSE_INT_DEF(_type) \
1524static inline ssize_t _fr_dbuff_in_##_type(uint8_t **pos_p, fr_dbuff_t *out, _type##_t num) \
1526 fr_assert(!out->is_const); \
1527 _FR_DBUFF_EXTEND_LOWAT_POS_OR_RETURN(pos_p, out, sizeof(_type##_t)); \
1528 fr_nbo_from_##_type((*pos_p), num); \
1529 return _fr_dbuff_set(pos_p, out, (*pos_p) + sizeof(_type##_t)); \
1531FR_DBUFF_PARSE_INT_DEF(uint16)
1532FR_DBUFF_PARSE_INT_DEF(uint32)
1533FR_DBUFF_PARSE_INT_DEF(uint64)
1534FR_DBUFF_PARSE_INT_DEF(int16)
1535FR_DBUFF_PARSE_INT_DEF(int32)
1536FR_DBUFF_PARSE_INT_DEF(int64)
1553 return _fr_dbuff_in_uint32(pos_p,
out, *(
uint32_t *)(&num));
1564 return _fr_dbuff_in_uint64(pos_p,
out, *(uint64_t *)(&num));
1576#define fr_dbuff_in(_dbuff_or_marker, _in) \
1578 int8_t : fr_dbuff_in_bytes(_dbuff_or_marker, (int8_t)_in), \
1579 int16_t : _fr_dbuff_in_int16(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), (int16_t)_in), \
1580 int32_t : _fr_dbuff_in_int32(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), (int32_t)_in), \
1581 int64_t : _fr_dbuff_in_int64(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), (int64_t)_in), \
1582 uint8_t : fr_dbuff_in_bytes(_dbuff_or_marker, (uint8_t)_in), \
1583 uint16_t : _fr_dbuff_in_uint16(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), (uint16_t)_in), \
1584 uint32_t : _fr_dbuff_in_uint32(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), (uint32_t)_in), \
1585 uint64_t : _fr_dbuff_in_uint64(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), (uint64_t)_in), \
1586 float : _fr_dbuff_in_float(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), (float)_in), \
1587 double : _fr_dbuff_in_double(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), (double)_in) \
1594#define FR_DBUFF_IN_RETURN(_dbuff_or_marker, _in) FR_DBUFF_RETURN(fr_dbuff_in, _dbuff_or_marker, _in)
1602 uint8_t swapped[
sizeof(uint64_t)];
1606 if (ret >
sizeof(uint64_t))
return -1;
1610 return _fr_dbuff_in_memcpy(pos_p, dbuff, (swapped + (
sizeof(uint64_t) - ret)), ret);
1621#define fr_dbuff_in_uint64v(_dbuff_or_marker, _num) \
1622 _fr_dbuff_in_uint64v(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), _num)
1628#define FR_DBUFF_IN_UINT64V(_dbuff_or_marker, _num) FR_DBUFF_RETURN(fr_dbuff_in_uint64v, _dbuff_or_marker, _num)
1665#define fr_dbuff_move(_out, _in, _len) \
1669 fr_dbuff_t * : _fr_dbuff_move_dbuff_to_dbuff((fr_dbuff_t *)_out, \
1670 (fr_dbuff_t *)_in, \
1672 fr_dbuff_marker_t * : _fr_dbuff_move_dbuff_marker_to_dbuff((fr_dbuff_t *)_out, \
1673 (fr_dbuff_marker_t *)_in, \
1676 fr_dbuff_marker_t * : \
1678 fr_dbuff_t * : _fr_dbuff_move_dbuff_to_dbuff_marker((fr_dbuff_marker_t *)_out, \
1679 (fr_dbuff_t *)_in, \
1681 fr_dbuff_marker_t * : _fr_dbuff_move_dbuff_marker_to_dbuff_marker((fr_dbuff_marker_t *)_out, \
1682 (fr_dbuff_marker_t *)_in, \
1698 size_t ext_len, to_copy, remaining;
1700 for (remaining = outlen; remaining > 0; remaining -= to_copy) {
1701 to_copy = remaining;
1702 ext_len = _fr_dbuff_extend_lowat(NULL,
in,
fr_dbuff_end(
in) - (*pos_p), 1);
1703 if (ext_len == 0)
return -remaining;
1704 if (ext_len < to_copy) to_copy = ext_len;
1705 out += _fr_dbuff_set(pos_p,
in,
1706 (*pos_p) + _fr_dbuff_safecpy(
out,
out + to_copy, (*pos_p), (*pos_p) + to_copy));
1717 if (outlen == SIZE_MAX) outlen = _fr_dbuff_extend_lowat(NULL,
out,
fr_dbuff_end(
out) - (*out_p), outlen);
1719 return _fr_dbuff_out_memcpy((*out_p), pos_p,
in, outlen);
1741#define fr_dbuff_out_memcpy(_out, _dbuff_or_marker, _outlen) \
1743 uint8_t * : _fr_dbuff_out_memcpy((uint8_t *)(_out), \
1744 _fr_dbuff_current_ptr(_dbuff_or_marker), \
1745 fr_dbuff_ptr(_dbuff_or_marker), \
1747 fr_dbuff_t * : _fr_dbuff_out_memcpy_dbuff(_fr_dbuff_current_ptr((fr_dbuff_t *)_out), \
1748 fr_dbuff_ptr((fr_dbuff_t *)(_out)), \
1749 _fr_dbuff_current_ptr(_dbuff_or_marker), \
1750 fr_dbuff_ptr(_dbuff_or_marker), _outlen), \
1751 fr_dbuff_marker_t * : _fr_dbuff_out_memcpy_dbuff(_fr_dbuff_current_ptr((fr_dbuff_marker_t *)_out), \
1752 fr_dbuff_ptr((fr_dbuff_marker_t *)(_out)), \
1753 _fr_dbuff_current_ptr(_dbuff_or_marker), \
1754 fr_dbuff_ptr(_dbuff_or_marker), _outlen) \
1761#define FR_DBUFF_OUT_MEMCPY_RETURN(_out, _dbuff_or_marker, _outlen) FR_DBUFF_RETURN(fr_dbuff_out_memcpy, _out, _dbuff_or_marker, _outlen)
1767#define FR_DBUFF_OUT_DEF(_type) \
1768static inline ssize_t _fr_dbuff_out_##_type(_type##_t *out, uint8_t **pos_p, fr_dbuff_t *in) \
1771 FR_DBUFF_EXTEND_LOWAT_OR_RETURN(in, sizeof(_type##_t)); \
1772 *out = fr_nbo_to_##_type((*pos_p)); \
1773 return _fr_dbuff_set(pos_p, in, (*pos_p) + sizeof(_type##_t)); \
1776FR_DBUFF_OUT_DEF(uint16)
1777FR_DBUFF_OUT_DEF(uint32)
1778FR_DBUFF_OUT_DEF(uint64)
1779FR_DBUFF_OUT_DEF(int16)
1780FR_DBUFF_OUT_DEF(int32)
1781FR_DBUFF_OUT_DEF(int64)
1783#define FR_DBUFF_OUT_DEF_NO_SWAP(_type) \
1784static inline ssize_t _fr_dbuff_out_##_type(_type##_t *out, uint8_t **pos_p, fr_dbuff_t *in) \
1787 FR_DBUFF_EXTEND_LOWAT_OR_RETURN(in, sizeof(_type##_t)); \
1789 return _fr_dbuff_set(pos_p, in, (*pos_p) + sizeof(_type##_t)); \
1792FR_DBUFF_OUT_DEF_NO_SWAP(uint8)
1793FR_DBUFF_OUT_DEF_NO_SWAP(int8)
1808#define fr_dbuff_out(_out, _dbuff_or_marker) \
1810 uint8_t * : _fr_dbuff_out_uint8((uint8_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1811 uint16_t * : _fr_dbuff_out_uint16((uint16_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1812 uint32_t * : _fr_dbuff_out_uint32((uint32_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1813 uint64_t * : _fr_dbuff_out_uint64((uint64_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1814 int8_t * : _fr_dbuff_out_int8((int8_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1815 int16_t * : _fr_dbuff_out_int16((int16_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1816 int32_t * : _fr_dbuff_out_int32((int32_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1817 int64_t * : _fr_dbuff_out_int64((int64_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1818 float * : _fr_dbuff_out_uint32((uint32_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1819 double * : _fr_dbuff_out_uint64((uint64_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)) \
1826#ifndef STATIC_ANALYZER
1827#define FR_DBUFF_OUT_RETURN(_out, _dbuff_or_marker) FR_DBUFF_RETURN(fr_dbuff_out, _out, _dbuff_or_marker)
1829#define FR_DBUFF_OUT_RETURN(_out, _dbuff_or_marker) do { *_out = 0; FR_DBUFF_RETURN(fr_dbuff_out, _out, _dbuff_or_marker); } while (0)
1839 fr_assert(length > 0 && length <=
sizeof(uint64_t));
1842 slen = _fr_dbuff_out_memcpy(((
uint8_t *) num) + (8 - length), pos_p, dbuff, length);
1843 if (slen <= 0)
return slen;
1860#define fr_dbuff_out_uint64v(_num, _dbuff_or_marker, _len) \
1861 _fr_dbuff_out_uint64v(_num, _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), _len)
1867#define FR_DBUFF_OUT_UINT64V_RETURN(_num, _dbuff_or_marker, _len) FR_DBUFF_RETURN(fr_dbuff_out_uint64v, _num, _dbuff_or_marker, _len)
1877 fr_assert(length > 0 && length <=
sizeof(uint64_t));
1880 slen = _fr_dbuff_out_memcpy(((
uint8_t *) num) + (8 - length), pos_p, dbuff, length);
1881 if (slen <= 0)
return slen;
1883 if (msb & 0x80) memset(((
uint8_t *)num), 0xff,
sizeof(*num) - length);
1900#define fr_dbuff_out_int64v(_num, _dbuff_or_marker, _len) \
1901 _fr_dbuff_out_int64v(_num, _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), _len)
1907#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