blob: 117fcb01e4f663549f2713e7d7880ce2a030e983 [file] [log] [blame]
Sylvain Munaut12ba7782014-06-16 10:13:40 +02001#pragma once
Harald Welteec8b4502010-02-20 20:34:29 +01002
3/* (C) 2008 by Harald Welte <laforge@gnumonks.org>
4 * All Rights Reserved
5 *
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
Harald Welteec8b4502010-02-20 20:34:29 +010016 */
17
18#include <stdint.h>
Pablo Neira Ayuso83419342011-03-22 16:36:13 +010019#include <osmocom/core/linuxlist.h>
20#include <osmocom/core/utils.h>
Max53777012014-06-04 19:07:41 +020021#include <osmocom/core/bits.h>
Neels Hofmeyrf45334b2016-09-16 00:15:56 +020022#include <osmocom/core/defs.h>
Harald Welteec8b4502010-02-20 20:34:29 +010023
Harald Welteba6988b2011-08-17 12:46:48 +020024/*! \defgroup msgb Message buffers
25 * @{
Neels Hofmeyr17518fe2017-06-20 04:35:06 +020026 * \file msgb.h */
Harald Weltebd598e32011-08-16 23:26:52 +020027
Harald Welte652a7232010-07-22 21:55:24 +020028#define MSGB_DEBUG
29
Neels Hofmeyr87e45502017-06-20 00:17:59 +020030/*! Osmocom message buffer */
Harald Welteec8b4502010-02-20 20:34:29 +010031struct msgb {
Neels Hofmeyr87e45502017-06-20 00:17:59 +020032 struct llist_head list; /*!< linked list header */
Harald Welteec8b4502010-02-20 20:34:29 +010033
Pablo Neira Ayuso29cbf612011-07-07 19:46:44 +020034
Harald Welteec8b4502010-02-20 20:34:29 +010035 /* Part of which TRX logical channel we were received / transmitted */
Harald Welte074c9f92010-04-30 14:29:11 +020036 /* FIXME: move them into the control buffer */
Harald Weltec5a0ded2011-07-18 16:59:27 +020037 union {
Neels Hofmeyr87e45502017-06-20 00:17:59 +020038 void *dst; /*!< reference of origin/destination */
Harald Weltec5a0ded2011-07-18 16:59:27 +020039 struct gsm_bts_trx *trx;
40 };
Neels Hofmeyr87e45502017-06-20 00:17:59 +020041 struct gsm_lchan *lchan; /*!< logical channel */
Harald Welteec8b4502010-02-20 20:34:29 +010042
Neels Hofmeyr87e45502017-06-20 00:17:59 +020043 unsigned char *l1h; /*!< pointer to Layer1 header (if any) */
44 unsigned char *l2h; /*!< pointer to A-bis layer 2 header: OML, RSL(RLL), NS */
45 unsigned char *l3h; /*!< pointer to Layer 3 header. For OML: FOM; RSL: 04.08; GPRS: BSSGP */
46 unsigned char *l4h; /*!< pointer to layer 4 header */
Harald Welteec8b4502010-02-20 20:34:29 +010047
Neels Hofmeyr87e45502017-06-20 00:17:59 +020048 unsigned long cb[5]; /*!< control buffer */
Harald Welte074c9f92010-04-30 14:29:11 +020049
Neels Hofmeyr87e45502017-06-20 00:17:59 +020050 uint16_t data_len; /*!< length of underlying data array */
51 uint16_t len; /*!< length of bytes used in msgb */
Harald Welteec8b4502010-02-20 20:34:29 +010052
Neels Hofmeyr87e45502017-06-20 00:17:59 +020053 unsigned char *head; /*!< start of underlying memory buffer */
54 unsigned char *tail; /*!< end of message in buffer */
55 unsigned char *data; /*!< start of message in buffer */
56 unsigned char _data[0]; /*!< optional immediate data array */
Harald Welteec8b4502010-02-20 20:34:29 +010057};
58
Harald Welte179f3572019-03-18 18:38:47 +010059extern struct msgb *msgb_alloc_c(const void *ctx, uint16_t size, const char *name);
Harald Welteec8b4502010-02-20 20:34:29 +010060extern struct msgb *msgb_alloc(uint16_t size, const char *name);
61extern void msgb_free(struct msgb *m);
62extern void msgb_enqueue(struct llist_head *queue, struct msgb *msg);
63extern struct msgb *msgb_dequeue(struct llist_head *queue);
64extern void msgb_reset(struct msgb *m);
Harald Welte972b5022012-09-08 19:58:59 +020065uint16_t msgb_length(const struct msgb *msg);
Jacob Erlbeckbaa225e2014-02-28 15:14:40 +010066extern const char *msgb_hexdump(const struct msgb *msg);
Harald Welte4a62eda2019-03-18 18:27:00 +010067char *msgb_hexdump_buf(char *buf, size_t buf_len, const struct msgb *msg);
Harald Welte179f3572019-03-18 18:38:47 +010068char *msgb_hexdump_c(const void *ctx, const struct msgb *msg);
Jacob Erlbeckcdd05f02015-11-27 13:26:13 +010069extern int msgb_resize_area(struct msgb *msg, uint8_t *area,
70 int old_size, int new_size);
71extern struct msgb *msgb_copy(const struct msgb *msg, const char *name);
Harald Welte179f3572019-03-18 18:38:47 +010072extern struct msgb *msgb_copy_c(const void *ctx, const struct msgb *msg, const char *name);
Jacob Erlbeck8db11342015-11-27 13:26:15 +010073static int msgb_test_invariant(const struct msgb *msg) __attribute__((pure));
Harald Welteec8b4502010-02-20 20:34:29 +010074
Neels Hofmeyr3fad5d72017-11-20 14:15:03 +010075/*! Free all msgbs from a queue built with msgb_enqueue().
76 * \param[in] queue list head of a msgb queue.
77 */
78static inline void msgb_queue_free(struct llist_head *queue)
79{
80 struct msgb *msg;
81 while ((msg = msgb_dequeue(queue))) msgb_free(msg);
82}
83
Pau Espin Pedrol8ce6f482018-08-17 10:33:57 +020084/*! Enqueue message buffer to tail of a queue and increment queue size counter
85 * \param[in] queue linked list header of queue
86 * \param[in] msg message buffer to be added to the queue
87 * \param[in] count pointer to variable holding size of the queue
88 *
89 * The function will append the specified message buffer \a msg to the queue
90 * implemented by \ref llist_head \a queue using function \ref msgb_enqueue_count,
91 * then increment \a count
92 */
93static inline void msgb_enqueue_count(struct llist_head *queue, struct msgb *msg,
94 unsigned int *count)
95{
96 msgb_enqueue(queue, msg);
97 (*count)++;
98}
99
100/*! Dequeue message buffer from head of queue and decrement queue size counter
101 * \param[in] queue linked list header of queue
102 * \param[in] count pointer to variable holding size of the queue
103 * \returns message buffer (if any) or NULL if queue empty
104 *
105 * The function will remove the first message buffer from the queue
106 * implemented by \ref llist_head \a queue using function \ref msgb_enqueue_count,
107 * and decrement \a count, all if queue is not empty.
108 */
109static inline struct msgb *msgb_dequeue_count(struct llist_head *queue,
110 unsigned int *count)
111{
112 struct msgb *msg = msgb_dequeue(queue);
113 if (msg)
114 (*count)--;
115 return msg;
116}
117
Harald Welte652a7232010-07-22 21:55:24 +0200118#ifdef MSGB_DEBUG
Pablo Neira Ayuso83419342011-03-22 16:36:13 +0100119#include <osmocom/core/panic.h>
Harald Welte929d8872010-11-05 07:47:41 +0100120#define MSGB_ABORT(msg, fmt, args ...) do { \
Harald Weltedab02872010-11-09 13:42:26 +0100121 osmo_panic("msgb(%p): " fmt, msg, ## args); \
Harald Welte929d8872010-11-05 07:47:41 +0100122 } while(0)
123#else
124#define MSGB_ABORT(msg, fmt, args ...)
Harald Welte652a7232010-07-22 21:55:24 +0200125#endif
126
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200127/*! obtain L1 header of msgb */
Harald Weltefdd0a702010-03-01 22:30:51 +0100128#define msgb_l1(m) ((void *)(m->l1h))
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200129/*! obtain L2 header of msgb */
Harald Welteec8b4502010-02-20 20:34:29 +0100130#define msgb_l2(m) ((void *)(m->l2h))
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200131/*! obtain L3 header of msgb */
Harald Welteec8b4502010-02-20 20:34:29 +0100132#define msgb_l3(m) ((void *)(m->l3h))
Maxcb02a392018-12-03 14:00:49 +0100133/*! obtain L4 header of msgb */
134#define msgb_l4(m) ((void *)(m->l4h))
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200135/*! obtain SMS header of msgb */
Maxcb02a392018-12-03 14:00:49 +0100136#define msgb_sms(m) msgb_l4(m)
Harald Welteec8b4502010-02-20 20:34:29 +0100137
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200138/*! determine length of L1 message
Harald Weltebd598e32011-08-16 23:26:52 +0200139 * \param[in] msgb message buffer
140 * \returns size of L1 message in bytes
141 *
142 * This function computes the number of bytes between the tail of the
143 * message and the layer 1 header.
144 */
Harald Weltefdd0a702010-03-01 22:30:51 +0100145static inline unsigned int msgb_l1len(const struct msgb *msgb)
146{
147 return msgb->tail - (uint8_t *)msgb_l1(msgb);
148}
149
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200150/*! determine length of L2 message
Harald Weltebd598e32011-08-16 23:26:52 +0200151 * \param[in] msgb message buffer
152 * \returns size of L2 message in bytes
153 *
154 * This function computes the number of bytes between the tail of the
155 * message and the layer 2 header.
156 */
Harald Welteec8b4502010-02-20 20:34:29 +0100157static inline unsigned int msgb_l2len(const struct msgb *msgb)
158{
159 return msgb->tail - (uint8_t *)msgb_l2(msgb);
160}
161
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200162/*! determine length of L3 message
Harald Weltebd598e32011-08-16 23:26:52 +0200163 * \param[in] msgb message buffer
164 * \returns size of L3 message in bytes
165 *
166 * This function computes the number of bytes between the tail of the
167 * message and the layer 3 header.
168 */
Harald Welteec8b4502010-02-20 20:34:29 +0100169static inline unsigned int msgb_l3len(const struct msgb *msgb)
170{
171 return msgb->tail - (uint8_t *)msgb_l3(msgb);
172}
173
Vadim Yanitskiye521ede2018-11-07 03:53:03 +0700174/*! determine length of L4 message
175 * \param[in] msgb message buffer
176 * \returns size of L4 message in bytes
177 *
178 * This function computes the number of bytes between the tail of the
179 * message and the layer 4 header.
180 */
181static inline unsigned int msgb_l4len(const struct msgb *msgb)
182{
183 return msgb->tail - (uint8_t *)msgb_sms(msgb);
184}
185
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200186/*! determine the length of the header
Harald Weltebd598e32011-08-16 23:26:52 +0200187 * \param[in] msgb message buffer
188 * \returns number of bytes between start of buffer and start of msg
189 *
190 * This function computes the length difference between the underlying
191 * data buffer and the used section of the \a msgb.
192 */
Harald Welteec8b4502010-02-20 20:34:29 +0100193static inline unsigned int msgb_headlen(const struct msgb *msgb)
194{
195 return msgb->len - msgb->data_len;
196}
Harald Welte652a7232010-07-22 21:55:24 +0200197
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200198/*! determine how much tail room is left in msgb
Harald Weltebd598e32011-08-16 23:26:52 +0200199 * \param[in] msgb message buffer
200 * \returns number of bytes remaining at end of msgb
201 *
202 * This function computes the amount of octets left in the underlying
203 * data buffer after the end of the message.
204 */
Harald Welte652a7232010-07-22 21:55:24 +0200205static inline int msgb_tailroom(const struct msgb *msgb)
206{
207 return (msgb->head + msgb->data_len) - msgb->tail;
208}
209
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200210/*! determine the amount of headroom in msgb
Harald Weltebd598e32011-08-16 23:26:52 +0200211 * \param[in] msgb message buffer
212 * \returns number of bytes left ahead of message start in msgb
213 *
214 * This function computes the amount of bytes left in the underlying
215 * data buffer before the start of the actual message.
216 */
Harald Welte652a7232010-07-22 21:55:24 +0200217static inline int msgb_headroom(const struct msgb *msgb)
218{
219 return (msgb->data - msgb->head);
220}
221
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200222/*! append data to end of message buffer
Harald Weltebd598e32011-08-16 23:26:52 +0200223 * \param[in] msgb message buffer
224 * \param[in] len number of bytes to append to message
225 * \returns pointer to start of newly-appended data
226 *
227 * This function will move the \a tail pointer of the message buffer \a
228 * len bytes further, thus enlarging the message by \a len bytes.
229 *
230 * The return value is a pointer to start of the newly added section at
231 * the end of the message and can be used for actually filling/copying
232 * data into it.
233 */
Harald Welteec8b4502010-02-20 20:34:29 +0100234static inline unsigned char *msgb_put(struct msgb *msgb, unsigned int len)
235{
236 unsigned char *tmp = msgb->tail;
Vadim Yanitskiy85c78312022-02-02 01:34:30 +0600237 if (OSMO_UNLIKELY(msgb_tailroom(msgb) < (int) len))
Neels Hofmeyr63cb9492019-11-11 05:51:43 +0100238 MSGB_ABORT(msgb, "Not enough tailroom msgb_put"
239 " (allocated %u, head at %u, len %u, tailroom %u < want tailroom %u)\n",
240 msgb->data_len - sizeof(struct msgb),
241 msgb->head - msgb->_data,
242 msgb->len,
Harald Welte929d8872010-11-05 07:47:41 +0100243 msgb_tailroom(msgb), len);
Harald Welteec8b4502010-02-20 20:34:29 +0100244 msgb->tail += len;
245 msgb->len += len;
246 return tmp;
247}
Harald Weltebd598e32011-08-16 23:26:52 +0200248
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200249/*! append a uint8 value to the end of the message
Harald Weltebd598e32011-08-16 23:26:52 +0200250 * \param[in] msgb message buffer
251 * \param[in] word unsigned 8bit byte to be appended
252 */
Ingo Albrecht48e17f82010-03-07 18:03:41 +0100253static inline void msgb_put_u8(struct msgb *msgb, uint8_t word)
254{
255 uint8_t *space = msgb_put(msgb, 1);
256 space[0] = word & 0xFF;
257}
Harald Weltebd598e32011-08-16 23:26:52 +0200258
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200259/*! append a uint16 value to the end of the message
Harald Weltebd598e32011-08-16 23:26:52 +0200260 * \param[in] msgb message buffer
261 * \param[in] word unsigned 16bit byte to be appended
262 */
Ingo Albrecht48e17f82010-03-07 18:03:41 +0100263static inline void msgb_put_u16(struct msgb *msgb, uint16_t word)
264{
265 uint8_t *space = msgb_put(msgb, 2);
Max53777012014-06-04 19:07:41 +0200266 osmo_store16be(word, space);
Ingo Albrecht48e17f82010-03-07 18:03:41 +0100267}
Harald Weltebd598e32011-08-16 23:26:52 +0200268
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200269/*! append a uint32 value to the end of the message
Harald Weltebd598e32011-08-16 23:26:52 +0200270 * \param[in] msgb message buffer
271 * \param[in] word unsigned 32bit byte to be appended
272 */
Ingo Albrecht48e17f82010-03-07 18:03:41 +0100273static inline void msgb_put_u32(struct msgb *msgb, uint32_t word)
274{
275 uint8_t *space = msgb_put(msgb, 4);
Max53777012014-06-04 19:07:41 +0200276 osmo_store32be(word, space);
Ingo Albrecht48e17f82010-03-07 18:03:41 +0100277}
Harald Weltebd598e32011-08-16 23:26:52 +0200278
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200279/*! remove data from end of message
Harald Weltebd598e32011-08-16 23:26:52 +0200280 * \param[in] msgb message buffer
281 * \param[in] len number of bytes to remove from end
282 */
Ingo Albrecht48e17f82010-03-07 18:03:41 +0100283static inline unsigned char *msgb_get(struct msgb *msgb, unsigned int len)
284{
Vadim Yanitskiy85c78312022-02-02 01:34:30 +0600285 if (OSMO_UNLIKELY(msgb_length(msgb) < len))
Harald Welte972b5022012-09-08 19:58:59 +0200286 MSGB_ABORT(msgb, "msgb too small to get %u (len %u)\n",
287 len, msgb_length(msgb));
288 msgb->tail -= len;
Ingo Albrecht48e17f82010-03-07 18:03:41 +0100289 msgb->len -= len;
Pau Espin Pedrol9fa09122018-04-11 20:09:13 +0200290 return msgb->tail;
Ingo Albrecht48e17f82010-03-07 18:03:41 +0100291}
Max53777012014-06-04 19:07:41 +0200292
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200293/*! remove uint8 from end of message
Harald Weltebd598e32011-08-16 23:26:52 +0200294 * \param[in] msgb message buffer
295 * \returns 8bit value taken from end of msgb
296 */
Ingo Albrecht48e17f82010-03-07 18:03:41 +0100297static inline uint8_t msgb_get_u8(struct msgb *msgb)
298{
299 uint8_t *space = msgb_get(msgb, 1);
300 return space[0];
301}
Max53777012014-06-04 19:07:41 +0200302
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200303/*! remove uint16 from end of message
Harald Weltebd598e32011-08-16 23:26:52 +0200304 * \param[in] msgb message buffer
305 * \returns 16bit value taken from end of msgb
306 */
Ingo Albrecht48e17f82010-03-07 18:03:41 +0100307static inline uint16_t msgb_get_u16(struct msgb *msgb)
308{
309 uint8_t *space = msgb_get(msgb, 2);
Max53777012014-06-04 19:07:41 +0200310 return osmo_load16be(space);
Ingo Albrecht48e17f82010-03-07 18:03:41 +0100311}
Max53777012014-06-04 19:07:41 +0200312
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200313/*! remove uint32 from end of message
Harald Weltebd598e32011-08-16 23:26:52 +0200314 * \param[in] msgb message buffer
315 * \returns 32bit value taken from end of msgb
316 */
Ingo Albrecht48e17f82010-03-07 18:03:41 +0100317static inline uint32_t msgb_get_u32(struct msgb *msgb)
318{
319 uint8_t *space = msgb_get(msgb, 4);
Max53777012014-06-04 19:07:41 +0200320 return osmo_load32be(space);
Ingo Albrecht48e17f82010-03-07 18:03:41 +0100321}
Harald Weltebd598e32011-08-16 23:26:52 +0200322
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200323/*! prepend (push) some data to start of message
Harald Weltebd598e32011-08-16 23:26:52 +0200324 * \param[in] msgb message buffer
325 * \param[in] len number of bytes to pre-pend
326 * \returns pointer to newly added portion at start of \a msgb
327 *
328 * This function moves the \a data pointer of the \ref msgb further
329 * to the front (by \a len bytes), thereby enlarging the message by \a
330 * len bytes.
331 *
332 * The return value is a pointer to the newly added section in the
333 * beginning of the message. It can be used to fill/copy data into it.
334 */
Harald Welteec8b4502010-02-20 20:34:29 +0100335static inline unsigned char *msgb_push(struct msgb *msgb, unsigned int len)
336{
Vadim Yanitskiy85c78312022-02-02 01:34:30 +0600337 if (OSMO_UNLIKELY(msgb_headroom(msgb) < (int) len))
Neels Hofmeyr63cb9492019-11-11 05:51:43 +0100338 MSGB_ABORT(msgb, "Not enough headroom msgb_push"
339 " (allocated %u, head at %u < want headroom %u, len %u, tailroom %u)\n",
340 msgb->data_len - sizeof(struct msgb),
341 msgb->head - msgb->_data,
342 len,
343 msgb->len,
344 msgb_tailroom(msgb));
Harald Welteec8b4502010-02-20 20:34:29 +0100345 msgb->data -= len;
346 msgb->len += len;
347 return msgb->data;
348}
Max53777012014-06-04 19:07:41 +0200349
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200350/*! prepend a uint8 value to the head of the message
Vadim Yanitskiy914a8ec2019-03-26 00:20:35 +0700351 * \param[in] msg message buffer
Harald Welte8593f1f2016-11-11 19:26:51 +0100352 * \param[in] word unsigned 8bit byte to be prepended
353 */
354static inline void msgb_push_u8(struct msgb *msg, uint8_t word)
355{
356 uint8_t *space = msgb_push(msg, 1);
357 space[0] = word;
358}
359
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200360/*! prepend a uint16 value to the head of the message
Vadim Yanitskiy914a8ec2019-03-26 00:20:35 +0700361 * \param[in] msg message buffer
Harald Welte8593f1f2016-11-11 19:26:51 +0100362 * \param[in] word unsigned 16bit byte to be prepended
363 */
364static inline void msgb_push_u16(struct msgb *msg, uint16_t word)
365{
366 uint16_t *space = (uint16_t *) msgb_push(msg, 2);
367 osmo_store16be(word, space);
368}
369
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200370/*! prepend a uint32 value to the head of the message
Vadim Yanitskiy914a8ec2019-03-26 00:20:35 +0700371 * \param[in] msg message buffer
Harald Welte8593f1f2016-11-11 19:26:51 +0100372 * \param[in] word unsigned 32bit byte to be prepended
373 */
374static inline void msgb_push_u32(struct msgb *msg, uint32_t word)
375{
376 uint32_t *space = (uint32_t *) msgb_push(msg, 4);
377 osmo_store32be(word, space);
378}
379
Maxb27e6fe2019-01-21 14:19:59 +0100380static inline unsigned char *msgb_push_tl(struct msgb *msgb, uint8_t tag)
Max84fb5bb2018-11-08 13:23:59 +0100381{
382 uint8_t *data = msgb_push(msgb, 2);
383
384 data[0] = tag;
385 data[1] = msgb->len - 2;
386 return data;
387}
388
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200389/*! remove (pull) a header from the front of the message buffer
Harald Weltebd598e32011-08-16 23:26:52 +0200390 * \param[in] msgb message buffer
391 * \param[in] len number of octets to be pulled
392 * \returns pointer to new start of msgb
393 *
394 * This function moves the \a data pointer of the \ref msgb further back
395 * in the message, thereby shrinking the size of the message by \a len
396 * bytes.
397 */
Harald Welteec8b4502010-02-20 20:34:29 +0100398static inline unsigned char *msgb_pull(struct msgb *msgb, unsigned int len)
399{
Vadim Yanitskiy85c78312022-02-02 01:34:30 +0600400 if (OSMO_UNLIKELY(msgb_length(msgb) < len))
Pau Espin Pedrolc62fc2d2018-04-11 20:14:08 +0200401 MSGB_ABORT(msgb, "msgb too small to pull %u (len %u)\n",
402 len, msgb_length(msgb));
Harald Welteec8b4502010-02-20 20:34:29 +0100403 msgb->len -= len;
404 return msgb->data += len;
405}
Harald Welteec8b4502010-02-20 20:34:29 +0100406
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200407/*! remove (pull) all headers in front of l3h from the message buffer.
Vadim Yanitskiy914a8ec2019-03-26 00:20:35 +0700408 * \param[in] msg message buffer with a valid l3h
Jacob Erlbeck8dac4152014-01-28 11:03:11 +0100409 * \returns pointer to new start of msgb (l3h)
410 *
411 * This function moves the \a data pointer of the \ref msgb further back
412 * in the message, thereby shrinking the size of the message.
413 * l1h and l2h will be cleared.
414 */
415static inline unsigned char *msgb_pull_to_l3(struct msgb *msg)
416{
417 unsigned char *ret = msgb_pull(msg, msg->l3h - msg->data);
418 msg->l1h = msg->l2h = NULL;
419 return ret;
420}
421
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200422/*! remove (pull) all headers in front of l2h from the message buffer.
Vadim Yanitskiy914a8ec2019-03-26 00:20:35 +0700423 * \param[in] msg message buffer with a valid l2h
Harald Welte59a9ebf2017-04-15 19:04:34 +0200424 * \returns pointer to new start of msgb (l2h)
425 *
426 * This function moves the \a data pointer of the \ref msgb further back
427 * in the message, thereby shrinking the size of the message.
428 * l1h will be cleared.
429 */
430static inline unsigned char *msgb_pull_to_l2(struct msgb *msg)
431{
432 unsigned char *ret = msgb_pull(msg, msg->l2h - msg->data);
433 msg->l1h = NULL;
434 return ret;
435}
436
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200437/*! remove uint8 from front of message
Harald Welte972b5022012-09-08 19:58:59 +0200438 * \param[in] msgb message buffer
439 * \returns 8bit value taken from end of msgb
440 */
441static inline uint8_t msgb_pull_u8(struct msgb *msgb)
442{
Steve Markgraf5905d5b2012-11-14 19:36:00 +0100443 uint8_t *space = msgb_pull(msgb, 1) - 1;
Harald Welte972b5022012-09-08 19:58:59 +0200444 return space[0];
445}
Max53777012014-06-04 19:07:41 +0200446
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200447/*! remove uint16 from front of message
Harald Welte972b5022012-09-08 19:58:59 +0200448 * \param[in] msgb message buffer
449 * \returns 16bit value taken from end of msgb
450 */
451static inline uint16_t msgb_pull_u16(struct msgb *msgb)
452{
Steve Markgraf5905d5b2012-11-14 19:36:00 +0100453 uint8_t *space = msgb_pull(msgb, 2) - 2;
Max53777012014-06-04 19:07:41 +0200454 return osmo_load16be(space);
Harald Welte972b5022012-09-08 19:58:59 +0200455}
Max53777012014-06-04 19:07:41 +0200456
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200457/*! remove uint32 from front of message
Harald Welte972b5022012-09-08 19:58:59 +0200458 * \param[in] msgb message buffer
459 * \returns 32bit value taken from end of msgb
460 */
461static inline uint32_t msgb_pull_u32(struct msgb *msgb)
462{
Steve Markgraf5905d5b2012-11-14 19:36:00 +0100463 uint8_t *space = msgb_pull(msgb, 4) - 4;
Max53777012014-06-04 19:07:41 +0200464 return osmo_load32be(space);
Harald Welte972b5022012-09-08 19:58:59 +0200465}
466
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200467/*! Increase headroom of empty msgb, reducing the tailroom
Harald Weltebd598e32011-08-16 23:26:52 +0200468 * \param[in] msg message buffer
469 * \param[in] len amount of extra octets to be reserved as headroom
470 *
471 * This function reserves some memory at the beginning of the underlying
472 * data buffer. The idea is to reserve space in case further headers
473 * have to be pushed to the \ref msgb during further processing.
474 *
475 * Calling this function leads to undefined reusults if it is called on
476 * a non-empty \ref msgb.
477 */
Harald Welteec8b4502010-02-20 20:34:29 +0100478static inline void msgb_reserve(struct msgb *msg, int len)
479{
480 msg->data += len;
481 msg->tail += len;
482}
483
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200484/*! Trim the msgb to a given absolute length
Harald Welteb0f91512012-01-14 21:53:41 +0100485 * \param[in] msg message buffer
486 * \param[in] len new total length of buffer
487 * \returns 0 in case of success, negative in case of error
488 */
489static inline int msgb_trim(struct msgb *msg, int len)
490{
Vadim Yanitskiy85c78312022-02-02 01:34:30 +0600491 if (OSMO_UNLIKELY(len < 0))
Jacob Erlbeckff42b262015-11-27 13:26:19 +0100492 MSGB_ABORT(msg, "Negative length is not allowed\n");
Vadim Yanitskiy85c78312022-02-02 01:34:30 +0600493 if (OSMO_UNLIKELY(len > msg->data_len))
Harald Welteb0f91512012-01-14 21:53:41 +0100494 return -1;
495
Harald Welte3068d572012-01-14 22:07:59 +0100496 msg->len = len;
497 msg->tail = msg->data + len;
Harald Welteb0f91512012-01-14 21:53:41 +0100498
499 return 0;
500}
501
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200502/*! Trim the msgb to a given layer3 length
Katerina Barone-Adesic28c6a02013-02-15 13:27:59 +0100503 * \param[in] msg message buffer
Harald Welteb0f91512012-01-14 21:53:41 +0100504 * \param[in] l3len new layer3 length
505 * \returns 0 in case of success, negative in case of error
506 */
507static inline int msgb_l3trim(struct msgb *msg, int l3len)
508{
509 return msgb_trim(msg, (msg->l3h - msg->data) + l3len);
510}
511
Harald Welte179f3572019-03-18 18:38:47 +0100512/*! Allocate message buffer with specified headroom from specified talloc context.
513 * \param[in] ctx talloc context from which to allocate
514 * \param[in] size size in bytes, including headroom
515 * \param[in] headroom headroom in bytes
516 * \param[in] name human-readable name
517 * \returns allocated message buffer with specified headroom
518 *
519 * This function is a convenience wrapper around \ref msgb_alloc
520 * followed by \ref msgb_reserve in order to create a new \ref msgb with
521 * user-specified amount of headroom.
522 */
Pau Espin Pedrolaf40b0b2021-06-10 14:33:46 +0200523static inline struct msgb *msgb_alloc_headroom_c(const void *ctx, uint16_t size, uint16_t headroom,
Harald Welte179f3572019-03-18 18:38:47 +0100524 const char *name)
525{
Pau Espin Pedrolec4b6272019-09-26 17:46:27 +0200526 osmo_static_assert(size >= headroom, headroom_bigger);
Harald Welte179f3572019-03-18 18:38:47 +0100527
528 struct msgb *msg = msgb_alloc_c(ctx, size, name);
Vadim Yanitskiy85c78312022-02-02 01:34:30 +0600529 if (OSMO_LIKELY(msg))
Harald Welte179f3572019-03-18 18:38:47 +0100530 msgb_reserve(msg, headroom);
531 return msg;
532}
533
534
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200535/*! Allocate message buffer with specified headroom
Harald Weltebd598e32011-08-16 23:26:52 +0200536 * \param[in] size size in bytes, including headroom
537 * \param[in] headroom headroom in bytes
538 * \param[in] name human-readable name
539 * \returns allocated message buffer with specified headroom
540 *
541 * This function is a convenience wrapper around \ref msgb_alloc
542 * followed by \ref msgb_reserve in order to create a new \ref msgb with
543 * user-specified amount of headroom.
544 */
Pau Espin Pedrolaf40b0b2021-06-10 14:33:46 +0200545static inline struct msgb *msgb_alloc_headroom(uint16_t size, uint16_t headroom,
Harald Welteec8b4502010-02-20 20:34:29 +0100546 const char *name)
547{
Pau Espin Pedrolec4b6272019-09-26 17:46:27 +0200548 osmo_static_assert(size >= headroom, headroom_bigger);
Holger Hans Peter Freyther5853f082011-01-16 17:38:22 +0100549
Harald Welteec8b4502010-02-20 20:34:29 +0100550 struct msgb *msg = msgb_alloc(size, name);
Vadim Yanitskiy85c78312022-02-02 01:34:30 +0600551 if (OSMO_LIKELY(msg))
Harald Welteec8b4502010-02-20 20:34:29 +0100552 msgb_reserve(msg, headroom);
553 return msg;
554}
555
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200556/*! Check a message buffer for consistency
Jacob Erlbeck8db11342015-11-27 13:26:15 +0100557 * \param[in] msg message buffer
558 * \returns 0 (false) if inconsistent, != 0 (true) otherwise
559 */
560static inline int msgb_test_invariant(const struct msgb *msg)
561{
562 const unsigned char *lbound;
563 if (!msg || !msg->data || !msg->tail ||
564 (msg->data + msg->len != msg->tail) ||
565 (msg->data < msg->head) ||
566 (msg->tail > msg->head + msg->data_len))
567 return 0;
568
569 lbound = msg->head;
570
571 if (msg->l1h) {
572 if (msg->l1h < lbound)
573 return 0;
574 lbound = msg->l1h;
575 }
576 if (msg->l2h) {
577 if (msg->l2h < lbound)
578 return 0;
579 lbound = msg->l2h;
580 }
581 if (msg->l3h) {
582 if (msg->l3h < lbound)
583 return 0;
584 lbound = msg->l3h;
585 }
586 if (msg->l4h) {
587 if (msg->l4h < lbound)
588 return 0;
589 lbound = msg->l4h;
590 }
591
592 return lbound <= msg->head + msg->data_len;
593}
594
Maxdb038252018-12-03 14:14:44 +0100595
596/* msgb data comparison helpers */
597
598/*! Compare: check data in msgb against given data
599 * \param[in] msg message buffer
600 * \param[in] data expected data
601 * \param[in] len length of data
602 * \returns boolean indicating whether msgb content is equal to the given data
603 */
604#define msgb_eq_data(msg, data, len) \
605 _msgb_eq(__FILE__, __LINE__, __func__, 0, msg, data, len, false)
606
607/*! Compare: check L1 data in msgb against given data
608 * \param[in] msg message buffer
609 * \param[in] data expected L1 data
610 * \param[in] len length of data
611 * \returns boolean indicating whether msgb L1 content is equal to the given data
612 */
613#define msgb_eq_l1_data(msg, data, len) \
614 _msgb_eq(__FILE__, __LINE__, __func__, 1, msg, data, len, false)
615
616/*! Compare: check L2 data in msgb against given data
617 * \param[in] msg message buffer
618 * \param[in] data expected L2 data
619 * \param[in] len length of data
620 * \returns boolean indicating whether msgb L2 content is equal to the given data
621 */
622#define msgb_eq_l2_data(msg, data, len) \
623 _msgb_eq(__FILE__, __LINE__, __func__, 2, msg, data, len, false)
624
625/*! Compare: check L3 data in msgb against given data
626 * \param[in] msg message buffer
627 * \param[in] data expected L3 data
628 * \param[in] len length of data
629 * \returns boolean indicating whether msgb L3 content is equal to the given data
630 */
631#define msgb_eq_l3_data(msg, data, len) \
632 _msgb_eq(__FILE__, __LINE__, __func__, 3, msg, data, len, false)
633
634/*! Compare: check L4 data in msgb against given data
635 * \param[in] msg message buffer
636 * \param[in] data expected L4 data
637 * \param[in] len length of data
638 * \returns boolean indicating whether msgb L4 content is equal to the given data
639 */
640#define msgb_eq_l4_data(msg, data, len) \
641 _msgb_eq(__FILE__, __LINE__, __func__, 4, msg, data, len, false)
642
643
644/* msgb test/debug helpers */
645
646/*! Compare and print: check data in msgb against given data and print errors if any
647 * \param[in] msg message buffer
648 * \param[in] data expected data
649 * \param[in] len length of data
650 * \returns boolean indicating whether msgb content is equal to the given data
651 */
652#define msgb_eq_data_print(msg, data, len) \
653 _msgb_eq(__FILE__, __LINE__, __func__, 0, msg, data, len, true)
654
655/*! Compare and print: check L1 data in msgb against given data and print errors if any
656 * \param[in] msg message buffer
657 * \param[in] data expected L1 data
658 * \param[in] len length of data
659 * \returns boolean indicating whether msgb L1 content is equal to the given data
660 */
661#define msgb_eq_l1_data_print(msg, data, len) \
662 _msgb_eq(__FILE__, __LINE__, __func__, 1, msg, data, len, true)
663
664/*! Compare and print: check L2 data in msgb against given data and print errors if any
665 * \param[in] msg message buffer
666 * \param[in] data expected L2 data
667 * \param[in] len length of data
668 * \returns boolean indicating whether msgb L2 content is equal to the given data
669 */
670#define msgb_eq_l2_data_print(msg, data, len) \
671 _msgb_eq(__FILE__, __LINE__, __func__, 2, msg, data, len, true)
672
673/*! Compare and print: check L3 data in msgb against given data and print errors if any
674 * \param[in] msg message buffer
675 * \param[in] data expected L3 data
676 * \param[in] len length of data
677 * \returns boolean indicating whether msgb L3 content is equal to the given data
678 */
679#define msgb_eq_l3_data_print(msg, data, len) \
680 _msgb_eq(__FILE__, __LINE__, __func__, 3, msg, data, len, true)
681
682
683/*! Compare and print: check L4 data in msgb against given data and print errors if any
684 * \param[in] msg message buffer
685 * \param[in] data expected L4 data
686 * \param[in] len length of data
687 * \returns boolean indicating whether msgb L4 content is equal to the given data
688 */
689#define msgb_eq_l4_data_print(msg, data, len) \
690 _msgb_eq(__FILE__, __LINE__, __func__, 4, msg, data, len, true)
691
692bool _msgb_eq(const char *file, size_t line, const char *func, uint8_t level,
693 const struct msgb *msg, const uint8_t *data, size_t len, bool print);
694
695
696/* msgb data comparison */
697
698/*! Compare msgbs
699 * \param[in] msg1 message buffer
700 * \param[in] msg2 reference message buffer
701 * \returns boolean indicating whether msgb content is equal
702 */
Vadim Yanitskiyba426e32019-03-25 23:41:07 +0700703#define msgb_eq(msg1, msg2) msgb_eq_data(msg1, msgb_data(msg2), msgb_length(msg2))
Maxdb038252018-12-03 14:14:44 +0100704
705/*! Compare msgbs L1 content
706 * \param[in] msg1 message buffer
707 * \param[in] msg2 reference message buffer
708 * \returns boolean indicating whether msgb L1 content is equal
709 */
Vadim Yanitskiyba426e32019-03-25 23:41:07 +0700710#define msgb_eq_l1(msg1, msg2) msgb_eq_l1_data(msg1, msgb_l1(msg2), msgb_l1len(msg2))
Maxdb038252018-12-03 14:14:44 +0100711
712/*! Compare msgbs L2 content
713 * \param[in] msg1 message buffer
714 * \param[in] msg2 reference message buffer
715 * \returns boolean indicating whether msgb L2 content is equal
716 */
Vadim Yanitskiyba426e32019-03-25 23:41:07 +0700717#define msgb_eq_l2(msg1, msg2) msgb_eq_l2_data(msg1, msgb_l2(msg2), msgb_l2len(msg2))
Maxdb038252018-12-03 14:14:44 +0100718
719/*! Compare msgbs L3 content
720 * \param[in] msg1 message buffer
721 * \param[in] msg2 reference message buffer
722 * \returns boolean indicating whether msgb L3 content is equal
723 */
Vadim Yanitskiyba426e32019-03-25 23:41:07 +0700724#define msgb_eq_l3(msg1, msg2) msgb_eq_l3_data(msg1, msgb_l3(msg2), msgb_l3len(msg2))
Maxdb038252018-12-03 14:14:44 +0100725
726/*! Compare msgbs L4 content
727 * \param[in] msg1 message buffer
728 * \param[in] msg2 reference message buffer
729 * \returns boolean indicating whether msgb L4 content is equal
730 */
Vadim Yanitskiyba426e32019-03-25 23:41:07 +0700731#define msgb_eq_l4(msg1, msg2) msgb_eq_l4_data(msg1, msgb_l4(msg2), msgb_l4len(msg2))
Maxdb038252018-12-03 14:14:44 +0100732
733
Holger Hans Peter Freytheracffb602010-10-18 18:22:31 +0200734/* non inline functions to ease binding */
Harald Weltebd598e32011-08-16 23:26:52 +0200735
Holger Hans Peter Freytheracffb602010-10-18 18:22:31 +0200736uint8_t *msgb_data(const struct msgb *msg);
Neels Hofmeyrf45334b2016-09-16 00:15:56 +0200737
738void *msgb_talloc_ctx_init(void *root_ctx, unsigned int pool_size);
739void msgb_set_talloc_ctx(void *ctx) OSMO_DEPRECATED("Use msgb_talloc_ctx_init() instead");
Philipp Maierc5b47cc2017-10-10 16:53:21 +0200740int msgb_printf(struct msgb *msgb, const char *format, ...);
Holger Hans Peter Freytheracffb602010-10-18 18:22:31 +0200741
Maxcb02a392018-12-03 14:00:49 +0100742static inline const char *msgb_hexdump_l1(const struct msgb *msg)
743{
744 if (!msgb_l1(msg) || !(msgb_l1len(msg)))
745 return "[]";
746 return osmo_hexdump((const unsigned char *) msgb_l1(msg), msgb_l1len(msg));
747}
748
Harald Welte57217702018-02-09 02:01:48 +0100749static inline const char *msgb_hexdump_l2(const struct msgb *msg)
750{
751 if (!msgb_l2(msg) || !(msgb_l2len(msg)))
752 return "[]";
Harald Welteebd2b0f2018-02-09 10:32:03 +0100753 return osmo_hexdump((const unsigned char *) msgb_l2(msg), msgb_l2len(msg));
Harald Welte57217702018-02-09 02:01:48 +0100754}
755
756static inline const char *msgb_hexdump_l3(const struct msgb *msg)
757{
758 if (!msgb_l3(msg) || !(msgb_l3len(msg)))
759 return "[]";
Harald Welteebd2b0f2018-02-09 10:32:03 +0100760 return osmo_hexdump((const unsigned char*) msgb_l3(msg), msgb_l3len(msg));
Harald Welte57217702018-02-09 02:01:48 +0100761}
762
Maxcb02a392018-12-03 14:00:49 +0100763static inline const char *msgb_hexdump_l4(const struct msgb *msg)
764{
765 if (!msgb_l4(msg) || !(msgb_l4len(msg)))
766 return "[]";
767 return osmo_hexdump((const unsigned char*) msgb_l4(msg), msgb_l4len(msg));
768}
769
Sylvain Munautdca7d2c2012-04-18 21:53:23 +0200770/*! @} */