26RCSIDH(dbuff_h,
"$Id: f613c11bd1cf7f12b7c883e81b37568639df5491 $")
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; \
202#define _FR_DBUFF(_dbuff_or_marker, _start, _adv_parent) \
204 .buff = fr_dbuff_buff(_dbuff_or_marker), \
206 .end = fr_dbuff_end(_dbuff_or_marker), \
207 .p = fr_dbuff_current(_dbuff_or_marker), \
208 .is_const = fr_dbuff_ptr(_dbuff_or_marker)->is_const, \
209 .adv_parent = (_adv_parent), \
210 .shifted = fr_dbuff_ptr(_dbuff_or_marker)->shifted, \
211 .extend = fr_dbuff_ptr(_dbuff_or_marker)->extend, \
212 .uctx = fr_dbuff_ptr(_dbuff_or_marker)->uctx, \
213 .parent = fr_dbuff_ptr(_dbuff_or_marker) \
224#define FR_DBUFF(_dbuff_or_marker) _FR_DBUFF(_dbuff_or_marker, fr_dbuff_current(_dbuff_or_marker), 0x00)
233#define FR_DBUFF_ABS(_dbuff_or_marker) _FR_DBUFF(_dbuff_or_marker, fr_dbuff_start(_dbuff_or_marker), 0x00)
242#define FR_DBUFF_BIND_CURRENT(_dbuff_or_marker) _FR_DBUFF(_dbuff_or_marker, fr_dbuff_current(_dbuff_or_marker), FR_DBUFF_ADV_PARENT_CURRENT)
251#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)
262#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)
269#define _FR_DBUFF_MAX(_dbuff_or_marker, _max, _adv_parent) \
271 .buff = fr_dbuff_buff(_dbuff_or_marker), \
272 .start = fr_dbuff_current(_dbuff_or_marker), \
273 .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))), \
274 .p = fr_dbuff_current(_dbuff_or_marker), \
275 .is_const = fr_dbuff_ptr(_dbuff_or_marker)->is_const, \
276 .adv_parent = _adv_parent, \
277 .shifted = fr_dbuff_ptr(_dbuff_or_marker)->shifted, \
280 .parent = fr_dbuff_ptr(_dbuff_or_marker) \
303#define FR_DBUFF_MAX(_dbuff_or_marker, _max) _FR_DBUFF_MAX(_dbuff_or_marker, _max, 0x00)
320#define FR_DBUFF_MAX_BIND_CURRENT(_dbuff_or_marker, _max) _FR_DBUFF_MAX(_dbuff_or_marker, _max, FR_DBUFF_ADV_PARENT_CURRENT)
326#if defined(__GNUC__) && __GNUC__ >= 11
338 if (
unlikely(end < start)) end = start;
356#define fr_dbuff_init(_out, _start, _len_or_end) \
357_fr_dbuff_init(_out, \
358 (uint8_t const *)(_start), \
359 _Generic((_len_or_end), \
360 size_t : (uint8_t const *)(_start) + (size_t)(_len_or_end), \
361 long : (uint8_t const *)(_start) + (size_t)(_len_or_end), \
362 int : (uint8_t const *)(_start) + (size_t)(_len_or_end), \
363 unsigned int : (uint8_t const *)(_start) + (size_t)(_len_or_end), \
364 uint8_t * : (uint8_t const *)(_len_or_end), \
365 uint8_t const * : (uint8_t const *)(_len_or_end), \
366 char * : (uint8_t const *)(_len_or_end), \
367 char const * : (uint8_t const *)(_len_or_end) \
371 uint8_t const * : true, \
373 char const * : true \
375#if defined(__GNUC__) && __GNUC__ >= 11
379#define FR_DBUFF_INIT(_out, _start, _len_or_end) do { \
380 fr_dbuff_init(_out, _start, _len_or_end); \
381 *(unsigned char *) _start = '\0'; \
400} fr_dbuff_uctx_talloc_t;
414 fr_dbuff_t *dbuff, fr_dbuff_uctx_talloc_t *tctx,
415 size_t init,
size_t max)
419 *tctx = (fr_dbuff_uctx_talloc_t){
436 memset(dbuff, 0,
sizeof(*dbuff));
457 TALLOC_FREE(dbuff->buff);
490 *fctx = (fr_dbuff_uctx_fd_t){
493 .buff_end =
buff + len
516#define FR_DBUFF_TMP(_start, _len_or_end) \
518 .buff_i = (uint8_t const *)(_start), \
519 .start_i = (uint8_t const *)(_start), \
520 .end_i = _Generic((_len_or_end), \
521 size_t : (uint8_t const *)(_start) + (size_t)(_len_or_end), \
522 long : (uint8_t const *)(_start) + (size_t)(_len_or_end), \
523 int : (uint8_t const *)(_start) + (size_t)(_len_or_end), \
524 unsigned int : (uint8_t const *)(_start) + (size_t)(_len_or_end), \
525 uint8_t * : (uint8_t const *)(_len_or_end), \
526 uint8_t const * : (uint8_t const *)(_len_or_end), \
527 char * : (uint8_t const *)(_len_or_end), \
528 char const * : (uint8_t const *)(_len_or_end) \
531 .is_const = _Generic((_start), \
533 uint8_t const * : true, \
535 char const * : true \
558#define FR_DBUFF_TALLOC_THREAD_LOCAL(_out, _init, _max) \
560 static _Thread_local fr_dbuff_thread_local_t *_dbuff_t_local; \
561 if (!_dbuff_t_local) { \
562 fr_dbuff_thread_local_t *dbtl = talloc_zero(NULL, fr_dbuff_thread_local_t); \
563 fr_dbuff_init_talloc(dbtl, &dbtl->dbuff, &dbtl->tctx, _init, _max); \
564 fr_atexit_thread_local(_dbuff_t_local, _dbuff_thread_local_free, dbtl); \
565 *(_out) = &_dbuff_t_local->dbuff; \
567 fr_dbuff_reset_talloc(&_dbuff_t_local->dbuff); \
568 *(_out) = &_dbuff_t_local->dbuff; \
584#define FR_DBUFF_FLAG_EXTENDABLE 0x01
588#define FR_DBUFF_FLAG_EXTENDED 0x02
612#define fr_dbuff_is_extendable(_status) ((_status) & FR_DBUFF_FLAG_EXTENDABLE)
616#define fr_dbuff_was_extended(_status) ((_status) & FR_DBUFF_FLAG_EXTENDED)
622 size_t remaining,
size_t lowat)
632 if (remaining >= lowat) {
637 if (!
in->extend || !(extended =
in->extend(
in, lowat - remaining)))
goto not_extendable;
641 return remaining + extended;
662#define fr_dbuff_extend_lowat(_status, _dbuff_or_marker, _lowat) \
663 _fr_dbuff_extend_lowat(_status, \
664 fr_dbuff_ptr(_dbuff_or_marker), \
665 fr_dbuff_remaining(_dbuff_or_marker), _lowat)
675#define FR_DBUFF_EXTEND_LOWAT_OR_RETURN(_dbuff_or_marker, _lowat) \
677 size_t _remaining = fr_dbuff_extend_lowat(NULL, _dbuff_or_marker, _lowat); \
678 if (_remaining < _lowat) return -(_lowat - _remaining); \
691#define _FR_DBUFF_EXTEND_LOWAT_POS_OR_RETURN(_pos_p, _dbuff_or_marker, _lowat) \
693 size_t _remaining = _fr_dbuff_extend_lowat(NULL, \
694 fr_dbuff_ptr(_dbuff_or_marker), \
695 fr_dbuff_end(_dbuff_or_marker) - (*(_pos_p)), _lowat); \
696 if (_remaining < _lowat) return -(_lowat - _remaining); \
707#define fr_dbuff_extend(_dbuff) fr_dbuff_extend_lowat(NULL, _dbuff, 1)
745#define fr_dbuff_remaining(_dbuff_or_marker) \
746 ((size_t)(fr_dbuff_end(_dbuff_or_marker) < fr_dbuff_current(_dbuff_or_marker) ? \
747 0 : (fr_dbuff_end(_dbuff_or_marker) - fr_dbuff_current(_dbuff_or_marker))))
763#define FR_DBUFF_REMAINING_RETURN(_dbuff_or_marker, _len) \
764 if ((_len) > fr_dbuff_remaining(_dbuff_or_marker)) return -((_len) - fr_dbuff_remaining(_dbuff_or_marker))
769#define fr_dbuff_used(_dbuff_or_marker) \
770 ((size_t)(fr_dbuff_start(_dbuff_or_marker) > fr_dbuff_current(_dbuff_or_marker) ? \
771 0 : (fr_dbuff_current(_dbuff_or_marker) - fr_dbuff_start(_dbuff_or_marker))))
778#define fr_dbuff_len(_dbuff_or_marker) \
779 ((size_t)(fr_dbuff_end(_dbuff_or_marker) - fr_dbuff_start(_dbuff_or_marker)))
788#define fr_dbuff_behind(_dbuff_or_marker) \
789 (fr_dbuff_current(_dbuff_or_marker) > fr_dbuff_current((_dbuff_or_marker)->parent) ? \
790 0 : fr_dbuff_current((_dbuff_or_marker)->parent) - fr_dbuff_current(_dbuff_or_marker))
798#define fr_dbuff_ahead(_dbuff_or_marker) \
799 (fr_dbuff_current((_dbuff_or_marker)->parent) > fr_dbuff_current(_dbuff_or_marker) ? \
800 0 : fr_dbuff_current(_dbuff_or_marker) - fr_dbuff_current((_dbuff_or_marker)->parent))
860#define fr_dbuff_ptr(_dbuff_or_marker) \
861 _Generic((_dbuff_or_marker), \
862 fr_dbuff_t * : ((fr_dbuff_t *)(_dbuff_or_marker)), \
863 fr_dbuff_marker_t * : (((fr_dbuff_marker_t *)(_dbuff_or_marker))->parent) \
871#define fr_dbuff_ptr_const(_dbuff_or_marker) \
872 _Generic((_dbuff_or_marker), \
873 fr_dbuff_t * : ((fr_dbuff_t const *)(_dbuff_or_marker)), \
874 fr_dbuff_t const * : ((fr_dbuff_t const *)(_dbuff_or_marker)), \
875 fr_dbuff_marker_t * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->parent), \
876 fr_dbuff_marker_t const * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->parent) \
884#define fr_dbuff_buff(_dbuff_or_marker) \
885 _Generic((_dbuff_or_marker), \
886 fr_dbuff_t * : (((fr_dbuff_t const *)(_dbuff_or_marker))->buff), \
887 fr_dbuff_t const * : (((fr_dbuff_t const *)(_dbuff_or_marker))->buff), \
888 fr_dbuff_marker_t * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->parent->buff), \
889 fr_dbuff_marker_t const * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->parent->buff) \
900#define fr_dbuff_start(_dbuff_or_marker) \
901 (_Generic((_dbuff_or_marker), \
902 fr_dbuff_t * : (((fr_dbuff_t const *)(_dbuff_or_marker))->start), \
903 fr_dbuff_t const * : (((fr_dbuff_t const *)(_dbuff_or_marker))->start), \
904 fr_dbuff_marker_t * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->parent->start), \
905 fr_dbuff_marker_t const * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->parent->start) \
913#define fr_dbuff_current(_dbuff_or_marker) \
914 (_Generic((_dbuff_or_marker), \
915 fr_dbuff_t * : (((fr_dbuff_t const *)(_dbuff_or_marker))->p), \
916 fr_dbuff_t const * : (((fr_dbuff_t const *)(_dbuff_or_marker))->p), \
917 fr_dbuff_marker_t * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->p), \
918 fr_dbuff_marker_t const * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->p) \
928#define _fr_dbuff_current_ptr(_dbuff_or_marker) \
929 (_Generic((_dbuff_or_marker), \
930 fr_dbuff_t * : &(((fr_dbuff_t *)(_dbuff_or_marker))->p), \
931 fr_dbuff_marker_t * : &(((fr_dbuff_marker_t *)(_dbuff_or_marker))->p) \
940#define fr_dbuff_end(_dbuff_or_marker) \
941 (_Generic((_dbuff_or_marker), \
942 fr_dbuff_t * : (((fr_dbuff_t const *)(_dbuff_or_marker))->end), \
943 fr_dbuff_t const * : (((fr_dbuff_t const *)(_dbuff_or_marker))->end), \
944 fr_dbuff_marker_t * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->parent->end), \
945 fr_dbuff_marker_t const * : (((fr_dbuff_marker_t const *)(_dbuff_or_marker))->parent->end) \
963 if (dbuff->adv_parent && dbuff->parent) _fr_dbuff_set_recurse(dbuff->parent, dbuff->adv_parent, p);
982 if (
unlikely(p > dbuff->end))
return -(p - dbuff->end);
983 if (
unlikely(p < dbuff->start))
return 0;
986 if (dbuff->adv_parent && dbuff->parent) _fr_dbuff_set_recurse(dbuff->parent, dbuff->adv_parent, p);
1006#define fr_dbuff_set(_dst, _src) \
1008 _fr_dbuff_current_ptr(_dst), fr_dbuff_ptr(_dst), \
1010 fr_dbuff_t * : fr_dbuff_current((fr_dbuff_t const *)(_src)), \
1011 fr_dbuff_marker_t * : fr_dbuff_current((fr_dbuff_marker_t const *)(_src)), \
1012 uint8_t const * : (uint8_t const *)(_src), \
1013 uint8_t * : (uint8_t const *)(_src), \
1014 size_t : (fr_dbuff_start(_dst) + (uintptr_t)(_src)), \
1015 long : (fr_dbuff_start(_dst) + (uintptr_t)(_src)), \
1016 int : (fr_dbuff_start(_dst) + (uintptr_t)(_src)) \
1024#define FR_DBUFF_SET_RETURN(_dst, _src) FR_DBUFF_RETURN(fr_dbuff_set, _dst, _src)
1037 if (
unlikely(p > dbuff->end))
return -(p - dbuff->end);
1038 if (
unlikely(p < dbuff->start))
return 0;
1042 return dbuff->end - p;
1053#define fr_dbuff_set_end(_dst, _end) \
1055 fr_dbuff_ptr(_dst), \
1057 fr_dbuff_t * : fr_dbuff_current((fr_dbuff_t const *)(_end)), \
1058 fr_dbuff_marker_t * : fr_dbuff_current((fr_dbuff_marker_t const *)(_end)), \
1059 uint8_t const * : (uint8_t const *)(_end), \
1060 uint8_t * : (uint8_t const *)(_end) \
1075#define fr_dbuff_advance(_dbuff_or_marker, _len) \
1076 fr_dbuff_set(_dbuff_or_marker, \
1077 (fr_dbuff_current(_dbuff_or_marker) + \
1079 unsigned char : (size_t)(_len), \
1080 unsigned short : (size_t)(_len), \
1081 unsigned int : (size_t)(_len), \
1082 unsigned long : (size_t)(_len), \
1083 unsigned long long : (size_t)(_len), \
1084 int : (size_t)(_len) \
1091#define FR_DBUFF_ADVANCE_RETURN(_dbuff_or_marker, _len) FR_DBUFF_RETURN(fr_dbuff_advance, _dbuff_or_marker, _len)
1111 if (p > dbuff->end) {
1112 size_t rel = p - dbuff->start;
1114 if (!dbuff->extend) {
1116 return -((dbuff->start + rel) - dbuff->end);
1119 dbuff->extend(dbuff, p - dbuff->end);
1120 if ((dbuff->start + rel) > dbuff->end)
goto oos;
1122 *pos_p = dbuff->start + rel;
1127 if (dbuff->adv_parent && dbuff->parent) _fr_dbuff_set_recurse(dbuff->parent, dbuff->adv_parent, *pos_p);
1142#define fr_dbuff_advance_extend(_dbuff_or_marker, _len) \
1143 _fr_dbuff_advance(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), \
1145 unsigned char : (size_t)(_len), \
1146 unsigned short : (size_t)(_len), \
1147 unsigned int : (size_t)(_len), \
1148 unsigned long : (size_t)(_len), \
1149 unsigned long long : (size_t)(_len), \
1150 int : (size_t)(_len) \
1153#define FR_DBUFF_BIND_EXTEND_RETURN(_dbuff_or_marker, _len) FR_DBUFF_RETURN(fr_dbuff_advance_extend, _dbuff_or_marker, _len)
1158#define fr_dbuff_set_to_start(_dbuff_or_marker) \
1159 fr_dbuff_set(_dbuff_or_marker, fr_dbuff_start(_dbuff_or_marker))
1164#define fr_dbuff_set_to_end(_dbuff_or_marker) \
1165 fr_dbuff_set(_dbuff_or_marker, fr_dbuff_end(_dbuff_or_marker))
1215 m->parent->m = m->next;
1218 memset(m, 0,
sizeof(*m));
1277static inline CC_HINT(always_inline)
size_t _fr_dbuff_safecpy(
uint8_t *o_start,
uint8_t *o_end,
1281 size_t i_len = i_end - i_start;
1283 if (
unlikely((o_end < o_start) || (i_end < i_start)))
return 0;
1285 diff = (o_end - o_start) - (i_len);
1286 if (diff < 0)
return 0;
1288 if ((i_start > o_end) || (i_end < o_start)) {
1289 memcpy(o_start, i_start, i_len);
1291 memmove(o_start, i_start, i_len);
1306 _FR_DBUFF_EXTEND_LOWAT_POS_OR_RETURN(pos_p,
out,
inlen);
1308 return _fr_dbuff_set(pos_p,
out, (*pos_p) + _fr_dbuff_safecpy((*pos_p), (*pos_p) +
inlen,
in,
in +
inlen));
1322 memcpy(&our_in, &
in,
sizeof(our_in));
1323 memcpy(&our_in_p, &in_p,
sizeof(our_in_p));
1325 if (
inlen == SIZE_MAX) {
1326 ext_len = _fr_dbuff_extend_lowat(NULL, our_in,
fr_dbuff_end(our_in) - (*our_in_p),
inlen);
1329 _FR_DBUFF_EXTEND_LOWAT_POS_OR_RETURN(our_in_p, our_in,
inlen);
1331 return _fr_dbuff_in_memcpy(pos_p,
out, *our_in_p,
inlen);
1353#define fr_dbuff_in_memcpy(_dbuff_or_marker, _in, _inlen) \
1355 uint8_t * : _fr_dbuff_in_memcpy(_fr_dbuff_current_ptr(_dbuff_or_marker), \
1356 fr_dbuff_ptr(_dbuff_or_marker), \
1357 (uint8_t const *)(_in), \
1359 uint8_t const * : _fr_dbuff_in_memcpy(_fr_dbuff_current_ptr(_dbuff_or_marker), \
1360 fr_dbuff_ptr(_dbuff_or_marker), \
1361 (uint8_t const *)(_in), \
1363 char * : _fr_dbuff_in_memcpy(_fr_dbuff_current_ptr(_dbuff_or_marker), \
1364 fr_dbuff_ptr(_dbuff_or_marker), \
1365 (uint8_t const *)(_in), \
1366 (size_t)(_inlen) == SIZE_MAX ? strlen((char const *)(_in)) : (_inlen)), \
1367 char const * : _fr_dbuff_in_memcpy(_fr_dbuff_current_ptr(_dbuff_or_marker), \
1368 fr_dbuff_ptr(_dbuff_or_marker), \
1369 (uint8_t const *)(_in), \
1370 (size_t)(_inlen) == SIZE_MAX ? strlen((char const *)(_in)) : (_inlen)), \
1371 fr_dbuff_t * : _fr_dbuff_in_memcpy_dbuff(_fr_dbuff_current_ptr(_dbuff_or_marker), \
1372 fr_dbuff_ptr(_dbuff_or_marker), \
1373 &((fr_dbuff_t const *)(_in))->p, \
1374 ((fr_dbuff_t const *)(_in)), \
1376 fr_dbuff_marker_t * : _fr_dbuff_in_memcpy_dbuff(_fr_dbuff_current_ptr(_dbuff_or_marker), \
1377 fr_dbuff_ptr(_dbuff_or_marker), \
1378 &((fr_dbuff_marker_t const *)(_in))->p, \
1379 ((fr_dbuff_marker_t const *)(_in))->parent, _inlen) \
1385#define FR_DBUFF_IN_MEMCPY_RETURN(_dbuff_or_marker, _in, _inlen) FR_DBUFF_RETURN(fr_dbuff_in_memcpy, _dbuff_or_marker, _in, _inlen)
1401 return _fr_dbuff_set(pos_p,
out, (*pos_p) + _fr_dbuff_safecpy((*pos_p), (*pos_p) +
inlen,
in,
in +
inlen));
1415 ext_len = _fr_dbuff_extend_lowat(NULL, our_in,
fr_dbuff_end(our_in) - (*our_in_p),
inlen);
1418 return _fr_dbuff_in_memcpy_partial(pos_p,
out, (*our_in_p),
inlen);
1441#define fr_dbuff_in_memcpy_partial(_out, _in, _inlen) \
1443 uint8_t * : _fr_dbuff_in_memcpy_partial(_fr_dbuff_current_ptr(_out), _out, (uint8_t const *)(_in), _inlen), \
1444 uint8_t const * : _fr_dbuff_in_memcpy_partial(_fr_dbuff_current_ptr(_out), _out, (uint8_t const *)(_in), _inlen), \
1445 char * : _fr_dbuff_in_memcpy_partial(_fr_dbuff_current_ptr(_out), _out, (uint8_t const *)(_in), _inlen == SIZE_MAX ? strlen((char const *)(_in)) : _inlen), \
1446 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), \
1447 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), \
1448 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) \
1458#define fr_dbuff_in_bytes_partial(_dbuff, ...) \
1459 fr_dbuff_in_memcpy_partial(_dbuff, ((uint8_t []){ __VA_ARGS__ }), sizeof((uint8_t []){ __VA_ARGS__ }))
1468#define fr_dbuff_in_bytes(_dbuff_or_marker, ...) \
1469 fr_dbuff_in_memcpy(_dbuff_or_marker, ((uint8_t []){ __VA_ARGS__ }), sizeof((uint8_t []){ __VA_ARGS__ }))
1475#define FR_DBUFF_IN_BYTES_RETURN(_dbuff_or_marker, ...) \
1476 FR_DBUFF_IN_MEMCPY_RETURN(_dbuff_or_marker, ((uint8_t []){ __VA_ARGS__ }), sizeof((uint8_t []){ __VA_ARGS__ }))
1486 _FR_DBUFF_EXTEND_LOWAT_POS_OR_RETURN(pos_p, dbuff,
inlen);
1488 memset((*pos_p), c,
inlen);
1490 return _fr_dbuff_set(pos_p, dbuff, (*pos_p) +
inlen);
1504#define fr_dbuff_memset(_dbuff_or_marker, _c, _inlen) \
1505 _fr_dbuff_memset(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), _c, _inlen)
1511#define FR_DBUFF_MEMSET_RETURN(_dbuff_or_marker, _c, _inlen) FR_DBUFF_RETURN(fr_dbuff_memset, _dbuff_or_marker, _c, _inlen)
1517#define FR_DBUFF_PARSE_INT_DEF(_type) \
1518static inline ssize_t _fr_dbuff_in_##_type(uint8_t **pos_p, fr_dbuff_t *out, _type##_t num) \
1520 fr_assert(!out->is_const); \
1521 _FR_DBUFF_EXTEND_LOWAT_POS_OR_RETURN(pos_p, out, sizeof(_type##_t)); \
1522 fr_nbo_from_##_type((*pos_p), num); \
1523 return _fr_dbuff_set(pos_p, out, (*pos_p) + sizeof(_type##_t)); \
1525FR_DBUFF_PARSE_INT_DEF(uint16)
1526FR_DBUFF_PARSE_INT_DEF(uint32)
1527FR_DBUFF_PARSE_INT_DEF(uint64)
1528FR_DBUFF_PARSE_INT_DEF(int16)
1529FR_DBUFF_PARSE_INT_DEF(int32)
1530FR_DBUFF_PARSE_INT_DEF(int64)
1547 return _fr_dbuff_in_uint32(pos_p,
out, *(
uint32_t *)(&num));
1558 return _fr_dbuff_in_uint64(pos_p,
out, *(uint64_t *)(&num));
1570#define fr_dbuff_in(_dbuff_or_marker, _in) \
1572 int8_t : fr_dbuff_in_bytes(_dbuff_or_marker, (int8_t)_in), \
1573 int16_t : _fr_dbuff_in_int16(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), (int16_t)_in), \
1574 int32_t : _fr_dbuff_in_int32(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), (int32_t)_in), \
1575 int64_t : _fr_dbuff_in_int64(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), (int64_t)_in), \
1576 uint8_t : fr_dbuff_in_bytes(_dbuff_or_marker, (uint8_t)_in), \
1577 uint16_t : _fr_dbuff_in_uint16(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), (uint16_t)_in), \
1578 uint32_t : _fr_dbuff_in_uint32(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), (uint32_t)_in), \
1579 uint64_t : _fr_dbuff_in_uint64(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), (uint64_t)_in), \
1580 float : _fr_dbuff_in_float(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), (float)_in), \
1581 double : _fr_dbuff_in_double(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), (double)_in) \
1588#define FR_DBUFF_IN_RETURN(_dbuff_or_marker, _in) FR_DBUFF_RETURN(fr_dbuff_in, _dbuff_or_marker, _in)
1596 uint8_t swapped[
sizeof(uint64_t)];
1600 if (ret >
sizeof(uint64_t))
return -1;
1604 return _fr_dbuff_in_memcpy(pos_p, dbuff, (swapped + (
sizeof(uint64_t) - ret)), ret);
1615#define fr_dbuff_in_uint64v(_dbuff_or_marker, _num) \
1616 _fr_dbuff_in_uint64v(_fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), _num)
1622#define FR_DBUFF_IN_UINT64V(_dbuff_or_marker, _num) FR_DBUFF_RETURN(fr_dbuff_in_uint64v, _dbuff_or_marker, _num)
1659#define fr_dbuff_move(_out, _in, _len) \
1663 fr_dbuff_t * : _fr_dbuff_move_dbuff_to_dbuff((fr_dbuff_t *)_out, \
1664 (fr_dbuff_t *)_in, \
1666 fr_dbuff_marker_t * : _fr_dbuff_move_dbuff_marker_to_dbuff((fr_dbuff_t *)_out, \
1667 (fr_dbuff_marker_t *)_in, \
1670 fr_dbuff_marker_t * : \
1672 fr_dbuff_t * : _fr_dbuff_move_dbuff_to_dbuff_marker((fr_dbuff_marker_t *)_out, \
1673 (fr_dbuff_t *)_in, \
1675 fr_dbuff_marker_t * : _fr_dbuff_move_dbuff_marker_to_dbuff_marker((fr_dbuff_marker_t *)_out, \
1676 (fr_dbuff_marker_t *)_in, \
1692 size_t ext_len, to_copy, remaining;
1694 for (remaining = outlen; remaining > 0; remaining -= to_copy) {
1695 to_copy = remaining;
1696 ext_len = _fr_dbuff_extend_lowat(NULL,
in,
fr_dbuff_end(
in) - (*pos_p), 1);
1697 if (ext_len == 0)
return -remaining;
1698 if (ext_len < to_copy) to_copy = ext_len;
1699 out += _fr_dbuff_set(pos_p,
in,
1700 (*pos_p) + _fr_dbuff_safecpy(
out,
out + to_copy, (*pos_p), (*pos_p) + to_copy));
1711 if (outlen == SIZE_MAX) outlen = _fr_dbuff_extend_lowat(NULL,
out,
fr_dbuff_end(
out) - (*out_p), outlen);
1713 return _fr_dbuff_out_memcpy((*out_p), pos_p,
in, outlen);
1735#define fr_dbuff_out_memcpy(_out, _dbuff_or_marker, _outlen) \
1737 uint8_t * : _fr_dbuff_out_memcpy((uint8_t *)(_out), \
1738 _fr_dbuff_current_ptr(_dbuff_or_marker), \
1739 fr_dbuff_ptr(_dbuff_or_marker), \
1741 fr_dbuff_t * : _fr_dbuff_out_memcpy_dbuff(_fr_dbuff_current_ptr((fr_dbuff_t *)_out), \
1742 fr_dbuff_ptr((fr_dbuff_t *)(_out)), \
1743 _fr_dbuff_current_ptr(_dbuff_or_marker), \
1744 fr_dbuff_ptr(_dbuff_or_marker), _outlen), \
1745 fr_dbuff_marker_t * : _fr_dbuff_out_memcpy_dbuff(_fr_dbuff_current_ptr((fr_dbuff_marker_t *)_out), \
1746 fr_dbuff_ptr((fr_dbuff_marker_t *)(_out)), \
1747 _fr_dbuff_current_ptr(_dbuff_or_marker), \
1748 fr_dbuff_ptr(_dbuff_or_marker), _outlen) \
1755#define FR_DBUFF_OUT_MEMCPY_RETURN(_out, _dbuff_or_marker, _outlen) FR_DBUFF_RETURN(fr_dbuff_out_memcpy, _out, _dbuff_or_marker, _outlen)
1761#define FR_DBUFF_OUT_DEF(_type) \
1762static inline ssize_t _fr_dbuff_out_##_type(_type##_t *out, uint8_t **pos_p, fr_dbuff_t *in) \
1765 FR_DBUFF_EXTEND_LOWAT_OR_RETURN(in, sizeof(_type##_t)); \
1766 *out = fr_nbo_to_##_type((*pos_p)); \
1767 return _fr_dbuff_set(pos_p, in, (*pos_p) + sizeof(_type##_t)); \
1770FR_DBUFF_OUT_DEF(uint16)
1771FR_DBUFF_OUT_DEF(uint32)
1772FR_DBUFF_OUT_DEF(uint64)
1773FR_DBUFF_OUT_DEF(int16)
1774FR_DBUFF_OUT_DEF(int32)
1775FR_DBUFF_OUT_DEF(int64)
1777#define FR_DBUFF_OUT_DEF_NO_SWAP(_type) \
1778static inline ssize_t _fr_dbuff_out_##_type(_type##_t *out, uint8_t **pos_p, fr_dbuff_t *in) \
1781 FR_DBUFF_EXTEND_LOWAT_OR_RETURN(in, sizeof(_type##_t)); \
1783 return _fr_dbuff_set(pos_p, in, (*pos_p) + sizeof(_type##_t)); \
1786FR_DBUFF_OUT_DEF_NO_SWAP(uint8)
1787FR_DBUFF_OUT_DEF_NO_SWAP(int8)
1802#define fr_dbuff_out(_out, _dbuff_or_marker) \
1804 uint8_t * : _fr_dbuff_out_uint8((uint8_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1805 uint16_t * : _fr_dbuff_out_uint16((uint16_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1806 uint32_t * : _fr_dbuff_out_uint32((uint32_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1807 uint64_t * : _fr_dbuff_out_uint64((uint64_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1808 int8_t * : _fr_dbuff_out_int8((int8_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1809 int16_t * : _fr_dbuff_out_int16((int16_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1810 int32_t * : _fr_dbuff_out_int32((int32_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1811 int64_t * : _fr_dbuff_out_int64((int64_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1812 float * : _fr_dbuff_out_uint32((uint32_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)), \
1813 double * : _fr_dbuff_out_uint64((uint64_t *)(_out), _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker)) \
1820#ifndef STATIC_ANALYZER
1821#define FR_DBUFF_OUT_RETURN(_out, _dbuff_or_marker) FR_DBUFF_RETURN(fr_dbuff_out, _out, _dbuff_or_marker)
1823#define FR_DBUFF_OUT_RETURN(_out, _dbuff_or_marker) do { *_out = 0; FR_DBUFF_RETURN(fr_dbuff_out, _out, _dbuff_or_marker); } while (0)
1833 fr_assert(length > 0 && length <=
sizeof(uint64_t));
1836 slen = _fr_dbuff_out_memcpy(((
uint8_t *) num) + (8 - length), pos_p, dbuff, length);
1837 if (slen <= 0)
return slen;
1854#define fr_dbuff_out_uint64v(_num, _dbuff_or_marker, _len) \
1855 _fr_dbuff_out_uint64v(_num, _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), _len)
1861#define FR_DBUFF_OUT_UINT64V_RETURN(_num, _dbuff_or_marker, _len) FR_DBUFF_RETURN(fr_dbuff_out_uint64v, _num, _dbuff_or_marker, _len)
1871 fr_assert(length > 0 && length <=
sizeof(uint64_t));
1874 slen = _fr_dbuff_out_memcpy(((
uint8_t *) num) + (8 - length), pos_p, dbuff, length);
1875 if (slen <= 0)
return slen;
1877 if (msb & 0x80) memset(((
uint8_t *)num), 0xff,
sizeof(*num) - length);
1894#define fr_dbuff_out_int64v(_num, _dbuff_or_marker, _len) \
1895 _fr_dbuff_out_int64v(_num, _fr_dbuff_current_ptr(_dbuff_or_marker), fr_dbuff_ptr(_dbuff_or_marker), _len)
1901#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