]> git.proxmox.com Git - mirror_frr.git/blame - lib/stream.h
bgpd: support for DF election in EVPN-MH
[mirror_frr.git] / lib / stream.h
CommitLineData
718e3744 1/*
2 * Packet interface
3 * Copyright (C) 1999 Kunihiro Ishiguro
4 *
5 * This file is part of GNU Zebra.
6 *
7 * GNU Zebra is free software; you can redistribute it and/or modify it
8 * under the terms of the GNU General Public License as published by the
9 * Free Software Foundation; either version 2, or (at your option) any
10 * later version.
11 *
12 * GNU Zebra is distributed in the hope that it will be useful, but
13 * WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * General Public License for more details.
16 *
896014f4
DL
17 * You should have received a copy of the GNU General Public License along
18 * with this program; see the file COPYING; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
718e3744 20 */
21
22#ifndef _ZEBRA_STREAM_H
23#define _ZEBRA_STREAM_H
24
363e24c6
QY
25#include <pthread.h>
26
27#include "frratomic.h"
9bedbb1e 28#include "mpls.h"
02ff83c5 29#include "prefix.h"
30
5e244469
RW
31#ifdef __cplusplus
32extern "C" {
33#endif
34
050c013a 35/*
36 * A stream is an arbitrary buffer, whose contents generally are assumed to
37 * be in network order.
38 *
39 * A stream has the following attributes associated with it:
40 *
41 * - size: the allocated, invariant size of the buffer.
42 *
43 * - getp: the get position marker, denoting the offset in the stream where
44 * the next read (or 'get') will be from. This getp marker is
45 * automatically adjusted when data is read from the stream, the
46 * user may also manipulate this offset as they wish, within limits
47 * (see below)
48 *
49 * - endp: the end position marker, denoting the offset in the stream where
50 * valid data ends, and if the user attempted to write (or
51 * 'put') data where that data would be written (or 'put') to.
52 *
53 * These attributes are all size_t values.
54 *
55 * Constraints:
56 *
57 * 1. getp can never exceed endp
58 *
59 * - hence if getp is equal to endp, there is no more valid data that can be
60 * gotten from the stream (though, the user may reposition getp to earlier in
61 * the stream, if they wish).
62 *
63 * 2. endp can never exceed size
64 *
65 * - hence, if endp is equal to size, then the stream is full, and no more
66 * data can be written to the stream.
67 *
68 * In other words the following must always be true, and the stream
69 * abstraction is allowed internally to assert that the following property
70 * holds true for a stream, as and when it wishes:
71 *
72 * getp <= endp <= size
73 *
74 * It is the users responsibility to ensure this property is never violated.
75 *
76 * A stream therefore can be thought of like this:
77 *
78 * ---------------------------------------------------
79 * |XXXXXXXXXXXXXXXXXXXXXXXX |
80 * ---------------------------------------------------
81 * ^ ^ ^
82 * getp endp size
83 *
84 * This shows a stream containing data (shown as 'X') up to the endp offset.
85 * The stream is empty from endp to size. Without adjusting getp, there are
86 * still endp-getp bytes of valid data to be read from the stream.
87 *
d62a17ae 88 * Methods are provided to get and put to/from the stream, as well as
050c013a 89 * retrieve the values of the 3 markers and manipulate the getp marker.
90 *
91 * Note:
92 * At the moment, newly allocated streams are zero filled. Hence, one can
93 * use stream_forward_endp() to effectively create arbitrary zero-fill
94 * padding. However, note that stream_reset() does *not* zero-out the
95 * stream. This property should **not** be relied upon.
0dab9303 96 *
97 * Best practice is to use stream_put (<stream *>, NULL, <size>) to zero out
98 * any part of a stream which isn't otherwise written to.
050c013a 99 */
100
718e3744 101/* Stream buffer. */
d62a17ae 102struct stream {
103 struct stream *next;
104
de75223e
DS
105 /*
106 * Remainder is ***private*** to stream
d62a17ae 107 * direct access is frowned upon!
108 * Use the appropriate functions/macros
109 */
de75223e
DS
110 size_t getp; /* next get position */
111 size_t endp; /* last valid data position */
112 size_t size; /* size of data segment */
83e7e1d7 113 unsigned char data[]; /* data pointer */
718e3744 114};
115
116/* First in first out queue structure. */
d62a17ae 117struct stream_fifo {
363e24c6
QY
118 /* lock for mt-safe operations */
119 pthread_mutex_t mtx;
120
121 /* number of streams in this fifo */
c8a65463 122 atomic_size_t count;
03ed85a6 123#if defined DEV_BUILD
c8a65463 124 atomic_size_t max_count;
03ed85a6 125#endif
718e3744 126
d62a17ae 127 struct stream *head;
128 struct stream *tail;
718e3744 129};
130
131/* Utility macros. */
718e3744 132#define STREAM_SIZE(S) ((S)->size)
d62a17ae 133/* number of bytes which can still be written */
050c013a 134#define STREAM_WRITEABLE(S) ((S)->size - (S)->endp)
d62a17ae 135/* number of bytes still to be read */
050c013a 136#define STREAM_READABLE(S) ((S)->endp - (S)->getp)
137
d62a17ae 138#define STREAM_CONCAT_REMAIN(S1, S2, size) ((size) - (S1)->endp - (S2)->endp)
8c71e481 139
2d34fb80
QY
140/* this macro is deprecated, but not slated for removal anytime soon */
141#define STREAM_DATA(S) ((S)->data)
142
d62a17ae 143/* Stream prototypes.
4b201d46 144 * For stream_{put,get}S, the S suffix mean:
145 *
146 * c: character (unsigned byte)
147 * w: word (two bytes)
148 * l: long (two words)
149 * q: quad (four words)
150 */
d62a17ae 151extern struct stream *stream_new(size_t);
152extern void stream_free(struct stream *);
f8c511cd
MS
153/* Copy 'src' into 'dest', returns 'dest' */
154extern struct stream *stream_copy(struct stream *dest,
155 const struct stream *src);
156extern struct stream *stream_dup(const struct stream *s);
43888669 157
43888669
DS
158extern size_t stream_resize_inplace(struct stream **sptr, size_t newsize);
159
f8c511cd
MS
160extern size_t stream_get_getp(const struct stream *s);
161extern size_t stream_get_endp(const struct stream *s);
162extern size_t stream_get_size(const struct stream *s);
8cc4198f 163
8c71e481
PM
164/**
165 * Create a new stream structure; copy offset bytes from s1 to the new
166 * stream; copy s2 data to the new stream; copy rest of s1 data to the
167 * new stream.
168 */
f8c511cd
MS
169extern struct stream *stream_dupcat(const struct stream *s1,
170 const struct stream *s2, size_t offset);
8c71e481 171
d62a17ae 172extern void stream_set_getp(struct stream *, size_t);
173extern void stream_set_endp(struct stream *, size_t);
174extern void stream_forward_getp(struct stream *, size_t);
4adddf0a 175extern bool stream_forward_getp2(struct stream *, size_t);
06cf2c0c
QY
176extern void stream_rewind_getp(struct stream *s, size_t size);
177extern bool stream_rewind_getp2(struct stream *s, size_t size);
d62a17ae 178extern void stream_forward_endp(struct stream *, size_t);
4adddf0a 179extern bool stream_forward_endp2(struct stream *, size_t);
8cc4198f 180
181/* steam_put: NULL source zeroes out size_t bytes of stream */
d62a17ae 182extern void stream_put(struct stream *, const void *, size_t);
d7c0a89a
QY
183extern int stream_putc(struct stream *, uint8_t);
184extern int stream_putc_at(struct stream *, size_t, uint8_t);
185extern int stream_putw(struct stream *, uint16_t);
186extern int stream_putw_at(struct stream *, size_t, uint16_t);
187extern int stream_put3(struct stream *, uint32_t);
188extern int stream_put3_at(struct stream *, size_t, uint32_t);
189extern int stream_putl(struct stream *, uint32_t);
190extern int stream_putl_at(struct stream *, size_t, uint32_t);
d62a17ae 191extern int stream_putq(struct stream *, uint64_t);
192extern int stream_putq_at(struct stream *, size_t, uint64_t);
d7c0a89a 193extern int stream_put_ipv4(struct stream *, uint32_t);
d3d77ec4 194extern int stream_put_in_addr(struct stream *s, const struct in_addr *addr);
31f937fb 195extern bool stream_put_ipaddr(struct stream *s, struct ipaddr *ip);
d3d77ec4
MS
196extern int stream_put_in_addr_at(struct stream *s, size_t putp,
197 const struct in_addr *addr);
198extern int stream_put_in6_addr_at(struct stream *s, size_t putp,
199 const struct in6_addr *addr);
200extern int stream_put_prefix_addpath(struct stream *s,
201 const struct prefix *p,
d62a17ae 202 int addpath_encode,
d7c0a89a 203 uint32_t addpath_tx_id);
d3d77ec4 204extern int stream_put_prefix(struct stream *s, const struct prefix *p);
5f040085 205extern int stream_put_labeled_prefix(struct stream *, const struct prefix *,
ec15e1b5
QY
206 mpls_label_t *, int addpath_encode,
207 uint32_t addpath_tx_id);
d62a17ae 208extern void stream_get(void *, struct stream *, size_t);
051cc28c 209extern bool stream_get2(void *data, struct stream *s, size_t size);
d62a17ae 210extern void stream_get_from(void *, struct stream *, size_t, size_t);
d7c0a89a
QY
211extern uint8_t stream_getc(struct stream *);
212extern bool stream_getc2(struct stream *s, uint8_t *byte);
213extern uint8_t stream_getc_from(struct stream *, size_t);
214extern uint16_t stream_getw(struct stream *);
051cc28c 215extern bool stream_getw2(struct stream *s, uint16_t *word);
d7c0a89a
QY
216extern uint16_t stream_getw_from(struct stream *, size_t);
217extern uint32_t stream_get3(struct stream *);
218extern uint32_t stream_get3_from(struct stream *, size_t);
219extern uint32_t stream_getl(struct stream *);
051cc28c 220extern bool stream_getl2(struct stream *s, uint32_t *l);
d7c0a89a 221extern uint32_t stream_getl_from(struct stream *, size_t);
d62a17ae 222extern uint64_t stream_getq(struct stream *);
223extern uint64_t stream_getq_from(struct stream *, size_t);
c2b5a4e5 224bool stream_getq2(struct stream *s, uint64_t *q);
d7c0a89a 225extern uint32_t stream_get_ipv4(struct stream *);
31f937fb 226extern bool stream_get_ipaddr(struct stream *s, struct ipaddr *ip);
718e3744 227
16f1b9ee 228/* IEEE-754 floats */
d62a17ae 229extern float stream_getf(struct stream *);
230extern double stream_getd(struct stream *);
231extern int stream_putf(struct stream *, float);
232extern int stream_putd(struct stream *, double);
16f1b9ee 233
718e3744 234#undef stream_read
235#undef stream_write
262feb1a 236
d62a17ae 237/* Deprecated: assumes blocking I/O. Will be removed.
262feb1a 238 Use stream_read_try instead. */
d62a17ae 239extern int stream_read(struct stream *, int, size_t);
262feb1a 240
262feb1a 241/* Read up to size bytes into the stream.
242 Return code:
243 >0: number of bytes read
244 0: end-of-file
245 -1: fatal error
246 -2: transient error, should retry later (i.e. EAGAIN or EINTR)
247 This is suitable for use with non-blocking file descriptors.
248 */
249extern ssize_t stream_read_try(struct stream *s, int fd, size_t size);
250
d62a17ae 251extern ssize_t stream_recvmsg(struct stream *s, int fd, struct msghdr *,
252 int flags, size_t size);
253extern ssize_t stream_recvfrom(struct stream *s, int fd, size_t len, int flags,
254 struct sockaddr *from, socklen_t *fromlen);
255extern size_t stream_write(struct stream *, const void *, size_t);
718e3744 256
8cc4198f 257/* reset the stream. See Note above */
d62a17ae 258extern void stream_reset(struct stream *);
259extern int stream_flush(struct stream *, int);
260extern int stream_empty(struct stream *); /* is the stream empty? */
050c013a 261
9d72660d 262/* debugging */
f1bc75da 263extern void stream_hexdump(const struct stream *s);
9d72660d 264
050c013a 265/* deprecated */
d7c0a89a 266extern uint8_t *stream_pnt(struct stream *);
718e3744 267
363e24c6
QY
268/*
269 * Operations on struct stream_fifo.
270 *
271 * Each function has a safe variant, which ensures that the operation performed
272 * is atomic with respect to the operations performed by all other safe
273 * variants. In other words, the safe variants lock the stream_fifo's mutex
274 * before performing their action. These are provided for convenience when
275 * using stream_fifo in a multithreaded context, to alleviate the need for the
276 * caller to implement their own synchronization around the stream_fifo.
277 *
278 * The following functions do not have safe variants. The caller must ensure
279 * that these operations are performed safely in a multithreaded context:
280 * - stream_fifo_new
281 * - stream_fifo_free
282 */
283
284/*
285 * Create a new stream_fifo.
286 *
287 * Returns:
288 * newly created stream_fifo
289 */
d62a17ae 290extern struct stream_fifo *stream_fifo_new(void);
363e24c6 291
f8c511cd
MS
292/*
293 * Init or re-init an on-stack fifo. This allows use of a fifo struct without
294 * requiring a malloc/free cycle.
295 * Note well that the fifo must be de-inited with the 'fifo_deinit' api.
296 */
297void stream_fifo_init(struct stream_fifo *fifo);
298
299/*
300 * Deinit an on-stack fifo.
301 */
302void stream_fifo_deinit(struct stream_fifo *fifo);
303
363e24c6
QY
304/*
305 * Push a stream onto a stream_fifo.
306 *
307 * fifo
308 * the stream_fifo to push onto
309 *
310 * s
311 * the stream to push onto the stream_fifo
312 */
d62a17ae 313extern void stream_fifo_push(struct stream_fifo *fifo, struct stream *s);
363e24c6
QY
314extern void stream_fifo_push_safe(struct stream_fifo *fifo, struct stream *s);
315
316/*
317 * Pop a stream off a stream_fifo.
318 *
319 * fifo
320 * the stream_fifo to pop from
321 *
322 * Returns:
323 * the next stream in the stream_fifo
324 */
d62a17ae 325extern struct stream *stream_fifo_pop(struct stream_fifo *fifo);
363e24c6
QY
326extern struct stream *stream_fifo_pop_safe(struct stream_fifo *fifo);
327
328/*
329 * Retrieve the next stream from a stream_fifo without popping it.
330 *
331 * fifo
332 * the stream_fifo to operate on
333 *
334 * Returns:
335 * the next stream that would be returned from stream_fifo_pop
336 */
d62a17ae 337extern struct stream *stream_fifo_head(struct stream_fifo *fifo);
363e24c6
QY
338extern struct stream *stream_fifo_head_safe(struct stream_fifo *fifo);
339
340/*
341 * Remove all streams from a stream_fifo.
342 *
343 * fifo
344 * the stream_fifo to clean
345 */
d62a17ae 346extern void stream_fifo_clean(struct stream_fifo *fifo);
363e24c6
QY
347extern void stream_fifo_clean_safe(struct stream_fifo *fifo);
348
349/*
350 * Retrieve number of streams on a stream_fifo.
351 *
352 * fifo
353 * the stream_fifo to retrieve the count for
354 *
355 * Returns:
356 * the number of streams on the stream_fifo
357 */
358extern size_t stream_fifo_count_safe(struct stream_fifo *fifo);
359
360/*
361 * Free a stream_fifo.
362 *
363 * Calls stream_fifo_clean, then deinitializes the stream_fifo and frees it.
364 *
365 * fifo
366 * the stream_fifo to free
367 */
d62a17ae 368extern void stream_fifo_free(struct stream_fifo *fifo);
718e3744 369
937652c6
DL
370/* This is here because "<< 24" is particularly problematic in C.
371 * This is because the left operand of << is integer-promoted, which means
372 * an uint8_t gets converted into a *signed* int. Shifting into the sign
373 * bit of a signed int is theoretically undefined behaviour, so - the left
374 * operand needs to be cast to unsigned.
375 *
376 * This is not a problem for 16- or 8-bit values (they don't reach the sign
377 * bit), for 64-bit values (you need to cast them anyway), and neither for
378 * encoding (because it's downcasted.)
379 */
1be1693e 380static inline const uint8_t *ptr_get_be32(const uint8_t *ptr, uint32_t *out)
937652c6
DL
381{
382 uint32_t tmp;
1be1693e 383
937652c6
DL
384 memcpy(&tmp, ptr, sizeof(tmp));
385 *out = ntohl(tmp);
386 return ptr + 4;
387}
388
74e2bd89
AK
389static inline uint8_t *ptr_get_be16(uint8_t *ptr, uint16_t *out)
390{
391 uint16_t tmp;
392
393 memcpy(&tmp, ptr, sizeof(tmp));
394 *out = ntohs(tmp);
395
396 return ptr + 2;
397}
398
051cc28c
DS
399/*
400 * so Normal stream_getX functions assert. Which is anathema
401 * to keeping a daemon up and running when something goes south
402 * Provide a stream_getX2 functions that do not assert.
403 * In addition provide these macro's that upon failure
404 * goto stream_failure. This is modeled upon some NL_XX
405 * macros in the linux kernel.
406 *
407 * This change allows for proper memory freeing
408 * after we've detected an error.
409 *
410 * In the future we will be removing the assert in
411 * the stream functions but we need a transition
412 * plan.
413 */
996c9314
LB
414#define STREAM_GETC(S, P) \
415 do { \
416 uint8_t _pval; \
417 if (!stream_getc2((S), &_pval)) \
418 goto stream_failure; \
419 (P) = _pval; \
051cc28c
DS
420 } while (0)
421
996c9314
LB
422#define STREAM_GETW(S, P) \
423 do { \
424 uint16_t _pval; \
425 if (!stream_getw2((S), &_pval)) \
426 goto stream_failure; \
427 (P) = _pval; \
051cc28c
DS
428 } while (0)
429
996c9314
LB
430#define STREAM_GETL(S, P) \
431 do { \
432 uint32_t _pval; \
433 if (!stream_getl2((S), &_pval)) \
434 goto stream_failure; \
435 (P) = _pval; \
051cc28c
DS
436 } while (0)
437
c2b5a4e5
QY
438#define STREAM_GETF(S, P) \
439 do { \
440 union { \
441 float r; \
442 uint32_t d; \
443 } _pval; \
741b1c4f 444 if (!stream_getl2((S), &_pval.d)) \
c2b5a4e5
QY
445 goto stream_failure; \
446 (P) = _pval.r; \
447 } while (0)
448
449#define STREAM_GETQ(S, P) \
450 do { \
451 uint64_t _pval; \
452 if (!stream_getq2((S), &_pval)) \
453 goto stream_failure; \
454 (P) = _pval; \
455 } while (0)
456
31f937fb
SM
457#define STREAM_GET_IPADDR(S, P) \
458 do { \
459 if (!stream_get_ipaddr((S), (P))) \
460 goto stream_failure; \
461 } while (0)
462
996c9314
LB
463#define STREAM_GET(P, STR, SIZE) \
464 do { \
465 if (!stream_get2((P), (STR), (SIZE))) \
466 goto stream_failure; \
051cc28c
DS
467 } while (0)
468
4adddf0a
QY
469#define STREAM_FORWARD_GETP(STR, SIZE) \
470 do { \
471 if (!stream_forward_getp2((STR), (SIZE))) \
472 goto stream_failure; \
473 } while (0)
474
06cf2c0c
QY
475#define STREAM_REWIND_GETP(STR, SIZE) \
476 do { \
477 if (!stream_rewind_getp2((STR), (SIZE))) \
478 goto stream_failure; \
479 } while (0)
480
4adddf0a
QY
481#define STREAM_FORWARD_ENDP(STR, SIZE) \
482 do { \
483 if (!stream_forward_endp2((STR), (SIZE))) \
484 goto stream_failure; \
485 } while (0)
486
5e244469
RW
487#ifdef __cplusplus
488}
489#endif
490
718e3744 491#endif /* _ZEBRA_STREAM_H */