The FreeRADIUS server $Id: 15bac2a4c627c01d1aa2047687b3418955ac7f00 $
Loading...
Searching...
No Matches
sbuff.h
Go to the documentation of this file.
1#pragma once
2/*
3 * This program is free software; you can redistribute it and/or modify
4 * it under the terms of the GNU General Public License as published by
5 * the Free Software Foundation; either version 2 of the License, or
6 * (at your option) any later version.
7 *
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU General Public License for more details.
12 *
13 * You should have received a copy of the GNU General Public License
14 * along with this program; if not, write to the Free Software
15 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
16 */
17
18/** A generic buffer structure for string printing and parsing strings
19 *
20 * Because doing manual length checks is error prone and a waste of everyones time.
21 *
22 * @file src/lib/util/sbuff.h
23 *
24 * @copyright 2020 Arran Cudbard-Bell <a.cudbardb@freeradius.org>
25 */
26RCSIDH(sbuff_h, "$Id: e058dbcd6c6251933f636751425e8fe43373b03d $")
27
28# ifdef __cplusplus
29extern "C" {
30# endif
31
32#include <ctype.h>
33#include <errno.h>
34#include <limits.h>
35#include <stdbool.h>
36#include <stdint.h>
37#include <stdio.h>
38#include <string.h>
39#include <sys/types.h>
40
41/** Represents number of bytes parsed or location of parse error
42 *
43 * Number of bytes parsed will be >= 0.
44 *
45 * If a parse error occurs the value will be the negative offset
46 * of the error -1. i.e. offset 0 will be -1.
47 *
48 * This is to disambiguate between 0 bytes parsed and error at
49 * offset 0.
50 */
52typedef struct fr_sbuff_s fr_sbuff_t;
54
55#include <freeradius-devel/util/atexit.h>
56#include <freeradius-devel/util/strerror.h>
57#include <freeradius-devel/util/table.h>
58#include <freeradius-devel/util/talloc.h>
59
60/** Whether the buffer is currently extendable and whether it was extended
61 *
62 */
63DIAG_OFF(attributes)
64typedef enum CC_HINT(flag_enum) {
65 FR_SBUFF_FLAG_EXTENDED = 0x01, //!< The last call to extend function actually extended the buffer.
66 FR_SBUFF_FLAG_EXTEND_ERROR = 0x02 //!< The last call to an extend function resulted in an error.
67 ///< Error should be provided using fr_strerror_const/fr_strerror_printf
68 ///< by the extension function.
70DIAG_OFF(attributes)
71
72/** Extension callback
73 *
74 * Retrieves additional data from a source and adds it to a buffer.
75 */
76typedef size_t(*fr_sbuff_extend_t)(fr_sbuff_extend_status_t *status, fr_sbuff_t *sbuff, size_t req_extension);
77
78/** For a given extension function, returns whether it is at EOF
79 *
80 */
81typedef bool(*fr_sbuff_eof_t)(fr_sbuff_t *sbuff);
82
83
85 union {
86 char const *p_i; //!< Immutable position pointer.
87 char *p; //!< Mutable position pointer.
88 };
89 fr_sbuff_marker_t *next; //!< Next m in the list.
90 fr_sbuff_t *parent; //!< Owner of the marker
91};
92
93struct fr_sbuff_s {
94 union {
95 char const *buff_i; //!< Immutable buffer pointer.
96 char *buff; //!< Mutable buffer pointer.
97 };
98
99 union {
100 char const *start_i; //!< Immutable start pointer.
101 char *start; //!< Mutable start pointer.
102 };
103
104 union {
105 char const *end_i; //!< Immutable end pointer.
106 char *end; //!< Mutable end pointer.
107 };
108
109 union {
110 char const *p_i; //!< Immutable position pointer.
111 char *p; //!< Mutable position pointer.
112 };
113
114 char const *err; //!< Where the last error occurred.
115
116 uint8_t is_const:1; //!< Can't be modified.
117 uint8_t adv_parent:1; //!< If true, advance the parent.
118 size_t shifted; //!< How many bytes this sbuff has been
119 ///< shifted since its creation.
120
121 fr_sbuff_extend_t extend; //!< Function to re-populate or extend
122 ///< the buffer.
123
124 fr_sbuff_eof_t eof; //!< Function to determine if the buffer is at EOF.
125
126 void *uctx; //!< Extend uctx data.
127
128 fr_sbuff_t *parent; //!< sbuff this sbuff was copied from.
129
130 fr_sbuff_marker_t *m; //!< Pointers to update if the underlying
131 ///< buffer changes.
132};
133
134/** Talloc sbuff extension structure
135 *
136 * Holds the data necessary for creating dynamically
137 * extensible buffers.
138 */
139typedef struct {
140 TALLOC_CTX *ctx; //!< Context to alloc new buffers in.
141 size_t init; //!< How much to allocate initially.
142 size_t max; //!< Maximum size of the buffer.
144
145/** File sbuff extension structure
146 *
147 * Holds the data necessary for creating dynamically
148 * extensible file buffers.
149 */
150typedef struct {
151 FILE *file; //!< FILE * we're reading from.
152 char *buff_end; //!< The true end of the buffer.
153 size_t max; //!< Maximum number of bytes to read.
154 size_t shifted; //!< How much we've read from this file.
155 bool eof; //!< are we at EOF?
157
158/** Terminal element with pre-calculated lengths
159 *
160 */
161typedef struct {
162 char const *str; //!< Terminal string
163 size_t len; //!< Length of string
165
166/** Set of terminal elements
167 *
168 * The elements MUST be listed in sorted order. If the inputs are
169 * not sorted, then all kinds of things will break.
170 */
171typedef struct {
172 size_t len; //!< Length of the list.
173 fr_sbuff_term_elem_t *elem; //!< A sorted list of terminal strings.
175
176/** Initialise a terminal structure with a single string
177 *
178 * @param[in] _str terminal string.
179 */
180#define FR_SBUFF_TERM(_str) \
181(fr_sbuff_term_t){ \
182 .len = 1, \
183 .elem = (fr_sbuff_term_elem_t[]){ L(_str) }, \
184}
185
186/** Initialise a terminal structure with a list of sorted strings
187 *
188 * Strings must be lexicographically sorted.
189 *
190 * @param[in] ... Lexicographically sorted list of terminal strings.
191 */
192#define FR_SBUFF_TERMS(...) \
193(fr_sbuff_term_t){ \
194 .len = (sizeof((fr_sbuff_term_elem_t[]){ __VA_ARGS__ }) / sizeof(fr_sbuff_term_elem_t)), \
195 .elem = (fr_sbuff_term_elem_t[]){ __VA_ARGS__ }, \
196}
197
198/** Set of parsing rules for *unescape_until functions
199 *
200 */
201typedef struct {
202 char const *name; //!< Name for rule set to aid we debugging.
203
204 char chr; //!< Character at the start of an escape sequence.
205 char subs[UINT8_MAX + 1]; //!< Special characters and their substitutions.
206 ///< Indexed by the printable representation i.e.
207 ///< 'n' for \n.
208 bool skip[UINT8_MAX + 1]; //!< Characters that are escaped, but left in the
209 ///< output along with the escape character.
210 ///< This is useful where we need to interpret escape
211 ///< sequences for parsing, but where the string will
212 ///< be passed off to a 3rd party library which will
213 ///< need to interpret the same sequences.
214
215 bool do_hex; //!< Process hex sequences i.e. @verbatim\x<hex><hex>.@endverbatim
216 bool do_oct; //!< Process oct sequences i.e. @verbatim<oct><oct><oct>.@endverbatim
218
219/** Set of parsing rules for *unescape_until functions
220 *
221 */
222typedef struct {
223 char const *name; //!< Name for rule set to aid we debugging.
224
225 char chr; //!< Character at the start of an escape sequence.
226
227 char subs[UINT8_MAX + 1]; //!< Special characters and their substitutions.
228 ///< Indexed by the binary representation i.e.
229 ///< 0x0a for \n.
230 bool esc[UINT8_MAX + 1]; //!< Characters that should be translated to hex or
231 ///< octal escape sequences.
232 bool do_utf8; //!< If true Don't apply escaping rules to valid UTF-8 sequences.
233
234 bool do_hex; //!< Represent escaped chars as hex sequences i.e.
235 ///< @verbatim\x<hex><hex>@endverbatim
236 bool do_oct; //!< Represent escapes chars as octal sequences i.e.
237 ///< @verbatim<oct><oct><oct>@endvertbatim
239
240/** A set of terminal sequences, and escape rules
241 *
242 */
243typedef struct {
244 fr_sbuff_unescape_rules_t const *escapes; //!< Escape characters
245
246 fr_sbuff_term_t const *terminals; //!< Terminal characters used as a hint
247 ///< that a token is not complete.
248} fr_sbuff_parse_rules_t;
249
250/** Standard parsing errors to be used by sbuff functions and other sbuff based parsing functions
251 *
252 */
253typedef enum {
254 FR_SBUFF_PARSE_OK = 0, //!< No error.
255 FR_SBUFF_PARSE_ERROR_NOT_FOUND = -1, //!< String does not contain a token
256 ///< matching the output type.
257 FR_SBUFF_PARSE_ERROR_TRAILING = -2, //!< Trailing characters found.
258 FR_SBUFF_PARSE_ERROR_FORMAT = -3, //!< Format of data was invalid.
259 FR_SBUFF_PARSE_ERROR_OUT_OF_SPACE = -4, //!< No space available in output buffer.
260 FR_SBUFF_PARSE_ERROR_NUM_OVERFLOW = -5, //!< Integer type would overflow.
261 FR_SBUFF_PARSE_ERROR_NUM_UNDERFLOW = -6 //!< Integer type would underflow.
263
265extern size_t sbuff_parse_error_table_len;
266
267/** Replace the contents of the thread local error stack with the string representation of a parse error
268 */
269static inline void fr_sbuff_parse_error_to_strerror(fr_sbuff_parse_error_t err)
270{
272}
273
274/** Return whether the sbuff is extendable
275 */
276static inline bool fr_sbuff_is_extendable(fr_sbuff_t *sbuff)
277{
278 return sbuff->extend && (!sbuff->eof || (sbuff->eof(sbuff) == false));
279}
280
281#define fr_sbuff_was_extended(_status) (_status & FR_SBUFF_FLAG_EXTENDED)
282
283extern bool const sbuff_char_class_uint[UINT8_MAX + 1];
284extern bool const sbuff_char_class_int[UINT8_MAX + 1];
285extern bool const sbuff_char_class_float[UINT8_MAX + 1];
286extern bool const sbuff_char_class_zero[UINT8_MAX + 1];
287extern bool const sbuff_char_class_hex[UINT8_MAX + 1];
288extern bool const sbuff_char_alpha_num[UINT8_MAX + 1];
289extern bool const sbuff_char_word[UINT8_MAX + 1];
290extern bool const sbuff_char_whitespace[UINT8_MAX + 1];
291extern bool const sbuff_char_line_endings[UINT8_MAX + 1];
292extern bool const sbuff_char_blank[UINT8_MAX + 1];
293extern bool const sbuff_char_class_hostname[UINT8_MAX + 1];
294
295/** Matches a-z,A-Z
296 */
297#define SBUFF_CHAR_CLASS_ALPHA \
298 ['a'] = true, ['b'] = true, ['c'] = true, ['d'] = true, ['e'] = true, \
299 ['f'] = true, ['g'] = true, ['h'] = true, ['i'] = true, ['j'] = true, \
300 ['k'] = true, ['l'] = true, ['m'] = true, ['n'] = true, ['o'] = true, \
301 ['p'] = true, ['q'] = true, ['r'] = true, ['s'] = true, ['t'] = true, \
302 ['u'] = true, ['v'] = true, ['w'] = true, ['x'] = true, ['y'] = true, \
303 ['z'] = true, \
304 ['A'] = true, ['B'] = true, ['C'] = true, ['D'] = true, ['E'] = true, \
305 ['F'] = true, ['G'] = true, ['H'] = true, ['I'] = true, ['J'] = true, \
306 ['K'] = true, ['L'] = true, ['M'] = true, ['N'] = true, ['O'] = true, \
307 ['P'] = true, ['Q'] = true, ['R'] = true, ['S'] = true, ['T'] = true, \
308 ['U'] = true, ['V'] = true, ['W'] = true, ['X'] = true, ['Y'] = true, \
309 ['Z'] = true
310
311/** Matches 0-9
312 */
313#define SBUFF_CHAR_CLASS_NUM \
314 ['0'] = true, ['1'] = true, ['2'] = true, ['3'] = true, ['4'] = true, \
315 ['5'] = true, ['6'] = true, ['7'] = true, ['8'] = true, ['9'] = true
316
317/** Matches 0-9,a-z,A-Z
318 */
319#define SBUFF_CHAR_CLASS_ALPHA_NUM \
320 SBUFF_CHAR_CLASS_ALPHA, \
321 SBUFF_CHAR_CLASS_NUM
322
323/** Matches 0-9,a-f,A-F
324 */
325#define SBUFF_CHAR_CLASS_HEX \
326 SBUFF_CHAR_CLASS_NUM, \
327 ['a'] = true, ['b'] = true, ['c'] = true, ['d'] = true, ['e'] = true, \
328 ['f'] = true, \
329 ['A'] = true, ['B'] = true, ['C'] = true, ['D'] = true, ['E'] = true, \
330 ['F'] = true
331
332#define SBUFF_CHAR_CLASS_SYMBOLS \
333 ['!'] = true, ['"'] = true, ['#'] = true, ['$'] = true, ['%'] = true, \
334 ['&'] = true, ['/'] = true, ['('] = true, [')'] = true, ['*'] = true, \
335 ['+'] = true, ['`'] = true, ['-'] = true, ['.'] = true, ['/'] = true, \
336 [':'] = true, [';'] = true, ['<'] = true, ['='] = true, ['>'] = true, \
337 ['?'] = true, ['@'] = true, ['['] = true, ['\''] = true, [']'] = true, \
338 ['^'] = true, ['_'] = true, ['`'] = true, ['{'] = true, ['|'] = true, \
339 ['}'] = true, ['~'] = true \
340
341/*
342 * If the additional tables need to be generated feel free to use this
343 * code snippet.
344 *
345 * @verbatim
346 #include <stdio.h>
347 #include <stdlib.h>
348
349 int main(int argc, char **argv)
350 {
351 int start, end, i;
352 start = atoi(argv[1]);
353 end = atoi(argv[2]);
354 for (i = start; i <= end; i++) {
355 printf("[0x%02x] = true, ", i);
356 if (!(i % 8)) printf("\\\n");
357 }
358 return 0;
359 }
360 * @endverbatim
361 */
362
363/** Unprintables (ascii range)
364 *
365 * We don't include characters in the extended range (128-255) as they're
366 * likely part of a multi-byte sequence and we don't want to break UTF8 strings.
367 */
368#define SBUFF_CHAR_UNPRINTABLES_LOW \
369 [0x00] = true, \
370 [0x01] = true, [0x02] = true, [0x03] = true, [0x04] = true, [0x05] = true, [0x06] = true, [0x07] = true, [0x08] = true, \
371 [0x09] = true, [0x0a] = true, [0x0b] = true, [0x0c] = true, [0x0d] = true, [0x0e] = true, [0x0f] = true, [0x10] = true, \
372 [0x11] = true, [0x12] = true, [0x13] = true, [0x14] = true, [0x15] = true, [0x16] = true, [0x17] = true, [0x18] = true, \
373 [0x19] = true, [0x1a] = true, [0x1b] = true, [0x1c] = true, [0x1d] = true, [0x1e] = true, [0x1f] = true, \
374 [0x7f] = true
375
376/** Unprintables (extended range)
377 *
378 * If these characters are being escaped, the function should also be passed
379 * the 'do_utf8' flag.
380 */
381#define SBUFF_CHAR_UNPRINTABLES_EXTENDED \
382 [0x80] = true, \
383 [0x81] = true, [0x82] = true, [0x83] = true, [0x84] = true, [0x85] = true, [0x86] = true, [0x87] = true, [0x88] = true, \
384 [0x89] = true, [0x8a] = true, [0x8b] = true, [0x8c] = true, [0x8d] = true, [0x8e] = true, [0x8f] = true, [0x90] = true, \
385 [0x91] = true, [0x92] = true, [0x93] = true, [0x94] = true, [0x95] = true, [0x96] = true, [0x97] = true, [0x98] = true, \
386 [0x99] = true, [0x9a] = true, [0x9b] = true, [0x9c] = true, [0x9d] = true, [0x9e] = true, [0x9f] = true, [0xa0] = true, \
387 [0xa1] = true, [0xa2] = true, [0xa3] = true, [0xa4] = true, [0xa5] = true, [0xa6] = true, [0xa7] = true, [0xa8] = true, \
388 [0xa9] = true, [0xaa] = true, [0xab] = true, [0xac] = true, [0xad] = true, [0xae] = true, [0xaf] = true, [0xb0] = true, \
389 [0xb1] = true, [0xb2] = true, [0xb3] = true, [0xb4] = true, [0xb5] = true, [0xb6] = true, [0xb7] = true, [0xb8] = true, \
390 [0xb9] = true, [0xba] = true, [0xbb] = true, [0xbc] = true, [0xbd] = true, [0xbe] = true, [0xbf] = true, [0xc0] = true, \
391 [0xc1] = true, [0xc2] = true, [0xc3] = true, [0xc4] = true, [0xc5] = true, [0xc6] = true, [0xc7] = true, [0xc8] = true, \
392 [0xc9] = true, [0xca] = true, [0xcb] = true, [0xcc] = true, [0xcd] = true, [0xce] = true, [0xcf] = true, [0xd0] = true, \
393 [0xd1] = true, [0xd2] = true, [0xd3] = true, [0xd4] = true, [0xd5] = true, [0xd6] = true, [0xd7] = true, [0xd8] = true, \
394 [0xd9] = true, [0xda] = true, [0xdb] = true, [0xdc] = true, [0xdd] = true, [0xde] = true, [0xdf] = true, [0xe0] = true, \
395 [0xe1] = true, [0xe2] = true, [0xe3] = true, [0xe4] = true, [0xe5] = true, [0xe6] = true, [0xe7] = true, [0xe8] = true, \
396 [0xe9] = true, [0xea] = true, [0xeb] = true, [0xec] = true, [0xed] = true, [0xee] = true, [0xef] = true, [0xf0] = true, \
397 [0xf1] = true, [0xf2] = true, [0xf3] = true, [0xf4] = true, [0xf5] = true, [0xf6] = true, [0xf7] = true, [0xf8] = true, \
398 [0xf9] = true, [0xfa] = true, [0xfb] = true, [0xfc] = true, [0xfd] = true, [0xfe] = true, [0xff] = true
399
400/** Generic wrapper macro to return if there's insufficient memory to satisfy the request on the sbuff
401 *
402 */
403#define FR_SBUFF_RETURN(_func, _sbuff, ...) \
404do { \
405 ssize_t _slen; \
406 _slen = _func(_sbuff, ## __VA_ARGS__ ); \
407 if (_slen < 0) return _slen; \
408} while (0)
409
410/** @name Ephemeral copying macros
411 * @{
412 */
413
414/** @cond */
415
416/** Copy another fr_sbuff_t, modifying it.
417 *
418 * @private
419 */
420#define _FR_SBUFF(_sbuff_or_marker, _start, _current, _end, _extend, _eof, _adv_parent) \
421((fr_sbuff_t){ \
422 .buff = fr_sbuff_buff(_sbuff_or_marker), \
423 .start = (_start), \
424 .end = (_end), \
425 .p = (_current), \
426 .is_const = fr_sbuff_ptr(_sbuff_or_marker)->is_const, \
427 .adv_parent = (_adv_parent), \
428 .shifted = 0, \
429 .extend = (_extend), \
430 .eof = (_eof), \
431 .uctx = fr_sbuff_ptr(_sbuff_or_marker)->uctx, \
432 .parent = fr_sbuff_ptr(_sbuff_or_marker) \
433})
434/* @endcond */
435
436/** Create a new sbuff pointing to the same underlying buffer
437 *
438 * - Parent will _NOT_ be advanced by operations on its child.
439 * - Child will have its `start` pointer set to the `p` pointer of the parent.
440 *
441 * @param[in] _sbuff_or_marker to make an ephemeral copy of.
442 */
443#define FR_SBUFF(_sbuff_or_marker) _FR_SBUFF(_sbuff_or_marker, \
444 fr_sbuff_current(_sbuff_or_marker), \
445 fr_sbuff_current(_sbuff_or_marker), \
446 fr_sbuff_end(_sbuff_or_marker), \
447 fr_sbuff_ptr(_sbuff_or_marker)->extend, \
448 fr_sbuff_ptr(_sbuff_or_marker)->eof, \
449 0x00)
450
451/** Create a new sbuff pointing to the same underlying buffer
452 *
453 * - Parent will _NOT_ be advanced by operations on its child.
454 * - Child will have its `start` pointer set to the `start` pointer of the parent.
455 *
456 * @param[in] _sbuff_or_marker to make an ephemeral copy of.
457 */
458#define FR_SBUFF_ABS(_sbuff_or_marker) _FR_SBUFF(_sbuff_or_marker, \
459 fr_sbuff_start(_sbuff_or_marker), \
460 fr_sbuff_current(_sbuff_or_marker), \
461 fr_sbuff_end(_sbuff_or_marker), \
462 fr_sbuff_ptr(_sbuff_or_marker)->extend, \
463 0x00)
464
465/** Create a new sbuff pointing to the same underlying buffer
466 *
467 * The intent of this sbuff type is to allow parsing operations to be performed
468 * on a subset of the buffer data.
469 *
470 * - Parent will _NOT_ be advanced by operations on its child.
471 * - Child will have its `start` pointer set to the `start` pointer of the parent.
472 * - Child will have its `end` pointer set to the `p` pointer of the parent.
473 * - Child will not extend parent.
474 *
475 * @param[in] _sbuff_or_marker to make an ephemeral copy of.
476 */
477#define FR_SBUFF_REPARSE(_sbuff_or_marker) _FR_SBUFF(_sbuff_or_marker, \
478 fr_sbuff_start(_sbuff_or_marker), \
479 fr_sbuff_start(_sbuff_or_marker), \
480 fr_sbuff_current(_sbuff_or_marker), \
481 NULL, \
482 NULL, \
483 0x00)
484
485/** Create a new sbuff pointing to the same underlying buffer
486 *
487 * - Parent `p` pointer will be advanced with child's `p` pointer.
488 * - Child will have its `start` pointer set to the `p` pointer of the parent.
489 *
490 * @param[in] _sbuff_or_marker to make an ephemeral copy of.
491 */
492#define FR_SBUFF_BIND_CURRENT(_sbuff_or_marker) _FR_SBUFF(_sbuff_or_marker, \
493 fr_sbuff_current(_sbuff_or_marker), \
494 fr_sbuff_current(_sbuff_or_marker), \
495 fr_sbuff_end(_sbuff_or_marker), \
496 fr_sbuff_ptr(_sbuff_or_marker)->extend, \
497 fr_sbuff_ptr(_sbuff_or_marker)->eof, \
498 0x01)
499
500/** Create a new sbuff pointing to the same underlying buffer
501 *
502 * - Parent `p` pointer will be advanced with child's `p` pointer.
503 * - Child will have its `start` pointer set to the `start` pointer of the parent.
504 *
505 * @param[in] _sbuff_or_marker to make an ephemeral copy of.
506 */
507#define FR_SBUFF_BIND_CURRENT_ABS(_sbuff_or_marker) FR_SBUFF_ABS(_sbuff_or_marker, \
508 fr_sbuff_start(_sbuff_or_marker), \
509 fr_sbuff_current(_sbuff_or_marker), \
510 fr_sbuff_end(_sbuff_or_marker), \
511 fr_sbuff_ptr(_sbuff_or_marker)->extend, \
512 fr_sbuff_ptr(_sbuff_or_marker)->eof, \
513 0x01)
514
515/** Creates a compound literal to pass into functions which accept a sbuff
516 *
517 * @note The return value of the function should be used to determine how much
518 * data was written to the buffer.
519 *
520 * @param[in] _start of the buffer.
521 * @param[in] _len_or_end Length of the buffer or the end pointer.
522 */
523#define FR_SBUFF_OUT(_start, _len_or_end) \
524((fr_sbuff_t){ \
525 .buff_i = _start, \
526 .start_i = _start, \
527 .end_i = _Generic((_len_or_end), \
528 size_t : (char const *)(_start) + ((size_t)(_len_or_end) - 1), \
529 long : (char const *)(_start) + ((size_t)(_len_or_end) - 1), \
530 int : (char const *)(_start) + ((size_t)(_len_or_end) - 1), \
531 unsigned int : (char const *)(_start) + ((size_t)(_len_or_end) - 1), \
532 char * : (char const *)(_len_or_end), \
533 char const * : (char const *)(_len_or_end) \
534 ), \
535 .p_i = _start, \
536 .is_const = IS_CONST(char *, _start) \
537})
538
539/** Creates a compound literal to pass into functions which accept a sbuff
540 *
541 * @note The return value of the function should be used to determine how much
542 * data was written to the buffer.
543 *
544 * @param[in] _start of the buffer.
545 * @param[in] _len_or_end Length of the buffer or the end pointer.
546 */
547#define FR_SBUFF_IN(_start, _len_or_end) \
548((fr_sbuff_t){ \
549 .buff_i = _start, \
550 .start_i = _start, \
551 .end_i = _Generic((_len_or_end), \
552 size_t : (char const *)(_start) + (size_t)(_len_or_end), \
553 long : (char const *)(_start) + (size_t)(_len_or_end), \
554 int : (char const *)(_start) + (size_t)(_len_or_end), \
555 unsigned int : (char const *)(_start) + (size_t)(_len_or_end), \
556 char * : (char const *)(_len_or_end), \
557 char const * : (char const *)(_len_or_end) \
558 ), \
559 .p_i = _start, \
560 .is_const = IS_CONST(char *, _start) \
561})
562
563#define FR_SBUFF_IN_STR(_start) FR_SBUFF_IN(_start, strlen(_start))
564
565/** Structure to encapsulate a thread local sbuff information
566 *
567 */
568typedef struct {
569 fr_sbuff_t sbuff; //!< Thread local sbuff.
570 fr_sbuff_uctx_talloc_t tctx; //!< Thread local tctx.
571} fr_sbuff_thread_local_t;
572
573static inline int _sbuff_thread_local_free(void *sbtl)
574{
575 return talloc_free(sbtl);
576}
577
578/** Create a function local and thread local extensible sbuff
579 *
580 * @param[out] _sbuff_out Where to write a pointer to the thread local sbuff
581 * @param[in] _init Initial size for the sbuff buffer.
582 * @param[in] _max Maximum size of the sbuff buffer.
583 */
584#define FR_SBUFF_TALLOC_THREAD_LOCAL(_out, _init, _max) \
585do { \
586 static _Thread_local fr_sbuff_thread_local_t *_sbuff_t_local; \
587 if (!_sbuff_t_local) { \
588 fr_sbuff_thread_local_t *sbtl = talloc_zero(NULL, fr_sbuff_thread_local_t); \
589 fr_sbuff_init_talloc(sbtl, &sbtl->sbuff, &sbtl->tctx, _init, _max); \
590 fr_atexit_thread_local(_sbuff_t_local, _sbuff_thread_local_free, sbtl); \
591 *(_out) = &_sbuff_t_local->sbuff; \
592 } else { \
593 fr_sbuff_reset_talloc(&_sbuff_t_local->sbuff); \
594 *(_out) = &_sbuff_t_local->sbuff; \
595 } \
596} while (0)
597
598void fr_sbuff_update(fr_sbuff_t *sbuff, char *new_buff, size_t new_len);
599
600size_t fr_sbuff_shift(fr_sbuff_t *sbuff, size_t shift, bool move_end);
601
602size_t fr_sbuff_extend_file(fr_sbuff_extend_status_t *status, fr_sbuff_t *sbuff, size_t extension);
603
604bool fr_sbuff_eof_file(fr_sbuff_t *sbuff);
605
606size_t fr_sbuff_extend_talloc(fr_sbuff_extend_status_t *status, fr_sbuff_t *sbuff, size_t extension);
607
608int fr_sbuff_trim_talloc(fr_sbuff_t *sbuff, size_t len);
609
610int fr_sbuff_reset_talloc(fr_sbuff_t *sbuff);
611
612static inline void fr_sbuff_terminate(fr_sbuff_t *sbuff)
613{
614 *sbuff->p = '\0';
615}
616
617static inline void _fr_sbuff_init(fr_sbuff_t *out, char const *start, char const *end, bool is_const)
618{
619 if (unlikely(end < start)) end = start; /* Could be an assert? */
620
621 *out = (fr_sbuff_t){
622 .buff_i = start,
623 .start_i = start,
624 .p_i = start,
625 .end_i = end,
626 .is_const = is_const
627 };
628}
629
630/*
631 * GCC is stupid and will warn about output variables
632 * being unnitialised, even if they're not dereferenced.
633 */
634#if defined(__GNUC__) && __GNUC__ >= 11
635DIAG_OFF(maybe-uninitialized)
636#endif
637
638/** Initialise an sbuff around a stack allocated buffer for printing
639 *
640 * Will \0 terminate the output buffer.
641 *
642 * @param[out] _out Pointer to buffer.
643 * @param[in] _start Start of the buffer. Cannot be const.
644 * @param[in] _len_or_end Either an end pointer or the length
645 * of the buffer. 'end' can be const.
646 */
647#define fr_sbuff_init_out(_out, _start, _len_or_end) \
648do { \
649 *(_start) = '\0'; \
650 _fr_sbuff_init(_out, _start, \
651 _Generic((_len_or_end), \
652 size_t : (char const *)(_start) + ((size_t)(_len_or_end) - 1), \
653 long : (char const *)(_start) + ((size_t)(_len_or_end) - 1), \
654 int : (char const *)(_start) + ((size_t)(_len_or_end) - 1), \
655 char * : (char const *)(_len_or_end), \
656 char const * : (char const *)(_len_or_end) \
657 ), \
658 false); \
659} while (0)
660
661#if defined(__GNUC__) && __GNUC__ >= 11
662DIAG_ON(maybe-uninitialized)
663#endif
664
665/** Initialise an sbuff around a stack allocated buffer for parsing
666 *
667 * @param[out] _out Pointer to buffer.
668 * @param[in] _start Start of the buffer.
669 * @param[in] _len_or_end Either an end pointer or the length
670 * of the buffer.
671 */
672#define fr_sbuff_init_in(_out, _start, _len_or_end) \
673_fr_sbuff_init(_out, _start, \
674_Generic((_len_or_end), \
675 size_t : (char const *)(_start) + (size_t)(_len_or_end), \
676 long : (char const *)(_start) + (size_t)(_len_or_end), \
677 int : (char const *)(_start) + (size_t)(_len_or_end), \
678 char * : (char const *)(_len_or_end), \
679 char const * : (char const *)(_len_or_end) \
680), \
681IS_CONST(char *, _start))
682
683/** Initialise a special sbuff which automatically reads in more data as the buffer is exhausted
684 *
685 * @param[out] sbuff to initialise.
686 * @param[out] fctx to initialise. Must have a lifetime >= to the sbuff.
687 * @param[in] buff Temporary buffer to use for storing file contents.
688 * @param[in] len Length of the temporary buffer.
689 * @param[in] file to read from.
690 * @param[in] max The maximum length of data to read from the file.
691 * @return
692 * - The passed sbuff on success.
693 * - NULL on failure.
694 */
695static inline fr_sbuff_t *fr_sbuff_init_file(fr_sbuff_t *sbuff, fr_sbuff_uctx_file_t *fctx,
696 char *buff, size_t len, FILE *file, size_t max)
697{
698 *fctx = (fr_sbuff_uctx_file_t){
699 .file = file,
700 .max = max,
701 .buff_end = buff + len //!< Store the real end
702 };
703
704 *sbuff = (fr_sbuff_t){
705 .buff = buff,
706 .start = buff,
707 .p = buff,
708 .end = buff, //!< Starts with 0 bytes available
709 .extend = fr_sbuff_extend_file,
710 .eof = fr_sbuff_eof_file,
711 .uctx = fctx
712 };
713
714 return sbuff;
715}
716
717/** Initialise a special sbuff which automatically extends as additional data is written
718 *
719 * @param[in] ctx to allocate buffer in.
720 * @param[out] sbuff to initialise.
721 * @param[out] tctx to initialise. Must have a lifetime >= to the sbuff.
722 * @param[in] init The length of the initial buffer, excluding \0 byte.
723 * @param[in] max The maximum length of the buffer.
724 * @return
725 * - The passed sbuff on success.
726 * - NULL on failure.
727 */
728static inline fr_sbuff_t *fr_sbuff_init_talloc(TALLOC_CTX *ctx,
729 fr_sbuff_t *sbuff, fr_sbuff_uctx_talloc_t *tctx,
730 size_t init, size_t max)
731{
732 char *buff;
733
734 *tctx = (fr_sbuff_uctx_talloc_t){
735 .ctx = ctx,
736 .init = init,
737 .max = max
738 };
739
740 /*
741 * Allocate the initial buffer
742 *
743 * We always allocate a buffer so we don't
744 * trigger ubsan errors by performing
745 * arithmetic on NULL pointers.
746 */
747 buff = talloc_zero_array(ctx, char, init + 1);
748 if (!buff) {
749 fr_strerror_printf("Failed allocating buffer of %zu bytes", init + 1);
750 memset(sbuff, 0, sizeof(*sbuff)); /* clang scan */
751 return NULL;
752 }
753
754 *sbuff = (fr_sbuff_t){
755 .buff = buff,
756 .start = buff,
757 .p = buff,
758 .end = buff + init,
759 .extend = fr_sbuff_extend_talloc,
760 .uctx = tctx
761 };
762
763 return sbuff;
764}
765/** @} */
766
767/** @name Accessors
768 *
769 * Caching the values of these pointers or the pointer values from the sbuff
770 * directly is strongly discouraged as they can become invalidated during
771 * stream parsing or when printing to an auto-expanding buffer.
772 *
773 * These functions should only be used to pass sbuff pointers into 3rd party
774 * APIs.
775 *
776 * @{
777 */
778
779/** Return a pointer to the sbuff
780 *
781 * @param[in] _sbuff_or_marker to return a pointer to.
782 * @return A pointer to the sbuff.
783 */
784#define fr_sbuff_ptr(_sbuff_or_marker) \
785 _Generic((_sbuff_or_marker), \
786 fr_sbuff_t * : ((fr_sbuff_t *)(_sbuff_or_marker)), \
787 fr_sbuff_marker_t * : ((fr_sbuff_marker_t *)(_sbuff_or_marker))->parent \
788 )
789
790/** Return a pointer to the start of the underlying buffer in an sbuff or one of its markers
791 *
792 * @param[in] _sbuff_or_marker to return the buffer for.
793 * @return A pointer to the start of the buffer.
794 */
795#define fr_sbuff_buff(_sbuff_or_marker) \
796 _Generic((_sbuff_or_marker), \
797 fr_sbuff_t * : ((fr_sbuff_t const *)(_sbuff_or_marker))->buff, \
798 fr_sbuff_t const * : ((fr_sbuff_t const *)(_sbuff_or_marker))->buff, \
799 fr_sbuff_marker_t * : ((fr_sbuff_marker_t const *)(_sbuff_or_marker))->parent->buff, \
800 fr_sbuff_marker_t const * : ((fr_sbuff_marker_t const *)(_sbuff_or_marker))->parent->buff \
801 )
802
803/** Return a pointer to the 'start' position of an sbuff or one of its markers
804 *
805 * The start position is not necessarily the start of the buffer, and is
806 * advanced every time an sbuff is copied.
807 *
808 * @param[in] _sbuff_or_marker to return the start position of.
809 * @return A pointer to the start position of the buffer.
810 */
811#define fr_sbuff_start(_sbuff_or_marker) \
812 (_Generic((_sbuff_or_marker), \
813 fr_sbuff_t * : ((fr_sbuff_t const *)(_sbuff_or_marker))->start, \
814 fr_sbuff_t const * : ((fr_sbuff_t const *)(_sbuff_or_marker))->start, \
815 fr_sbuff_marker_t * : ((fr_sbuff_marker_t const *)(_sbuff_or_marker))->parent->start, \
816 fr_sbuff_marker_t const * : ((fr_sbuff_marker_t const *)(_sbuff_or_marker))->parent->start \
817 ))
818
819/** Return a pointer to the 'current' position of an sbuff or one of its markers
820 *
821 * @note Should not be derferenced as it may point past the end of the buffer.
822 * Use #fr_sbuff_char to get the current char.
823 *
824 * @param[in] _sbuff_or_marker to return the current position of.
825 * @return A pointer to the current position of the buffer or marker.
826 */
827#define fr_sbuff_current(_sbuff_or_marker) \
828 (_Generic((_sbuff_or_marker), \
829 fr_sbuff_t * : ((fr_sbuff_t const *)(_sbuff_or_marker))->p, \
830 fr_sbuff_t const * : ((fr_sbuff_t const *)(_sbuff_or_marker))->p, \
831 fr_sbuff_marker_t * : ((fr_sbuff_marker_t const *)(_sbuff_or_marker))->p, \
832 fr_sbuff_marker_t const * : ((fr_sbuff_marker_t const *)(_sbuff_or_marker))->p \
833 ))
834
835/** Return a pointer to the 'end' position of an sbuff or one of its markers
836 *
837 * @param[in] _sbuff_or_marker to return the end position of.
838 * @return A pointer to the end position of the buffer or marker.
839 */
840#define fr_sbuff_end(_sbuff_or_marker) \
841 (_Generic((_sbuff_or_marker), \
842 fr_sbuff_t * : ((fr_sbuff_t const *)(_sbuff_or_marker))->end, \
843 fr_sbuff_t const * : ((fr_sbuff_t const *)(_sbuff_or_marker))->end, \
844 fr_sbuff_marker_t * : ((fr_sbuff_marker_t const *)(_sbuff_or_marker))->parent->end, \
845 fr_sbuff_marker_t const * : ((fr_sbuff_marker_t const *)(_sbuff_or_marker))->parent->end \
846 ))
847
848/** Return the value of the shifted field
849 *
850 * @param[in] _sbuff_or_marker to return the position of.
851 * @return the number of bytes the buffer has been shifted.
852 */
853#define fr_sbuff_shifted(_sbuff_or_marker) \
854 (_Generic((_sbuff_or_marker), \
855 fr_sbuff_t * : ((fr_sbuff_t const *)(_sbuff_or_marker))->shifted, \
856 fr_sbuff_t const * : ((fr_sbuff_t const *)(_sbuff_or_marker))->shifted, \
857 fr_sbuff_marker_t * : ((fr_sbuff_marker_t const *)(_sbuff_or_marker))->parent->shifted, \
858 fr_sbuff_marker_t const * : ((fr_sbuff_marker_t const *)(_sbuff_or_marker))->parent->shifted \
859 ))
860
861/** Return the current char pointed to by the sbuff or '\0' if no more chars remain
862 *
863 * @note Should be used in place of #fr_sbuff_current if switching over the current char.
864 *
865 * @param[in] _sbuff_or_marker to return the current char from.
866 * @param[in] _eob char used to indicate End of Buffer, usually '\0'.
867 * @return The current char pointed to be the sbuff.
868 */
869#define fr_sbuff_char(_sbuff_or_marker, _eob) \
870 (fr_sbuff_current(_sbuff_or_marker) >= fr_sbuff_end(_sbuff_or_marker) ? _eob : *fr_sbuff_current(_sbuff_or_marker))
871
872/** Start a switch block over the current sbuff char
873 *
874 * @note '\0' is used to indicate EOB.
875 *
876 * @param[in] _sbuff_or_marker to return the current char from.
877 * @param[in] _eob char used to indicate End of Buffer, usually '\0'.
878 */
879#define fr_sbuff_switch(_sbuff_or_marker, _eob) \
880 switch (fr_sbuff_char(_sbuff_or_marker, _eob))
881/** @} */
882
883/** @name Length calculations
884 * @{
885 */
886
887/** Return the difference in position between the two sbuffs or markers
888 *
889 * @param[in] _a The first sbuff or marker.
890 * @param[in] _b The second sbuff or marker.
891 * @return
892 * - >0 the number of bytes _a is ahead of _b.
893 * - 0 _a and _b are the same position.
894 * - <0 the number of bytes _a is behind of _b.
895 */
896#define fr_sbuff_diff(_a, _b) \
897 ((ssize_t)(fr_sbuff_current(_a) - fr_sbuff_current(_b)))
898
899/** Return the number of bytes remaining between the sbuff or marker and the end of the buffer
900 *
901 * @note Do not use this in functions that may be used for stream parsing
902 * unless you're sure you know what you're doing.
903 * The value return does not reflect the number of bytes that may
904 * be potentially read from the stream, only the number of bytes
905 * until the end of the current chunk.
906 *
907 * @param[in] _sbuff_or_marker to return the number of bytes remaining for.
908 * @return
909 * - >0 the number of bytes remaining before we reach the end of the buffer.
910 * - -0 we're at the end of the buffer.
911 */
912#define fr_sbuff_remaining(_sbuff_or_marker) \
913 ((size_t)(fr_sbuff_end(_sbuff_or_marker) < fr_sbuff_current(_sbuff_or_marker) ? \
914 0 : (fr_sbuff_end(_sbuff_or_marker) - fr_sbuff_current(_sbuff_or_marker))))
915
916/** Return the number of bytes remaining between the start of the sbuff or marker and the current position
917 *
918 * @param[in] _sbuff_or_marker to return the number of bytes used for.
919 * @return
920 * - >0 the number of bytes the current position has advanced past the start.
921 * - -0 the current position is at the start of the buffer.
922 */
923#define fr_sbuff_used(_sbuff_or_marker) \
924 ((size_t)(fr_sbuff_start(_sbuff_or_marker) > fr_sbuff_current(_sbuff_or_marker) ? \
925 0 : (fr_sbuff_current(_sbuff_or_marker) - fr_sbuff_start(_sbuff_or_marker))))
926
927/** Sets an error marker in the parent
928 *
929 * If an error already exists at this level it will be used instead of the provided error.
930 *
931 * @param[in] sbuff who's parent we'll set the error marker in.
932 * @param[in] err marker to set.
933 * @return <0 the negative offset of the error.
934 */
935static inline fr_slen_t _fr_sbuff_error(fr_sbuff_t *sbuff, char const *err)
936{
937 fr_sbuff_t *parent = sbuff->parent;
938 fr_slen_t slen;
939
940 if (sbuff->err) err = sbuff->err;
941 if (parent) parent->err = err;
942
943 slen = -((err - fr_sbuff_start(sbuff)) + 1);
944
945#ifdef __clang_analyzer__
946 /*
947 * Convince clang that the return value
948 * is always negative. It never can be
949 * else the sbuff code is very broken.
950 */
951 if (slen >= 0) return -1;
952#endif
953
954 return slen;
955}
956
957/** Return the current position as an error marker
958 *
959 * @param[in] _sbuff_or_marker Error marker will be set from the current position of this sbuff.
960 *
961 * +1 is added to the position to disambiguate with 0 meaning "parsed no data".
962 *
963 * An error at offset 0 will be returned as -1.
964 */
965#define fr_sbuff_error(_sbuff_or_marker) \
966 _fr_sbuff_error(fr_sbuff_ptr(_sbuff_or_marker), fr_sbuff_current(_sbuff_or_marker))
967
968/** Like fr_sbuff_used, but adjusts for the value returned for the amount shifted
969 *
970 * @param[in] _sbuff_or_marker to return the number of bytes used for.
971 * @return
972 * - >0 the number of bytes the current position has advanced past the start +
973 * the amount the buffer has shifted.
974 * - -0 the current position is at the start of the buffer (and hasn't been shifted).
975 */
976#define fr_sbuff_used_total(_sbuff_or_marker) \
977 ((size_t)((fr_sbuff_current(_sbuff_or_marker) + fr_sbuff_shifted(_sbuff_or_marker)) - fr_sbuff_start(_sbuff_or_marker)))
978
979/** The length of the underlying buffer
980 *
981 * @param[in] _sbuff_or_marker to return the length of.
982 * @return The length of the underlying buffer (minus 1 byte for \0).
983 */
984#define fr_sbuff_len(_sbuff_or_marker) \
985 ((size_t)(fr_sbuff_end(_sbuff_or_marker) - fr_sbuff_buff(_sbuff_or_marker)))
986
987/** How many bytes the sbuff or marker is behind its parent
988 *
989 * @param[in] _sbuff_or_marker
990 * @return
991 * - 0 the sbuff or marker is ahead of its parent.
992 * - >0 the number of bytes the marker is behind its parent.
993 */
994#define fr_sbuff_behind(_sbuff_or_marker) \
995 (fr_sbuff_current(_sbuff_or_marker) > fr_sbuff_current((_sbuff_or_marker)->parent) ? \
996 0 : fr_sbuff_current((_sbuff_or_marker)->parent) - fr_sbuff_current(_sbuff_or_marker))
997
998/** How many bytes the sbuff or marker is ahead of its parent
999 *
1000 * @return
1001 * - 0 the sbuff or marker is behind its parent.
1002 * - >0 the number of bytes the marker is ahead of its parent.
1003 */
1004#define fr_sbuff_ahead(_sbuff_or_marker) \
1005 (fr_sbuff_current((_sbuff_or_marker)->parent) > fr_sbuff_current(_sbuff_or_marker) ? \
1006 0 : fr_sbuff_current(_sbuff_or_marker) - fr_sbuff_current((_sbuff_or_marker)->parent))
1007
1008/** Return the current position in the sbuff as a negative offset
1009 *
1010 */
1011#define FR_SBUFF_ERROR_RETURN(_sbuff_or_marker) return fr_sbuff_error(_sbuff_or_marker)
1012
1013/** Check if _len bytes are available in the sbuff, and if not return the number of bytes we'd need
1014 *
1015 */
1016#define FR_SBUFF_CHECK_REMAINING_RETURN(_sbuff, _len) \
1017 if ((_len) > fr_sbuff_remaining(_sbuff)) return -((_len) - fr_sbuff_remaining(_sbuff))
1018
1019static inline size_t _fr_sbuff_extend_lowat(fr_sbuff_extend_status_t *status, fr_sbuff_t *in, size_t remaining, size_t lowat)
1020{
1021 size_t extended;
1022 fr_sbuff_extend_status_t our_status = 0;
1023
1024 if (!fr_sbuff_is_extendable(in)) {
1025 no_extend:
1026 if (status) *status = our_status;
1027 return remaining;
1028 }
1029
1030 /* Still have data remaining, no need to try and extend */
1031 if (remaining >= lowat) goto no_extend;
1032
1033 if (!in->extend || ((extended = in->extend(&our_status, in, lowat - remaining)) == 0)) {
1034 goto no_extend;
1035 }
1036
1037 our_status |= FR_SBUFF_FLAG_EXTENDED;
1038
1039 if (status) *status = our_status;
1040 return remaining + extended;
1041}
1042
1043/** Extend a buffer if we're below the low water mark
1044 *
1045 * @param[out] _status Should be initialised to FR_SBUFF_EXTENDABLE
1046 * for the first call to this function if used
1047 * as a loop condition.
1048 * Will be filled with the result of the previous
1049 * call, and can be used to determine if the buffer
1050 * was extended.
1051 * @param[in] _sbuff_or_marker to extend.
1052 * @param[in] _lowat If bytes remaining are below the amount, extend.
1053 * @return
1054 * - 0 if there are no bytes left in the buffer and we couldn't extend.
1055 * - >0 the number of bytes in the buffer after extending.
1056 */
1057#define fr_sbuff_extend_lowat(_status, _sbuff_or_marker, _lowat) \
1058 _fr_sbuff_extend_lowat(_status, \
1059 fr_sbuff_ptr(_sbuff_or_marker), \
1060 fr_sbuff_remaining(_sbuff_or_marker), _lowat)
1061
1062/** Check if _len bytes are available in the sbuff and extend the buffer if possible
1063 *
1064 * If we do not have _len bytes in the sbuff after extending, then return.
1065 *
1066 * @param[in] _sbuff to extend.
1067 * @param[in] _len The minimum amount the sbuff should be extended by.
1068 * @return The number of bytes we would need to satisfy _len as a negative integer.
1069 */
1070#define FR_SBUFF_EXTEND_LOWAT_OR_RETURN(_sbuff, _len) \
1071do { \
1072 size_t _remaining = fr_sbuff_extend_lowat(NULL, _sbuff, _len); \
1073 if (_remaining < _len) return -(_len - _remaining); \
1074} while (0)
1075
1076/** Extend a buffer if no space remains
1077 *
1078 * @param[in] _sbuff_or_marker to extend.
1079 * @return
1080 * - 0 if there are no bytes left in the buffer and we couldn't extend.
1081 * - >0 the number of bytes in the buffer after extending.
1082 */
1083#define fr_sbuff_extend(_sbuff_or_marker) fr_sbuff_extend_lowat(NULL, _sbuff_or_marker, 1)
1084
1085/** @} */
1086
1087/** @name Position modification (recursive)
1088 *
1089 * Change the current position of pointers in the sbuff and their children.
1090 * @{
1091 */
1092
1093/** Update the position of p in a list of sbuffs
1094 *
1095 * @note Do not call directly.
1096 */
1097static inline void _fr_sbuff_set_recurse(fr_sbuff_t *sbuff, char const *p)
1098{
1099 sbuff->p_i = p;
1100 sbuff->err = NULL; /* Modifying the position of the sbuff clears the error */
1101
1102 if (sbuff->adv_parent && sbuff->parent) _fr_sbuff_set_recurse(sbuff->parent, p);
1103}
1104
1105static inline ssize_t _fr_sbuff_marker_set(fr_sbuff_marker_t *m, char const *p)
1106{
1107 fr_sbuff_t *sbuff = m->parent;
1108 char *current = m->p;
1109
1110 if (unlikely(p > sbuff->end)) return -(p - sbuff->end);
1111 if (unlikely(p < sbuff->start)) return 0;
1112
1113 sbuff->err = NULL; /* Modifying the position of any markers clears the error, unsure if this is correct? */
1114 m->p_i = p;
1115
1116 return p - current;
1117}
1118
1119static inline ssize_t _fr_sbuff_set(fr_sbuff_t *sbuff, char const *p)
1120{
1121 char const *c;
1122
1123 if (unlikely(p > sbuff->end)) return -(p - sbuff->end);
1124 if (unlikely(p < sbuff->start)) return 0;
1125
1126 c = sbuff->p;
1127 _fr_sbuff_set_recurse(sbuff, p);
1128
1129 return p - c;
1130}
1131
1132/** Set the position in a sbuff using another sbuff, a char pointer, or a length
1133 *
1134 * @param[in] _dst sbuff or marker to set the position for.
1135 * @param[in] _src Variable to glean new position from. Behaviour here
1136 * depends on the type of the variable.
1137 * - sbuff, the current position of the sbuff.
1138 * - marker, the current position of the marker.
1139 * - pointer, the position of the pointer.
1140 * - size_t, _dst->start + _src.
1141 * @return
1142 * - 0 not advanced.
1143 * - >0 the number of bytes the sbuff was advanced by.
1144 * - <0 the number of bytes required to complete the advancement
1145 */
1146#define fr_sbuff_set(_dst, _src) \
1147_Generic((_dst), \
1148 fr_sbuff_t * : _fr_sbuff_set, \
1149 fr_sbuff_marker_t * : _fr_sbuff_marker_set \
1150)(_dst, \
1151_Generic((_src), \
1152 fr_sbuff_t * : fr_sbuff_current((fr_sbuff_t const *)(_src)), \
1153 fr_sbuff_t const * : fr_sbuff_current((fr_sbuff_t const *)(_src)), \
1154 fr_sbuff_marker_t * : fr_sbuff_current((fr_sbuff_marker_t const *)(_src)), \
1155 fr_sbuff_marker_t const * : fr_sbuff_current((fr_sbuff_marker_t const *)(_src)), \
1156 char const * : (char const *)(_src), \
1157 char * : (char const *)(_src), \
1158 size_t : (fr_sbuff_start(_dst) + (uintptr_t)(_src)) \
1159))
1160
1161/** Set an sbuff to a given position, and return how much it was advanced
1162 *
1163 * @param[in] _dst to modify.
1164 * @param[in] _src position to set it too.
1165 * @return The number of bytes _dst was advanced.
1166 */
1167#define FR_SBUFF_SET_RETURN(_dst, _src) return fr_sbuff_set(_dst, _src)
1168
1169/** Advance position in sbuff by N bytes
1170 *
1171 * @param[in] _sbuff_or_marker to advance.
1172 * @param[in] _len How much to advance sbuff by.
1173 * @return
1174 * - 0 not advanced.
1175 * - >0 the number of bytes the sbuff was advanced by.
1176 * - <0 the number of bytes required to complete the advancement
1177 */
1178#define fr_sbuff_advance(_sbuff_or_marker, _len) fr_sbuff_set(_sbuff_or_marker, (fr_sbuff_current(_sbuff_or_marker) + (_len)))
1179#define FR_SBUFF_ADVANCE_RETURN(_sbuff, _len) FR_SBUFF_RETURN(fr_sbuff_advance, _sbuff, _len)
1180
1181/** Reset the current position of the sbuff to the start of the string
1182 *
1183 */
1184static inline void fr_sbuff_set_to_start(fr_sbuff_t *sbuff)
1185{
1186 _fr_sbuff_set_recurse(sbuff, sbuff->start);
1187}
1188
1189/** Reset the current position of the sbuff to the end of the string
1190 *
1191 */
1192static inline void fr_sbuff_set_to_end(fr_sbuff_t *sbuff)
1193{
1194 _fr_sbuff_set_recurse(sbuff, sbuff->end);
1195}
1196/** @} */
1197
1198/** @name Add a marker to an sbuff
1199 *
1200 * Markers are used to indicate an area of the code is working at a particular
1201 * point in a string buffer.
1202 *
1203 * If the sbuff is performing stream parsing, then markers are used to update
1204 * any pointers to the buffer, as the data in the buffer is shifted to make
1205 * room for new data from the stream.
1206 *
1207 * If the sbuff is being used to create strings, then the markers are updated
1208 * if the buffer is re-allocated.
1209 * @{
1210 */
1211
1212/** Adds a new pointer to the beginning of the list of pointers to update
1213 *
1214 * @param[out] m to initialise.
1215 * @param[in] sbuff to associate marker with.
1216 * @return The position the marker was set to.
1217 */
1218static inline char *fr_sbuff_marker(fr_sbuff_marker_t *m, fr_sbuff_t *sbuff)
1219{
1220 *m = (fr_sbuff_marker_t){
1221 .next = sbuff->m, /* Link into the head */
1222 .p = sbuff->p, /* Set the current position in the sbuff */
1223 .parent = sbuff /* Record which sbuff this marker was associated with */
1224 };
1225 sbuff->m = m;
1226
1227 return sbuff->p;
1228}
1229
1230/** Used to update the position of an 'end' position marker
1231 *
1232 * Updates a marker so that it represents a 'constrained' end.
1233 * If max > fr_sbuff_remaining() + fr_sbuff_used_total(), then the marker will
1234 * be set to the end of the sbuff.
1235 *
1236 * Otherwise the marker will be set to the position indicated by
1237 * start + (max - fr_sbuff_used_total()).
1238 *
1239 * This is used to add a constraint on the amount data that can be copied from
1240 * an extendable buffer.
1241 *
1242 * @param[in] m Marker created with #fr_sbuff_marker.
1243 * @param[in] max amount of data we allow to be read from
1244 * the sbuff. May be SIZE_MAX in which
1245 * case there is no constraint imposed
1246 * and the marker is set to the end of the buffer.
1247 */
1248static inline char *fr_sbuff_marker_update_end(fr_sbuff_marker_t *m, size_t max)
1249{
1250 fr_sbuff_t *sbuff = m->parent;
1251 size_t used = fr_sbuff_used_total(sbuff);
1252
1253 m->p = (((max) - (used)) > fr_sbuff_remaining(sbuff) ?
1254 fr_sbuff_end(sbuff) :
1255 fr_sbuff_current(sbuff) + ((max) - (used)));
1256
1257 return m->p;
1258}
1259
1260/** Trims the linked list back to the specified pointer
1261 *
1262 * Pointers should be released in the inverse order to allocation.
1263 *
1264 * Alternatively the oldest pointer can be released, resulting in any newer pointer
1265 * also being removed from the list.
1266 *
1267 * @param[in] m to release.
1268 */
1269static inline void fr_sbuff_marker_release(fr_sbuff_marker_t *m)
1270{
1271 m->parent->m = m->next;
1272
1273#ifndef NDEBUF
1274 memset(m, 0, sizeof(*m)); /* Use after release */
1275#endif
1276}
1277
1278/** Trims the linked list back to the specified pointer and return how many bytes marker was behind p
1279 *
1280 * Pointers should be released in the inverse order to allocation.
1281 *
1282 * Alternatively the oldest pointer can be released, resulting in any newer pointer
1283 * also being removed from the list.
1284 *
1285 * @param[in] m to release.
1286 * @return
1287 * - 0 marker is ahead of p.
1288 * - >0 the number of bytes the marker is behind p.
1289 */
1290static inline size_t fr_sbuff_marker_release_behind(fr_sbuff_marker_t *m)
1291{
1292 size_t len = fr_sbuff_behind(m);
1293 fr_sbuff_marker_release(m);
1294 return len;
1295}
1296
1297/** Trims the linked list back to the specified pointer and return how many bytes marker was ahead of p
1298 *
1299 * Pointers should be released in the inverse order to allocation.
1300 *
1301 * Alternatively the oldest pointer can be released, resulting in any newer pointer
1302 * also being removed from the list.
1303 *
1304 * @param[in] m to release.
1305 * @return
1306 * - 0 marker is ahead of p.
1307 * - >0 the number of bytes the marker is behind p.
1308 */
1309static inline size_t fr_sbuff_marker_release_ahead(fr_sbuff_marker_t *m)
1310{
1311 size_t len = fr_sbuff_ahead(m);
1312 fr_sbuff_marker_release(m);
1313 return len;
1314}
1315
1316/** Trims the linked list back to the specified pointer and return how many bytes marker was behind p
1317 *
1318 * Pointers should be released in the inverse order to allocation.
1319 *
1320 * Alternatively the oldest pointer can be released, resulting in any newer pointer
1321 * also being removed from the list.
1322 *
1323 * @param[in] m to release.
1324 * @return
1325 * - 0 marker is ahead of p.
1326 * - >0 the number of bytes the marker is behind p.
1327 */
1328static inline size_t fr_sbuff_marker_release_reset_behind(fr_sbuff_marker_t *m)
1329{
1330 size_t len = fr_sbuff_behind(m);
1331 fr_sbuff_set(m->parent, m);
1332 fr_sbuff_marker_release(m);
1333 return len;
1334}
1335
1336/** Trims the linked list back to the specified pointer and return how many bytes marker was ahead of p
1337 *
1338 * Pointers should be released in the inverse order to allocation.
1339 *
1340 * Alternatively the oldest pointer can be released, resulting in any newer pointer
1341 * also being removed from the list.
1342 *
1343 * @param[in] m to release.
1344 * @return
1345 * - 0 marker is ahead of p.
1346 * - >0 the number of bytes the marker is behind p.
1347 */
1348static inline size_t fr_sbuff_marker_release_reset_ahead(fr_sbuff_marker_t *m)
1349{
1350 size_t len = fr_sbuff_ahead(m);
1351 fr_sbuff_set(m->parent, m);
1352 fr_sbuff_marker_release(m);
1353 return len;
1354}
1355/** @} */
1356
1357/** @name Copy data between an sbuff/marker
1358 *
1359 * These functions are typically used for moving data between sbuffs
1360 *
1361 * @{
1362 */
1363size_t _fr_sbuff_move_sbuff_to_sbuff(fr_sbuff_t *out, fr_sbuff_t *in, size_t len);
1364
1365size_t _fr_sbuff_move_marker_to_sbuff(fr_sbuff_t *out, fr_sbuff_marker_t *in, size_t len);
1366
1367size_t _fr_sbuff_move_marker_to_marker(fr_sbuff_marker_t *out, fr_sbuff_marker_t *in, size_t len);
1368
1369size_t _fr_sbuff_move_sbuff_to_marker(fr_sbuff_marker_t *out, fr_sbuff_t *in, size_t len);
1370
1371/** Copy in as many bytes as possible from one sbuff or marker to another
1372 *
1373 * @param[in] _out to copy into.
1374 * @param[in] _in to copy from.
1375 * @param[in] _len The maximum length to copy.
1376 * @return Number of bytes copied.
1377 */
1378#define fr_sbuff_move(_out, _in, _len) \
1379 _Generic((_out), \
1380 fr_sbuff_t * : \
1381 _Generic((_in), \
1382 fr_sbuff_t * : _fr_sbuff_move_sbuff_to_sbuff((fr_sbuff_t *)_out, (fr_sbuff_t *)_in, _len), \
1383 fr_sbuff_marker_t * : _fr_sbuff_move_marker_to_sbuff((fr_sbuff_t *)_out, (fr_sbuff_marker_t *)_in, _len) \
1384 ), \
1385 fr_sbuff_marker_t * : \
1386 _Generic((_in), \
1387 fr_sbuff_t * : _fr_sbuff_move_sbuff_to_marker((fr_sbuff_marker_t *)_out, (fr_sbuff_t *)_in, _len), \
1388 fr_sbuff_marker_t * : _fr_sbuff_move_marker_to_marker((fr_sbuff_marker_t *)_out, (fr_sbuff_marker_t *)_in, _len) \
1389 ) \
1390 )
1391/** @} */
1392
1393/** @name Copy/print complete input data to an sbuff
1394 *
1395 * These functions are typically used for printing.
1396 *
1397 * @{
1398 */
1399#define fr_sbuff_in_char(_sbuff, ...) fr_sbuff_in_bstrncpy(_sbuff, ((char []){ __VA_ARGS__ }), sizeof((char []){ __VA_ARGS__ }))
1400#define FR_SBUFF_IN_CHAR_RETURN(_sbuff, ...) FR_SBUFF_RETURN(fr_sbuff_in_bstrncpy, _sbuff, ((char []){ __VA_ARGS__ }), sizeof((char []){ __VA_ARGS__ }))
1401
1402ssize_t fr_sbuff_in_strcpy(fr_sbuff_t *sbuff, char const *str);
1403#define FR_SBUFF_IN_STRCPY_RETURN(...) FR_SBUFF_RETURN(fr_sbuff_in_strcpy, ##__VA_ARGS__)
1404
1405ssize_t fr_sbuff_in_bstrncpy(fr_sbuff_t *sbuff, char const *str, size_t len);
1406#define FR_SBUFF_IN_BSTRNCPY_RETURN(...) FR_SBUFF_RETURN(fr_sbuff_in_bstrncpy, ##__VA_ARGS__)
1407
1408#define fr_sbuff_in_strcpy_literal(_sbuff, _str) fr_sbuff_in_bstrncpy(_sbuff, _str, sizeof(_str) - 1)
1409#define FR_SBUFF_IN_STRCPY_LITERAL_RETURN(_sbuff, _str) FR_SBUFF_RETURN(fr_sbuff_in_bstrncpy, _sbuff, _str, sizeof(_str) - 1)
1410
1411ssize_t fr_sbuff_in_bstrcpy_buffer(fr_sbuff_t *sbuff, char const *str);
1412#define FR_SBUFF_IN_BSTRCPY_BUFFER_RETURN(...) FR_SBUFF_RETURN(fr_sbuff_in_bstrcpy_buffer, ##__VA_ARGS__)
1413
1414ssize_t fr_sbuff_in_vsprintf(fr_sbuff_t *sbuff, char const *fmt, va_list ap);
1415#define FR_SBUFF_IN_VSPRINTF_RETURN(...) FR_SBUFF_RETURN(fr_sbuff_in_vsprintf, ##__VA_ARGS__)
1416
1417ssize_t fr_sbuff_in_sprintf(fr_sbuff_t *sbuff, char const *fmt, ...);
1418#define FR_SBUFF_IN_SPRINTF_RETURN(...) FR_SBUFF_RETURN(fr_sbuff_in_sprintf, ##__VA_ARGS__)
1419
1420ssize_t fr_sbuff_in_escape(fr_sbuff_t *sbuff, char const *in, size_t inlen, fr_sbuff_escape_rules_t const *e_rules);
1421#define FR_SBUFF_IN_ESCAPE_RETURN(...) FR_SBUFF_RETURN(fr_sbuff_in_escape, ##__VA_ARGS__)
1422
1423ssize_t fr_sbuff_in_escape_buffer(fr_sbuff_t *sbuff, char const *in, fr_sbuff_escape_rules_t const *e_rules);
1424#define FR_SBUFF_IN_ESCAPE_BUFFER_RETURN(...) FR_SBUFF_RETURN(fr_sbuff_in_escape_buffer, ##__VA_ARGS__)
1425
1426ssize_t fr_sbuff_in_array(fr_sbuff_t *sbuff, char const * const *array, char const *sep);
1427#define FR_SBUFF_IN_ARRAY(...) FR_SBUFF_RETURN(fr_sbuff_in_array, ##__VA_ARGS__)
1428
1429/** Lookup a string in a table using an integer value, and copy it to the sbuff
1430 *
1431 * @param[out] _slen Where to write the return value.
1432 * @param[in] _sbuff to search in.
1433 * @param[in] _table to search for number in.
1434 * @param[in] _number to search for.
1435 * @param[in] _def Default string value.
1436 */
1437#define fr_sbuff_in_table_str(_slen, _sbuff, _table, _number, _def) \
1438 _slen = fr_sbuff_in_strcpy(_sbuff, fr_table_str_by_value(_table, _number, _def))
1439#define FR_SBUFF_IN_TABLE_STR_RETURN(_sbuff, _table, _number, _def) \
1440do { \
1441 ssize_t _slen; \
1442 fr_sbuff_in_table_str(_slen, _sbuff, _table, _number, _def); \
1443 if (_slen < 0) return _slen; \
1444} while (0)
1445/** @} */
1446
1447/** @name Copy data out of an sbuff
1448 *
1449 * These functions are typically used for parsing.
1450 *
1451 * @{
1452 */
1453
1454/** Toggle any chars to 'true' in out, that were present in, out or in
1455 *
1456 */
1457static inline void fr_sbuff_allowed_merge(bool out[static UINT8_MAX + 1], bool const in[static UINT8_MAX + 1])
1458{
1459 for (size_t i = 0; i <= UINT8_MAX; i++) out[i] = out[i] || in[i];
1460}
1461
1462fr_sbuff_term_t *fr_sbuff_terminals_amerge(TALLOC_CTX *ctx,
1463 fr_sbuff_term_t const *a, fr_sbuff_term_t const *b);
1464
1465size_t fr_sbuff_out_bstrncpy(fr_sbuff_t *out, fr_sbuff_t *in, size_t len);
1466
1467ssize_t fr_sbuff_out_bstrncpy_exact(fr_sbuff_t *out, fr_sbuff_t *in, size_t len);
1468
1469size_t fr_sbuff_out_bstrncpy_allowed(fr_sbuff_t *out, fr_sbuff_t *in, size_t len,
1470 bool const allowed[static UINT8_MAX + 1]);
1471
1472size_t fr_sbuff_out_bstrncpy_until(fr_sbuff_t *out, fr_sbuff_t *in, size_t len,
1473 fr_sbuff_term_t const *tt,
1474 fr_sbuff_unescape_rules_t const *u_rules);
1475
1476size_t fr_sbuff_out_unescape_until(fr_sbuff_t *out, fr_sbuff_t *in, size_t len,
1477 fr_sbuff_term_t const *tt,
1478 fr_sbuff_unescape_rules_t const *u_rules);
1479
1480/** Find the longest prefix in an sbuff
1481 *
1482 * @param[out] _match_len The length of the matched string.
1483 * May be NULL.
1484 * @param[out] _out The value resolved in the table.
1485 * @param[in] _table to find longest match in.
1486 * @param[in] _sbuff containing the needle.
1487 * @param[in] _def Default value if no match is found.
1488 */
1489#define fr_sbuff_out_by_longest_prefix(_match_len, _out, _table, _sbuff, _def) \
1490do { \
1491 size_t _match_len_tmp; \
1492 fr_sbuff_extend_lowat(NULL, _sbuff, fr_table_max_needle_len(_table)); \
1493 *(_out) = fr_table_value_by_longest_prefix(&_match_len_tmp, _table, \
1494 fr_sbuff_current(_sbuff), fr_sbuff_remaining(_sbuff), \
1495 _def); \
1496 (void) fr_sbuff_advance(_sbuff, _match_len_tmp); /* can't fail */ \
1497 *(_match_len) = _match_len_tmp; \
1498} while (0)
1499
1500/** Build a talloc wrapper function for a fr_sbuff_out_* function
1501 *
1502 * @param[in] _func to call.
1503 * @param[in] _in input sbuff arg.
1504 * @param[in] _len expected output len.
1505 * @param[in] ... additional arguments to pass to _func.
1506 */
1507#define SBUFF_OUT_TALLOC_ERR_FUNC_DEF(_func, _in, _len, ...) \
1508{ \
1509 fr_sbuff_t sbuff; \
1510 fr_sbuff_uctx_talloc_t tctx; \
1511 fr_sbuff_parse_error_t err; \
1512 fr_slen_t slen = -1; \
1513 if (unlikely(fr_sbuff_init_talloc(ctx, &sbuff, &tctx, \
1514 ((_len) != SIZE_MAX) ? (_len) : 1024, \
1515 ((_len) != SIZE_MAX) ? (_len) : SIZE_MAX) == NULL)) { \
1516 error: \
1517 TALLOC_FREE(sbuff.buff); \
1518 *out = NULL; \
1519 return slen; \
1520 } \
1521 slen = _func(&err, &sbuff, _in, _len, ##__VA_ARGS__); \
1522 if (slen < 0) goto error; \
1523 if (unlikely(fr_sbuff_trim_talloc(&sbuff, SIZE_MAX) < 0)) { \
1524 slen = -1; \
1525 goto error; \
1526 } \
1527 *out = sbuff.buff; \
1528 return slen; \
1529}
1530
1531/** Build a talloc wrapper function for a fr_sbuff_out_* function
1532 *
1533 * @param[in] _func to call.
1534 * @param[in] _in input sbuff arg.
1535 * @param[in] _len expected output len.
1536 * @param[in] ... additional arguments to pass to _func.
1537 */
1538#define SBUFF_OUT_TALLOC_FUNC_DEF(_func, _in, _len, ...) \
1539{ \
1540 fr_sbuff_t sbuff; \
1541 fr_sbuff_uctx_talloc_t tctx; \
1542 fr_slen_t slen = -1; \
1543 if (unlikely(fr_sbuff_init_talloc(ctx, &sbuff, &tctx, \
1544 ((_len) != SIZE_MAX) ? (_len) : 1024, \
1545 ((_len) != SIZE_MAX) ? (_len) : SIZE_MAX) == NULL)) { \
1546 error: \
1547 TALLOC_FREE(sbuff.buff); \
1548 *out = NULL; \
1549 return slen; \
1550 } \
1551 slen = _func(&sbuff, _in, _len, ##__VA_ARGS__); \
1552 if (slen < 0) goto error; \
1553 if (unlikely(fr_sbuff_trim_talloc(&sbuff, SIZE_MAX) < 0)) { \
1554 slen = -1; \
1555 goto error; \
1556 } \
1557 *out = sbuff.buff; \
1558 return slen; \
1559}
1560
1561/** Build a talloc wrapper function for a fr_sbuff_out_* function
1562 *
1563 * @param[in] _func to call.
1564 * @param[in] ... additional arguments to pass to _func.
1565 */
1566#define SBUFF_OUT_TALLOC_FUNC_NO_LEN_DEF(_func, ...) \
1567{ \
1568 fr_sbuff_t sbuff; \
1569 fr_sbuff_uctx_talloc_t tctx; \
1570 fr_slen_t slen = -1; \
1571 if (unlikely(fr_sbuff_init_talloc(ctx, &sbuff, &tctx, 0, SIZE_MAX) == NULL)) { \
1572 error: \
1573 TALLOC_FREE(sbuff.buff); \
1574 *out = NULL; \
1575 return slen; \
1576 } \
1577 slen = _func(&sbuff, ##__VA_ARGS__); \
1578 if (slen < 0) goto error; \
1579 if (unlikely(fr_sbuff_trim_talloc(&sbuff, SIZE_MAX) < 0)) { \
1580 slen = -1; \
1581 goto error; \
1582 } \
1583 *out = sbuff.buff; \
1584 return slen; \
1585}
1586
1587static inline fr_slen_t fr_sbuff_out_abstrncpy(TALLOC_CTX *ctx, char **out, fr_sbuff_t *in, size_t len)
1588SBUFF_OUT_TALLOC_FUNC_DEF(fr_sbuff_out_bstrncpy, in, len)
1589
1590static inline fr_slen_t fr_sbuff_out_abstrncpy_exact(TALLOC_CTX *ctx, char **out, fr_sbuff_t *in, size_t len)
1591SBUFF_OUT_TALLOC_FUNC_DEF(fr_sbuff_out_bstrncpy_exact, in, len)
1592
1593static inline fr_slen_t fr_sbuff_out_abstrncpy_allowed(TALLOC_CTX *ctx, char **out, fr_sbuff_t *in, size_t len,
1594 bool const allowed[static UINT8_MAX + 1])
1595SBUFF_OUT_TALLOC_FUNC_DEF(fr_sbuff_out_bstrncpy_allowed, in, len, allowed)
1596
1597static inline fr_slen_t fr_sbuff_out_abstrncpy_until(TALLOC_CTX *ctx, char **out, fr_sbuff_t *in, size_t len,
1598 fr_sbuff_term_t const *tt,
1599 fr_sbuff_unescape_rules_t const *u_rules)
1600SBUFF_OUT_TALLOC_FUNC_DEF(fr_sbuff_out_bstrncpy_until, in, len, tt, u_rules)
1601
1602static inline fr_slen_t fr_sbuff_out_aunescape_until(TALLOC_CTX *ctx, char **out, fr_sbuff_t *in, size_t len,
1603 fr_sbuff_term_t const *tt,
1604 fr_sbuff_unescape_rules_t const *u_rules)
1605SBUFF_OUT_TALLOC_FUNC_DEF(fr_sbuff_out_unescape_until, in, len, tt, u_rules)
1606/** @} */
1607
1608/** @name Look for a token in a particular format, parse it, and write it to the output pointer
1609 *
1610 * These functions should not be called directly. #fr_sbuff_out should be used instead
1611 * so that if the output variable type changes, the parse rules are automatically changed.
1612 * @{
1613 */
1614fr_slen_t fr_sbuff_out_bool(bool *out, fr_sbuff_t *in);
1615
1616fr_slen_t fr_sbuff_out_int8(fr_sbuff_parse_error_t *err, int8_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1617fr_slen_t fr_sbuff_out_int16(fr_sbuff_parse_error_t *err, int16_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1618fr_slen_t fr_sbuff_out_int32(fr_sbuff_parse_error_t *err, int32_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1619fr_slen_t fr_sbuff_out_int64(fr_sbuff_parse_error_t *err, int64_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1620fr_slen_t fr_sbuff_out_ssize(fr_sbuff_parse_error_t *err, ssize_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1621fr_slen_t fr_sbuff_out_uint8(fr_sbuff_parse_error_t *err, uint8_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1622fr_slen_t fr_sbuff_out_uint16(fr_sbuff_parse_error_t *err, uint16_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1623fr_slen_t fr_sbuff_out_uint32(fr_sbuff_parse_error_t *err, uint32_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1624fr_slen_t fr_sbuff_out_uint64(fr_sbuff_parse_error_t *err, uint64_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1625fr_slen_t fr_sbuff_out_size(fr_sbuff_parse_error_t *err, size_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1626
1627fr_slen_t fr_sbuff_out_uint8_dec(fr_sbuff_parse_error_t *err, uint8_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1628fr_slen_t fr_sbuff_out_uint16_dec(fr_sbuff_parse_error_t *err, uint16_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1629fr_slen_t fr_sbuff_out_uint32_dec(fr_sbuff_parse_error_t *err, uint32_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1630fr_slen_t fr_sbuff_out_uint64_dec(fr_sbuff_parse_error_t *err, uint64_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1631fr_slen_t fr_sbuff_out_size_dec(fr_sbuff_parse_error_t *err, size_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1632
1633fr_slen_t fr_sbuff_out_uint8_oct(fr_sbuff_parse_error_t *err, uint8_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1634fr_slen_t fr_sbuff_out_uint16_oct(fr_sbuff_parse_error_t *err, uint16_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1635fr_slen_t fr_sbuff_out_uint32_oct(fr_sbuff_parse_error_t *err, uint32_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1636fr_slen_t fr_sbuff_out_uint64_oct(fr_sbuff_parse_error_t *err, uint64_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1637fr_slen_t fr_sbuff_out_size_oct(fr_sbuff_parse_error_t *err, size_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1638
1639fr_slen_t fr_sbuff_out_uint8_hex(fr_sbuff_parse_error_t *err, uint8_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1640fr_slen_t fr_sbuff_out_uint16_hex(fr_sbuff_parse_error_t *err, uint16_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1641fr_slen_t fr_sbuff_out_uint32_hex(fr_sbuff_parse_error_t *err, uint32_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1642fr_slen_t fr_sbuff_out_uint64_hex(fr_sbuff_parse_error_t *err, uint64_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1643fr_slen_t fr_sbuff_out_size_hex(fr_sbuff_parse_error_t *err, size_t *out, fr_sbuff_t *sbuff, bool no_trailing);
1644
1645fr_slen_t fr_sbuff_out_float32(fr_sbuff_parse_error_t *err, float *out, fr_sbuff_t *sbuff, bool no_trailing);
1646fr_slen_t fr_sbuff_out_float64(fr_sbuff_parse_error_t *err, double *out, fr_sbuff_t *sbuff, bool no_trailing);
1647
1648#ifndef SIZE_SAME_AS_UINT64
1649# define _fr_sbuff_out_size(_err, _out, _in) size_t * : fr_sbuff_out_size(_err, (size_t *)_out, _in, true),
1650#else
1651# define _fr_sbuff_out_size(_err, _out, _in)
1652#endif
1653
1654#ifndef SSIZE_SAME_AS_INT64
1655# define _fr_sbuff_out_ssize(_err, _out, _in) ssize_t * : fr_sbuff_out_ssize(_err, (ssize_t *)_out, _in, true),
1656#else
1657# define _fr_sbuff_out_ssize(_err, _out, _in)
1658#endif
1659
1660/** Parse a value based on the output type
1661 *
1662 * @param[out] _err If not NULL a value describing the parse error
1663 * will be written to err.
1664 * @param[out] _out Pointer to an integer type.
1665 * @param[in] _in Sbuff to parse integer from.
1666 * @return The number of bytes parsed (even on error).
1667 */
1668#define fr_sbuff_out(_err, _out, _in) \
1669 _Generic((_out), \
1670 bool * : fr_sbuff_out_bool((bool *)_out, _in), \
1671 int8_t * : fr_sbuff_out_int8(_err, (int8_t *)_out, _in, true), \
1672 int16_t * : fr_sbuff_out_int16(_err, (int16_t *)_out, _in, true), \
1673 int32_t * : fr_sbuff_out_int32(_err, (int32_t *)_out, _in, true), \
1674 int64_t * : fr_sbuff_out_int64(_err, (int64_t *)_out, _in, true), \
1675 _fr_sbuff_out_ssize(_err, _out, _in) \
1676 uint8_t * : fr_sbuff_out_uint8(_err, (uint8_t *)_out, _in, true), \
1677 uint16_t * : fr_sbuff_out_uint16(_err, (uint16_t *)_out, _in, true), \
1678 uint32_t * : fr_sbuff_out_uint32(_err, (uint32_t *)_out, _in, true), \
1679 uint64_t * : fr_sbuff_out_uint64(_err, (uint64_t *)_out, _in, true), \
1680 _fr_sbuff_out_size(_err, _out, _in) \
1681 float * : fr_sbuff_out_float32(_err, (float *)_out, _in, true), \
1682 double * : fr_sbuff_out_float64(_err, (double *)_out, _in, true) \
1683 )
1684/** @} */
1685
1686
1687/** @name Conditional advancement
1688 *
1689 * These functions are typically used for parsing when trying to locate
1690 * a sequence of characters in the sbuff.
1691 *
1692 * @{
1693 */
1694size_t fr_sbuff_adv_past_str(fr_sbuff_t *sbuff, char const *needle, size_t need_len);
1695
1696#define fr_sbuff_adv_past_str_literal(_sbuff, _needle) fr_sbuff_adv_past_str(_sbuff, _needle, sizeof(_needle) - 1)
1697
1698size_t fr_sbuff_adv_past_strcase(fr_sbuff_t *sbuff, char const *needle, size_t need_len);
1699
1700#define fr_sbuff_adv_past_strcase_literal(_sbuff, _needle) fr_sbuff_adv_past_strcase(_sbuff, _needle, sizeof(_needle) - 1)
1701
1702size_t fr_sbuff_adv_past_allowed(fr_sbuff_t *sbuff, size_t len,
1703 bool const allowed[static UINT8_MAX + 1], fr_sbuff_term_t const *tt);
1704
1705#define fr_sbuff_adv_past_zeros(_sbuff, _len, _tt) fr_sbuff_adv_past_allowed(_sbuff, _len, sbuff_char_class_zero, _tt)
1706
1707#define fr_sbuff_adv_past_whitespace(_sbuff, _len, _tt) fr_sbuff_adv_past_allowed(_sbuff, _len, sbuff_char_whitespace, _tt)
1708
1709#define fr_sbuff_adv_past_blank(_sbuff, _len, _tt) fr_sbuff_adv_past_allowed(_sbuff, _len, sbuff_char_blank, _tt)
1710
1711size_t fr_sbuff_adv_until(fr_sbuff_t *sbuff, size_t len, fr_sbuff_term_t const *tt, char escape_chr);
1712
1713char *fr_sbuff_adv_to_chr_utf8(fr_sbuff_t *in, size_t len, char const *chr);
1714
1715char *fr_sbuff_adv_to_chr(fr_sbuff_t *in, size_t len, char c);
1716
1717char *fr_sbuff_adv_to_str(fr_sbuff_t *sbuff, size_t len, char const *needle, size_t needle_len);
1718
1719#define fr_sbuff_adv_to_str_literal(_sbuff, _len, _needle) fr_sbuff_adv_to_str(_sbuff, _len, _needle, sizeof(_needle) - 1)
1720
1721char *fr_sbuff_adv_to_strcase(fr_sbuff_t *sbuff, size_t len, char const *needle, size_t needle_len);
1722
1723#define fr_sbuff_adv_to_strcase_literal(_sbuff, _len, _needle) fr_sbuff_adv_to_strcase(_sbuff, _len, _needle, sizeof(_needle) - 1)
1724
1725bool fr_sbuff_next_if_char(fr_sbuff_t *sbuff, char c);
1726
1727bool fr_sbuff_next_unless_char(fr_sbuff_t *sbuff, char c);
1728
1729/** Advance the sbuff by one char
1730 *
1731 */
1732static inline char fr_sbuff_next(fr_sbuff_t *sbuff)
1733{
1734 if (!fr_sbuff_extend(sbuff)) return '\0';
1735 return fr_sbuff_advance(sbuff, 1);
1736}
1737/** @} */
1738
1739/** @name Remove chars from a buffer and re-terminate
1740 *
1741 * @{
1742 */
1743size_t fr_sbuff_trim(fr_sbuff_t *sbuff, bool const to_trim[static UINT8_MAX + 1]);
1744/** @} */
1745
1746/** @name Conditions
1747 *
1748 * These functions are typically used in recursive decent parsing for
1749 * look ahead.
1750 * @{
1751 */
1752bool fr_sbuff_is_terminal(fr_sbuff_t *in, fr_sbuff_term_t const *tt);
1753
1754static inline bool fr_sbuff_is_in_charset(fr_sbuff_t *sbuff, bool const chars[static UINT8_MAX + 1])
1755{
1756 if (!fr_sbuff_extend(sbuff)) return false;
1757 return chars[(uint8_t)*sbuff->p];
1758}
1759
1760static inline bool fr_sbuff_is_str(fr_sbuff_t *sbuff, char const *str, size_t len)
1761{
1762 if (len == SIZE_MAX) len = strlen(str);
1763 if (!fr_sbuff_extend_lowat(NULL, sbuff, len)) return false;
1764 return memcmp(sbuff->p, str, len) == 0;
1765}
1766#define fr_sbuff_is_str_literal(_sbuff, _str) fr_sbuff_is_str(_sbuff, _str, sizeof(_str) - 1)
1767
1768#define fr_sbuff_eof(_x) (!fr_sbuff_extend(_x))
1769
1770static inline bool _fr_sbuff_is_char(fr_sbuff_t *sbuff, char *p, char c)
1771{
1772 if (!fr_sbuff_extend(sbuff)) return false;
1773 return *p == c;
1774}
1775static inline bool _fr_marker_is_char(fr_sbuff_marker_t *marker, char *p, char c)
1776{
1777 if (!fr_sbuff_extend(marker)) return false;
1778 return *p == c;
1779}
1780#define fr_sbuff_is_char(_sbuff_or_marker, _c) \
1781 _Generic((_sbuff_or_marker), \
1782 fr_sbuff_t * : _fr_sbuff_is_char((fr_sbuff_t *)(_sbuff_or_marker), fr_sbuff_current(_sbuff_or_marker), _c), \
1783 fr_sbuff_marker_t * : _fr_marker_is_char((fr_sbuff_marker_t *)(_sbuff_or_marker), fr_sbuff_current(_sbuff_or_marker), _c) \
1784 )
1785
1786#define SBUFF_IS_FUNC(_name, _test) \
1787 static inline bool _fr_sbuff_is_ ## _name(fr_sbuff_t *sbuff, char *p) \
1788 { \
1789 if (!fr_sbuff_extend(sbuff)) return false; \
1790 return _test; \
1791 }\
1792 static inline bool _fr_marker_is_ ## _name(fr_sbuff_marker_t *marker, char *p) \
1793 { \
1794 if (!fr_sbuff_extend(marker)) return false; \
1795 return _test; \
1796 }
1797
1798#define SBUFF_IS_GENERIC(_sbuff_or_marker, _name) \
1799 _Generic((_sbuff_or_marker), \
1800 fr_sbuff_t * : _fr_sbuff_is_ ## _name((fr_sbuff_t *)(_sbuff_or_marker), fr_sbuff_current(_sbuff_or_marker)), \
1801 fr_sbuff_marker_t * : _fr_marker_is_ ## _name((fr_sbuff_marker_t *)(_sbuff_or_marker), fr_sbuff_current(_sbuff_or_marker)) \
1802 )
1803
1804SBUFF_IS_FUNC(digit, isdigit((uint8_t) *p))
1805#define fr_sbuff_is_digit(_sbuff_or_marker) \
1806 SBUFF_IS_GENERIC(_sbuff_or_marker, digit)
1807
1808SBUFF_IS_FUNC(upper, isupper((uint8_t) *p))
1809#define fr_sbuff_is_upper(_sbuff_or_marker) \
1810 SBUFF_IS_GENERIC(_sbuff_or_marker, upper)
1811
1812SBUFF_IS_FUNC(lower, islower((uint8_t) *p))
1813#define fr_sbuff_is_lower(_sbuff_or_marker) \
1814 SBUFF_IS_GENERIC(_sbuff_or_marker, lower)
1815
1816SBUFF_IS_FUNC(alpha, isalpha((uint8_t) *p))
1817#define fr_sbuff_is_alpha(_sbuff_or_marker) \
1818 SBUFF_IS_GENERIC(_sbuff_or_marker, alpha)
1819
1820SBUFF_IS_FUNC(alnum, isalnum((uint8_t) *p))
1821#define fr_sbuff_is_alnum(_sbuff_or_marker) \
1822 SBUFF_IS_GENERIC(_sbuff_or_marker, alnum)
1823
1824SBUFF_IS_FUNC(space, isspace((uint8_t) *p))
1825#define fr_sbuff_is_space(_sbuff_or_marker) \
1826 SBUFF_IS_GENERIC(_sbuff_or_marker, space)
1827
1828SBUFF_IS_FUNC(hex, isxdigit((uint8_t) *p))
1829#define fr_sbuff_is_hex(_sbuff_or_marker) \
1830 SBUFF_IS_GENERIC(_sbuff_or_marker, hex)
1831
1832/** @} */
1833
1834void fr_sbuff_unescape_debug(FILE *fp, fr_sbuff_unescape_rules_t const *escapes);
1835
1836void fr_sbuff_terminal_debug(FILE *fp, fr_sbuff_term_t const *tt);
1837
1838void fr_sbuff_parse_rules_debug(FILE *fp, fr_sbuff_parse_rules_t const *p_rules);
1839
1840/*
1841 * ...printf("foo %.*s", fr_sbuff_as_percent_s(&sbuff));
1842 */
1843#define fr_sbuff_as_percent_s(_sbuff) (int) fr_sbuff_remaining(_sbuff), fr_sbuff_current(_sbuff)
1844
1845#ifdef __cplusplus
1846}
1847#endif
#define RCSIDH(h, id)
Definition build.h:486
#define DIAG_OFF(_x)
Definition build.h:459
static fr_slen_t err
Definition dict.h:846
static const bool escapes[UINT8_MAX+1]
Definition util.c:40
long int ssize_t
unsigned char uint8_t
#define UINT8_MAX
fr_sbuff_parse_error_t
@ FR_SBUFF_PARSE_ERROR_NUM_OVERFLOW
Integer type would overflow.
@ FR_SBUFF_PARSE_ERROR_NUM_UNDERFLOW
Integer type would underflow.
@ FR_SBUFF_PARSE_ERROR_NOT_FOUND
String does not contain a token matching the output type.
@ FR_SBUFF_PARSE_ERROR_FORMAT
Format of data was invalid.
@ FR_SBUFF_PARSE_OK
No error.
@ FR_SBUFF_PARSE_ERROR_OUT_OF_SPACE
No space available in output buffer.
@ FR_SBUFF_PARSE_ERROR_TRAILING
Trailing characters found.
static char const * name
bool const sbuff_char_class_float[UINT8_MAX+1]
Definition sbuff.c:70
bool const sbuff_char_line_endings[UINT8_MAX+1]
Definition sbuff.c:104
bool const sbuff_char_class_hex[UINT8_MAX+1]
Definition sbuff.c:94
bool const sbuff_char_class_uint[UINT8_MAX+1]
Definition sbuff.c:60
size_t sbuff_parse_error_table_len
Definition sbuff.c:52
bool const sbuff_char_class_zero[UINT8_MAX+1]
Definition sbuff.c:75
bool const sbuff_char_class_hostname[UINT8_MAX+1]
Definition sbuff.c:82
bool const sbuff_char_whitespace[UINT8_MAX+1]
Definition sbuff.c:100
bool const sbuff_char_blank[UINT8_MAX+1]
Definition sbuff.c:108
bool const sbuff_char_class_int[UINT8_MAX+1]
Definition sbuff.c:65
fr_table_num_ordered_t const sbuff_parse_error_table[]
Definition sbuff.c:43
bool const sbuff_char_word[UINT8_MAX+1]
Definition sbuff.c:96
bool const sbuff_char_alpha_num[UINT8_MAX+1]
Definition sbuff.c:95
TALLOC_CTX * ctx
Context to alloc new buffers in.
Definition sbuff.h:140
bool(* fr_sbuff_eof_t)(fr_sbuff_t *sbuff)
For a given extension function, returns whether it is at EOF.
Definition sbuff.h:81
size_t shifted
How much we've read from this file.
Definition sbuff.h:154
size_t shifted
How many bytes this sbuff has been shifted since its creation.
Definition sbuff.h:118
fr_sbuff_extend_t extend
Function to re-populate or extend the buffer.
Definition sbuff.h:121
char const * str
Terminal string.
Definition sbuff.h:162
void * uctx
Extend uctx data.
Definition sbuff.h:126
size_t init
How much to allocate initially.
Definition sbuff.h:141
char chr
Character at the start of an escape sequence.
Definition sbuff.h:204
char const * name
Name for rule set to aid we debugging.
Definition sbuff.h:202
bool do_oct
Process oct sequences i.e.
Definition sbuff.h:216
fr_sbuff_marker_t * m
Pointers to update if the underlying buffer changes.
Definition sbuff.h:130
size_t len
Length of the list.
Definition sbuff.h:172
char const * err
Where the last error occurred.
Definition sbuff.h:114
bool do_hex
Process hex sequences i.e.
Definition sbuff.h:215
size_t max
Maximum size of the buffer.
Definition sbuff.h:142
size_t len
Length of string.
Definition sbuff.h:163
char * buff_end
The true end of the buffer.
Definition sbuff.h:152
fr_sbuff_t * parent
sbuff this sbuff was copied from.
Definition sbuff.h:128
uint8_t is_const
Can't be modified.
Definition sbuff.h:116
fr_sbuff_marker_t * next
Next m in the list.
Definition sbuff.h:89
bool eof
are we at EOF?
Definition sbuff.h:155
uint8_t adv_parent
If true, advance the parent.
Definition sbuff.h:117
fr_sbuff_term_elem_t * elem
A sorted list of terminal strings.
Definition sbuff.h:173
ssize_t fr_slen_t
Represents number of bytes parsed or location of parse error.
Definition sbuff.h:51
fr_sbuff_t * parent
Owner of the marker.
Definition sbuff.h:90
fr_sbuff_eof_t eof
Function to determine if the buffer is at EOF.
Definition sbuff.h:124
size_t(* fr_sbuff_extend_t)(fr_sbuff_extend_status_t *status, fr_sbuff_t *sbuff, size_t req_extension)
Extension callback.
Definition sbuff.h:76
FILE * file
FILE * we're reading from.
Definition sbuff.h:151
size_t max
Maximum number of bytes to read.
Definition sbuff.h:153
fr_sbuff_extend_status_t
Whether the buffer is currently extendable and whether it was extended.
Definition sbuff.h:64
@ FR_SBUFF_FLAG_EXTEND_ERROR
The last call to an extend function resulted in an error.
Definition sbuff.h:66
@ FR_SBUFF_FLAG_EXTENDED
The last call to extend function actually extended the buffer.
Definition sbuff.h:65
Terminal element with pre-calculated lengths.
Definition sbuff.h:161
Set of terminal elements.
File sbuff extension structure.
Definition sbuff.h:150
Talloc sbuff extension structure.
Definition sbuff.h:139
Set of parsing rules for *unescape_until functions.
static char buff[sizeof("18446744073709551615")+3]
Definition size_tests.c:41
#define fr_table_str_by_value(_table, _number, _def)
Convert an integer to a string.
Definition table.h:772
An element in an arbitrarily ordered array of name to num mappings.
Definition table.h:57
#define fr_strerror_const(_msg)
Definition strerror.h:223